From 2e7702904813dbdbfb2f1fa24979e80222bb2e27 Mon Sep 17 00:00:00 2001
From: Ruben Pollan <meskio@sindominio.net>
Date: Wed, 31 Mar 2021 18:14:20 +0200
Subject: Add a proper reconnection mechanism

---
 pkg/vpn/bonafide/gateways.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'pkg/vpn/bonafide/gateways.go')

diff --git a/pkg/vpn/bonafide/gateways.go b/pkg/vpn/bonafide/gateways.go
index 460d80f..50359e9 100644
--- a/pkg/vpn/bonafide/gateways.go
+++ b/pkg/vpn/bonafide/gateways.go
@@ -59,6 +59,10 @@ type gatewayPool struct {
 	locations map[string]Location
 }
 
+func (gw Gateway) isTransport(transport string) bool {
+	return transport == "any" || gw.Transport == transport
+}
+
 func (p *gatewayPool) populateLocationList() {
 	for i, gw := range p.available {
 		p.byLocation[gw.Location] = append(p.byLocation[gw.Location], &p.available[i])
@@ -118,7 +122,7 @@ func (p *gatewayPool) getRandomGatewaysByLocation(location, transport string) ([
 
 	var gateways []Gateway
 	for _, gw := range gws {
-		if gw.Transport == transport {
+		if gw.isTransport(transport) {
 			gateways = append(gateways, *gw)
 		}
 		if len(gateways) == maxGateways {
@@ -143,7 +147,7 @@ func (p *gatewayPool) getGatewaysFromMenshenByLocation(location, transport strin
 
 	var gateways []Gateway
 	for _, gw := range p.recommended {
-		if gw.gateway.Transport != transport {
+		if !gw.gateway.isTransport(transport) {
 			continue
 		}
 		for _, locatedGw := range gws {
@@ -275,7 +279,7 @@ func (p *gatewayPool) getAll(transport string, tz int) ([]Gateway, error) {
 func (p *gatewayPool) getGatewaysFromMenshen(transport string, max int) ([]Gateway, error) {
 	gws := make([]Gateway, 0)
 	for _, gw := range p.recommended {
-		if gw.gateway.Transport != transport {
+		if !gw.gateway.isTransport(transport) {
 			continue
 		}
 		gws = append(gws, *gw.gateway)
@@ -292,7 +296,7 @@ func (p *gatewayPool) getGatewaysByTimezone(transport string, tzOffsetHours, max
 	gwVector := []gatewayDistance{}
 
 	for _, gw := range p.available {
-		if gw.Transport != transport {
+		if !gw.isTransport(transport) {
 			continue
 		}
 		distance := 13
-- 
cgit v1.2.3