admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-03 01:27:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: close PE's

Browse Source
This commit is contained in:
Dobin
2024-02-18 11:45:07 +00:00
parent dc13ff3c93
commit ef65b92b9a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pehelper.py
+2 -1
View File
@@ -18,6 +18,7 @@ def extract_code_from_exe(exe_file: FilePath) -> bytes:
logger.info(" > 0x{:X} Code Size: {} (raw code section size: {})".format( logger.info(" > 0x{:X} Code Size: {} (raw code section size: {})".format(
section.VirtualAddress, section.VirtualAddress,
len(data), section.SizeOfRawData)) len(data), section.SizeOfRawData))
pe.close()
return data return data
@@ -28,7 +29,7 @@ def write_code_section(exe_file: FilePath, new_data: bytes):
with open(exe_file, 'r+b') as f: with open(exe_file, 'r+b') as f:
f.seek(file_offset) f.seek(file_offset)
f.write(new_data) f.write(new_data)
pe.close()
def get_code_section(pe: pefile.PE) -> pefile.SectionStructure: def get_code_section(pe: pefile.PE) -> pefile.SectionStructure:
entrypoint = pe.OPTIONAL_HEADER.AddressOfEntryPoint entrypoint = pe.OPTIONAL_HEADER.AddressOfEntryPoint