fix: add paused check and blacklist check for updateAgent (#573)

* feat: add paused check and blacklist check for updateAgent * chores: add comment * chores: add comment
bnb-chain · Jul 19, 2024 · 5ce49d3 · 5ce49d3
1 parent f1f21ce
commit 5ce49d3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/contracts/BC_fusion/StakeHub.sol b/contracts/BC_fusion/StakeHub.sol
@@ -325,7 +325,10 @@ contract StakeHub is System, Initializable, Protectable {
     }
 
     /*----------------- external functions -----------------*/
-    function updateAgent(address newAgent) external validatorExist(msg.sender) {
+    /**
+     * @param newAgent the new agent address of the validator, updating to address(0) means remove the old agent.
+     */
+    function updateAgent(address newAgent) external validatorExist(msg.sender) whenNotPaused notInBlackList {
         if (agentToOperator[newAgent] != address(0)) revert InvalidAgent();
         if (_validatorSet.contains(newAgent)) revert InvalidAgent();