also catch key error

6 years ago · 7e78a77acb
parent 2b2698e1fb
commit 7e78a77acb
1 changed files with 1 additions and 1 deletions
--- a/mythril/disassembler/disassembly.py
+++ b/mythril/disassembler/disassembly.py
@ -88,7 +88,7 @@ def get_function_info(index: int, instruction_list: list, signature_database: Si
    try:
        offset = instruction_list[index + 2]["argument"]
        entry_point = int(offset, 16)
-    except IndexError:
+    except (KeyError, IndexError):
        return function_hash, None, None

    return function_hash, entry_point, function_name