Fix bug in gzclose_w() when gzwrite() fails to allocate memory.

1 files changed, 8 insertions, 6 deletions

diff --git a/gzwrite.c b/gzwrite.c
index 6e6d8a2..27cb342 100644
--- a/gzwrite.c
+++ b/gzwrite.c
@@ -547,13 +547,15 @@ int ZEXPORT gzclose_w(file)
     }
 
     /* flush, free memory, and close file */
-    if (gz_comp(state, Z_FINISH) == -1)
-        ret = state->err;
-    if (!state->direct) {
-        (void)deflateEnd(&(state->strm));
-        free(state->out);
+    if (state->size) {
+        if (gz_comp(state, Z_FINISH) == -1)
+            ret = state->err;
+        if (!state->direct) {
+            (void)deflateEnd(&(state->strm));
+            free(state->out);
+        }
+        free(state->in);
     }
-    free(state->in);
     gz_error(state, Z_OK, NULL);
     free(state->path);
     if (close(state->fd) == -1)