From a5db54eca5faa6eee311947ca5ab86a34da4546b Mon Sep 17 00:00:00 2001 From: elseif Date: Wed, 18 Jun 2025 18:11:41 +0800 Subject: [PATCH] Update patch.py Signed-off-by: elseif --- patch.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/patch.py b/patch.py index f15b852..5f5bce6 100644 --- a/patch.py +++ b/patch.py @@ -281,11 +281,22 @@ def patch_kernel(data:bytes,key_dict): else: raise Exception('unknown kernel format') +def patch_loader(loader_file): + try: + from loader import patch_loader + if 'ARCH' in os.environ and os.environ['ARCH'] == '-arm64': + arch = 'arm64' + else: + arch = 'x86' + patch_loader(loader_file,loader_file,arch) + except ImportError: + print("loader module not found. cannot run patch_loader.py") + def patch_squashfs(path,key_dict): for root, dirs, files in os.walk(path): for file in files: if file =='loader': - continue + patch_loader(loader_file) file = os.path.join(root,file) if os.path.isfile(file): data = open(file,'rb').read()