WindyMadman/qBittorrent-Enhanced-Edition
1
0
Fork 0
mirror of https://github.com/c0re100/qBittorrent-Enhanced-Edition.git synced 2025-01-07 03:16:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set HTTP method restriction on WebAPI actions

Browse Source
This commit is contained in:
Chocobo1 2022-08-06 15:54:07 +08:00
parent c1e8849b40
commit 99b5983143
2 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/webui/webapplication.cpp
View File

@ -276,6 +276,20 @@ void WebApplication::doProcessRequest()
if (!session() && !isPublicAPI(scope, action))
throw ForbiddenHTTPError();
// Filter HTTP methods
const auto allowedMethodIter = m_allowedMethod.find({scope, action});
if (allowedMethodIter == m_allowedMethod.end())
{
// by default allow both GET, POST methods
if ((m_request.method != Http::METHOD_GET) && (m_request.method != Http::METHOD_POST))
throw MethodNotAllowedHTTPError();
}
else
{
if (*allowedMethodIter != m_request.method)
throw MethodNotAllowedHTTPError();
}
DataMap data;
for (const Http::UploadedFile &torrent : request().files)
data[torrent.filename] = torrent.data;

16
src/webui/webapplication.h
View File

@ -28,6 +28,8 @@
#pragma once
#include <utility>
#include <QDateTime>
#include <QElapsedTimer>
#include <QHash>
@ -130,6 +132,20 @@ private:
QHash<QString, APIController *> m_apiControllers;
QSet<QString> m_publicAPIs;
const QHash<std::pair<QString, QString>, QString> m_allowedMethod =
{
// <<controller name, action name>, HTTP method>
// TODO: this list is incomplete
{{QLatin1String("app"), QLatin1String("setPreferences")}, Http::METHOD_POST},
{{QLatin1String("app"), QLatin1String("shutdown")}, Http::METHOD_POST},
{{QLatin1String("auth"), QLatin1String("login")}, Http::METHOD_POST},
{{QLatin1String("auth"), QLatin1String("logout")}, Http::METHOD_POST},
{{QLatin1String("rss"), QLatin1String("addFeed")}, Http::METHOD_POST},
{{QLatin1String("search"), QLatin1String("installPlugin")}, Http::METHOD_POST},
{{QLatin1String("torrents"), QLatin1String("add")}, Http::METHOD_POST},
{{QLatin1String("torrents"), QLatin1String("addPeers")}, Http::METHOD_POST},
{{QLatin1String("torrents"), QLatin1String("addTrackers")}, Http::METHOD_POST}
};
bool m_isAltUIUsed = false;
QString m_rootFolder;