add zero address check

1inch · Mar 29, 2022 · 5b97ae7 · 5b97ae7
1 parent f41b029
commit 5b97ae7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/contracts/FarmingPool.sol b/contracts/FarmingPool.sol
@@ -32,6 +32,8 @@ contract FarmingPool is IFarmingPool, Ownable, ERC20 {
             string(abi.encodePacked("farm", stakingToken_.symbol()))
         )
     {
+        require(address(stakingToken_) != address(0), "FP: stakingToken is zero");
+        require(address(rewardsToken_) != address(0), "FP: rewardsToken is zero");
         stakingToken = stakingToken_;
         rewardsToken = rewardsToken_;
     }