crypto/contracts-imtbl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改pause和unpause的执行权限， 可以跳过timelock执行

Browse Source
This commit is contained in:
CounterFire2023 2024-09-05 17:02:38 +08:00
parent 24ee9282e9
commit 8d1bf19299
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
contracts/activity/CECDistributor.sol
View File

@ -6,13 +6,14 @@ import {Pausable} from "@openzeppelin/contracts/security/Pausable.sol";
import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import {SafeERC20} from "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
import {Ownable} from "@openzeppelin/contracts/access/Ownable.sol";
import {Governable} from "../core/Governable.sol";
/**
* @title CECDistributor
* @dev CECDistributor is a contract for distributing CEC token with unlock time
* after all data is set, transfer owner to timelock contract
*/
contract CECDistributor is ReentrancyGuard, Pausable, Ownable {
contract CECDistributor is ReentrancyGuard, Pausable, Ownable, Governable {
using SafeERC20 for IERC20;
mapping(address account => uint256 amount) public balanceMap;
@ -29,17 +30,31 @@ contract CECDistributor is ReentrancyGuard, Pausable, Ownable {
cecToken = IERC20(_cecToken);
unlockTime = _unlockTime;
}
/**
* @dev Throws if called by any account other than the owner or gov.
*/
modifier ownerOrGov() {
require(msg.sender == owner() || msg.sender == gov, "CECDistributor: forbidden");
_;
}
function setGov(address _gov) external override onlyOwner {
gov = _gov;
}
/**
* @dev update pause state
* When encountering special circumstances that require an emergency pause of the contract,
* the pause function can be called by the gov account to quickly pause the contract and minimize losses.
*/
function pause() external onlyOwner {
function pause() external ownerOrGov {
_pause();
}
/**
* @dev update unpause state
*/
function unpause() external onlyOwner {
function unpause() external ownerOrGov {
_unpause();
}