Make the hash function for generic case

6 years ago · 31f3a4a5c3
parent 02534db7bd
commit 31f3a4a5c3
1 changed files with 2 additions and 1 deletions
--- a/mythril/support/support_utils.py
+++ b/mythril/support/support_utils.py
@ -31,9 +31,10 @@ def get_code_hash(code):
    :param code: bytecode
    :return: Returns hash of the given bytecode
    """
+    code = code[2:] if code[:2] == "0x" else code
    try:
        keccak = sha3.keccak_256()
-        keccak.update(bytes.fromhex(code[2:]))
+        keccak.update(bytes.fromhex(code))
        return "0x" + keccak.hexdigest()
    except ValueError:
        log.debug("Unable to change the bytecode to bytes. Bytecode: {}".format(code))