Merge pull request #378 from b-mueller/master

Fix function signature search

mythril/ether/ethcontract.py

@@ -38,15 +38,6 @@ class ETHContract(persistent.Persistent):
         str_eval = ''
         easm_code = None
 
-        matches = re.findall(r'func#([a-zA-Z0-9\s_,(\\)\[\]]+)#', expression)
-
-        for m in matches:
-            # Calculate function signature hashes
-
-            sign_hash = utils.sha3(m)[:4].hex()
-
-            expression = expression.replace(m, sign_hash)
-
         tokens = filter(None, re.split("(and|or|not)", expression.replace(" ", ""), re.IGNORECASE))
 
         for token in tokens:
@@ -65,11 +56,14 @@ class ETHContract(persistent.Persistent):
                 str_eval += "\"" + code + "\" in easm_code"
                 continue
 
-            m = re.match(r'^func#([a-fA-F0-9]+)#$', token)
+            m = re.match(r'^func#([a-zA-Z0-9\s_,(\\)\[\]]+)#$', token)
 
             if (m):
-                str_eval += "\"" + m.group(1) + "\" in self.disassembly.func_hashes"
 
-                continue
+                sign_hash = "0x" + utils.sha3(m.group(1))[:4].hex()
+
+                str_eval += "\"" + sign_hash + "\" in self.disassembly.func_hashes"
 
+                continue
+	
         return eval(str_eval.strip())

mythril/leveldb/client.py

@@ -203,7 +203,7 @@ class EthLevelDB(object):
 
             cnt += 1
 
-            if not cnt % 10:
+            if not cnt % 1000:
                 logging.info("Searched %d contracts" % cnt)
 
     def contract_hash_to_address(self, hash):