From 78abd58e7c9d35092aeb509f6cfc6b8454bf7e38 Mon Sep 17 00:00:00 2001 From: Denis Fadeev Date: Thu, 30 Jan 2025 18:40:27 +0300 Subject: [PATCH 1/2] fix: check for zero address in setConnected --- contracts/nft/contracts/zetachain/UniversalNFTCore.sol | 1 + contracts/token/contracts/zetachain/UniversalTokenCore.sol | 1 + 2 files changed, 2 insertions(+) diff --git a/contracts/nft/contracts/zetachain/UniversalNFTCore.sol b/contracts/nft/contracts/zetachain/UniversalNFTCore.sol index 7c81233..f6a982c 100644 --- a/contracts/nft/contracts/zetachain/UniversalNFTCore.sol +++ b/contracts/nft/contracts/zetachain/UniversalNFTCore.sol @@ -104,6 +104,7 @@ abstract contract UniversalNFTCore is address zrc20, address contractAddress ) external onlyOwner { + if (zrc20 == 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 index c49ee3b..20e97a8 100644 --- a/contracts/token/contracts/zetachain/UniversalTokenCore.sol +++ b/contracts/token/contracts/zetachain/UniversalTokenCore.sol @@ -102,6 +102,7 @@ abstract contract UniversalTokenCore is address zrc20, address contractAddress ) external onlyOwner { + if (zrc20 == address(0)) revert InvalidAddress(); connected[zrc20] = contractAddress; emit SetConnected(zrc20, contractAddress); } From 2c95dfc3c466c96eee5f707c8af29d34ee062655 Mon Sep 17 00:00:00 2001 From: Denis Fadeev Date: Thu, 30 Jan 2025 18:42:02 +0300 Subject: [PATCH 2/2] fix: check for zero address in setConnected --- contracts/nft/contracts/zetachain/UniversalNFTCore.sol | 1 + contracts/token/contracts/zetachain/UniversalTokenCore.sol | 1 + 2 files changed, 2 insertions(+) diff --git a/contracts/nft/contracts/zetachain/UniversalNFTCore.sol b/contracts/nft/contracts/zetachain/UniversalNFTCore.sol index f6a982c..8ede9ed 100644 --- a/contracts/nft/contracts/zetachain/UniversalNFTCore.sol +++ b/contracts/nft/contracts/zetachain/UniversalNFTCore.sol @@ -105,6 +105,7 @@ abstract contract UniversalNFTCore is 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 index 20e97a8..5873c47 100644 --- a/contracts/token/contracts/zetachain/UniversalTokenCore.sol +++ b/contracts/token/contracts/zetachain/UniversalTokenCore.sol @@ -103,6 +103,7 @@ abstract contract UniversalTokenCore is address contractAddress ) external onlyOwner { if (zrc20 == address(0)) revert InvalidAddress(); + if (contractAddress == address(0)) revert InvalidAddress(); connected[zrc20] = contractAddress; emit SetConnected(zrc20, contractAddress); }