Add uncompressed chunk padding (#29)

aff4 · Aug 12, 2021 · 94a3583 · 94a3583
1 parent 2629bf5
commit 94a3583
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pyaff4/aff4_image.py b/pyaff4/aff4_image.py
@@ -86,6 +86,11 @@ def read(self, _):
             self.bevy_length += compressedLen
             return compressed_chunk
         else:
+            # On final chunks that aren't compressed, pad if they are less than chunk_size
+            # so that at decompression we won't try to decompress an already decompressed chunk.
+            if chunkLen < self.owner.chunk_size:
+                padding = self.owner.chunk_size - chunkLen
+                chunk += b"\x00" * padding
             self.bevy_index.append((self.bevy_length, self.owner.chunk_size))
             self.bevy_length += self.owner.chunk_size
             return chunk