Fix runlists becoming too large (#9)

fox-it · Nov 11, 2022 · abd3482 · abd3482
1 parent 18d589b
commit abd3482
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dissect/ffs/ffs.py b/dissect/ffs/ffs.py
@@ -302,7 +302,7 @@ def open(self):
         return RunlistStream(self.fs.fh, self.dataruns(), self.size, self.fs.fragment_size)
 
     def _iter_blocks(self):
-        num_blocks = (self.size + self.fs.block_size - 1) // self.fs.block_size
+        num_blocks = (self.size + (self.fs.block_size - 1)) // self.fs.block_size
         num_direct_blocks = min(num_blocks, c_ffs.UFS_NDADDR)
 
         blocks = self.inode.di_db[:num_direct_blocks]
@@ -320,6 +320,9 @@ def _iter_blocks(self):
                     yield block
                     num_blocks -= 1
 
+                    if num_blocks == 0:
+                        return
+
     def _walk_indirect(self, block, level, num_blocks):
         yield block, level