Merge pull request #70 from ionicprotocol/feat/pause-guardian

Feat/pause guardian

Author: rhlsthrm

contracts/GlobalPauser.sol

@@ -0,0 +1,54 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.0;
+
+import { IonicComptroller } from "./compound/ComptrollerInterface.sol";
+import { ICErc20 } from "./compound/CTokenInterfaces.sol";
+import { Ownable2Step } from "openzeppelin-contracts/contracts/access/Ownable2Step.sol";
+
+interface IPoolDirectory {
+  struct Pool {
+    string name;
+    address creator;
+    address comptroller;
+    uint256 blockPosted;
+    uint256 timestampPosted;
+  }
+
+  function getActivePools() external view returns (uint256, Pool[] memory);
+}
+
+contract GlobalPauser is Ownable2Step {
+  IPoolDirectory public poolDirectory;
+  mapping(address => bool) public pauseGuardian;
+
+  modifier onlyPauseGuardian() {
+    require(pauseGuardian[msg.sender], "!guardian");
+    _;
+  }
+
+  constructor(address _poolDirectory) Ownable2Step() {
+    poolDirectory = IPoolDirectory(_poolDirectory);
+  }
+
+  function setPauseGuardian(address _pauseGuardian, bool _isPauseGuardian) external onlyOwner {
+    pauseGuardian[_pauseGuardian] = _isPauseGuardian;
+  }
+
+  function pauseAll() external onlyPauseGuardian {
+    (, IPoolDirectory.Pool[] memory pools) = poolDirectory.getActivePools();
+    for (uint256 i = 0; i < pools.length; i++) {
+      ICErc20[] memory markets = IonicComptroller(pools[i].comptroller).getAllMarkets();
+      for (uint256 j = 0; j < markets.length; j++) {
+        bool isPaused = IonicComptroller(pools[i].comptroller).borrowGuardianPaused(address(markets[j]));
+        if (!isPaused) {
+          IonicComptroller(pools[i].comptroller)._setBorrowPaused(markets[j], true);
+        }
+
+        isPaused = IonicComptroller(pools[i].comptroller).mintGuardianPaused(address(markets[j]));
+        if (!isPaused) {
+          IonicComptroller(pools[i].comptroller)._setMintPaused(markets[j], true);
+        }
+      }
+    }
+  }
+}

contracts/test/GlobalPauser.t.sol

