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

4
contracts/staking/RewardDistributor.sol
View File

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