modified: patch.py

This commit is contained in:
zyb 2024-06-19 00:06:21 +08:00
parent 8544764c6f
commit d51d6ad547

View File

@ -63,7 +63,7 @@ def patch_bootloader(key_dict,boot_dev):
inode = int(tmp[1])
print(f'inode : {inode}')
#sudo debugfs /dev/sda1 -R 'stat <12>' 2> /dev/null | sed -n '11p'
stdout,stderr = run_shell_command(f"debugfs /dev/sda1 -R 'stat <12>' 2> /dev/null | sed -n '11p' ")
stdout,stderr = run_shell_command(f"debugfs {boot_dev} -R 'stat <12>' 2> /dev/null | sed -n '11p' ")
blocks_info = stdout.decode().strip().split(',')
blocks = []
ind_block_id = None
@ -76,7 +76,7 @@ def patch_bootloader(key_dict,boot_dev):
block_range = _tmp[1].strip().replace('(','').replace(')','').split('-')
blocks += [id for id in range(int(block_range[0]),int(block_range[1])+1)]
print(f' blocks : {len(blocks)} ind_block_id : {ind_block_id}')
stdout,stderr = run_shell_command(f"debugfs /dev/sda1 -R 'cat <{inode}>' 2> /dev/null")
stdout,stderr = run_shell_command(f"debugfs {boot_dev} -R 'cat <{inode}>' 2> /dev/null")
bzImage = stdout
new_bzImage = patch_bzimage(bzImage,key_dict)
print(f'write block {len(blocks)} : [',end="")