diff options
author | Dominik Maier <domenukk@gmail.com> | 2021-04-15 23:56:58 +0200 |
---|---|---|
committer | Dominik Maier <domenukk@gmail.com> | 2021-04-15 23:56:58 +0200 |
commit | c8e96e52536d47ee41967657202574d8e61562ee (patch) | |
tree | fb9ea197aa4b89d2c44f983cbd4ff0581af9e867 /frida_mode/test/testinstr.py | |
parent | 4f93220c4bfbffc51e18159d30e08884a4d7dfc1 (diff) | |
download | afl++-c8e96e52536d47ee41967657202574d8e61562ee.tar.gz |
autoformat with black
Diffstat (limited to 'frida_mode/test/testinstr.py')
-rwxr-xr-x | frida_mode/test/testinstr.py | 49 |
1 files changed, 33 insertions, 16 deletions
diff --git a/frida_mode/test/testinstr.py b/frida_mode/test/testinstr.py index 8f5fe886..f648808b 100755 --- a/frida_mode/test/testinstr.py +++ b/frida_mode/test/testinstr.py @@ -1,32 +1,49 @@ -#!/usr/bin/python3 +#!/usr/bin/env python3 import argparse from elftools.elf.elffile import ELFFile + def process_file(file, section, base): - with open(file, 'rb') as f: + with open(file, "rb") as f: for sect in ELFFile(f).iter_sections(): - if (sect.name == section): - start = base + sect.header['sh_offset'] - end = start + sect.header['sh_size'] - print ("0x%016x-0x%016x" % (start, end)) + if sect.name == section: + start = base + sect.header["sh_offset"] + end = start + sect.header["sh_size"] + print("0x%016x-0x%016x" % (start, end)) return - print ("Section '%s' not found in '%s'" % (section, file)) + print("Section '%s' not found in '%s'" % (section, file)) + def hex_value(x): return int(x, 16) + def main(): - parser = argparse.ArgumentParser(description='Process some integers.') - parser.add_argument('-f', '--file', dest='file', type=str, - help='elf file name', required=True) - parser.add_argument('-s', '--section', dest='section', type=str, - help='elf section name', required=True) - parser.add_argument('-b', '--base', dest='base', type=hex_value, - help='elf base address', required=True) + parser = argparse.ArgumentParser(description="Process some integers.") + parser.add_argument( + "-f", "--file", dest="file", type=str, help="elf file name", required=True + ) + parser.add_argument( + "-s", + "--section", + dest="section", + type=str, + help="elf section name", + required=True, + ) + parser.add_argument( + "-b", + "--base", + dest="base", + type=hex_value, + help="elf base address", + required=True, + ) args = parser.parse_args() - process_file (args.file, args.section, args.base) + process_file(args.file, args.section, args.base) + if __name__ == "__main__": - main() \ No newline at end of file + main() |