admin/gluetun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore(provider): add safety connection count check

Browse Source
This commit is contained in:
Quentin McGaw
2022-04-17 16:23:53 +00:00
parent 7ad6fc8e73
commit 0dd723b29f
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/provider/utils/pick.go
View File

@@ -11,6 +11,8 @@ import (
"github.com/qdm12/gluetun/internal/models" "github.com/qdm12/gluetun/internal/models"
) )
var ErrNoConnectionToPickFrom = errors.New("no connection to pick from")
// PickConnection picks a connection from a pool of connections. // PickConnection picks a connection from a pool of connections.
// If the VPN protocol is Wireguard and the target IP is set, // If the VPN protocol is Wireguard and the target IP is set,
// it finds the connection corresponding to this target IP. // it finds the connection corresponding to this target IP.
@@ -19,6 +21,10 @@ import (
func PickConnection(connections []models.Connection, func PickConnection(connections []models.Connection,
selection settings.ServerSelection, randSource rand.Source) ( selection settings.ServerSelection, randSource rand.Source) (
connection models.Connection, err error) { connection models.Connection, err error) {
if len(connections) == 0 {
return connection, ErrNoConnectionToPickFrom
}
if len(selection.TargetIP) > 0 && selection.VPN == constants.Wireguard { if len(selection.TargetIP) > 0 && selection.VPN == constants.Wireguard {
// we need the right public key // we need the right public key
return getTargetIPConnection(connections, selection.TargetIP) return getTargetIPConnection(connections, selection.TargetIP)