[bug] fix memory leakage lloyd/easylzma#4 and lloyd/easylzma#7, close #…

…11
NeuroJSON · Aug 18, 2024 · 66c8e88 · 66c8e88
1 parent d756c37
commit 66c8e88
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/example/c/testzmat.c b/example/c/testzmat.c
@@ -36,7 +36,7 @@ int main(void) {
     /*=====================================*/
 
     /*first, perform zlib compression use the highest compression (-9); one can use zmat_encode as well*/
-    ret = zmat_run(strlen(test[0]), (unsigned char*)test[0], &compressedlen, &compressed, zmZlib, &status, flags.iscompress);
+    ret = zmat_run(strlen(test[0]), (unsigned char*)test[0], &compressedlen, &compressed, zmLzma, &status, flags.iscompress);
 
     if (ret == 0) {
         /*next, encode the compressed data using base64*/
@@ -68,7 +68,7 @@ int main(void) {
 
     if (ret == 0) {
         /*next, decompress using zlib (deflate) */
-        ret = zmat_decode(decodelen, decoded, &decompressedlen, &decompressed, zmZlib, &status);
+        ret = zmat_decode(decodelen, decoded, &decompressedlen, &decompressed, zmLzma, &status);
 
         if (ret == 0) {
             printf("\t\"decompressed\":\"%s\",\n", decompressed);

diff --git a/src/zmatlib.c b/src/zmatlib.c
@@ -965,6 +965,8 @@ simpleCompress(elzma_file_format format, const unsigned char* inData,
         *outLen = ds.outLen;
     }
 
+    elzma_compress_free(&hand);
+
     return rc;
 }
 
@@ -1009,6 +1011,8 @@ simpleDecompress(elzma_file_format format, const unsigned char* inData,
             return rc;
         }
 
+        elzma_decompress_free(&hand);
+
         *outData = ds.outData;
         *outLen = ds.outLen;
     }