Resolve: MCC-12

CubFinance · Feb 18, 2022 · a10db84 · a10db84
1 parent 9a029e2
commit a10db84
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/contracts/MasterChef.sol b/contracts/MasterChef.sol
@@ -118,6 +118,9 @@ contract MasterChef is Ownable, ReentrancyGuard {
      * @param newRewardToken The address of the reward token
      */
     constructor(address newPenaltyAddress, uint256 newStartBlock, address newRewardToken) public {
+      require(newPenaltyAddress != address(0), '!address(0)');
+      require(newRewardToken != address(0), '!address(0)');
+
       penaltyAddress = newPenaltyAddress;
       rewardToken = newRewardToken;
 
@@ -664,6 +667,7 @@ contract MasterChef is Ownable, ReentrancyGuard {
      * @param _newPenaltyAddress new penalty address
      */
     function setPenaltyAddress(address _newPenaltyAddress) external onlyOwner {
+      require(_newPenaltyAddress != address(0), '!address(0)');
       penaltyAddress = _newPenaltyAddress;
       emit updatePenaltyAddress(_newPenaltyAddress);
     }