feedc0de/qt-creator
1
0
Fork 0
forked from qt-creator/qt-creator
Code Pull Requests Activity

CmdBridge: Fix handle partial package marker

Browse Source 
Change-Id: Ifbae08fdd04a292735b0bf515d7c9b43f75cc7d9
Reviewed-by: hjk <hjk@qt.io>
This commit is contained in:
Marcus Tillmanns
2024-10-11 08:36:32 +02:00
parent e2547608c0
commit ae6a0bb706
1 changed files with 51 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
src/libs/gocmdbridge/client/cmdbridgeclient.cpp
View File

@@ -320,54 +320,64 @@ expected_str<QFuture<Environment>> Client::start()
QThread::currentThread()->quit();
});
auto stateMachine = [state = int(0), packetSize(0), packetData = QByteArray(), this](
QByteArray &buffer) mutable {
static const QByteArray MagicCode{GOBRIDGE_MAGIC_PACKET_MARKER};
auto stateMachine =
[markerOffset = 0, state = int(0), packetSize(0), packetData = QByteArray(), this](
QByteArray &buffer) mutable {
static const QByteArrayView MagicCode{GOBRIDGE_MAGIC_PACKET_MARKER};
if (state == 0) {
int start = buffer.indexOf(MagicCode);
if (start == -1) {
// Broken package, search for next magic marker
qCWarning(clientLog) << "Magic marker was not found";
// If we don't find a magic marker, the rest of the buffer is trash.
buffer.clear();
} else {
buffer.remove(0, start + MagicCode.size());
state = 1;
if (state == 0) {
const auto offsetMagicCode = MagicCode.mid(markerOffset);
int start = buffer.indexOf(offsetMagicCode);
if (start == -1) {
if (buffer.size() < offsetMagicCode.size()
&& offsetMagicCode.startsWith(buffer)) {
// Partial magic marker?
markerOffset += buffer.size();
buffer.clear();
return;
}
// Broken package, search for next magic marker
qCWarning(clientLog) << "Magic marker was not found";
// If we don't find a magic marker, the rest of the buffer is trash.
buffer.clear();
} else {
buffer.remove(0, start + offsetMagicCode.size());
state = 1;
}
markerOffset = 0;
}
}
if (state == 1) {
QDataStream ds(buffer);
ds >> packetSize;
// TODO: Enforce max size in bridge.
if (packetSize > 0 && packetSize < 16384) {
state = 2;
buffer.remove(0, sizeof(packetSize));
} else {
// Broken package, search for next magic marker
qCWarning(clientLog) << "Invalid packet size" << packetSize;
state = 0;
if (state == 1) {
QDataStream ds(buffer);
ds >> packetSize;
// TODO: Enforce max size in bridge.
if (packetSize > 0 && packetSize < 16384) {
state = 2;
buffer.remove(0, sizeof(packetSize));
} else {
// Broken package, search for next magic marker
qCWarning(clientLog) << "Invalid packet size" << packetSize;
state = 0;
}
}
}
if (state == 2) {
auto packetDataRemaining = packetSize - packetData.size();
auto dataAvailable = buffer.size();
auto availablePacketData = qMin(packetDataRemaining, dataAvailable);
packetData.append(buffer.left(availablePacketData));
buffer.remove(0, availablePacketData);
if (state == 2) {
auto packetDataRemaining = packetSize - packetData.size();
auto dataAvailable = buffer.size();
auto availablePacketData = qMin(packetDataRemaining, dataAvailable);
packetData.append(buffer.left(availablePacketData));
buffer.remove(0, availablePacketData);
if (packetData.size() == packetSize) {
QCborStreamReader reader;
reader.addData(packetData);
packetData.clear();
state = 0;
auto result = d->readPacket(reader);
QTC_CHECK_EXPECTED(result);
if (packetData.size() == packetSize) {
QCborStreamReader reader;
reader.addData(packetData);
packetData.clear();
state = 0;
auto result = d->readPacket(reader);
QTC_CHECK_EXPECTED(result);
}
}
}
};
};
connect(
d->process,