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

Add new variable borrowedAssetLeftovers to improve readability

Browse Source
This commit is contained in:
David Racero 2021-02-01 13:24:07 +01:00
parent 37ac8b5297
commit 211dca7a07
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
contracts/adapters/FlashLiquidationAdapter.sol
View File

@ -37,6 +37,7 @@ contract FlashLiquidationAdapter is BaseUniswapAdapter {
uint256 flashLoanDebt; uint256 flashLoanDebt;
uint256 soldAmount; uint256 soldAmount;
uint256 remainingTokens; uint256 remainingTokens;
uint256 borrowedAssetLeftovers;
} }
constructor( constructor(
@ -104,7 +105,7 @@ contract FlashLiquidationAdapter is BaseUniswapAdapter {
address user, address user,
uint256 debtToCover, uint256 debtToCover,
bool useEthPath, bool useEthPath,
uint256 flashBorrowedAmount, // 1000 uint256 flashBorrowedAmount,
uint256 premium, uint256 premium,
address initiator address initiator
) internal { ) internal {
@ -112,8 +113,11 @@ contract FlashLiquidationAdapter is BaseUniswapAdapter {
vars.initCollateralBalance = IERC20(collateralAsset).balanceOf(address(this)); vars.initCollateralBalance = IERC20(collateralAsset).balanceOf(address(this));
if (collateralAsset != borrowedAsset) { if (collateralAsset != borrowedAsset) {
vars.initFlashBorrowedBalance = IERC20(borrowedAsset).balanceOf(address(this)); vars.initFlashBorrowedBalance = IERC20(borrowedAsset).balanceOf(address(this));
// Track leftover balance to rescue funds in case of external transfers into this contract
vars.borrowedAssetLeftovers = vars.initFlashBorrowedBalance.sub(flashBorrowedAmount);
} }
vars.flashLoanDebt = flashBorrowedAmount.add(premium); // 1010 vars.flashLoanDebt = flashBorrowedAmount.add(premium);
// Approve LendingPool to use debt token for liquidation // Approve LendingPool to use debt token for liquidation
IERC20(borrowedAsset).approve(address(LENDING_POOL), debtToCover); IERC20(borrowedAsset).approve(address(LENDING_POOL), debtToCover);
@ -124,15 +128,16 @@ contract FlashLiquidationAdapter is BaseUniswapAdapter {
// Discover the liquidated tokens // Discover the liquidated tokens
uint256 collateralBalanceAfter = IERC20(collateralAsset).balanceOf(address(this)); uint256 collateralBalanceAfter = IERC20(collateralAsset).balanceOf(address(this));
// Track only collateral released, not current asset balance of the contract
vars.diffCollateralBalance = collateralBalanceAfter.sub(vars.initCollateralBalance); vars.diffCollateralBalance = collateralBalanceAfter.sub(vars.initCollateralBalance);
if (collateralAsset != borrowedAsset) { if (collateralAsset != borrowedAsset) {
// Discover flash loan balance // Discover flash loan balance after the liquidation
uint256 flashBorrowedAssetAfter = IERC20(borrowedAsset).balanceOf(address(this)); uint256 flashBorrowedAssetAfter = IERC20(borrowedAsset).balanceOf(address(this));
vars.diffFlashBorrowedBalance = flashBorrowedAssetAfter.sub( // Use only flash loan borrowed assets, not current asset balance of the contract
vars.initFlashBorrowedBalance.sub(flashBorrowedAmount) vars.diffFlashBorrowedBalance = flashBorrowedAssetAfter.sub(vars.borrowedAssetLeftovers);
);
// Swap released collateral into the debt asset, to repay the flash loan // Swap released collateral into the debt asset, to repay the flash loan
vars.soldAmount = _swapTokensForExactTokens( vars.soldAmount = _swapTokensForExactTokens(
collateralAsset, collateralAsset,