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