mirror of
https://github.com/juanfont/headscale.git
synced 2026-10-27 16:56:49 +09:00
Kristoffer Dalby
4933bb83d2
Every node re-parsed every entry into an IPSet on every policy change; under churn that was most of the batcher's CPU.
1417 lines
33 KiB
Go
1417 lines
33 KiB
Go
package policyutil_test
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"net/netip"
|
||
"testing"
|
||
|
||
"github.com/google/go-cmp/cmp"
|
||
"github.com/juanfont/headscale/hscontrol/policy"
|
||
"github.com/juanfont/headscale/hscontrol/policy/policyutil"
|
||
"github.com/juanfont/headscale/hscontrol/types"
|
||
"github.com/juanfont/headscale/hscontrol/util"
|
||
"github.com/rs/zerolog/log"
|
||
"github.com/stretchr/testify/require"
|
||
"tailscale.com/net/tsaddr"
|
||
"tailscale.com/tailcfg"
|
||
"tailscale.com/util/must"
|
||
)
|
||
|
||
var ap = func(ipStr string) *netip.Addr {
|
||
ip := netip.MustParseAddr(ipStr)
|
||
return &ip
|
||
}
|
||
|
||
var p = func(prefStr string) netip.Prefix {
|
||
ip := netip.MustParsePrefix(prefStr)
|
||
return ip
|
||
}
|
||
|
||
// hsExitNodeDestForTest is the list of destination IP ranges that are allowed when
|
||
// we use headscale "autogroup:internet".
|
||
var hsExitNodeDestForTest = []tailcfg.NetPortRange{
|
||
{IP: "0.0.0.0/5", Ports: tailcfg.PortRangeAny},
|
||
{IP: "8.0.0.0/7", Ports: tailcfg.PortRangeAny},
|
||
{IP: "11.0.0.0/8", Ports: tailcfg.PortRangeAny},
|
||
{IP: "12.0.0.0/6", Ports: tailcfg.PortRangeAny},
|
||
{IP: "16.0.0.0/4", Ports: tailcfg.PortRangeAny},
|
||
{IP: "32.0.0.0/3", Ports: tailcfg.PortRangeAny},
|
||
{IP: "64.0.0.0/3", Ports: tailcfg.PortRangeAny},
|
||
{IP: "96.0.0.0/6", Ports: tailcfg.PortRangeAny},
|
||
{IP: "100.0.0.0/10", Ports: tailcfg.PortRangeAny},
|
||
{IP: "100.128.0.0/9", Ports: tailcfg.PortRangeAny},
|
||
{IP: "101.0.0.0/8", Ports: tailcfg.PortRangeAny},
|
||
{IP: "102.0.0.0/7", Ports: tailcfg.PortRangeAny},
|
||
{IP: "104.0.0.0/5", Ports: tailcfg.PortRangeAny},
|
||
{IP: "112.0.0.0/4", Ports: tailcfg.PortRangeAny},
|
||
{IP: "128.0.0.0/3", Ports: tailcfg.PortRangeAny},
|
||
{IP: "160.0.0.0/5", Ports: tailcfg.PortRangeAny},
|
||
{IP: "168.0.0.0/8", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.0.0.0/9", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.128.0.0/10", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.192.0.0/11", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.224.0.0/12", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.240.0.0/13", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.248.0.0/14", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.252.0.0/15", Ports: tailcfg.PortRangeAny},
|
||
{IP: "169.255.0.0/16", Ports: tailcfg.PortRangeAny},
|
||
{IP: "170.0.0.0/7", Ports: tailcfg.PortRangeAny},
|
||
{IP: "172.0.0.0/12", Ports: tailcfg.PortRangeAny},
|
||
{IP: "172.32.0.0/11", Ports: tailcfg.PortRangeAny},
|
||
{IP: "172.64.0.0/10", Ports: tailcfg.PortRangeAny},
|
||
{IP: "172.128.0.0/9", Ports: tailcfg.PortRangeAny},
|
||
{IP: "173.0.0.0/8", Ports: tailcfg.PortRangeAny},
|
||
{IP: "174.0.0.0/7", Ports: tailcfg.PortRangeAny},
|
||
{IP: "176.0.0.0/4", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.0.0.0/9", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.128.0.0/11", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.160.0.0/13", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.169.0.0/16", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.170.0.0/15", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.172.0.0/14", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.176.0.0/12", Ports: tailcfg.PortRangeAny},
|
||
{IP: "192.192.0.0/10", Ports: tailcfg.PortRangeAny},
|
||
{IP: "193.0.0.0/8", Ports: tailcfg.PortRangeAny},
|
||
{IP: "194.0.0.0/7", Ports: tailcfg.PortRangeAny},
|
||
{IP: "196.0.0.0/6", Ports: tailcfg.PortRangeAny},
|
||
{IP: "200.0.0.0/5", Ports: tailcfg.PortRangeAny},
|
||
{IP: "208.0.0.0/4", Ports: tailcfg.PortRangeAny},
|
||
{IP: "224.0.0.0/3", Ports: tailcfg.PortRangeAny},
|
||
{IP: "2000::/3", Ports: tailcfg.PortRangeAny},
|
||
}
|
||
|
||
func TestTheInternet(t *testing.T) {
|
||
internetSet := util.TheInternet()
|
||
|
||
internetPrefs := internetSet.Prefixes()
|
||
|
||
for i := range internetPrefs {
|
||
if internetPrefs[i].String() != hsExitNodeDestForTest[i].IP {
|
||
t.Errorf(
|
||
"prefix from internet set %q != hsExit list %q",
|
||
internetPrefs[i].String(),
|
||
hsExitNodeDestForTest[i].IP,
|
||
)
|
||
}
|
||
}
|
||
|
||
if len(internetPrefs) != len(hsExitNodeDestForTest) {
|
||
t.Fatalf(
|
||
"expected same length of prefixes, internet: %d, hsExit: %d",
|
||
len(internetPrefs),
|
||
len(hsExitNodeDestForTest),
|
||
)
|
||
}
|
||
}
|
||
|
||
func TestReduceFilterRules(t *testing.T) {
|
||
users := types.Users{
|
||
types.User{ID: 1, Name: "mickael"},
|
||
types.User{ID: 2, Name: "user1"},
|
||
types.User{ID: 3, Name: "user2"},
|
||
types.User{ID: 4, Name: "user100"},
|
||
types.User{ID: 5, Name: "user3"},
|
||
}
|
||
|
||
tests := []struct {
|
||
name string
|
||
node *types.Node
|
||
peers types.Nodes
|
||
pol string
|
||
want []tailcfg.FilterRule
|
||
}{
|
||
{
|
||
name: "host1-can-reach-host2-no-rules",
|
||
pol: `
|
||
{
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"100.64.0.1"
|
||
],
|
||
"dst": [
|
||
"100.64.0.2:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0:ab12:4843:2222:6273:2221"),
|
||
User: new(users[0]),
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0:ab12:4843:2222:6273:2222"),
|
||
User: new(users[0]),
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{},
|
||
},
|
||
{
|
||
name: "1604-subnet-routers-are-preserved",
|
||
pol: `
|
||
{
|
||
"groups": {
|
||
"group:admins": [
|
||
"user1@"
|
||
]
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:admins"
|
||
],
|
||
"dst": [
|
||
"group:admins:*"
|
||
]
|
||
},
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:admins"
|
||
],
|
||
"dst": [
|
||
"10.33.0.0/16:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0::2"),
|
||
User: new(users[1]),
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
// Merged: Both ACL rules combined (same SrcIPs)
|
||
{
|
||
SrcIPs: []string{
|
||
"100.64.0.1-100.64.0.2",
|
||
"fd7a:115c:a1e0::1-fd7a:115c:a1e0::2",
|
||
},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{
|
||
IP: "100.64.0.1",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "fd7a:115c:a1e0::1",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "10.33.0.0/16",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "1786-reducing-breaks-exit-nodes-the-client",
|
||
pol: `
|
||
{
|
||
"groups": {
|
||
"group:team": [
|
||
"user3@",
|
||
"user2@",
|
||
"user1@"
|
||
]
|
||
},
|
||
"hosts": {
|
||
"internal": "100.64.0.100/32"
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"internal:*"
|
||
]
|
||
},
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"autogroup:internet:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0::2"),
|
||
User: new(users[2]),
|
||
},
|
||
// "internal" exit node
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.100"),
|
||
IPv6: ap("fd7a:115c:a1e0::100"),
|
||
User: new(users[3]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: tsaddr.ExitRoutes(),
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{},
|
||
},
|
||
{
|
||
name: "1786-reducing-breaks-exit-nodes-the-exit",
|
||
pol: `
|
||
{
|
||
"groups": {
|
||
"group:team": [
|
||
"user3@",
|
||
"user2@",
|
||
"user1@"
|
||
]
|
||
},
|
||
"hosts": {
|
||
"internal": "100.64.0.100/32"
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"internal:*"
|
||
]
|
||
},
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"autogroup:internet:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.100"),
|
||
IPv6: ap("fd7a:115c:a1e0::100"),
|
||
User: new(users[3]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: tsaddr.ExitRoutes(),
|
||
},
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0::2"),
|
||
User: new(users[2]),
|
||
},
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
// Only the internal:* rule generates filters.
|
||
// autogroup:internet does NOT generate packet filters - it's handled
|
||
// by exit node routing via AllowedIPs, not by packet filtering.
|
||
{
|
||
SrcIPs: []string{
|
||
"100.64.0.1-100.64.0.2",
|
||
"fd7a:115c:a1e0::1-fd7a:115c:a1e0::2",
|
||
},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{
|
||
IP: "100.64.0.100",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "1786-reducing-breaks-exit-nodes-the-example-from-issue",
|
||
pol: `
|
||
{
|
||
"groups": {
|
||
"group:team": [
|
||
"user3@",
|
||
"user2@",
|
||
"user1@"
|
||
]
|
||
},
|
||
"hosts": {
|
||
"internal": "100.64.0.100/32"
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"internal:*"
|
||
]
|
||
},
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"0.0.0.0/5:*",
|
||
"8.0.0.0/7:*",
|
||
"11.0.0.0/8:*",
|
||
"12.0.0.0/6:*",
|
||
"16.0.0.0/4:*",
|
||
"32.0.0.0/3:*",
|
||
"64.0.0.0/2:*",
|
||
"128.0.0.0/3:*",
|
||
"160.0.0.0/5:*",
|
||
"168.0.0.0/6:*",
|
||
"172.0.0.0/12:*",
|
||
"172.32.0.0/11:*",
|
||
"172.64.0.0/10:*",
|
||
"172.128.0.0/9:*",
|
||
"173.0.0.0/8:*",
|
||
"174.0.0.0/7:*",
|
||
"176.0.0.0/4:*",
|
||
"192.0.0.0/9:*",
|
||
"192.128.0.0/11:*",
|
||
"192.160.0.0/13:*",
|
||
"192.169.0.0/16:*",
|
||
"192.170.0.0/15:*",
|
||
"192.172.0.0/14:*",
|
||
"192.176.0.0/12:*",
|
||
"192.192.0.0/10:*",
|
||
"193.0.0.0/8:*",
|
||
"194.0.0.0/7:*",
|
||
"196.0.0.0/6:*",
|
||
"200.0.0.0/5:*",
|
||
"208.0.0.0/4:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.100"),
|
||
IPv6: ap("fd7a:115c:a1e0::100"),
|
||
User: new(users[3]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: tsaddr.ExitRoutes(),
|
||
},
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0::2"),
|
||
User: new(users[2]),
|
||
},
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
// Exit routes (0.0.0.0/0, ::/0) are skipped when checking RoutableIPs
|
||
// overlap, matching Tailscale SaaS behavior. Only destinations that
|
||
// contain the node's own Tailscale IP (via InIPSet) are kept.
|
||
// Here, 64.0.0.0/2 contains 100.64.0.100 (CGNAT range), so it matches.
|
||
{
|
||
SrcIPs: []string{
|
||
"100.64.0.1-100.64.0.2",
|
||
"fd7a:115c:a1e0::1-fd7a:115c:a1e0::2",
|
||
},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{
|
||
IP: "100.64.0.100",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{IP: "64.0.0.0/2", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "1786-reducing-breaks-exit-nodes-app-connector-like",
|
||
pol: `
|
||
{
|
||
"groups": {
|
||
"group:team": [
|
||
"user3@",
|
||
"user2@",
|
||
"user1@"
|
||
]
|
||
},
|
||
"hosts": {
|
||
"internal": "100.64.0.100/32"
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"internal:*"
|
||
]
|
||
},
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"8.0.0.0/8:*",
|
||
"16.0.0.0/8:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.100"),
|
||
IPv6: ap("fd7a:115c:a1e0::100"),
|
||
User: new(users[3]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{netip.MustParsePrefix("8.0.0.0/16"), netip.MustParsePrefix("16.0.0.0/16")},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{netip.MustParsePrefix("8.0.0.0/16"), netip.MustParsePrefix("16.0.0.0/16")},
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0::2"),
|
||
User: new(users[2]),
|
||
},
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
// Merged: Both ACL rules combined (same SrcIPs)
|
||
{
|
||
SrcIPs: []string{
|
||
"100.64.0.1-100.64.0.2",
|
||
"fd7a:115c:a1e0::1-fd7a:115c:a1e0::2",
|
||
},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{
|
||
IP: "100.64.0.100",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "8.0.0.0/8",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "16.0.0.0/8",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "1786-reducing-breaks-exit-nodes-app-connector-like2",
|
||
pol: `
|
||
{
|
||
"groups": {
|
||
"group:team": [
|
||
"user3@",
|
||
"user2@",
|
||
"user1@"
|
||
]
|
||
},
|
||
"hosts": {
|
||
"internal": "100.64.0.100/32"
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"internal:*"
|
||
]
|
||
},
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:team"
|
||
],
|
||
"dst": [
|
||
"8.0.0.0/16:*",
|
||
"16.0.0.0/16:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.100"),
|
||
IPv6: ap("fd7a:115c:a1e0::100"),
|
||
User: new(users[3]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{netip.MustParsePrefix("8.0.0.0/8"), netip.MustParsePrefix("16.0.0.0/8")},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{netip.MustParsePrefix("8.0.0.0/8"), netip.MustParsePrefix("16.0.0.0/8")},
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0::2"),
|
||
User: new(users[2]),
|
||
},
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
// Merged: Both ACL rules combined (same SrcIPs)
|
||
{
|
||
SrcIPs: []string{
|
||
"100.64.0.1-100.64.0.2",
|
||
"fd7a:115c:a1e0::1-fd7a:115c:a1e0::2",
|
||
},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{
|
||
IP: "100.64.0.100",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "8.0.0.0/16",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "16.0.0.0/16",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "1817-reduce-breaks-32-mask",
|
||
pol: `
|
||
{
|
||
"tagOwners": {
|
||
"tag:access-servers": ["user100@"],
|
||
},
|
||
"groups": {
|
||
"group:access": [
|
||
"user1@"
|
||
]
|
||
},
|
||
"hosts": {
|
||
"dns1": "172.16.0.21/32",
|
||
"vlan1": "172.16.0.0/24"
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"proto": "",
|
||
"src": [
|
||
"group:access"
|
||
],
|
||
"dst": [
|
||
"tag:access-servers:*",
|
||
"dns1:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.100"),
|
||
IPv6: ap("fd7a:115c:a1e0::100"),
|
||
User: new(users[3]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{netip.MustParsePrefix("172.16.0.0/24")},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{netip.MustParsePrefix("172.16.0.0/24")},
|
||
Tags: []string{"tag:access-servers"},
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{
|
||
"100.64.0.1",
|
||
"fd7a:115c:a1e0::1",
|
||
},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{
|
||
IP: "100.64.0.100",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "fd7a:115c:a1e0::100",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
{
|
||
IP: "172.16.0.21",
|
||
Ports: tailcfg.PortRangeAny,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "2365-only-route-policy",
|
||
pol: `
|
||
{
|
||
"hosts": {
|
||
"router": "100.64.0.1/32",
|
||
"node": "100.64.0.2/32"
|
||
},
|
||
"acls": [
|
||
{
|
||
"action": "accept",
|
||
"src": [
|
||
"*"
|
||
],
|
||
"dst": [
|
||
"router:8000"
|
||
]
|
||
},
|
||
{
|
||
"action": "accept",
|
||
"src": [
|
||
"node"
|
||
],
|
||
"dst": [
|
||
"172.26.0.0/16:*"
|
||
]
|
||
}
|
||
],
|
||
}
|
||
`,
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.2"),
|
||
IPv6: ap("fd7a:115c:a1e0::2"),
|
||
User: new(users[3]),
|
||
},
|
||
peers: types.Nodes{
|
||
&types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
User: new(users[1]),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{p("172.16.0.0/24"), p("10.10.11.0/24"), p("10.10.12.0/24")},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{p("172.16.0.0/24"), p("10.10.11.0/24"), p("10.10.12.0/24")},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{},
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
for idx, pmf := range policy.PolicyManagerFuncsForTest(
|
||
[]byte(tt.pol),
|
||
) {
|
||
t.Run(fmt.Sprintf("%s-index%d", tt.name, idx), func(t *testing.T) {
|
||
var (
|
||
pm policy.PolicyManager
|
||
err error
|
||
)
|
||
|
||
pm, err = pmf(users, append(tt.peers, tt.node).ViewSlice())
|
||
require.NoError(t, err)
|
||
|
||
got, _ := pm.Filter()
|
||
t.Logf("full filter:\n%s", must.Get(json.MarshalIndent(got, "", " ")))
|
||
got = policyutil.ReduceFilterRules(tt.node.View(), got)
|
||
|
||
if diff := cmp.Diff(tt.want, got); diff != "" {
|
||
log.Trace().Interface("got", got).Msg("result")
|
||
t.Errorf("TestReduceFilterRules() unexpected result (-want +got):\n%s", diff)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
}
|
||
|
||
// TestReduceFilterRulesPartialApproval verifies that ReduceFilterRules
|
||
// only preserves filter rules for routes that are both advertised
|
||
// (RoutableIPs) AND approved (ApprovedRoutes), matching Tailscale
|
||
// SaaS behavior. Advertised-but-unapproved routes do not cause rule
|
||
// preservation: SaaS never generates filter rules for unapproved
|
||
// routes, and headscale consults node.SubnetRoutes() (which filters
|
||
// by approval) rather than Hostinfo.RoutableIPs() (which does not).
|
||
func TestReduceFilterRulesPartialApproval(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
node *types.Node
|
||
rules []tailcfg.FilterRule
|
||
wantCount int
|
||
wantRoutes []string
|
||
}{
|
||
{
|
||
name: "approved-route-included",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{IP: "10.33.0.0/16", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
},
|
||
wantCount: 1,
|
||
wantRoutes: []string{"10.33.0.0/16"},
|
||
},
|
||
{
|
||
name: "unapproved-route-excluded",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
// Advertised but NOT approved:
|
||
netip.MustParsePrefix("172.16.0.0/24"),
|
||
},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{
|
||
// Only 10.33.0.0/16 approved
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
// Targets the unapproved route
|
||
{IP: "172.16.0.0/24", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
},
|
||
// SubnetRoutes() does NOT contain 172.16.0.0/24
|
||
// (only approved routes), and the ACL dst does not
|
||
// overlap the node's own IPs, so the rule is
|
||
// dropped. This matches Tailscale SaaS behavior.
|
||
wantCount: 0,
|
||
},
|
||
{
|
||
name: "neither-advertised-nor-approved-excluded",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
// Not advertised, not approved
|
||
{IP: "192.168.0.0/16", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
},
|
||
wantCount: 0,
|
||
},
|
||
{
|
||
// Tailscale SaaS delivers every rule to an approved exit
|
||
// node: its exit routes contain every destination.
|
||
name: "approved-exit-route-includes-tailnet-host",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{RoutableIPs: tsaddr.ExitRoutes()},
|
||
|
||
ApprovedRoutes: tsaddr.ExitRoutes(),
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"100.64.0.17"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{IP: "100.64.0.16", Ports: tailcfg.PortRange{First: 53, Last: 53}},
|
||
{IP: "fd7a:115c:a1e0::10", Ports: tailcfg.PortRange{First: 53, Last: 53}},
|
||
},
|
||
},
|
||
},
|
||
wantCount: 1,
|
||
wantRoutes: []string{"100.64.0.16", "fd7a:115c:a1e0::10"},
|
||
},
|
||
{
|
||
name: "approved-exit-route-includes-private-subnet",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{RoutableIPs: tsaddr.ExitRoutes()},
|
||
|
||
ApprovedRoutes: tsaddr.ExitRoutes(),
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"100.64.0.17"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{IP: "10.33.0.0/16", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
},
|
||
wantCount: 1,
|
||
wantRoutes: []string{"10.33.0.0/16"},
|
||
},
|
||
{
|
||
name: "unapproved-exit-route-excluded",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{RoutableIPs: tsaddr.ExitRoutes()},
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"100.64.0.17"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{IP: "100.64.0.16", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
},
|
||
wantCount: 0,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
got := policyutil.ReduceFilterRules(
|
||
tt.node.View(), tt.rules,
|
||
)
|
||
require.Len(t, got, tt.wantCount,
|
||
"rule count mismatch")
|
||
|
||
if tt.wantCount > 0 {
|
||
var gotRoutes []string
|
||
for _, dp := range got[0].DstPorts {
|
||
gotRoutes = append(gotRoutes, dp.IP)
|
||
}
|
||
|
||
require.Equal(t, tt.wantRoutes, gotRoutes)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestReduceFilterRulesCapGrant tests the CapGrant branch of
|
||
// ReduceFilterRules, which was previously untested. All existing
|
||
// test cases use ACL-only policies with DstPorts rules.
|
||
func TestReduceFilterRulesCapGrant(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
node *types.Node
|
||
rules []tailcfg.FilterRule
|
||
want []tailcfg.FilterRule
|
||
}{
|
||
{
|
||
name: "capgrant-matches-node-ip-narrowed",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
// Broad IPv4 prefix containing node IP
|
||
netip.MustParsePrefix("100.64.0.0/10"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
// Only IPv4 narrowed (IPv6 not in /10)
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "capgrant-no-match-filtered-out",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
// Different IP — doesn't match this node
|
||
netip.MustParsePrefix("100.64.0.99/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{},
|
||
},
|
||
{
|
||
name: "capgrant-with-subnet-route-overlap",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{
|
||
netip.MustParsePrefix("10.33.0.0/16"),
|
||
},
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
// Subnet route overlap
|
||
netip.MustParsePrefix("10.0.0.0/8"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/relay-target": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
// 10.0.0.0/8 doesn't contain node's
|
||
// CGNAT IP (100.64.0.1), so no IP
|
||
// narrowing. Only route overlap kept
|
||
// as original prefix.
|
||
netip.MustParsePrefix("10.0.0.0/8"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/relay-target": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "capgrant-exit-route-skipped",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: tsaddr.ExitRoutes(),
|
||
},
|
||
ApprovedRoutes: tsaddr.ExitRoutes(),
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
// 0.0.0.0/0 overlaps the exit route
|
||
// but exit routes should be skipped
|
||
netip.MustParsePrefix("0.0.0.0/0"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
// Node IP narrowed (0.0.0.0/0
|
||
// contains the node's IP)
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "mixed-dstports-and-capgrant-rules",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
// DstPorts rule that matches
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{IP: "100.64.0.1", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
{
|
||
// CapGrant rule that matches
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
// CapGrant rule that doesn't match
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.99/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
DstPorts: []tailcfg.NetPortRange{
|
||
{IP: "100.64.0.1", Ports: tailcfg.PortRangeAny},
|
||
},
|
||
},
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
// Third rule filtered out — doesn't match node
|
||
},
|
||
},
|
||
{
|
||
name: "capgrant-ipv4-only-node",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
// IPv6 is nil, single-IP node
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
name: "capgrant-zero-ip-node-no-panic",
|
||
node: &types.Node{
|
||
// Both IPv4 and IPv6 are nil, zero IPs.
|
||
// Must not panic.
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"10.0.0.0/8"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/drive-sharer": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{},
|
||
},
|
||
{
|
||
// A broad prefix that both contains the node IP (/32 narrow)
|
||
// and overlaps one of its approved subnet routes (route
|
||
// preserved) would otherwise be emitted twice.
|
||
name: "capgrant-ip-and-route-overlap-dedup",
|
||
node: &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.0/24"),
|
||
},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.0/24"),
|
||
},
|
||
},
|
||
rules: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.0/24"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/relay-target": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
want: []tailcfg.FilterRule{
|
||
{
|
||
SrcIPs: []string{"*"},
|
||
CapGrant: []tailcfg.CapGrant{
|
||
{
|
||
Dsts: []netip.Prefix{
|
||
netip.MustParsePrefix("100.64.0.1/32"),
|
||
netip.MustParsePrefix("100.64.0.0/24"),
|
||
},
|
||
CapMap: tailcfg.PeerCapMap{
|
||
"tailscale.com/cap/relay-target": nil,
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
got := policyutil.ReduceFilterRules(tt.node.View(), tt.rules)
|
||
|
||
require.Len(t, got, len(tt.want),
|
||
"rule count mismatch")
|
||
|
||
for i := range tt.want {
|
||
require.Equal(t, tt.want[i].SrcIPs, got[i].SrcIPs,
|
||
"rule[%d] SrcIPs", i)
|
||
require.Len(t, got[i].DstPorts, len(tt.want[i].DstPorts),
|
||
"rule[%d] DstPorts count", i)
|
||
require.Len(t, got[i].CapGrant, len(tt.want[i].CapGrant),
|
||
"rule[%d] CapGrant count", i)
|
||
|
||
for j := range tt.want[i].CapGrant {
|
||
require.ElementsMatch(t,
|
||
tt.want[i].CapGrant[j].Dsts,
|
||
got[i].CapGrant[j].Dsts,
|
||
"rule[%d].CapGrant[%d] Dsts", i, j,
|
||
)
|
||
}
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestReduceFilterRulesDestForms reduces one rule holding every DstPorts form
|
||
// for a node at 100.64.0.1 routing 10.0.0.0/24, and pins which entries reach
|
||
// it. Malformed entries fail closed.
|
||
func TestReduceFilterRulesDestForms(t *testing.T) {
|
||
node := &types.Node{
|
||
IPv4: ap("100.64.0.1"),
|
||
IPv6: ap("fd7a:115c:a1e0::1"),
|
||
Hostinfo: &tailcfg.Hostinfo{
|
||
RoutableIPs: []netip.Prefix{p("10.0.0.0/24")},
|
||
},
|
||
ApprovedRoutes: []netip.Prefix{p("10.0.0.0/24")},
|
||
}
|
||
|
||
dsts := map[string]bool{
|
||
"*": true,
|
||
"100.64.0.1": true,
|
||
"fd7a:115c:a1e0::1": true,
|
||
"100.64.0.2": false,
|
||
"100.64.0.0/10": true,
|
||
"100.128.0.0/10": false,
|
||
"10.0.0.128/25": true,
|
||
"10.0.1.0/24": false,
|
||
"100.64.0.0-100.64.0.9": true,
|
||
"100.64.0.2-100.64.0.9": false,
|
||
"10.0.0.200-10.0.1.5": true,
|
||
"100.64.0.1/10": false, // host bits set
|
||
"100.64.0.9-100.64.0.1": false, // reversed range
|
||
"100.64.0.1-fd7a:115c::1": false, // mixed families
|
||
"not-an-ip": false,
|
||
"100.64.0.1-100.64.0.2-100": false,
|
||
}
|
||
|
||
rule := tailcfg.FilterRule{SrcIPs: []string{"*"}}
|
||
for dst := range dsts {
|
||
rule.DstPorts = append(rule.DstPorts, tailcfg.NetPortRange{IP: dst, Ports: tailcfg.PortRangeAny})
|
||
}
|
||
|
||
got := policyutil.ReduceFilterRules(node.View(), []tailcfg.FilterRule{rule})
|
||
require.Len(t, got, 1)
|
||
|
||
kept := make(map[string]bool)
|
||
for _, dp := range got[0].DstPorts {
|
||
kept[dp.IP] = true
|
||
}
|
||
|
||
for dst, want := range dsts {
|
||
require.Equal(t, want, kept[dst], "DstPorts %q", dst)
|
||
}
|
||
}
|
||
|
||
// TestReduceFilterRulesAllocsIndependentOfDests reduces the global filter as
|
||
// every node does on every policy change. Its DstPorts are plain strings, so
|
||
// anything allocated per entry, such as building an IPSet to test one
|
||
// address, is paid nodes × entries times per change.
|
||
func TestReduceFilterRulesAllocsIndependentOfDests(t *testing.T) {
|
||
node := (&types.Node{IPv4: ap("100.64.0.1"), IPv6: ap("fd7a:115c:a1e0::1")}).View()
|
||
|
||
allocs := func(n int) int {
|
||
rule := tailcfg.FilterRule{SrcIPs: []string{"*"}}
|
||
for i := range n {
|
||
rule.DstPorts = append(rule.DstPorts, tailcfg.NetPortRange{
|
||
IP: fmt.Sprintf("100.64.%d.%d", i/250, i%250+2),
|
||
Ports: tailcfg.PortRangeAny,
|
||
})
|
||
}
|
||
|
||
rules := []tailcfg.FilterRule{rule}
|
||
|
||
return int(testing.AllocsPerRun(10, func() { policyutil.ReduceFilterRules(node, rules) }))
|
||
}
|
||
|
||
require.Equal(t, allocs(10), allocs(1000),
|
||
"reducing a filter must not allocate per DstPorts entry")
|
||
}
|