fix: honor pranks during contract creation (#140)

Co-authored-by: Daejun Park <daejunpark@gmail.com>
a16z · Jul 22, 2023 · d7e3a4e · d7e3a4e
1 parent 50ebba8
commit d7e3a4e
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 1 deletion.
diff --git a/src/halmos/sevm.py b/src/halmos/sevm.py
@@ -1752,7 +1752,7 @@ def create(
                 #
                 calldata=[],
                 callvalue=value,
-                caller=ex.this,
+                caller=caller,
                 this=new_addr,
                 #
                 pc=0,

diff --git a/tests/expected/all.json b/tests/expected/all.json
@@ -689,6 +689,14 @@
                 "time": null,
                 "num_bounded_loops": null
             },
+            {
+                "name": "checkPrankConstructor(address)",
+                "exitcode": 0,
+                "num_models": 0,
+                "num_paths": null,
+                "time": null,
+                "num_bounded_loops": null
+            },
             {
                 "name": "checkPrankExternal(address)",
                 "exitcode": 0,

diff --git a/tests/test/Prank.t.sol b/tests/test/Prank.t.sol
@@ -17,6 +17,14 @@ contract Target {
     }
 }
 
+contract ConstructorRecorder {
+    address public caller;
+
+    constructor() {
+        caller = msg.sender;
+    }
+}
+
 contract Ext is Test {
     function prank(address user) public {
         vm.prank(user);
@@ -131,4 +139,17 @@ contract PrankTest is Test {
         target.recordCaller();
         assert(target.caller() == address(this));
     }
+
+    function checkPrankConstructor(address user) public {
+        vm.prank(user);
+        ConstructorRecorder recorder = new ConstructorRecorder();
+        assert(recorder.caller() == user);
+    }
+
+    // TODO: uncomment when we add CREATE2 support
+    // function checkPrankConstructorCreate2(address user, bytes32 salt) public {
+    //     vm.prank(user);
+    //     ConstructorRecorder recorder = new ConstructorRecorder{salt:salt}();
+    //     assert(recorder.caller() == user);
+    // }
 }