mirror of
https://github.com/qdm12/gluetun.git
synced 2026-05-06 20:10:11 +02:00
Quentin McGaw
4a78989d9d
- main reason being it's a burden to always define sentinel errors at global scope, wrap them with `%w` instead of using a string directly - only use sentinel errors when it has to be checked using `errors.Is` - replace all usage of these sentinel errors in `fmt.Errorf` with direct strings that were in the sentinel error - exclude the sentinel error definition requirement from .golangci.yml - update unit tests to use ContainersError instead of ErrorIs so it stays as a "not a change detector test" without requiring a sentinel error
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package openvpn
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/netip"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
func ExtractProto(b []byte) (tcp, udp bool, err error) {
|
|
lines := strings.Split(string(b), "\n")
|
|
const protoPrefix = "proto "
|
|
for _, line := range lines {
|
|
if !strings.HasPrefix(line, protoPrefix) {
|
|
continue
|
|
}
|
|
s := strings.TrimPrefix(line, protoPrefix)
|
|
s = strings.TrimSpace(s)
|
|
s = strings.ToLower(s)
|
|
switch s {
|
|
case "tcp", "tcp4", "tcp6", "tcp-client":
|
|
return true, false, nil
|
|
case "udp", "udp4", "udp6":
|
|
return false, true, nil
|
|
default:
|
|
return false, false, fmt.Errorf("unknown protocol: %s", s)
|
|
}
|
|
}
|
|
|
|
// default is UDP if unspecified in openvpn configuration
|
|
return false, true, nil
|
|
}
|
|
|
|
func ExtractHost(b []byte) (host, warning string, err error) {
|
|
const (
|
|
rejectIP = true
|
|
rejectDomain = false
|
|
)
|
|
hosts := extractRemoteHosts(b, rejectIP, rejectDomain)
|
|
if len(hosts) == 0 {
|
|
return "", "", errors.New("remote host not found")
|
|
} else if len(hosts) > 1 {
|
|
warning = fmt.Sprintf(
|
|
"only using the first host %q and discarding %d other hosts",
|
|
hosts[0], len(hosts)-1)
|
|
}
|
|
return hosts[0], warning, nil
|
|
}
|
|
|
|
func ExtractIPs(b []byte) (ips []netip.Addr, err error) {
|
|
const rejectIP, rejectDomain = false, true
|
|
ipStrings := extractRemoteHosts(b, rejectIP, rejectDomain)
|
|
if len(ipStrings) == 0 {
|
|
return nil, errors.New("remote IP not found")
|
|
}
|
|
|
|
sort.Slice(ipStrings, func(i, j int) bool {
|
|
return ipStrings[i] < ipStrings[j]
|
|
})
|
|
|
|
ips = make([]netip.Addr, len(ipStrings))
|
|
for i := range ipStrings {
|
|
ips[i], err = netip.ParseAddr(ipStrings[i])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("parsing IP address: %w", err)
|
|
}
|
|
}
|
|
|
|
return ips, nil
|
|
}
|
|
|
|
func extractRemoteHosts(content []byte, rejectIP, rejectDomain bool) (hosts []string) {
|
|
lines := strings.Split(string(content), "\n")
|
|
for _, line := range lines {
|
|
line = strings.TrimSpace(line)
|
|
if !strings.HasPrefix(line, "remote ") {
|
|
continue
|
|
}
|
|
fields := strings.Fields(line)
|
|
if len(fields) == 1 || fields[1] == "" {
|
|
continue
|
|
}
|
|
host := fields[1]
|
|
_, err := netip.ParseAddr(host)
|
|
if (rejectIP && err == nil) ||
|
|
(rejectDomain && err != nil) {
|
|
continue
|
|
}
|
|
hosts = append(hosts, host)
|
|
}
|
|
return hosts
|
|
}
|