|
|
|
// Copyright (c) Tailscale Inc & AUTHORS
|
|
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
|
|
|
|
package ipnlocal
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"encoding/json"
|
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"math"
|
|
|
|
"net"
|
|
|
|
"net/http"
|
|
|
|
"net/netip"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"reflect"
|
|
|
|
"slices"
|
ipn/{ipnauth,ipnlocal,ipnserver}: send the auth URL to the user who started interactive login
We add the ClientID() method to the ipnauth.Actor interface and updated ipnserver.actor to implement it.
This method returns a unique ID of the connected client if the actor represents one. It helps link a series
of interactions initiated by the client, such as when a notification needs to be sent back to a specific session,
rather than all active sessions, in response to a certain request.
We also add LocalBackend.WatchNotificationsAs and LocalBackend.StartLoginInteractiveAs methods,
which are like WatchNotifications and StartLoginInteractive but accept an additional parameter
specifying an ipnauth.Actor who initiates the operation. We store these actor identities in
watchSession.owner and LocalBackend.authActor, respectively,and implement LocalBackend.sendTo
and related helper methods to enable sending notifications to watchSessions associated with actors
(or, more broadly, identifiable recipients).
We then use the above to change who receives the BrowseToURL notifications:
- For user-initiated, interactive logins, the notification is delivered only to the user who initiated the
process. If the initiating actor represents a specific connected client, the URL notification is sent back
to the same LocalAPI client that called StartLoginInteractive. Otherwise, the notification is sent to all
clients connected as that user.
Currently, we only differentiate between users on Windows, as it is inherently a multi-user OS.
- In all other cases (e.g., node key expiration), we send the notification to all connected users.
Updates tailscale/corp#18342
Signed-off-by: Nick Khyl <nickk@tailscale.com>
1 month ago
|
|
|
"strings"
|
|
|
|
"sync"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
|
|
"go4.org/netipx"
|
|
|
|
"golang.org/x/net/dns/dnsmessage"
|
|
|
|
"tailscale.com/appc"
|
|
|
|
"tailscale.com/appc/appctest"
|
|
|
|
"tailscale.com/clientupdate"
|
|
|
|
"tailscale.com/control/controlclient"
|
|
|
|
"tailscale.com/drive"
|
|
|
|
"tailscale.com/drive/driveimpl"
|
|
|
|
"tailscale.com/health"
|
|
|
|
"tailscale.com/hostinfo"
|
|
|
|
"tailscale.com/ipn"
|
|
|
|
"tailscale.com/ipn/conffile"
|
ipn/{ipnauth,ipnlocal,ipnserver}: send the auth URL to the user who started interactive login
We add the ClientID() method to the ipnauth.Actor interface and updated ipnserver.actor to implement it.
This method returns a unique ID of the connected client if the actor represents one. It helps link a series
of interactions initiated by the client, such as when a notification needs to be sent back to a specific session,
rather than all active sessions, in response to a certain request.
We also add LocalBackend.WatchNotificationsAs and LocalBackend.StartLoginInteractiveAs methods,
which are like WatchNotifications and StartLoginInteractive but accept an additional parameter
specifying an ipnauth.Actor who initiates the operation. We store these actor identities in
watchSession.owner and LocalBackend.authActor, respectively,and implement LocalBackend.sendTo
and related helper methods to enable sending notifications to watchSessions associated with actors
(or, more broadly, identifiable recipients).
We then use the above to change who receives the BrowseToURL notifications:
- For user-initiated, interactive logins, the notification is delivered only to the user who initiated the
process. If the initiating actor represents a specific connected client, the URL notification is sent back
to the same LocalAPI client that called StartLoginInteractive. Otherwise, the notification is sent to all
clients connected as that user.
Currently, we only differentiate between users on Windows, as it is inherently a multi-user OS.
- In all other cases (e.g., node key expiration), we send the notification to all connected users.
Updates tailscale/corp#18342
Signed-off-by: Nick Khyl <nickk@tailscale.com>
1 month ago
|
|
|
"tailscale.com/ipn/ipnauth"
|
|
|
|
"tailscale.com/ipn/store/mem"
|
|
|
|
"tailscale.com/net/netcheck"
|
|
|
|
"tailscale.com/net/netmon"
|
|
|
|
"tailscale.com/net/tsaddr"
|
|
|
|
"tailscale.com/net/tsdial"
|
|
|
|
"tailscale.com/tailcfg"
|
|
|
|
"tailscale.com/tsd"
|
|
|
|
"tailscale.com/tstest"
|
|
|
|
"tailscale.com/types/dnstype"
|
|
|
|
"tailscale.com/types/key"
|
|
|
|
"tailscale.com/types/logger"
|
|
|
|
"tailscale.com/types/logid"
|
|
|
|
"tailscale.com/types/netmap"
|
|
|
|
"tailscale.com/types/opt"
|
|
|
|
"tailscale.com/types/ptr"
|
|
|
|
"tailscale.com/types/views"
|
|
|
|
"tailscale.com/util/dnsname"
|
|
|
|
"tailscale.com/util/mak"
|
|
|
|
"tailscale.com/util/must"
|
|
|
|
"tailscale.com/util/set"
|
|
|
|
"tailscale.com/util/syspolicy"
|
|
|
|
"tailscale.com/util/syspolicy/setting"
|
|
|
|
"tailscale.com/util/syspolicy/source"
|
|
|
|
"tailscale.com/wgengine"
|
|
|
|
"tailscale.com/wgengine/filter"
|
|
|
|
"tailscale.com/wgengine/wgcfg"
|
|
|
|
)
|
|
|
|
|
|
|
|
func fakeStoreRoutes(*appc.RouteInfo) error { return nil }
|
|
|
|
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
func inRemove(ip netip.Addr) bool {
|
|
|
|
for _, pfx := range removeFromDefaultRoute {
|
|
|
|
if pfx.Contains(ip) {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestShrinkDefaultRoute(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
route string
|
|
|
|
in []string
|
|
|
|
out []string
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
localIPFn func(netip.Addr) bool // true if this machine's local IP address should be "in" after shrinking.
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
route: "0.0.0.0/0",
|
|
|
|
in: []string{"1.2.3.4", "25.0.0.1"},
|
|
|
|
out: []string{
|
|
|
|
"10.0.0.1",
|
|
|
|
"10.255.255.255",
|
|
|
|
"192.168.0.1",
|
|
|
|
"192.168.255.255",
|
|
|
|
"172.16.0.1",
|
|
|
|
"172.31.255.255",
|
|
|
|
"100.101.102.103",
|
|
|
|
"224.0.0.1",
|
|
|
|
"169.254.169.254",
|
|
|
|
// Some random IPv6 stuff that shouldn't be in a v4
|
|
|
|
// default route.
|
|
|
|
"fe80::",
|
|
|
|
"2601::1",
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
localIPFn: func(ip netip.Addr) bool { return !inRemove(ip) && ip.Is4() },
|
|
|
|
},
|
|
|
|
{
|
|
|
|
route: "::/0",
|
|
|
|
in: []string{"::1", "2601::1"},
|
|
|
|
out: []string{
|
|
|
|
"fe80::1",
|
|
|
|
"ff00::1",
|
|
|
|
tsaddr.TailscaleULARange().Addr().String(),
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
localIPFn: func(ip netip.Addr) bool { return !inRemove(ip) && ip.Is6() },
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
// Construct a fake local network environment to make this test hermetic.
|
|
|
|
// localInterfaceRoutes and hostIPs would normally come from calling interfaceRoutes,
|
|
|
|
// and localAddresses would normally come from calling interfaces.LocalAddresses.
|
|
|
|
var b netipx.IPSetBuilder
|
|
|
|
for _, c := range []string{"127.0.0.0/8", "192.168.9.0/24", "fe80::/32"} {
|
|
|
|
p := netip.MustParsePrefix(c)
|
|
|
|
b.AddPrefix(p)
|
|
|
|
}
|
|
|
|
localInterfaceRoutes, err := b.IPSet()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
hostIPs := []netip.Addr{
|
|
|
|
netip.MustParseAddr("127.0.0.1"),
|
|
|
|
netip.MustParseAddr("192.168.9.39"),
|
|
|
|
netip.MustParseAddr("fe80::1"),
|
|
|
|
netip.MustParseAddr("fe80::437d:feff:feca:49a7"),
|
|
|
|
}
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
localAddresses := []netip.Addr{
|
|
|
|
netip.MustParseAddr("192.168.9.39"),
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, test := range tests {
|
|
|
|
def := netip.MustParsePrefix(test.route)
|
|
|
|
got, err := shrinkDefaultRoute(def, localInterfaceRoutes, hostIPs)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("shrinkDefaultRoute(%q): %v", test.route, err)
|
|
|
|
}
|
|
|
|
for _, ip := range test.in {
|
|
|
|
if !got.Contains(netip.MustParseAddr(ip)) {
|
|
|
|
t.Errorf("shrink(%q).Contains(%v) = false, want true", test.route, ip)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for _, ip := range test.out {
|
|
|
|
if got.Contains(netip.MustParseAddr(ip)) {
|
|
|
|
t.Errorf("shrink(%q).Contains(%v) = true, want false", test.route, ip)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for _, ip := range localAddresses {
|
|
|
|
want := test.localIPFn(ip)
|
|
|
|
if gotContains := got.Contains(ip); gotContains != want {
|
|
|
|
t.Errorf("shrink(%q).Contains(%v) = %v, want %v", test.route, ip, gotContains, want)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestPeerRoutes(t *testing.T) {
|
|
|
|
pp := netip.MustParsePrefix
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
peers []wgcfg.Peer
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
want []netip.Prefix
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "small_v4",
|
|
|
|
peers: []wgcfg.Peer{
|
|
|
|
{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("100.101.102.103/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
want: []netip.Prefix{
|
|
|
|
pp("100.101.102.103/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "big_v4",
|
|
|
|
peers: []wgcfg.Peer{
|
|
|
|
{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("100.101.102.103/32"),
|
|
|
|
pp("100.101.102.104/32"),
|
|
|
|
pp("100.101.102.105/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
want: []netip.Prefix{
|
|
|
|
pp("100.64.0.0/10"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "has_1_v6",
|
|
|
|
peers: []wgcfg.Peer{
|
|
|
|
{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("fd7a:115c:a1e0:ab12:4843:cd96:6258:b240/128"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
want: []netip.Prefix{
|
|
|
|
pp("fd7a:115c:a1e0::/48"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "has_2_v6",
|
|
|
|
peers: []wgcfg.Peer{
|
|
|
|
{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("fd7a:115c:a1e0:ab12:4843:cd96:6258:b240/128"),
|
|
|
|
pp("fd7a:115c:a1e0:ab12:4843:cd96:6258:b241/128"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
want: []netip.Prefix{
|
|
|
|
pp("fd7a:115c:a1e0::/48"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "big_v4_big_v6",
|
|
|
|
peers: []wgcfg.Peer{
|
|
|
|
{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("100.101.102.103/32"),
|
|
|
|
pp("100.101.102.104/32"),
|
|
|
|
pp("100.101.102.105/32"),
|
|
|
|
pp("fd7a:115c:a1e0:ab12:4843:cd96:6258:b240/128"),
|
|
|
|
pp("fd7a:115c:a1e0:ab12:4843:cd96:6258:b241/128"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
want: []netip.Prefix{
|
|
|
|
pp("100.64.0.0/10"),
|
|
|
|
pp("fd7a:115c:a1e0::/48"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "output-should-be-sorted",
|
|
|
|
peers: []wgcfg.Peer{
|
|
|
|
{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("100.64.0.2/32"),
|
|
|
|
pp("10.0.0.0/16"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("100.64.0.1/32"),
|
|
|
|
pp("10.0.0.0/8"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
want: []netip.Prefix{
|
|
|
|
pp("10.0.0.0/8"),
|
|
|
|
pp("10.0.0.0/16"),
|
|
|
|
pp("100.64.0.1/32"),
|
|
|
|
pp("100.64.0.2/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "skip-unmasked-prefixes",
|
|
|
|
peers: []wgcfg.Peer{
|
|
|
|
{
|
|
|
|
PublicKey: key.NewNode().Public(),
|
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
pp("100.64.0.2/32"),
|
|
|
|
pp("10.0.0.100/16"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: []netip.Prefix{
|
|
|
|
pp("100.64.0.2/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got := peerRoutes(t.Logf, tt.peers, 2)
|
|
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
|
|
t.Errorf("got = %v; want %v", got, tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestPeerAPIBase(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
nm *netmap.NetworkMap
|
|
|
|
peer *tailcfg.Node
|
|
|
|
want string
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "nil_netmap",
|
|
|
|
peer: new(tailcfg.Node),
|
|
|
|
want: "",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "nil_peer",
|
|
|
|
nm: new(netmap.NetworkMap),
|
|
|
|
want: "",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "self_only_4_them_both",
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.1/32"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
peer: &tailcfg.Node{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.2/32"),
|
|
|
|
netip.MustParsePrefix("fe70::2/128"),
|
|
|
|
},
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{
|
|
|
|
Services: []tailcfg.Service{
|
|
|
|
{Proto: "peerapi4", Port: 444},
|
|
|
|
{Proto: "peerapi6", Port: 666},
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
want: "http://100.64.1.2:444",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "self_only_6_them_both",
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("fe70::1/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
peer: &tailcfg.Node{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.2/32"),
|
|
|
|
netip.MustParsePrefix("fe70::2/128"),
|
|
|
|
},
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{
|
|
|
|
Services: []tailcfg.Service{
|
|
|
|
{Proto: "peerapi4", Port: 444},
|
|
|
|
{Proto: "peerapi6", Port: 666},
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
want: "http://[fe70::2]:666",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "self_both_them_only_4",
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.1/32"),
|
|
|
|
netip.MustParsePrefix("fe70::1/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
peer: &tailcfg.Node{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.2/32"),
|
|
|
|
netip.MustParsePrefix("fe70::2/128"),
|
|
|
|
},
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{
|
|
|
|
Services: []tailcfg.Service{
|
|
|
|
{Proto: "peerapi4", Port: 444},
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
want: "http://100.64.1.2:444",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "self_both_them_only_6",
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.1/32"),
|
|
|
|
netip.MustParsePrefix("fe70::1/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
peer: &tailcfg.Node{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.2/32"),
|
|
|
|
netip.MustParsePrefix("fe70::2/128"),
|
|
|
|
},
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{
|
|
|
|
Services: []tailcfg.Service{
|
|
|
|
{Proto: "peerapi6", Port: 666},
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
want: "http://[fe70::2]:666",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "self_both_them_no_peerapi_service",
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.1/32"),
|
|
|
|
netip.MustParsePrefix("fe70::1/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
peer: &tailcfg.Node{
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.2/32"),
|
|
|
|
netip.MustParsePrefix("fe70::2/128"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: "",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got := peerAPIBase(tt.nm, tt.peer.View())
|
|
|
|
if got != tt.want {
|
|
|
|
t.Errorf("got %q; want %q", got, tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
type panicOnUseTransport struct{}
|
|
|
|
|
|
|
|
func (panicOnUseTransport) RoundTrip(*http.Request) (*http.Response, error) {
|
|
|
|
panic("unexpected HTTP request")
|
|
|
|
}
|
|
|
|
|
|
|
|
func newTestLocalBackend(t testing.TB) *LocalBackend {
|
|
|
|
return newTestLocalBackendWithSys(t, new(tsd.System))
|
|
|
|
}
|
|
|
|
|
|
|
|
// newTestLocalBackendWithSys creates a new LocalBackend with the given tsd.System.
|
|
|
|
// If the state store or engine are not set in sys, they will be set to a new
|
|
|
|
// in-memory store and fake userspace engine, respectively.
|
|
|
|
func newTestLocalBackendWithSys(t testing.TB, sys *tsd.System) *LocalBackend {
|
|
|
|
var logf logger.Logf = logger.Discard
|
|
|
|
if _, ok := sys.StateStore.GetOK(); !ok {
|
|
|
|
sys.Set(new(mem.Store))
|
|
|
|
}
|
|
|
|
if _, ok := sys.Engine.GetOK(); !ok {
|
|
|
|
eng, err := wgengine.NewFakeUserspaceEngine(logf, sys.Set, sys.HealthTracker(), sys.UserMetricsRegistry())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("NewFakeUserspaceEngine: %v", err)
|
|
|
|
}
|
|
|
|
t.Cleanup(eng.Close)
|
|
|
|
sys.Set(eng)
|
|
|
|
}
|
|
|
|
lb, err := NewLocalBackend(logf, logid.PublicID{}, sys, 0)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("NewLocalBackend: %v", err)
|
|
|
|
}
|
|
|
|
return lb
|
|
|
|
}
|
|
|
|
|
|
|
|
// Issue 1573: don't generate a machine key if we don't want to be running.
|
|
|
|
func TestLazyMachineKeyGeneration(t *testing.T) {
|
|
|
|
tstest.Replace(t, &panicOnMachineKeyGeneration, func() bool { return true })
|
|
|
|
|
|
|
|
lb := newTestLocalBackend(t)
|
|
|
|
lb.SetHTTPTestClient(&http.Client{
|
|
|
|
Transport: panicOnUseTransport{}, // validate we don't send HTTP requests
|
|
|
|
})
|
|
|
|
|
|
|
|
if err := lb.Start(ipn.Options{}); err != nil {
|
|
|
|
t.Fatalf("Start: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Give the controlclient package goroutines (if they're
|
|
|
|
// accidentally started) extra time to schedule and run (and thus
|
|
|
|
// hit panicOnUseTransport).
|
|
|
|
time.Sleep(500 * time.Millisecond)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestZeroExitNodeViaLocalAPI(t *testing.T) {
|
|
|
|
lb := newTestLocalBackend(t)
|
|
|
|
|
|
|
|
// Give it an initial exit node in use.
|
|
|
|
if _, err := lb.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
ExitNodeIDSet: true,
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
ExitNodeID: "foo",
|
|
|
|
},
|
|
|
|
}); err != nil {
|
|
|
|
t.Fatalf("enabling first exit node: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// SetUseExitNodeEnabled(false) "remembers" the prior exit node.
|
|
|
|
if _, err := lb.SetUseExitNodeEnabled(false); err != nil {
|
|
|
|
t.Fatal("expected failure")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Zero the exit node
|
|
|
|
pv, err := lb.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
ExitNodeIDSet: true,
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
ExitNodeID: "",
|
|
|
|
},
|
|
|
|
})
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("enabling first exit node: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// We just set the internal exit node to the empty string, so InternalExitNodePrior should
|
|
|
|
// also be zero'd
|
|
|
|
if got, want := pv.InternalExitNodePrior(), tailcfg.StableNodeID(""); got != want {
|
|
|
|
t.Fatalf("unexpected InternalExitNodePrior %q, want: %q", got, want)
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSetUseExitNodeEnabled(t *testing.T) {
|
|
|
|
lb := newTestLocalBackend(t)
|
|
|
|
|
|
|
|
// Can't turn it on if it never had an old value.
|
|
|
|
if _, err := lb.SetUseExitNodeEnabled(true); err == nil {
|
|
|
|
t.Fatal("expected success")
|
|
|
|
}
|
|
|
|
|
|
|
|
// But we can turn it off when it's already off.
|
|
|
|
if _, err := lb.SetUseExitNodeEnabled(false); err != nil {
|
|
|
|
t.Fatal("expected failure")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Give it an initial exit node in use.
|
|
|
|
if _, err := lb.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
ExitNodeIDSet: true,
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
ExitNodeID: "foo",
|
|
|
|
},
|
|
|
|
}); err != nil {
|
|
|
|
t.Fatalf("enabling first exit node: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Now turn off that exit node.
|
|
|
|
if prefs, err := lb.SetUseExitNodeEnabled(false); err != nil {
|
|
|
|
t.Fatal("expected failure")
|
|
|
|
} else {
|
|
|
|
if g, w := prefs.ExitNodeID(), tailcfg.StableNodeID(""); g != w {
|
|
|
|
t.Fatalf("unexpected exit node ID %q; want %q", g, w)
|
|
|
|
}
|
|
|
|
if g, w := prefs.InternalExitNodePrior(), tailcfg.StableNodeID("foo"); g != w {
|
|
|
|
t.Fatalf("unexpected exit node prior %q; want %q", g, w)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// And turn it back on.
|
|
|
|
if prefs, err := lb.SetUseExitNodeEnabled(true); err != nil {
|
|
|
|
t.Fatal("expected failure")
|
|
|
|
} else {
|
|
|
|
if g, w := prefs.ExitNodeID(), tailcfg.StableNodeID("foo"); g != w {
|
|
|
|
t.Fatalf("unexpected exit node ID %q; want %q", g, w)
|
|
|
|
}
|
|
|
|
if g, w := prefs.InternalExitNodePrior(), tailcfg.StableNodeID("foo"); g != w {
|
|
|
|
t.Fatalf("unexpected exit node prior %q; want %q", g, w)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Verify we block setting an Internal field.
|
|
|
|
if _, err := lb.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
InternalExitNodePriorSet: true,
|
|
|
|
}); err == nil {
|
|
|
|
t.Fatalf("unexpected success; want an error trying to set an internal field")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestFileTargets(t *testing.T) {
|
|
|
|
b := new(LocalBackend)
|
|
|
|
_, err := b.FileTargets()
|
|
|
|
if got, want := fmt.Sprint(err), "not connected to the tailnet"; got != want {
|
|
|
|
t.Errorf("before connect: got %q; want %q", got, want)
|
|
|
|
}
|
|
|
|
|
|
|
|
b.netMap = new(netmap.NetworkMap)
|
|
|
|
_, err = b.FileTargets()
|
|
|
|
if got, want := fmt.Sprint(err), "not connected to the tailnet"; got != want {
|
|
|
|
t.Errorf("non-running netmap: got %q; want %q", got, want)
|
|
|
|
}
|
|
|
|
|
|
|
|
b.state = ipn.Running
|
|
|
|
_, err = b.FileTargets()
|
|
|
|
if got, want := fmt.Sprint(err), "file sharing not enabled by Tailscale admin"; got != want {
|
|
|
|
t.Errorf("without cap: got %q; want %q", got, want)
|
|
|
|
}
|
|
|
|
|
|
|
|
b.capFileSharing = true
|
|
|
|
got, err := b.FileTargets()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if len(got) != 0 {
|
|
|
|
t.Fatalf("unexpected %d peers", len(got))
|
|
|
|
}
|
|
|
|
|
|
|
|
var peerMap map[tailcfg.NodeID]tailcfg.NodeView
|
|
|
|
mak.NonNil(&peerMap)
|
|
|
|
var nodeID tailcfg.NodeID
|
|
|
|
nodeID = 1234
|
|
|
|
peer := &tailcfg.Node{
|
|
|
|
ID: 1234,
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{OS: "tvOS"}).View(),
|
|
|
|
}
|
|
|
|
peerMap[nodeID] = peer.View()
|
|
|
|
b.peers = peerMap
|
|
|
|
got, err = b.FileTargets()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if len(got) != 0 {
|
|
|
|
t.Fatalf("unexpected %d peers", len(got))
|
|
|
|
}
|
|
|
|
// (other cases handled by TestPeerAPIBase above)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestInternalAndExternalInterfaces(t *testing.T) {
|
|
|
|
type interfacePrefix struct {
|
|
|
|
i netmon.Interface
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
pfx netip.Prefix
|
|
|
|
}
|
|
|
|
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
masked := func(ips ...interfacePrefix) (pfxs []netip.Prefix) {
|
|
|
|
for _, ip := range ips {
|
|
|
|
pfxs = append(pfxs, ip.pfx.Masked())
|
|
|
|
}
|
|
|
|
return pfxs
|
|
|
|
}
|
|
|
|
iList := func(ips ...interfacePrefix) (il netmon.InterfaceList) {
|
|
|
|
for _, ip := range ips {
|
|
|
|
il = append(il, ip.i)
|
|
|
|
}
|
|
|
|
return il
|
|
|
|
}
|
|
|
|
newInterface := func(name, pfx string, wsl2, loopback bool) interfacePrefix {
|
|
|
|
ippfx := netip.MustParsePrefix(pfx)
|
|
|
|
ip := netmon.Interface{
|
|
|
|
Interface: &net.Interface{},
|
|
|
|
AltAddrs: []net.Addr{
|
|
|
|
netipx.PrefixIPNet(ippfx),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
if loopback {
|
|
|
|
ip.Flags = net.FlagLoopback
|
|
|
|
}
|
|
|
|
if wsl2 {
|
|
|
|
ip.HardwareAddr = []byte{0x00, 0x15, 0x5d, 0x00, 0x00, 0x00}
|
|
|
|
}
|
|
|
|
return interfacePrefix{i: ip, pfx: ippfx}
|
|
|
|
}
|
|
|
|
var (
|
|
|
|
en0 = newInterface("en0", "10.20.2.5/16", false, false)
|
|
|
|
en1 = newInterface("en1", "192.168.1.237/24", false, false)
|
|
|
|
wsl = newInterface("wsl", "192.168.5.34/24", true, false)
|
|
|
|
loopback = newInterface("lo0", "127.0.0.1/8", false, true)
|
|
|
|
)
|
|
|
|
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
goos string
|
|
|
|
il netmon.InterfaceList
|
all: convert more code to use net/netip directly
perl -i -npe 's,netaddr.IPPrefixFrom,netip.PrefixFrom,' $(git grep -l -F netaddr.)
perl -i -npe 's,netaddr.IPPortFrom,netip.AddrPortFrom,' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPrefix,netip.Prefix,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPPort,netip.AddrPort,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IP\b,netip.Addr,g' $(git grep -l -F netaddr. )
perl -i -npe 's,netaddr.IPv6Raw\b,netip.AddrFrom16,g' $(git grep -l -F netaddr. )
goimports -w .
Then delete some stuff from the net/netaddr shim package which is no
longer neeed.
Updates #5162
Change-Id: Ia7a86893fe21c7e3ee1ec823e8aba288d4566cd8
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
2 years ago
|
|
|
wantInt []netip.Prefix
|
|
|
|
wantExt []netip.Prefix
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "single-interface",
|
|
|
|
goos: "linux",
|
|
|
|
il: iList(
|
|
|
|
en0,
|
|
|
|
loopback,
|
|
|
|
),
|
|
|
|
wantInt: masked(loopback),
|
|
|
|
wantExt: masked(en0),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "multiple-interfaces",
|
|
|
|
goos: "linux",
|
|
|
|
il: iList(
|
|
|
|
en0,
|
|
|
|
en1,
|
|
|
|
wsl,
|
|
|
|
loopback,
|
|
|
|
),
|
|
|
|
wantInt: masked(loopback),
|
|
|
|
wantExt: masked(en0, en1, wsl),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wsl2",
|
|
|
|
goos: "windows",
|
|
|
|
il: iList(
|
|
|
|
en0,
|
|
|
|
en1,
|
|
|
|
wsl,
|
|
|
|
loopback,
|
|
|
|
),
|
|
|
|
wantInt: masked(loopback, wsl),
|
|
|
|
wantExt: masked(en0, en1),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tc := range tests {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
gotInt, gotExt, err := internalAndExternalInterfacesFrom(tc.il, tc.goos)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(gotInt, tc.wantInt) {
|
|
|
|
t.Errorf("unexpected internal prefixes\ngot %v\nwant %v", gotInt, tc.wantInt)
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(gotExt, tc.wantExt) {
|
|
|
|
t.Errorf("unexpected external prefixes\ngot %v\nwant %v", gotExt, tc.wantExt)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestPacketFilterPermitsUnlockedNodes(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
peers []*tailcfg.Node
|
|
|
|
filter []filter.Match
|
|
|
|
want bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "empty",
|
|
|
|
want: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "no-unsigned",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{ID: 1},
|
|
|
|
},
|
|
|
|
want: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unsigned-good",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{ID: 1, UnsignedPeerAPIOnly: true},
|
|
|
|
},
|
|
|
|
want: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unsigned-bad",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
UnsignedPeerAPIOnly: true,
|
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.0.0/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
filter: []filter.Match{
|
|
|
|
{
|
|
|
|
Srcs: []netip.Prefix{netip.MustParsePrefix("100.64.0.0/32")},
|
|
|
|
Dsts: []filter.NetPortRange{
|
|
|
|
{
|
|
|
|
Net: netip.MustParsePrefix("100.99.0.0/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unsigned-bad-src-is-superset",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
UnsignedPeerAPIOnly: true,
|
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.0.0/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
filter: []filter.Match{
|
|
|
|
{
|
|
|
|
Srcs: []netip.Prefix{netip.MustParsePrefix("100.64.0.0/24")},
|
|
|
|
Dsts: []filter.NetPortRange{
|
|
|
|
{
|
|
|
|
Net: netip.MustParsePrefix("100.99.0.0/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unsigned-okay-because-no-dsts",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
UnsignedPeerAPIOnly: true,
|
|
|
|
AllowedIPs: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.0.0/32"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
filter: []filter.Match{
|
|
|
|
{
|
|
|
|
Srcs: []netip.Prefix{netip.MustParsePrefix("100.64.0.0/32")},
|
|
|
|
Caps: []filter.CapMatch{
|
|
|
|
{
|
|
|
|
Dst: netip.MustParsePrefix("100.99.0.0/32"),
|
|
|
|
Cap: "foo",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
if got := packetFilterPermitsUnlockedNodes(peersMap(nodeViews(tt.peers)), tt.filter); got != tt.want {
|
|
|
|
t.Errorf("got %v, want %v", got, tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestStatusPeerCapabilities(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
peers []tailcfg.NodeView
|
|
|
|
expectedPeerCapabilities map[tailcfg.StableNodeID][]tailcfg.NodeCapability
|
|
|
|
expectedPeerCapMap map[tailcfg.StableNodeID]tailcfg.NodeCapMap
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "peers-with-capabilities",
|
|
|
|
peers: []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
ID: 1,
|
|
|
|
StableID: "foo",
|
|
|
|
IsWireGuardOnly: true,
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{}).View(),
|
|
|
|
Capabilities: []tailcfg.NodeCapability{tailcfg.CapabilitySSH},
|
|
|
|
CapMap: (tailcfg.NodeCapMap)(map[tailcfg.NodeCapability][]tailcfg.RawMessage{
|
|
|
|
tailcfg.CapabilitySSH: nil,
|
|
|
|
}),
|
|
|
|
}).View(),
|
|
|
|
(&tailcfg.Node{
|
|
|
|
ID: 2,
|
|
|
|
StableID: "bar",
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{}).View(),
|
|
|
|
Capabilities: []tailcfg.NodeCapability{tailcfg.CapabilityAdmin},
|
|
|
|
CapMap: (tailcfg.NodeCapMap)(map[tailcfg.NodeCapability][]tailcfg.RawMessage{
|
|
|
|
tailcfg.CapabilityAdmin: {`{"test": "true}`},
|
|
|
|
}),
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
expectedPeerCapabilities: map[tailcfg.StableNodeID][]tailcfg.NodeCapability{
|
|
|
|
tailcfg.StableNodeID("foo"): {tailcfg.CapabilitySSH},
|
|
|
|
tailcfg.StableNodeID("bar"): {tailcfg.CapabilityAdmin},
|
|
|
|
},
|
|
|
|
expectedPeerCapMap: map[tailcfg.StableNodeID]tailcfg.NodeCapMap{
|
|
|
|
tailcfg.StableNodeID("foo"): (tailcfg.NodeCapMap)(map[tailcfg.NodeCapability][]tailcfg.RawMessage{
|
|
|
|
tailcfg.CapabilitySSH: nil,
|
|
|
|
}),
|
|
|
|
tailcfg.StableNodeID("bar"): (tailcfg.NodeCapMap)(map[tailcfg.NodeCapability][]tailcfg.RawMessage{
|
|
|
|
tailcfg.CapabilityAdmin: {`{"test": "true}`},
|
|
|
|
}),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "peers-without-capabilities",
|
|
|
|
peers: []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
ID: 1,
|
|
|
|
StableID: "foo",
|
|
|
|
IsWireGuardOnly: true,
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{}).View(),
|
|
|
|
}).View(),
|
|
|
|
(&tailcfg.Node{
|
|
|
|
ID: 2,
|
|
|
|
StableID: "bar",
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{}).View(),
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
b := newTestLocalBackend(t)
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
b.setNetMapLocked(&netmap.NetworkMap{
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
MachineAuthorized: true,
|
|
|
|
Addresses: ipps("100.101.101.101"),
|
|
|
|
}).View(),
|
|
|
|
Peers: tt.peers,
|
|
|
|
})
|
|
|
|
got := b.Status()
|
|
|
|
for _, peer := range got.Peer {
|
|
|
|
if !reflect.DeepEqual(peer.Capabilities, tt.expectedPeerCapabilities[peer.ID]) {
|
|
|
|
t.Errorf("peer capabilities: expected %v got %v", tt.expectedPeerCapabilities, peer.Capabilities)
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(peer.CapMap, tt.expectedPeerCapMap[peer.ID]) {
|
|
|
|
t.Errorf("peer capmap: expected %v got %v", tt.expectedPeerCapMap, peer.CapMap)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// legacyBackend was the interface between Tailscale frontends
|
|
|
|
// (e.g. cmd/tailscale, iOS/MacOS/Windows GUIs) and the tailscale
|
|
|
|
// backend (e.g. cmd/tailscaled) running on the same machine.
|
|
|
|
// (It has nothing to do with the interface between the backends
|
|
|
|
// and the cloud control plane.)
|
|
|
|
type legacyBackend interface {
|
|
|
|
// SetNotifyCallback sets the callback to be called on updates
|
|
|
|
// from the backend to the client.
|
|
|
|
SetNotifyCallback(func(ipn.Notify))
|
|
|
|
// Start starts or restarts the backend, typically when a
|
|
|
|
// frontend client connects.
|
|
|
|
Start(ipn.Options) error
|
|
|
|
}
|
|
|
|
|
|
|
|
// Verify that LocalBackend still implements the legacyBackend interface
|
|
|
|
// for now, at least until the macOS and iOS clients move off of it.
|
|
|
|
var _ legacyBackend = (*LocalBackend)(nil)
|
|
|
|
|
|
|
|
func TestWatchNotificationsCallbacks(t *testing.T) {
|
|
|
|
b := new(LocalBackend)
|
|
|
|
n := new(ipn.Notify)
|
|
|
|
b.WatchNotifications(context.Background(), 0, func() {
|
|
|
|
b.mu.Lock()
|
|
|
|
defer b.mu.Unlock()
|
|
|
|
|
|
|
|
// Ensure a watcher has been installed.
|
|
|
|
if len(b.notifyWatchers) != 1 {
|
|
|
|
t.Fatalf("unexpected number of watchers in new LocalBackend, want: 1 got: %v", len(b.notifyWatchers))
|
|
|
|
}
|
|
|
|
// Send a notification. Range over notifyWatchers to get the channel
|
|
|
|
// because WatchNotifications doesn't expose the handle for it.
|
|
|
|
for _, sess := range b.notifyWatchers {
|
|
|
|
select {
|
|
|
|
case sess.ch <- n:
|
|
|
|
default:
|
|
|
|
t.Fatalf("could not send notification")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}, func(roNotify *ipn.Notify) bool {
|
|
|
|
if roNotify != n {
|
|
|
|
t.Fatalf("unexpected notification received. want: %v got: %v", n, roNotify)
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
})
|
|
|
|
|
|
|
|
// Ensure watchers have been cleaned up.
|
|
|
|
b.mu.Lock()
|
|
|
|
defer b.mu.Unlock()
|
|
|
|
if len(b.notifyWatchers) != 0 {
|
|
|
|
t.Fatalf("unexpected number of watchers in new LocalBackend, want: 0 got: %v", len(b.notifyWatchers))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// tests LocalBackend.updateNetmapDeltaLocked
|
|
|
|
func TestUpdateNetmapDelta(t *testing.T) {
|
|
|
|
b := newTestLocalBackend(t)
|
|
|
|
if b.updateNetmapDeltaLocked(nil) {
|
|
|
|
t.Errorf("updateNetmapDeltaLocked() = true, want false with nil netmap")
|
|
|
|
}
|
|
|
|
|
|
|
|
b.netMap = &netmap.NetworkMap{}
|
|
|
|
for i := range 5 {
|
|
|
|
b.netMap.Peers = append(b.netMap.Peers, (&tailcfg.Node{ID: (tailcfg.NodeID(i) + 1)}).View())
|
|
|
|
}
|
|
|
|
b.updatePeersFromNetmapLocked(b.netMap)
|
|
|
|
|
|
|
|
someTime := time.Unix(123, 0)
|
|
|
|
muts, ok := netmap.MutationsFromMapResponse(&tailcfg.MapResponse{
|
|
|
|
PeersChangedPatch: []*tailcfg.PeerChange{
|
|
|
|
{
|
|
|
|
NodeID: 1,
|
|
|
|
DERPRegion: 1,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
NodeID: 2,
|
|
|
|
Online: ptr.To(true),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
NodeID: 3,
|
|
|
|
Online: ptr.To(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
NodeID: 4,
|
|
|
|
LastSeen: ptr.To(someTime),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}, someTime)
|
|
|
|
if !ok {
|
|
|
|
t.Fatal("netmap.MutationsFromMapResponse failed")
|
|
|
|
}
|
|
|
|
|
|
|
|
if !b.updateNetmapDeltaLocked(muts) {
|
|
|
|
t.Fatalf("updateNetmapDeltaLocked() = false, want true with new netmap")
|
|
|
|
}
|
|
|
|
|
|
|
|
wants := []*tailcfg.Node{
|
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
DERP: "127.3.3.40:1",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 2,
|
|
|
|
Online: ptr.To(true),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 3,
|
|
|
|
Online: ptr.To(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 4,
|
|
|
|
LastSeen: ptr.To(someTime),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, want := range wants {
|
|
|
|
gotv, ok := b.peers[want.ID]
|
|
|
|
if !ok {
|
|
|
|
t.Errorf("netmap.Peer %v missing from b.peers", want.ID)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
got := gotv.AsStruct()
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
|
|
t.Errorf("netmap.Peer %v wrong.\n got: %v\nwant: %v", want.ID, logger.AsJSON(got), logger.AsJSON(want))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// tests WhoIs and indirectly that setNetMapLocked updates b.nodeByAddr correctly.
|
|
|
|
func TestWhoIs(t *testing.T) {
|
|
|
|
b := newTestLocalBackend(t)
|
|
|
|
b.setNetMapLocked(&netmap.NetworkMap{
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
ID: 1,
|
|
|
|
User: 10,
|
|
|
|
Addresses: []netip.Prefix{netip.MustParsePrefix("100.101.102.103/32")},
|
|
|
|
}).View(),
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
ID: 2,
|
|
|
|
User: 20,
|
|
|
|
Addresses: []netip.Prefix{netip.MustParsePrefix("100.200.200.200/32")},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
UserProfiles: map[tailcfg.UserID]tailcfg.UserProfile{
|
|
|
|
10: {
|
|
|
|
DisplayName: "Myself",
|
|
|
|
},
|
|
|
|
20: {
|
|
|
|
DisplayName: "Peer",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
})
|
|
|
|
tests := []struct {
|
|
|
|
q string
|
|
|
|
want tailcfg.NodeID // 0 means want ok=false
|
|
|
|
wantName string
|
|
|
|
}{
|
|
|
|
{"100.101.102.103:0", 1, "Myself"},
|
|
|
|
{"100.101.102.103:123", 1, "Myself"},
|
|
|
|
{"100.200.200.200:0", 2, "Peer"},
|
|
|
|
{"100.200.200.200:123", 2, "Peer"},
|
|
|
|
{"100.4.0.4:404", 0, ""},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.q, func(t *testing.T) {
|
|
|
|
nv, up, ok := b.WhoIs("", netip.MustParseAddrPort(tt.q))
|
|
|
|
var got tailcfg.NodeID
|
|
|
|
if ok {
|
|
|
|
got = nv.ID()
|
|
|
|
}
|
|
|
|
if got != tt.want {
|
|
|
|
t.Errorf("got nodeID %v; want %v", got, tt.want)
|
|
|
|
}
|
|
|
|
if up.DisplayName != tt.wantName {
|
|
|
|
t.Errorf("got name %q; want %q", up.DisplayName, tt.wantName)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestWireguardExitNodeDNSResolvers(t *testing.T) {
|
|
|
|
type tc struct {
|
|
|
|
name string
|
|
|
|
id tailcfg.StableNodeID
|
|
|
|
peers []*tailcfg.Node
|
|
|
|
wantOK bool
|
|
|
|
wantResolvers []*dnstype.Resolver
|
|
|
|
}
|
|
|
|
|
|
|
|
tests := []tc{
|
|
|
|
{
|
|
|
|
name: "no peers",
|
|
|
|
id: "1",
|
|
|
|
wantOK: false,
|
|
|
|
wantResolvers: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "non wireguard peer",
|
|
|
|
id: "1",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
StableID: "1",
|
|
|
|
IsWireGuardOnly: false,
|
|
|
|
ExitNodeDNSResolvers: []*dnstype.Resolver{{Addr: "dns.example.com"}},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantOK: false,
|
|
|
|
wantResolvers: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "no matching IDs",
|
|
|
|
id: "2",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
StableID: "1",
|
|
|
|
IsWireGuardOnly: true,
|
|
|
|
ExitNodeDNSResolvers: []*dnstype.Resolver{{Addr: "dns.example.com"}},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantOK: false,
|
|
|
|
wantResolvers: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wireguard peer",
|
|
|
|
id: "1",
|
|
|
|
peers: []*tailcfg.Node{
|
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
StableID: "1",
|
|
|
|
IsWireGuardOnly: true,
|
|
|
|
ExitNodeDNSResolvers: []*dnstype.Resolver{{Addr: "dns.example.com"}},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantOK: true,
|
|
|
|
wantResolvers: []*dnstype.Resolver{{Addr: "dns.example.com"}},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tc := range tests {
|
|
|
|
peers := peersMap(nodeViews(tc.peers))
|
|
|
|
nm := &netmap.NetworkMap{}
|
|
|
|
gotResolvers, gotOK := wireguardExitNodeDNSResolvers(nm, peers, tc.id)
|
|
|
|
|
|
|
|
if gotOK != tc.wantOK || !resolversEqual(t, gotResolvers, tc.wantResolvers) {
|
|
|
|
t.Errorf("case: %s: got %v, %v, want %v, %v", tc.name, gotOK, gotResolvers, tc.wantOK, tc.wantResolvers)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNSConfigForNetmapForExitNodeConfigs(t *testing.T) {
|
|
|
|
type tc struct {
|
|
|
|
name string
|
|
|
|
exitNode tailcfg.StableNodeID
|
|
|
|
peers []tailcfg.NodeView
|
|
|
|
dnsConfig *tailcfg.DNSConfig
|
|
|
|
wantDefaultResolvers []*dnstype.Resolver
|
|
|
|
wantRoutes map[dnsname.FQDN][]*dnstype.Resolver
|
|
|
|
}
|
|
|
|
|
|
|
|
defaultResolvers := []*dnstype.Resolver{{Addr: "default.example.com"}}
|
|
|
|
wgResolvers := []*dnstype.Resolver{{Addr: "wg.example.com"}}
|
|
|
|
peers := []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
ID: 1,
|
|
|
|
StableID: "wg",
|
|
|
|
IsWireGuardOnly: true,
|
|
|
|
ExitNodeDNSResolvers: wgResolvers,
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{}).View(),
|
|
|
|
}).View(),
|
|
|
|
// regular tailscale exit node with DNS capabilities
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Cap: 26,
|
|
|
|
ID: 2,
|
|
|
|
StableID: "ts",
|
|
|
|
Hostinfo: (&tailcfg.Hostinfo{}).View(),
|
|
|
|
}).View(),
|
|
|
|
}
|
|
|
|
exitDOH := peerAPIBase(&netmap.NetworkMap{Peers: peers}, peers[0]) + "/dns-query"
|
|
|
|
routes := map[dnsname.FQDN][]*dnstype.Resolver{
|
|
|
|
"route.example.com.": {{Addr: "route.example.com"}},
|
|
|
|
}
|
|
|
|
stringifyRoutes := func(routes map[dnsname.FQDN][]*dnstype.Resolver) map[string][]*dnstype.Resolver {
|
|
|
|
if routes == nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
m := make(map[string][]*dnstype.Resolver)
|
|
|
|
for k, v := range routes {
|
|
|
|
m[string(k)] = v
|
|
|
|
}
|
|
|
|
return m
|
|
|
|
}
|
|
|
|
|
|
|
|
tests := []tc{
|
|
|
|
{
|
|
|
|
name: "noExit/noRoutes/noResolver",
|
|
|
|
exitNode: "",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{},
|
|
|
|
wantDefaultResolvers: nil,
|
|
|
|
wantRoutes: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "tsExit/noRoutes/noResolver",
|
|
|
|
exitNode: "ts",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{},
|
|
|
|
wantDefaultResolvers: []*dnstype.Resolver{{Addr: exitDOH}},
|
|
|
|
wantRoutes: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "tsExit/noRoutes/defaultResolver",
|
|
|
|
exitNode: "ts",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{Resolvers: defaultResolvers},
|
|
|
|
wantDefaultResolvers: []*dnstype.Resolver{{Addr: exitDOH}},
|
|
|
|
wantRoutes: nil,
|
|
|
|
},
|
|
|
|
|
|
|
|
// The following two cases may need to be revisited. For a shared-in
|
|
|
|
// exit node split-DNS may effectively break, furthermore in the future
|
|
|
|
// if different nodes observe different DNS configurations, even a
|
|
|
|
// tailnet local exit node may present a different DNS configuration,
|
|
|
|
// which may not meet expectations in some use cases.
|
|
|
|
// In the case where a default resolver is set, the default resolver
|
|
|
|
// should also perhaps take precedence also.
|
|
|
|
{
|
|
|
|
name: "tsExit/routes/noResolver",
|
|
|
|
exitNode: "ts",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{Routes: stringifyRoutes(routes)},
|
|
|
|
wantDefaultResolvers: []*dnstype.Resolver{{Addr: exitDOH}},
|
|
|
|
wantRoutes: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "tsExit/routes/defaultResolver",
|
|
|
|
exitNode: "ts",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{Routes: stringifyRoutes(routes), Resolvers: defaultResolvers},
|
|
|
|
wantDefaultResolvers: []*dnstype.Resolver{{Addr: exitDOH}},
|
|
|
|
wantRoutes: nil,
|
|
|
|
},
|
|
|
|
|
|
|
|
// WireGuard exit nodes with DNS capabilities provide a "fallback" type
|
|
|
|
// behavior, they have a lower precedence than a default resolver, but
|
|
|
|
// otherwise allow split-DNS to operate as normal, and are used when
|
|
|
|
// there is no default resolver.
|
|
|
|
{
|
|
|
|
name: "wgExit/noRoutes/noResolver",
|
|
|
|
exitNode: "wg",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{},
|
|
|
|
wantDefaultResolvers: wgResolvers,
|
|
|
|
wantRoutes: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wgExit/noRoutes/defaultResolver",
|
|
|
|
exitNode: "wg",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{Resolvers: defaultResolvers},
|
|
|
|
wantDefaultResolvers: defaultResolvers,
|
|
|
|
wantRoutes: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wgExit/routes/defaultResolver",
|
|
|
|
exitNode: "wg",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{Routes: stringifyRoutes(routes), Resolvers: defaultResolvers},
|
|
|
|
wantDefaultResolvers: defaultResolvers,
|
|
|
|
wantRoutes: routes,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wgExit/routes/noResolver",
|
|
|
|
exitNode: "wg",
|
|
|
|
peers: peers,
|
|
|
|
dnsConfig: &tailcfg.DNSConfig{Routes: stringifyRoutes(routes)},
|
|
|
|
wantDefaultResolvers: wgResolvers,
|
|
|
|
wantRoutes: routes,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tc := range tests {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
nm := &netmap.NetworkMap{
|
|
|
|
Peers: tc.peers,
|
|
|
|
DNS: *tc.dnsConfig,
|
|
|
|
}
|
|
|
|
|
|
|
|
prefs := &ipn.Prefs{ExitNodeID: tc.exitNode, CorpDNS: true}
|
|
|
|
got := dnsConfigForNetmap(nm, peersMap(tc.peers), prefs.View(), false, t.Logf, "")
|
|
|
|
if !resolversEqual(t, got.DefaultResolvers, tc.wantDefaultResolvers) {
|
|
|
|
t.Errorf("DefaultResolvers: got %#v, want %#v", got.DefaultResolvers, tc.wantDefaultResolvers)
|
|
|
|
}
|
|
|
|
if !routesEqual(t, got.Routes, tc.wantRoutes) {
|
|
|
|
t.Errorf("Routes: got %#v, want %#v", got.Routes, tc.wantRoutes)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestOfferingAppConnector(t *testing.T) {
|
|
|
|
for _, shouldStore := range []bool{false, true} {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
if b.OfferingAppConnector() {
|
|
|
|
t.Fatal("unexpected offering app connector")
|
|
|
|
}
|
|
|
|
if shouldStore {
|
|
|
|
b.appConnector = appc.NewAppConnector(t.Logf, nil, &appc.RouteInfo{}, fakeStoreRoutes)
|
|
|
|
} else {
|
|
|
|
b.appConnector = appc.NewAppConnector(t.Logf, nil, nil, nil)
|
|
|
|
}
|
|
|
|
if !b.OfferingAppConnector() {
|
|
|
|
t.Fatal("unexpected not offering app connector")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestRouteAdvertiser(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
testPrefix := netip.MustParsePrefix("192.0.0.8/32")
|
|
|
|
|
|
|
|
ra := appc.RouteAdvertiser(b)
|
|
|
|
must.Do(ra.AdvertiseRoute(testPrefix))
|
|
|
|
|
|
|
|
routes := b.Prefs().AdvertiseRoutes()
|
|
|
|
if routes.Len() != 1 || routes.At(0) != testPrefix {
|
|
|
|
t.Fatalf("got routes %v, want %v", routes, []netip.Prefix{testPrefix})
|
|
|
|
}
|
|
|
|
|
|
|
|
must.Do(ra.UnadvertiseRoute(testPrefix))
|
|
|
|
|
|
|
|
routes = b.Prefs().AdvertiseRoutes()
|
|
|
|
if routes.Len() != 0 {
|
|
|
|
t.Fatalf("got routes %v, want none", routes)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestRouterAdvertiserIgnoresContainedRoutes(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
testPrefix := netip.MustParsePrefix("192.0.0.0/24")
|
|
|
|
ra := appc.RouteAdvertiser(b)
|
|
|
|
must.Do(ra.AdvertiseRoute(testPrefix))
|
|
|
|
|
|
|
|
routes := b.Prefs().AdvertiseRoutes()
|
|
|
|
if routes.Len() != 1 || routes.At(0) != testPrefix {
|
|
|
|
t.Fatalf("got routes %v, want %v", routes, []netip.Prefix{testPrefix})
|
|
|
|
}
|
|
|
|
|
|
|
|
must.Do(ra.AdvertiseRoute(netip.MustParsePrefix("192.0.0.8/32")))
|
|
|
|
|
|
|
|
// the above /32 is not added as it is contained within the /24
|
|
|
|
routes = b.Prefs().AdvertiseRoutes()
|
|
|
|
if routes.Len() != 1 || routes.At(0) != testPrefix {
|
|
|
|
t.Fatalf("got routes %v, want %v", routes, []netip.Prefix{testPrefix})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestObserveDNSResponse(t *testing.T) {
|
|
|
|
for _, shouldStore := range []bool{false, true} {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
|
|
|
|
// ensure no error when no app connector is configured
|
|
|
|
b.ObserveDNSResponse(dnsResponse("example.com.", "192.0.0.8"))
|
|
|
|
|
|
|
|
rc := &appctest.RouteCollector{}
|
|
|
|
if shouldStore {
|
|
|
|
b.appConnector = appc.NewAppConnector(t.Logf, rc, &appc.RouteInfo{}, fakeStoreRoutes)
|
|
|
|
} else {
|
|
|
|
b.appConnector = appc.NewAppConnector(t.Logf, rc, nil, nil)
|
|
|
|
}
|
|
|
|
b.appConnector.UpdateDomains([]string{"example.com"})
|
|
|
|
b.appConnector.Wait(context.Background())
|
|
|
|
|
|
|
|
b.ObserveDNSResponse(dnsResponse("example.com.", "192.0.0.8"))
|
|
|
|
b.appConnector.Wait(context.Background())
|
|
|
|
wantRoutes := []netip.Prefix{netip.MustParsePrefix("192.0.0.8/32")}
|
|
|
|
if !slices.Equal(rc.Routes(), wantRoutes) {
|
|
|
|
t.Fatalf("got routes %v, want %v", rc.Routes(), wantRoutes)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestCoveredRouteRangeNoDefault(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
existingRoute netip.Prefix
|
|
|
|
newRoute netip.Prefix
|
|
|
|
want bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
existingRoute: netip.MustParsePrefix("192.0.0.1/32"),
|
|
|
|
newRoute: netip.MustParsePrefix("192.0.0.1/32"),
|
|
|
|
want: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
existingRoute: netip.MustParsePrefix("192.0.0.1/32"),
|
|
|
|
newRoute: netip.MustParsePrefix("192.0.0.2/32"),
|
|
|
|
want: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
existingRoute: netip.MustParsePrefix("192.0.0.0/24"),
|
|
|
|
newRoute: netip.MustParsePrefix("192.0.0.1/32"),
|
|
|
|
want: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
existingRoute: netip.MustParsePrefix("192.0.0.0/16"),
|
|
|
|
newRoute: netip.MustParsePrefix("192.0.0.0/24"),
|
|
|
|
want: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
existingRoute: netip.MustParsePrefix("0.0.0.0/0"),
|
|
|
|
newRoute: netip.MustParsePrefix("192.0.0.0/24"),
|
|
|
|
want: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
existingRoute: netip.MustParsePrefix("::/0"),
|
|
|
|
newRoute: netip.MustParsePrefix("2001:db8::/32"),
|
|
|
|
want: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
got := coveredRouteRangeNoDefault([]netip.Prefix{tt.existingRoute}, tt.newRoute)
|
|
|
|
if got != tt.want {
|
|
|
|
t.Errorf("coveredRouteRange(%v, %v) = %v, want %v", tt.existingRoute, tt.newRoute, got, tt.want)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestReconfigureAppConnector(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
b.reconfigAppConnectorLocked(b.netMap, b.pm.prefs)
|
|
|
|
if b.appConnector != nil {
|
|
|
|
t.Fatal("unexpected app connector")
|
|
|
|
}
|
|
|
|
|
|
|
|
b.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
AppConnector: ipn.AppConnectorPrefs{
|
|
|
|
Advertise: true,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
AppConnectorSet: true,
|
|
|
|
})
|
|
|
|
b.reconfigAppConnectorLocked(b.netMap, b.pm.prefs)
|
|
|
|
if b.appConnector == nil {
|
|
|
|
t.Fatal("expected app connector")
|
|
|
|
}
|
|
|
|
|
|
|
|
appCfg := `{
|
|
|
|
"name": "example",
|
|
|
|
"domains": ["example.com"],
|
|
|
|
"connectors": ["tag:example"]
|
|
|
|
}`
|
|
|
|
|
|
|
|
b.netMap.SelfNode = (&tailcfg.Node{
|
|
|
|
Name: "example.ts.net",
|
|
|
|
Tags: []string{"tag:example"},
|
|
|
|
CapMap: (tailcfg.NodeCapMap)(map[tailcfg.NodeCapability][]tailcfg.RawMessage{
|
|
|
|
"tailscale.com/app-connectors": {tailcfg.RawMessage(appCfg)},
|
|
|
|
}),
|
|
|
|
}).View()
|
|
|
|
|
|
|
|
b.reconfigAppConnectorLocked(b.netMap, b.pm.prefs)
|
|
|
|
b.appConnector.Wait(context.Background())
|
|
|
|
|
|
|
|
want := []string{"example.com"}
|
|
|
|
if !slices.Equal(b.appConnector.Domains().AsSlice(), want) {
|
|
|
|
t.Fatalf("got domains %v, want %v", b.appConnector.Domains(), want)
|
|
|
|
}
|
|
|
|
if v, _ := b.hostinfo.AppConnector.Get(); !v {
|
|
|
|
t.Fatalf("expected app connector service")
|
|
|
|
}
|
|
|
|
|
|
|
|
// disable the connector in order to assert that the service is removed
|
|
|
|
b.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
AppConnector: ipn.AppConnectorPrefs{
|
|
|
|
Advertise: false,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
AppConnectorSet: true,
|
|
|
|
})
|
|
|
|
b.reconfigAppConnectorLocked(b.netMap, b.pm.prefs)
|
|
|
|
if b.appConnector != nil {
|
|
|
|
t.Fatal("expected no app connector")
|
|
|
|
}
|
|
|
|
if v, _ := b.hostinfo.AppConnector.Get(); v {
|
|
|
|
t.Fatalf("expected no app connector service")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func resolversEqual(t *testing.T, a, b []*dnstype.Resolver) bool {
|
|
|
|
if a == nil && b == nil {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
if a == nil || b == nil {
|
|
|
|
t.Errorf("resolversEqual: a == nil || b == nil : %#v != %#v", a, b)
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
if len(a) != len(b) {
|
|
|
|
t.Errorf("resolversEqual: len(a) != len(b) : %#v != %#v", a, b)
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
for i := range a {
|
|
|
|
if !a[i].Equal(b[i]) {
|
|
|
|
t.Errorf("resolversEqual: a != b [%d]: %v != %v", i, *a[i], *b[i])
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
func routesEqual(t *testing.T, a, b map[dnsname.FQDN][]*dnstype.Resolver) bool {
|
|
|
|
if len(a) != len(b) {
|
|
|
|
t.Logf("routes: len(a) != len(b): %d != %d", len(a), len(b))
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
for name := range a {
|
|
|
|
if !resolversEqual(t, a[name], b[name]) {
|
|
|
|
t.Logf("routes: a != b [%s]: %v != %v", name, a[name], b[name])
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// dnsResponse is a test helper that creates a DNS response buffer for the given domain and address
|
|
|
|
func dnsResponse(domain, address string) []byte {
|
|
|
|
addr := netip.MustParseAddr(address)
|
|
|
|
b := dnsmessage.NewBuilder(nil, dnsmessage.Header{})
|
|
|
|
b.EnableCompression()
|
|
|
|
b.StartAnswers()
|
|
|
|
switch addr.BitLen() {
|
|
|
|
case 32:
|
|
|
|
b.AResource(
|
|
|
|
dnsmessage.ResourceHeader{
|
|
|
|
Name: dnsmessage.MustNewName(domain),
|
|
|
|
Type: dnsmessage.TypeA,
|
|
|
|
Class: dnsmessage.ClassINET,
|
|
|
|
TTL: 0,
|
|
|
|
},
|
|
|
|
dnsmessage.AResource{
|
|
|
|
A: addr.As4(),
|
|
|
|
},
|
|
|
|
)
|
|
|
|
case 128:
|
|
|
|
b.AAAAResource(
|
|
|
|
dnsmessage.ResourceHeader{
|
|
|
|
Name: dnsmessage.MustNewName(domain),
|
|
|
|
Type: dnsmessage.TypeAAAA,
|
|
|
|
Class: dnsmessage.ClassINET,
|
|
|
|
TTL: 0,
|
|
|
|
},
|
|
|
|
dnsmessage.AAAAResource{
|
|
|
|
AAAA: addr.As16(),
|
|
|
|
},
|
|
|
|
)
|
|
|
|
default:
|
|
|
|
panic("invalid address length")
|
|
|
|
}
|
|
|
|
return must.Get(b.Finish())
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSetExitNodeIDPolicy(t *testing.T) {
|
|
|
|
pfx := netip.MustParsePrefix
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
exitNodeIPKey bool
|
|
|
|
exitNodeIDKey bool
|
|
|
|
exitNodeID string
|
|
|
|
exitNodeIP string
|
|
|
|
prefs *ipn.Prefs
|
|
|
|
exitNodeIPWant string
|
|
|
|
exitNodeIDWant string
|
|
|
|
prefsChanged bool
|
|
|
|
nm *netmap.NetworkMap
|
|
|
|
lastSuggestedExitNode tailcfg.StableNodeID
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "ExitNodeID key is set",
|
|
|
|
exitNodeIDKey: true,
|
|
|
|
exitNodeID: "123",
|
|
|
|
exitNodeIDWant: "123",
|
|
|
|
prefsChanged: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeID key not set",
|
|
|
|
exitNodeIDKey: true,
|
|
|
|
exitNodeIDWant: "",
|
|
|
|
prefsChanged: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeID key set, ExitNodeIP preference set",
|
|
|
|
exitNodeIDKey: true,
|
|
|
|
exitNodeID: "123",
|
|
|
|
prefs: &ipn.Prefs{ExitNodeIP: netip.MustParseAddr("127.0.0.1")},
|
|
|
|
exitNodeIDWant: "123",
|
|
|
|
prefsChanged: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeID key not set, ExitNodeIP key set",
|
|
|
|
exitNodeIPKey: true,
|
|
|
|
exitNodeIP: "127.0.0.1",
|
|
|
|
prefs: &ipn.Prefs{ExitNodeIP: netip.MustParseAddr("127.0.0.1")},
|
|
|
|
exitNodeIPWant: "127.0.0.1",
|
|
|
|
prefsChanged: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeIP key set, existing ExitNodeIP pref",
|
|
|
|
exitNodeIPKey: true,
|
|
|
|
exitNodeIP: "127.0.0.1",
|
|
|
|
prefs: &ipn.Prefs{ExitNodeIP: netip.MustParseAddr("127.0.0.1")},
|
|
|
|
exitNodeIPWant: "127.0.0.1",
|
|
|
|
prefsChanged: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "existing preferences match policy",
|
|
|
|
exitNodeIDKey: true,
|
|
|
|
exitNodeID: "123",
|
|
|
|
prefs: &ipn.Prefs{ExitNodeID: tailcfg.StableNodeID("123")},
|
|
|
|
exitNodeIDWant: "123",
|
|
|
|
prefsChanged: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeIP set if net map does not have corresponding node",
|
|
|
|
exitNodeIPKey: true,
|
|
|
|
prefs: &ipn.Prefs{ExitNodeIP: netip.MustParseAddr("127.0.0.1")},
|
|
|
|
exitNodeIP: "127.0.0.1",
|
|
|
|
exitNodeIPWant: "127.0.0.1",
|
|
|
|
prefsChanged: false,
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
Name: "foo.tailnet",
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100.102.103.104/32"),
|
|
|
|
pfx("100::123/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "a.tailnet",
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100.0.0.201/32"),
|
|
|
|
pfx("100::201/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "b.tailnet",
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100::202/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeIP cleared if net map has corresponding node - policy matches prefs",
|
|
|
|
prefs: &ipn.Prefs{ExitNodeIP: netip.MustParseAddr("127.0.0.1")},
|
|
|
|
exitNodeIPKey: true,
|
|
|
|
exitNodeIP: "127.0.0.1",
|
|
|
|
exitNodeIPWant: "",
|
|
|
|
exitNodeIDWant: "123",
|
|
|
|
prefsChanged: true,
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
Name: "foo.tailnet",
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100.102.103.104/32"),
|
|
|
|
pfx("100::123/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "a.tailnet",
|
|
|
|
StableID: tailcfg.StableNodeID("123"),
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("127.0.0.1/32"),
|
|
|
|
pfx("100::201/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "b.tailnet",
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100::202/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeIP cleared if net map has corresponding node - no policy set",
|
|
|
|
prefs: &ipn.Prefs{ExitNodeIP: netip.MustParseAddr("127.0.0.1")},
|
|
|
|
exitNodeIPWant: "",
|
|
|
|
exitNodeIDWant: "123",
|
|
|
|
prefsChanged: true,
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
Name: "foo.tailnet",
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100.102.103.104/32"),
|
|
|
|
pfx("100::123/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "a.tailnet",
|
|
|
|
StableID: tailcfg.StableNodeID("123"),
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("127.0.0.1/32"),
|
|
|
|
pfx("100::201/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "b.tailnet",
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100::202/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeIP cleared if net map has corresponding node - different exit node IP in policy",
|
|
|
|
exitNodeIPKey: true,
|
|
|
|
prefs: &ipn.Prefs{ExitNodeIP: netip.MustParseAddr("127.0.0.1")},
|
|
|
|
exitNodeIP: "100.64.5.6",
|
|
|
|
exitNodeIPWant: "",
|
|
|
|
exitNodeIDWant: "123",
|
|
|
|
prefsChanged: true,
|
|
|
|
nm: &netmap.NetworkMap{
|
|
|
|
Name: "foo.tailnet",
|
|
|
|
SelfNode: (&tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100.102.103.104/32"),
|
|
|
|
pfx("100::123/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "a.tailnet",
|
|
|
|
StableID: tailcfg.StableNodeID("123"),
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100.64.5.6/32"),
|
|
|
|
pfx("100::201/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
(&tailcfg.Node{
|
|
|
|
Name: "b.tailnet",
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
pfx("100::202/128"),
|
|
|
|
},
|
|
|
|
}).View(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeID key is set to auto and last suggested exit node is populated",
|
|
|
|
exitNodeIDKey: true,
|
|
|
|
exitNodeID: "auto:any",
|
|
|
|
lastSuggestedExitNode: "123",
|
|
|
|
exitNodeIDWant: "123",
|
|
|
|
prefsChanged: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ExitNodeID key is set to auto and last suggested exit node is not populated",
|
|
|
|
exitNodeIDKey: true,
|
|
|
|
exitNodeID: "auto:any",
|
|
|
|
prefsChanged: true,
|
|
|
|
exitNodeIDWant: "auto:any",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
|
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
|
|
|
|
policyStore := source.NewTestStoreOf(t,
|
|
|
|
source.TestSettingOf(syspolicy.ExitNodeID, test.exitNodeID),
|
|
|
|
source.TestSettingOf(syspolicy.ExitNodeIP, test.exitNodeIP),
|
|
|
|
)
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
|
|
|
|
if test.nm == nil {
|
|
|
|
test.nm = new(netmap.NetworkMap)
|
|
|
|
}
|
|
|
|
if test.prefs == nil {
|
|
|
|
test.prefs = ipn.NewPrefs()
|
|
|
|
}
|
|
|
|
pm := must.Get(newProfileManager(new(mem.Store), t.Logf, new(health.Tracker)))
|
|
|
|
pm.prefs = test.prefs.View()
|
|
|
|
b.netMap = test.nm
|
|
|
|
b.pm = pm
|
|
|
|
b.lastSuggestedExitNode = test.lastSuggestedExitNode
|
|
|
|
changed := setExitNodeID(b.pm.prefs.AsStruct(), test.nm, tailcfg.StableNodeID(test.lastSuggestedExitNode))
|
|
|
|
b.SetPrefsForTest(pm.CurrentPrefs().AsStruct())
|
|
|
|
|
|
|
|
if got := b.pm.prefs.ExitNodeID(); got != tailcfg.StableNodeID(test.exitNodeIDWant) {
|
|
|
|
t.Errorf("got %v want %v", got, test.exitNodeIDWant)
|
|
|
|
}
|
|
|
|
if got := b.pm.prefs.ExitNodeIP(); test.exitNodeIPWant == "" {
|
|
|
|
if got.String() != "invalid IP" {
|
|
|
|
t.Errorf("got %v want invalid IP", got)
|
|
|
|
}
|
|
|
|
} else if got.String() != test.exitNodeIPWant {
|
|
|
|
t.Errorf("got %v want %v", got, test.exitNodeIPWant)
|
|
|
|
}
|
|
|
|
|
|
|
|
if changed != test.prefsChanged {
|
|
|
|
t.Errorf("wanted prefs changed %v, got prefs changed %v", test.prefsChanged, changed)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestUpdateNetmapDeltaAutoExitNode(t *testing.T) {
|
|
|
|
peer1 := makePeer(1, withCap(26), withSuggest(), withExitRoutes())
|
|
|
|
peer2 := makePeer(2, withCap(26), withSuggest(), withExitRoutes())
|
|
|
|
derpMap := &tailcfg.DERPMap{
|
|
|
|
Regions: map[int]*tailcfg.DERPRegion{
|
|
|
|
1: {
|
|
|
|
Nodes: []*tailcfg.DERPNode{
|
|
|
|
{
|
|
|
|
Name: "t1",
|
|
|
|
RegionID: 1,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
2: {
|
|
|
|
Nodes: []*tailcfg.DERPNode{
|
|
|
|
{
|
|
|
|
Name: "t2",
|
|
|
|
RegionID: 2,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
report := &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10 * time.Millisecond,
|
|
|
|
2: 5 * time.Millisecond,
|
|
|
|
3: 30 * time.Millisecond,
|
|
|
|
},
|
|
|
|
PreferredDERP: 2,
|
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
lastSuggestedExitNode tailcfg.StableNodeID
|
|
|
|
netmap *netmap.NetworkMap
|
|
|
|
muts []*tailcfg.PeerChange
|
|
|
|
exitNodeIDWant tailcfg.StableNodeID
|
|
|
|
updateNetmapDeltaResponse bool
|
|
|
|
report *netcheck.Report
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "selected auto exit node goes offline",
|
|
|
|
lastSuggestedExitNode: peer1.StableID(),
|
|
|
|
netmap: &netmap.NetworkMap{
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer2,
|
|
|
|
},
|
|
|
|
DERPMap: derpMap,
|
|
|
|
},
|
|
|
|
muts: []*tailcfg.PeerChange{
|
|
|
|
{
|
|
|
|
NodeID: 1,
|
|
|
|
Online: ptr.To(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
NodeID: 2,
|
|
|
|
Online: ptr.To(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
exitNodeIDWant: peer2.StableID(),
|
|
|
|
updateNetmapDeltaResponse: false,
|
|
|
|
report: report,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "other exit node goes offline doesn't change selected auto exit node that's still online",
|
|
|
|
lastSuggestedExitNode: peer2.StableID(),
|
|
|
|
netmap: &netmap.NetworkMap{
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer2,
|
|
|
|
},
|
|
|
|
DERPMap: derpMap,
|
|
|
|
},
|
|
|
|
muts: []*tailcfg.PeerChange{
|
|
|
|
{
|
|
|
|
NodeID: 1,
|
|
|
|
Online: ptr.To(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
NodeID: 2,
|
|
|
|
Online: ptr.To(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
exitNodeIDWant: peer2.StableID(),
|
|
|
|
updateNetmapDeltaResponse: true,
|
|
|
|
report: report,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
policyStore := source.NewTestStoreOf(t, source.TestSettingOf(
|
|
|
|
syspolicy.ExitNodeID, "auto:any",
|
|
|
|
))
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
b := newTestLocalBackend(t)
|
|
|
|
b.netMap = tt.netmap
|
|
|
|
b.updatePeersFromNetmapLocked(b.netMap)
|
|
|
|
b.lastSuggestedExitNode = tt.lastSuggestedExitNode
|
|
|
|
b.sys.MagicSock.Get().SetLastNetcheckReportForTest(b.ctx, tt.report)
|
|
|
|
b.SetPrefsForTest(b.pm.CurrentPrefs().AsStruct())
|
|
|
|
someTime := time.Unix(123, 0)
|
|
|
|
muts, ok := netmap.MutationsFromMapResponse(&tailcfg.MapResponse{
|
|
|
|
PeersChangedPatch: tt.muts,
|
|
|
|
}, someTime)
|
|
|
|
if !ok {
|
|
|
|
t.Fatal("netmap.MutationsFromMapResponse failed")
|
|
|
|
}
|
|
|
|
if b.pm.prefs.ExitNodeID() != tt.lastSuggestedExitNode {
|
|
|
|
t.Fatalf("did not set exit node ID to last suggested exit node despite auto policy")
|
|
|
|
}
|
|
|
|
|
|
|
|
got := b.UpdateNetmapDelta(muts)
|
|
|
|
if got != tt.updateNetmapDeltaResponse {
|
|
|
|
t.Fatalf("got %v expected %v from UpdateNetmapDelta", got, tt.updateNetmapDeltaResponse)
|
|
|
|
}
|
|
|
|
if b.pm.prefs.ExitNodeID() != tt.exitNodeIDWant {
|
|
|
|
t.Fatalf("did not get expected exit node id after UpdateNetmapDelta")
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestAutoExitNodeSetNetInfoCallback(t *testing.T) {
|
|
|
|
b := newTestLocalBackend(t)
|
|
|
|
hi := hostinfo.New()
|
|
|
|
ni := tailcfg.NetInfo{LinkType: "wired"}
|
|
|
|
hi.NetInfo = &ni
|
|
|
|
b.hostinfo = hi
|
|
|
|
k := key.NewMachine()
|
|
|
|
var cc *mockControl
|
|
|
|
opts := controlclient.Options{
|
|
|
|
ServerURL: "https://example.com",
|
|
|
|
GetMachinePrivateKey: func() (key.MachinePrivate, error) {
|
|
|
|
return k, nil
|
|
|
|
},
|
|
|
|
Dialer: tsdial.NewDialer(netmon.NewStatic()),
|
|
|
|
Logf: b.logf,
|
|
|
|
}
|
|
|
|
cc = newClient(t, opts)
|
|
|
|
b.cc = cc
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
policyStore := source.NewTestStoreOf(t, source.TestSettingOf(
|
|
|
|
syspolicy.ExitNodeID, "auto:any",
|
|
|
|
))
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
peer1 := makePeer(1, withCap(26), withDERP(3), withSuggest(), withExitRoutes())
|
|
|
|
peer2 := makePeer(2, withCap(26), withDERP(2), withSuggest(), withExitRoutes())
|
|
|
|
selfNode := tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.1/32"),
|
|
|
|
netip.MustParsePrefix("fe70::1/128"),
|
|
|
|
},
|
|
|
|
DERP: "127.3.3.40:2",
|
|
|
|
}
|
|
|
|
defaultDERPMap := &tailcfg.DERPMap{
|
|
|
|
Regions: map[int]*tailcfg.DERPRegion{
|
|
|
|
1: {
|
|
|
|
Nodes: []*tailcfg.DERPNode{
|
|
|
|
{
|
|
|
|
Name: "t1",
|
|
|
|
RegionID: 1,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
2: {
|
|
|
|
Nodes: []*tailcfg.DERPNode{
|
|
|
|
{
|
|
|
|
Name: "t2",
|
|
|
|
RegionID: 2,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
3: {
|
|
|
|
Nodes: []*tailcfg.DERPNode{
|
|
|
|
{
|
|
|
|
Name: "t3",
|
|
|
|
RegionID: 3,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
b.netMap = &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer2,
|
|
|
|
},
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
}
|
|
|
|
b.lastSuggestedExitNode = peer1.StableID()
|
|
|
|
b.SetPrefsForTest(b.pm.CurrentPrefs().AsStruct())
|
|
|
|
if eid := b.Prefs().ExitNodeID(); eid != peer1.StableID() {
|
|
|
|
t.Errorf("got initial exit node %v, want %v", eid, peer1.StableID())
|
|
|
|
}
|
|
|
|
b.refreshAutoExitNode = true
|
|
|
|
b.sys.MagicSock.Get().SetLastNetcheckReportForTest(b.ctx, &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10 * time.Millisecond,
|
|
|
|
2: 5 * time.Millisecond,
|
|
|
|
3: 30 * time.Millisecond,
|
|
|
|
},
|
|
|
|
PreferredDERP: 2,
|
|
|
|
})
|
|
|
|
b.setNetInfo(&ni)
|
|
|
|
if eid := b.Prefs().ExitNodeID(); eid != peer2.StableID() {
|
|
|
|
t.Errorf("got final exit node %v, want %v", eid, peer2.StableID())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSetControlClientStatusAutoExitNode(t *testing.T) {
|
|
|
|
peer1 := makePeer(1, withCap(26), withSuggest(), withExitRoutes(), withNodeKey())
|
|
|
|
peer2 := makePeer(2, withCap(26), withSuggest(), withExitRoutes(), withNodeKey())
|
|
|
|
derpMap := &tailcfg.DERPMap{
|
|
|
|
Regions: map[int]*tailcfg.DERPRegion{
|
|
|
|
1: {
|
|
|
|
Nodes: []*tailcfg.DERPNode{
|
|
|
|
{
|
|
|
|
Name: "t1",
|
|
|
|
RegionID: 1,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
2: {
|
|
|
|
Nodes: []*tailcfg.DERPNode{
|
|
|
|
{
|
|
|
|
Name: "t2",
|
|
|
|
RegionID: 2,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
report := &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10 * time.Millisecond,
|
|
|
|
2: 5 * time.Millisecond,
|
|
|
|
3: 30 * time.Millisecond,
|
|
|
|
},
|
|
|
|
PreferredDERP: 1,
|
|
|
|
}
|
|
|
|
nm := &netmap.NetworkMap{
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer2,
|
|
|
|
},
|
|
|
|
DERPMap: derpMap,
|
|
|
|
}
|
|
|
|
b := newTestLocalBackend(t)
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
policyStore := source.NewTestStoreOf(t, source.TestSettingOf(
|
|
|
|
syspolicy.ExitNodeID, "auto:any",
|
|
|
|
))
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
b.netMap = nm
|
|
|
|
b.lastSuggestedExitNode = peer1.StableID()
|
|
|
|
b.sys.MagicSock.Get().SetLastNetcheckReportForTest(b.ctx, report)
|
|
|
|
b.SetPrefsForTest(b.pm.CurrentPrefs().AsStruct())
|
|
|
|
firstExitNode := b.Prefs().ExitNodeID()
|
|
|
|
newPeer1 := makePeer(1, withCap(26), withSuggest(), withExitRoutes(), withOnline(false), withNodeKey())
|
|
|
|
updatedNetmap := &netmap.NetworkMap{
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
newPeer1,
|
|
|
|
peer2,
|
|
|
|
},
|
|
|
|
DERPMap: derpMap,
|
|
|
|
}
|
|
|
|
b.SetControlClientStatus(b.cc, controlclient.Status{NetMap: updatedNetmap})
|
|
|
|
lastExitNode := b.Prefs().ExitNodeID()
|
|
|
|
if firstExitNode == lastExitNode {
|
|
|
|
t.Errorf("did not switch exit nodes despite auto exit node going offline")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestApplySysPolicy(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
prefs ipn.Prefs
|
|
|
|
wantPrefs ipn.Prefs
|
|
|
|
wantAnyChange bool
|
|
|
|
stringPolicies map[syspolicy.Key]string
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "empty prefs without policies",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prefs set without policies",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
ExitNodeAllowLANAccess: true,
|
|
|
|
CorpDNS: true,
|
|
|
|
RouteAll: true,
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
ExitNodeAllowLANAccess: true,
|
|
|
|
CorpDNS: true,
|
|
|
|
RouteAll: true,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "empty prefs with policies",
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
ExitNodeAllowLANAccess: true,
|
|
|
|
CorpDNS: true,
|
|
|
|
RouteAll: true,
|
|
|
|
},
|
|
|
|
wantAnyChange: true,
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.ControlURL: "1",
|
|
|
|
syspolicy.EnableIncomingConnections: "never",
|
|
|
|
syspolicy.EnableServerMode: "always",
|
|
|
|
syspolicy.ExitNodeAllowLANAccess: "always",
|
|
|
|
syspolicy.EnableTailscaleDNS: "always",
|
|
|
|
syspolicy.EnableTailscaleSubnets: "always",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prefs set with matching policies",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
},
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.ControlURL: "1",
|
|
|
|
syspolicy.EnableIncomingConnections: "never",
|
|
|
|
syspolicy.EnableServerMode: "always",
|
|
|
|
syspolicy.ExitNodeAllowLANAccess: "never",
|
|
|
|
syspolicy.EnableTailscaleDNS: "never",
|
|
|
|
syspolicy.EnableTailscaleSubnets: "never",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prefs set with conflicting policies",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
ExitNodeAllowLANAccess: false,
|
|
|
|
CorpDNS: true,
|
|
|
|
RouteAll: false,
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
ControlURL: "2",
|
|
|
|
ShieldsUp: false,
|
|
|
|
ForceDaemon: false,
|
|
|
|
ExitNodeAllowLANAccess: true,
|
|
|
|
CorpDNS: false,
|
|
|
|
RouteAll: true,
|
|
|
|
},
|
|
|
|
wantAnyChange: true,
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.ControlURL: "2",
|
|
|
|
syspolicy.EnableIncomingConnections: "always",
|
|
|
|
syspolicy.EnableServerMode: "never",
|
|
|
|
syspolicy.ExitNodeAllowLANAccess: "always",
|
|
|
|
syspolicy.EnableTailscaleDNS: "never",
|
|
|
|
syspolicy.EnableTailscaleSubnets: "always",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prefs set with neutral policies",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
ExitNodeAllowLANAccess: false,
|
|
|
|
CorpDNS: true,
|
|
|
|
RouteAll: true,
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
ControlURL: "1",
|
|
|
|
ShieldsUp: true,
|
|
|
|
ForceDaemon: true,
|
|
|
|
ExitNodeAllowLANAccess: false,
|
|
|
|
CorpDNS: true,
|
|
|
|
RouteAll: true,
|
|
|
|
},
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.EnableIncomingConnections: "user-decides",
|
|
|
|
syspolicy.EnableServerMode: "user-decides",
|
|
|
|
syspolicy.ExitNodeAllowLANAccess: "user-decides",
|
|
|
|
syspolicy.EnableTailscaleDNS: "user-decides",
|
|
|
|
syspolicy.EnableTailscaleSubnets: "user-decides",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ControlURL",
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
ControlURL: "set",
|
|
|
|
},
|
|
|
|
wantAnyChange: true,
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.ControlURL: "set",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "enable AutoUpdate apply does not unset check",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: true,
|
|
|
|
Apply: opt.NewBool(false),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: true,
|
|
|
|
Apply: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantAnyChange: true,
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.ApplyUpdates: "always",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "disable AutoUpdate apply does not unset check",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: true,
|
|
|
|
Apply: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: true,
|
|
|
|
Apply: opt.NewBool(false),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantAnyChange: true,
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.ApplyUpdates: "never",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "enable AutoUpdate check does not unset apply",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: false,
|
|
|
|
Apply: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: true,
|
|
|
|
Apply: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantAnyChange: true,
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.CheckUpdates: "always",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "disable AutoUpdate check does not unset apply",
|
|
|
|
prefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: true,
|
|
|
|
Apply: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantPrefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Check: false,
|
|
|
|
Apply: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantAnyChange: true,
|
|
|
|
stringPolicies: map[syspolicy.Key]string{
|
|
|
|
syspolicy.CheckUpdates: "never",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
settings := make([]source.TestSetting[string], 0, len(tt.stringPolicies))
|
|
|
|
for p, v := range tt.stringPolicies {
|
|
|
|
settings = append(settings, source.TestSettingOf(p, v))
|
|
|
|
}
|
|
|
|
policyStore := source.NewTestStoreOf(t, settings...)
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
|
|
|
|
t.Run("unit", func(t *testing.T) {
|
|
|
|
prefs := tt.prefs.Clone()
|
|
|
|
|
|
|
|
gotAnyChange := applySysPolicy(prefs)
|
|
|
|
|
|
|
|
if gotAnyChange && prefs.Equals(&tt.prefs) {
|
|
|
|
t.Errorf("anyChange but prefs is unchanged: %v", prefs.Pretty())
|
|
|
|
}
|
|
|
|
if !gotAnyChange && !prefs.Equals(&tt.prefs) {
|
|
|
|
t.Errorf("!anyChange but prefs changed from %v to %v", tt.prefs.Pretty(), prefs.Pretty())
|
|
|
|
}
|
|
|
|
if gotAnyChange != tt.wantAnyChange {
|
|
|
|
t.Errorf("anyChange=%v, want %v", gotAnyChange, tt.wantAnyChange)
|
|
|
|
}
|
|
|
|
if !prefs.Equals(&tt.wantPrefs) {
|
|
|
|
t.Errorf("prefs=%v, want %v", prefs.Pretty(), tt.wantPrefs.Pretty())
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("status update", func(t *testing.T) {
|
|
|
|
// Profile manager fills in blank ControlURL but it's not set
|
|
|
|
// in most test cases to avoid cluttering them, so adjust for
|
|
|
|
// that.
|
|
|
|
usePrefs := tt.prefs.Clone()
|
|
|
|
if usePrefs.ControlURL == "" {
|
|
|
|
usePrefs.ControlURL = ipn.DefaultControlURL
|
|
|
|
}
|
|
|
|
wantPrefs := tt.wantPrefs.Clone()
|
|
|
|
if wantPrefs.ControlURL == "" {
|
|
|
|
wantPrefs.ControlURL = ipn.DefaultControlURL
|
|
|
|
}
|
|
|
|
|
|
|
|
pm := must.Get(newProfileManager(new(mem.Store), t.Logf, new(health.Tracker)))
|
|
|
|
pm.prefs = usePrefs.View()
|
|
|
|
|
|
|
|
b := newTestBackend(t)
|
|
|
|
b.mu.Lock()
|
|
|
|
b.pm = pm
|
|
|
|
b.mu.Unlock()
|
|
|
|
|
|
|
|
b.SetControlClientStatus(b.cc, controlclient.Status{})
|
|
|
|
if !b.Prefs().Equals(wantPrefs.View()) {
|
|
|
|
t.Errorf("prefs=%v, want %v", b.Prefs().Pretty(), wantPrefs.Pretty())
|
|
|
|
}
|
|
|
|
})
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestPreferencePolicyInfo(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
initialValue bool
|
|
|
|
wantValue bool
|
|
|
|
wantChange bool
|
|
|
|
policyValue string
|
|
|
|
policyError error
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "force enable modify",
|
|
|
|
initialValue: false,
|
|
|
|
wantValue: true,
|
|
|
|
wantChange: true,
|
|
|
|
policyValue: "always",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "force enable unchanged",
|
|
|
|
initialValue: true,
|
|
|
|
wantValue: true,
|
|
|
|
policyValue: "always",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "force disable modify",
|
|
|
|
initialValue: true,
|
|
|
|
wantValue: false,
|
|
|
|
wantChange: true,
|
|
|
|
policyValue: "never",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "force disable unchanged",
|
|
|
|
initialValue: false,
|
|
|
|
wantValue: false,
|
|
|
|
policyValue: "never",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unforced enabled",
|
|
|
|
initialValue: true,
|
|
|
|
wantValue: true,
|
|
|
|
policyValue: "user-decides",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unforced disabled",
|
|
|
|
initialValue: false,
|
|
|
|
wantValue: false,
|
|
|
|
policyValue: "user-decides",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "blank enabled",
|
|
|
|
initialValue: true,
|
|
|
|
wantValue: true,
|
|
|
|
policyValue: "",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "blank disabled",
|
|
|
|
initialValue: false,
|
|
|
|
wantValue: false,
|
|
|
|
policyValue: "",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unset enabled",
|
|
|
|
initialValue: true,
|
|
|
|
wantValue: true,
|
|
|
|
policyError: syspolicy.ErrNoSuchKey,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unset disabled",
|
|
|
|
initialValue: false,
|
|
|
|
wantValue: false,
|
|
|
|
policyError: syspolicy.ErrNoSuchKey,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "error enabled",
|
|
|
|
initialValue: true,
|
|
|
|
wantValue: true,
|
|
|
|
policyError: errors.New("test error"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "error disabled",
|
|
|
|
initialValue: false,
|
|
|
|
wantValue: false,
|
|
|
|
policyError: errors.New("test error"),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
for _, pp := range preferencePolicies {
|
|
|
|
t.Run(string(pp.key), func(t *testing.T) {
|
|
|
|
s := source.TestSetting[string]{
|
|
|
|
Key: pp.key,
|
|
|
|
Error: tt.policyError,
|
|
|
|
Value: tt.policyValue,
|
|
|
|
}
|
|
|
|
policyStore := source.NewTestStoreOf(t, s)
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
|
|
|
|
prefs := defaultPrefs.AsStruct()
|
|
|
|
pp.set(prefs, tt.initialValue)
|
|
|
|
|
|
|
|
gotAnyChange := applySysPolicy(prefs)
|
|
|
|
|
|
|
|
if gotAnyChange != tt.wantChange {
|
|
|
|
t.Errorf("anyChange=%v, want %v", gotAnyChange, tt.wantChange)
|
|
|
|
}
|
|
|
|
got := pp.get(prefs.View())
|
|
|
|
if got != tt.wantValue {
|
|
|
|
t.Errorf("pref=%v, want %v", got, tt.wantValue)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestOnTailnetDefaultAutoUpdate(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
before, after opt.Bool
|
|
|
|
container opt.Bool
|
|
|
|
tailnetDefault bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
before: opt.Bool(""),
|
|
|
|
tailnetDefault: true,
|
|
|
|
after: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.Bool(""),
|
|
|
|
tailnetDefault: false,
|
|
|
|
after: opt.NewBool(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.Bool("unset"),
|
|
|
|
tailnetDefault: true,
|
|
|
|
after: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.Bool("unset"),
|
|
|
|
tailnetDefault: false,
|
|
|
|
after: opt.NewBool(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.NewBool(false),
|
|
|
|
tailnetDefault: true,
|
|
|
|
after: opt.NewBool(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.NewBool(true),
|
|
|
|
tailnetDefault: false,
|
|
|
|
after: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.Bool(""),
|
|
|
|
container: opt.NewBool(true),
|
|
|
|
tailnetDefault: true,
|
|
|
|
after: opt.Bool(""),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.NewBool(false),
|
|
|
|
container: opt.NewBool(true),
|
|
|
|
tailnetDefault: true,
|
|
|
|
after: opt.NewBool(false),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
before: opt.NewBool(true),
|
|
|
|
container: opt.NewBool(true),
|
|
|
|
tailnetDefault: false,
|
|
|
|
after: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(fmt.Sprintf("before=%s,after=%s", tt.before, tt.after), func(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
b.hostinfo = hostinfo.New()
|
|
|
|
b.hostinfo.Container = tt.container
|
|
|
|
p := ipn.NewPrefs()
|
|
|
|
p.AutoUpdate.Apply = tt.before
|
ipn/ipnlocal: refactor and cleanup profileManager
In preparation for multi-user and unattended mode improvements, we are
refactoring and cleaning up `ipn/ipnlocal.profileManager`. The concept of the
"current user", which is only relevant on Windows, is being deprecated and will
soon be removed to allow more than one Windows user to connect and utilize
`LocalBackend` according to that user's access rights to the device and specific
Tailscale profiles.
We plan to pass the user's identity down to the `profileManager`, where it can
be used to determine the user's access rights to a given `LoginProfile`. While
the new permission model in `ipnauth` requires more work and is currently
blocked pending PR reviews, we are updating the `profileManager` to reduce its
reliance on the concept of a single OS user being connected to the backend at
the same time.
We extract the switching to the default Tailscale profile, which may also
trigger legacy profile migration, from `profileManager.SetCurrentUserID`. This
introduces `profileManager.DefaultUserProfileID`, which returns the default
profile ID for the current user, and `profileManager.SwitchToDefaultProfile`,
which is essentially a shorthand for `pm.SwitchProfile(pm.DefaultUserProfileID())`.
Both methods will eventually be updated to accept the user's identity and
utilize that user's default profile.
We make access checks more explicit by introducing the `profileManager.checkProfileAccess`
method. The current implementation continues to use `profileManager.currentUserID`
and `LoginProfile.LocalUserID` to determine whether access to a given profile
should be granted. This will be updated to utilize the `ipnauth` package and the
new permissions model once it's ready. We also expand access checks to be used
more widely in the `profileManager`, not just when switching or listing
profiles. This includes access checks in methods like `SetPrefs` and, most notably,
`DeleteProfile` and `DeleteAllProfiles`, preventing unprivileged Windows users
from deleting Tailscale profiles owned by other users on the same device,
including profiles owned by local admins.
We extract `profileManager.ProfilePrefs` and `profileManager.SetProfilePrefs`
methods that can be used to get and set preferences of a given `LoginProfile` if
`profileManager.checkProfileAccess` permits access to it.
We also update `profileManager.setUnattendedModeAsConfigured` to always enable
unattended mode on Windows if `Prefs.ForceDaemon` is true in the current
`LoginProfile`, even if `profileManager.currentUserID` is `""`. This facilitates
enabling unattended mode via `tailscale up --unattended` even if
`tailscale-ipn.exe` is not running, such as when a Group Policy or MDM-deployed
script runs at boot time, or when Tailscale is used on a Server Code or otherwise
headless Windows environments. See #12239, #2137, #3186 and
https://github.com/tailscale/tailscale/pull/6255#issuecomment-2016623838 for
details.
Fixes #12239
Updates tailscale/corp#18342
Updates #3186
Updates #2137
Signed-off-by: Nick Khyl <nickk@tailscale.com>
3 months ago
|
|
|
if err := b.pm.setPrefsNoPermCheck(p.View()); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
b.onTailnetDefaultAutoUpdate(tt.tailnetDefault)
|
|
|
|
want := tt.after
|
|
|
|
// On platforms that don't support auto-update we can never
|
|
|
|
// transition to auto-updates being enabled. The value should
|
|
|
|
// remain unchanged after onTailnetDefaultAutoUpdate.
|
|
|
|
if !clientupdate.CanAutoUpdate() && want.EqualBool(true) {
|
|
|
|
want = tt.before
|
|
|
|
}
|
|
|
|
if got := b.pm.CurrentPrefs().AutoUpdate().Apply; got != want {
|
|
|
|
t.Errorf("got: %q, want %q", got, want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTCPHandlerForDst(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
|
|
|
|
tests := []struct {
|
|
|
|
desc string
|
|
|
|
dst string
|
|
|
|
intercept bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
desc: "intercept port 80 (Web UI) on quad100 IPv4",
|
|
|
|
dst: "100.100.100.100:80",
|
|
|
|
intercept: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "intercept port 80 (Web UI) on quad100 IPv6",
|
|
|
|
dst: "[fd7a:115c:a1e0::53]:80",
|
|
|
|
intercept: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "don't intercept port 80 on local ip",
|
|
|
|
dst: "100.100.103.100:80",
|
|
|
|
intercept: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "intercept port 8080 (Taildrive) on quad100 IPv4",
|
|
|
|
dst: "100.100.100.100:8080",
|
|
|
|
intercept: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "intercept port 8080 (Taildrive) on quad100 IPv6",
|
|
|
|
dst: "[fd7a:115c:a1e0::53]:8080",
|
|
|
|
intercept: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "don't intercept port 8080 on local ip",
|
|
|
|
dst: "100.100.103.100:8080",
|
|
|
|
intercept: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "don't intercept port 9080 on quad100 IPv4",
|
|
|
|
dst: "100.100.100.100:9080",
|
|
|
|
intercept: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "don't intercept port 9080 on quad100 IPv6",
|
|
|
|
dst: "[fd7a:115c:a1e0::53]:9080",
|
|
|
|
intercept: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
desc: "don't intercept port 9080 on local ip",
|
|
|
|
dst: "100.100.103.100:9080",
|
|
|
|
intercept: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.dst, func(t *testing.T) {
|
|
|
|
t.Log(tt.desc)
|
|
|
|
src := netip.MustParseAddrPort("100.100.102.100:51234")
|
|
|
|
h, _ := b.TCPHandlerForDst(src, netip.MustParseAddrPort(tt.dst))
|
|
|
|
if !tt.intercept && h != nil {
|
|
|
|
t.Error("intercepted traffic we shouldn't have")
|
|
|
|
} else if tt.intercept && h == nil {
|
|
|
|
t.Error("failed to intercept traffic we should have")
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDriveManageShares(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
disabled bool
|
|
|
|
existing []*drive.Share
|
|
|
|
add *drive.Share
|
|
|
|
remove string
|
|
|
|
rename [2]string
|
|
|
|
expect any
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "append",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "d"},
|
|
|
|
},
|
|
|
|
add: &drive.Share{Name: " E "},
|
|
|
|
expect: []*drive.Share{
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "d"},
|
|
|
|
{Name: "e"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prepend",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "d"},
|
|
|
|
},
|
|
|
|
add: &drive.Share{Name: " A "},
|
|
|
|
expect: []*drive.Share{
|
|
|
|
{Name: "a"},
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "d"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "insert",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "d"},
|
|
|
|
},
|
|
|
|
add: &drive.Share{Name: " C "},
|
|
|
|
expect: []*drive.Share{
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "c"},
|
|
|
|
{Name: "d"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "replace",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "b", Path: "i"},
|
|
|
|
{Name: "d"},
|
|
|
|
},
|
|
|
|
add: &drive.Share{Name: " B ", Path: "ii"},
|
|
|
|
expect: []*drive.Share{
|
|
|
|
{Name: "b", Path: "ii"},
|
|
|
|
{Name: "d"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "add_bad_name",
|
|
|
|
add: &drive.Share{Name: "$"},
|
|
|
|
expect: drive.ErrInvalidShareName,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "add_disabled",
|
|
|
|
disabled: true,
|
|
|
|
add: &drive.Share{Name: "a"},
|
|
|
|
expect: drive.ErrDriveNotEnabled,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "remove",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "a"},
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "c"},
|
|
|
|
},
|
|
|
|
remove: "b",
|
|
|
|
expect: []*drive.Share{
|
|
|
|
{Name: "a"},
|
|
|
|
{Name: "c"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "remove_non_existing",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "a"},
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "c"},
|
|
|
|
},
|
|
|
|
remove: "D",
|
|
|
|
expect: os.ErrNotExist,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "remove_disabled",
|
|
|
|
disabled: true,
|
|
|
|
remove: "b",
|
|
|
|
expect: drive.ErrDriveNotEnabled,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rename",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "a"},
|
|
|
|
{Name: "b"},
|
|
|
|
},
|
|
|
|
rename: [2]string{"a", " C "},
|
|
|
|
expect: []*drive.Share{
|
|
|
|
{Name: "b"},
|
|
|
|
{Name: "c"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rename_not_exist",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "a"},
|
|
|
|
{Name: "b"},
|
|
|
|
},
|
|
|
|
rename: [2]string{"d", "c"},
|
|
|
|
expect: os.ErrNotExist,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rename_exists",
|
|
|
|
existing: []*drive.Share{
|
|
|
|
{Name: "a"},
|
|
|
|
{Name: "b"},
|
|
|
|
},
|
|
|
|
rename: [2]string{"a", "b"},
|
|
|
|
expect: os.ErrExist,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rename_bad_name",
|
|
|
|
rename: [2]string{"a", "$"},
|
|
|
|
expect: drive.ErrInvalidShareName,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rename_disabled",
|
|
|
|
disabled: true,
|
|
|
|
rename: [2]string{"a", "c"},
|
|
|
|
expect: drive.ErrDriveNotEnabled,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
drive.DisallowShareAs = true
|
|
|
|
t.Cleanup(func() {
|
|
|
|
drive.DisallowShareAs = false
|
|
|
|
})
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
b.mu.Lock()
|
|
|
|
if tt.existing != nil {
|
|
|
|
b.driveSetSharesLocked(tt.existing)
|
|
|
|
}
|
|
|
|
if !tt.disabled {
|
|
|
|
self := b.netMap.SelfNode.AsStruct()
|
|
|
|
self.CapMap = tailcfg.NodeCapMap{tailcfg.NodeAttrsTaildriveShare: nil}
|
|
|
|
b.netMap.SelfNode = self.View()
|
|
|
|
b.sys.Set(driveimpl.NewFileSystemForRemote(b.logf))
|
|
|
|
}
|
|
|
|
b.mu.Unlock()
|
|
|
|
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
|
|
|
t.Cleanup(cancel)
|
|
|
|
|
|
|
|
result := make(chan views.SliceView[*drive.Share, drive.ShareView], 1)
|
|
|
|
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(1)
|
|
|
|
go b.WatchNotifications(
|
|
|
|
ctx,
|
|
|
|
0,
|
|
|
|
func() { wg.Done() },
|
|
|
|
func(n *ipn.Notify) bool {
|
|
|
|
select {
|
|
|
|
case result <- n.DriveShares:
|
|
|
|
default:
|
|
|
|
//
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
},
|
|
|
|
)
|
|
|
|
wg.Wait()
|
|
|
|
|
|
|
|
var err error
|
|
|
|
switch {
|
|
|
|
case tt.add != nil:
|
|
|
|
err = b.DriveSetShare(tt.add)
|
|
|
|
case tt.remove != "":
|
|
|
|
err = b.DriveRemoveShare(tt.remove)
|
|
|
|
default:
|
|
|
|
err = b.DriveRenameShare(tt.rename[0], tt.rename[1])
|
|
|
|
}
|
|
|
|
|
|
|
|
switch e := tt.expect.(type) {
|
|
|
|
case error:
|
|
|
|
if !errors.Is(err, e) {
|
|
|
|
t.Errorf("expected error, want: %v got: %v", e, err)
|
|
|
|
}
|
|
|
|
case []*drive.Share:
|
|
|
|
if err != nil {
|
|
|
|
t.Errorf("unexpected error: %v", err)
|
|
|
|
} else {
|
|
|
|
r := <-result
|
|
|
|
|
|
|
|
got, err := json.MarshalIndent(r, "", " ")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("can't marshal got: %v", err)
|
|
|
|
}
|
|
|
|
want, err := json.MarshalIndent(e, "", " ")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("can't marshal want: %v", err)
|
|
|
|
}
|
|
|
|
if diff := cmp.Diff(string(got), string(want)); diff != "" {
|
|
|
|
t.Errorf("wrong shares; (-got+want):%v", diff)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestValidPopBrowserURL(t *testing.T) {
|
|
|
|
b := newTestBackend(t)
|
|
|
|
tests := []struct {
|
|
|
|
desc string
|
|
|
|
controlURL string
|
|
|
|
popBrowserURL string
|
|
|
|
want bool
|
|
|
|
}{
|
|
|
|
{"saas_login", "https://login.tailscale.com", "https://login.tailscale.com/a/foo", true},
|
|
|
|
{"saas_controlplane", "https://controlplane.tailscale.com", "https://controlplane.tailscale.com/a/foo", true},
|
|
|
|
{"saas_root", "https://login.tailscale.com", "https://tailscale.com/", true},
|
|
|
|
{"saas_bad_hostname", "https://login.tailscale.com", "https://example.com/a/foo", false},
|
|
|
|
{"localhost", "http://localhost", "http://localhost/a/foo", true},
|
|
|
|
{"custom_control_url_https", "https://example.com", "https://example.com/a/foo", true},
|
|
|
|
{"custom_control_url_https_diff_domain", "https://example.com", "https://other.com/a/foo", true},
|
|
|
|
{"custom_control_url_http", "http://example.com", "http://example.com/a/foo", true},
|
|
|
|
{"custom_control_url_http_diff_domain", "http://example.com", "http://other.com/a/foo", true},
|
|
|
|
{"bad_scheme", "https://example.com", "http://example.com/a/foo", false},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.desc, func(t *testing.T) {
|
|
|
|
if _, err := b.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
ControlURLSet: true,
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
ControlURL: tt.controlURL,
|
|
|
|
},
|
|
|
|
}); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
got := b.validPopBrowserURL(tt.popBrowserURL)
|
|
|
|
if got != tt.want {
|
|
|
|
t.Errorf("got %v, want %v", got, tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestRoundTraffic(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
bytes int64
|
|
|
|
want float64
|
|
|
|
}{
|
|
|
|
{name: "under 5 bytes", bytes: 4, want: 4},
|
|
|
|
{name: "under 1000 bytes", bytes: 987, want: 990},
|
|
|
|
{name: "under 10_000 bytes", bytes: 8875, want: 8900},
|
|
|
|
{name: "under 100_000 bytes", bytes: 77777, want: 78000},
|
|
|
|
{name: "under 1_000_000 bytes", bytes: 666523, want: 670000},
|
|
|
|
{name: "under 10_000_000 bytes", bytes: 22556677, want: 23000000},
|
|
|
|
{name: "under 1_000_000_000 bytes", bytes: 1234234234, want: 1200000000},
|
|
|
|
{name: "under 1_000_000_000 bytes", bytes: 123423423499, want: 123400000000},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
if result := roundTraffic(tt.bytes); result != tt.want {
|
|
|
|
t.Errorf("unexpected rounding got %v want %v", result, tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *LocalBackend) SetPrefsForTest(newp *ipn.Prefs) {
|
|
|
|
if newp == nil {
|
|
|
|
panic("SetPrefsForTest got nil prefs")
|
|
|
|
}
|
|
|
|
unlock := b.lockAndGetUnlock()
|
|
|
|
defer unlock()
|
|
|
|
b.setPrefsLockedOnEntry(newp, unlock)
|
|
|
|
}
|
|
|
|
|
|
|
|
type peerOptFunc func(*tailcfg.Node)
|
|
|
|
|
|
|
|
func makePeer(id tailcfg.NodeID, opts ...peerOptFunc) tailcfg.NodeView {
|
|
|
|
node := &tailcfg.Node{
|
|
|
|
ID: id,
|
|
|
|
StableID: tailcfg.StableNodeID(fmt.Sprintf("stable%d", id)),
|
|
|
|
Name: fmt.Sprintf("peer%d", id),
|
|
|
|
DERP: fmt.Sprintf("127.3.3.40:%d", id),
|
|
|
|
}
|
|
|
|
for _, opt := range opts {
|
|
|
|
opt(node)
|
|
|
|
}
|
|
|
|
return node.View()
|
|
|
|
}
|
|
|
|
|
|
|
|
func withName(name string) peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
n.Name = name
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withDERP(region int) peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
n.DERP = fmt.Sprintf("127.3.3.40:%d", region)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withoutDERP() peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
n.DERP = ""
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withLocation(loc tailcfg.LocationView) peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
var hi *tailcfg.Hostinfo
|
|
|
|
if n.Hostinfo.Valid() {
|
|
|
|
hi = n.Hostinfo.AsStruct()
|
|
|
|
} else {
|
|
|
|
hi = new(tailcfg.Hostinfo)
|
|
|
|
}
|
|
|
|
hi.Location = loc.AsStruct()
|
|
|
|
|
|
|
|
n.Hostinfo = hi.View()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withExitRoutes() peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
n.AllowedIPs = append(n.AllowedIPs, tsaddr.ExitRoutes()...)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withSuggest() peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
mak.Set(&n.CapMap, tailcfg.NodeAttrSuggestExitNode, []tailcfg.RawMessage{})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withCap(version tailcfg.CapabilityVersion) peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
n.Cap = version
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withOnline(isOnline bool) peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
n.Online = &isOnline
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func withNodeKey() peerOptFunc {
|
|
|
|
return func(n *tailcfg.Node) {
|
|
|
|
n.Key = key.NewNode().Public()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func deterministicRegionForTest(t testing.TB, want views.Slice[int], use int) selectRegionFunc {
|
|
|
|
t.Helper()
|
|
|
|
|
|
|
|
if !views.SliceContains(want, use) {
|
|
|
|
t.Errorf("invalid test: use %v is not in want %v", use, want)
|
|
|
|
}
|
|
|
|
|
|
|
|
return func(got views.Slice[int]) int {
|
|
|
|
if !views.SliceEqualAnyOrder(got, want) {
|
|
|
|
t.Errorf("candidate regions = %v, want %v", got, want)
|
|
|
|
}
|
|
|
|
return use
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func deterministicNodeForTest(t testing.TB, want views.Slice[tailcfg.StableNodeID], wantLast tailcfg.StableNodeID, use tailcfg.StableNodeID) selectNodeFunc {
|
|
|
|
t.Helper()
|
|
|
|
|
|
|
|
if !views.SliceContains(want, use) {
|
|
|
|
t.Errorf("invalid test: use %v is not in want %v", use, want)
|
|
|
|
}
|
|
|
|
|
|
|
|
return func(got views.Slice[tailcfg.NodeView], last tailcfg.StableNodeID) tailcfg.NodeView {
|
|
|
|
var ret tailcfg.NodeView
|
|
|
|
|
|
|
|
gotIDs := make([]tailcfg.StableNodeID, got.Len())
|
|
|
|
for i := range got.Len() {
|
|
|
|
nv := got.At(i)
|
|
|
|
if !nv.Valid() {
|
|
|
|
t.Fatalf("invalid node at index %v", i)
|
|
|
|
}
|
|
|
|
|
|
|
|
gotIDs[i] = nv.StableID()
|
|
|
|
if nv.StableID() == use {
|
|
|
|
ret = nv
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if !views.SliceEqualAnyOrder(views.SliceOf(gotIDs), want) {
|
|
|
|
t.Errorf("candidate nodes = %v, want %v", gotIDs, want)
|
|
|
|
}
|
|
|
|
if last != wantLast {
|
|
|
|
t.Errorf("last node = %v, want %v", last, wantLast)
|
|
|
|
}
|
|
|
|
if !ret.Valid() {
|
|
|
|
t.Fatalf("did not find matching node in %v, want %v", gotIDs, use)
|
|
|
|
}
|
|
|
|
|
|
|
|
return ret
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSuggestExitNode(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
|
|
|
|
defaultDERPMap := &tailcfg.DERPMap{
|
|
|
|
Regions: map[int]*tailcfg.DERPRegion{
|
|
|
|
1: {
|
|
|
|
Latitude: 32,
|
|
|
|
Longitude: -97,
|
|
|
|
},
|
|
|
|
2: {},
|
|
|
|
3: {},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
preferred1Report := &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10 * time.Millisecond,
|
|
|
|
2: 20 * time.Millisecond,
|
|
|
|
3: 30 * time.Millisecond,
|
|
|
|
},
|
|
|
|
PreferredDERP: 1,
|
|
|
|
}
|
|
|
|
noLatency1Report := &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 0,
|
|
|
|
2: 0,
|
|
|
|
3: 0,
|
|
|
|
},
|
|
|
|
PreferredDERP: 1,
|
|
|
|
}
|
|
|
|
preferredNoneReport := &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10 * time.Millisecond,
|
|
|
|
2: 20 * time.Millisecond,
|
|
|
|
3: 30 * time.Millisecond,
|
|
|
|
},
|
|
|
|
PreferredDERP: 0,
|
|
|
|
}
|
|
|
|
|
|
|
|
dallas := tailcfg.Location{
|
|
|
|
Latitude: 32.779167,
|
|
|
|
Longitude: -96.808889,
|
|
|
|
Priority: 100,
|
|
|
|
}
|
|
|
|
sanJose := tailcfg.Location{
|
|
|
|
Latitude: 37.3382082,
|
|
|
|
Longitude: -121.8863286,
|
|
|
|
Priority: 20,
|
|
|
|
}
|
|
|
|
fortWorth := tailcfg.Location{
|
|
|
|
Latitude: 32.756389,
|
|
|
|
Longitude: -97.3325,
|
|
|
|
Priority: 150,
|
|
|
|
}
|
|
|
|
fortWorthLowPriority := tailcfg.Location{
|
|
|
|
Latitude: 32.756389,
|
|
|
|
Longitude: -97.3325,
|
|
|
|
Priority: 100,
|
|
|
|
}
|
|
|
|
|
|
|
|
peer1 := makePeer(1,
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest())
|
|
|
|
peer2DERP1 := makePeer(2,
|
|
|
|
withDERP(1),
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest())
|
|
|
|
peer3 := makePeer(3,
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest())
|
|
|
|
peer4DERP3 := makePeer(4,
|
|
|
|
withDERP(3),
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest())
|
|
|
|
dallasPeer5 := makePeer(5,
|
|
|
|
withName("Dallas"),
|
|
|
|
withoutDERP(),
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest(),
|
|
|
|
withLocation(dallas.View()))
|
|
|
|
sanJosePeer6 := makePeer(6,
|
|
|
|
withName("San Jose"),
|
|
|
|
withoutDERP(),
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest(),
|
|
|
|
withLocation(sanJose.View()))
|
|
|
|
fortWorthPeer7 := makePeer(7,
|
|
|
|
withName("Fort Worth"),
|
|
|
|
withoutDERP(),
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest(),
|
|
|
|
withLocation(fortWorth.View()))
|
|
|
|
fortWorthPeer8LowPriority := makePeer(8,
|
|
|
|
withName("Fort Worth Low"),
|
|
|
|
withoutDERP(),
|
|
|
|
withExitRoutes(),
|
|
|
|
withSuggest(),
|
|
|
|
withLocation(fortWorthLowPriority.View()))
|
|
|
|
|
|
|
|
selfNode := tailcfg.Node{
|
|
|
|
Addresses: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("100.64.1.1/32"),
|
|
|
|
netip.MustParsePrefix("fe70::1/128"),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
defaultNetmap := &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer2DERP1,
|
|
|
|
peer3,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
locationNetmap := &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
dallasPeer5,
|
|
|
|
sanJosePeer6,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
largeNetmap := &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer2DERP1,
|
|
|
|
peer3,
|
|
|
|
peer4DERP3,
|
|
|
|
dallasPeer5,
|
|
|
|
sanJosePeer6,
|
|
|
|
fortWorthPeer7,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
|
|
|
|
lastReport *netcheck.Report
|
|
|
|
netMap *netmap.NetworkMap
|
|
|
|
lastSuggestion tailcfg.StableNodeID
|
|
|
|
|
|
|
|
allowPolicy []tailcfg.StableNodeID
|
|
|
|
|
|
|
|
wantRegions []int
|
|
|
|
useRegion int
|
|
|
|
|
|
|
|
wantNodes []tailcfg.StableNodeID
|
|
|
|
|
|
|
|
wantID tailcfg.StableNodeID
|
|
|
|
wantName string
|
|
|
|
wantLocation tailcfg.LocationView
|
|
|
|
|
|
|
|
wantError error
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "2 exit nodes in same region",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer2DERP1,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantNodes: []tailcfg.StableNodeID{
|
|
|
|
"stable1",
|
|
|
|
"stable2",
|
|
|
|
},
|
|
|
|
wantName: "peer1",
|
|
|
|
wantID: "stable1",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "2 exit nodes different regions unknown latency",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: defaultNetmap,
|
|
|
|
wantRegions: []int{1, 3}, // the only regions with peers
|
|
|
|
useRegion: 1,
|
|
|
|
wantName: "peer2",
|
|
|
|
wantID: "stable2",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "2 derp based exit nodes, different regions, equal latency",
|
|
|
|
lastReport: &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10,
|
|
|
|
2: 20,
|
|
|
|
3: 10,
|
|
|
|
},
|
|
|
|
PreferredDERP: 1,
|
|
|
|
},
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer3,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantRegions: []int{1, 2},
|
|
|
|
useRegion: 1,
|
|
|
|
wantName: "peer1",
|
|
|
|
wantID: "stable1",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "mullvad nodes, no derp based exit nodes",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: locationNetmap,
|
|
|
|
wantID: "stable5",
|
|
|
|
wantLocation: dallas.View(),
|
|
|
|
wantName: "Dallas",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "nearby mullvad nodes with different priorities",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
dallasPeer5,
|
|
|
|
sanJosePeer6,
|
|
|
|
fortWorthPeer7,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantID: "stable7",
|
|
|
|
wantLocation: fortWorth.View(),
|
|
|
|
wantName: "Fort Worth",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "nearby mullvad nodes with same priorities",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
dallasPeer5,
|
|
|
|
sanJosePeer6,
|
|
|
|
fortWorthPeer8LowPriority,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantNodes: []tailcfg.StableNodeID{"stable5", "stable8"},
|
|
|
|
wantID: "stable5",
|
|
|
|
wantLocation: dallas.View(),
|
|
|
|
wantName: "Dallas",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "mullvad nodes, remaining node is not in preferred derp",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
dallasPeer5,
|
|
|
|
sanJosePeer6,
|
|
|
|
peer4DERP3,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
useRegion: 3,
|
|
|
|
wantID: "stable4",
|
|
|
|
wantName: "peer4",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "no peers",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "nil report",
|
|
|
|
lastReport: nil,
|
|
|
|
netMap: largeNetmap,
|
|
|
|
wantError: ErrNoPreferredDERP,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "no preferred derp region",
|
|
|
|
lastReport: preferredNoneReport,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
},
|
|
|
|
wantError: ErrNoPreferredDERP,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "nil netmap",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: nil,
|
|
|
|
wantError: ErrNoPreferredDERP,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "nil derpmap",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: nil,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
dallasPeer5,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantError: ErrNoPreferredDERP,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "missing suggestion capability",
|
|
|
|
lastReport: noLatency1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
makePeer(1, withExitRoutes()),
|
|
|
|
makePeer(2, withLocation(dallas.View()), withExitRoutes()),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prefer last node",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
peer1,
|
|
|
|
peer2DERP1,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
lastSuggestion: "stable2",
|
|
|
|
wantNodes: []tailcfg.StableNodeID{
|
|
|
|
"stable1",
|
|
|
|
"stable2",
|
|
|
|
},
|
|
|
|
wantName: "peer2",
|
|
|
|
wantID: "stable2",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "found better derp node",
|
|
|
|
lastSuggestion: "stable3",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: defaultNetmap,
|
|
|
|
wantID: "stable2",
|
|
|
|
wantName: "peer2",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prefer last mullvad node",
|
|
|
|
lastSuggestion: "stable2",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
dallasPeer5,
|
|
|
|
sanJosePeer6,
|
|
|
|
fortWorthPeer8LowPriority,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantNodes: []tailcfg.StableNodeID{"stable5", "stable8"},
|
|
|
|
wantID: "stable5",
|
|
|
|
wantName: "Dallas",
|
|
|
|
wantLocation: dallas.View(),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "prefer better mullvad node",
|
|
|
|
lastSuggestion: "stable2",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: &netmap.NetworkMap{
|
|
|
|
SelfNode: selfNode.View(),
|
|
|
|
DERPMap: defaultDERPMap,
|
|
|
|
Peers: []tailcfg.NodeView{
|
|
|
|
dallasPeer5,
|
|
|
|
sanJosePeer6,
|
|
|
|
fortWorthPeer7,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantNodes: []tailcfg.StableNodeID{"stable7"},
|
|
|
|
wantID: "stable7",
|
|
|
|
wantName: "Fort Worth",
|
|
|
|
wantLocation: fortWorth.View(),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "large netmap",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: largeNetmap,
|
|
|
|
wantNodes: []tailcfg.StableNodeID{"stable1", "stable2"},
|
|
|
|
wantID: "stable2",
|
|
|
|
wantName: "peer2",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "no allowed suggestions",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: largeNetmap,
|
|
|
|
allowPolicy: []tailcfg.StableNodeID{},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "only derp suggestions",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: largeNetmap,
|
|
|
|
allowPolicy: []tailcfg.StableNodeID{"stable1", "stable2", "stable3"},
|
|
|
|
wantNodes: []tailcfg.StableNodeID{"stable1", "stable2"},
|
|
|
|
wantID: "stable2",
|
|
|
|
wantName: "peer2",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "only mullvad suggestions",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: largeNetmap,
|
|
|
|
allowPolicy: []tailcfg.StableNodeID{"stable5", "stable6", "stable7"},
|
|
|
|
wantID: "stable7",
|
|
|
|
wantName: "Fort Worth",
|
|
|
|
wantLocation: fortWorth.View(),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "only worst derp",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: largeNetmap,
|
|
|
|
allowPolicy: []tailcfg.StableNodeID{"stable3"},
|
|
|
|
wantID: "stable3",
|
|
|
|
wantName: "peer3",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "only worst mullvad",
|
|
|
|
lastReport: preferred1Report,
|
|
|
|
netMap: largeNetmap,
|
|
|
|
allowPolicy: []tailcfg.StableNodeID{"stable6"},
|
|
|
|
wantID: "stable6",
|
|
|
|
wantName: "San Jose",
|
|
|
|
wantLocation: sanJose.View(),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
wantRegions := tt.wantRegions
|
|
|
|
if wantRegions == nil {
|
|
|
|
wantRegions = []int{tt.useRegion}
|
|
|
|
}
|
|
|
|
selectRegion := deterministicRegionForTest(t, views.SliceOf(wantRegions), tt.useRegion)
|
|
|
|
|
|
|
|
wantNodes := tt.wantNodes
|
|
|
|
if wantNodes == nil {
|
|
|
|
wantNodes = []tailcfg.StableNodeID{tt.wantID}
|
|
|
|
}
|
|
|
|
selectNode := deterministicNodeForTest(t, views.SliceOf(wantNodes), tt.lastSuggestion, tt.wantID)
|
|
|
|
|
|
|
|
var allowList set.Set[tailcfg.StableNodeID]
|
|
|
|
if tt.allowPolicy != nil {
|
|
|
|
allowList = set.SetOf(tt.allowPolicy)
|
|
|
|
}
|
|
|
|
|
|
|
|
got, err := suggestExitNode(tt.lastReport, tt.netMap, tt.lastSuggestion, selectRegion, selectNode, allowList)
|
|
|
|
if got.Name != tt.wantName {
|
|
|
|
t.Errorf("name=%v, want %v", got.Name, tt.wantName)
|
|
|
|
}
|
|
|
|
if got.ID != tt.wantID {
|
|
|
|
t.Errorf("ID=%v, want %v", got.ID, tt.wantID)
|
|
|
|
}
|
|
|
|
if tt.wantError == nil && err != nil {
|
|
|
|
t.Errorf("err=%v, want no error", err)
|
|
|
|
}
|
|
|
|
if tt.wantError != nil && !errors.Is(err, tt.wantError) {
|
|
|
|
t.Errorf("err=%v, want %v", err, tt.wantError)
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(got.Location, tt.wantLocation) {
|
|
|
|
t.Errorf("location=%v, want %v", got.Location, tt.wantLocation)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSuggestExitNodePickWeighted(t *testing.T) {
|
|
|
|
location10 := tailcfg.Location{
|
|
|
|
Priority: 10,
|
|
|
|
}
|
|
|
|
location20 := tailcfg.Location{
|
|
|
|
Priority: 20,
|
|
|
|
}
|
|
|
|
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
candidates []tailcfg.NodeView
|
|
|
|
wantIDs []tailcfg.StableNodeID
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "different priorities",
|
|
|
|
candidates: []tailcfg.NodeView{
|
|
|
|
makePeer(2, withExitRoutes(), withLocation(location20.View())),
|
|
|
|
makePeer(3, withExitRoutes(), withLocation(location10.View())),
|
|
|
|
},
|
|
|
|
wantIDs: []tailcfg.StableNodeID{"stable2"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "same priorities",
|
|
|
|
candidates: []tailcfg.NodeView{
|
|
|
|
makePeer(2, withExitRoutes(), withLocation(location10.View())),
|
|
|
|
makePeer(3, withExitRoutes(), withLocation(location10.View())),
|
|
|
|
},
|
|
|
|
wantIDs: []tailcfg.StableNodeID{"stable2", "stable3"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "<1 candidates",
|
|
|
|
candidates: []tailcfg.NodeView{},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "1 candidate",
|
|
|
|
candidates: []tailcfg.NodeView{
|
|
|
|
makePeer(2, withExitRoutes(), withLocation(location20.View())),
|
|
|
|
},
|
|
|
|
wantIDs: []tailcfg.StableNodeID{"stable2"},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got := pickWeighted(tt.candidates)
|
|
|
|
gotIDs := make([]tailcfg.StableNodeID, 0, len(got))
|
|
|
|
for _, n := range got {
|
|
|
|
if !n.Valid() {
|
|
|
|
gotIDs = append(gotIDs, "<invalid>")
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
gotIDs = append(gotIDs, n.StableID())
|
|
|
|
}
|
|
|
|
if !views.SliceEqualAnyOrder(views.SliceOf(gotIDs), views.SliceOf(tt.wantIDs)) {
|
|
|
|
t.Errorf("node IDs = %v, want %v", gotIDs, tt.wantIDs)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSuggestExitNodeLongLatDistance(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
fromLat float64
|
|
|
|
fromLong float64
|
|
|
|
toLat float64
|
|
|
|
toLong float64
|
|
|
|
want float64
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "zero values",
|
|
|
|
fromLat: 0,
|
|
|
|
fromLong: 0,
|
|
|
|
toLat: 0,
|
|
|
|
toLong: 0,
|
|
|
|
want: 0,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "valid values",
|
|
|
|
fromLat: 40.73061,
|
|
|
|
fromLong: -73.935242,
|
|
|
|
toLat: 37.3382082,
|
|
|
|
toLong: -121.8863286,
|
|
|
|
want: 4117266.873301274,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "valid values, locations in north and south of equator",
|
|
|
|
fromLat: 40.73061,
|
|
|
|
fromLong: -73.935242,
|
|
|
|
toLat: -33.861481,
|
|
|
|
toLong: 151.205475,
|
|
|
|
want: 15994089.144368416,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
// The wanted values are computed using a more precise algorithm using the WGS84 model but
|
|
|
|
// longLatDistance uses a spherical approximation for simplicity. To account for this, we allow for
|
|
|
|
// 10km of error.
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got := longLatDistance(tt.fromLat, tt.fromLong, tt.toLat, tt.toLong)
|
|
|
|
const maxError = 10000 // 10km
|
|
|
|
if math.Abs(got-tt.want) > maxError {
|
|
|
|
t.Errorf("distance=%vm, want within %vm of %vm", got, maxError, tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestMinLatencyDERPregion(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
regions []int
|
|
|
|
report *netcheck.Report
|
|
|
|
wantRegion int
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "regions, no latency values",
|
|
|
|
regions: []int{1, 2, 3},
|
|
|
|
wantRegion: 0,
|
|
|
|
report: &netcheck.Report{},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "regions, different latency values",
|
|
|
|
regions: []int{1, 2, 3},
|
|
|
|
wantRegion: 2,
|
|
|
|
report: &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10 * time.Millisecond,
|
|
|
|
2: 5 * time.Millisecond,
|
|
|
|
3: 30 * time.Millisecond,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "regions, same values",
|
|
|
|
regions: []int{1, 2, 3},
|
|
|
|
wantRegion: 1,
|
|
|
|
report: &netcheck.Report{
|
|
|
|
RegionLatency: map[int]time.Duration{
|
|
|
|
1: 10 * time.Millisecond,
|
|
|
|
2: 10 * time.Millisecond,
|
|
|
|
3: 10 * time.Millisecond,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got := minLatencyDERPRegion(tt.regions, tt.report)
|
|
|
|
if got != tt.wantRegion {
|
|
|
|
t.Errorf("got region %v want region %v", got, tt.wantRegion)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestShouldAutoExitNode(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
exitNodeIDPolicyValue string
|
|
|
|
expectedBool bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "auto:any",
|
|
|
|
exitNodeIDPolicyValue: "auto:any",
|
|
|
|
expectedBool: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "no auto prefix",
|
|
|
|
exitNodeIDPolicyValue: "foo",
|
|
|
|
expectedBool: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "auto prefix but empty suffix",
|
|
|
|
exitNodeIDPolicyValue: "auto:",
|
|
|
|
expectedBool: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "auto prefix no colon",
|
|
|
|
exitNodeIDPolicyValue: "auto",
|
|
|
|
expectedBool: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "auto prefix invalid suffix",
|
|
|
|
exitNodeIDPolicyValue: "auto:foo",
|
|
|
|
expectedBool: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
policyStore := source.NewTestStoreOf(t, source.TestSettingOf(
|
|
|
|
syspolicy.ExitNodeID, tt.exitNodeIDPolicyValue,
|
|
|
|
))
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
|
|
|
|
got := shouldAutoExitNode()
|
|
|
|
if got != tt.expectedBool {
|
|
|
|
t.Fatalf("expected %v got %v for %v policy value", tt.expectedBool, got, tt.exitNodeIDPolicyValue)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestEnableAutoUpdates(t *testing.T) {
|
|
|
|
lb := newTestLocalBackend(t)
|
|
|
|
|
|
|
|
_, err := lb.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
AutoUpdateSet: ipn.AutoUpdatePrefsMask{
|
|
|
|
ApplySet: true,
|
|
|
|
},
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Apply: opt.NewBool(true),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
})
|
|
|
|
// Enabling may fail, depending on which environment we are running this
|
|
|
|
// test in.
|
|
|
|
wantErr := !clientupdate.CanAutoUpdate()
|
|
|
|
gotErr := err != nil
|
|
|
|
if gotErr != wantErr {
|
|
|
|
t.Fatalf("enabling auto-updates: got error: %v (%v); want error: %v", gotErr, err, wantErr)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Disabling should always succeed.
|
|
|
|
if _, err := lb.EditPrefs(&ipn.MaskedPrefs{
|
|
|
|
AutoUpdateSet: ipn.AutoUpdatePrefsMask{
|
|
|
|
ApplySet: true,
|
|
|
|
},
|
|
|
|
Prefs: ipn.Prefs{
|
|
|
|
AutoUpdate: ipn.AutoUpdatePrefs{
|
|
|
|
Apply: opt.NewBool(false),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}); err != nil {
|
|
|
|
t.Fatalf("disabling auto-updates: got error: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestReadWriteRouteInfo(t *testing.T) {
|
|
|
|
// set up a backend with more than one profile
|
|
|
|
b := newTestBackend(t)
|
|
|
|
prof1 := ipn.LoginProfile{ID: "id1", Key: "key1"}
|
|
|
|
prof2 := ipn.LoginProfile{ID: "id2", Key: "key2"}
|
|
|
|
b.pm.knownProfiles["id1"] = &prof1
|
|
|
|
b.pm.knownProfiles["id2"] = &prof2
|
|
|
|
b.pm.currentProfile = &prof1
|
|
|
|
|
|
|
|
// set up routeInfo
|
|
|
|
ri1 := &appc.RouteInfo{}
|
|
|
|
ri1.Wildcards = []string{"1"}
|
|
|
|
|
|
|
|
ri2 := &appc.RouteInfo{}
|
|
|
|
ri2.Wildcards = []string{"2"}
|
|
|
|
|
|
|
|
// read before write
|
|
|
|
readRi, err := b.readRouteInfoLocked()
|
|
|
|
if readRi != nil {
|
|
|
|
t.Fatalf("read before writing: want nil, got %v", readRi)
|
|
|
|
}
|
|
|
|
if err != ipn.ErrStateNotExist {
|
|
|
|
t.Fatalf("read before writing: want %v, got %v", ipn.ErrStateNotExist, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// write the first routeInfo
|
|
|
|
if err := b.storeRouteInfo(ri1); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// write the other routeInfo as the other profile
|
|
|
|
if err := b.pm.SwitchProfile("id2"); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if err := b.storeRouteInfo(ri2); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// read the routeInfo of the first profile
|
|
|
|
if err := b.pm.SwitchProfile("id1"); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
readRi, err = b.readRouteInfoLocked()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if !slices.Equal(readRi.Wildcards, ri1.Wildcards) {
|
|
|
|
t.Fatalf("read prof1 routeInfo wildcards: want %v, got %v", ri1.Wildcards, readRi.Wildcards)
|
|
|
|
}
|
|
|
|
|
|
|
|
// read the routeInfo of the second profile
|
|
|
|
if err := b.pm.SwitchProfile("id2"); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
readRi, err = b.readRouteInfoLocked()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if !slices.Equal(readRi.Wildcards, ri2.Wildcards) {
|
|
|
|
t.Fatalf("read prof2 routeInfo wildcards: want %v, got %v", ri2.Wildcards, readRi.Wildcards)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestFillAllowedSuggestions(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
allowPolicy []string
|
|
|
|
want []tailcfg.StableNodeID
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "unset",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "zero",
|
|
|
|
allowPolicy: []string{},
|
|
|
|
want: []tailcfg.StableNodeID{},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "one",
|
|
|
|
allowPolicy: []string{"one"},
|
|
|
|
want: []tailcfg.StableNodeID{"one"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "many",
|
|
|
|
allowPolicy: []string{"one", "two", "three", "four"},
|
|
|
|
want: []tailcfg.StableNodeID{"one", "three", "four", "two"}, // order should not matter
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "preserve case",
|
|
|
|
allowPolicy: []string{"ABC", "def", "gHiJ"},
|
|
|
|
want: []tailcfg.StableNodeID{"ABC", "def", "gHiJ"},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
syspolicy.RegisterWellKnownSettingsForTest(t)
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
policyStore := source.NewTestStoreOf(t, source.TestSettingOf(
|
|
|
|
syspolicy.AllowedSuggestedExitNodes, tt.allowPolicy,
|
|
|
|
))
|
|
|
|
syspolicy.MustRegisterStoreForTest(t, "TestStore", setting.DeviceScope, policyStore)
|
|
|
|
|
|
|
|
got := fillAllowedSuggestions()
|
|
|
|
if got == nil {
|
|
|
|
if tt.want == nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
t.Errorf("got nil, want %v", tt.want)
|
|
|
|
}
|
|
|
|
if tt.want == nil {
|
|
|
|
t.Errorf("got %v, want nil", got)
|
|
|
|
}
|
|
|
|
|
|
|
|
if !got.Equal(set.SetOf(tt.want)) {
|
|
|
|
t.Errorf("got %v, want %v", got, tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
ipn/{ipnauth,ipnlocal,ipnserver}: send the auth URL to the user who started interactive login
We add the ClientID() method to the ipnauth.Actor interface and updated ipnserver.actor to implement it.
This method returns a unique ID of the connected client if the actor represents one. It helps link a series
of interactions initiated by the client, such as when a notification needs to be sent back to a specific session,
rather than all active sessions, in response to a certain request.
We also add LocalBackend.WatchNotificationsAs and LocalBackend.StartLoginInteractiveAs methods,
which are like WatchNotifications and StartLoginInteractive but accept an additional parameter
specifying an ipnauth.Actor who initiates the operation. We store these actor identities in
watchSession.owner and LocalBackend.authActor, respectively,and implement LocalBackend.sendTo
and related helper methods to enable sending notifications to watchSessions associated with actors
(or, more broadly, identifiable recipients).
We then use the above to change who receives the BrowseToURL notifications:
- For user-initiated, interactive logins, the notification is delivered only to the user who initiated the
process. If the initiating actor represents a specific connected client, the URL notification is sent back
to the same LocalAPI client that called StartLoginInteractive. Otherwise, the notification is sent to all
clients connected as that user.
Currently, we only differentiate between users on Windows, as it is inherently a multi-user OS.
- In all other cases (e.g., node key expiration), we send the notification to all connected users.
Updates tailscale/corp#18342
Signed-off-by: Nick Khyl <nickk@tailscale.com>
1 month ago
|
|
|
|
|
|
|
func TestNotificationTargetMatch(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
target notificationTarget
|
|
|
|
actor ipnauth.Actor
|
|
|
|
wantMatch bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "AllClients/Nil",
|
|
|
|
target: allClients,
|
|
|
|
actor: nil,
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "AllClients/NoUID/NoCID",
|
|
|
|
target: allClients,
|
|
|
|
actor: &ipnauth.TestActor{},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "AllClients/WithUID/NoCID",
|
|
|
|
target: allClients,
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4", CID: ipnauth.NoClientID},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "AllClients/NoUID/WithCID",
|
|
|
|
target: allClients,
|
|
|
|
actor: &ipnauth.TestActor{CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "AllClients/WithUID/WithCID",
|
|
|
|
target: allClients,
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4", CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID/Nil",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: nil,
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID/NoUID/NoCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: &ipnauth.TestActor{},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID/NoUID/WithCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: &ipnauth.TestActor{CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID/SameUID/NoCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4"},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID/DifferentUID/NoCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-5-6-7-8"},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID/SameUID/WithCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4", CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID/DifferentUID/WithCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-5-6-7-8", CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByCID/Nil",
|
|
|
|
target: notificationTarget{clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: nil,
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByCID/NoUID/NoCID",
|
|
|
|
target: notificationTarget{clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByCID/NoUID/SameCID",
|
|
|
|
target: notificationTarget{clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByCID/NoUID/DifferentCID",
|
|
|
|
target: notificationTarget{clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{CID: ipnauth.ClientIDFrom("B")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByCID/WithUID/NoCID",
|
|
|
|
target: notificationTarget{clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4"},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByCID/WithUID/SameCID",
|
|
|
|
target: notificationTarget{clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4", CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByCID/WithUID/DifferentCID",
|
|
|
|
target: notificationTarget{clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4", CID: ipnauth.ClientIDFrom("B")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/Nil",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4"},
|
|
|
|
actor: nil,
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/NoUID/NoCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/NoUID/SameCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/NoUID/DifferentCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{CID: ipnauth.ClientIDFrom("B")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/SameUID/NoCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4"},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/SameUID/SameCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4", CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/SameUID/DifferentCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-1-2-3-4", CID: ipnauth.ClientIDFrom("B")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/DifferentUID/NoCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-5-6-7-8"},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/DifferentUID/SameCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-5-6-7-8", CID: ipnauth.ClientIDFrom("A")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "FilterByUID+CID/DifferentUID/DifferentCID",
|
|
|
|
target: notificationTarget{userID: "S-1-5-21-1-2-3-4", clientID: ipnauth.ClientIDFrom("A")},
|
|
|
|
actor: &ipnauth.TestActor{UID: "S-1-5-21-5-6-7-8", CID: ipnauth.ClientIDFrom("B")},
|
|
|
|
wantMatch: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
gotMatch := tt.target.match(tt.actor)
|
|
|
|
if gotMatch != tt.wantMatch {
|
|
|
|
t.Errorf("match: got %v; want %v", gotMatch, tt.wantMatch)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
type newTestControlFn func(tb testing.TB, opts controlclient.Options) controlclient.Client
|
|
|
|
|
|
|
|
func newLocalBackendWithTestControl(t *testing.T, enableLogging bool, newControl newTestControlFn) *LocalBackend {
|
|
|
|
logf := logger.Discard
|
|
|
|
if enableLogging {
|
|
|
|
logf = tstest.WhileTestRunningLogger(t)
|
|
|
|
}
|
|
|
|
sys := new(tsd.System)
|
|
|
|
store := new(mem.Store)
|
|
|
|
sys.Set(store)
|
|
|
|
e, err := wgengine.NewFakeUserspaceEngine(logf, sys.Set, sys.HealthTracker(), sys.UserMetricsRegistry())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("NewFakeUserspaceEngine: %v", err)
|
|
|
|
}
|
|
|
|
t.Cleanup(e.Close)
|
|
|
|
sys.Set(e)
|
|
|
|
|
|
|
|
b, err := NewLocalBackend(logf, logid.PublicID{}, sys, 0)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("NewLocalBackend: %v", err)
|
|
|
|
}
|
|
|
|
b.DisablePortMapperForTest()
|
|
|
|
|
|
|
|
b.SetControlClientGetterForTesting(func(opts controlclient.Options) (controlclient.Client, error) {
|
|
|
|
return newControl(t, opts), nil
|
|
|
|
})
|
|
|
|
return b
|
|
|
|
}
|
|
|
|
|
|
|
|
// notificationHandler is any function that can process (e.g., check) a notification.
|
|
|
|
// It returns whether the notification has been handled or should be passed to the next handler.
|
|
|
|
// The handler may be called from any goroutine, so it must avoid calling functions
|
|
|
|
// that are restricted to the goroutine running the test or benchmark function,
|
|
|
|
// such as [testing.common.FailNow] and [testing.common.Fatalf].
|
|
|
|
type notificationHandler func(testing.TB, ipnauth.Actor, *ipn.Notify) bool
|
|
|
|
|
|
|
|
// wantedNotification names a [notificationHandler] that processes a notification
|
|
|
|
// the test expects and wants to receive. The name is used to report notifications
|
|
|
|
// that haven't been received within the expected timeout.
|
|
|
|
type wantedNotification struct {
|
|
|
|
name string
|
|
|
|
cond notificationHandler
|
|
|
|
}
|
|
|
|
|
|
|
|
// notificationWatcher observes [LocalBackend] notifications as the specified actor,
|
|
|
|
// reporting missing but expected notifications using [testing.common.Error],
|
|
|
|
// and delegating the handling of unexpected notifications to the [notificationHandler]s.
|
|
|
|
type notificationWatcher struct {
|
|
|
|
tb testing.TB
|
|
|
|
lb *LocalBackend
|
|
|
|
actor ipnauth.Actor
|
|
|
|
|
|
|
|
mu sync.Mutex
|
|
|
|
mask ipn.NotifyWatchOpt
|
|
|
|
want []wantedNotification // notifications we want to receive
|
|
|
|
unexpected []notificationHandler // funcs that are called to check any other notifications
|
|
|
|
ctxCancel context.CancelFunc // cancels the outstanding [LocalBackend.WatchNotificationsAs] call
|
|
|
|
got []*ipn.Notify // all notifications, both wanted and unexpected, we've received so far
|
|
|
|
gotWanted []*ipn.Notify // only the expected notifications; holds nil for any notification that hasn't been received
|
|
|
|
gotWantedCh chan struct{} // closed when we have received the last wanted notification
|
|
|
|
doneCh chan struct{} // closed when [LocalBackend.WatchNotificationsAs] returns
|
|
|
|
}
|
|
|
|
|
|
|
|
func newNotificationWatcher(tb testing.TB, lb *LocalBackend, actor ipnauth.Actor) *notificationWatcher {
|
|
|
|
return ¬ificationWatcher{tb: tb, lb: lb, actor: actor}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *notificationWatcher) watch(mask ipn.NotifyWatchOpt, wanted []wantedNotification, unexpected ...notificationHandler) {
|
|
|
|
w.tb.Helper()
|
|
|
|
|
|
|
|
// Cancel any outstanding [LocalBackend.WatchNotificationsAs] calls.
|
|
|
|
w.mu.Lock()
|
|
|
|
ctxCancel := w.ctxCancel
|
|
|
|
doneCh := w.doneCh
|
|
|
|
w.mu.Unlock()
|
|
|
|
if doneCh != nil {
|
|
|
|
ctxCancel()
|
|
|
|
<-doneCh
|
|
|
|
}
|
|
|
|
|
|
|
|
doneCh = make(chan struct{})
|
|
|
|
gotWantedCh := make(chan struct{})
|
|
|
|
ctx, ctxCancel := context.WithCancel(context.Background())
|
|
|
|
w.tb.Cleanup(func() {
|
|
|
|
ctxCancel()
|
|
|
|
<-doneCh
|
|
|
|
})
|
|
|
|
|
|
|
|
w.mu.Lock()
|
|
|
|
w.mask = mask
|
|
|
|
w.want = wanted
|
|
|
|
w.unexpected = unexpected
|
|
|
|
w.ctxCancel = ctxCancel
|
|
|
|
w.got = nil
|
|
|
|
w.gotWanted = make([]*ipn.Notify, len(wanted))
|
|
|
|
w.gotWantedCh = gotWantedCh
|
|
|
|
w.doneCh = doneCh
|
|
|
|
w.mu.Unlock()
|
|
|
|
|
|
|
|
watchAddedCh := make(chan struct{})
|
|
|
|
go func() {
|
|
|
|
defer close(doneCh)
|
|
|
|
if len(wanted) == 0 {
|
|
|
|
close(gotWantedCh)
|
|
|
|
if len(unexpected) == 0 {
|
|
|
|
close(watchAddedCh)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var nextWantIdx int
|
|
|
|
w.lb.WatchNotificationsAs(ctx, w.actor, w.mask, func() { close(watchAddedCh) }, func(notify *ipn.Notify) (keepGoing bool) {
|
|
|
|
w.tb.Helper()
|
|
|
|
|
|
|
|
w.mu.Lock()
|
|
|
|
defer w.mu.Unlock()
|
|
|
|
w.got = append(w.got, notify)
|
|
|
|
|
|
|
|
wanted := false
|
|
|
|
for i := nextWantIdx; i < len(w.want); i++ {
|
|
|
|
if wanted = w.want[i].cond(w.tb, w.actor, notify); wanted {
|
|
|
|
w.gotWanted[i] = notify
|
|
|
|
nextWantIdx = i + 1
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if wanted && nextWantIdx == len(w.want) {
|
|
|
|
close(w.gotWantedCh)
|
|
|
|
if len(w.unexpected) == 0 {
|
|
|
|
// If we have received the last wanted notification,
|
|
|
|
// and we don't have any handlers for the unexpected notifications,
|
|
|
|
// we can stop the watcher right away.
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if !wanted {
|
|
|
|
// If we've received a notification we didn't expect,
|
|
|
|
// it could either be an unwanted notification caused by a bug
|
|
|
|
// or just a miscellaneous one that's irrelevant for the current test.
|
|
|
|
// Call unexpected notification handlers, if any, to
|
|
|
|
// check and fail the test if necessary.
|
|
|
|
for _, h := range w.unexpected {
|
|
|
|
if h(w.tb, w.actor, notify) {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true
|
|
|
|
})
|
|
|
|
|
|
|
|
}()
|
|
|
|
<-watchAddedCh
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *notificationWatcher) check() []*ipn.Notify {
|
|
|
|
w.tb.Helper()
|
|
|
|
|
|
|
|
w.mu.Lock()
|
|
|
|
cancel := w.ctxCancel
|
|
|
|
gotWantedCh := w.gotWantedCh
|
|
|
|
checkUnexpected := len(w.unexpected) != 0
|
|
|
|
doneCh := w.doneCh
|
|
|
|
w.mu.Unlock()
|
|
|
|
|
|
|
|
// Wait for up to 10 seconds to receive expected notifications.
|
|
|
|
timeout := 10 * time.Second
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-gotWantedCh:
|
|
|
|
if checkUnexpected {
|
|
|
|
gotWantedCh = nil
|
|
|
|
// But do not wait longer than 500ms for unexpected notifications after
|
|
|
|
// the expected notifications have been received.
|
|
|
|
timeout = 500 * time.Millisecond
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
case <-doneCh:
|
|
|
|
// [LocalBackend.WatchNotificationsAs] has already returned, so no further
|
|
|
|
// notifications will be received. There's no reason to wait any longer.
|
|
|
|
case <-time.After(timeout):
|
|
|
|
}
|
|
|
|
cancel()
|
|
|
|
<-doneCh
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
// Report missing notifications, if any, and log all received notifications,
|
|
|
|
// including both expected and unexpected ones.
|
|
|
|
w.mu.Lock()
|
|
|
|
defer w.mu.Unlock()
|
|
|
|
if hasMissing := slices.Contains(w.gotWanted, nil); hasMissing {
|
|
|
|
want := make([]string, len(w.want))
|
|
|
|
got := make([]string, 0, len(w.want))
|
|
|
|
for i, wn := range w.want {
|
|
|
|
want[i] = wn.name
|
|
|
|
if w.gotWanted[i] != nil {
|
|
|
|
got = append(got, wn.name)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
w.tb.Errorf("Notifications(%s): got %q; want %q", actorDescriptionForTest(w.actor), strings.Join(got, ", "), strings.Join(want, ", "))
|
|
|
|
for i, n := range w.got {
|
|
|
|
w.tb.Logf("%d. %v", i, n)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
return w.gotWanted
|
|
|
|
}
|
|
|
|
|
|
|
|
func actorDescriptionForTest(actor ipnauth.Actor) string {
|
|
|
|
var parts []string
|
|
|
|
if actor != nil {
|
|
|
|
if name, _ := actor.Username(); name != "" {
|
|
|
|
parts = append(parts, name)
|
|
|
|
}
|
|
|
|
if uid := actor.UserID(); uid != "" {
|
|
|
|
parts = append(parts, string(uid))
|
|
|
|
}
|
|
|
|
if clientID, _ := actor.ClientID(); clientID != ipnauth.NoClientID {
|
|
|
|
parts = append(parts, clientID.String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return fmt.Sprintf("Actor{%s}", strings.Join(parts, ", "))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestLoginNotifications(t *testing.T) {
|
|
|
|
const (
|
|
|
|
enableLogging = true
|
|
|
|
controlURL = "https://localhost:1/"
|
|
|
|
loginURL = "https://localhost:1/1"
|
|
|
|
)
|
|
|
|
|
|
|
|
wantBrowseToURL := wantedNotification{
|
|
|
|
name: "BrowseToURL",
|
|
|
|
cond: func(t testing.TB, actor ipnauth.Actor, n *ipn.Notify) bool {
|
|
|
|
if n.BrowseToURL != nil && *n.BrowseToURL != loginURL {
|
|
|
|
t.Errorf("BrowseToURL (%s): got %q; want %q", actorDescriptionForTest(actor), *n.BrowseToURL, loginURL)
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
return n.BrowseToURL != nil
|
|
|
|
},
|
|
|
|
}
|
|
|
|
unexpectedBrowseToURL := func(t testing.TB, actor ipnauth.Actor, n *ipn.Notify) bool {
|
|
|
|
if n.BrowseToURL != nil {
|
|
|
|
t.Errorf("Unexpected BrowseToURL(%s): %v", actorDescriptionForTest(actor), n)
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
logInAs ipnauth.Actor
|
|
|
|
urlExpectedBy []ipnauth.Actor
|
|
|
|
urlUnexpectedBy []ipnauth.Actor
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "NoObservers",
|
|
|
|
logInAs: &ipnauth.TestActor{UID: "A"},
|
|
|
|
urlExpectedBy: []ipnauth.Actor{}, // ensure that it does not panic if no one is watching
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "SingleUser",
|
|
|
|
logInAs: &ipnauth.TestActor{UID: "A"},
|
|
|
|
urlExpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A"}},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "SameUser/TwoSessions/NoCID",
|
|
|
|
logInAs: &ipnauth.TestActor{UID: "A"},
|
|
|
|
urlExpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A"}, &ipnauth.TestActor{UID: "A"}},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "SameUser/TwoSessions/OneWithCID",
|
|
|
|
logInAs: &ipnauth.TestActor{UID: "A", CID: ipnauth.ClientIDFrom("123")},
|
|
|
|
urlExpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A", CID: ipnauth.ClientIDFrom("123")}},
|
|
|
|
urlUnexpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A"}},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "SameUser/TwoSessions/BothWithCID",
|
|
|
|
logInAs: &ipnauth.TestActor{UID: "A", CID: ipnauth.ClientIDFrom("123")},
|
|
|
|
urlExpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A", CID: ipnauth.ClientIDFrom("123")}},
|
|
|
|
urlUnexpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A", CID: ipnauth.ClientIDFrom("456")}},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "DifferentUsers/NoCID",
|
|
|
|
logInAs: &ipnauth.TestActor{UID: "A"},
|
|
|
|
urlExpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A"}},
|
|
|
|
urlUnexpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "B"}},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "DifferentUsers/SameCID",
|
|
|
|
logInAs: &ipnauth.TestActor{UID: "A"},
|
|
|
|
urlExpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "A", CID: ipnauth.ClientIDFrom("123")}},
|
|
|
|
urlUnexpectedBy: []ipnauth.Actor{&ipnauth.TestActor{UID: "B", CID: ipnauth.ClientIDFrom("123")}},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
|
|
|
|
lb := newLocalBackendWithTestControl(t, enableLogging, func(tb testing.TB, opts controlclient.Options) controlclient.Client {
|
|
|
|
return newClient(tb, opts)
|
|
|
|
})
|
|
|
|
if _, err := lb.EditPrefs(&ipn.MaskedPrefs{ControlURLSet: true, Prefs: ipn.Prefs{ControlURL: controlURL}}); err != nil {
|
|
|
|
t.Fatalf("(*EditPrefs).Start(): %v", err)
|
|
|
|
}
|
|
|
|
if err := lb.Start(ipn.Options{}); err != nil {
|
|
|
|
t.Fatalf("(*LocalBackend).Start(): %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
sessions := make([]*notificationWatcher, 0, len(tt.urlExpectedBy)+len(tt.urlUnexpectedBy))
|
|
|
|
for _, actor := range tt.urlExpectedBy {
|
|
|
|
session := newNotificationWatcher(t, lb, actor)
|
|
|
|
session.watch(0, []wantedNotification{wantBrowseToURL})
|
|
|
|
sessions = append(sessions, session)
|
|
|
|
}
|
|
|
|
for _, actor := range tt.urlUnexpectedBy {
|
|
|
|
session := newNotificationWatcher(t, lb, actor)
|
|
|
|
session.watch(0, nil, unexpectedBrowseToURL)
|
|
|
|
sessions = append(sessions, session)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := lb.StartLoginInteractiveAs(context.Background(), tt.logInAs); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
lb.cc.(*mockControl).send(nil, loginURL, false, nil)
|
|
|
|
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(len(sessions))
|
|
|
|
for _, sess := range sessions {
|
|
|
|
go func() { // check all sessions in parallel
|
|
|
|
sess.check()
|
|
|
|
wg.Done()
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TestConfigFileReload tests that the LocalBackend reloads its configuration
|
|
|
|
// when the configuration file changes.
|
|
|
|
func TestConfigFileReload(t *testing.T) {
|
|
|
|
cfg1 := `{"Hostname": "foo", "Version": "alpha0"}`
|
|
|
|
f := filepath.Join(t.TempDir(), "cfg")
|
|
|
|
must.Do(os.WriteFile(f, []byte(cfg1), 0600))
|
|
|
|
sys := new(tsd.System)
|
|
|
|
sys.InitialConfig = must.Get(conffile.Load(f))
|
|
|
|
lb := newTestLocalBackendWithSys(t, sys)
|
|
|
|
must.Do(lb.Start(ipn.Options{}))
|
|
|
|
|
|
|
|
lb.mu.Lock()
|
|
|
|
hn := lb.hostinfo.Hostname
|
|
|
|
lb.mu.Unlock()
|
|
|
|
if hn != "foo" {
|
|
|
|
t.Fatalf("got %q; want %q", hn, "foo")
|
|
|
|
}
|
|
|
|
|
|
|
|
cfg2 := `{"Hostname": "bar", "Version": "alpha0"}`
|
|
|
|
must.Do(os.WriteFile(f, []byte(cfg2), 0600))
|
|
|
|
if !must.Get(lb.ReloadConfig()) {
|
|
|
|
t.Fatal("reload failed")
|
|
|
|
}
|
|
|
|
|
|
|
|
lb.mu.Lock()
|
|
|
|
hn = lb.hostinfo.Hostname
|
|
|
|
lb.mu.Unlock()
|
|
|
|
if hn != "bar" {
|
|
|
|
t.Fatalf("got %q; want %q", hn, "bar")
|
|
|
|
}
|
|
|
|
}
|