Add exception handling for disassembler

7 years ago · 974c04ba27
parent 322426028f
commit 974c04ba27
1 changed files with 4 additions and 1 deletions
--- a/mythril/mythril
+++ b/mythril/mythril
@ -77,7 +77,10 @@ if (args.disassemble):
    else:
        exitWithError("Disassembler: Provide the input bytecode via -c BYTECODE or --id ID")

-    disassembly = asm.disassemble(util.safe_decode(encoded_bytecode))
+    try:
+        disassembly = asm.disassemble(util.safe_decode(encoded_bytecode))
+    except binascii.Error:
+        exitWithError("Disassembler: Invalid code string.")

    easm_text = asm.disassembly_to_easm(disassembly)