Merge pull request #635 from onekey-sec/634-fix-zstd-infinite-loop

fix(handlers): fix infinite loop in zstd handler.

Author: qkaiser

tests/integration/compression/zstd/__input__/truncated.dos.zstd

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:052ba0688dc57289ad3bb5863ac952b4ff8d91156e3a3a274e60ef2ad1746015
+size 9

tests/integration/compression/zstd/__output__/truncated.dos.zstd_extract/.gitkeep

@@ -58,9 +58,11 @@ def calculate_chunk(self, file: File, start_offset: int) -> Optional[ValidChunk]
 
         last_block = False
         while not last_block:
-            block_header = int.from_bytes(
-                file.read(BLOCK_HEADER_LEN), byteorder="little"
-            )
+            block_header_val = file.read(BLOCK_HEADER_LEN)
+            # EOF
+            if not block_header_val:
+                raise InvalidInputFormat("Premature end of ZSTD stream.")
+            block_header = int.from_bytes(block_header_val, byteorder="little")
             last_block = block_header >> 0 & 0b1
             block_type = block_header >> 1 & 0b11