Something went wrong. Try again.
forked from tailscale/tailscale
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328// Copyright (c) Tailscale Inc & contributors// SPDX-License-Identifier: BSD-3-Clause
package netmap
import ( "encoding/hex" "net/netip" "reflect" "testing"
"go4.org/mem" "tailscale.com/net/netaddr" "tailscale.com/tailcfg" "tailscale.com/tstest/typewalk" "tailscale.com/types/key")
func testNodeKey(b byte) (ret key.NodePublic) { var bs [key.NodePublicRawLen]byte for i := range bs { bs[i] = b } return key.NodePublicFromRaw32(mem.B(bs[:]))}
func testDiscoKey(hexPrefix string) (ret key.DiscoPublic) { b, err := hex.DecodeString(hexPrefix) if err != nil { panic(err) } // this function is used with short hexes, so zero-extend the raw // value. var bs [32]byte copy(bs[:], b) return key.DiscoPublicFromRaw32(mem.B(bs[:]))}
func nodeViews(v []*tailcfg.Node) []tailcfg.NodeView { nv := make([]tailcfg.NodeView, len(v)) for i, n := range v { nv[i] = n.View() } return nv}
func eps(s ...string) []netip.AddrPort { var eps []netip.AddrPort for _, ep := range s { eps = append(eps, netip.MustParseAddrPort(ep)) } return eps}
func TestNetworkMapConcise(t *testing.T) { for _, tt := range []struct { name string nm *NetworkMap want string }{ { name: "basic", nm: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, { Key: testNodeKey(3), HomeDERP: 4, Endpoints: eps("10.2.0.100:12", "10.1.0.100:12345"), }, }), }, want: "netmap: self: [AQEBA] auth=machine-unknown u=? []\n [AgICA] D2 : 192.168.0.100:12 192.168.0.100:12354\n [AwMDA] D4 : 10.2.0.100:12 10.1.0.100:12345\n", }, } { t.Run(tt.name, func(t *testing.T) { var got string n := int(testing.AllocsPerRun(1000, func() { got = tt.nm.Concise() })) t.Logf("Allocs = %d", n) if got != tt.want { t.Errorf("Wrong output\n Got: %q\nWant: %q\n## Got (unescaped):\n%s\n## Want (unescaped):\n%s\n", got, tt.want, got, tt.want) } }) }}
func TestConciseDiffFrom(t *testing.T) { for _, tt := range []struct { name string a, b *NetworkMap want string }{ { name: "no_change", a: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, b: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, want: "", }, { name: "header_change", a: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, b: &NetworkMap{ NodeKey: testNodeKey(2), Peers: nodeViews([]*tailcfg.Node{ { Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, want: "-netmap: self: [AQEBA] auth=machine-unknown u=? []\n+netmap: self: [AgICA] auth=machine-unknown u=? []\n", }, { name: "peer_add", a: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, b: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 1, Key: testNodeKey(1), HomeDERP: 1, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, { ID: 3, Key: testNodeKey(3), HomeDERP: 3, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, want: "+ [AQEBA] D1 : 192.168.0.100:12 192.168.0.100:12354\n+ [AwMDA] D3 : 192.168.0.100:12 192.168.0.100:12354\n", }, { name: "peer_remove", a: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 1, Key: testNodeKey(1), HomeDERP: 1, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, { ID: 3, Key: testNodeKey(3), HomeDERP: 3, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, b: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "192.168.0.100:12354"), }, }), }, want: "- [AQEBA] D1 : 192.168.0.100:12 192.168.0.100:12354\n- [AwMDA] D3 : 192.168.0.100:12 192.168.0.100:12354\n", }, { name: "peer_port_change", a: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "1.1.1.1:1"), }, }), }, b: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:12", "1.1.1.1:2"), }, }), }, want: "- [AgICA] D2 : 192.168.0.100:12 1.1.1.1:1 \n+ [AgICA] D2 : 192.168.0.100:12 1.1.1.1:2 \n", }, { name: "disco_key_only_change", a: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:41641", "1.1.1.1:41641"), DiscoKey: testDiscoKey("f00f00f00f"), AllowedIPs: []netip.Prefix{netip.PrefixFrom(netaddr.IPv4(100, 102, 103, 104), 32)}, }, }), }, b: &NetworkMap{ NodeKey: testNodeKey(1), Peers: nodeViews([]*tailcfg.Node{ { ID: 2, Key: testNodeKey(2), HomeDERP: 2, Endpoints: eps("192.168.0.100:41641", "1.1.1.1:41641"), DiscoKey: testDiscoKey("ba4ba4ba4b"), AllowedIPs: []netip.Prefix{netip.PrefixFrom(netaddr.IPv4(100, 102, 103, 104), 32)}, }, }), }, want: "- [AgICA] d:f00f00f00f000000 D2 100.102.103.104 : 192.168.0.100:41641 1.1.1.1:41641\n+ [AgICA] d:ba4ba4ba4b000000 D2 100.102.103.104 : 192.168.0.100:41641 1.1.1.1:41641\n", }, } { t.Run(tt.name, func(t *testing.T) { var got string n := int(testing.AllocsPerRun(50, func() { got = tt.b.ConciseDiffFrom(tt.a) })) t.Logf("Allocs = %d", n) if got != tt.want { t.Errorf("Wrong output\n Got: %q\nWant: %q\n## Got (unescaped):\n%s\n## Want (unescaped):\n%s\n", got, tt.want, got, tt.want) } }) }}
func TestPeerIndexByNodeID(t *testing.T) { var nilPtr *NetworkMap if nilPtr.PeerIndexByNodeID(123) != -1 { t.Errorf("nil PeerIndexByNodeID should return -1") } var nm NetworkMap const min = 2 const max = 10000 const hole = max / 2 for nid := tailcfg.NodeID(2); nid <= max; nid++ { if nid == hole { continue } nm.Peers = append(nm.Peers, (&tailcfg.Node{ID: nid}).View()) } for want, nv := range nm.Peers { got := nm.PeerIndexByNodeID(nv.ID()) if got != want { t.Errorf("PeerIndexByNodeID(%v) = %v; want %v", nv.ID(), got, want) } } for _, miss := range []tailcfg.NodeID{min - 1, hole, max + 1} { if got := nm.PeerIndexByNodeID(miss); got != -1 { t.Errorf("PeerIndexByNodeID(%v) = %v; want -1", miss, got) } }}
func TestNoPrivateKeyMaterial(t *testing.T) { private := key.PrivateTypesForTest() for path := range typewalk.MatchingPaths(reflect.TypeFor[NetworkMap](), private.Contains) { t.Errorf("NetworkMap contains private key material at path: %q", path.Name) }}