fix: socket close properly

src/benchUtil.c

@@ -9,7 +9,6 @@
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  * FITNESS FOR A PARTICULAR PURPOSE.
  */
-
 #include <bench.h>
 
 char resEncodingChunk[] = "Encoding: chunked";
@@ -1161,6 +1160,45 @@ int createSockFd() {
 }
 
 void destroySockFd(int sockfd) {
+    // shutdown the connection since no more data will be sent
+    int result, how;
+#ifdef WINDOWS
+    how = SD_SEND;
+#else
+    how = SHUT_WR;
+    #define SOCKET_ERROR  -1
+#endif
+    result = shutdown(sockfd, how);
+    if (SOCKET_ERROR == result) {
+        errorPrint("Socket shutdown failed with error: %d\n",
+#ifdef WINDOWS
+                   WSAGetLastError());
+        closesocket(sockfd);
+        WSACleanup();
+#else
+                   result);
+       close(sockfd);
+#endif
+        return;
+    }
+    // Receive until the peer closes the connection
+    do {
+        int recvbuflen = LARGE_BUFF_LEN;
+        char recvbuf[LARGE_BUFF_LEN];
+        result = recv(sockfd, recvbuf, recvbuflen, 0);
+        if ( result > 0 )
+            debugPrint("Socket bytes received: %d\n", result);
+        else if ( result == 0 )
+            infoPrint("Connection closed with result %d\n", result);
+        else
+            errorPrint("Socket recv failed with error: %d\n",
+#ifdef WINDOWS
+                       WSAGetLastError()
+#else
+                        result
+#endif
+                       );
+    } while( result > 0 );
 #ifdef WINDOWS
     closesocket(sockfd);
     WSACleanup();