fix: sherlock-audit/2024-06-leveraged-vaults-judging#72

notional-finance · Aug 9, 2024 · 69c6280 · 69c6280
1 parent a78f2cc
commit 69c6280
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/contracts/vaults/staking/BaseStakingVault.sol b/contracts/vaults/staking/BaseStakingVault.sol
@@ -10,6 +10,7 @@ import {WithdrawRequestBase, WithdrawRequest, SplitWithdrawRequest} from "../com
 import {BaseStrategyVault, IERC20, NotionalProxy} from "../common/BaseStrategyVault.sol";
 import {ITradingModule, Trade, TradeType} from "@interfaces/trading/ITradingModule.sol";
 import {VaultAccountHealthFactors} from "@interfaces/notional/IVaultController.sol";
+import {ClonedCoolDownHolder} from "@contracts/vaults/staking/protocols/ClonedCoolDownHolder.sol";
 
 struct RedeemParams {
     uint8 dexId;
@@ -267,4 +268,10 @@ abstract contract BaseStakingVault is WithdrawRequestBase, BaseStrategyVault {
     function finalizeWithdrawsManual(address account) external {
         return _finalizeWithdrawsManual(account);
     }
+
+    function rescueTokens(
+        address cooldownHolder, IERC20 token, address receiver, uint256 amount
+    ) external onlyRole(EMERGENCY_EXIT_ROLE) {
+        ClonedCoolDownHolder(cooldownHolder).rescueTokens(token, receiver, amount);
+    }
 }