Skip to content

Commit a744b1c

Browse files
committed
Do not bother with rETHBefore
Sending rETH to this contract means sending it to whatever lucky NO calls depositAndMint next.
1 parent c5b03be commit a744b1c

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

contracts/contract/node/RocketNodeDeposit.sol

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -88,14 +88,11 @@ contract RocketNodeDeposit is RocketBase, RocketNodeDepositInterface {
8888
}
8989

9090
function _mint(uint256 _mintEthAmount) private {
91-
// Save current rETH balance (for calculating how much rETH is minted)
92-
RocketTokenRETHInterface rocketTokenRETH = RocketTokenRETHInterface(getContractAddress("rocketTokenRETH"));
93-
uint256 rETHBefore = rocketTokenRETH.balanceOf(address(this));
9491
// Deposit ETH to mint rETH
95-
RocketDepositPoolInterface rocketDepositPool = RocketDepositPoolInterface(getContractAddress("rocketDepositPool"));
96-
rocketDepositPool.deposit{value: _mintEthAmount}();
92+
RocketDepositPoolInterface(getContractAddress("rocketDepositPool")).deposit{value: _mintEthAmount}();
9793
// Transfer minted rETH back to the caller
98-
rocketTokenRETH.transfer(msg.sender, rocketTokenRETH.balanceOf(address(this)) - rETHBefore);
94+
RocketTokenRETHInterface rocketTokenRETH = RocketTokenRETHInterface(getContractAddress("rocketTokenRETH"));
95+
rocketTokenRETH.transfer(msg.sender, rocketTokenRETH.balanceOf(address(this)));
9996
}
10097

10198
/// @notice Returns true if the given amount is a valid deposit amount

0 commit comments

Comments
 (0)