fix: make sure _veToken isn't zero

YAcademy-Residents · Mar 26, 2022 · eb44ee4 · eb44ee4
1 parent 4c619f1
commit eb44ee4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/contracts/Gauge.sol b/contracts/Gauge.sol
@@ -105,9 +105,10 @@ contract Gauge is BaseGauge, IGauge {
         );
     }
 
-    function setVe(address _ve) external onlyOwner {
-        veToken = _ve;
-        emit UpdatedVeToken(_ve);
+    function setVe(address _veToken) external onlyOwner {
+        require(address(_veToken) != address(0x0), "_veToken 0x0 address");
+        veToken = _veToken;
+        emit UpdatedVeToken(_veToken);
     }
 
     /** @return total of the staked vault token

diff --git a/contracts/VeYfiRewards.sol b/contracts/VeYfiRewards.sol
@@ -23,12 +23,14 @@ contract VeYfiRewards is BaseGauge {
         address _owner
     ) {
         __initialize(_rewardToken, _owner);
+        require(address(_veToken) != address(0x0), "_veToken 0x0 address");
         veToken = _veToken;
     }
 
-    function setVe(address _ve) external onlyOwner {
-        veToken = _ve;
-        emit UpdatedVeToken(_ve);
+    function setVe(address _veToken) external onlyOwner {
+        require(address(_veToken) != address(0x0), "_veToken 0x0 address");
+        veToken = _veToken;
+        emit UpdatedVeToken(_veToken);
     }
 
     function _updateReward(address account) internal override {