Improve exception when reading out of bounds EWF segment (#21)

fox-it · Feb 19, 2024 · 464f32c · 464f32c
1 parent 950dbaf
commit 464f32c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dissect/evidence/ewf.py b/dissect/evidence/ewf.py
@@ -228,7 +228,11 @@ def _read(self, offset: int, length: int) -> bytes:
         sector_count = (length + self.sector_size - 1) // self.sector_size
 
         segment_idx = bisect_right(self.ewf._segment_offsets, sector_offset)
+
         while sector_count > 0:
+            if segment_idx > len(self.ewf._segment_offsets):
+                raise EWFError(f"Missing EWF file for segment index: {segment_idx}")
+
             segment = self.ewf.open_segment(segment_idx)
 
             segment_remaining_sectors = segment.sector_count - (sector_offset - segment.sector_offset)