ensured that large messages will get processed properly if they arriv…

…e in multiple packets
boomanaiden154 · Mar 7, 2020 · 59fe29c · 59fe29c
1 parent ff9c9ea
commit 59fe29c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/websocket.hpp b/websocket.hpp
@@ -153,11 +153,16 @@ class websocket
                     payloadLength += (uint64_t)payloadLengthBuffer[7];
             }
             char* textBuffer = new char[payloadLength + 1];
-            SSL_read(SSLConnection, textBuffer, payloadLength);
+            uint32_t bytesRecieved = 0;
+            while(bytesRecieved < payloadLength)
+            {
+                bytesRecieved += SSL_read(SSLConnection, textBuffer, payloadLength - bytesRecieved);
+            }
             textBuffer[payloadLength] = '\0';
             listenerCallback(textBuffer, payloadLength);
         }
         pthread_exit(0);
+        return NULL;
     }
 
     int connectSocket(std::string address, int port, void (*listenerCallback)(char*,int))