fix: check for zero address in setConnected (#47)

zeta-chain · Feb 5, 2025 · 32d34c7 · 32d34c7
1 parent 38a6a05
commit 32d34c7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/contracts/nft/contracts/zetachain/UniversalNFTCore.sol b/contracts/nft/contracts/zetachain/UniversalNFTCore.sol
@@ -104,6 +104,8 @@ abstract contract UniversalNFTCore is
         address zrc20,
         address contractAddress
     ) external onlyOwner {
+        if (zrc20 == address(0)) revert InvalidAddress();
+        if (contractAddress == address(0)) revert InvalidAddress();
         connected[zrc20] = contractAddress;
         emit SetConnected(zrc20, contractAddress);
     }

diff --git a/contracts/token/contracts/zetachain/UniversalTokenCore.sol b/contracts/token/contracts/zetachain/UniversalTokenCore.sol
@@ -102,6 +102,8 @@ abstract contract UniversalTokenCore is
         address zrc20,
         address contractAddress
     ) external onlyOwner {
+        if (zrc20 == address(0)) revert InvalidAddress();
+        if (contractAddress == address(0)) revert InvalidAddress();
         connected[zrc20] = contractAddress;
         emit SetConnected(zrc20, contractAddress);
     }