From eec40b651986b60a65a4e68af832759f3c18f4af Mon Sep 17 00:00:00 2001
From: Jocke <joachimpetterson@gmail.com>
Date: Fri, 4 Mar 2016 05:10:18 +0100
Subject: [PATCH] We can now read larger packets than 65536 bytes in
 TCPClient::readBuffer()

---
 src/Engine/Network/TCPClient.cpp | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/Engine/Network/TCPClient.cpp b/src/Engine/Network/TCPClient.cpp
index 8bfa9ded..b3aa7d1b 100644
--- a/src/Engine/Network/TCPClient.cpp
+++ b/src/Engine/Network/TCPClient.cpp
@@ -76,7 +76,7 @@ size_t TCPClient::readBuffer()
     memcpy(&sizeOfPacket, m_ReadBuffer, sizeof(int));
     if (sizeOfPacket > m_Socket->available()) {
         LOG_WARNING("TCPClient::readBuffer(): We haven't got the whole packet yet.");
-        return 0;
+        //return 0;
     }
     // if the buffer is to small increase the size of it
     // TODO if message is huge 1 time the buffer will not decrease.
@@ -85,12 +85,15 @@ size_t TCPClient::readBuffer()
         m_ReadBuffer = new char[sizeOfPacket];
         m_BufferSize = sizeOfPacket;
     }
-    // Read the rest of the message
-    size_t bytesReceived = m_Socket->read_some(boost
-        ::asio::buffer((void*)(m_ReadBuffer), sizeOfPacket),
-        error);
-    if (error) {
-        //LOG_ERROR("receive: %s", error.message().c_str());
+    size_t bytesReceived = 0;
+    while (sizeOfPacket > bytesReceived) {
+        // Read the rest of the message
+        bytesReceived += m_Socket->read_some(boost
+            ::asio::buffer((void*)(m_ReadBuffer), sizeOfPacket - bytesReceived),
+            error);
+        if (error) {
+            //LOG_ERROR("receive: %s", error.message().c_str());
+        }
     }
     if (sizeOfPacket > 1000000)
         LOG_WARNING("The packets received are bigger than 1MB");