Add missing isSupported check to bulkWithdraw

Se7en-Seas · Apr 4, 2024 · 59a1804 · 59a1804
1 parent 61d9a48
commit 59a1804
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/base/Roles/TellerWithMultiAssetSupport.sol b/src/base/Roles/TellerWithMultiAssetSupport.sol
@@ -288,6 +288,7 @@ contract TellerWithMultiAssetSupport is Auth, BeforeTransferHook, ReentrancyGuar
         returns (uint256 shares)
     {
         if (!isSupported[depositAsset]) revert TellerWithMultiAssetSupport__AssetNotSupported();
+
         shares = _erc20Deposit(depositAsset, depositAmount, minimumMint, to);
         emit BulkDeposit(address(depositAsset), depositAmount);
     }
@@ -300,6 +301,8 @@ contract TellerWithMultiAssetSupport is Auth, BeforeTransferHook, ReentrancyGuar
         requiresAuth
         returns (uint256 assetsOut)
     {
+        if (!isSupported[withdrawAsset]) revert TellerWithMultiAssetSupport__AssetNotSupported();
+
         if (shareAmount == 0) revert TellerWithMultiAssetSupport__ZeroShares();
         assetsOut = shareAmount.mulDivDown(accountant.getRateInQuoteSafe(withdrawAsset), ONE_SHARE);
         if (assetsOut < minimumAssets) revert TellerWithMultiAssetSupport__MinimumAssetsNotMet();