fix: SRS-03 check the address(0) for comptroller address in swapRoute…

…r constructor
VenusProtocol · May 9, 2023 · 48eb87c · 48eb87c
1 parent 774eed1
commit 48eb87c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/contracts/Swap/SwapRouter.sol b/contracts/Swap/SwapRouter.sol
@@ -56,6 +56,9 @@ contract SwapRouter is Ownable2Step, RouterHelper, IPancakeSwapV2Router {
     /// @notice Constructor for the implementation contract. Sets immutable variables.
     /// @custom:oz-upgrades-unsafe-allow constructor
     constructor(address WBNB_, address factory_, address _comptrollerAddress) RouterHelper(WBNB_, factory_) {
+        if (_comptrollerAddress == address(0)) {
+            revert ZeroAddress();
+        }
         comptrollerAddress = _comptrollerAddress;
     }