ctrld/internal/clientinfo/client_info_test.go

package clientinfo

import (
	"sync"
	"testing"
)

// TestTable_SetSelfIP_NilDHCP ensures SetSelfIP does not panic when t.dhcp is
// nil, which happens when DHCP discovery is disabled and the network-change
// callback fires before or without initialisation.
func TestTable_SetSelfIP_NilDHCP(t *testing.T) {
	table := &Table{} // dhcp is nil
	// Must not panic.
	table.SetSelfIP("192.168.1.1")
	if got := table.SelfIP(); got != "192.168.1.1" {
		t.Fatalf("SelfIP() = %q, want %q", got, "192.168.1.1")
	}
}

// TestTable_SetSelfIP_UpdatesDHCP ensures SetSelfIP propagates the new IP to
// the dhcp discover and calls addSelf when dhcp is initialised.
func TestTable_SetSelfIP_UpdatesDHCP(t *testing.T) {
	table := &Table{
		dhcp: &dhcp{selfIP: "10.0.0.1"},
	}
	table.SetSelfIP("10.0.0.2")
	if got := table.SelfIP(); got != "10.0.0.2" {
		t.Fatalf("SelfIP() = %q, want %q", got, "10.0.0.2")
	}
	if table.dhcp.selfIP != "10.0.0.2" {
		t.Fatalf("dhcp.selfIP = %q, want %q", table.dhcp.selfIP, "10.0.0.2")
	}
}

// TestTable_SetSelfIP_Concurrent ensures concurrent calls to SetSelfIP do not
// race, regardless of whether dhcp is nil or not.
func TestTable_SetSelfIP_Concurrent(t *testing.T) {
	for _, tc := range []struct {
		name  string
		table *Table
	}{
		{"nil dhcp", &Table{}},
		{"with dhcp", &Table{dhcp: &dhcp{}}},
	} {
		tc := tc
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			var wg sync.WaitGroup
			for range 10 {
				wg.Add(1)
				go func() {
					defer wg.Done()
					tc.table.SetSelfIP("192.168.1.1")
					_ = tc.table.SelfIP()
				}()
			}
			wg.Wait()
		})
	}
}

func Test_normalizeIP(t *testing.T) {
	tests := []struct {
		name string
		in   string
		want string
	}{
		{"v4", "127.0.0.1", "127.0.0.1"},
		{"v4 with index", "127.0.0.1%lo", "127.0.0.1"},
		{"v6", "fe80::1", "fe80::1"},
		{"v6 with index", "fe80::1%22002", "fe80::1"},
	}
	for _, tc := range tests {
		tc := tc
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			if got := normalizeIP(tc.in); got != tc.want {
				t.Errorf("normalizeIP() = %v, want %v", got, tc.want)
			}
		})
	}
}

func TestTable_LookupRFC1918IPv4(t *testing.T) {
	table := &Table{
		dhcp: &dhcp{},
		arp:  &arpDiscover{},
	}

	table.ipResolvers = append(table.ipResolvers, table.dhcp)
	table.ipResolvers = append(table.ipResolvers, table.arp)

	macAddress := "cc:19:f9:8a:49:e6"
	rfc1918IPv4 := "10.0.10.245"
	table.dhcp.ip.Store(macAddress, "127.0.0.1")
	table.arp.ip.Store(macAddress, rfc1918IPv4)

	if got := table.LookupRFC1918IPv4(macAddress); got != rfc1918IPv4 {
		t.Fatalf("unexpected result, want: %s, got: %s", rfc1918IPv4, got)
	}
}

func TestTable_ListClients(t *testing.T) {
	mac := "74:56:3c:44:eb:5e"
	ipv6_1 := "2405:4803:a04b:4190:fbe9:cd14:d522:bbae"
	ipv6_2 := "2405:4803:a04b:4190:fbe9:cd14:d522:bbab"
	table := &Table{}

	// NDP init.
	table.ndp = &ndpDiscover{}
	table.ndp.mac.Store(ipv6_1, mac)
	table.ndp.mac.Store(ipv6_2, mac)
	table.ndp.ip.Store(mac, ipv6_1)
	table.ndp.ip.Store(mac, ipv6_2)
	table.ipResolvers = append(table.ipResolvers, table.ndp)
	table.macResolvers = append(table.macResolvers, table.ndp)

	hostname := "foo"
	// mdns init.
	table.mdns = &mdns{}
	table.mdns.name.Store(ipv6_2, hostname)
	table.hostnameResolvers = append(table.hostnameResolvers, table.mdns)

	for _, c := range table.ListClients() {
		if c.Hostname != hostname {
			t.Fatalf("missing hostname for client: %v", c)
		}
	}
}