igzip: [SHIM] fix memory leaks

Signed-off-by: Pablo de Lara <[email protected]>

Author: pablodelara

igzip/shim/shim_deflate.c

@@ -81,8 +81,15 @@ deflateInit2_(z_streamp strm, int level, int method, int windowBits, int memLeve
                 return -1;
         }
 
+        if (!isal_strm->level_buf) {
+                free(isal_strm);
+                fprintf(stderr, "Error: Memory allocation for level_buf failed\n");
+                return -1;
+        }
         deflate_state *s = (deflate_state *) malloc(sizeof(deflate_state));
         if (!s) {
+                free(isal_strm->level_buf);
+                free(isal_strm);
                 fprintf(stderr, "Error: Memory allocation for deflate_state failed\n");
                 return -1;
         }

igzip/shim/shim_inflate.c

@@ -61,6 +61,7 @@ inflateInit2_(z_streamp strm, int windowBits)
 
         inflate_state2 *s = (inflate_state2 *) malloc(sizeof(inflate_state2));
         if (!s) {
+                free(isal_strm_inflate);
                 fprintf(stderr, "Error: Memory allocation for inflate_state2 failed\n");
                 return -1;
         }
@@ -210,6 +211,7 @@ inflateEnd(z_streamp strm)
         }
 
         inflate_state2 *s = (inflate_state2 *) strm->state;
+        free(s->isal_strm_inflate);
         free(s);
 
         strm->state = NULL;