From e986a294ec3e328bd48bb4c423fe835d1aa2dc69 Mon Sep 17 00:00:00 2001
From: cyBerta <cyberta@riseup.net>
Date: Fri, 24 Jul 2020 16:30:02 +0200
Subject: fix bug in gateway selection - always iterate through all gateways
 and count those supporting the requested transport in order to get the
 nClosest gateway; adding more tests

---
 .../se/leap/bitmaskclient/eip/GatewaysManager.java  | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

(limited to 'app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java')

diff --git a/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java b/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java
index aa10ffae..f3773758 100644
--- a/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java
+++ b/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java
@@ -81,22 +81,29 @@ public class GatewaysManager {
         List<Gateway> list = new ArrayList<>(gateways.values());
         GatewaySelector gatewaySelector = new GatewaySelector(list);
         Gateway gateway;
-        while ((gateway = gatewaySelector.select(nClosest)) != null) {
+        int found  = 0;
+        int i = 0;
+        while ((gateway = gatewaySelector.select(i)) != null) {
             if (gateway.getProfile(transportType) != null) {
-                return gateway;
+                if (found == nClosest) {
+                    return gateway;
+                }
+                found++;
             }
-            nClosest++;
+            i++;
         }
         return null;
     }
 
     private Gateway getGatewayFromPresortedList(int nClosest, Connection.TransportType transportType) {
-        while (nClosest < presortedList.size()) {
-            Gateway gateway = presortedList.get(nClosest);
+        int found = 0;
+        for (Gateway gateway : presortedList) {
             if (gateway.getProfile(transportType) != null) {
-                return gateway;
+                if (found == nClosest) {
+                    return gateway;
+                }
+                found++;
             }
-            nClosest++;
         }
         return null;
     }
-- 
cgit v1.2.3