test_upgrade: fail if connection is closed when it shouldn't

1 files changed, 3 insertions, 1 deletions

diff --git a/src/microhttpd/test_upgrade.c b/src/microhttpd/test_upgrade.c
index 1e5430bc..81a36004 100644
--- a/src/microhttpd/test_upgrade.c
+++ b/src/microhttpd/test_upgrade.c
@@ -813,7 +813,7 @@ recv_hdr (struct wr_socket *sock)
       externalErrorExitDesc ("recv() failed");
     }
     if (0 == ret)
-      continue; /* TODO: replace with error exit! */
+      mhdErrorExitDesc ("The server unexpectedly closed connection");
     if (c == next)
     {
       i++;
@@ -860,6 +860,8 @@ recv_all (struct wr_socket *sock,
       }
       externalErrorExitDesc ("recv() failed");
     }
+    if (0 == ret)
+      mhdErrorExitDesc ("The server unexpectedly closed connection");
   }
   if (0 != strncmp (text, buf, len))
   {