client/server: zlib error code fixup
Signed-off-by: Jens Axboe <axboe@kernel.dk>
diff --git a/client.c b/client.c
index f7b050b..678a627 100644
--- a/client.c
+++ b/client.c
@@ -1000,7 +1000,8 @@
stream.avail_out = this_chunk;
stream.next_out = p;
err = inflate(&stream, Z_NO_FLUSH);
- if (err != Z_OK) {
+ /* may be Z_OK, or Z_STREAM_END */
+ if (err < 0) {
log_err("fio: inflate error %d\n", err);
free(ret);
ret = NULL;
diff --git a/server.c b/server.c
index 2da0c83..7ab01ef 100644
--- a/server.c
+++ b/server.c
@@ -955,7 +955,10 @@
stream.avail_out = FIO_SERVER_MAX_FRAGMENT_PDU;
stream.next_out = out_pdu;
- assert(deflate(&stream, Z_FINISH) == Z_OK);
+ ret = deflate(&stream, Z_FINISH);
+ /* may be Z_OK, or Z_STREAM_END */
+ if (ret < 0)
+ goto err_zlib;
this_len = FIO_SERVER_MAX_FRAGMENT_PDU - stream.avail_out;