From 16bf043a817facbca5e7c9e55e4bebfaa05490c6 Mon Sep 17 00:00:00 2001
From: Samyak Jain <34437877+KaymasJain@users.noreply.github.com>
Date: Thu, 15 Apr 2021 04:51:02 +0530
Subject: [PATCH] token repeatation

---
 contracts/senders/aave-v2-migrator/main.sol | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/contracts/senders/aave-v2-migrator/main.sol b/contracts/senders/aave-v2-migrator/main.sol
index 13201cc..0a6ee1e 100644
--- a/contracts/senders/aave-v2-migrator/main.sol
+++ b/contracts/senders/aave-v2-migrator/main.sol
@@ -112,6 +112,26 @@ contract MigrateResolver is LiquidityResolver {
             "invalid-length"
         );
 
+        for (uint i = 0; i < _data.supplyTokens.length; i++) {
+            address _token = _data.supplyTokens[i];
+            for (uint j = 0; j < _data.supplyTokens.length; j++) {
+                if (j != i) {
+                    require(j != i, "token-repeated");
+                }
+            }
+            require(_token != wethAddr, "should-be-eth-address");
+        }
+
+        for (uint i = 0; i < _data.borrowTokens.length; i++) {
+            address _token = _data.borrowTokens[i];
+            for (uint j = 0; j < _data.borrowTokens.length; j++) {
+                if (j != i) {
+                    require(j != i, "token-repeated");
+                }
+            }
+            require(_token != wethAddr, "should-be-eth-address");
+        }
+
         (uint[] memory stableBorrows, uint[] memory variableBorrows, uint[] memory totalBorrows) = _PaybackCalculate(aave, _data, sourceDsa);
 
         _PaybackStable(_data.borrowTokens.length, aave, _data.borrowTokens, stableBorrows, sourceDsa);