@@ -0,0 +1,88 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity >=0.8.0;
+
+import { ERC20 } from "solmate/tokens/ERC20.sol";
+import { TransparentUpgradeableProxy } from "@openzeppelin/contracts/proxy/transparent/TransparentUpgradeableProxy.sol";
+
+import "./config/BaseTest.t.sol";
+import { IonicComptroller } from "../compound/ComptrollerInterface.sol";
+import { IComptroller } from "../external/compound/IComptroller.sol";
+import { GlobalPauser } from "../GlobalPauser.sol";
+import { PoolDirectory } from "../PoolDirectory.sol";
+import { ICErc20 } from "../compound/CTokenInterfaces.sol";
+
+import "forge-std/console.sol";
+
+contract GlobalPauserTest is BaseTest {
+  address public poolDirectory = 0x39C353Cf9041CcF467A04d0e78B63d961E81458a;
+  address public pauseGuardian = 0xD9677b0eeafdCe6BF322d9774Bb65B1f42cF0404;
+  address public multisig = 0x8Fba84867Ba458E7c6E2c024D2DE3d0b5C3ea1C2;
+  GlobalPauser public pauser; // = GlobalPauser(0xe646D8Be18e545244C5E79F121202f75FA3880c8);
+
+  function afterForkSetUp() internal override {
+    super.afterForkSetUp();
+    pauser = new GlobalPauser(poolDirectory);
+    pauser.setPauseGuardian(pauseGuardian, true);
+    (, PoolDirectory.Pool[] memory pools) = PoolDirectory(poolDirectory).getActivePools();
+    for (uint256 i = 0; i < pools.length; i++) {
+      vm.prank(IonicComptroller(pools[i].comptroller).admin());
+      IonicComptroller(pools[i].comptroller)._setPauseGuardian(address(pauser));
+    }
+  }
+
+  function testPauseNotGuardian(address sender) public debuggingOnly forkAtBlock(MODE_MAINNET, 9269895) {
+    vm.assume(sender != pauseGuardian);
+    vm.expectRevert(bytes("!guardian"));
+    pauser.pauseAll();
+  }
+
+  function testPauseAllMode() public debuggingOnly forkAtBlock(MODE_MAINNET, 9269895) {
+    (, PoolDirectory.Pool[] memory pools) = PoolDirectory(poolDirectory).getActivePools();
+    for (uint256 i = 0; i < pools.length; i++) {
+      ICErc20[] memory markets = IonicComptroller(pools[i].comptroller).getAllMarkets();
+      for (uint256 j = 0; j < markets.length; j++) {
+        bool isPaused = IonicComptroller(pools[i].comptroller).borrowGuardianPaused(address(markets[j]));
+        assertEq(isPaused, false);
+        isPaused = IonicComptroller(pools[i].comptroller).mintGuardianPaused(address(markets[j]));
+        assertEq(isPaused, false);
+      }
+    }
+    vm.prank(pauseGuardian);
+    pauser.pauseAll();
+    for (uint256 i = 0; i < pools.length; i++) {
+      ICErc20[] memory markets = IonicComptroller(pools[i].comptroller).getAllMarkets();
+      for (uint256 j = 0; j < markets.length; j++) {
+        bool isPaused = IonicComptroller(pools[i].comptroller).borrowGuardianPaused(address(markets[j]));
+        assertEq(isPaused, true);
+        isPaused = IonicComptroller(pools[i].comptroller).mintGuardianPaused(address(markets[j]));
+        assertEq(isPaused, true);
+      }
+    }
+  }
+
+  function testPauseAllBase() public debuggingOnly forkAtBlock(BASE_MAINNET, 15970403) {
+    address _poolDirectory = 0xE1A3006be645a80F206311d9f18C866c204bA02f;
+    pauser = GlobalPauser(0x48F0F46F56C2Ca5def59fd673fF69495b7272Eb0);
+    (, PoolDirectory.Pool[] memory pools) = PoolDirectory(_poolDirectory).getActivePools();
+    for (uint256 i = 0; i < pools.length; i++) {
+      ICErc20[] memory markets = IonicComptroller(pools[i].comptroller).getAllMarkets();
+      for (uint256 j = 0; j < markets.length; j++) {
+        bool isPaused = IonicComptroller(pools[i].comptroller).borrowGuardianPaused(address(markets[j]));
+        assertEq(isPaused, false);
+        isPaused = IonicComptroller(pools[i].comptroller).mintGuardianPaused(address(markets[j]));
+        assertEq(isPaused, false);
+      }
+    }
+    vm.prank(pauseGuardian);
+    pauser.pauseAll();
+    for (uint256 i = 0; i < pools.length; i++) {
+      ICErc20[] memory markets = IonicComptroller(pools[i].comptroller).getAllMarkets();
+      for (uint256 j = 0; j < markets.length; j++) {
+        bool isPaused = IonicComptroller(pools[i].comptroller).borrowGuardianPaused(address(markets[j]));
+        assertEq(isPaused, true);
+        isPaused = IonicComptroller(pools[i].comptroller).mintGuardianPaused(address(markets[j]));
+        assertEq(isPaused, true);
+      }
+    }
+  }
+}

contracts/test/config/BaseTest.t.sol

@@ -21,6 +21,7 @@ abstract contract BaseTest is Test {
   uint128 constant LINEA_MAINNET = 59144;
   uint128 constant ZKEVM_MAINNET = 1101;
   uint128 constant MODE_MAINNET = 34443;
+  uint128 constant BASE_MAINNET = 8453;
 
   // taken from ERC1967Upgrade
   bytes32 internal constant _ADMIN_SLOT = 0xb53127684a568b3173ae13b9f8a6016e243e63b6e8ee1178d6a717850b5d6103;