chore(vpn): moved wireguard settings helpers from provider/utils to vpn as unexported functions

2026-05-09 20:29:23 +02:00 · 2026-03-11 14:05:55 +00:00
parent e6fc792f4f
commit 97ccadfd33
3 changed files with 76 additions and 87 deletions
@@ -0,0 +1,97 @@
+package vpn
+
+import (
+	"net/netip"
+	"testing"
+	"time"
+
+	"github.com/qdm12/gluetun/internal/configuration/settings"
+	"github.com/qdm12/gluetun/internal/models"
+	"github.com/qdm12/gluetun/internal/wireguard"
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_buildWireguardSettings(t *testing.T) {
+	t.Parallel()
+
+	testCases := map[string]struct {
+		connection    models.Connection
+		userSettings  settings.Wireguard
+		ipv6Supported bool
+		settings      wireguard.Settings
+	}{
+		"some_settings": {
+			connection: models.Connection{
+				IP:     netip.AddrFrom4([4]byte{1, 2, 3, 4}),
+				Port:   51821,
+				PubKey: "public",
+			},
+			userSettings: settings.Wireguard{
+				PrivateKey:   ptrTo("private"),
+				PreSharedKey: ptrTo("pre-shared"),
+				Addresses: []netip.Prefix{
+					netip.PrefixFrom(netip.AddrFrom4([4]byte{1, 1, 1, 1}), 32),
+					netip.PrefixFrom(netip.AddrFrom16([16]byte{}), 32),
+				},
+				AllowedIPs: []netip.Prefix{
+					netip.PrefixFrom(netip.AddrFrom4([4]byte{2, 2, 2, 2}), 32),
+					netip.PrefixFrom(netip.AddrFrom16([16]byte{}), 32),
+				},
+				PersistentKeepaliveInterval: ptrTo(time.Hour),
+				Interface:                   "wg1",
+				MTU:                         ptrTo(uint32(1000)),
+				AmneziaWG: settings.AmneziaWg{
+					JunkPacketCount: ptrTo(uint16(1)),
+					JunkPacketMin:   ptrTo(uint16(0)),
+					JunkPacketMax:   ptrTo(uint16(0)),
+					PaddingS1:       ptrTo(uint16(0)),
+					PaddingS2:       ptrTo(uint16(0)),
+					PaddingS3:       ptrTo(uint16(0)),
+					PaddingS4:       ptrTo(uint16(0)),
+					HeaderH1:        ptrTo("x"),
+					HeaderH2:        ptrTo(""),
+					HeaderH3:        ptrTo(""),
+					HeaderH4:        ptrTo(""),
+					InitPacketI1:    ptrTo(""),
+					InitPacketI2:    ptrTo(""),
+					InitPacketI3:    ptrTo(""),
+					InitPacketI4:    ptrTo(""),
+					InitPacketI5:    ptrTo(""),
+				},
+			},
+			ipv6Supported: false,
+			settings: wireguard.Settings{
+				InterfaceName: "wg1",
+				PrivateKey:    "private",
+				PublicKey:     "public",
+				PreSharedKey:  "pre-shared",
+				Endpoint:      netip.AddrPortFrom(netip.AddrFrom4([4]byte{1, 2, 3, 4}), 51821),
+				Addresses: []netip.Prefix{
+					netip.PrefixFrom(netip.AddrFrom4([4]byte{1, 1, 1, 1}), 32),
+				},
+				AllowedIPs: []netip.Prefix{
+					netip.PrefixFrom(netip.AddrFrom4([4]byte{2, 2, 2, 2}), 32),
+				},
+				PersistentKeepaliveInterval: time.Hour,
+				RulePriority:                101,
+				IPv6:                        ptrTo(false),
+				MTU:                         1000,
+				AmneziaWG: wireguard.AmneziaSettings{
+					JunkPacketCount: 1,
+					HeaderH1:        "x",
+				},
+			},
+		},
+	}
+
+	for name, testCase := range testCases {
+		t.Run(name, func(t *testing.T) {
+			t.Parallel()
+
+			settings := buildWireguardSettings(testCase.connection,
+				testCase.userSettings, testCase.ipv6Supported)
+
+			assert.Equal(t, testCase.settings, settings)
+		})
+	}
+}