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

Initial testing

Browse Source
This commit is contained in:
Mubaris NK 2021-04-23 17:38:58 +05:30
parent d8463fbcdf
commit cab583e945
No known key found for this signature in database
GPG Key ID: 9AC09AD0F8D68561
5 changed files with 316 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

113
contracts/mock/aave-v2-migrator/main.sol Normal file
View File

@ -0,0 +1,113 @@
pragma solidity ^0.7.0;
pragma experimental ABIEncoderV2;
import { Variables } from "./variables.sol";
/**
* @title InstaAccountV2 Implementation-M2.
* @dev DeFi Smart Account Wallet for polygon migration.
*/
interface ConnectorsInterface {
function isConnectors(string[] calldata connectorNames) external view returns (bool, address[] memory);
}
contract Constants is Variables {
// InstaIndex Address.
address internal constant instaIndex = 0xA9B99766E6C676Cf1975c0D3166F96C0848fF5ad;
// Migration contract Address.
address internal constant migrationContract = 0xDc64a140Aa3E981100a9becA4E685f962f0cF6C9;
// Connnectors Address.
address public constant connectorsM1 = 0x0a0a82D2F86b9E46AE60E22FCE4e8b916F858Ddc;
}
contract InstaImplementationM2 is Constants {
function decodeEvent(bytes memory response) internal pure returns (string memory _eventCode, bytes memory _eventParams) {
if (response.length > 0) {
(_eventCode, _eventParams) = abi.decode(response, (string, bytes));
}
}
event LogCastMigrate(
address indexed origin,
address indexed sender,
uint256 value,
string[] targetsNames,
address[] targets,
string[] eventNames,
bytes[] eventParams
);
receive() external payable {}
/**
* @dev Delegate the calls to Connector.
* @param _target Connector address
* @param _data CallData of function.
*/
function spell(address _target, bytes memory _data) internal returns (bytes memory response) {
require(_target != address(0), "target-invalid");
assembly {
let succeeded := delegatecall(gas(), _target, add(_data, 0x20), mload(_data), 0, 0)
let size := returndatasize()
response := mload(0x40)
mstore(0x40, add(response, and(add(add(size, 0x20), 0x1f), not(0x1f))))
mstore(response, size)
returndatacopy(add(response, 0x20), 0, size)
switch iszero(succeeded)
case 1 {
// throw if delegatecall failed
returndatacopy(0x00, 0x00, size)
revert(0x00, size)
}
}
}
/**
* @dev This is the main function, Where all the different functions are called
* from Smart Account.
* @param _targetNames Array of Connector address.
* @param _datas Array of Calldata.
*/
function castMigrate(
string[] calldata _targetNames,
bytes[] calldata _datas,
address _origin
)
external
payable
returns (bytes32) // Dummy return to fix instaIndex buildWithCast function
{
uint256 _length = _targetNames.length;
require(msg.sender == migrationContract, "2: permission-denied");
require(_length != 0, "2: length-invalid");
require(_length == _datas.length , "2: array-length-invalid");
string[] memory eventNames = new string[](_length);
bytes[] memory eventParams = new bytes[](_length);
// TODO: restrict migration contract to run something specific? or give is all access as it doesn't have power to run anything else
(bool isOk, address[] memory _targets) = ConnectorsInterface(connectorsM1).isConnectors(_targetNames);
require(isOk, "2: not-connector");
for (uint i = 0; i < _length; i++) {
bytes memory response = spell(_targets[i], _datas[i]);
(eventNames[i], eventParams[i]) = decodeEvent(response);
}
emit LogCastMigrate(
_origin,
msg.sender,
msg.value,
_targetNames,
_targets,
eventNames,
eventParams
);
}
}

6
contracts/mock/aave-v2-migrator/variables.sol Normal file
View File

@ -0,0 +1,6 @@
pragma solidity ^0.7.0;
contract Variables {
// Auth Module(Address of Auth => bool).
mapping (address => bool) internal _auth;
}

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

