aboutsummaryrefslogtreecommitdiff
path: root/frida_mode
diff options
context:
space:
mode:
authorDominik Maier <domenukk@gmail.com>2021-04-15 23:56:58 +0200
committerDominik Maier <domenukk@gmail.com>2021-04-15 23:56:58 +0200
commitc8e96e52536d47ee41967657202574d8e61562ee (patch)
treefb9ea197aa4b89d2c44f983cbd4ff0581af9e867 /frida_mode
parent4f93220c4bfbffc51e18159d30e08884a4d7dfc1 (diff)
downloadafl++-c8e96e52536d47ee41967657202574d8e61562ee.tar.gz
autoformat with black
Diffstat (limited to 'frida_mode')
-rwxr-xr-xfrida_mode/test/testinstr.py49
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()