Fragments forming a long message are not caught early

2019-12-31 12:56:45 +01:00
parent b9062ec770
commit 54dd866a84
2 changed files with 7 additions and 2 deletions
@@ -100,6 +100,11 @@ private:
                if (!webSocketData->fragmentBuffer.length()) {
                    webSocketData->fragmentBuffer.reserve(length + remainingBytes);
                }
+                /* Fragments forming a big message are not caught until appending them */
+                if (refusePayloadLength(length + webSocketData->fragmentBuffer.length(), webSocketState, s)) {
+                    forceClose(webSocketState, s);
+                    return true;
+                }
                webSocketData->fragmentBuffer.append(data, length);

                /* Are we done now? */