@ -177,7 +177,7 @@ contract InstaFlash is AaveV2Migrator {
}
contract InstaAaveV2MigratorReceiverImplementation is AaveV2Migrator {
contract InstaAaveV2MigratorReceiverImplementation is InstaFlash {
function migrate(uint _id) external {
bytes memory _data = positions[_id];

165
test/receiver_flashloan.test.js Normal file
View File

@ -0,0 +1,165 @@
const hre = require("hardhat");
const { expect } = require("chai");
const { ethers, network, waffle } = hre;
const { provider, deployContract } = waffle
const Receiver = require("../artifacts/contracts/receivers/aave-v2-receiver/main.sol/InstaAaveV2MigratorReceiverImplementation.json")
const Implementation = require("../artifacts/contracts/mock/aave-v2-migrator/main.sol/InstaImplementationM2.json")
describe("Receiver", function() {
let accounts, account, receiver, master, implementation
const usdc = '0x2791Bca1f2de4661ED88A30C99A7a9449Aa84174'
const usdt = '0xc2132D05D31c914a87C6611C10748AEb04B58e8F'
const dai = '0x8f3Cf7ad23Cd3CaDbD9735AFf958023239c6A063'
const wbtc = '0x1BFD67037B42Cf73acF2047067bd4F2C47D9BfD6'
const weth = '0x7ceB23fD6bC0adD59E62ac25578270cFf1b9f619'
const wmatic = '0x0d500B1d8E8eF31E21C99d1Db9A6444d3ADf1270'
const instaImplementations = '0x39d3d5e7c11D61E072511485878dd84711c19d4A'
const maxValue = '115792089237316195423570985008687907853269984665640564039457584007913129639935'
const erc20Abi = [
"function balanceOf(address) view returns (uint)",
"function transfer(address to, uint amount)",
"function approve(address spender, uint amount)"
]
const instaImplementationABI = [
"function addImplementation(address _implementation, bytes4[] calldata _sigs)",
"function getImplementation(bytes4 _sig) view returns (address)",
"function removeImplementation(address _implementation)"
]
const implementations_m2Sigs = ["0x5a19a5eb"]
before(async function() {
accounts = await ethers.getSigners()
account = accounts[0]
masterAddress = "0x31de2088f38ed7F8a4231dE03973814edA1f8773"
await hre.network.provider.request({
method: "hardhat_impersonateAccount",
params: [ masterAddress ]
})
master = ethers.provider.getSigner(masterAddress)
receiver = await deployContract(account, Receiver, [])
implementation = await deployContract(account, Implementation, [])
console.log("Receiver deployed: ", receiver.address)
console.log("ImplementationM2 deployed: ", implementation.address)
await master.sendTransaction({
to: receiver.address,
value: ethers.utils.parseEther("45400000")
})
const usdcHolderAddr = '0x986a2fCa9eDa0e06fBf7839B89BfC006eE2a23Dd' // 1,000,000
await account.sendTransaction({ to: usdcHolderAddr, value: ethers.utils.parseEther('1') })
await hre.network.provider.request({
method: "hardhat_impersonateAccount",
params: [ usdcHolderAddr ]
})
const usdcHolder = ethers.provider.getSigner(usdcHolderAddr)
const usdcContract = new ethers.Contract(usdc, erc20Abi, usdcHolder)
await usdcContract.transfer(receiver.address, ethers.utils.parseUnits('1000000', 6))
})
// it("should change the implementation", async function() {
// const funder = await deployContract(master, Funder, [])
// await master.sendTransaction({
// to: funder.address,
// value: ethers.utils.parseEther("10")
// })
// await funder.kill()
// const proxyAbi = [
// "function upgradeTo(address newImplementation)"
// ]
// const masterProxyAddr = '0x697860CeE594c577F18f71cAf3d8B68D913c7366'
// const masterProxySigner = ethers.provider.getSigner(masterProxyAddr)
// const receiverProxy = '0x4A090897f47993C2504144419751D6A91D79AbF4'
// const receiverProxyContract = new ethers.Contract(receiverProxy, proxyAbi, masterProxySigner)
// await receiverProxyContract.upgradeTo(receiver.address)
// })
it("should match matic receiver", async function() {
const maticReceiver = await receiver.maticReceiver()
expect(maticReceiver).to.be.equal('0x0000000000000000000000000000000000001001')
})
it("should set implementationsM2", async function() {
const instaImplementationsContract = new ethers.Contract(instaImplementations, instaImplementationABI, master)
await instaImplementationsContract.connect(master).removeImplementation('0xEAac5739eB532110431b14D01017506DBA8f7b07')
await instaImplementationsContract.connect(master).addImplementation(implementation.address, implementations_m2Sigs)
})
it("single token flashloan", async function() {
const dsaAddr = '0x150Acc42e6751776c9E784EfF830cB4f35aE98f3'
await hre.network.provider.request({
method: "hardhat_impersonateAccount",
params: [ dsaAddr ]
})
const dsaSigner = ethers.provider.getSigner(dsaAddr)
const aave = new ethers.utils.Interface([
'function deposit(address,uint256,uint256,uint256)',
'function withdraw(address,uint256,uint256,uint256)'
])
const basic = new ethers.utils.Interface([
'function withdraw(address,uint256,address,uint256,uint256)'
])
const abiCoder = ethers.utils.defaultAbiCoder
const abiData = abiCoder.encode(
['address', 'uint256', 'address[]', 'uint256[]', 'string[]', 'bytes[]'],
[
dsaAddr,
0,
[usdc],
[ethers.utils.parseUnits('1000', 6)],
['AAVE-V2-A', 'AAVE-V2-A', 'BASIC-A'],
[
aave.encodeFunctionData('deposit', [usdc, ethers.utils.parseUnits('1000', 6), 0, 0]),
aave.encodeFunctionData('withdraw', [usdc, ethers.utils.parseUnits('1000', 6), 0, 0]),
basic.encodeFunctionData('withdraw', [usdc, ethers.utils.parseUnits('1000', 6), receiver.address, 0, 0])
]
]
)
// console.log('abiData', abiData)
const tx = await receiver.connect(dsaSigner).initiateFlashLoan(
[usdc],
[ethers.utils.parseUnits('1000', 6)],
0,
abiData
// {
// dsa: dsaAddr,
// route: 0,
// tokens: [usdc],
// amounts: [ethers.utils.parseUnits('1000', 6)],
// dsaTargets: ['AAVE-V2-A', 'AAVE-V2-A'],
// dsaData: [
// aave.encodeFunctionData('deposit', (usdc, ethers.utils.parseUnits('1000', 6), 0, 0)),
// aave.encodeFunctionData('withdraw', (usdc, ethers.utils.parseUnits('1000', 6), 0, 0))
// ]
// }
)
await tx.wait()
})
})

43
test/receiver_l2.test.js
View File

@ -5,7 +5,8 @@ const { provider, deployContract } = waffle
const Migrator = require("../artifacts/contracts/receivers/aave-v2-receiver/main.sol/InstaAaveV2MigratorReceiverImplementation.json")
const TokenMappingContract = require("../artifacts/contracts/receivers/mapping/main.sol/InstaPolygonTokenMapping.json")
const Implementations_m2Contract = require("../artifacts/contracts/implementation/aave-v2-migrator/main.sol/InstaImplementationM1.json")
const Implementations_m2Contract = require("../artifacts/contracts/implementation/aave-v2-migrator/main.sol/InstaImplementationM2.json")
const Funder = require("../artifacts/contracts/mock/funder.sol/Funder.json");
const tokenMaps = {
// polygon address : main net address
@ -21,7 +22,7 @@ const tokenMaps = {
//
describe("Migrator", function() {
let accounts, masterAddress, master, migrator, ethereum, instapool, tokenMapping, implementations_m2Contract
let accounts, masterAddress, master, migrator, ethereum, instapool, tokenMapping, receiverSigner
const erc20Abi = [
"function balanceOf(address) view returns (uint)",
@ -47,6 +48,8 @@ describe("Migrator", function() {
const instaConnectorsV2 = '0x0a0a82D2F86b9E46AE60E22FCE4e8b916F858Ddc'
const instaImplementations = '0x39d3d5e7c11D61E072511485878dd84711c19d4A'
const receiverMsgSender = '0x0000000000000000000000000000000000001001'
const maxValue = '115792089237316195423570985008687907853269984665640564039457584007913129639935'
@ -60,19 +63,26 @@ describe("Migrator", function() {
params: [ masterAddress ]
})
await hre.network.provider.request({
method: "hardhat_impersonateAccount",
params: [ receiverMsgSender ]
})
accounts = await ethers.getSigners();
master = ethers.provider.getSigner(masterAddress)
receiverSigner = ethers.provider.getSigner(receiverMsgSender)
migrator = await deployContract(master, Migrator, [])
// instapool = await deployContract(master, InstaPool, [])
tokenMapping = await deployContract(master, TokenMappingContract, [Object.values(tokenMaps), Object.keys(tokenMaps)])
console.log("Migrator deployed: ", migrator.address)
implementations_m2Contract = await deployContract(master, Implementations_m2Contract, [instaConnectorsV2, migrator.address])
// implementations_m2Contract = await deployContract(master, Implementations_m2Contract, [])
console.log("Migrator deployed: ", migrator.address)
console.log("tokenMapping deployed: ", tokenMapping.address)
console.log("implementations_m2Contract deployed: ", implementations_m2Contract.address)
// console.log("implementations_m2Contract deployed: ", implementations_m2Contract.address)
await master.sendTransaction({
to: migrator.address,
@ -81,11 +91,11 @@ describe("Migrator", function() {
ethereum = network.provider
})
it("should set implementationsV2", async function() {
// it("should set implementationsV2", async function() {
const instaImplementationsContract = new ethers.Contract(instaImplementations, instaImplementationABI, master)
await instaImplementationsContract.connect(master).addImplementation(implementations_m2Contract.address, implementations_m2Sigs)
})
// const instaImplementationsContract = new ethers.Contract(instaImplementations, instaImplementationABI, master)
// await instaImplementationsContract.connect(master).addImplementation(implementations_m2Contract.address, implementations_m2Sigs)
// })
it("should set tokens", async function() {
const tx = await migrator.connect(master).addTokenSupport([...Object.keys(tokenMaps).slice(1, 3), "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE"])
@ -95,8 +105,17 @@ describe("Migrator", function() {
expect(isMatic).to.be.true;
})
it("fund receiver data setter", async function() {
const funder = await deployContract(master, Funder, [])
await master.sendTransaction({
to: funder.address,
value: ethers.utils.parseEther("1")
});
await funder.kill()
})
it("should set data", async function() {
const tx = await migrator.onStateReceive("345", migrateData)
const tx = await migrator.connect(receiverSigner).onStateReceive("345", migrateData)
const receipt = await tx.wait()
const _migrateData = await migrator.positions("345")
@ -226,7 +245,7 @@ describe("Migrator", function() {
it("test migrate 2", async function() {
const positionData = '0x0000000000000000000000000000000000000000000000000000000000000020000000000000000000000000150acc42e6751776c9e784eff830cb4f35ae98f300000000000000000000000000000000000000000000000000000000000000a000000000000000000000000000000000000000000000000000000000000000e0000000000000000000000000000000000000000000000000000000000000016000000000000000000000000000000000000000000000000000000000000001a000000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000056900d33ca7fc0000000000000000000000000000000000000000000000000000000000000000000300000000000000000000000000000000000000000000000000000004a817c80000000000000000000000000000000000000000000000000000000004a817c800000000000000000000000000000000000000000000000878678326eac90000000000000000000000000000000000000000000000000000000000000000000001000000000000000000000000c02aaa39b223fe8d0a0e5c4f27ead9083c756cc20000000000000000000000000000000000000000000000000000000000000003000000000000000000000000a0b86991c6218b36c1d19d4a2e9eb0ce3606eb48000000000000000000000000dac17f958d2ee523a2206206994597c13d831ec70000000000000000000000006b175474e89094c44da98b954eedeac495271d0f'
let tx = await migrator.onStateReceive("346", positionData)
let tx = await migrator.connect(receiverSigner).onStateReceive("346", positionData)
await tx.wait()
tx = await migrator.migrate("346")
@ -242,7 +261,7 @@ describe("Migrator", function() {
it("test migrate 3", async function() {
const positionData = '0x000000000000000000000000000000000000000000000000000000000000002000000000000000000000000032d99500f7621c6dc5391395d419236383dbff9700000000000000000000000000000000000000000000000000000000000000a000000000000000000000000000000000000000000000000000000000000000e00000000000000000000000000000000000000000000000000000000000000140000000000000000000000000000000000000000000000000000000000000018000000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000361a08405e8fd8000000000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000df847580000000000000000000000000000000000000000000000000000000009502f90000000000000000000000000000000000000000000000000000000000000000001000000000000000000000000c02aaa39b223fe8d0a0e5c4f27ead9083c756cc20000000000000000000000000000000000000000000000000000000000000002000000000000000000000000a0b86991c6218b36c1d19d4a2e9eb0ce3606eb48000000000000000000000000dac17f958d2ee523a2206206994597c13d831ec7'
let tx = await migrator.onStateReceive("347", positionData)
let tx = await migrator.connect(receiverSigner).onStateReceive("347", positionData)
await tx.wait()
tx = await migrator.migrate("347")
@ -258,7 +277,7 @@ describe("Migrator", function() {
it("test migrate 4", async function() {
const positionData = '0x000000000000000000000000000000000000000000000000000000000000002000000000000000000000000029601161ad5da8c54435f4065af3a0ee30cb24dd00000000000000000000000000000000000000000000000000000000000000a00000000000000000000000000000000000000000000000000000000000000100000000000000000000000000000000000000000000000000000000000000014000000000000000000000000000000000000000000000000000000000000001a000000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000d86821017a3f60000000000000000000000000000000000000000000000000000000000000b274d080000000000000000000000000000000000000000000000000000000000000000100000000000000000000000000000000000000000000000000000574fbde60000000000000000000000000000000000000000000000000000000000000000002000000000000000000000000c02aaa39b223fe8d0a0e5c4f27ead9083c756cc20000000000000000000000002260fac5e5542a773aa44fbcfedf7c193bc2c5990000000000000000000000000000000000000000000000000000000000000001000000000000000000000000a0b86991c6218b36c1d19d4a2e9eb0ce3606eb48'
let tx = await migrator.onStateReceive("348", positionData)
let tx = await migrator.connect(receiverSigner).onStateReceive("348", positionData)
await tx.wait()
tx = await migrator.migrate("348")