Contract pausing

contracts/Account.sol

@@ -7,13 +7,14 @@ import "./Managed.sol";
 import "./common/UUPSOwnableUpgradeable.sol";
 import "./common/UsingRegistryUpgradeable.sol";
 import "./interfaces/IAccount.sol";
+import "./Pausable.sol";
 
 /**
  * @title A contract that facilitates voting on behalf of StakedCelo.sol.
  * @notice This contract depends on the Manager to decide how to distribute votes and how to
  * keep track of ownership of CELO voted via this contract.
  */
-contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, IAccount {
+contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, IAccount, Pausable {
     /**
      * @notice Used to keep track of a pending withdrawal. A similar data structure
      * exists within LockedGold.sol, but it only keeps track of pending withdrawals
@@ -225,6 +226,14 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         }
     }
 
+    /**
+     * @notice Sets that address permissioned to pause/unpause this contract.
+     * @param _pauser The address that can pause/unpause this contract.
+     */
+    function setPauser(address _pauser) external onlyOwner {
+        _setPauser(_pauser);
+    }
+
     /**
      * @notice Deposits CELO sent via msg.value as unlocked CELO intended as
      * votes for groups.
@@ -238,6 +247,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         external
         payable
         onlyManager
+        onlyWhenNotPaused
     {
         if (groups.length != votes.length) {
             revert GroupsAndVotesArrayLengthsMismatch();
@@ -268,7 +278,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         uint256[] calldata fromVotes,
         address[] calldata toGroups,
         uint256[] calldata toVotes
-    ) external onlyManager {
+    ) external onlyManager onlyWhenNotPaused {
         if (fromGroups.length != fromVotes.length || toGroups.length != toVotes.length) {
             revert GroupsAndVotesArrayLengthsMismatch();
         }
@@ -301,7 +311,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         address beneficiary,
         address[] calldata groups,
         uint256[] calldata withdrawals
-    ) external onlyManager {
+    ) external onlyManager onlyWhenNotPaused {
         if (groups.length != withdrawals.length) {
             revert GroupsAndVotesArrayLengthsMismatch();
         }
@@ -355,7 +365,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         address lesserAfterActiveRevoke,
         address greaterAfterActiveRevoke,
         uint256 index
-    ) external returns (uint256) {
+    ) external onlyWhenNotPaused returns (uint256) {
         uint256 withdrawalAmount = scheduledVotes[group].toWithdrawFor[beneficiary];
         if (withdrawalAmount == 0) {
             revert NoScheduledWithdrawal(beneficiary, group);
@@ -431,7 +441,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         address group,
         address voteLesser,
         address voteGreater
-    ) external {
+    ) external onlyWhenNotPaused {
         IElection election = getElection();
 
         // The amount of unlocked CELO for group that we want to lock and vote with.
@@ -491,7 +501,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         address beneficiary,
         uint256 localPendingWithdrawalIndex,
         uint256 lockedGoldPendingWithdrawalIndex
-    ) external returns (uint256 amount) {
+    ) external onlyWhenNotPaused returns (uint256 amount) {
         (uint256 value, uint256 timestamp) = validatePendingWithdrawalRequest(
             beneficiary,
             localPendingWithdrawalIndex,
@@ -544,7 +554,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         uint256 yesVotes,
         uint256 noVotes,
         uint256 abstainVotes
-    ) external onlyManager {
+    ) external onlyManager onlyWhenNotPaused {
         bool voteResult = getGovernance().votePartially(
             proposalId,
             index,
@@ -697,7 +707,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
             uint256
         )
     {
-        return (1, 2, 0, 0);
+        return (1, 2, 1, 0);
     }
 
     /**
@@ -728,7 +738,7 @@ contract Account is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Managed, I
         address lesserAfterActiveRevoke,
         address greaterAfterActiveRevoke,
         uint256 index
-    ) public {
+    ) public onlyWhenNotPaused {
         (, uint256 revokeAmount) = getAndUpdateToVoteAndToRevoke(group, 0, 0);
 
         if (revokeAmount == 0) {

contracts/DefaultStrategy.sol

@@ -11,12 +11,13 @@ import "./interfaces/IGroupHealth.sol";
 import "./interfaces/IManager.sol";
 import "./interfaces/ISpecificGroupStrategy.sol";
 import "./Managed.sol";
+import "./Pausable.sol";
 
 /**
  * @title DefaultStrategy is responsible for handling any deposit/withdrawal
  * for accounts without any specific strategy.
  */
-contract DefaultStrategy is UUPSOwnableUpgradeable, Managed {
+contract DefaultStrategy is UUPSOwnableUpgradeable, Managed, Pausable {
     using EnumerableSet for EnumerableSet.AddressSet;
     using AddressSortedLinkedList for SortedLinkedList.List;
 
@@ -218,6 +219,14 @@ contract DefaultStrategy is UUPSOwnableUpgradeable, Managed {
         emit SortedFlagUpdated(sorted);
     }
 
+    /**
+     * @notice Sets that address permissioned to pause/unpause this contract.
+     * @param _pauser The address that can pause/unpause this contract.
+     */
+    function setPauser(address _pauser) external onlyOwner {
+        _setPauser(_pauser);
+    }
+
     /**
      * @notice Set this contract's dependencies in the StakedCelo system.
      * @param _account The address of the Account contract.
@@ -285,7 +294,7 @@ contract DefaultStrategy is UUPSOwnableUpgradeable, Managed {
         address group,
         address lesserKey,
         address greaterKey
-    ) external {
+    ) external onlyWhenNotPaused {
         if (!unsortedGroups.contains(group)) {
             revert NotUnsortedGroup();
         }
@@ -347,7 +356,7 @@ contract DefaultStrategy is UUPSOwnableUpgradeable, Managed {
      * @param fromGroup The from group.
      * @param toGroup The to group.
      */
-    function rebalance(address fromGroup, address toGroup) external {
+    function rebalance(address fromGroup, address toGroup) external onlyWhenNotPaused {
         if (!activeGroups.contains(fromGroup)) {
             revert InvalidFromGroup(fromGroup);
         }
@@ -461,7 +470,7 @@ contract DefaultStrategy is UUPSOwnableUpgradeable, Managed {
      * @notice Deactivates an unhealthy group.
      * @param group The group to deactivate if unhealthy.
      */
-    function deactivateUnhealthyGroup(address group) external {
+    function deactivateUnhealthyGroup(address group) external onlyWhenNotPaused {
         if (groupHealth.isGroupValid(group)) {
             revert HealthyGroup(group);
         }
@@ -552,7 +561,7 @@ contract DefaultStrategy is UUPSOwnableUpgradeable, Managed {
             uint256
         )
     {
-        return (1, 1, 0, 0);
+        return (1, 1, 1, 0);
     }
 
     /**

contracts/GroupHealth.sol

@@ -6,11 +6,12 @@ import "@openzeppelin/contracts/utils/structs/EnumerableSet.sol";
 
 import "./common/UsingRegistryUpgradeable.sol";
 import "./common/UUPSOwnableUpgradeable.sol";
+import "./Pausable.sol";
 
 /**
  * @title GroupHealth stores and updates info about validator group health.
  */
-contract GroupHealth is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
+contract GroupHealth is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Pausable {
     using EnumerableSet for EnumerableSet.AddressSet;
 
     /**
@@ -59,6 +60,14 @@ contract GroupHealth is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
         __UsingRegistry_init(_registry);
     }
 
+    /**
+     * @notice Sets that address permissioned to pause/unpause this contract.
+     * @param _pauser The address that can pause/unpause this contract.
+     */
+    function setPauser(address _pauser) external onlyOwner {
+        _setPauser(_pauser);
+    }
+
     /**
      * @notice Returns the storage, major, minor, and patch version of the contract.
      * @return Storage version of the contract.
@@ -76,14 +85,14 @@ contract GroupHealth is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
             uint256
         )
     {
-        return (1, 1, 0, 1);
+        return (1, 1, 1, 0);
     }
 
     /**
      * @notice Updates validator group health.
      * @param group The group address.
      */
-    function updateGroupHealth(address group) public {
+    function updateGroupHealth(address group) public onlyWhenNotPaused {
         IValidators validators = getValidators();
 
         (bool valid, address[] memory members) = _isGroupPartiallyValid(validators, group);
@@ -104,7 +113,10 @@ contract GroupHealth is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      * This array needs to have same length as all (even not elected) members of validator group.
      * Index of not elected member can be any uint256 number.
      */
-    function markGroupHealthy(address group, uint256[] calldata membersElectedIndex) public {
+    function markGroupHealthy(address group, uint256[] calldata membersElectedIndex)
+        public
+        onlyWhenNotPaused
+    {
         if (isGroupValid[group] == true) {
             revert GroupHealthy(group);
         }

contracts/Manager.sol

@@ -12,13 +12,14 @@ import "./interfaces/IVote.sol";
 import "./interfaces/IGroupHealth.sol";
 import "./interfaces/ISpecificGroupStrategy.sol";
 import "./interfaces/IDefaultStrategy.sol";
+import "./Pausable.sol";
 
 /**
  * @title Manages the StakedCelo system, by controlling the minting and burning
  * of stCELO and implementing strategies for voting and unvoting of deposited or
  * withdrawn CELO.
  */
-contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
+contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable, Pausable {
     using EnumerableSet for EnumerableSet.AddressSet;
 
     /**
@@ -223,6 +224,14 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
         emit VoteContractSet(_vote);
     }
 
+    /**
+     * @notice Sets that address permissioned to pause/unpause this contract.
+     * @param _pauser The address that can pause/unpause this contract.
+     */
+    function setPauser(address _pauser) external onlyOwner {
+        _setPauser(_pauser);
+    }
+
     /**
      * @notice Used to withdraw CELO from the system, in exchange for burning
      * stCELO.
@@ -233,7 +242,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      * @dev The funds need to be withdrawn using calls to `Account.withdraw` and
      * `Account.finishPendingWithdrawal`.
      */
-    function withdraw(uint256 stCeloAmount) external {
+    function withdraw(uint256 stCeloAmount) external onlyWhenNotPaused {
         (
             address[] memory groupsWithdrawn,
             uint256[] memory withdrawalsPerGroup
@@ -248,7 +257,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      * @param proposalId The ID of the proposal to revoke from.
      * @param index The index of the proposal ID in `dequeued`.
      */
-    function revokeVotes(uint256 proposalId, uint256 index) external {
+    function revokeVotes(uint256 proposalId, uint256 index) external onlyWhenNotPaused {
         IVote vote = IVote(voteContract);
 
         (uint256 totalYesVotes, uint256 totalNoVotes, uint256 totalAbstainVotes) = vote.revokeVotes(
@@ -265,6 +274,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      */
     function updateHistoryAndReturnLockedStCeloInVoting(address beneficiary)
         external
+        onlyWhenNotPaused
         returns (uint256)
     {
         IVote vote = IVote(voteContract);
@@ -277,7 +287,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      * will be scheduled to be voted for with the Account contract.
      * The CELO will be distributed based on address strategy.
      */
-    function deposit() external payable {
+    function deposit() external payable onlyWhenNotPaused {
         uint256 stCeloAmount = toStakedCelo(msg.value);
         (address[] memory finalGroups, uint256[] memory finalVotes) = distributeVotes(
             msg.value,
@@ -327,7 +337,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
             uint256
         )
     {
-        return (1, 3, 0, 0);
+        return (1, 3, 1, 0);
     }
 
     /**
@@ -351,7 +361,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      * !address(0) = voting for allowed validator group. Group needs to be in allowed
      * @param newStrategy The from account.
      */
-    function changeStrategy(address newStrategy) public {
+    function changeStrategy(address newStrategy) public onlyWhenNotPaused {
         if (
             newStrategy != address(0) &&
             (specificGroupStrategy.isBlockedGroup(newStrategy) ||
@@ -384,7 +394,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      * @param fromGroup The from group.
      * @param toGroup The to group.
      */
-    function rebalance(address fromGroup, address toGroup) public {
+    function rebalance(address fromGroup, address toGroup) public onlyWhenNotPaused {
         if (!defaultStrategy.isActive(toGroup) && specificGroupStrategy.isBlockedGroup(toGroup)) {
             // rebalancing to deactivated/non-existent group is not allowed
             revert InvalidToGroup(toGroup);
@@ -427,7 +437,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
      * @param fromGroup The group to unschedule votes from.
      * @param toGroup The group to schedule votes to.
      */
-    function rebalanceOverflow(address fromGroup, address toGroup) public {
+    function rebalanceOverflow(address fromGroup, address toGroup) public onlyWhenNotPaused {
         if (!defaultStrategy.isActive(toGroup)) {
             revert InvalidToGroup(toGroup);
         }
@@ -491,7 +501,7 @@ contract Manager is UUPSOwnableUpgradeable, UsingRegistryUpgradeable {
         uint256 yesVotes,
         uint256 noVotes,
         uint256 abstainVotes
-    ) public {
+    ) public onlyWhenNotPaused {
         IVote vote = IVote(voteContract);
 
         (