From cbd74e10a3e7fe4ae27e32d91e2b1cd1b95e3729 Mon Sep 17 00:00:00 2001
From: Simon Rettberg
Date: Wed, 10 Jun 2020 16:14:50 +0200
Subject: [SERVER] Know when to stop

---
 src/server/altservers.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/server/altservers.c b/src/server/altservers.c
index fbbac81..838a475 100644
--- a/src/server/altservers.c
+++ b/src/server/altservers.c
@@ -598,7 +598,7 @@ static void altservers_findUplinkInternal(dnbd3_uplink_t *uplink)
 		char buffer[DNBD3_BLOCK_SIZE];
 		uint32_t todo = length;
 		ssize_t ret;
-		while ( ( ret = recv( sock, buffer, MIN( DNBD3_BLOCK_SIZE, todo ), MSG_WAITALL ) ) > 0 ) {
+		while ( todo != 0 && ( ret = recv( sock, buffer, MIN( DNBD3_BLOCK_SIZE, todo ), MSG_WAITALL ) ) > 0 ) {
 			todo -= (uint32_t)ret;
 		}
 		if ( todo != 0 ) {
-- 
cgit v1.2.3-55-g7522