use zstd stream reader since image may not have content size header

Signed-off-by: Jannik Glückert <[email protected]> Closes: #12 Signed-off-by: Michał Górny <[email protected]>
projg2 · Aug 10, 2021 · b85d7dd · b85d7dd
1 parent 1393f84
commit b85d7dd
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ecleankernel/file.py b/ecleankernel/file.py
@@ -123,7 +123,14 @@ def decompress_raw(self) -> bytes:
                     # Technically a redundant import, this is just
                     # to make your IDE happy :)
                     import zstandard
-                    return zstandard.ZstdDecompressor().decompress(f.read())
+                    reader = zstandard.ZstdDecompressor().stream_reader(f)
+                    decomp = b''
+                    while True:
+                        chunk = reader.read(1024*1024)
+                        if not chunk:
+                            break
+                        decomp += chunk
+                    return decomp
                 else:
                     return getattr(mod, 'decompress')(f.read())
         return f.read()