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

Limit the amount of data the server will read

Browse Source 
Set a hard limit on 30KB that the server is willing to read, over two sequential reads. If the client sends more than 30KB combined, the server will disconnect the client.
This commit is contained in:
MangosArentLiterature 2021-04-20 11:51:22 -05:00
parent 9c3cd12202
commit 2f69b51280
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/aoclient.h
View File

@ -2049,6 +2049,11 @@ class AOClient : public QObject {
* @param incoming_message QString to be decoded.
*/
QString decodeMessage(QString incoming_message);
/**
* @brief The size, in bytes, of the last data the client sent to the server.
*/
int last_read;
};
#endif // AOCLIENT_H

5
src/aoclient.cpp
View File

@ -19,7 +19,12 @@
void AOClient::clientData()
{
if (last_read + socket->bytesAvailable() > 30720) { // Client can send a max of 30KB to the server over two sequential reads
socket->close();
}
QString data = QString::fromUtf8(socket->readAll());
last_read = data.size();
if (is_partial) {
data = partial_packet + data;