Revize 285c6fe5

Re #8899 - Server - Implement variable-length message reception

deltarobot/curvedataserver.cpp

@@ -5 +5 @@
 #include <QAbstractSocket>
 #include <QVector4D>
 #include <chrono>
+#include <string>
+#include <cstring>
 
 #include "../element/trajectory.hpp"
 
@@ -173 +175 @@
     Client& client = *it;
 
     while (sender->bytesAvailable()) {
+        // read the message ID
         if (client.bufUsed < Client::MESSAGE_ID_SIZE) {
             qint64 toRead = qMin(qint64(Client::MESSAGE_ID_SIZE - client.bufUsed), sender->bytesAvailable());
-            client.bufUsed += sender->read(&client.buf[client.bufUsed], toRead);
+            if (toRead > 0) {
+                client.bufUsed += sender->read(&client.buf[client.bufUsed], toRead);
+            }
         }
 
         if (client.bufUsed < 2) {
@@ -184 +189 @@
         }
 
         uint16_t messageId = *((uint16_t*) client.buf);
+        bool variableLength = false;
         qint64 contentSize;
         switch (messageId & 0xF000) {
         case 0x0000:
@@ -202 +208 @@
             contentSize = 16;
             break;
         case 0xF000:
-            // TODO - implement support for variable length messages
+            variableLength = true;
             contentSize = 0;
             break;
         default:
@@ -210 +216 @@
             return;
         }
 
-        qint64 toRead = qMin(qint64(contentSize - (client.bufUsed - Client::MESSAGE_ID_SIZE)), sender->bytesAvailable());
+        size_t contentOffset = Client::MESSAGE_ID_SIZE;
+
+        if (variableLength) {
+            // read the message's content length
+            qint64 toRead = qMin(
+                        qint64(Client::MESSAGE_LENGTH_SIZE - (client.bufUsed - Client::MESSAGE_ID_SIZE)),
+                        sender->bytesAvailable());
+            if (toRead > 0) {
+                client.bufUsed += sender->read(&client.buf[client.bufUsed], toRead);
+            }
+
+            if (client.bufUsed < (Client::MESSAGE_ID_SIZE + Client::MESSAGE_LENGTH_SIZE)) {
+                // we do not yet know the content size, do not proceed any further
+                return;
+            }
+
+            contentSize = qint64(*((uint32_t*) &client.buf[Client::MESSAGE_ID_SIZE]));
+            contentOffset = Client::MESSAGE_ID_SIZE + Client::MESSAGE_LENGTH_SIZE;
+        }
+
+        // read message content
+        qint64 toRead = qMin(
+                    qint64(contentSize - (client.bufUsed - contentOffset)),
+                    sender->bytesAvailable());
         if (toRead > 0) {
-            client.bufUsed += sender->read(&client.buf[client.bufUsed], toRead);
+            qint64 bufRemaining = qint64(Client::BUF_SIZE - client.bufUsed);
+            if (toRead > bufRemaining) {
+                // message would overflow the buffer - we'll read whatever fits and skip the rest
+                if (bufRemaining > 0) {
+                    qint64 actuallyRead = sender->read(&client.buf[client.bufUsed], bufRemaining);
+                    client.bufUsed += actuallyRead;
+                    toRead -= actuallyRead;
+                }
+
+                if (toRead > 0) {
+                    sender->skip(toRead);
+                }
+            } else {
+                client.bufUsed += sender->read(&client.buf[client.bufUsed], toRead);
+            }
         }
 
-        if (client.bufUsed == (Client::MESSAGE_ID_SIZE + contentSize)) {
-            handleMessage(client, messageId, &client.buf[Client::MESSAGE_ID_SIZE]);
+        if (client.bufUsed == (contentOffset + contentSize)) {
+            handleMessage(client, messageId, &client.buf[contentOffset], contentSize);
             client.bufUsed = 0;
         }
     }
 }
 
-void CurveDataServer::handleMessage(Client& client, uint16_t messageId, void *content) {
+void CurveDataServer::handleMessage(Client& client, uint16_t messageId, void *content, size_t contentSize) {
     switch (messageId) {
     case 0x3001:
         qDebug() << "Magic arrived!";
@@ -238 +281 @@
         break;
     case 0xF006:
         qDebug() << "EOT!";
-        handleEOT(client, (char *) content);
+        handleEOT(client, (char *) content, contentSize);
         break;
     }
 }
@@ -276 +319 @@
     //qDebug() << "prisel pong" << number;
 }
 
-void CurveDataServer::handleEOT(Client& client, char* content)
+void CurveDataServer::handleEOT(Client& client, char* content, size_t contentSize)
 {
-    qDebug() << "Client" << client.socket->peerName() << "disconnected:" << content;
+    char cstrContent[contentSize + 1];
+    std::memcpy(cstrContent, content, contentSize);
+    cstrContent[contentSize] = '\0';
+
+    qDebug() << "Client" << client.socket->peerName() << "disconnected:" << cstrContent;
     terminateConnection(client, "Recieved EOT from client");
 }