mirror of
https://huihui.cat/mirrors/MikroTikPatch.git
synced 2025-12-10 03:24:37 +01:00
Update patch.py
Signed-off-by: elseif <elseif@live.cn>
This commit is contained in:
parent
c657f2e736
commit
39af3326bd
1 changed files with 2 additions and 5 deletions
7
patch.py
7
patch.py
|
|
@ -10,10 +10,10 @@ def replace_key(old,new,data,name=''):
|
||||||
pattern_bytes = b''.join(pattern_parts)
|
pattern_bytes = b''.join(pattern_parts)
|
||||||
pattern = re.compile(pattern_bytes, flags=re.DOTALL)
|
pattern = re.compile(pattern_bytes, flags=re.DOTALL)
|
||||||
def replace_match(match):
|
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 = b''.join([new_chunks[i] + match.group(i+1) for i in range(len(new_chunks) - 1)])
|
||||||
replaced += 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()}...')
|
print(f'{name} public key patched {old[:16].hex().upper()}...')
|
||||||
return replaced
|
return replaced
|
||||||
return re.sub(pattern, replace_match, data)
|
return re.sub(pattern, replace_match, data)
|
||||||
|
|
@ -269,9 +269,6 @@ def patch_squashfs(path,key_dict):
|
||||||
for file in files:
|
for file in files:
|
||||||
file = os.path.join(root,file)
|
file = os.path.join(root,file)
|
||||||
if os.path.isfile(file):
|
if os.path.isfile(file):
|
||||||
if 'sys2' in file:
|
|
||||||
os.remove(file)
|
|
||||||
continue
|
|
||||||
data = open(file,'rb').read()
|
data = open(file,'rb').read()
|
||||||
for old_public_key,new_public_key in key_dict.items():
|
for old_public_key,new_public_key in key_dict.items():
|
||||||
_data = replace_key(old_public_key,new_public_key,data,file)
|
_data = replace_key(old_public_key,new_public_key,data,file)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue