AMT-Cheif
/
extractor
mirror of https://github.com/AMT-Cheif/extractor.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support lz4-compressed super.img

This commit is contained in:
Jakob Lell 2021-05-25 11:33:44 +02:00
parent 10c9753eb7
commit f2746822b5
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
extractor.py
View File

@ -1091,6 +1091,8 @@ class Lz4Handler(FileHandler):
return CheckFileResult.HANDLER_NO_MATCH # Handled by BootImageHandler
if self.fn.lower() == b"recovery.img.lz4":
return CheckFileResult.HANDLER_NO_MATCH # Handled by RecoveryImageHandler
if self.fn.lower() == b"super.img.lz4":
return CheckFileResult.ARCHIVE
if self.fn.lower().startswith(b"persist."):
return CheckFileResult.IGNORE
if self.fn.lower().startswith(b"userdata."): # userdata partition contains stuff like dalvik cache etc.
@ -1138,13 +1140,13 @@ class Lz4Handler(FileHandler):
with open(abs_out_fn, 'wb') as f:
subprocess.check_call(cmd, stdout=f)
assert os.path.exists(abs_out_fn)
HANDLER_TYPES = [ExtfsHandler, SparseImageHandler, ErofsHandler]
HANDLER_TYPES = [ExtfsHandler, SparseImageHandler, ErofsHandler, SuperImageHandler]
handlers = []
for handler_type in HANDLER_TYPES:
handler = handler_type(self.extractor, self.extractor.rel_path(abs_out_fn), image_type=self.image_type, file_type=get_file_type(abs_out_fn))
handler_result = handler.check()
if handler_result != CheckFileResult.HANDLER_NO_MATCH:
assert handler_result in (CheckFileResult.SYSTEM_IMG, CheckFileResult.VENDOR_IMG, CheckFileResult.SYSTEM_OR_VENDOR), "Unexpected handler_result=%r from handler %r" % (handler_result, handler.__class__.__name__)
assert handler_result in (CheckFileResult.SYSTEM_IMG, CheckFileResult.VENDOR_IMG, CheckFileResult.SYSTEM_OR_VENDOR, CheckFileResult.ARCHIVE), "Unexpected handler_result=%r from handler %r" % (handler_result, handler.__class__.__name__)
handlers.append(handler)
if len(handlers) > 1:
raise MultipleHandlerMatchError("File %r: %r" % (abs_out_fn, [x.__class__.__name__ for x in handlers]))