Audit preview/yneigen withdrawals

.gas-snapshot

@@ -0,0 +1,2 @@
+ynLSDeDepositAdapterTest:testTotalAssetsGas() (gas: 760757)
+ynLSDeWithdrawalsTest:testTotalAssetsGas() (gas: 760757)

.gitmodules

@@ -17,4 +17,7 @@
 [submodule "lib/eigenlayer-contracts"]
 	path = lib/eigenlayer-contracts
 	url = https://github.com/layr-labs/eigenlayer-contracts
-	commit = bda003385c5fec59e35196dc14d01f17d1eb7001
+	commit = bda003385c5fec59e35196dc14d01f17d1eb7001
+[submodule "lib/solady"]
+	path = lib/solady
+	url = https://github.com/Vectorized/solady

src/interfaces/IAssetRegistry.sol

@@ -15,11 +15,13 @@ interface IAssetRegistry {
         AssetStatus status;
     }
 
+    function getAssets() external view returns (IERC20[] memory);
     function assetData(IERC20 asset) external view returns (AssetData memory);
     function addAsset(IERC20 asset) external;
     function disableAsset(IERC20 asset) external;
     function deleteAsset(IERC20 asset) external;
     function totalAssets() external view returns (uint256);
     function convertToUnitOfAccount(IERC20 asset, uint256 amount) external view returns (uint256);
+    function convertFromUnitOfAccount(IERC20 asset, uint256 amount) external view returns (uint256);
     function assetIsSupported(IERC20 asset) external view returns (bool);
 }

src/interfaces/IEigenStrategyManager.sol

@@ -5,7 +5,8 @@ import {IStrategy} from "lib/eigenlayer-contracts/src/contracts/interfaces/IStra
 import {IERC20} from "lib/openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
 import {IwstETH} from "src/external/lido/IwstETH.sol";
 import {IERC4626} from "@openzeppelin/contracts/interfaces/IERC4626.sol";
