More flexible regex for replacing library placeholders

Consensys · May 8, 2018 · 5a69a52 · 5a69a52
1 parent e7c1d49
commit 5a69a52
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mythril/ether/ethcontract.py b/mythril/ether/ethcontract.py
@@ -14,7 +14,9 @@ def __init__(self, code, creation_code="", name="Unknown"):
         # Workaround: We currently do not support compile-time linking.
         # Dynamic contract addresses of the format __[contract-name]_____________ are replaced with a generic address
 
-        code = re.sub(r'(_+[A-Za-z0-9]+_+)', 'aa' * 20, code)
+        code = re.sub(r'(_+.*_+)', 'aa' * 20, code)
+
+        print(code)
 
         self.code = code
         self.disassembly = Disassembly(self.code)