Mamanyonyo/akashi-esquizolandia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ability to read forwarded-for header

Browse Source 
This helps when the websockets are behind a proxyserver, like Cloudflare
This commit is contained in:
Salanto 2022-03-21 01:30:03 +01:00
parent 0f6b6c3c4b
commit b38a6e7ee9
3 changed files with 39 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
core/include/ws_client.h
View File

@ -44,8 +44,7 @@ public:
* @pre This class will not connect up the ports to each other in any way. Unless some setup is done, this class * @pre This class will not connect up the ports to each other in any way. Unless some setup is done, this class
* by default will never be prompted to read and/or write from/to either of the sockets. * by default will never be prompted to read and/or write from/to either of the sockets.
*/ */
WSClient(QTcpSocket* p_tcp_socket, QWebSocket* p_web_socket, QObject* parent = nullptr) WSClient(QTcpSocket* p_tcp_socket, QWebSocket* p_web_socket, QObject* parent = nullptr);
: QObject(parent), tcp_socket(p_tcp_socket), web_socket(p_web_socket) {};
/** /**
* @brief Destructor for the WSClient class. * @brief Destructor for the WSClient class.
@ -114,6 +113,11 @@ private:
* @brief Flag that is set when packets are segmented * @brief Flag that is set when packets are segmented
*/ */
bool is_segmented = false; bool is_segmented = false;
/**
* @brief The IP send in the WSIP packet
*/
QString websocket_ip;
}; };
#endif // WS_CLIENT_H #endif // WS_CLIENT_H

17
core/src/advertiser.cpp
View File

@ -1,3 +1,20 @@
//////////////////////////////////////////////////////////////////////////////////////
// akashi - a server for Attorney Online 2 //
// Copyright (C) 2020 scatterflower //
// //
// This program is free software: you can redistribute it and/or modify //
// it under the terms of the GNU Affero General Public License as //
// published by the Free Software Foundation, either version 3 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 Affero General Public License for more details. //
// //
// You should have received a copy of the GNU Affero General Public License //
// along with this program. If not, see <https://www.gnu.org/licenses/>. //
//////////////////////////////////////////////////////////////////////////////////////
#include "include/advertiser.h" #include "include/advertiser.h"
Advertiser::Advertiser() Advertiser::Advertiser()

17
core/src/ws_client.cpp
View File

@ -61,10 +61,25 @@ void WSClient::onTcpDisconnect()
void WSClient::onTcpConnect() void WSClient::onTcpConnect()
{ {
tcp_socket->write(QString("WSIP#" + web_socket->peerAddress().toString() + "#%").toUtf8()); tcp_socket->write(QString("WSIP#" + websocket_ip + "#%").toUtf8());
tcp_socket->flush(); tcp_socket->flush();
} }
WSClient::WSClient(QTcpSocket *p_tcp_socket, QWebSocket *p_web_socket, QObject *parent)
: QObject(parent),
tcp_socket(p_tcp_socket),
web_socket(p_web_socket)
{
QNetworkRequest l_request = web_socket->request();
if (l_request.hasRawHeader("x-forwarded-for")) {
websocket_ip = l_request.rawHeader("x-forwarded-for");
}
else {
websocket_ip = web_socket->peerAddress().toString();
}
qDebug() << websocket_ip;
}
WSClient::~WSClient() WSClient::~WSClient()
{ {
tcp_socket->deleteLater(); tcp_socket->deleteLater();