TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-08-14 23:17:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

net/tsaddr: add WithoutExitRoutes and IsExitRoute

Browse Source 
Updates #cleanup

Signed-off-by: Kristoffer Dalby <kristoffer@tailscale.com>
This commit is contained in:
Kristoffer Dalby
2024-09-25 11:08:58 +02:00
committed by Kristoffer Dalby
parent c90c9938c8
commit 3dc33a0a5b
2 changed files with 181 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

133
net/tsaddr/tsaddr_test.go
View File

@@ -7,7 +7,10 @@ import (
"net/netip"
"testing"
"github.com/google/go-cmp/cmp"
"github.com/google/go-cmp/cmp/cmpopts"
"tailscale.com/net/netaddr"
"tailscale.com/types/views"
)
func TestInCrostiniRange(t *testing.T) {
@@ -89,3 +92,133 @@ func TestUnmapVia(t *testing.T) {
}
}
}
func TestIsExitNodeRoute(t *testing.T) {
tests := []struct {
pref netip.Prefix
want bool
}{
{
pref: AllIPv4(),
want: true,
},
{
pref: AllIPv6(),
want: true,
},
{
pref: netip.MustParsePrefix("1.1.1.1/0"),
want: false,
},
{
pref: netip.MustParsePrefix("1.1.1.1/1"),
want: false,
},
{
pref: netip.MustParsePrefix("192.168.0.0/24"),
want: false,
},
}
for _, tt := range tests {
if got := IsExitRoute(tt.pref); got != tt.want {
t.Errorf("for %q: got %v, want %v", tt.pref, got, tt.want)
}
}
}
func TestWithoutExitRoutes(t *testing.T) {
tests := []struct {
prefs []netip.Prefix
want []netip.Prefix
}{
{
prefs: []netip.Prefix{AllIPv4(), AllIPv6()},
want: []netip.Prefix{},
},
{
prefs: []netip.Prefix{AllIPv4()},
want: []netip.Prefix{AllIPv4()},
},
{
prefs: []netip.Prefix{AllIPv4(), AllIPv6(), netip.MustParsePrefix("10.0.0.0/10")},
want: []netip.Prefix{netip.MustParsePrefix("10.0.0.0/10")},
},
{
prefs: []netip.Prefix{AllIPv6(), netip.MustParsePrefix("10.0.0.0/10")},
want: []netip.Prefix{AllIPv6(), netip.MustParsePrefix("10.0.0.0/10")},
},
}
for _, tt := range tests {
got := WithoutExitRoutes(views.SliceOf(tt.prefs))
if diff := cmp.Diff(tt.want, got.AsSlice(), cmpopts.EquateEmpty(), cmp.Comparer(func(a, b netip.Prefix) bool { return a == b })); diff != "" {
t.Errorf("unexpected route difference (-want +got):\n%s", diff)
}
}
}
func TestWithoutExitRoute(t *testing.T) {
tests := []struct {
prefs []netip.Prefix
want []netip.Prefix
}{
{
prefs: []netip.Prefix{AllIPv4(), AllIPv6()},
want: []netip.Prefix{},
},
{
prefs: []netip.Prefix{AllIPv4()},
want: []netip.Prefix{},
},
{
prefs: []netip.Prefix{AllIPv4(), AllIPv6(), netip.MustParsePrefix("10.0.0.0/10")},
want: []netip.Prefix{netip.MustParsePrefix("10.0.0.0/10")},
},
{
prefs: []netip.Prefix{AllIPv6(), netip.MustParsePrefix("10.0.0.0/10")},
want: []netip.Prefix{netip.MustParsePrefix("10.0.0.0/10")},
},
}
for _, tt := range tests {
got := WithoutExitRoute(views.SliceOf(tt.prefs))
if diff := cmp.Diff(tt.want, got.AsSlice(), cmpopts.EquateEmpty(), cmp.Comparer(func(a, b netip.Prefix) bool { return a == b })); diff != "" {
t.Errorf("unexpected route difference (-want +got):\n%s", diff)
}
}
}
func TestContainsExitRoute(t *testing.T) {
tests := []struct {
prefs []netip.Prefix
want bool
}{
{
prefs: []netip.Prefix{AllIPv4(), AllIPv6()},
want: true,
},
{
prefs: []netip.Prefix{AllIPv4()},
want: true,
},
{
prefs: []netip.Prefix{AllIPv4(), AllIPv6(), netip.MustParsePrefix("10.0.0.0/10")},
want: true,
},
{
prefs: []netip.Prefix{AllIPv6(), netip.MustParsePrefix("10.0.0.0/10")},
want: true,
},
{
prefs: []netip.Prefix{netip.MustParsePrefix("10.0.0.0/10")},
want: false,
},
}
for _, tt := range tests {
if got := ContainsExitRoute(views.SliceOf(tt.prefs)); got != tt.want {
t.Errorf("for %q: got %v, want %v", tt.prefs, got, tt.want)
}
}
}