Apple-2-Tools/atrcopy
1
0
Fork 0
mirror of https://github.com/robmcmullen/atrcopy.git synced 2024-12-02 00:51:41 +00:00
Code Issues Projects Releases Wiki Activity

Added check for incomplete file in container unpacking

Browse Source
This commit is contained in:
Rob McMullen 2018-07-16 07:03:47 -07:00
parent cdac32238e
commit 5f5e911d10
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
atrcopy/container.py
View File

@ -23,7 +23,10 @@ class DiskImageContainer:
def __unpack_raw_data(self, data): def __unpack_raw_data(self, data):
raw = data.tobytes() raw = data.tobytes()
try:
unpacked = self.unpack_bytes(raw) unpacked = self.unpack_bytes(raw)
except EOFError as e:
raise errors.InvalidContainer(e)
return to_numpy(unpacked) return to_numpy(unpacked)
def unpack_bytes(self, byte_data): def unpack_bytes(self, byte_data):