X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/ef21f8dbffe6bad5c72add02f291a6e81b63ae60..280cbb85375b7968e4f81c86ceb303c7f1ee563b:/token.c

diff --git a/token.c b/token.c
index 35b980b8..5056569a 100644
--- a/token.c
+++ b/token.c
@@ -241,6 +241,7 @@ recv_deflated_token(int f, char **data)
 		inflateReset(&rx_strm);
 	    }
 	    recv_state = r_idle;
+	    rx_token = 0;
 	    break;
 	    
 	case r_idle:
@@ -292,14 +293,17 @@ recv_deflated_token(int f, char **data)
 	    rx_strm.avail_out = CHUNK_SIZE;
 	    r = inflate(&rx_strm, Z_NO_FLUSH);
 	    n = CHUNK_SIZE - rx_strm.avail_out;
-	    if (r != Z_OK || n == 0) {
+	    if (r != Z_OK) {
 		fprintf(FERROR, "inflate returned %d (%d bytes)\n", r, n);
 		exit_cleanup(1);
 	    }
 	    if (rx_strm.avail_out != 0)
 		recv_state = r_inflated;
-	    *data = dbuf;
-	    return n;
+	    if (n != 0) {
+		*data = dbuf;
+		return n;
+	    }
+	    break;
 
 	case r_running:
 	    ++rx_token;