danger/program
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Upload files to "src"

Browse Source
This commit is contained in:
danger
2024-11-04 20:38:25 +08:00
parent a7fdf479a5
commit 0675b67377
1 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
src/binary_patch.py
View File

@@ -88,14 +88,12 @@ def patch_strcpy(lief_binary, nbytes, save_path, output=True):
new_segment = add_segment(lief_binary, types = lief._lief.ELF.Segment.TYPE.LOAD, flags = 5, content=patch_code)
new_segment_address = new_segment.virtual_address
#lief_binary.patch_pltgot("strcpy", new_segment.virtual_address)
os.system("rm " + save_path)
lief_binary.write(save_path)
os.system("chmod +x " + save_path)
elf_patch = ELF(save_path)
print("save_path --> " + save_path)
print("elf_patch.plt.strcpy --> " + hex(elf_patch.plt.strcpy))
patch_by_pltsec_jmp(elf_patch, 'strcpy', elf_patch.plt.strcpy, new_segment_address, len(patch_code), save_path)
patch_by_pltsec_jmp(elf_patch, 'strcpy', elf_patch.plt['strcpy'], new_segment_address, len(patch_code), save_path)
# to do
def patch_dprintf(lief_binary, save_path, output=True):
@@ -115,40 +113,42 @@ def patch_dprintf(lief_binary, save_path, output=True):
new_segment = add_segment(lief_binary, types = lief._lief.ELF.Segment.TYPE.LOAD, flags = 5, content=patch_code)
new_segment_address = new_segment.virtual_address
#lief_binary.patch_pltgot("dprintf", new_segment.virtual_address)
os.system("rm " + save_path)
lief_binary.write(save_path)
os.system("chmod +x " + save_path)
elf_patch = ELF(save_path)
patch_by_pltsec_jmp(elf_patch, 'dprintf', elf_patch.plt.dprintf, new_segment_address, len(patch_code), save_path)
patch_by_pltsec_jmp(elf_patch, 'dprintf', elf_patch.plt['dprintf'], new_segment_address, len(patch_code), save_path)
def patch_recv(lief_binary, nbytes, save_path, output=True):
patch_recv_code = f"""
mov rdx, {nbytes}
mov r10, rcx;
xor r8, r8;
xor r9, r9;
push 45;
pop rax;
syscall;
ret;
mov rdx, {nbytes};
"""
patch_code = asm(patch_recv_code)
if output:
print("the assmebly code :\n %s" % patch_recv_code)
print("the machine code :\n %s" % patch_code)
new_segment = add_segment(lief_binary, types = lief._lief.ELF.Segment.TYPE.LOAD, flags = 5, content=patch_code)
lief_binary.patch_pltgot("recv", new_segment.virtual_address)
new_segment_address = new_segment.virtual_address
os.system("rm " + save_path)
lief_binary.write(save_path)
os.system("chmod +x " + save_path)
# if __name__ == '__main__':
# argv = sys.argv
# argc = len(sys.argv)
# path = sys.argv[1]
# save_path = path + "_patch"
# lief_binary, pwn_binary = load_binary_file_information(path)
# if sys.argv[2] == 'dprintf':
# patch_dprintf(save_path)
# elif sys.argv[2] == 'strcpy':
# patch_strcpy(int(sys.argv[3]), save_path)
elf_patch = ELF(save_path)
patch_by_pltsec_jmp(elf_patch, 'recv', elf_patch.plt['recv'], new_segment_address, len(patch_code), save_path)
if __name__ == '__main__':
argv = sys.argv
argc = len(sys.argv)
path = sys.argv[1]
save_path = path + "_patch"
lief_binary, pwn_binary = load_binary_file_information(path)
if sys.argv[2] == 'dprintf':
patch_dprintf(lief_binary, save_path)
elif sys.argv[2] == 'strcpy':
patch_strcpy(lief_binary, int(sys.argv[3]), save_path)
elif sys.argv[2] == 'recv':
patch_recv(lief_binary, int(sys.argv[3]), save_path)