Add extra safety check to FarmingPool constructor

1inch · Sep 25, 2023 · 44d7868 · 44d7868
1 parent c320d35
commit 44d7868
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/contracts/FarmingPool.sol b/contracts/FarmingPool.sol
@@ -16,6 +16,7 @@ contract FarmingPool is IFarmingPool, Ownable, ERC20 {
     using Address for address payable;
     using FarmingLib for FarmingLib.Info;
 
+    error SameStakingAndRewardsTokens();
     error ZeroStakingTokenAddress();
     error ZeroRewardsTokenAddress();
     error ZeroDistributorAddress();
@@ -43,6 +44,7 @@ contract FarmingPool is IFarmingPool, Ownable, ERC20 {
             string(abi.encodePacked("farm", stakingToken_.symbol()))
         )
     {
+        if (stakingToken_ == rewardsToken_) revert SameStakingAndRewardsTokens();
         if (address(stakingToken_) == address(0)) revert ZeroStakingTokenAddress();
         if (address(rewardsToken_) == address(0)) revert ZeroRewardsTokenAddress();
         stakingToken = stakingToken_;