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

distribute时检查分发的token是否超过balance

Browse Source
This commit is contained in:
CounterFire2023 2024-08-27 15:24:09 +08:00
parent 8a3d4313a3
commit 34e2977c2d
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
contracts/staking/RewardDistributor.sol
View File

@ -79,10 +79,8 @@ contract RewardDistributor is IRewardDistributor, ReentrancyGuard, Governable {
uint256 tokenAmount = amount * _amount / (10**_decimals);
uint256 balance = IERC20(rewardToken).balanceOf(address(this));
if (tokenAmount > balance) {
tokenAmount = balance;
}
require(tokenAmount >= balance, "RewardDistributor: insufficient balance");
IERC20(rewardToken).safeTransfer(msg.sender, tokenAmount);
emit Distribute(tokenAmount);