-
+import {IynEigen} from "src/interfaces/IynEigen.sol";
+import {ITokenStakingNodesManager} from "src/interfaces/ITokenStakingNodesManager.sol";
 
 interface IEigenStrategyManager {
 
@@ -21,4 +22,6 @@ interface IEigenStrategyManager {
     function woETH() external view returns (IERC4626);
     function oETH() external view returns (IERC20);
     function stETH() external view returns (IERC20);
+    function ynEigen() external view returns (IynEigen);
+    function tokenStakingNodesManager() external view returns (ITokenStakingNodesManager);
 }

src/interfaces/ITokenStakingNode.sol

@@ -5,6 +5,7 @@ import {IERC20} from "lib/openzeppelin-contracts/contracts/token/ERC20/IERC20.so
 import {ISignatureUtils} from "lib/eigenlayer-contracts/src/contracts/interfaces/ISignatureUtils.sol";
 import {ITokenStakingNodesManager} from "src/interfaces/ITokenStakingNodesManager.sol";
 import {IStrategy} from "lib/eigenlayer-contracts/src/contracts/interfaces/IStrategy.sol";
+import {IDelegationManager} from "lib/eigenlayer-contracts/src/contracts/interfaces/IDelegationManager.sol";
 
 interface ITokenStakingNode {
 
@@ -37,4 +38,13 @@ interface ITokenStakingNode {
     ) external;
 
     function undelegate() external;
+
+    function getQueuedSharesAndWithdrawn(IStrategy _strategy, IERC20 _asset) external view returns (uint256, uint256);
+    function queueWithdrawals(IStrategy _strategy, uint256 _shares) external returns (bytes32[] memory _fullWithdrawalRoots);
+    function completeQueuedWithdrawals(uint256 _nonce, uint32 _startBlock, uint256 _shares, IStrategy _strategy, uint256[] memory _middlewareTimesIndexes) external;
+
+    function deallocateTokens(IERC20 _token, uint256 _amount) external;
+
+    function queuedShares(IStrategy _strategy) external view returns (uint256);
+    function withdrawn(IERC20 _token) external view returns (uint256);
 }

src/interfaces/ITokenStakingNodesManager.sol

@@ -24,4 +24,6 @@ interface ITokenStakingNodesManager {
     function hasYieldNestStrategyManagerRole(address) external view returns (bool);
 
     function getNodeById(uint256 nodeId) external view returns (ITokenStakingNode);
+
+    function yieldNestStrategyManager() external view returns (address);
 }

src/interfaces/IWrapper.sol

@@ -0,0 +1,27 @@
+// SPDX-License-Identifier: BSD 3-Clause License
+pragma solidity ^0.8.24;
+
+import {IERC20} from "lib/openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
+
+interface IWrapper {
+
+    /// @notice Wraps the given amount of the given token.
+    /// @param _amount The amount to wrap.
+    /// @param _token The token to wrap.
+    /// @return The amount of wrapped tokens and the wrapped token.
+    function wrap(uint256 _amount, IERC20 _token) external returns (uint256, IERC20);
+
+    /// @notice Unwraps the given amount of the given token.
+    /// @param _amount The amount to unwrap.
+    /// @param _token The token to unwrap.
+    /// @return The amount of unwrapped tokens and the unwrapped token.
+    function unwrap(uint256 _amount, IERC20 _token) external returns (uint256, IERC20);
+
+    /// @notice Converts the user's underlying asset amount to the equivalent user asset amount.
+    /// @dev This function handles the conversion for wrapped staked ETH (wstETH) and wrapped other ETH (woETH),
+    ///      returning the equivalent amount in the respective wrapped token.
+    /// @param _asset The ERC20 token for which the conversion is being made.
+    /// @param _userUnderlyingView The amount of the underlying asset.
+    /// @return The equivalent amount in the user asset denomination.
+    function toUserAssetAmount(IERC20 _asset, uint256 _userUnderlyingView) external view returns (uint256);
+}

src/interfaces/IYieldNestStrategyManager.sol

@@ -2,9 +2,23 @@
 pragma solidity ^0.8.24;
 
 import {IERC20} from "lib/openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
+import {ITokenStakingNodesManager} from "src/interfaces/ITokenStakingNodesManager.sol";
+import {IRedemptionAssetsVault} from "src/interfaces/IRedemptionAssetsVault.sol";
+
+interface IRedemptionAssetsVaultExt is IRedemptionAssetsVault {
+    function deposit(uint256 amount, address asset) external;
+    function balances(address asset) external view returns (uint256 amount);
+}
 
 interface IYieldNestStrategyManager {
 
+    struct WithdrawalAction {
+        uint256 nodeId;
+        uint256 amountToReinvest;
+        uint256 amountToQueue;
+        address asset;
+    }
+
     function getStakedAssetsBalances(
         IERC20[] calldata assets
     ) external view returns (uint256[] memory stakedBalances);
@@ -13,4 +27,9 @@ interface IYieldNestStrategyManager {
 
     function supportsAsset(IERC20 asset) external view returns (bool);
 
+    function tokenStakingNodesManager() external view returns (ITokenStakingNodesManager);
+
+    function processPrincipalWithdrawals(WithdrawalAction[] calldata _actions) external;
+
+    function redemptionAssetsVault() external view returns (IRedemptionAssetsVaultExt);
 }

src/interfaces/IynEigen.sol

@@ -17,11 +17,20 @@ interface IynEigen is IERC20 {
 
     function previewDeposit(IERC20 asset, uint256 amount) external view returns (uint256);
 
+    function burn(uint256 amount) external;
+
+    function previewRedeem(IERC20 asset, uint256 shares) external view returns (uint256 assets);
+    function previewRedeem(uint256 shares) external view returns (uint256);
+
+    function convertToAssets(IERC20 asset, uint256 shares) external view returns (uint256);
+
     function retrieveAssets(
         IERC20[] calldata assetsToRetrieve,
         uint256[] calldata amounts
     ) external;
 
+    function processWithdrawn(uint256 _amount, address _asset) external;
+
     function assetBalances(IERC20[] calldata assetsArray) external view returns (uint256[] memory balances);
     function assetBalance(IERC20 asset) external view returns (uint256 balance);
 

src/ynEIGEN/AssetRegistry.sol

@@ -10,7 +10,7 @@ import {IRateProvider} from "src/interfaces/IRateProvider.sol";
 import {IAssetRegistry} from "src/interfaces/IAssetRegistry.sol";
 import "lib/openzeppelin-contracts-upgradeable/contracts/access/AccessControlUpgradeable.sol";
 import {IYieldNestStrategyManager} from "src/interfaces/IYieldNestStrategyManager.sol";
-
+import {IRedemptionAssetsVaultExt} from "src/interfaces/IYieldNestStrategyManager.sol";
 
 interface IAssetRegistryEvents {
     event AssetAdded(address indexed asset);
@@ -286,8 +286,10 @@ interface IAssetRegistryEvents {
             revert LengthMismatch(assetsCount, stakedAssetBalances.length);
         }
 
+        IRedemptionAssetsVaultExt redemptionAssetsVault = strategyManager.redemptionAssetsVault();
         for (uint256 i = 0; i < assetsCount; i++) {
             assetBalances[i] += stakedAssetBalances[i];
+            if (address(redemptionAssetsVault) != address(0)) assetBalances[i] += redemptionAssetsVault.balances(address(assets[i]));
         }
     }
 
@@ -315,6 +317,14 @@ interface IAssetRegistryEvents {
             : assetRate * amount / 1e18;
     }
 
+    function convertFromUnitOfAccount(IERC20 asset, uint256 amount) public view returns (uint256) {
+        uint256 assetRate = rateProvider.rate(address(asset));
+        uint8 assetDecimals = IERC20Metadata(address(asset)).decimals();
+        return assetDecimals != 18
+            ? amount * (10 ** assetDecimals) / assetRate
+            : amount * 1e18 / assetRate;
+    }
+
     //--------------------------------------------------------------------------------------
     //----------------------------------  PAUSING  -----------------------------------------
     //--------------------------------------------------------------------------------------