Vladimir Golovnev (Glassez)
10 years ago
18 changed files with 1533 additions and 992 deletions
@ -0,0 +1,174 @@
@@ -0,0 +1,174 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2014 Vladimir Golovnev <glassez@yandex.ru> |
||||
* Copyright (C) 2012, Christophe Dumez <chris@qbittorrent.org> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#include <QDebug> |
||||
#include <QDir> |
||||
#include <QTemporaryFile> |
||||
#include <QNetworkCookie> |
||||
#include "preferences.h" |
||||
#include "webapplication.h" |
||||
#include "abstractrequesthandler.h" |
||||
|
||||
AbstractRequestHandler::AbstractRequestHandler(const HttpRequest &request, const HttpEnvironment &env, WebApplication *app) |
||||
: app_(app), session_(0), request_(request), env_(env) |
||||
{ |
||||
if (isBanned()) |
||||
{ |
||||
status(403, "Forbidden"); |
||||
print(QObject::tr("Your IP address has been banned after too many failed authentication attempts.")); |
||||
return; |
||||
} |
||||
|
||||
sessionInitialize(); |
||||
|
||||
if (!sessionActive() && !isAuthNeeded()) sessionStart(); |
||||
} |
||||
|
||||
HttpResponse AbstractRequestHandler::run() |
||||
{ |
||||
response_ = HttpResponse(); |
||||
processRequest(); |
||||
return response_; |
||||
} |
||||
|
||||
bool AbstractRequestHandler::isAuthNeeded() |
||||
{ |
||||
return |
||||
( |
||||
env_.clientAddress != QHostAddress::LocalHost && |
||||
env_.clientAddress != QHostAddress::LocalHostIPv6 |
||||
) || Preferences::instance()->isWebUiLocalAuthEnabled(); |
||||
} |
||||
|
||||
void AbstractRequestHandler::status(uint code, const QString& text) |
||||
{ |
||||
response_.status = HttpResponseStatus(code, text); |
||||
} |
||||
|
||||
void AbstractRequestHandler::header(const QString& name, const QString& value) |
||||
{ |
||||
response_.headers[name] = value; |
||||
} |
||||
|
||||
void AbstractRequestHandler::print(const QString& text, const QString& type) |
||||
{ |
||||
print_impl(text.toUtf8(), type); |
||||
} |
||||
|
||||
void AbstractRequestHandler::print(const QByteArray& data, const QString& type) |
||||
{ |
||||
print_impl(data, type); |
||||
} |
||||
|
||||
void AbstractRequestHandler::print_impl(const QByteArray& data, const QString& type) |
||||
{ |
||||
if (!response_.headers.contains(HEADER_CONTENT_TYPE)) |
||||
response_.headers[HEADER_CONTENT_TYPE] = type; |
||||
|
||||
response_.content += data; |
||||
} |
||||
|
||||
void AbstractRequestHandler::printFile(const QString& path) |
||||
{ |
||||
QByteArray data; |
||||
QString type; |
||||
|
||||
if (!app_->readFile(path, data, type)) |
||||
{ |
||||
status(404, "Not Found"); |
||||
return; |
||||
} |
||||
|
||||
print(data, type); |
||||
} |
||||
|
||||
void AbstractRequestHandler::sessionInitialize() |
||||
{ |
||||
app_->sessionInitialize(this); |
||||
} |
||||
|
||||
void AbstractRequestHandler::sessionStart() |
||||
{ |
||||
if (app_->sessionStart(this)) |
||||
{ |
||||
QNetworkCookie cookie(C_SID.toUtf8(), session_->id.toUtf8()); |
||||
cookie.setPath("/"); |
||||
header(HEADER_SET_COOKIE, cookie.toRawForm()); |
||||
} |
||||
} |
||||
|
||||
void AbstractRequestHandler::sessionEnd() |
||||
{ |
||||
if (sessionActive()) |
||||
{ |
||||
QNetworkCookie cookie(C_SID.toUtf8(), session_->id.toUtf8()); |
||||
cookie.setPath("/"); |
||||
cookie.setExpirationDate(QDateTime::currentDateTime()); |
||||
|
||||
if (app_->sessionEnd(this)) |
||||
{ |
||||
header(HEADER_SET_COOKIE, cookie.toRawForm()); |
||||
} |
||||
} |
||||
} |
||||
|
||||
bool AbstractRequestHandler::isBanned() const |
||||
{ |
||||
return app_->isBanned(this); |
||||
} |
||||
|
||||
int AbstractRequestHandler::failedAttempts() const |
||||
{ |
||||
return app_->failedAttempts(this); |
||||
} |
||||
|
||||
void AbstractRequestHandler::resetFailedAttempts() |
||||
{ |
||||
app_->resetFailedAttempts(this); |
||||
} |
||||
|
||||
void AbstractRequestHandler::increaseFailedAttempts() |
||||
{ |
||||
app_->increaseFailedAttempts(this); |
||||
} |
||||
|
||||
QString AbstractRequestHandler::saveTmpFile(const QByteArray &data) |
||||
{ |
||||
QTemporaryFile tmpfile(QDir::temp().absoluteFilePath("qBT-XXXXXX.torrent")); |
||||
tmpfile.setAutoRemove(false); |
||||
if (tmpfile.open()) |
||||
{ |
||||
tmpfile.write(data); |
||||
tmpfile.close(); |
||||
return tmpfile.fileName(); |
||||
} |
||||
|
||||
qWarning() << "I/O Error: Could not create temporary file"; |
||||
return QString(); |
||||
} |
@ -0,0 +1,89 @@
@@ -0,0 +1,89 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2014 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#ifndef ABSTRACTREQUESTHANDLER_H |
||||
#define ABSTRACTREQUESTHANDLER_H |
||||
|
||||
#include <QString> |
||||
#include "httptypes.h" |
||||
|
||||
class WebApplication; |
||||
struct WebSession; |
||||
|
||||
class AbstractRequestHandler |
||||
{ |
||||
friend class WebApplication; |
||||
|
||||
public: |
||||
AbstractRequestHandler( |
||||
const HttpRequest& request, const HttpEnvironment& env, |
||||
WebApplication* app); |
||||
|
||||
HttpResponse run(); |
||||
|
||||
protected: |
||||
virtual void processRequest() = 0; |
||||
|
||||
void status(uint code, const QString& text); |
||||
void header(const QString& name, const QString& value); |
||||
void print(const QString& text, const QString& type = CONTENT_TYPE_HTML); |
||||
void print(const QByteArray& data, const QString& type = CONTENT_TYPE_HTML); |
||||
void printFile(const QString& path); |
||||
|
||||
// Session management
|
||||
bool sessionActive() const { return session_ != 0; } |
||||
void sessionInitialize(); |
||||
void sessionStart(); |
||||
void sessionEnd(); |
||||
|
||||
// Ban management
|
||||
bool isBanned() const; |
||||
int failedAttempts() const; |
||||
void resetFailedAttempts(); |
||||
void increaseFailedAttempts(); |
||||
|
||||
bool isAuthNeeded(); |
||||
|
||||
// save data to temporary file on disk and return its name (or empty string if fails)
|
||||
static QString saveTmpFile(const QByteArray& data); |
||||
|
||||
inline WebSession* session() { return session_; } |
||||
inline HttpRequest request() const { return request_; } |
||||
inline HttpEnvironment env() const { return env_; } |
||||
|
||||
private: |
||||
WebApplication* app_; |
||||
WebSession* session_; |
||||
const HttpRequest request_; |
||||
const HttpEnvironment env_; |
||||
HttpResponse response_; |
||||
|
||||
void print_impl(const QByteArray& data, const QString& type); |
||||
}; |
||||
|
||||
#endif // ABSTRACTREQUESTHANDLER_H
|
@ -0,0 +1,75 @@
@@ -0,0 +1,75 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2014 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#ifndef EXTRA_TRANSLATIONS_H |
||||
#define EXTRA_TRANSLATIONS_H |
||||
|
||||
#include <QObject> |
||||
|
||||
// Additional translations for Web UI
|
||||
static const char *__TRANSLATIONS__[] = { |
||||
QT_TRANSLATE_NOOP("HttpServer", "File"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Edit"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Help"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Download Torrents from their URL or Magnet link"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Only one link per line"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Download local torrent"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Torrent files were correctly added to download list."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Point to torrent file"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Download"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Are you sure you want to delete the selected torrents from the transfer list and hard disk?"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Download rate limit must be greater than 0 or disabled."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Upload rate limit must be greater than 0 or disabled."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Maximum number of connections limit must be greater than 0 or disabled."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Maximum number of connections per torrent limit must be greater than 0 or disabled."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Maximum number of upload slots per torrent limit must be greater than 0 or disabled."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Unable to save program preferences, qBittorrent is probably unreachable."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Language"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "The port used for incoming connections must be greater than 1024 and less than 65535."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "The port used for the Web UI must be greater than 1024 and less than 65535."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "The Web UI username must be at least 3 characters long."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "The Web UI password must be at least 3 characters long."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Save"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "qBittorrent client is not reachable"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "HTTP Server"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "The following parameters are supported:"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Torrent path"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Torrent name"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "qBittorrent has been shutdown."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Unable to log in, qBittorrent is probably unreachable."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Invalid Username or Password."), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Password"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "Login"), |
||||
QT_TRANSLATE_NOOP("HttpServer", "qBittorrent web User Interface") |
||||
}; |
||||
|
||||
static const struct { const char *source; const char *comment; } __COMMENTED_TRANSLATIONS__[] = { |
||||
QT_TRANSLATE_NOOP3("HttpServer", "Downloaded", "Is the file downloaded or not?") |
||||
}; |
||||
|
||||
#endif // EXTRA_TRANSLATIONS_H
|
@ -0,0 +1,588 @@
@@ -0,0 +1,588 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2014 Vladimir Golovnev <glassez@yandex.ru> |
||||
* Copyright (C) 2012, Christophe Dumez <chris@qbittorrent.org> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#include <QDebug> |
||||
#ifdef DISABLE_GUI |
||||
#include <QCoreApplication> |
||||
#else |
||||
#include <QApplication> |
||||
#endif |
||||
#include <QTimer> |
||||
#include <QCryptographicHash> |
||||
#include <queue> |
||||
#include <vector> |
||||
#include <libtorrent/session.hpp> |
||||
#ifndef DISABLE_GUI |
||||
#include "iconprovider.h" |
||||
#endif |
||||
#include "misc.h" |
||||
#include "fs_utils.h" |
||||
#include "preferences.h" |
||||
#include "btjson.h" |
||||
#include "prefjson.h" |
||||
#include "qbtsession.h" |
||||
#include "requesthandler.h" |
||||
|
||||
using namespace libtorrent; |
||||
|
||||
const QString WWW_FOLDER = ":/www/public/"; |
||||
const QString PRIVATE_FOLDER = ":/www/private/"; |
||||
const QString DEFAULT_SCOPE = "public"; |
||||
const QString SCOPE_IMAGES = "images"; |
||||
const QString SCOPE_THEME = "theme"; |
||||
const QString DEFAULT_ACTION = "index"; |
||||
const QString WEBUI_ACTION = "webui"; |
||||
|
||||
#define ADD_ACTION(scope, action) actions[#scope][#action] = &RequestHandler::action_##scope##_##action |
||||
|
||||
QMap<QString, QMap<QString, RequestHandler::Action> > RequestHandler::initializeActions() |
||||
{ |
||||
QMap<QString,QMap<QString, RequestHandler::Action> > actions; |
||||
|
||||
ADD_ACTION(public, webui); |
||||
ADD_ACTION(public, index); |
||||
ADD_ACTION(public, login); |
||||
ADD_ACTION(public, logout); |
||||
ADD_ACTION(public, theme); |
||||
ADD_ACTION(public, images); |
||||
ADD_ACTION(json, torrents); |
||||
ADD_ACTION(json, preferences); |
||||
ADD_ACTION(json, transferInfo); |
||||
ADD_ACTION(json, propertiesGeneral); |
||||
ADD_ACTION(json, propertiesTrackers); |
||||
ADD_ACTION(json, propertiesFiles); |
||||
ADD_ACTION(command, shutdown); |
||||
ADD_ACTION(command, download); |
||||
ADD_ACTION(command, upload); |
||||
ADD_ACTION(command, addTrackers); |
||||
ADD_ACTION(command, resumeAll); |
||||
ADD_ACTION(command, pauseAll); |
||||
ADD_ACTION(command, resume); |
||||
ADD_ACTION(command, pause); |
||||
ADD_ACTION(command, setPreferences); |
||||
ADD_ACTION(command, setFilePrio); |
||||
ADD_ACTION(command, getGlobalUpLimit); |
||||
ADD_ACTION(command, getGlobalDlLimit); |
||||
ADD_ACTION(command, setGlobalUpLimit); |
||||
ADD_ACTION(command, setGlobalDlLimit); |
||||
ADD_ACTION(command, getTorrentUpLimit); |
||||
ADD_ACTION(command, getTorrentDlLimit); |
||||
ADD_ACTION(command, setTorrentUpLimit); |
||||
ADD_ACTION(command, setTorrentDlLimit); |
||||
ADD_ACTION(command, delete); |
||||
ADD_ACTION(command, deletePerm); |
||||
ADD_ACTION(command, increasePrio); |
||||
ADD_ACTION(command, decreasePrio); |
||||
ADD_ACTION(command, topPrio); |
||||
ADD_ACTION(command, bottomPrio); |
||||
ADD_ACTION(command, recheck); |
||||
|
||||
return actions; |
||||
} |
||||
|
||||
void RequestHandler::action_public_index() |
||||
{ |
||||
QString path; |
||||
if (!args_.isEmpty()) |
||||
{ |
||||
if (args_.back() == "favicon.ico") |
||||
path = ":/Icons/skin/qbittorrent16.png"; |
||||
else |
||||
path = WWW_FOLDER + args_.join("/"); |
||||
} |
||||
|
||||
printFile(path); |
||||
} |
||||
|
||||
void RequestHandler::action_public_webui() |
||||
{ |
||||
if (!sessionActive()) |
||||
printFile(PRIVATE_FOLDER + "login.html"); |
||||
else |
||||
printFile(PRIVATE_FOLDER + "index.html"); |
||||
} |
||||
|
||||
void RequestHandler::action_public_login() |
||||
{ |
||||
const Preferences* const pref = Preferences::instance(); |
||||
QCryptographicHash md5(QCryptographicHash::Md5); |
||||
|
||||
md5.addData(request().posts["password"].toLocal8Bit()); |
||||
QString pass = md5.result().toHex(); |
||||
|
||||
if ((request().posts["username"] == pref->getWebUiUsername()) && (pass == pref->getWebUiPassword())) |
||||
{ |
||||
sessionStart(); |
||||
print(QByteArray("Ok."), CONTENT_TYPE_TXT); |
||||
} |
||||
else |
||||
{ |
||||
QString addr = env().clientAddress.toString(); |
||||
increaseFailedAttempts(); |
||||
qDebug("client IP: %s (%d failed attempts)", qPrintable(addr), failedAttempts()); |
||||
print(QByteArray("Fails."), CONTENT_TYPE_TXT); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_public_logout() |
||||
{ |
||||
sessionEnd(); |
||||
} |
||||
|
||||
void RequestHandler::action_public_theme() |
||||
{ |
||||
if (args_.size() != 1) |
||||
{ |
||||
status(404, "Not Found"); |
||||
return; |
||||
} |
||||
|
||||
#ifdef DISABLE_GUI |
||||
QString url = ":/Icons/oxygen/" + args_.front() + ".png"; |
||||
#else |
||||
QString url = IconProvider::instance()->getIconPath(args_.front()); |
||||
#endif |
||||
qDebug() << Q_FUNC_INFO << "There icon:" << url; |
||||
|
||||
printFile(url); |
||||
} |
||||
|
||||
void RequestHandler::action_public_images() |
||||
{ |
||||
const QString path = ":/Icons/" + args_.join("/"); |
||||
printFile(path); |
||||
} |
||||
|
||||
void RequestHandler::action_json_torrents() |
||||
{ |
||||
print(btjson::getTorrents(), CONTENT_TYPE_JS); |
||||
} |
||||
|
||||
void RequestHandler::action_json_preferences() |
||||
{ |
||||
print(prefjson::getPreferences(), CONTENT_TYPE_JS); |
||||
} |
||||
|
||||
void RequestHandler::action_json_transferInfo() |
||||
{ |
||||
print(btjson::getTransferInfo(), CONTENT_TYPE_JS); |
||||
} |
||||
|
||||
void RequestHandler::action_json_propertiesGeneral() |
||||
{ |
||||
print(btjson::getPropertiesForTorrent(args_.front()), CONTENT_TYPE_JS); |
||||
} |
||||
|
||||
void RequestHandler::action_json_propertiesTrackers() |
||||
{ |
||||
print(btjson::getTrackersForTorrent(args_.front()), CONTENT_TYPE_JS); |
||||
} |
||||
|
||||
void RequestHandler::action_json_propertiesFiles() |
||||
{ |
||||
print(btjson::getFilesForTorrent(args_.front()), CONTENT_TYPE_JS); |
||||
} |
||||
|
||||
void RequestHandler::action_command_shutdown() |
||||
{ |
||||
qDebug() << "Shutdown request from Web UI"; |
||||
// Special case handling for shutdown, we
|
||||
// need to reply to the Web UI before
|
||||
// actually shutting down.
|
||||
|
||||
QTimer::singleShot(0, qApp, SLOT(quit())); |
||||
} |
||||
|
||||
void RequestHandler::action_command_download() |
||||
{ |
||||
QString urls = request().posts["urls"]; |
||||
QStringList list = urls.split('\n'); |
||||
|
||||
foreach (QString url, list) |
||||
{ |
||||
url = url.trimmed(); |
||||
if (!url.isEmpty()) |
||||
{ |
||||
if (url.startsWith("bc://bt/", Qt::CaseInsensitive)) |
||||
{ |
||||
qDebug("Converting bc link to magnet link"); |
||||
url = misc::bcLinkToMagnet(url); |
||||
} |
||||
else if (url.startsWith("magnet:", Qt::CaseInsensitive)) |
||||
{ |
||||
QBtSession::instance()->addMagnetSkipAddDlg(url); |
||||
} |
||||
else |
||||
{ |
||||
qDebug("Downloading url: %s", qPrintable(url)); |
||||
QBtSession::instance()->downloadUrlAndSkipDialog(url); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_upload() |
||||
{ |
||||
qDebug() << Q_FUNC_INFO; |
||||
|
||||
foreach(const UploadedFile& torrent, request().files) |
||||
{ |
||||
QString filePath = saveTmpFile(torrent.data); |
||||
|
||||
if (!filePath.isEmpty()) |
||||
{ |
||||
QBtSession::instance()->addTorrent(filePath); |
||||
// Clean up
|
||||
fsutils::forceRemove(filePath); |
||||
print(QLatin1String("<script type=\"text/javascript\">window.parent.hideAll();</script>")); |
||||
} |
||||
else |
||||
{ |
||||
qWarning() << "I/O Error: Could not create temporary file"; |
||||
} |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_addTrackers() |
||||
{ |
||||
QString hash = request().posts["hash"]; |
||||
|
||||
if (!hash.isEmpty()) |
||||
{ |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
|
||||
if (h.is_valid() && h.has_metadata()) |
||||
{ |
||||
QString urls = request().posts["urls"]; |
||||
QStringList list = urls.split('\n'); |
||||
|
||||
foreach (const QString& url, list) |
||||
{ |
||||
announce_entry e(url.toStdString()); |
||||
h.add_tracker(e); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_resumeAll() |
||||
{ |
||||
QBtSession::instance()->resumeAllTorrents(); |
||||
} |
||||
|
||||
void RequestHandler::action_command_pauseAll() |
||||
{ |
||||
QBtSession::instance()->pauseAllTorrents(); |
||||
} |
||||
|
||||
void RequestHandler::action_command_resume() |
||||
{ |
||||
QBtSession::instance()->resumeTorrent(request().posts["hash"]); |
||||
} |
||||
|
||||
void RequestHandler::action_command_pause() |
||||
{ |
||||
QBtSession::instance()->pauseTorrent(request().posts["hash"]); |
||||
} |
||||
|
||||
void RequestHandler::action_command_setPreferences() |
||||
{ |
||||
prefjson::setPreferences(request().posts["json"]); |
||||
} |
||||
|
||||
void RequestHandler::action_command_setFilePrio() |
||||
{ |
||||
QString hash = request().posts["hash"]; |
||||
int file_id = request().posts["id"].toInt(); |
||||
int priority = request().posts["priority"].toInt(); |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
|
||||
if (h.is_valid() && h.has_metadata()) |
||||
{ |
||||
h.file_priority(file_id, priority); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_getGlobalUpLimit() |
||||
{ |
||||
print(QByteArray::number(QBtSession::instance()->getSession()->settings().upload_rate_limit)); |
||||
} |
||||
|
||||
void RequestHandler::action_command_getGlobalDlLimit() |
||||
{ |
||||
print(QByteArray::number(QBtSession::instance()->getSession()->settings().download_rate_limit)); |
||||
} |
||||
|
||||
void RequestHandler::action_command_setGlobalUpLimit() |
||||
{ |
||||
qlonglong limit = request().posts["limit"].toLongLong(); |
||||
if (limit == 0) limit = -1; |
||||
|
||||
QBtSession::instance()->setUploadRateLimit(limit); |
||||
Preferences::instance()->setGlobalUploadLimit(limit/1024.); |
||||
} |
||||
|
||||
void RequestHandler::action_command_setGlobalDlLimit() |
||||
{ |
||||
qlonglong limit = request().posts["limit"].toLongLong(); |
||||
if (limit == 0) limit = -1; |
||||
|
||||
QBtSession::instance()->setDownloadRateLimit(limit); |
||||
Preferences::instance()->setGlobalDownloadLimit(limit/1024.); |
||||
} |
||||
|
||||
void RequestHandler::action_command_getTorrentUpLimit() |
||||
{ |
||||
QString hash = request().posts["hash"]; |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
|
||||
if (h.is_valid()) |
||||
{ |
||||
print(QByteArray::number(h.upload_limit())); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_getTorrentDlLimit() |
||||
{ |
||||
QString hash = request().posts["hash"]; |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
|
||||
if (h.is_valid()) |
||||
{ |
||||
print(QByteArray::number(h.download_limit())); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_setTorrentUpLimit() |
||||
{ |
||||
QString hash = request().posts["hash"]; |
||||
qlonglong limit = request().posts["limit"].toLongLong(); |
||||
if (limit == 0) limit = -1; |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
|
||||
if (h.is_valid()) |
||||
{ |
||||
h.set_upload_limit(limit); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_setTorrentDlLimit() |
||||
{ |
||||
QString hash = request().posts["hash"]; |
||||
qlonglong limit = request().posts["limit"].toLongLong(); |
||||
if (limit == 0) limit = -1; |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
|
||||
if (h.is_valid()) |
||||
{ |
||||
h.set_download_limit(limit); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_delete() |
||||
{ |
||||
QStringList hashes = request().posts["hashes"].split("|"); |
||||
|
||||
foreach (const QString &hash, hashes) |
||||
{ |
||||
QBtSession::instance()->deleteTorrent(hash, false); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_deletePerm() |
||||
{ |
||||
QStringList hashes = request().posts["hashes"].split("|"); |
||||
|
||||
foreach (const QString &hash, hashes) |
||||
{ |
||||
QBtSession::instance()->deleteTorrent(hash, true); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_increasePrio() |
||||
{ |
||||
QStringList hashes = request().posts["hashes"].split("|"); |
||||
std::priority_queue<QPair<int, QTorrentHandle>, |
||||
std::vector<QPair<int, QTorrentHandle> >, |
||||
std::greater<QPair<int, QTorrentHandle> > > torrent_queue; |
||||
|
||||
// Sort torrents by priority
|
||||
foreach (const QString &hash, hashes) |
||||
{ |
||||
try |
||||
{ |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
if (!h.is_seed()) |
||||
{ |
||||
torrent_queue.push(qMakePair(h.queue_position(), h)); |
||||
} |
||||
} |
||||
catch(invalid_handle&) {} |
||||
} |
||||
|
||||
// Increase torrents priority (starting with the ones with highest priority)
|
||||
while(!torrent_queue.empty()) |
||||
{ |
||||
QTorrentHandle h = torrent_queue.top().second; |
||||
|
||||
try |
||||
{ |
||||
h.queue_position_up(); |
||||
} |
||||
catch(invalid_handle&) {} |
||||
|
||||
torrent_queue.pop(); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_decreasePrio() |
||||
{ |
||||
QStringList hashes = request().posts["hashes"].split("|"); |
||||
std::priority_queue<QPair<int, QTorrentHandle>, |
||||
std::vector<QPair<int, QTorrentHandle> >, |
||||
std::less<QPair<int, QTorrentHandle> > > torrent_queue; |
||||
|
||||
// Sort torrents by priority
|
||||
foreach (const QString &hash, hashes) |
||||
{ |
||||
try |
||||
{ |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
|
||||
if (!h.is_seed()) |
||||
{ |
||||
torrent_queue.push(qMakePair(h.queue_position(), h)); |
||||
} |
||||
} |
||||
catch(invalid_handle&) {} |
||||
} |
||||
|
||||
// Decrease torrents priority (starting with the ones with lowest priority)
|
||||
while(!torrent_queue.empty()) |
||||
{ |
||||
QTorrentHandle h = torrent_queue.top().second; |
||||
|
||||
try |
||||
{ |
||||
h.queue_position_down(); |
||||
} |
||||
catch(invalid_handle&) {} |
||||
|
||||
torrent_queue.pop(); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_topPrio() |
||||
{ |
||||
foreach (const QString &hash, request().posts["hashes"].split("|")) |
||||
{ |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
if (h.is_valid()) h.queue_position_top(); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_bottomPrio() |
||||
{ |
||||
foreach (const QString &hash, request().posts["hashes"].split("|")) |
||||
{ |
||||
QTorrentHandle h = QBtSession::instance()->getTorrentHandle(hash); |
||||
if (h.is_valid()) h.queue_position_bottom(); |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::action_command_recheck() |
||||
{ |
||||
QBtSession::instance()->recheckTorrent(request().posts["hash"]); |
||||
} |
||||
|
||||
bool RequestHandler::isPublicScope() |
||||
{ |
||||
return (scope_ == DEFAULT_SCOPE); |
||||
} |
||||
|
||||
void RequestHandler::processRequest() |
||||
{ |
||||
if (args_.contains(".") || args_.contains("..")) |
||||
{ |
||||
qDebug() << Q_FUNC_INFO << "Invalid path:" << request().path; |
||||
status(404, "Not Found"); |
||||
return; |
||||
} |
||||
|
||||
if (!isPublicScope() && !sessionActive()) |
||||
{ |
||||
status(403, "Forbidden"); |
||||
return; |
||||
} |
||||
|
||||
if (actions_.value(scope_).value(action_) != 0) |
||||
{ |
||||
(this->*(actions_[scope_][action_]))(); |
||||
} |
||||
else |
||||
{ |
||||
status(404, "Not Found"); |
||||
qDebug() << Q_FUNC_INFO << "Resource not found:" << request().path; |
||||
} |
||||
} |
||||
|
||||
void RequestHandler::parsePath() |
||||
{ |
||||
if(request().path == "/") action_ = WEBUI_ACTION; |
||||
|
||||
// check action for requested path
|
||||
QStringList pathItems = request().path.split('/', QString::SkipEmptyParts); |
||||
if (!pathItems.empty()) |
||||
{ |
||||
if (actions_.contains(pathItems.front())) |
||||
{ |
||||
scope_ = pathItems.front(); |
||||
pathItems.pop_front(); |
||||
} |
||||
} |
||||
|
||||
if (!pathItems.empty()) |
||||
{ |
||||
if (actions_[scope_].contains(pathItems.front())) |
||||
{ |
||||
action_ = pathItems.front(); |
||||
pathItems.pop_front(); |
||||
} |
||||
} |
||||
|
||||
args_ = pathItems; |
||||
} |
||||
|
||||
RequestHandler::RequestHandler(const HttpRequest &request, const HttpEnvironment &env, WebApplication *app) |
||||
: AbstractRequestHandler(request, env, app), scope_(DEFAULT_SCOPE), action_(DEFAULT_ACTION) |
||||
{ |
||||
parsePath(); |
||||
} |
||||
|
||||
QMap<QString, QMap<QString, RequestHandler::Action> > RequestHandler::actions_ = RequestHandler::initializeActions(); |
@ -0,0 +1,100 @@
@@ -0,0 +1,100 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2014 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#ifndef REQUESTHANDLER_H |
||||
#define REQUESTHANDLER_H |
||||
|
||||
#include <QStringList> |
||||
#include "httptypes.h" |
||||
#include "abstractrequesthandler.h" |
||||
|
||||
class WebApplication; |
||||
|
||||
class RequestHandler: public AbstractRequestHandler |
||||
{ |
||||
public: |
||||
RequestHandler( |
||||
const HttpRequest& request, const HttpEnvironment& env, |
||||
WebApplication* app); |
||||
|
||||
private: |
||||
// Actions
|
||||
void action_public_webui(); |
||||
void action_public_index(); |
||||
void action_public_login(); |
||||
void action_public_logout(); |
||||
void action_public_theme(); |
||||
void action_public_images(); |
||||
void action_json_torrents(); |
||||
void action_json_preferences(); |
||||
void action_json_transferInfo(); |
||||
void action_json_propertiesGeneral(); |
||||
void action_json_propertiesTrackers(); |
||||
void action_json_propertiesFiles(); |
||||
void action_command_shutdown(); |
||||
void action_command_download(); |
||||
void action_command_upload(); |
||||
void action_command_addTrackers(); |
||||
void action_command_resumeAll(); |
||||
void action_command_pauseAll(); |
||||
void action_command_resume(); |
||||
void action_command_pause(); |
||||
void action_command_setPreferences(); |
||||
void action_command_setFilePrio(); |
||||
void action_command_getGlobalUpLimit(); |
||||
void action_command_getGlobalDlLimit(); |
||||
void action_command_setGlobalUpLimit(); |
||||
void action_command_setGlobalDlLimit(); |
||||
void action_command_getTorrentUpLimit(); |
||||
void action_command_getTorrentDlLimit(); |
||||
void action_command_setTorrentUpLimit(); |
||||
void action_command_setTorrentDlLimit(); |
||||
void action_command_delete(); |
||||
void action_command_deletePerm(); |
||||
void action_command_increasePrio(); |
||||
void action_command_decreasePrio(); |
||||
void action_command_topPrio(); |
||||
void action_command_bottomPrio(); |
||||
void action_command_recheck(); |
||||
|
||||
typedef void (RequestHandler::*Action)(); |
||||
|
||||
QString scope_; |
||||
QString action_; |
||||
QStringList args_; |
||||
|
||||
void processRequest(); |
||||
|
||||
bool isPublicScope(); |
||||
void parsePath(); |
||||
|
||||
static QMap<QString, QMap<QString, Action> > initializeActions(); |
||||
static QMap<QString, QMap<QString, Action> > actions_; |
||||
}; |
||||
|
||||
#endif // REQUESTHANDLER_H
|
@ -0,0 +1,309 @@
@@ -0,0 +1,309 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2014 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#ifdef DISABLE_GUI |
||||
#include <QCoreApplication> |
||||
#else |
||||
#include <QApplication> |
||||
#endif |
||||
#include <QDateTime> |
||||
#include <QTimer> |
||||
#include <QFile> |
||||
#include <QDebug> |
||||
#include "preferences.h" |
||||
#include "requesthandler.h" |
||||
#include "webapplication.h" |
||||
|
||||
// UnbanTimer
|
||||
|
||||
class UnbanTimer: public QTimer |
||||
{ |
||||
public: |
||||
UnbanTimer(const QHostAddress& peer_ip, QObject *parent) |
||||
: QTimer(parent), m_peerIp(peer_ip) |
||||
{ |
||||
setSingleShot(true); |
||||
setInterval(BAN_TIME); |
||||
} |
||||
|
||||
inline const QHostAddress& peerIp() const { return m_peerIp; } |
||||
|
||||
private: |
||||
QHostAddress m_peerIp; |
||||
}; |
||||
|
||||
// WebApplication
|
||||
|
||||
WebApplication::WebApplication(QObject *parent) |
||||
: QObject(parent) |
||||
{ |
||||
} |
||||
|
||||
WebApplication::~WebApplication() |
||||
{ |
||||
// cleanup sessions data
|
||||
foreach (WebSession* session, sessions_.values()) |
||||
delete session; |
||||
} |
||||
|
||||
WebApplication *WebApplication::instance() |
||||
{ |
||||
static WebApplication inst; |
||||
return &inst; |
||||
} |
||||
|
||||
void WebApplication::UnbanTimerEvent() |
||||
{ |
||||
UnbanTimer* ubantimer = static_cast<UnbanTimer*>(sender()); |
||||
qDebug("Ban period has expired for %s", qPrintable(ubantimer->peerIp().toString())); |
||||
clientFailedAttempts_.remove(ubantimer->peerIp()); |
||||
ubantimer->deleteLater(); |
||||
} |
||||
|
||||
bool WebApplication::sessionInitialize(AbstractRequestHandler* _this) |
||||
{ |
||||
if (_this->session_ == 0) |
||||
{ |
||||
QString cookie = _this->request_.headers.value("cookie"); |
||||
//qDebug() << Q_FUNC_INFO << "cookie: " << cookie;
|
||||
|
||||
QString sessionId; |
||||
const QString SID_START = C_SID + "="; |
||||
int pos = cookie.indexOf(SID_START); |
||||
if (pos >= 0) |
||||
{ |
||||
pos += SID_START.length(); |
||||
int end = cookie.indexOf(QRegExp("[,;]"), pos); |
||||
sessionId = cookie.mid(pos, end >= 0 ? end - pos : end); |
||||
} |
||||
|
||||
// TODO: Additional session check
|
||||
|
||||
if (!sessionId.isNull()) |
||||
{ |
||||
if (sessions_.contains(sessionId)) |
||||
{ |
||||
_this->session_ = sessions_[sessionId]; |
||||
return true; |
||||
} |
||||
else |
||||
{ |
||||
qDebug() << Q_FUNC_INFO << "session does not exist!"; |
||||
} |
||||
} |
||||
} |
||||
|
||||
return false; |
||||
} |
||||
|
||||
bool WebApplication::readFile(const QString& path, QByteArray &data, QString &type) |
||||
{ |
||||
QString ext = ""; |
||||
int index = path.lastIndexOf('.') + 1; |
||||
if (index > 0) |
||||
ext = path.mid(index); |
||||
|
||||
// find translated file in cache
|
||||
if (translatedFiles_.contains(path)) |
||||
{ |
||||
data = translatedFiles_[path]; |
||||
} |
||||
else |
||||
{ |
||||
QFile file(path); |
||||
if (!file.open(QIODevice::ReadOnly)) |
||||
{ |
||||
qDebug("File %s was not found!", qPrintable(path)); |
||||
return false; |
||||
} |
||||
|
||||
data = file.readAll(); |
||||
file.close(); |
||||
|
||||
// Translate the file
|
||||
if ((ext == "html") || ((ext == "js") && !path.endsWith("excanvas-compressed.js"))) |
||||
{ |
||||
QString dataStr = QString::fromUtf8(data.constData()); |
||||
translateDocument(dataStr); |
||||
|
||||
if (path.endsWith("about.html")) |
||||
{ |
||||
dataStr.replace("${VERSION}", VERSION); |
||||
} |
||||
|
||||
data = dataStr.toUtf8(); |
||||
translatedFiles_[path] = data; // cashing translated file
|
||||
} |
||||
} |
||||
|
||||
type = CONTENT_TYPE_BY_EXT[ext]; |
||||
return true; |
||||
} |
||||
|
||||
QString WebApplication::generateSid() |
||||
{ |
||||
QString sid; |
||||
|
||||
qsrand(QDateTime::currentDateTime().toTime_t()); |
||||
do |
||||
{ |
||||
const size_t size = 6; |
||||
quint32 tmp[size]; |
||||
|
||||
for (size_t i = 0; i < size; ++i) |
||||
tmp[i] = qrand(); |
||||
|
||||
sid = QByteArray::fromRawData(reinterpret_cast<const char *>(tmp), sizeof(quint32) * size).toBase64(); |
||||
} |
||||
while (sessions_.contains(sid)); |
||||
|
||||
return sid; |
||||
} |
||||
|
||||
void WebApplication::translateDocument(QString& data) |
||||
{ |
||||
const QRegExp regex(QString::fromUtf8("_\\(([\\w\\s?!:\\/\\(\\),%µ&\\-\\.]+)\\)")); |
||||
const QRegExp mnemonic("\\(?&([a-zA-Z]?\\))?"); |
||||
const std::string contexts[] = { |
||||
"TransferListFiltersWidget", "TransferListWidget", "PropertiesWidget", |
||||
"HttpServer", "confirmDeletionDlg", "TrackerList", "TorrentFilesModel", |
||||
"options_imp", "Preferences", "TrackersAdditionDlg", "ScanFoldersModel", |
||||
"PropTabBar", "TorrentModel", "downloadFromURL", "MainWindow", "misc" |
||||
}; |
||||
const size_t context_count = sizeof(contexts) / sizeof(contexts[0]); |
||||
int i = 0; |
||||
bool found = true; |
||||
|
||||
const QString locale = Preferences::instance()->getLocale(); |
||||
bool isTranslationNeeded = !locale.startsWith("en") || locale.startsWith("en_AU") || locale.startsWith("en_GB"); |
||||
|
||||
while(i < data.size() && found) |
||||
{ |
||||
i = regex.indexIn(data, i); |
||||
if (i >= 0) |
||||
{ |
||||
//qDebug("Found translatable string: %s", regex.cap(1).toUtf8().data());
|
||||
QByteArray word = regex.cap(1).toUtf8(); |
||||
|
||||
QString translation = word; |
||||
if (isTranslationNeeded) |
||||
{ |
||||
size_t context_index = 0; |
||||
while ((context_index < context_count) && (translation == word)) |
||||
{ |
||||
#if (QT_VERSION < QT_VERSION_CHECK(5, 0, 0)) |
||||
translation = qApp->translate(contexts[context_index].c_str(), word.constData(), 0, QCoreApplication::UnicodeUTF8, 1); |
||||
#else |
||||
translation = qApp->translate(contexts[context_index].c_str(), word.constData(), 0, 1); |
||||
#endif |
||||
++context_index; |
||||
} |
||||
} |
||||
// Remove keyboard shortcuts
|
||||
translation.replace(mnemonic, ""); |
||||
|
||||
data.replace(i, regex.matchedLength(), translation); |
||||
i += translation.length(); |
||||
} |
||||
else |
||||
{ |
||||
found = false; // no more translatable strings
|
||||
} |
||||
} |
||||
} |
||||
|
||||
bool WebApplication::isBanned(const AbstractRequestHandler *_this) const |
||||
{ |
||||
return clientFailedAttempts_.value(_this->env_.clientAddress, 0) >= MAX_AUTH_FAILED_ATTEMPTS; |
||||
} |
||||
|
||||
int WebApplication::failedAttempts(const AbstractRequestHandler* _this) const |
||||
{ |
||||
return clientFailedAttempts_.value(_this->env_.clientAddress, 0); |
||||
} |
||||
|
||||
void WebApplication::resetFailedAttempts(AbstractRequestHandler* _this) |
||||
{ |
||||
clientFailedAttempts_.remove(_this->env_.clientAddress); |
||||
} |
||||
|
||||
void WebApplication::increaseFailedAttempts(AbstractRequestHandler* _this) |
||||
{ |
||||
const int nb_fail = clientFailedAttempts_.value(_this->env_.clientAddress, 0) + 1; |
||||
|
||||
clientFailedAttempts_[_this->env_.clientAddress] = nb_fail; |
||||
if (nb_fail == MAX_AUTH_FAILED_ATTEMPTS) |
||||
{ |
||||
// Max number of failed attempts reached
|
||||
// Start ban period
|
||||
UnbanTimer* ubantimer = new UnbanTimer(_this->env_.clientAddress, this); |
||||
connect(ubantimer, SIGNAL(timeout()), SLOT(UnbanTimerEvent())); |
||||
ubantimer->start(); |
||||
} |
||||
} |
||||
|
||||
bool WebApplication::sessionStart(AbstractRequestHandler *_this) |
||||
{ |
||||
if (_this->session_ == 0) |
||||
{ |
||||
_this->session_ = new WebSession(generateSid()); |
||||
sessions_[_this->session_->id] = _this->session_; |
||||
return true; |
||||
} |
||||
|
||||
return false; |
||||
} |
||||
|
||||
bool WebApplication::sessionEnd(AbstractRequestHandler *_this) |
||||
{ |
||||
if ((_this->session_ != 0) && (sessions_.contains(_this->session_->id))) |
||||
{ |
||||
sessions_.remove(_this->session_->id); |
||||
delete _this->session_; |
||||
_this->session_ = 0; |
||||
return true; |
||||
} |
||||
|
||||
return false; |
||||
} |
||||
|
||||
QStringMap WebApplication::initializeContentTypeByExtMap() |
||||
{ |
||||
QStringMap map; |
||||
|
||||
map["htm"] = CONTENT_TYPE_HTML; |
||||
map["html"] = CONTENT_TYPE_HTML; |
||||
map["css"] = CONTENT_TYPE_CSS; |
||||
map["gif"] = CONTENT_TYPE_GIF; |
||||
map["png"] = CONTENT_TYPE_PNG; |
||||
map["js"] = CONTENT_TYPE_JS; |
||||
|
||||
return map; |
||||
} |
||||
|
||||
const QStringMap WebApplication::CONTENT_TYPE_BY_EXT = WebApplication::initializeContentTypeByExtMap(); |
@ -0,0 +1,87 @@
@@ -0,0 +1,87 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2014 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* This program is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU General Public License |
||||
* as published by the Free Software Foundation; either version 2 |
||||
* of the License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
*/ |
||||
|
||||
#ifndef WEBAPPLICATION_H |
||||
#define WEBAPPLICATION_H |
||||
|
||||
#include <QObject> |
||||
#include <QMap> |
||||
#include <QHash> |
||||
#include "httptypes.h" |
||||
|
||||
struct WebSession |
||||
{ |
||||
const QString id; |
||||
|
||||
WebSession(const QString& id): id(id) {} |
||||
}; |
||||
|
||||
const QString C_SID = "SID"; // name of session id cookie
|
||||
const int BAN_TIME = 3600000; // 1 hour
|
||||
const int MAX_AUTH_FAILED_ATTEMPTS = 5; |
||||
|
||||
class AbstractRequestHandler; |
||||
|
||||
class WebApplication: public QObject |
||||
{ |
||||
Q_OBJECT |
||||
Q_DISABLE_COPY(WebApplication) |
||||
|
||||
public: |
||||
WebApplication(QObject* parent = 0); |
||||
virtual ~WebApplication(); |
||||
|
||||
static WebApplication* instance(); |
||||
|
||||
bool isBanned(const AbstractRequestHandler* _this) const; |
||||
int failedAttempts(const AbstractRequestHandler *_this) const; |
||||
void resetFailedAttempts(AbstractRequestHandler* _this); |
||||
void increaseFailedAttempts(AbstractRequestHandler* _this); |
||||
|
||||
bool sessionStart(AbstractRequestHandler* _this); |
||||
bool sessionEnd(AbstractRequestHandler* _this); |
||||
bool sessionInitialize(AbstractRequestHandler* _this); |
||||
|
||||
bool readFile(const QString &path, QByteArray& data, QString& type); |
||||
|
||||
private slots: |
||||
void UnbanTimerEvent(); |
||||
|
||||
private: |
||||
QMap<QString, WebSession*> sessions_; |
||||
QHash<QHostAddress, int> clientFailedAttempts_; |
||||
QMap<QString, QByteArray> translatedFiles_; |
||||
|
||||
QString generateSid(); |
||||
static void translateDocument(QString& data); |
||||
|
||||
static const QStringMap CONTENT_TYPE_BY_EXT; |
||||
static QStringMap initializeContentTypeByExtMap(); |
||||
}; |
||||
|
||||
#endif // WEBAPPLICATION_H
|
Loading…
Reference in new issue