kaiyga/MikroTikPatch
1
0
Fork 0
mirror of https://github.com/elseif/MikroTikPatch synced 2026-07-17 18:49:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update patch.py

Browse Source 
Signed-off-by: elseif <elseif@live.cn>
This commit is contained in:
elseif 2025-02-05 22:47:38 +08:00 committed by GitHub
parent 5b909283a0
commit ea10a1a4c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
patch.py
View File

@ -1,7 +1,22 @@
import subprocess,lzma import subprocess,lzma
import struct,os import struct,os,re
from npk import NovaPackage,NpkPartID,NpkFileContainer from npk import NovaPackage,NpkPartID,NpkFileContainer
def replace_key(old,new,data):
old_chunks = [old[i:i+4] for i in range(0, len(old), 4)]
new_chunks = [new[i:i+4] for i in range(0, len(new), 4)]
pattern_parts = [chunk + b'(.{0,6})' for chunk in old_chunks[:-1]]
pattern_parts.append(old_chunks[-1])
pattern_bytes = b''.join(pattern_parts)
pattern = re.compile(pattern_bytes)
def replace_match(match):
print(f'public key patched {old[:16].hex().upper()}...')
replaced = b''.join([new_chunks[i] + match.group(i+1) for i in range(len(new_chunks) - 1)])
replaced += new_chunks[-1]
return replaced
return re.sub(pattern, replace_match, data)
def patch_bzimage(data:bytes,key_dict:dict): def patch_bzimage(data:bytes,key_dict:dict):
PE_TEXT_SECTION_OFFSET = 414 PE_TEXT_SECTION_OFFSET = 414
HEADER_PAYLOAD_OFFSET = 584 HEADER_PAYLOAD_OFFSET = 584
@ -22,9 +37,7 @@ def patch_bzimage(data:bytes,key_dict:dict):
initramfs = initramfs[:cpio_offset2] initramfs = initramfs[:cpio_offset2]
new_initramfs = initramfs new_initramfs = initramfs
for old_public_key,new_public_key in key_dict.items(): for old_public_key,new_public_key in key_dict.items():
if old_public_key in new_initramfs: new_initramfs = replace_key(old_public_key,new_public_key,new_initramfs)
print(f'initramfs public key patched {old_public_key[:16].hex().upper()}...')
new_initramfs = new_initramfs.replace(old_public_key,new_public_key)
new_vmlinux = vmlinux.replace(initramfs,new_initramfs) new_vmlinux = vmlinux.replace(initramfs,new_initramfs)
new_vmlinux_xz = lzma.compress(new_vmlinux,check=lzma.CHECK_CRC32,filters=[ new_vmlinux_xz = lzma.compress(new_vmlinux,check=lzma.CHECK_CRC32,filters=[
{"id": lzma.FILTER_X86}, {"id": lzma.FILTER_X86},
@ -83,9 +96,7 @@ def patch_initrd_xz(initrd_xz:bytes,key_dict:dict,ljust=True):
initrd = lzma.decompress(initrd_xz) initrd = lzma.decompress(initrd_xz)
new_initrd = initrd new_initrd = initrd
for old_public_key,new_public_key in key_dict.items(): for old_public_key,new_public_key in key_dict.items():
if old_public_key in new_initrd: new_initrd = replace_key(old_public_key,new_public_key,new_initrd)
print(f'initrd public key patched {old_public_key[:16].hex().upper()}...')
new_initrd = new_initrd.replace(old_public_key,new_public_key)
preset = 6 preset = 6
new_initrd_xz = lzma.compress(new_initrd,check=lzma.CHECK_CRC32,filters=[{"id": lzma.FILTER_LZMA2, "preset": preset }] ) new_initrd_xz = lzma.compress(new_initrd,check=lzma.CHECK_CRC32,filters=[{"id": lzma.FILTER_LZMA2, "preset": preset }] )
while len(new_initrd_xz) > len(initrd_xz) and preset < 9: while len(new_initrd_xz) > len(initrd_xz) and preset < 9:
@ -258,10 +269,9 @@ def patch_squashfs(path,key_dict):
if os.path.isfile(file): if os.path.isfile(file):
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():
if old_public_key in data: _data = replace_key(old_public_key,new_public_key,data)
print(f'{file} public key patched {old_public_key[:16].hex().upper()}...') if _data != data:
data = data.replace(old_public_key,new_public_key) open(file,'wb').write(_data)
open(file,'wb').write(data)
url_dict = { url_dict = {
os.environ['MIKRO_LICENCE_URL'].encode():os.environ['CUSTOM_LICENCE_URL'].encode(), os.environ['MIKRO_LICENCE_URL'].encode():os.environ['CUSTOM_LICENCE_URL'].encode(),
os.environ['MIKRO_UPGRADE_URL'].encode():os.environ['CUSTOM_UPGRADE_URL'].encode(), os.environ['MIKRO_UPGRADE_URL'].encode():os.environ['CUSTOM_UPGRADE_URL'].encode(),