mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-01 22:15:51 +00:00
6298018704
NetworkMap text diffs being empty were currently used to short-circuit calling magicsock's SetNetworkMap (via Engine.SetNetworkMap), but that went away inc7582dc2(0.100.0-230) Prior toc7582dc2(notably, in 0.100.0-225 and below, down to 0.100.0), a change in only disco key (as when a node restarts) but without endpoint changes (as would happen for a client not behind a NAT with random ports) could result in a "netmap diff: (none)" being printed, as well as Engine.SetNetworkMap being skipped, leading to broken discovery endpoints.c7582dc2fixed the Engine.SetNetworkMap skippage. This change fixes the "netmap diff: (none)" print so we'll actually see when a peer restarts with identical endpoints but a new discovery key.
285 lines
7.3 KiB
Go
285 lines
7.3 KiB
Go
// Copyright (c) 2020 Tailscale Inc & AUTHORS All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package controlclient
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"testing"
|
|
|
|
"github.com/tailscale/wireguard-go/wgcfg"
|
|
"tailscale.com/tailcfg"
|
|
)
|
|
|
|
func testNodeKey(b byte) (ret tailcfg.NodeKey) {
|
|
for i := range ret {
|
|
ret[i] = b
|
|
}
|
|
return
|
|
}
|
|
|
|
func testDiscoKey(hexPrefix string) (ret tailcfg.DiscoKey) {
|
|
b, err := hex.DecodeString(hexPrefix)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
copy(ret[:], b)
|
|
return
|
|
}
|
|
|
|
func TestNetworkMapConcise(t *testing.T) {
|
|
for _, tt := range []struct {
|
|
name string
|
|
nm *NetworkMap
|
|
want string
|
|
}{
|
|
{
|
|
name: "basic",
|
|
nm: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
{
|
|
Key: testNodeKey(3),
|
|
DERP: "127.3.3.40:4",
|
|
Endpoints: []string{"10.2.0.100:12", "10.1.0.100:12345"},
|
|
},
|
|
},
|
|
},
|
|
want: "netmap: self: [AQEBA] auth=machine-unknown []\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",
|
|
},
|
|
{
|
|
name: "debug_non_nil",
|
|
nm: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Debug: &tailcfg.Debug{},
|
|
},
|
|
want: "netmap: self: [AQEBA] auth=machine-unknown debug={} []\n",
|
|
},
|
|
{
|
|
name: "debug_values",
|
|
nm: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Debug: &tailcfg.Debug{LogHeapPprof: true},
|
|
},
|
|
want: "netmap: self: [AQEBA] auth=machine-unknown debug={\"LogHeapPprof\":true} []\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: []*tailcfg.Node{
|
|
{
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
},
|
|
},
|
|
b: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
},
|
|
},
|
|
want: "",
|
|
},
|
|
{
|
|
name: "header_change",
|
|
a: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
},
|
|
},
|
|
b: &NetworkMap{
|
|
NodeKey: testNodeKey(2),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
},
|
|
},
|
|
want: "-netmap: self: [AQEBA] auth=machine-unknown []\n+netmap: self: [AgICA] auth=machine-unknown []\n",
|
|
},
|
|
{
|
|
name: "peer_add",
|
|
a: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
},
|
|
},
|
|
b: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
ID: 1,
|
|
Key: testNodeKey(1),
|
|
DERP: "127.3.3.40:1",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
{
|
|
ID: 3,
|
|
Key: testNodeKey(3),
|
|
DERP: "127.3.3.40:3",
|
|
Endpoints: []string{"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: []*tailcfg.Node{
|
|
{
|
|
ID: 1,
|
|
Key: testNodeKey(1),
|
|
DERP: "127.3.3.40:1",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
{
|
|
ID: 3,
|
|
Key: testNodeKey(3),
|
|
DERP: "127.3.3.40:3",
|
|
Endpoints: []string{"192.168.0.100:12", "192.168.0.100:12354"},
|
|
},
|
|
},
|
|
},
|
|
b: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"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: []*tailcfg.Node{
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:12", "1.1.1.1:1"},
|
|
},
|
|
},
|
|
},
|
|
b: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"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: []*tailcfg.Node{
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:41641", "1.1.1.1:41641"},
|
|
DiscoKey: testDiscoKey("f00f00f00f"),
|
|
AllowedIPs: []wgcfg.CIDR{{IP: wgcfg.IPv4(100, 102, 103, 104), Mask: 32}},
|
|
},
|
|
},
|
|
},
|
|
b: &NetworkMap{
|
|
NodeKey: testNodeKey(1),
|
|
Peers: []*tailcfg.Node{
|
|
{
|
|
ID: 2,
|
|
Key: testNodeKey(2),
|
|
DERP: "127.3.3.40:2",
|
|
Endpoints: []string{"192.168.0.100:41641", "1.1.1.1:41641"},
|
|
DiscoKey: testDiscoKey("ba4ba4ba4b"),
|
|
AllowedIPs: []wgcfg.CIDR{{IP: wgcfg.IPv4(100, 102, 103, 104), Mask: 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)
|
|
}
|
|
})
|
|
}
|
|
}
|