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

Added double token check

Browse Source
This commit is contained in:
Thrilok Kumar 2021-04-29 23:42:13 +05:30
parent ce8cc951bc
commit 8952099463
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -148,6 +148,15 @@ contract InstaFlash is AaveV2Migrator {
) external nonReentrant isDSA { ) external nonReentrant isDSA {
uint _length = _tokens.length; uint _length = _tokens.length;
require(_length == _amounts.length, "not-equal-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); uint[] memory iniBal = new uint[](_length);
IERC20[] memory _tokenContracts = new IERC20[](_length); IERC20[] memory _tokenContracts = new IERC20[](_length);
for (uint i = 0; i < _length; i++) { for (uint i = 0; i < _length; i++) {