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

feat: split flashoan fees between fees to protocol and fees to LPs

Browse Source
This commit is contained in:
Hadrien Charlanes 2021-06-09 18:12:33 +02:00
parent 4a8d0748da
commit 9cbda77b27
2 changed files with 28 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
contracts/protocol/lendingpool/LendingPool.sol
View File

@ -87,8 +87,9 @@ contract LendingPool is VersionedInitializable, ILendingPool, LendingPoolStorage
function initialize(ILendingPoolAddressesProvider provider) public initializer { function initialize(ILendingPoolAddressesProvider provider) public initializer {
_addressesProvider = provider; _addressesProvider = provider;
_maxStableRateBorrowSizePercent = 2500; _maxStableRateBorrowSizePercent = 2500;
_flashLoanPremiumTotal = 9; _flashLoanPremiumToLP = 9;
_maxNumberOfReserves = 128; _maxNumberOfReserves = 128;
_flashLoanPremiumToProtocol = 0;
} }
/** /**
@ -432,10 +433,13 @@ contract LendingPool is VersionedInitializable, ILendingPool, LendingPoolStorage
address currentAsset; address currentAsset;
address currentATokenAddress; address currentATokenAddress;
uint256 currentAmount; uint256 currentAmount;
uint256 currentPremium; uint256 currentPremiumToLP;
uint256 currentPremiumToProtocol;
uint256 currentAmountPlusPremium; uint256 currentAmountPlusPremium;
uint256 currentAmountPlusPremiumToProtocol;
address debtToken; address debtToken;
uint256 flashloanPremiumTotal; uint256 flashloanPremiumToLP;
uint256 flashloanPremiumToProtocol;
} }
/** /**
@ -469,36 +473,45 @@ contract LendingPool is VersionedInitializable, ILendingPool, LendingPoolStorage
ValidationLogic.validateFlashloan(assets, amounts, _reserves); ValidationLogic.validateFlashloan(assets, amounts, _reserves);
address[] memory aTokenAddresses = new address[](assets.length); address[] memory aTokenAddresses = new address[](assets.length);
uint256[] memory premiums = new uint256[](assets.length); uint256[] memory totalPremiums = new uint256[](assets.length);
vars.receiver = IFlashLoanReceiver(receiverAddress); vars.receiver = IFlashLoanReceiver(receiverAddress);
vars.flashloanPremiumTotal = _authorizedFlashBorrowers[msg.sender] ? 0 : _flashLoanPremiumTotal; (vars.flashloanPremiumToLP, vars.flashloanPremiumToProtocol) = _authorizedFlashBorrowers[
msg.sender
]
? (0, 0)
: (_flashLoanPremiumToLP, _flashLoanPremiumToProtocol);
for (vars.i = 0; vars.i < assets.length; vars.i++) { for (vars.i = 0; vars.i < assets.length; vars.i++) {
aTokenAddresses[vars.i] = _reserves[assets[vars.i]].aTokenAddress; aTokenAddresses[vars.i] = _reserves[assets[vars.i]].aTokenAddress;
premiums[vars.i] = amounts[vars.i].percentMul(vars.flashloanPremiumTotal); vars.currentPremiumToLP = amounts[vars.i].percentMul(vars.flashloanPremiumToLP);
vars.currentPremiumToProtocol = amounts[vars.i].percentMul(vars.flashloanPremiumToProtocol);
totalPremiums[vars.i] = vars.currentPremiumToLP.add(vars.currentPremiumToProtocol);
IAToken(aTokenAddresses[vars.i]).transferUnderlyingTo(receiverAddress, amounts[vars.i]); IAToken(aTokenAddresses[vars.i]).transferUnderlyingTo(receiverAddress, amounts[vars.i]);
} }
require( require(
vars.receiver.executeOperation(assets, amounts, premiums, msg.sender, params), vars.receiver.executeOperation(assets, amounts, totalPremiums, msg.sender, params),
Errors.LP_INVALID_FLASH_LOAN_EXECUTOR_RETURN Errors.LP_INVALID_FLASH_LOAN_EXECUTOR_RETURN
); );
for (vars.i = 0; vars.i < assets.length; vars.i++) { for (vars.i = 0; vars.i < assets.length; vars.i++) {
vars.currentAsset = assets[vars.i]; vars.currentAsset = assets[vars.i];
vars.currentAmount = amounts[vars.i]; vars.currentAmount = amounts[vars.i];
vars.currentPremium = premiums[vars.i];
vars.currentATokenAddress = aTokenAddresses[vars.i]; vars.currentATokenAddress = aTokenAddresses[vars.i];
vars.currentAmountPlusPremium = vars.currentAmount.add(vars.currentPremium); vars.currentAmountPlusPremium = vars.currentAmount.add(totalPremiums[vars.i]);
if (DataTypes.InterestRateMode(modes[vars.i]) == DataTypes.InterestRateMode.NONE) { if (DataTypes.InterestRateMode(modes[vars.i]) == DataTypes.InterestRateMode.NONE) {
_reserves[vars.currentAsset].updateState(); _reserves[vars.currentAsset].updateState();
_reserves[vars.currentAsset].cumulateToLiquidityIndex( _reserves[vars.currentAsset].cumulateToLiquidityIndex(
IERC20(vars.currentATokenAddress).totalSupply(), IERC20(vars.currentATokenAddress).totalSupply(),
vars.currentPremium vars.currentPremiumToLP
); );
_reserves[vars.currentAsset].accruedToTreasury = _reserves[vars.currentAsset]
.accruedToTreasury
.add(vars.currentPremiumToProtocol.rayDiv(_reserves[vars.currentAsset].liquidityIndex));
_reserves[vars.currentAsset].updateInterestRates( _reserves[vars.currentAsset].updateInterestRates(
vars.currentAsset, vars.currentAsset,
vars.currentATokenAddress, vars.currentATokenAddress,
@ -532,7 +545,7 @@ contract LendingPool is VersionedInitializable, ILendingPool, LendingPoolStorage
msg.sender, msg.sender,
vars.currentAsset, vars.currentAsset,
vars.currentAmount, vars.currentAmount,
vars.currentPremium, vars.currentPremiumToLP,
referralCode referralCode
); );
} }
@ -733,7 +746,7 @@ contract LendingPool is VersionedInitializable, ILendingPool, LendingPoolStorage
* @dev Returns the fee on flash loans * @dev Returns the fee on flash loans
*/ */
function FLASHLOAN_PREMIUM_TOTAL() public view returns (uint256) { function FLASHLOAN_PREMIUM_TOTAL() public view returns (uint256) {
return _flashLoanPremiumTotal; return _flashLoanPremiumToLP;
} }
/** /**

4
contracts/protocol/lendingpool/LendingPoolStorage.sol
View File

@ -26,9 +26,11 @@ contract LendingPoolStorage {
uint256 internal _maxStableRateBorrowSizePercent; uint256 internal _maxStableRateBorrowSizePercent;
uint256 internal _flashLoanPremiumTotal; uint256 internal _flashLoanPremiumToLP;
uint256 internal _maxNumberOfReserves; uint256 internal _maxNumberOfReserves;
mapping(address => bool) _authorizedFlashBorrowers; mapping(address => bool) _authorizedFlashBorrowers;
uint256 internal _flashLoanPremiumToProtocol;
} }