fix: process request args

imua-xyz · Jun 21, 2024 · fbeb696 · fbeb696
1 parent 6687eed
commit fbeb696
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 11 deletions.
diff --git a/src/core/ClientGatewayLzReceiver.sol b/src/core/ClientGatewayLzReceiver.sol
@@ -115,7 +115,7 @@ abstract contract ClientGatewayLzReceiver is PausableUpgradeable, OAppReceiverUp
             abi.decode(requestPayload, (address, address, uint256));
 
         bool success = (uint8(bytes1(responsePayload[0])) == 1);
-        uint256 lastlyUpdatedPrincipleBalance = uint256(bytes32(responsePayload[1:33]));
+        uint256 lastlyUpdatedPrincipalBalance = uint256(bytes32(responsePayload[1:33]));
 
         if (!success) {
             revert WithdrawShouldNotFailOnExocore(token, withdrawer);
@@ -124,8 +124,8 @@ abstract contract ClientGatewayLzReceiver is PausableUpgradeable, OAppReceiverUp
         if (token == VIRTUAL_STAKED_ETH_ADDRESS) {
             IExoCapsule capsule = _getCapsule(withdrawer);
 
-            capsule.updatePrincipleBalance(lastlyUpdatedPrincipleBalance);
-            capsule.updateWithdrawableBalance(unlockPrincipleAmount);
+            capsule.updatePrincipalBalance(lastlyUpdatedPrincipalBalance);
+            capsule.updateWithdrawableBalance(unlockPrincipalAmount);
         } else {
             IVault vault = _getVault(token);
 

diff --git a/src/core/ExoCapsule.sol b/src/core/ExoCapsule.sol
@@ -196,8 +196,8 @@ contract ExoCapsule is Initializable, ExoCapsuleStorage, IExoCapsule {
         emit NonBeaconChainETHWithdrawn(recipient, amountToWithdraw);
     }
 
-    function updatePrincipleBalance(uint256 lastlyUpdatedPrincipleBalance) external onlyGateway {
-        principleBalance = lastlyUpdatedPrincipleBalance;
+    function updatePrincipalBalance(uint256 lastlyUpdatedPrincipalBalance) external onlyGateway {
+        principleBalance = lastlyUpdatedPrincipalBalance;
 
         emit PrincipalBalanceUpdated(capsuleOwner, lastlyUpdatedPrincipalBalance);
     }

diff --git a/src/core/NativeRestakingController.sol b/src/core/NativeRestakingController.sol
@@ -80,13 +80,12 @@ abstract contract NativeRestakingController is
             capsule.verifyWithdrawalProof(validatorContainer, validatorProof, withdrawalContainer, withdrawalProof);
         if (!partialWithdrawal) {
             // request full withdraw
-            _processRequest(
-                VIRTUAL_STAKED_ETH_ADDRESS,
-                msg.sender,
-                withdrawalAmount,
-                Action.REQUEST_WITHDRAW_PRINCIPLE_FROM_EXOCORE,
-                ""
+            bytes memory actionArgs = abi.encodePacked(
+                bytes32(bytes20(VIRTUAL_STAKED_ETH_ADDRESS)), bytes32(bytes20(msg.sender)), withdrawalAmount
             );
+            bytes memory encodedRequest = abi.encode(VIRTUAL_STAKED_ETH_ADDRESS, msg.sender, withdrawalAmount);
+
+            _processRequest(Action.REQUEST_WITHDRAW_PRINCIPAL_FROM_EXOCORE, actionArgs, encodedRequest);
         }
     }