Instadapp/dsa-polygon-migration
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-polygon-migration.git synced 2024-07-29 22:27:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Polygon flashloan

Browse Source
This commit is contained in:
Samyak Jain 2021-04-22 06:29:07 +05:30
parent 182d13a019
commit 9865addb2a
2 changed files with 59 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
contracts/receivers/aave-v2-receiver/interfaces.sol
View File

@ -25,6 +25,15 @@ struct AaveData {
address[] borrowTokens;
}
struct CastData {
address dsa;
uint route;
address[] tokens;
uint[] amounts;
address[] dsaTargets;
bytes[] dsaData;
}
interface IndexInterface {
function master() external view returns (address);
}
@ -99,5 +108,9 @@ interface AaveInterface {
}
interface InstaListInterface {
function accountID(address) external view returns (uint);
function accountID(address) external view returns (uint64);
}
interface DSAInterface {
function cast(address[] calldata _targets, bytes[] calldata _datas, address _origin) external payable returns (bytes32);
}

46
contracts/receivers/aave-v2-receiver/main.sol
View File

@ -4,7 +4,7 @@ pragma experimental ABIEncoderV2;
import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import { TokenInterface } from "../../common/interfaces.sol";
import { AccountInterface, AaveData, AaveInterface, IndexInterface } from "./interfaces.sol";
import { AccountInterface, AaveData, AaveInterface, IndexInterface, CastData, DSAInterface } from "./interfaces.sol";
import { Events } from "./events.sol";
import { Helpers } from "./helpers.sol";
@ -126,6 +126,50 @@ contract AaveV2Migrator is MigrateResolver {
}
}
contract InstaFlash is AaveV2Migrator {
using SafeERC20 for IERC20;
/**
* FOR SECURITY PURPOSE
* only Smart DEFI Account can access the liquidity pool contract
*/
modifier isDSA {
uint64 id = instaList.accountID(msg.sender);
require(id != 0, "not-dsa-id");
_;
}
function initiateFlashLoan(
address[] calldata _tokens,
uint256[] calldata _amounts,
uint _route, // no use of route but just to follow current flashloan pattern
bytes calldata data
) external isDSA {
uint _length = _tokens.length;
require(_length == _amounts.length, "not-equal-length");
uint[] memory iniBal = new uint[](_length);
IERC20[] memory _tokenContracts = new IERC20[](_length);
for (uint i = 0; i < _length; i++) {
_tokenContracts[i] = IERC20(_tokens[i]);
iniBal[i] = _tokenContracts[i].balanceOf(address(this));
_tokenContracts[i].safeTransfer(msg.sender, _amounts[i]);
}
CastData memory cd;
(cd.dsa, cd.route, cd.tokens, cd.amounts, cd.dsaTargets, cd.dsaData) = abi.decode(
data,
(address, uint256, address[], uint256[], address[], bytes[])
);
DSAInterface(msg.sender).cast(cd.dsaTargets, cd.dsaData, 0xB7fA44c2E964B6EB24893f7082Ecc08c8d0c0F87);
for (uint i = 0; i < _length; i++) {
uint _finBal = _tokenContracts[i].balanceOf(address(this));
require(_finBal >= iniBal[i], "flashloan-not-returned");
}
// TODO: emit event
}
}
contract InstaAaveV2MigratorReceiverImplementation is AaveV2Migrator {
function migrate(uint _id) external {
bytes memory _data = positions[_id];