Allow underscore in function names

7 years ago · b5add50230
parent ffb5779886
commit b5add50230
1 changed files with 1 additions and 2 deletions
--- a/mythril/ether/ethcontract.py
+++ b/mythril/ether/ethcontract.py
@ -44,7 +44,7 @@ class ETHContract(persistent.Persistent):
        easm_code = self.get_easm()
        str_eval = ''

-        matches = re.findall(r'func#([a-zA-Z0-9\s,(\\)\[\]]+)#', expression)
+        matches = re.findall(r'func#([a-zA-Z0-9\s_,(\\)\[\]]+)#', expression)

        for m in matches:
            # Calculate function signature hashes
@ -53,7 +53,6 @@ class ETHContract(persistent.Persistent):

            expression = expression.replace(m, sign_hash)

-
        tokens = re.split("( and | or )", expression, re.IGNORECASE)

        for token in tokens: