improved testcase B a little

2 files changed, 3 insertions, 7 deletions

diff --git a/contrib/testclient.c b/contrib/testclient.c
index 2c5324b..bc93336 100644
--- a/contrib/testclient.c
+++ b/contrib/testclient.c
@@ -164,7 +164,7 @@ int main(int argc, char* argv[])
       int nbread = recv(c, buf, sizeof(buf), 0);
       if(nbread <= 0) {
         if(!nbread) {
-          fprintf(stderr, "connection closed\n");
+          fprintf(stderr, "connection closed ... finished\n");
           return 0;
         } else if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) {
           perror("recv()");
@@ -174,12 +174,7 @@ int main(int argc, char* argv[])
       }
 
       rtot += nbread;
-      printf("%d bytes received, total = %d", nbread, rtot);
-      if(rtot >= 1234567) {
-        printf(" .. finished\n");
-        return 0;
-      }
-      printf("\n");
+      printf("%d bytes received, total = %d\n", nbread, rtot);
     }
     break;
   }
diff --git a/contrib/testserver.c b/contrib/testserver.c
index 3200379..00e0ac4 100644
--- a/contrib/testserver.c
+++ b/contrib/testserver.c
@@ -185,6 +185,7 @@ int main(int argc, char* argv[])
         break;
       }
     }
+    shutdown(c, SHUT_WR);
     while(recv(c, buf, 1, 0) > 0);
 
     break;