mirror of
https://github.com/brain-hackers/boot4u
synced 2024-12-22 20:20:05 +09:00
35 lines
831 B
Python
35 lines
831 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from elftools.elf.elffile import ELFFile # pip install pyelftools
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
if len(sys.argv) < 3:
|
||
|
|
print(f'Usage: {sys.argv[0]} in.elf out.bin')
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
with open(sys.argv[-2], 'rb') as f:
|
||
|
|
extract(ELFFile(f))
|
||
|
|
|
||
|
|
|
||
|
|
def extract(elf):
|
||
|
|
text = elf.get_section_by_name('.text')
|
||
|
|
if text is None:
|
||
|
|
print('Input ELF has no .text section', file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
with open(sys.argv[-1], 'wb') as f:
|
||
|
|
if '-p' in sys.argv:
|
||
|
|
print(f'Pure .text mode is enabled')
|
||
|
|
f.write(text.data())
|
||
|
|
else:
|
||
|
|
elf.stream.seek(0)
|
||
|
|
elf.stream.read(text.header.sh_offset)
|
||
|
|
f.write(elf.stream.read())
|
||
|
|
|
||
|
|
print(f'Successfully extracted the necessary sections to "{sys.argv[2]}"')
|
||
|
|
|
||
|
|
|
||
|
|
main()
|