src/webui/webui.cpp

   1 /*
   2  * Bittorrent Client using Qt and libtorrent.
   3  * Copyright (C) 2015  Vladimir Golovnev <glassez@yandex.ru>
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License
   7  * as published by the Free Software Foundation; either version 2
   8  * of the License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  18  *
  19  * In addition, as a special exception, the copyright holders give permission to
  20  * link this program with the OpenSSL project's "OpenSSL" library (or with
  21  * modified versions of it that use the same license as the "OpenSSL" library),
  22  * and distribute the linked executables. You must obey the GNU General Public
  23  * License in all respects for all of the code used other than "OpenSSL".  If you
  24  * modify file(s), you may extend this exception to your version of the file(s),
  25  * but you are not obligated to do so. If you do not wish to do so, delete this
  26  * exception statement from your version.
  27  */
  28
  29 #include "webui.h"
  30
  31 #include <QFile>
  32
  33 #include "base/http/server.h"
  34 #include "base/logger.h"
  35 #include "base/net/dnsupdater.h"
  36 #include "base/net/portforwarder.h"
  37 #include "base/preferences.h"
  38 #include "base/utils/net.h"
  39 #include "webapplication.h"
  40
  41 WebUI::WebUI()
  42     : m_isErrored(false)
  43     , m_port(0)
  44 {
  45     configure();
  46     connect(Preferences::instance(), &Preferences::changed, this, &WebUI::configure);
  47 }
  48
  49 void WebUI::configure()
  50 {
  51     m_isErrored = false; // clear previous error state
  52
  53     Logger *const logger = Logger::instance();
  54     Preferences *const pref = Preferences::instance();
  55
  56     const quint16 oldPort = m_port;
  57     m_port = pref->getWebUiPort();
  58
  59     if (pref->isWebUiEnabled())
  60     {
  61         // UPnP/NAT-PMP
  62         if (pref->useUPnPForWebUIPort())
  63         {
  64             if (m_port != oldPort)
  65             {
  66                 Net::PortForwarder::instance()->deletePort(oldPort);
  67                 Net::PortForwarder::instance()->addPort(m_port);
  68             }
  69         }
  70         else
  71         {
  72             Net::PortForwarder::instance()->deletePort(oldPort);
  73         }
  74
  75         // http server
  76         const QString serverAddressString = pref->getWebUiAddress();
  77         if (!m_httpServer)
  78         {
  79             m_webapp = new WebApplication(this);
  80             m_httpServer = new Http::Server(m_webapp, this);
  81         }
  82         else
  83         {
  84             if ((m_httpServer->serverAddress().toString() != serverAddressString)
  85                     || (m_httpServer->serverPort() != m_port))
  86                 m_httpServer->close();
  87         }
  88
  89         if (pref->isWebUiHttpsEnabled())
  90         {
  91             const auto readData = [](const QString &path) -> QByteArray
  92             {
  93                 QFile file(path);
  94                 if (!file.open(QIODevice::ReadOnly))
  95                     return {};
  96                 return file.read(Utils::Net::MAX_SSL_FILE_SIZE);
  97             };
  98             const QByteArray cert = readData(pref->getWebUIHttpsCertificatePath());
  99             const QByteArray key = readData(pref->getWebUIHttpsKeyPath());
 100
 101             const bool success = m_httpServer->setupHttps(cert, key);
 102             if (success)
 103                 logger->addMessage(tr("Web UI: HTTPS setup successful"));
 104             else
 105                 logger->addMessage(tr("Web UI: HTTPS setup failed, fallback to HTTP"), Log::CRITICAL);
 106         }
 107         else
 108         {
 109             m_httpServer->disableHttps();
 110         }
 111
 112         if (!m_httpServer->isListening())
 113         {
 114             const auto address = (serverAddressString == "*" || serverAddressString.isEmpty())
 115                 ? QHostAddress::Any : QHostAddress(serverAddressString);
 116             bool success = m_httpServer->listen(address, m_port);
 117             if (success)
 118             {
 119                 logger->addMessage(tr("Web UI: Now listening on IP: %1, port: %2").arg(serverAddressString).arg(m_port));
 120             }
 121             else
 122             {
 123                 const QString errorMsg = tr("Web UI: Unable to bind to IP: %1, port: %2. Reason: %3")
 124                     .arg(serverAddressString).arg(m_port).arg(m_httpServer->errorString());
 125                 logger->addMessage(errorMsg, Log::CRITICAL);
 126                 qCritical() << errorMsg;
 127
 128                 m_isErrored = true;
 129                 emit fatalError();
 130             }
 131         }
 132
 133         // DynDNS
 134         if (pref->isDynDNSEnabled())
 135         {
 136             if (!m_dnsUpdater)
 137                 m_dnsUpdater = new Net::DNSUpdater(this);
 138             else
 139                 m_dnsUpdater->updateCredentials();
 140         }
 141         else
 142         {
 143             delete m_dnsUpdater;
 144         }
 145     }
 146     else
 147     {
 148         Net::PortForwarder::instance()->deletePort(oldPort);
 149
 150         delete m_httpServer;
 151         delete m_webapp;
 152         delete m_dnsUpdater;
 153     }
 154 }
 155
 156 bool WebUI::isErrored() const
 157 {
 158     return m_isErrored;
 159 }