BUGFIX: delegate correct amount of shares in `EigenPod.restakeBeaconC…

…hainETH`
Layr-Labs · Oct 6, 2023 · 5dacb7d · 5dacb7d
1 parent 9235ece
commit 5dacb7d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/contracts/pods/EigenPodManager.sol b/src/contracts/pods/EigenPodManager.sol
@@ -123,11 +123,11 @@ contract EigenPodManager is
         address podOwner,
         uint256 amountWei
     ) external onlyEigenPod(podOwner) onlyNotFrozen(podOwner) nonReentrant {
-        _addShares(podOwner, amountWei);
+        uint256 sharesAddedAboveZero = _addShares(podOwner, amountWei);
         delegationManager.increaseDelegatedShares({
             staker: podOwner,
             strategy: beaconChainETHStrategy,
-            shares: amountWei
+            shares: sharesAddedAboveZero
         });
         emit BeaconChainETHDeposited(podOwner, amountWei);
     }

diff --git a/src/test/EigenPod.t.sol b/src/test/EigenPod.t.sol
@@ -1183,7 +1183,7 @@ contract EigenPodTests is ProofParsing, EigenPodPausingConstants {
             "withdrawableRestakedExecutionLayerGwei not decremented correctly");
     }
 */
-    function _verifyEigenPodBalanceSharesInvariant(address podowner, IEigenPod pod, bytes32 validatorPubkeyHash) internal {
+    function _verifyEigenPodBalanceSharesInvariant(address podowner, IEigenPod pod, bytes32 validatorPubkeyHash) internal view {
         int256 shares = eigenPodManager.podOwnerShares(podowner);
         uint64 withdrawableRestakedExecutionLayerGwei = pod.withdrawableRestakedExecutionLayerGwei();