badguardhome/internal/aghnet/net_linux_test.go

//go:build linux
// +build linux

package aghnet

import (
	"bytes"
	"net"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

const nl = "\n"

func TestDHCPCDStaticConfig(t *testing.T) {
	const iface interfaceName = `wlan0`

	testCases := []struct {
		name     string
		data     []byte
		wantCont bool
	}{{
		name: "has_not",
		data: []byte(`#comment` + nl +
			`# comment` + nl +
			`interface eth0` + nl +
			`static ip_address=192.168.0.1/24` + nl +
			`# interface ` + iface + nl +
			`static ip_address=192.168.1.1/24` + nl +
			`# comment` + nl,
		),
		wantCont: true,
	}, {
		name: "has",
		data: []byte(`#comment` + nl +
			`# comment` + nl +
			`interface eth0` + nl +
			`static ip_address=192.168.0.1/24` + nl +
			`# interface ` + iface + nl +
			`static ip_address=192.168.1.1/24` + nl +
			`# comment` + nl +
			`interface ` + iface + nl +
			`# comment` + nl +
			`static ip_address=192.168.2.1/24` + nl,
		),
		wantCont: false,
	}}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			r := bytes.NewReader(tc.data)
			_, cont, err := iface.dhcpcdStaticConfig(r)
			require.NoError(t, err)

			assert.Equal(t, tc.wantCont, cont)
		})
	}
}

func TestIfacesStaticConfig(t *testing.T) {
	const iface interfaceName = `enp0s3`

	testCases := []struct {
		name         string
		data         []byte
		wantCont     bool
		wantPatterns []string
	}{{
		name: "has_not",
		data: []byte(`allow-hotplug ` + iface + nl +
			`#iface enp0s3 inet static` + nl +
			`#  address 192.168.0.200` + nl +
			`#  netmask 255.255.255.0` + nl +
			`#  gateway 192.168.0.1` + nl +
			`iface ` + iface + ` inet dhcp` + nl,
		),
		wantCont:     true,
		wantPatterns: []string{},
	}, {
		name: "has",
		data: []byte(`allow-hotplug ` + iface + nl +
			`iface ` + iface + ` inet static` + nl +
			`  address 192.168.0.200` + nl +
			`  netmask 255.255.255.0` + nl +
			`  gateway 192.168.0.1` + nl +
			`#iface ` + iface + ` inet dhcp` + nl,
		),
		wantCont:     false,
		wantPatterns: []string{},
	}, {
		name: "return_patterns",
		data: []byte(`source hello` + nl +
			`source world` + nl +
			`#iface ` + iface + ` inet static` + nl,
		),
		wantCont:     true,
		wantPatterns: []string{"hello", "world"},
	}, {
		// This one tests if the first found valid interface prevents
		// checking files under the `source` directive.
		name: "ignore_patterns",
		data: []byte(`source hello` + nl +
			`source world` + nl +
			`iface ` + iface + ` inet static` + nl,
		),
		wantCont:     false,
		wantPatterns: []string{},
	}}

	for _, tc := range testCases {
		r := bytes.NewReader(tc.data)
		t.Run(tc.name, func(t *testing.T) {
			patterns, has, err := iface.ifacesStaticConfig(r)
			require.NoError(t, err)

			assert.Equal(t, tc.wantCont, has)
			assert.ElementsMatch(t, tc.wantPatterns, patterns)
		})
	}
}

func TestSetStaticIPdhcpcdConf(t *testing.T) {
	testCases := []struct {
		name       string
		dhcpcdConf string
		routers    net.IP
	}{{
		name: "with_gateway",
		dhcpcdConf: nl + `# wlan0 added by AdGuard Home.` + nl +
			`interface wlan0` + nl +
			`static ip_address=192.168.0.2/24` + nl +
			`static routers=192.168.0.1` + nl +
			`static domain_name_servers=192.168.0.2` + nl + nl,
		routers: net.IP{192, 168, 0, 1},
	}, {
		name: "without_gateway",
		dhcpcdConf: nl + `# wlan0 added by AdGuard Home.` + nl +
			`interface wlan0` + nl +
			`static ip_address=192.168.0.2/24` + nl +
			`static domain_name_servers=192.168.0.2` + nl + nl,
		routers: nil,
	}}

	ipNet := &net.IPNet{
		IP:   net.IP{192, 168, 0, 2},
		Mask: net.IPMask{255, 255, 255, 0},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			s := dhcpcdConfIface("wlan0", ipNet, tc.routers, net.IP{192, 168, 0, 2})
			assert.Equal(t, tc.dhcpcdConf, s)
		})
	}
}