From 89520994632fc1f6b6c489e0920d5a66d1167a65 Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
Date: Thu, 29 Apr 2021 23:42:13 +0530
Subject: [PATCH] Added double token check

---
 contracts/receivers/aave-v2-receiver/main.sol | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/contracts/receivers/aave-v2-receiver/main.sol b/contracts/receivers/aave-v2-receiver/main.sol
index e2c3f68..54c8531 100644
--- a/contracts/receivers/aave-v2-receiver/main.sol
+++ b/contracts/receivers/aave-v2-receiver/main.sol
@@ -148,6 +148,15 @@ contract InstaFlash is AaveV2Migrator {
     ) 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++) {