diff --git a/atrcopy/container.py b/atrcopy/container.py index 499dfe2..dbb8f66 100644 --- a/atrcopy/container.py +++ b/atrcopy/container.py @@ -23,7 +23,10 @@ class DiskImageContainer: def __unpack_raw_data(self, data): raw = data.tobytes() - unpacked = self.unpack_bytes(raw) + try: + unpacked = self.unpack_bytes(raw) + except EOFError as e: + raise errors.InvalidContainer(e) return to_numpy(unpacked) def unpack_bytes(self, byte_data):