fix: allocate one more byte to add null termination

eminfedar · Aug 2, 2023 · 78641cf · 78641cf
1 parent d66588d
commit 78641cf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/async-sockets/include/tcpsocket.hpp b/async-sockets/include/tcpsocket.hpp
@@ -96,7 +96,7 @@ class TCPSocket : public BaseSocket
 private:
     static void Receive(TCPSocket* socket)
     {
-        char tempBuffer[BUFFER_SIZE];
+        char tempBuffer[BUFFER_SIZE+1];
         ssize_t messageLength;
 
         while ((messageLength = recv(socket->sock, tempBuffer, BUFFER_SIZE, 0)) > 0)

diff --git a/async-sockets/include/udpsocket.hpp b/async-sockets/include/udpsocket.hpp
@@ -135,7 +135,7 @@ class UDPSocket : public BaseSocket
 private:
     static void Receive(UDPSocket* udpSocket)
     {
-        char tempBuffer[BUFFER_SIZE];
+        char tempBuffer[BUFFER_SIZE+1];
         ssize_t messageLength;
 
         while ((messageLength = recv(udpSocket->sock, tempBuffer, BUFFER_SIZE, 0)) != -1)
@@ -154,7 +154,7 @@ class UDPSocket : public BaseSocket
         sockaddr_in hostAddr;
         socklen_t hostAddrSize = sizeof(hostAddr);
 
-        char tempBuffer[BUFFER_SIZE];
+        char tempBuffer[BUFFER_SIZE+1];
         ssize_t messageLength;
 
         while ((messageLength = recvfrom(udpSocket->sock, tempBuffer, BUFFER_SIZE, 0, (sockaddr* )&hostAddr, &hostAddrSize)) != -1)