Eljakim/amlogic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove argparsing

Browse Source
This commit is contained in:
Eljakim Herrewijnen
2024-04-28 22:26:57 +02:00
parent add3a1feaa
commit 558248ac82
1 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
source/bootrom_emulator/fuzzer.py
View File

@@ -23,8 +23,8 @@ def test_fb_cmd(cmd=b'getvar:version', device="S905X3"):
res = emulator.run_fastboot_cmd()
pass
def afl_fuzzer():
emulator = Amlogic_Emulator()
def afl_fuzzer(device="S905X3"):
emulator = Amlogic_Emulator(device=device)
# emulator.debug = True
def _place_fb_command(uc, input, persistent_round, data):
@@ -44,21 +44,23 @@ def afl_fuzzer():
unicornafl.uc_afl_fuzz_custom(emulator.uc, "input/getvar", _place_fb_command, _run, persistent_iters=1)
if __name__ == "__main__":
args = argparse.ArgumentParser("Amlogic BootROM Fuzzer")
args.add_argument("--device", "-d", help="Device to test", default="S905X3")
args.add_argument("--test", "-t", help="Test fastboot command", default=False, action="store_true")
args.add_argument("--input", "-i", help="Input file for crash", default=None)
# print(sys.argv)
# args = argparse.ArgumentParser("Amlogic BootROM Fuzzer")
# args.add_argument("--device", "-d", help="Device to test", default="S905X3")
# args.add_argument("--test", "-t", help="Test fastboot command", default=False, action="store_true")
# args.add_argument("--input", "-i", help="Input file for crash", default=None)
args = args.parse_args()
# args = args.parse_args()
if args.test:
test_fb_cmd(device=args.device)
print("", flush=True)
exit(0)
# if args.test:
# test_fb_cmd(device=args.device)
# print("", flush=True)
# exit(0)
if args.input:
# Run a single comand
test_fb_cmd(open(args.input, 'rb').read())
else:
# Run AFL
afl_fuzzer()
# if args.input:
# # Run a single comand
# test_fb_cmd(open(args.input, 'rb').read())
# else:
# pass
# Run AFL
afl_fuzzer()