From 39af3326bd22627d4ca91cbc216ceb8b9e5a97a7 Mon Sep 17 00:00:00 2001 From: elseif Date: Fri, 7 Feb 2025 03:39:14 +0800 Subject: [PATCH] Update patch.py Signed-off-by: elseif --- patch.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/patch.py b/patch.py index c71a2f8..a51a265 100644 --- a/patch.py +++ b/patch.py @@ -10,10 +10,10 @@ def replace_key(old,new,data,name=''): pattern_bytes = b''.join(pattern_parts) pattern = re.compile(pattern_bytes, flags=re.DOTALL) def replace_match(match): - print(f'{name} found public key at {match.start()} {match.group().hex().upper()}') + # print(f'{name} found public key at {match.start()} {match.group().hex().upper()}') replaced = b''.join([new_chunks[i] + match.group(i+1) for i in range(len(new_chunks) - 1)]) replaced += new_chunks[-1] - print(f'{name} replace public key {replaced.hex().upper()}') + # print(f'{name} replace public key {replaced.hex().upper()}') print(f'{name} public key patched {old[:16].hex().upper()}...') return replaced return re.sub(pattern, replace_match, data) @@ -269,9 +269,6 @@ def patch_squashfs(path,key_dict): for file in files: file = os.path.join(root,file) if os.path.isfile(file): - if 'sys2' in file: - os.remove(file) - continue data = open(file,'rb').read() for old_public_key,new_public_key in key_dict.items(): _data = replace_key(old_public_key,new_public_key,data,file)