From d51d6ad547762994e42e6d5dcdaddf339d527b77 Mon Sep 17 00:00:00 2001 From: zyb Date: Wed, 19 Jun 2024 00:06:21 +0800 Subject: [PATCH] modified: patch.py --- patch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/patch.py b/patch.py index dbcc28d..ec24e0f 100644 --- a/patch.py +++ b/patch.py @@ -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="")