Instadapp/aave-protocol-v2
1
0
Fork 0
mirror of https://github.com/Instadapp/aave-protocol-v2.git synced 2024-07-29 21:47:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Replace _safeRewardsTransfer by IERC20 safeTransfer directly

Browse Source
This commit is contained in:
Lasse Herskind 2021-06-01 11:04:52 +02:00
parent f42b7590f4
commit 7eb610c890
1 changed files with 1 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
contracts/protocol/tokenization/StaticATokenLM.sol
View File

@ -457,7 +457,6 @@ contract StaticATokenLM is ERC20 {
* @dev Claims rewards from the `_incentivesController` and update `accRewardstokenPerShare`
*/
function updateRewards() public {
// Only need to claim rewards when once per block. Check with the _incentivesController
if (block.number > lastRewardBlock) {
lastRewardBlock = block.number;
uint256 _supply = totalSupply();
@ -496,26 +495,11 @@ contract StaticATokenLM is ERC20 {
uint256 reward = _getClaimableRewards(user, balance); // Remember that this is converting to wad
if (reward > 0) {
unclaimedRewards[user] = 0;
_safeRewardTransfer(user, reward);
IERC20(currentRewardToken).safeTransfer(user, reward);
_updateRewardDebt(user, balance);
}
}
/**
* @dev Safe transfer of the rewards, taking into account rounding errors to
* allow the last user to retrieve funds if 1 wei off etc
* @param to The address to receive rewards
* @param amount The amount of rewards to receive in WAD
*/
function _safeRewardTransfer(address to, uint256 amount) internal {
uint256 rewardBal = IERC20(currentRewardToken).balanceOf(address(this));
if (amount > rewardBal) {
IERC20(currentRewardToken).safeTransfer(to, rewardBal);
} else {
IERC20(currentRewardToken).safeTransfer(to, amount);
}
}
/**
* @dev Update the rewardDebt for a user with balance as his balance
* @param user The user to update