Update patch.py

Signed-off-by: elseif <elseif@live.cn>
This commit is contained in:
elseif 2025-06-18 18:11:41 +08:00 committed by GitHub
parent dedfe06698
commit a5db54eca5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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()