From 09f8e8dbce07544a9efbb5a2e9298cab8e8057cd Mon Sep 17 00:00:00 2001 From: kevincheng96 Date: Thu, 26 Sep 2024 17:58:10 -0700 Subject: [PATCH] Switch from callcode to delegatecall --- README.md | 4 +- diagrams/diagrams.md | 51 ++++--------------- src/quark-core/src/QuarkScript.sol | 21 -------- src/quark-core/src/QuarkWallet.sol | 6 +-- test/lib/ExecuteOtherOperation.sol | 2 +- test/lib/GetMessageDetails.sol | 4 +- ...{CallcodeReentrancy.sol => Reentrancy.sol} | 11 ++-- test/lib/Transfer.sol | 4 +- test/quark-core-scripts/Multicall.t.sol | 2 +- test/quark-core/Callbacks.t.sol | 29 ++++++----- test/quark-core/QuarkWallet.t.sol | 28 +++++----- .../quark-proxy/QuarkWalletProxyFactory.t.sol | 14 ++--- 12 files changed, 66 insertions(+), 110 deletions(-) rename test/lib/{CallcodeReentrancy.sol => Reentrancy.sol} (87%) diff --git a/README.md b/README.md index 691bf1ca..ae4ca00e 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ ## Overview -Quark is an Ethereum smart contract wallet system, designed to run custom code — termed Quark Operations — with each transaction. This functionality is achieved through Quark wallet's capability to execute code from a separate contract via a `callcode` or `delegatecall` operation. The system leverages _Code Jar_, using `CREATE2` to deploy EVM bytecode for efficient code re-use. Additionally, the _Quark Nonce Manager_ contract plays a pivotal role in managing nonces for each wallet operation. The system also includes a wallet factory for deterministic wallet creation and a suite of Core Scripts — audited, versatile contracts that form the foundation for complex Quark Operations such as multicalls and flash-loans. +Quark is an Ethereum smart contract wallet system, designed to run custom code — termed Quark Operations — with each transaction. This functionality is achieved through Quark wallet's capability to execute code from a separate contract via a `delegatecall` operation. The system leverages _Code Jar_, using `CREATE2` to deploy EVM bytecode for efficient code re-use. Additionally, the _Quark Nonce Manager_ contract plays a pivotal role in managing nonces for each wallet operation. The system also includes a wallet factory for deterministic wallet creation and a suite of Core Scripts — audited, versatile contracts that form the foundation for complex Quark Operations such as multicalls and flash-loans. ## Contracts @@ -49,7 +49,7 @@ flowchart TB factory -- 1. createAndExecute --> wallet wallet -- 2. saveCode --> jar jar -- 3. CREATE2 --> script - wallet -- 4. Executes script\nusing callcode --> script + wallet -- 4. Executes script\nusing delegatecall --> script ``` ## Quark Wallet Features diff --git a/diagrams/diagrams.md b/diagrams/diagrams.md index 88123921..02ad3c07 100644 --- a/diagrams/diagrams.md +++ b/diagrams/diagrams.md @@ -21,8 +21,8 @@ sequenceDiagram title: Quark Operation via Factory UW -->> QW: [4] Delegatecall [sender=Factory] QW -->> CJ: [5] Save Script CJ -->> QW: [6] Return Script Address - QW -->> S: [7] Callcode [sender=User Wallet] - S -->> S: [8] Executes User Code [sender=User Wallet] + QW -->> S: [7] Delegatecall [sender=Factory] [address(this)=User Wallet] + S -->> S: [8] Executes User Code [sender=Factory] [address(this)=User Wallet] ``` ## Execute Quark Operation @@ -46,8 +46,8 @@ sequenceDiagram title: Execute Quark Operation UW -->> QW: [2] Delegatecall [sender=EOA] QW -->> CJ: [3] Save Script CJ -->> QW: [4] Return Script Address - QW -->> S: [5] Callcode [sender=User Wallet] - S -->> S: [6] Executes User Code [sender=User Wallet] + QW -->> S: [5] Delegatecall [sender=EOA] [address(this)=User Wallet] + S -->> S: [6] Executes User Code [sender=EOA] [address(this)=User Wallet] ``` ## Execute Quark Operation Direct @@ -71,8 +71,8 @@ sequenceDiagram title: Execute Quark Operation Direct UW -->> QW: [2] Delegatecall [sender=User] QW -->> CJ: [3] Save Script CJ -->> QW: [4] Return Script Address - QW -->> S: [5] Callcode [sender=User Wallet] - S -->> S: [6] Executes User Code [sender=User Wallet] + QW -->> S: [5] Delegatecall [sender=User] [address(this)=User Wallet] + S -->> S: [6] Executes User Code [sender=User] [address(this)=User Wallet] ``` ## Execute Quark Operation Erc20 Transfer @@ -97,8 +97,8 @@ sequenceDiagram title: Execute Quark Operation Erc20 Transfer UW -->> QW: [2] Delegatecall [sender=EOA] QW -->> CJ: [3] Save Script CJ -->> QW: [4] Return Script Address - QW -->> S: [5] Callcode [sender=User Wallet] - S -->> S: [6] Executes "Ethcall" Script [sender=User Wallet] + QW -->> S: [5] Delegatecall [sender=EOA] [address(this)=User Wallet] + S -->> S: [6] Executes "Ethcall" Script [sender=EOA] [address(this)=User Wallet] S -->> T: [7] Erc20 Transfer [sender=User Wallet] ``` @@ -126,9 +126,9 @@ sequenceDiagram title: Execute Quark Operation with Callback QW -->> CJ: [3] Save Script CJ -->> QW: [4] Return Script Address QW -->> QW: [5] Set Code Address - QW -->> S: [6] Callcode [sender=User Wallet] - S -->> S: [7] Executes "FlashMulticall" Script [sender=User Wallet] - S -->> U: [8] Uniswap Flash [sender=User Wallet] + QW -->> S: [6] Delegatecall [sender=EOA] [address(this)=User Wallet] + S -->> S: [7] Executes "FlashMulticall" Script [sender=EOA] [address(this)=User Wallet] + S -->> U: [8] Uniswap Flash [sender=EOA] [address(this)=User Wallet] U -->> T: [9] Erc20 Transfer [sender=Uniswap] U -->> UW: [10] Flash Callback [sender=Uniswap] UW -->> QW: [11] Delegatecall [sender=Uniswap] @@ -137,32 +137,3 @@ sequenceDiagram title: Execute Quark Operation with Callback S -->> S: [14] Run Script S -->> T: [15] Erc20 Transfer "Repay" [sender=User Wallet] ``` - -## Upgrade Quark Wallet - -```mermaid -sequenceDiagram title: Upgrade Quark Wallet - %%{init: {'theme': 'forest' } }%% - actor User - participant F as Factory - - box lightblue Executes as Wallet - participant UW as User Wallet [TUP] - participant QW as QuarkWallet - end - participant CJ as Code Jar - box lightblue Executes as Wallet - participant S as Script - end - participant PA as Proxy Admin - - User -->> UW: [1] Execute Quark Operation [sender=EOA] - UW -->> QW: [2] Delegatecall [sender=EOA] - QW -->> CJ: [3] Save Script - CJ -->> QW: [4] Return Script Address - QW -->> S: [5] Callcode [sender=User Wallet] - S -->> S: [6] Executes User Code [sender=User Wallet] - S -->> PA: [7] Call upgradeAndCall [sender=User Wallet] - PA -->> UW: [8] Call upgradeToAndCall [sender=Proxy Admin] - UW -->> UW: [9] Upgrade Wallet -``` diff --git a/src/quark-core/src/QuarkScript.sol b/src/quark-core/src/QuarkScript.sol index df3d8469..c77e2192 100644 --- a/src/quark-core/src/QuarkScript.sol +++ b/src/quark-core/src/QuarkScript.sol @@ -38,27 +38,6 @@ abstract contract QuarkScript { } } - /** - * @notice A cheaper, but weaker reentrancy guard that does not prevent recursive reentrancy (e.g. script calling itself) - * @dev Use with caution; this guard should only be used if the function being guarded cannot recursively call itself - * There are currently two ways to do this from a script: - * 1. The script uses `delegatecall` and the target can be itself (technically the wallet). The script - * has to also enable callbacks for this reentrancy to succeed. - * 2. The script defines circular codepaths that can be used to reenter the function using internal - * functions. - * @dev A side-effect of using this guard is that the guarded function can no longer be called as part of the Quark wallet - * callback flow. This is because the fallback in Quark wallet makes a `delegatecall` instead of a `callcode`. The - * guarded function would still be able to be called if a calling contract calls into the Quark wallet fallback using - * a `delegatecall`, but most calling contracts are likely to make a `call` into the Quark wallet fallback instead. - */ - modifier onlyWallet() { - if (msg.sender != address(this)) { - revert ReentrantCall(); - } - - _; - } - /// @notice Returns the `signer` of the wallet function signer() internal view returns (address) { return IHasSignerExecutor(address(this)).signer(); diff --git a/src/quark-core/src/QuarkWallet.sol b/src/quark-core/src/QuarkWallet.sol index 17b3f8bc..e6827c7d 100644 --- a/src/quark-core/src/QuarkWallet.sol +++ b/src/quark-core/src/QuarkWallet.sol @@ -487,7 +487,7 @@ contract QuarkWallet is IERC1271 { bytes32 oldActiveSubmissionToken; address oldCallback; assembly { - // Cache the previous values in each of the transient slots so they can be restored after the callcode + // Cache the previous values in each of the transient slots so they can be restored after executing the script oldActiveScript := tload(activeScriptSlot) oldActiveNonce := tload(activeNonceSlot) oldActiveSubmissionToken := tload(activeSubmissionTokenSlot) @@ -505,9 +505,7 @@ contract QuarkWallet is IERC1271 { // Transiently set the callback slot to 0 tstore(callbackSlot, 0) - // Note: CALLCODE is used to set the QuarkWallet as the `msg.sender` - success := - callcode(gas(), scriptAddress, /* value */ 0, add(scriptCalldata, 0x20), scriptCalldataLen, 0x0, 0) + success := delegatecall(gas(), scriptAddress, add(scriptCalldata, 0x20), scriptCalldataLen, 0x0, 0) returnSize := returndatasize() // Transiently restore the active script diff --git a/test/lib/ExecuteOtherOperation.sol b/test/lib/ExecuteOtherOperation.sol index 6e853230..78def12c 100644 --- a/test/lib/ExecuteOtherOperation.sol +++ b/test/lib/ExecuteOtherOperation.sol @@ -8,6 +8,6 @@ contract ExecuteOtherOperation is QuarkScript { function run(QuarkWallet.QuarkOperation memory op, uint8 v, bytes32 r, bytes32 s) external returns (bytes memory) { // XXX: this should just be run(uint256,address,bytes) and use direct execute path allowCallback(); - return QuarkWallet(payable(msg.sender)).executeQuarkOperation(op, v, r, s); + return QuarkWallet(payable(address(this))).executeQuarkOperation(op, v, r, s); } } diff --git a/test/lib/GetMessageDetails.sol b/test/lib/GetMessageDetails.sol index 10412d26..50536f3a 100644 --- a/test/lib/GetMessageDetails.sol +++ b/test/lib/GetMessageDetails.sol @@ -2,7 +2,7 @@ pragma solidity 0.8.27; contract GetMessageDetails { - function getMsgSenderAndValue() external payable returns (address, uint256) { - return (msg.sender, msg.value); + function getMsgDetails() external payable returns (address, address, uint256) { + return (msg.sender, address(this), msg.value); } } diff --git a/test/lib/CallcodeReentrancy.sol b/test/lib/Reentrancy.sol similarity index 87% rename from test/lib/CallcodeReentrancy.sol rename to test/lib/Reentrancy.sol index 93468649..ff28c3d7 100644 --- a/test/lib/CallcodeReentrancy.sol +++ b/test/lib/Reentrancy.sol @@ -10,8 +10,10 @@ interface CallbackReceiver { function receiveCallback() external; } +// Note: This used to be exploitable when the script was protected using the `onlyWallet` modifier. Now that we +// switched over to a standard reentrancy guard (`nonReentrant`), this script is no longer exploitable. contract ExploitableScript is QuarkScript, CallbackReceiver { - // we expect a callback, but we do not guard gainst re-entrancy, allowing the caller to steal funds + // We expect a callback, but we do not guard gainst re-entrancy, allowing the caller to steal funds function callMeBack(address target, bytes calldata call, uint256 fee) external payable returns (bytes memory) { allowCallback(); (bool success, bytes memory result) = target.call{value: fee}(call); @@ -23,11 +25,12 @@ contract ExploitableScript is QuarkScript, CallbackReceiver { return result; } - // protected by `onlyWallet`, but still susceptible to recursive re-entrancy due to using `delegatecall` + // Would be susceptible to recursive reentrancy if protected by `onlyWallet` instead of `nonReentrant`due to + // using delegatecall function callMeBackDelegateCall(address target, bytes calldata call, uint256 fee) external payable - onlyWallet + nonReentrant returns (bytes memory) { allowCallback(); @@ -56,7 +59,7 @@ contract ProtectedScript is QuarkScript, CallbackReceiver { function callMeBack(address target, bytes calldata call, uint256 fee) external payable - onlyWallet + nonReentrant returns (bytes memory) { allowCallback(); diff --git a/test/lib/Transfer.sol b/test/lib/Transfer.sol index f3146c70..25d71d63 100644 --- a/test/lib/Transfer.sol +++ b/test/lib/Transfer.sol @@ -17,7 +17,7 @@ contract TransferActions is QuarkScript { * @param recipient The recipient address * @param amount The amount to transfer */ - function transferERC20Token(address token, address recipient, uint256 amount) external onlyWallet { + function transferERC20Token(address token, address recipient, uint256 amount) external nonReentrant { IERC20(token).safeTransfer(recipient, amount); } @@ -26,7 +26,7 @@ contract TransferActions is QuarkScript { * @param recipient The recipient address * @param amount The amount to transfer */ - function transferNativeToken(address recipient, uint256 amount) external onlyWallet { + function transferNativeToken(address recipient, uint256 amount) external nonReentrant { (bool success, bytes memory data) = payable(recipient).call{value: amount}(""); if (!success) { revert TransferFailed(data); diff --git a/test/quark-core-scripts/Multicall.t.sol b/test/quark-core-scripts/Multicall.t.sol index 5024e0bb..59ca3ccb 100644 --- a/test/quark-core-scripts/Multicall.t.sol +++ b/test/quark-core-scripts/Multicall.t.sol @@ -81,7 +81,7 @@ contract MulticallTest is Test { multicallContract.run(callContracts, callDatas); } - function testCallcodeToMulticallSucceedsWhenUninitialized() public { + function testDelegatecallToMulticallSucceedsWhenInitialized() public { // gas: do not meter set-up vm.pauseGasMetering(); QuarkWallet wallet = QuarkWallet(factory.create(alice, address(0))); diff --git a/test/quark-core/Callbacks.t.sol b/test/quark-core/Callbacks.t.sol index cf546b3f..9f664378 100644 --- a/test/quark-core/Callbacks.t.sol +++ b/test/quark-core/Callbacks.t.sol @@ -24,7 +24,7 @@ import {QuarkOperationHelper, ScriptType} from "test/lib/QuarkOperationHelper.so import {CounterScript} from "test/lib/CounterScript.sol"; import {ExecuteOnBehalf} from "test/lib/ExecuteOnBehalf.sol"; import {CallbackFromCounter} from "test/lib/CallbackFromCounter.sol"; -import {CallbackCaller, ExploitableScript, ProtectedScript} from "test/lib/CallcodeReentrancy.sol"; +import {CallbackCaller, ExploitableScript, ProtectedScript} from "test/lib/Reentrancy.sol"; import {Ethcall} from "quark-core-scripts/src/Ethcall.sol"; @@ -289,7 +289,10 @@ contract CallbacksTest is Test { /* ===== callback reentrancy tests ===== */ - function testCallcodeReentrancyExploitWithUnprotectedScript() public { + function testDelegatecallReentrancyExploitWithUnprotectedScript() public { + // Note: The explanation below is no longer relevant because we moved away from using + // `callcode`. However, we are leaving the explanation as documentation for posterity. + /* * Notably, Quark uses `callcode` instead of `delegatecall` to execute script bytecode in * the context of a wallet. Consequently, it is possible to construct a sort of "only-self" @@ -323,8 +326,8 @@ contract CallbacksTest is Test { * recursive callbacks and exploiting the wallet. */ vm.pauseGasMetering(); - bytes memory exploitableScript = new YulHelper().getCode("CallcodeReentrancy.sol/ExploitableScript.json"); - bytes memory callbackCaller = new YulHelper().getCode("CallcodeReentrancy.sol/CallbackCaller.json"); + bytes memory exploitableScript = new YulHelper().getCode("Reentrancy.sol/ExploitableScript.json"); + bytes memory callbackCaller = new YulHelper().getCode("Reentrancy.sol/CallbackCaller.json"); address callbackCallerAddress = codeJar.saveCode(callbackCaller); @@ -349,11 +352,11 @@ contract CallbacksTest is Test { assertEq(callbackCallerAddress.balance, 1000 wei); } - function testCallcodeReentrancyProtectionWithProtectedScript() public { + function testDelegatecallReentrancyProtectionWithProtectedScript() public { // gas: do not meter set-up vm.pauseGasMetering(); - bytes memory protectedScript = new YulHelper().getCode("CallcodeReentrancy.sol/ProtectedScript.json"); - bytes memory callbackCaller = new YulHelper().getCode("CallcodeReentrancy.sol/CallbackCaller.json"); + bytes memory protectedScript = new YulHelper().getCode("Reentrancy.sol/ProtectedScript.json"); + bytes memory callbackCaller = new YulHelper().getCode("Reentrancy.sol/CallbackCaller.json"); address callbackCallerAddress = codeJar.saveCode(callbackCaller); @@ -401,12 +404,13 @@ contract CallbacksTest is Test { assertEq(callbackCallerAddress.balance, 500 wei); } - // This exploit is possible despite the use of `onlyWallet` guard because it uses recursive reentrancy (using delegatecalls) - function testCallcodeReentrancyExploitWithProtectedScript() public { + // Note: This used to be exploitable when the script was protected using the `onlyWallet` modifier. Now that we + // switched over to a standard reentrancy guard (`nonReentrant`), this script is no longer exploitable. + function testReentrancyGuardProtectsAgainstDoubleDipping() public { // gas: do not meter set-up vm.pauseGasMetering(); - bytes memory exploitableScript = new YulHelper().getCode("CallcodeReentrancy.sol/ExploitableScript.json"); - bytes memory callbackCaller = new YulHelper().getCode("CallcodeReentrancy.sol/CallbackCaller.json"); + bytes memory exploitableScript = new YulHelper().getCode("Reentrancy.sol/ExploitableScript.json"); + bytes memory callbackCaller = new YulHelper().getCode("Reentrancy.sol/CallbackCaller.json"); address callbackCallerAddress = codeJar.saveCode(callbackCaller); @@ -428,6 +432,7 @@ contract CallbacksTest is Test { // gas: meter execute vm.resumeGasMetering(); aliceWallet.executeQuarkOperation(op, v, r, s); - assertEq(callbackCallerAddress.balance, 2 ether); + // Note: If this was exploitable, the callback caller would have 2 ether + assertEq(callbackCallerAddress.balance, 1 ether); } } diff --git a/test/quark-core/QuarkWallet.t.sol b/test/quark-core/QuarkWallet.t.sol index b560cec2..ea2ff5d9 100644 --- a/test/quark-core/QuarkWallet.t.sol +++ b/test/quark-core/QuarkWallet.t.sol @@ -106,7 +106,7 @@ contract QuarkWalletTest is Test { vm.pauseGasMetering(); bytes memory getMessageDetails = new YulHelper().getCode("GetMessageDetails.sol/GetMessageDetails.json"); QuarkWallet.QuarkOperation memory op = new QuarkOperationHelper().newBasicOpWithCalldata( - aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgSenderAndValue()"), ScriptType.ScriptSource + aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgDetails()"), ScriptType.ScriptSource ); (uint8 v, bytes32 r, bytes32 s) = new SignatureHelper().signOp(alicePrivateKey, aliceWallet, op); @@ -114,8 +114,9 @@ contract QuarkWalletTest is Test { vm.resumeGasMetering(); bytes memory result = aliceWallet.executeQuarkOperation(op, v, r, s); - (address msgSender, uint256 msgValue) = abi.decode(result, (address, uint256)); - assertEq(msgSender, address(aliceWallet)); + (address msgSender, address addressThis, uint256 msgValue) = abi.decode(result, (address, address, uint256)); + assertEq(msgSender, address(this), "Message sender should be address(this)"); + assertEq(addressThis, address(aliceWallet), "address(this) should be Alice's wallet"); assertEq(msgValue, 0); } @@ -126,7 +127,7 @@ contract QuarkWalletTest is Test { bytes memory getMessageDetails = new YulHelper().getCode("GetMessageDetails.sol/GetMessageDetails.json"); bytes32 nonce = new QuarkOperationHelper().semiRandomNonce(nonceManager, aliceWalletExecutable); address scriptAddress = codeJar.saveCode(getMessageDetails); - bytes memory call = abi.encodeWithSignature("getMsgSenderAndValue()"); + bytes memory call = abi.encodeWithSignature("getMsgDetails()"); vm.startPrank(aliceAccount); @@ -136,8 +137,9 @@ contract QuarkWalletTest is Test { vm.stopPrank(); - (address msgSender, uint256 msgValue) = abi.decode(result, (address, uint256)); - assertEq(msgSender, address(aliceWalletExecutable)); + (address msgSender, address addressThis, uint256 msgValue) = abi.decode(result, (address, address, uint256)); + assertEq(msgSender, aliceAccount, "Message sender should be Alice's account"); + assertEq(addressThis, address(aliceWalletExecutable), "address(this) should be Alice's wallet"); assertEq(msgValue, 0); } @@ -148,12 +150,12 @@ contract QuarkWalletTest is Test { vm.pauseGasMetering(); bytes memory getMessageDetails = new YulHelper().getCode("GetMessageDetails.sol/GetMessageDetails.json"); QuarkWallet.QuarkOperation memory opWithScriptAddress = new QuarkOperationHelper().newBasicOpWithCalldata( - aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgSenderAndValue()"), ScriptType.ScriptAddress + aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgDetails()"), ScriptType.ScriptAddress ); (uint8 v, bytes32 r, bytes32 s) = new SignatureHelper().signOp(alicePrivateKey, aliceWallet, opWithScriptAddress); QuarkWallet.QuarkOperation memory opWithScriptSource = new QuarkOperationHelper().newBasicOpWithCalldata( - aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgSenderAndValue()"), ScriptType.ScriptSource + aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgDetails()"), ScriptType.ScriptSource ); opWithScriptSource.nonce = new QuarkOperationHelper().incrementNonce(opWithScriptSource.nonce); (uint8 v2, bytes32 r2, bytes32 s2) = @@ -191,11 +193,7 @@ contract QuarkWalletTest is Test { bytes memory getMessageDetails = new YulHelper().getCode("GetMessageDetails.sol/GetMessageDetails.json"); (QuarkWallet.QuarkOperation memory opWithScriptAddress, bytes32[] memory submissionTokens) = new QuarkOperationHelper( ).newReplayableOpWithCalldata( - aliceWallet, - getMessageDetails, - abi.encodeWithSignature("getMsgSenderAndValue()"), - ScriptType.ScriptAddress, - 2 + aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgDetails()"), ScriptType.ScriptAddress, 2 ); address scriptAddress = opWithScriptAddress.scriptAddress; (uint8 v, bytes32 r, bytes32 s) = @@ -236,7 +234,7 @@ contract QuarkWalletTest is Test { bytes memory getMessageDetails = new YulHelper().getCode("GetMessageDetails.sol/GetMessageDetails.json"); bytes32 nonce = new QuarkOperationHelper().semiRandomNonce(nonceManager, aliceWalletExecutable); address scriptAddress = codeJar.saveCode(getMessageDetails); - bytes memory call = abi.encodeWithSignature("getMsgSenderAndValue()"); + bytes memory call = abi.encodeWithSignature("getMsgDetails()"); vm.startPrank(aliceAccount); @@ -291,7 +289,7 @@ contract QuarkWalletTest is Test { bytes memory getMessageDetails = new YulHelper().getCode("GetMessageDetails.sol/GetMessageDetails.json"); QuarkWallet.QuarkOperation memory op = new QuarkOperationHelper().newBasicOpWithCalldata( - aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgSenderAndValue()"), ScriptType.ScriptSource + aliceWallet, getMessageDetails, abi.encodeWithSignature("getMsgDetails()"), ScriptType.ScriptSource ); (uint8 v, bytes32 r, bytes32 s) = new SignatureHelper().signOp(alicePrivateKey, aliceWallet, op); diff --git a/test/quark-proxy/QuarkWalletProxyFactory.t.sol b/test/quark-proxy/QuarkWalletProxyFactory.t.sol index 1af416b4..383b9a5e 100644 --- a/test/quark-proxy/QuarkWalletProxyFactory.t.sol +++ b/test/quark-proxy/QuarkWalletProxyFactory.t.sol @@ -447,7 +447,7 @@ contract QuarkWalletProxyFactoryTest is Test { QuarkWallet.QuarkOperation memory op = QuarkWallet.QuarkOperation({ scriptAddress: getMessageDetailsAddress, scriptSources: scriptSources, - scriptCalldata: abi.encodeWithSignature("getMsgSenderAndValue()"), + scriptCalldata: abi.encodeWithSignature("getMsgDetails()"), nonce: nonce, isReplayable: false, expiry: block.timestamp + 1000 @@ -463,8 +463,9 @@ contract QuarkWalletProxyFactoryTest is Test { emit WalletDeploy(alice, address(0), aliceWallet, bytes32(0)); bytes memory result = factory.createAndExecute(alice, address(0), op, v, r, s); - (address msgSender, uint256 msgValue) = abi.decode(result, (address, uint256)); - assertEq(msgSender, address(aliceWallet)); + (address msgSender, address addressThis, uint256 msgValue) = abi.decode(result, (address, address, uint256)); + assertEq(msgSender, address(factory)); + assertEq(addressThis, address(aliceWallet)); assertEq(msgValue, 0); // uses up the operation's nonce @@ -483,7 +484,7 @@ contract QuarkWalletProxyFactoryTest is Test { QuarkWallet.QuarkOperation memory op = QuarkWallet.QuarkOperation({ scriptAddress: getMessageDetailsAddress, scriptSources: new bytes[](0), - scriptCalldata: abi.encodeWithSignature("getMsgSenderAndValue()"), + scriptCalldata: abi.encodeWithSignature("getMsgDetails()"), nonce: nonce, isReplayable: false, expiry: block.timestamp + 1000 @@ -514,8 +515,9 @@ contract QuarkWalletProxyFactoryTest is Test { emit WalletDeploy(alice, address(0), aliceWallet, salt); bytes memory result = factory.createAndExecute(alice, address(0), salt, op, v, r, s); - (address msgSender, uint256 msgValue) = abi.decode(result, (address, uint256)); - assertEq(msgSender, address(aliceWallet)); + (address msgSender, address addressThis, uint256 msgValue) = abi.decode(result, (address, address, uint256)); + assertEq(msgSender, address(factory)); + assertEq(addressThis, address(aliceWallet)); assertEq(msgValue, 0); // uses up the operation's nonce