From 558248ac82f733d2e6074b8d60cc31cf0376645d Mon Sep 17 00:00:00 2001
From: Eljakim Herrewijnen <eljakim.herrewijnen@gmail.com>
Date: Sun, 28 Apr 2024 22:26:57 +0200
Subject: [PATCH] remove argparsing

---
 source/bootrom_emulator/fuzzer.py | 36 ++++++++++++++++---------------
 1 file changed, 19 insertions(+), 17 deletions(-)

diff --git a/source/bootrom_emulator/fuzzer.py b/source/bootrom_emulator/fuzzer.py
index bfe0be0..8091e03 100644
--- a/source/bootrom_emulator/fuzzer.py
+++ b/source/bootrom_emulator/fuzzer.py
@@ -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()