Add the return condition (#1216)

5 years ago · d8bf5021c7
parent 638f9369b9
commit d8bf5021c7
1 changed files with 4 additions and 0 deletions
--- a/mythril/analysis/modules/delegatecall.py
+++ b/mythril/analysis/modules/delegatecall.py
@ -56,6 +56,10 @@ class DelegateCallModule(DetectionModule):
        constraints = [
            to == ATTACKER_ADDRESS,
            UGT(gas, symbol_factory.BitVecVal(2300, 256)),
+            state.new_bitvec(
+                "retval_{}".format(state.get_current_instruction()["address"]), 256
+            )
+            == 1,
        ]

        for tx in state.world_state.transaction_sequence: