taynpg/frelay
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ser：服务端接收大量无效数据时踢出客户端。

Browse Source
This commit is contained in:
taynpg
2025-11-15 17:26:09 +08:00
parent de792bb68f
commit 7a69db774d
3 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Gui/main.cpp
View File

@@ -30,10 +30,10 @@ int main(int argc, char* argv[])
frelayGUI w; frelayGUI w;
QFile file(":/QtTheme/theme/Flat/Light/Blue/Pink.qss"); // QFile file(":/QtTheme/theme/Flat/Light/Blue/Pink.qss");
if (file.open(QFile::ReadOnly)) { // if (file.open(QFile::ReadOnly)) {
a.setStyleSheet(file.readAll()); // a.setStyleSheet(file.readAll());
} // }
QObject::connect(&a, &SingleApplication::instanceStarted, &w, [&w]() { QObject::connect(&a, &SingleApplication::instanceStarted, &w, [&w]() {
w.showNormal(); w.showNormal();

2
Protocol/Protocol.h
View File

@@ -13,6 +13,8 @@ constexpr quint32 CHUNK_BUF_SIZE = 1 * 1024 * 512;
constexpr quint32 FLOW_BACK_MULTIPLE = 50; constexpr quint32 FLOW_BACK_MULTIPLE = 50;
// 阻塞等级放大倍率 // 阻塞等级放大倍率
constexpr quint32 BLOCK_LEVEL_MULTIPLE = 5; constexpr quint32 BLOCK_LEVEL_MULTIPLE = 5;
// 允许最大的无效数据包大小
constexpr quint32 MAX_INVALID_PACKET_SIZE = CHUNK_BUF_SIZE * 5;
// It is specified here that the first 30 contents (inclusive) are // It is specified here that the first 30 contents (inclusive) are
// used for communication with the server. // used for communication with the server.

6
Server/Server.cpp
View File

@@ -103,6 +103,12 @@ void Server::onReadyRead()
} }
if (client) { if (client) {
if (client->buffer.size() > MAX_INVALID_PACKET_SIZE) {
auto mg = QString("Client %1 buffer size exceeded, XXXXX...").arg(client->id);
qWarning() << mg;
socket->disconnectFromHost();
return;
}
client->buffer.append(socket->readAll()); client->buffer.append(socket->readAll());
processClientData(client); processClientData(client);
} }