From 1cfa1784317c49e858327a30322d1bebb98c5766 Mon Sep 17 00:00:00 2001
From: Natalia Portillo <claunia@claunia.com>
Date: Thu, 21 Oct 2021 05:02:02 +0100
Subject: [PATCH] Fix possible partial decompression of LZIP.

---
 lzip.c | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/lzip.c b/lzip.c
index 59a5ca6..83cc836 100644
--- a/lzip.c
+++ b/lzip.c
@@ -32,7 +32,7 @@ AARU_EXPORT int32_t AARU_CALL lzip_decode_buffer(uint8_t*       dst_buffer,
     void*         ctx;
     enum LZ_Errno lz_err;
     int32_t       in_pos = 0, out_pos = 0, in_size;
-
+    int           rd;
     // Initialize the decoder
     ctx    = LZ_decompress_open();
     lz_err = LZ_decompress_errno(ctx);
@@ -57,13 +57,26 @@ AARU_EXPORT int32_t AARU_CALL lzip_decode_buffer(uint8_t*       dst_buffer,
 
         in_pos += in_size;
 
-        int rd = LZ_decompress_read(ctx, dst_buffer + out_pos, dst_size);
+        rd = LZ_decompress_read(ctx, dst_buffer + out_pos, dst_size);
 
         out_pos += rd;
 
         if(LZ_decompress_finished(ctx) == 1) break;
     }
 
+    LZ_decompress_finish(ctx);
+
+    if(out_pos < dst_size)
+    {
+        do {
+            rd = LZ_decompress_read(ctx, dst_buffer + out_pos, dst_size);
+
+            out_pos += rd;
+
+            if(LZ_compress_finished(ctx) == 1) break;
+        } while(rd > 0 && out_pos < dst_size);
+    }
+
     // Free buffers
     LZ_decompress_close(ctx);