Fix memory leak in gz_compress() and add NULL check.

diff --git a/test/minigzip.c b/test/minigzip.c
index 4cbb30f4b3bcb089c442629304fde07c07f49b55..21cf8a36830b3d9771bbcecf0612450d725ca14f 100644 (file)
--- a/test/minigzip.c
+++ b/test/minigzip.c
@@ -101,9 +101,12 @@ void gz_compress(FILE *in, gzFile out) {
     if (gz_compress_mmap(in, out) == Z_OK) return;
 #endif
     buf = (char *)calloc(BUFLEN, 1);
+    if (buf == NULL) perror("out of memory");
+
     for (;;) {
         len = (int)fread(buf, 1, BUFLEN, in);
         if (ferror(in)) {
+            free(buf);
             perror("fread");
             exit(1);
         }