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

Merge pull request #13 from Instadapp/flashloan-changes

Browse Source 
Flashloan changes
This commit is contained in:
Thrilok kumar 2021-04-29 23:43:01 +05:30 committed by GitHub
commit e3fc1a92fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 44 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -145,9 +145,18 @@ contract InstaFlash is AaveV2Migrator {
uint256[] calldata _amounts,
uint /*_route */, // no use of route but just to follow current flashloan pattern
bytes calldata data
) external isDSA {
) external nonReentrant isDSA {
uint _length = _tokens.length;
require(_length == _amounts.length, "not-equal-length");
for (uint i = 0; i < _length; i++) {
for (uint j = 0; j < _length; j++) {
if (j != i) {
require(_tokens[j] != _tokens[i], "token-repeated");
}
}
}
uint[] memory iniBal = new uint[](_length);
IERC20[] memory _tokenContracts = new IERC20[](_length);
for (uint i = 0; i < _length; i++) {

34
contracts/receivers/aave-v2-receiver/variables.sol
View File

@ -41,6 +41,12 @@ contract Variables {
*/
address public constant maticReceiver = 0x0000000000000000000000000000000000001001;
/**
* @dev ReentrancyGuard
*/
uint256 private constant _NOT_ENTERED = 1;
uint256 private constant _ENTERED = 2;
// Storage variables //
@ -66,4 +72,32 @@ contract Variables {
* @dev last stateId from the onStateReceive
*/
uint256 internal lastStateId;
/**
* @dev ReentrancyGuard status variable
*/
uint256 private _reentrancyStatus;
// Modifer //
/**
* @dev Prevents a contract from calling itself, directly or indirectly.
* Calling a `nonReentrant` function from another `nonReentrant`
* function is not supported. It is possible to prevent this from happening
* by making the `nonReentrant` function external, and make it call a
* `private` function that does the actual work.
*/
modifier nonReentrant() {
// On the first call to nonReentrant, _notEntered will be true
require(_reentrancyStatus != _ENTERED, "ReentrancyGuard: reentrant call");
// Any calls to nonReentrant after this point will fail
_reentrancyStatus = _ENTERED;
_;
// By storing the original value once again, a refund is triggered (see
// https://eips.ethereum.org/EIPS/eip-2200)
_reentrancyStatus = _NOT_ENTERED;
}
}