mirrors/fort
1
0
Fork 0
mirror of https://github.com/tnodir/fort synced 2024-11-15 17:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

UI: ControlWorker: Refactor data header handling.

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2021-04-05 10:59:33 +03:00
parent cc6ef0909b
commit 4c45410cb0
1 changed files with 33 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
src/ui/control/controlworker.cpp
View File

@ -6,7 +6,23 @@
namespace { namespace {
constexpr int commandMaxArgs = 7; constexpr int commandMaxArgs = 7;
constexpr int commandArgMaxSize = 256; constexpr int commandArgMaxSize = 4 * 1024;
constexpr int dataMaxSize = 4 * 1024 * 1024;
struct DataHeader
{
DataHeader(Control::Command command = Control::CommandNone, int dataSize = 0) :
m_command(command), m_dataSize(dataSize)
{
}
Control::Command command() const { return static_cast<Control::Command>(m_command); }
int dataSize() const { return m_dataSize; }
private:
quint32 m_command : 8;
quint32 m_dataSize : 24;
};
} }
@ -89,18 +105,21 @@ bool ControlWorker::readRequest()
void ControlWorker::writeDataHeader(Control::Command command, int dataSize) void ControlWorker::writeDataHeader(Control::Command command, int dataSize)
{ {
const quint32 dataHeader = (quint32(command) << 24) | dataSize; DataHeader dataHeader(command, dataSize);
socket()->write((const char *) &dataHeader, sizeof(quint32)); socket()->write((const char *) &dataHeader, sizeof(DataHeader));
} }
bool ControlWorker::readDataHeader(Control::Command &command, int &dataSize) bool ControlWorker::readDataHeader(Control::Command &command, int &dataSize)
{ {
quint32 dataHeader = 0; DataHeader dataHeader;
if (socket()->read((char *) &dataHeader, sizeof(quint32)) != sizeof(quint32)) if (socket()->read((char *) &dataHeader, sizeof(DataHeader)) != sizeof(DataHeader))
return false; return false;
command = static_cast<Control::Command>(dataHeader >> 24); command = dataHeader.command();
dataSize = dataHeader & 0xFFFFFF; dataSize = dataHeader.dataSize();
if (dataSize > dataMaxSize)
return false;
return true; return true;
} }
@ -125,11 +144,11 @@ bool ControlWorker::buildArgsData(QByteArray &data, const QStringList &args)
#endif #endif
); );
const int argsSize = args.size(); const int argsCount = args.count();
if (argsSize > commandMaxArgs) if (argsCount > commandMaxArgs)
return false; return false;
stream << qint8(argsSize); stream << qint8(argsCount);
for (const auto &arg : args) { for (const auto &arg : args) {
if (arg.size() > commandArgMaxSize) if (arg.size() > commandArgMaxSize)
@ -145,12 +164,12 @@ bool ControlWorker::parseArgsData(const QByteArray &data, QStringList &args)
{ {
QDataStream stream(data); QDataStream stream(data);
qint8 argsSize; qint8 argsCount;
stream >> argsSize; stream >> argsCount;
if (argsSize > commandMaxArgs) if (argsCount > commandMaxArgs)
return false; return false;
while (--argsSize >= 0) { while (--argsCount >= 0) {
QString arg; QString arg;
stream >> arg; stream >> arg;
if (arg.size() > commandArgMaxSize) if (arg.size() > commandArgMaxSize)