about summary refs log tree commit diff stats
path: root/example/disas_and_graph.py
diff options
context:
space:
mode:
Diffstat (limited to 'example/disas_and_graph.py')
-rwxr-xr-xexample/disas_and_graph.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/example/disas_and_graph.py b/example/disas_and_graph.py
index 1984bd49..23c92d0d 100755
--- a/example/disas_and_graph.py
+++ b/example/disas_and_graph.py
@@ -34,7 +34,7 @@ if data.startswith("MZ"):
     if len(sys.argv) <=2:
         ad_to_dis = e.rva2virt(e.Opthdr.AddressOfEntryPoint)
     else:
-        ad_to_dis = int(sys.argv[1], 16)
+        ad_to_dis = int(sys.argv[2], 16)
     in_str = bin_stream.bin_stream(e.virt)
     try:
         dll_dyn_funcs = get_import_address(e)
@@ -47,7 +47,7 @@ elif data.startswith("\x7fELF") :
     if len(sys.argv) <=2:
         ad_to_dis = e.Ehdr.entry
     else:
-        ad_to_dis = int(sys.argv[1], 16)
+        ad_to_dis = int(sys.argv[2], 16)
     in_str = bin_stream.bin_stream(e.virt)
     try:
         dll_dyn_funcs = get_import_address_elf(e)