add check for the validity of the componentToken

plumenetwork · Dec 5, 2024 · f4deb45 · f4deb45
1 parent ae513f7
commit f4deb45
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 our list
+        if (!_getAggregateTokenStorage().componentTokenMap[componentToken]) {
+            revert ComponentTokenNotListed(componentToken);
+        }
         IERC20(componentToken.asset()).approve(address(componentToken), amount);
     }