Do not set full flag if remaining size if 0

arduino-libraries · Aug 24, 2016 · 41f60df · 41f60df
1 parent e17e08b
commit 41f60df
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/socket/source/socket_buffer.c b/src/socket/source/socket_buffer.c
@@ -110,7 +110,9 @@ void socketBufferCb(SOCKET sock, uint8 u8Msg, void *pvMsg)
 
 				/* Buffer full, stop reception. */
 				if (SOCKET_BUFFER_TCP_SIZE - *(gastrSocketBuffer[sock].head) < SOCKET_BUFFER_MTU) {
-					*(gastrSocketBuffer[sock].flag) |= SOCKET_BUFFER_FLAG_FULL;
+					if (pstrRecv->u16RemainingSize != 0) {
+						*(gastrSocketBuffer[sock].flag) |= SOCKET_BUFFER_FLAG_FULL;
+					}
 				}
 				else {
 					recv(sock, gastrSocketBuffer[sock].buffer + *(gastrSocketBuffer[sock].head),
@@ -168,7 +170,9 @@ void socketBufferCb(SOCKET sock, uint8 u8Msg, void *pvMsg)
 
 				/* Buffer full, stop reception. */
 				if (SOCKET_BUFFER_UDP_SIZE - *(gastrSocketBuffer[sock].head) < SOCKET_BUFFER_MTU + SOCKET_BUFFER_UDP_HEADER_SIZE) {
-					*(gastrSocketBuffer[sock].flag) |= SOCKET_BUFFER_FLAG_FULL;
+					if (pstrRecv->u16RemainingSize != 0) {
+						*(gastrSocketBuffer[sock].flag) |= SOCKET_BUFFER_FLAG_FULL;
+					}
 				}
 				else {
 					if (hif_small_xfer && hif_small_xfer != 3) {