[NES-267] [Fix:] [N7] add check for the validity of the componentToken (

#111) * add check for the validity of the componentToken * formatting
plumenetwork · Dec 6, 2024 · 15d06a8 · 15d06a8
1 parent ba7ddbe
commit 15d06a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nest/src/AggregateToken.sol b/nest/src/AggregateToken.sol
@@ -201,6 +201,10 @@ contract AggregateToken is ComponentToken, IAggregateToken, ERC1155Holder {
         IComponentToken componentToken,
         uint256 amount
     ) external nonReentrant onlyRole(ADMIN_ROLE) {
+        // Verify the componentToken is in componentTokenMap
+        if (!_getAggregateTokenStorage().componentTokenMap[componentToken]) {
+            revert ComponentTokenNotListed(componentToken);
+        }
         IERC20(componentToken.asset()).approve(address(componentToken), amount);
     }