aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--unicorn_mode/samples/python_simple/simple_test_harness.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/unicorn_mode/samples/python_simple/simple_test_harness.py b/unicorn_mode/samples/python_simple/simple_test_harness.py
index 8c5239b6..8758192e 100644
--- a/unicorn_mode/samples/python_simple/simple_test_harness.py
+++ b/unicorn_mode/samples/python_simple/simple_test_harness.py
@@ -19,7 +19,7 @@ import os
import signal
from unicornafl import *
-from unicornafl.mips_const import *
+from unicorn.mips_const import *
# Path to the file containing the binary to emulate
BINARY_FILE = os.path.join(
@@ -156,7 +156,7 @@ def main():
uc.mem_write(DATA_ADDRESS, input)
# Start the fuzzer.
- uc.afl_fuzz(args.input_file, place_input_callback, [end_address])
+ uc_afl_fuzz(uc=uc, input_file=args.input_file, place_input_callback=place_input_callback, exits=[end_address])
if __name__ == "__main__":