2023-05-21 16:37:59 +00:00
|
|
|
package policy
|
2021-07-03 09:55:32 +00:00
|
|
|
|
|
|
|
import (
|
2022-02-05 16:18:39 +00:00
|
|
|
"errors"
|
2022-09-01 22:05:43 +00:00
|
|
|
"net/netip"
|
2022-02-07 15:12:05 +00:00
|
|
|
"testing"
|
2022-02-05 16:18:39 +00:00
|
|
|
|
2023-04-26 12:04:12 +00:00
|
|
|
"github.com/google/go-cmp/cmp"
|
2023-05-21 16:37:59 +00:00
|
|
|
"github.com/juanfont/headscale/hscontrol/types"
|
|
|
|
"github.com/juanfont/headscale/hscontrol/util"
|
2023-04-26 12:04:12 +00:00
|
|
|
"github.com/rs/zerolog/log"
|
2023-06-12 13:29:34 +00:00
|
|
|
"github.com/spf13/viper"
|
2023-06-08 17:50:59 +00:00
|
|
|
"github.com/stretchr/testify/assert"
|
2023-04-28 14:11:02 +00:00
|
|
|
"go4.org/netipx"
|
2021-07-03 09:55:32 +00:00
|
|
|
"gopkg.in/check.v1"
|
2022-02-07 15:12:05 +00:00
|
|
|
"tailscale.com/tailcfg"
|
2021-07-03 09:55:32 +00:00
|
|
|
)
|
|
|
|
|
2023-05-21 16:37:59 +00:00
|
|
|
func Test(t *testing.T) {
|
|
|
|
check.TestingT(t)
|
|
|
|
}
|
|
|
|
|
|
|
|
var _ = check.Suite(&Suite{})
|
|
|
|
|
|
|
|
type Suite struct{}
|
|
|
|
|
2021-07-03 09:55:32 +00:00
|
|
|
func (s *Suite) TestWrongPath(c *check.C) {
|
2023-05-21 16:37:59 +00:00
|
|
|
_, err := LoadACLPolicyFromPath("asdfg")
|
2021-07-03 09:55:32 +00:00
|
|
|
c.Assert(err, check.NotNil)
|
|
|
|
}
|
|
|
|
|
2023-06-13 08:03:22 +00:00
|
|
|
func TestParsing(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
format string
|
|
|
|
acl string
|
|
|
|
want []tailcfg.FilterRule
|
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "invalid-hujson",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
2023-05-10 08:19:16 +00:00
|
|
|
{
|
2023-06-13 08:03:22 +00:00
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{},
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "valid-hujson-invalid-content",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
|
|
|
{
|
|
|
|
"valid_json": true,
|
|
|
|
"but_a_policy_though": false
|
2021-07-03 09:55:32 +00:00
|
|
|
}
|
2023-06-13 08:03:22 +00:00
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{},
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "invalid-cidr",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
|
|
|
{"example-host-1": "100.100.100.100/42"}
|
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{},
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "basic-rule",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
|
|
|
{
|
|
|
|
"hosts": {
|
|
|
|
"host-1": "100.100.100.100",
|
|
|
|
"subnet-1": "100.100.101.100/24",
|
|
|
|
},
|
2021-07-03 09:55:32 +00:00
|
|
|
|
2023-06-13 08:03:22 +00:00
|
|
|
"acls": [
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"subnet-1",
|
|
|
|
"192.168.1.0/24"
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"*:22,3389",
|
|
|
|
"host-1:*",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.100.101.0/24", "192.168.1.0/24"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "0.0.0.0/0", Ports: tailcfg.PortRange{First: 22, Last: 22}},
|
|
|
|
{IP: "0.0.0.0/0", Ports: tailcfg.PortRange{First: 3389, Last: 3389}},
|
|
|
|
{IP: "::/0", Ports: tailcfg.PortRange{First: 22, Last: 22}},
|
|
|
|
{IP: "::/0", Ports: tailcfg.PortRange{First: 3389, Last: 3389}},
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "parse-protocol",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
|
|
|
{
|
|
|
|
"hosts": {
|
|
|
|
"host-1": "100.100.100.100",
|
|
|
|
"subnet-1": "100.100.101.100/24",
|
|
|
|
},
|
|
|
|
|
|
|
|
"acls": [
|
|
|
|
{
|
|
|
|
"Action": "accept",
|
|
|
|
"src": [
|
|
|
|
"*",
|
|
|
|
],
|
|
|
|
"proto": "tcp",
|
|
|
|
"dst": [
|
|
|
|
"host-1:*",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"Action": "accept",
|
|
|
|
"src": [
|
|
|
|
"*",
|
|
|
|
],
|
|
|
|
"proto": "udp",
|
|
|
|
"dst": [
|
|
|
|
"host-1:53",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"Action": "accept",
|
|
|
|
"src": [
|
|
|
|
"*",
|
|
|
|
],
|
|
|
|
"proto": "icmp",
|
|
|
|
"dst": [
|
|
|
|
"host-1:*",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
IPProto: []int{protocolTCP},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRange{First: 53, Last: 53}},
|
|
|
|
},
|
|
|
|
IPProto: []int{protocolUDP},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
IPProto: []int{protocolICMP, protocolIPv6ICMP},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "port-wildcard",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
2023-05-10 08:19:16 +00:00
|
|
|
{
|
2023-06-13 08:03:22 +00:00
|
|
|
"hosts": {
|
|
|
|
"host-1": "100.100.100.100",
|
|
|
|
"subnet-1": "100.100.101.100/24",
|
|
|
|
},
|
|
|
|
|
|
|
|
"acls": [
|
|
|
|
{
|
|
|
|
"Action": "accept",
|
|
|
|
"src": [
|
|
|
|
"*",
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"host-1:*",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
2023-05-10 08:19:16 +00:00
|
|
|
}
|
2023-06-13 08:03:22 +00:00
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "port-range",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
|
|
|
{
|
|
|
|
"hosts": {
|
|
|
|
"host-1": "100.100.100.100",
|
|
|
|
"subnet-1": "100.100.101.100/24",
|
|
|
|
},
|
|
|
|
|
|
|
|
"acls": [
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"subnet-1",
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"host-1:5400-5500",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
2021-07-03 09:55:32 +00:00
|
|
|
}
|
2023-06-13 08:03:22 +00:00
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.100.101.0/24"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{
|
|
|
|
IP: "100.100.100.100/32",
|
|
|
|
Ports: tailcfg.PortRange{First: 5400, Last: 5500},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "port-group",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
|
|
|
{
|
|
|
|
"groups": {
|
|
|
|
"group:example": [
|
|
|
|
"testuser",
|
|
|
|
],
|
|
|
|
},
|
2021-07-03 09:55:32 +00:00
|
|
|
|
2023-06-13 08:03:22 +00:00
|
|
|
"hosts": {
|
|
|
|
"host-1": "100.100.100.100",
|
|
|
|
"subnet-1": "100.100.101.100/24",
|
|
|
|
},
|
|
|
|
|
|
|
|
"acls": [
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"group:example",
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"host-1:*",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
2021-07-03 15:31:32 +00:00
|
|
|
}
|
2023-06-13 08:03:22 +00:00
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"200.200.200.200/32"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "port-user",
|
|
|
|
format: "hujson",
|
|
|
|
acl: `
|
|
|
|
{
|
|
|
|
"hosts": {
|
|
|
|
"host-1": "100.100.100.100",
|
|
|
|
"subnet-1": "100.100.101.100/24",
|
|
|
|
},
|
2021-07-03 15:31:32 +00:00
|
|
|
|
2023-06-13 08:03:22 +00:00
|
|
|
"acls": [
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"testuser",
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"host-1:*",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"200.200.200.200/32"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "port-wildcard-yaml",
|
|
|
|
format: "yaml",
|
|
|
|
acl: `
|
|
|
|
---
|
|
|
|
hosts:
|
|
|
|
host-1: 100.100.100.100/32
|
|
|
|
subnet-1: 100.100.101.100/24
|
|
|
|
acls:
|
|
|
|
- action: accept
|
|
|
|
src:
|
|
|
|
- "*"
|
|
|
|
dst:
|
|
|
|
- host-1:*
|
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "ipv6-yaml",
|
|
|
|
format: "yaml",
|
|
|
|
acl: `
|
|
|
|
---
|
|
|
|
hosts:
|
|
|
|
host-1: 100.100.100.100/32
|
|
|
|
subnet-1: 100.100.101.100/24
|
|
|
|
acls:
|
|
|
|
- action: accept
|
|
|
|
src:
|
|
|
|
- "*"
|
|
|
|
dst:
|
|
|
|
- host-1:*
|
|
|
|
`,
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.100.100.100/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
pol, err := LoadACLPolicyFromBytes([]byte(tt.acl), tt.format)
|
|
|
|
|
|
|
|
if tt.wantErr && err == nil {
|
|
|
|
t.Errorf("parsing() error = %v, wantErr %v", err, tt.wantErr)
|
|
|
|
|
|
|
|
return
|
|
|
|
} else if !tt.wantErr && err != nil {
|
|
|
|
t.Errorf("parsing() error = %v, wantErr %v", err, tt.wantErr)
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
rules, err := pol.generateFilterRules(&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-06-13 08:03:22 +00:00
|
|
|
netip.MustParseAddr("100.100.100.100"),
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
}, types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-06-13 08:03:22 +00:00
|
|
|
netip.MustParseAddr("200.200.200.200"),
|
|
|
|
},
|
|
|
|
User: types.User{
|
|
|
|
Name: "testuser",
|
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2023-06-13 08:03:22 +00:00
|
|
|
},
|
|
|
|
})
|
|
|
|
|
|
|
|
if (err != nil) != tt.wantErr {
|
|
|
|
t.Errorf("parsing() error = %v, wantErr %v", err, tt.wantErr)
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if diff := cmp.Diff(tt.want, rules); diff != "" {
|
|
|
|
t.Errorf("parsing() unexpected result (-want +got):\n%s", diff)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
2021-07-03 15:31:32 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Suite) TestRuleInvalidGeneration(c *check.C) {
|
2023-05-10 08:19:16 +00:00
|
|
|
acl := []byte(`
|
|
|
|
{
|
2023-05-11 07:09:18 +00:00
|
|
|
// Declare static groups of users beyond those in the identity service.
|
|
|
|
"groups": {
|
|
|
|
"group:example": [
|
|
|
|
"user1@example.com",
|
|
|
|
"user2@example.com",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
// Declare hostname aliases to use in place of IP addresses or subnets.
|
|
|
|
"hosts": {
|
|
|
|
"example-host-1": "100.100.100.100",
|
|
|
|
"example-host-2": "100.100.101.100/24",
|
|
|
|
},
|
|
|
|
// Define who is allowed to use which tags.
|
|
|
|
"tagOwners": {
|
|
|
|
// Everyone in the montreal-admins or global-admins group are
|
|
|
|
// allowed to tag servers as montreal-webserver.
|
|
|
|
"tag:montreal-webserver": [
|
|
|
|
"group:montreal-admins",
|
|
|
|
"group:global-admins",
|
|
|
|
],
|
|
|
|
// Only a few admins are allowed to create API servers.
|
|
|
|
"tag:api-server": [
|
|
|
|
"group:global-admins",
|
|
|
|
"example-host-1",
|
|
|
|
],
|
|
|
|
},
|
|
|
|
// Access control lists.
|
|
|
|
"acls": [
|
|
|
|
// Engineering users, plus the president, can access port 22 (ssh)
|
|
|
|
// and port 3389 (remote desktop protocol) on all servers, and all
|
|
|
|
// ports on git-server or ci-server.
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"group:engineering",
|
|
|
|
"president@example.com"
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"*:22,3389",
|
|
|
|
"git-server:*",
|
|
|
|
"ci-server:*"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
// Allow engineer users to access any port on a device tagged with
|
|
|
|
// tag:production.
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"group:engineers"
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"tag:production:*"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
// Allow servers in the my-subnet host and 192.168.1.0/24 to access hosts
|
|
|
|
// on both networks.
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"my-subnet",
|
|
|
|
"192.168.1.0/24"
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"my-subnet:*",
|
|
|
|
"192.168.1.0/24:*"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
// Allow every user of your network to access anything on the network.
|
|
|
|
// Comment out this section if you want to define specific ACL
|
|
|
|
// restrictions above.
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"*"
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"*:*"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
// All users in Montreal are allowed to access the Montreal web
|
|
|
|
// servers.
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"group:montreal-users"
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"tag:montreal-webserver:80,443"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
// Montreal web servers are allowed to make outgoing connections to
|
|
|
|
// the API servers, but only on https port 443.
|
|
|
|
// In contrast, this doesn't grant API servers the right to initiate
|
|
|
|
// any connections.
|
|
|
|
{
|
|
|
|
"action": "accept",
|
|
|
|
"src": [
|
|
|
|
"tag:montreal-webserver"
|
|
|
|
],
|
|
|
|
"dst": [
|
|
|
|
"tag:api-server:443"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
|
|
|
// Declare tests to check functionality of ACL rules
|
|
|
|
"tests": [
|
|
|
|
{
|
|
|
|
"src": "user1@example.com",
|
|
|
|
"accept": [
|
|
|
|
"example-host-1:22",
|
|
|
|
"example-host-2:80"
|
|
|
|
],
|
|
|
|
"deny": [
|
|
|
|
"exapmle-host-2:100"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"src": "user2@example.com",
|
|
|
|
"accept": [
|
|
|
|
"100.60.3.4:22"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
2023-05-10 08:19:16 +00:00
|
|
|
}
|
|
|
|
`)
|
2023-05-21 16:37:59 +00:00
|
|
|
pol, err := LoadACLPolicyFromBytes(acl, "hujson")
|
|
|
|
c.Assert(pol.ACLs, check.HasLen, 6)
|
|
|
|
c.Assert(err, check.IsNil)
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
rules, err := pol.generateFilterRules(&types.Node{}, types.Nodes{})
|
2021-07-03 15:31:32 +00:00
|
|
|
c.Assert(err, check.NotNil)
|
2023-05-21 16:37:59 +00:00
|
|
|
c.Assert(rules, check.IsNil)
|
2021-07-03 15:31:32 +00:00
|
|
|
}
|
|
|
|
|
2022-02-21 20:48:05 +00:00
|
|
|
// TODO(kradalby): Make tests values safe, independent and descriptive.
|
2022-02-05 16:18:39 +00:00
|
|
|
func (s *Suite) TestInvalidAction(c *check.C) {
|
2023-05-21 16:37:59 +00:00
|
|
|
pol := &ACLPolicy{
|
2022-02-05 16:18:39 +00:00
|
|
|
ACLs: []ACL{
|
2022-08-04 08:47:00 +00:00
|
|
|
{
|
|
|
|
Action: "invalidAction",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
2022-02-05 16:18:39 +00:00
|
|
|
},
|
|
|
|
}
|
2023-09-24 11:42:05 +00:00
|
|
|
_, _, err := GenerateFilterAndSSHRules(pol, &types.Node{}, types.Nodes{})
|
2023-05-21 16:37:59 +00:00
|
|
|
c.Assert(errors.Is(err, ErrInvalidAction), check.Equals, true)
|
2022-09-30 18:44:23 +00:00
|
|
|
}
|
|
|
|
|
2022-02-05 16:18:39 +00:00
|
|
|
func (s *Suite) TestInvalidGroupInGroup(c *check.C) {
|
2022-06-08 11:40:15 +00:00
|
|
|
// this ACL is wrong because the group in Sources sections doesn't exist
|
2023-05-21 16:37:59 +00:00
|
|
|
pol := &ACLPolicy{
|
2022-02-14 14:54:51 +00:00
|
|
|
Groups: Groups{
|
|
|
|
"group:test": []string{"foo"},
|
|
|
|
"group:error": []string{"foo", "group:test"},
|
|
|
|
},
|
2022-02-05 16:18:39 +00:00
|
|
|
ACLs: []ACL{
|
2022-08-04 08:47:00 +00:00
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"group:error"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
2022-02-05 16:18:39 +00:00
|
|
|
},
|
|
|
|
}
|
2023-09-24 11:42:05 +00:00
|
|
|
_, _, err := GenerateFilterAndSSHRules(pol, &types.Node{}, types.Nodes{})
|
2023-05-21 16:37:59 +00:00
|
|
|
c.Assert(errors.Is(err, ErrInvalidGroup), check.Equals, true)
|
2022-02-05 16:18:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Suite) TestInvalidTagOwners(c *check.C) {
|
|
|
|
// this ACL is wrong because no tagOwners own the requested tag for the server
|
2023-05-21 16:37:59 +00:00
|
|
|
pol := &ACLPolicy{
|
2022-02-05 16:18:39 +00:00
|
|
|
ACLs: []ACL{
|
2022-08-04 08:47:00 +00:00
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"tag:foo"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
2022-02-05 16:18:39 +00:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
_, _, err := GenerateFilterAndSSHRules(pol, &types.Node{}, types.Nodes{})
|
2023-05-21 16:37:59 +00:00
|
|
|
c.Assert(errors.Is(err, ErrInvalidTag), check.Equals, true)
|
2022-02-05 16:18:39 +00:00
|
|
|
}
|
|
|
|
|
2022-02-07 15:12:05 +00:00
|
|
|
func Test_expandGroup(t *testing.T) {
|
2023-04-26 08:58:26 +00:00
|
|
|
type field struct {
|
|
|
|
pol ACLPolicy
|
|
|
|
}
|
2022-02-07 15:12:05 +00:00
|
|
|
type args struct {
|
2023-06-12 13:29:34 +00:00
|
|
|
group string
|
|
|
|
stripEmail bool
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
2023-04-26 08:58:26 +00:00
|
|
|
field field
|
2022-02-07 15:12:05 +00:00
|
|
|
args args
|
|
|
|
want []string
|
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "simple test",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
2022-02-14 14:54:51 +00:00
|
|
|
Groups: Groups{
|
2022-02-20 22:00:31 +00:00
|
|
|
"group:test": []string{"user1", "user2", "user3"},
|
|
|
|
"group:foo": []string{"user2", "user3"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-04-26 08:58:26 +00:00
|
|
|
},
|
|
|
|
args: args{
|
2023-06-12 13:29:34 +00:00
|
|
|
group: "group:test",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2022-02-20 22:00:31 +00:00
|
|
|
want: []string{"user1", "user2", "user3"},
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "InexistantGroup",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
2022-02-14 14:54:51 +00:00
|
|
|
Groups: Groups{
|
2022-02-20 22:00:31 +00:00
|
|
|
"group:test": []string{"user1", "user2", "user3"},
|
|
|
|
"group:foo": []string{"user2", "user3"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-04-26 08:58:26 +00:00
|
|
|
},
|
|
|
|
args: args{
|
2023-06-12 13:29:34 +00:00
|
|
|
group: "group:undefined",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
want: []string{},
|
|
|
|
wantErr: true,
|
|
|
|
},
|
2022-03-01 20:01:46 +00:00
|
|
|
{
|
2023-06-12 13:29:34 +00:00
|
|
|
name: "Expand emails in group strip domains",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
2022-03-01 20:01:46 +00:00
|
|
|
Groups: Groups{
|
|
|
|
"group:admin": []string{
|
|
|
|
"joe.bar@gmail.com",
|
|
|
|
"john.doe@yahoo.fr",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-04-26 08:58:26 +00:00
|
|
|
},
|
|
|
|
args: args{
|
2023-06-12 13:29:34 +00:00
|
|
|
group: "group:admin",
|
|
|
|
stripEmail: true,
|
2022-03-01 20:01:46 +00:00
|
|
|
},
|
|
|
|
want: []string{"joe.bar", "john.doe"},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "Expand emails in group",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
2022-03-01 20:01:46 +00:00
|
|
|
Groups: Groups{
|
|
|
|
"group:admin": []string{
|
|
|
|
"joe.bar@gmail.com",
|
|
|
|
"john.doe@yahoo.fr",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-04-26 08:58:26 +00:00
|
|
|
},
|
|
|
|
args: args{
|
2023-06-12 13:29:34 +00:00
|
|
|
group: "group:admin",
|
2022-03-01 20:01:46 +00:00
|
|
|
},
|
|
|
|
want: []string{"joe.bar.gmail.com", "john.doe.yahoo.fr"},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
2022-02-14 14:26:54 +00:00
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
2023-06-12 13:29:34 +00:00
|
|
|
viper.Set("oidc.strip_email_domain", test.args.stripEmail)
|
|
|
|
|
2023-06-19 07:17:50 +00:00
|
|
|
got, err := test.field.pol.expandUsersFromGroup(
|
2022-03-01 20:01:46 +00:00
|
|
|
test.args.group,
|
|
|
|
)
|
2023-06-12 13:29:34 +00:00
|
|
|
|
2022-02-14 14:26:54 +00:00
|
|
|
if (err != nil) != test.wantErr {
|
|
|
|
t.Errorf("expandGroup() error = %v, wantErr %v", err, test.wantErr)
|
|
|
|
|
2022-02-07 15:12:05 +00:00
|
|
|
return
|
|
|
|
}
|
2023-06-12 13:29:34 +00:00
|
|
|
|
|
|
|
if diff := cmp.Diff(test.want, got); diff != "" {
|
|
|
|
t.Errorf("expandGroup() unexpected result (-want +got):\n%s", diff)
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_expandTagOwners(t *testing.T) {
|
|
|
|
type args struct {
|
2023-06-12 13:29:34 +00:00
|
|
|
aclPolicy *ACLPolicy
|
|
|
|
tag string
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
want []string
|
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
2022-02-20 22:00:31 +00:00
|
|
|
name: "simple tag expansion",
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-02-20 22:00:31 +00:00
|
|
|
TagOwners: TagOwners{"tag:test": []string{"user1"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
tag: "tag:test",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2022-02-20 22:00:31 +00:00
|
|
|
want: []string{"user1"},
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
2022-02-20 22:00:31 +00:00
|
|
|
name: "expand with tag and group",
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-02-20 22:00:31 +00:00
|
|
|
Groups: Groups{"group:foo": []string{"user1", "user2"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
TagOwners: TagOwners{"tag:test": []string{"group:foo"}},
|
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
tag: "tag:test",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2022-02-20 22:00:31 +00:00
|
|
|
want: []string{"user1", "user2"},
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
2023-01-17 16:43:44 +00:00
|
|
|
name: "expand with user and group",
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-02-20 22:00:31 +00:00
|
|
|
Groups: Groups{"group:foo": []string{"user1", "user2"}},
|
|
|
|
TagOwners: TagOwners{"tag:test": []string{"group:foo", "user3"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
tag: "tag:test",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2022-02-20 22:00:31 +00:00
|
|
|
want: []string{"user1", "user2", "user3"},
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "invalid tag",
|
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-02-20 22:00:31 +00:00
|
|
|
TagOwners: TagOwners{"tag:foo": []string{"group:foo", "user1"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
tag: "tag:test",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
want: []string{},
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "invalid group",
|
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-02-20 22:00:31 +00:00
|
|
|
Groups: Groups{"group:bar": []string{"user1", "user2"}},
|
|
|
|
TagOwners: TagOwners{"tag:test": []string{"group:foo", "user2"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
tag: "tag:test",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
want: []string{},
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
}
|
2022-02-14 14:26:54 +00:00
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
2023-06-19 07:17:50 +00:00
|
|
|
got, err := expandOwnersFromTag(
|
2022-03-01 20:01:46 +00:00
|
|
|
test.args.aclPolicy,
|
|
|
|
test.args.tag,
|
|
|
|
)
|
2022-02-14 14:26:54 +00:00
|
|
|
if (err != nil) != test.wantErr {
|
|
|
|
t.Errorf("expandTagOwners() error = %v, wantErr %v", err, test.wantErr)
|
|
|
|
|
2022-02-07 15:12:05 +00:00
|
|
|
return
|
|
|
|
}
|
2023-06-21 06:12:24 +00:00
|
|
|
if diff := cmp.Diff(test.want, got); diff != "" {
|
|
|
|
t.Errorf("expandTagOwners() = (-want +got):\n%s", diff)
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_expandPorts(t *testing.T) {
|
|
|
|
type args struct {
|
2022-06-08 15:43:59 +00:00
|
|
|
portsStr string
|
|
|
|
needsWildcard bool
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
want *[]tailcfg.PortRange
|
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "wildcard",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "*", needsWildcard: true},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: &[]tailcfg.PortRange{
|
|
|
|
{First: portRangeBegin, Last: portRangeEnd},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
2022-06-08 15:43:59 +00:00
|
|
|
{
|
|
|
|
name: "needs wildcard but does not require it",
|
|
|
|
args: args{portsStr: "*", needsWildcard: false},
|
|
|
|
want: &[]tailcfg.PortRange{
|
|
|
|
{First: portRangeBegin, Last: portRangeEnd},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "needs wildcard but gets port",
|
|
|
|
args: args{portsStr: "80,443", needsWildcard: true},
|
|
|
|
want: nil,
|
|
|
|
wantErr: true,
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
{
|
2022-06-08 11:40:15 +00:00
|
|
|
name: "two Destinations",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "80,443", needsWildcard: false},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: &[]tailcfg.PortRange{
|
|
|
|
{First: 80, Last: 80},
|
|
|
|
{First: 443, Last: 443},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "a range and a port",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "80-1024,443", needsWildcard: false},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: &[]tailcfg.PortRange{
|
|
|
|
{First: 80, Last: 1024},
|
|
|
|
{First: 443, Last: 443},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "out of bounds",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "854038", needsWildcard: false},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: nil,
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wrong port",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "85a38", needsWildcard: false},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: nil,
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wrong port in first",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "a-80", needsWildcard: false},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: nil,
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wrong port in last",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "80-85a38", needsWildcard: false},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: nil,
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wrong port format",
|
2022-06-08 15:43:59 +00:00
|
|
|
args: args{portsStr: "80-85a38-3", needsWildcard: false},
|
2022-02-07 15:12:05 +00:00
|
|
|
want: nil,
|
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
}
|
2022-02-14 14:26:54 +00:00
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
2022-06-08 15:43:59 +00:00
|
|
|
got, err := expandPorts(test.args.portsStr, test.args.needsWildcard)
|
2022-02-14 14:26:54 +00:00
|
|
|
if (err != nil) != test.wantErr {
|
|
|
|
t.Errorf("expandPorts() error = %v, wantErr %v", err, test.wantErr)
|
|
|
|
|
2022-02-07 15:12:05 +00:00
|
|
|
return
|
|
|
|
}
|
2023-06-21 06:12:24 +00:00
|
|
|
if diff := cmp.Diff(test.want, got); diff != "" {
|
|
|
|
t.Errorf("expandPorts() = (-want +got):\n%s", diff)
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
func Test_listNodesInUser(t *testing.T) {
|
2022-02-07 15:12:05 +00:00
|
|
|
type args struct {
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes types.Nodes
|
|
|
|
user string
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
2023-09-24 11:42:05 +00:00
|
|
|
want types.Nodes
|
2022-02-07 15:12:05 +00:00
|
|
|
}{
|
|
|
|
{
|
2023-09-24 11:42:05 +00:00
|
|
|
name: "1 node in user",
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{User: types.User{Name: "joe"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-01-17 16:43:44 +00:00
|
|
|
user: "joe",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{User: types.User{Name: "joe"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2023-09-24 11:42:05 +00:00
|
|
|
name: "3 nodes, 2 in user",
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{ID: 1, User: types.User{Name: "joe"}},
|
|
|
|
&types.Node{ID: 2, User: types.User{Name: "marc"}},
|
|
|
|
&types.Node{ID: 3, User: types.User{Name: "marc"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-01-17 16:43:44 +00:00
|
|
|
user: "marc",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{ID: 2, User: types.User{Name: "marc"}},
|
|
|
|
&types.Node{ID: 3, User: types.User{Name: "marc"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2023-09-24 11:42:05 +00:00
|
|
|
name: "5 nodes, 0 in user",
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{ID: 1, User: types.User{Name: "joe"}},
|
|
|
|
&types.Node{ID: 2, User: types.User{Name: "marc"}},
|
|
|
|
&types.Node{ID: 3, User: types.User{Name: "marc"}},
|
|
|
|
&types.Node{ID: 4, User: types.User{Name: "marc"}},
|
|
|
|
&types.Node{ID: 5, User: types.User{Name: "marc"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-01-17 16:43:44 +00:00
|
|
|
user: "mickael",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
}
|
2022-02-14 17:24:37 +00:00
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
2023-09-24 11:42:05 +00:00
|
|
|
got := filterNodesByUser(test.args.nodes, test.args.user)
|
2023-06-21 06:12:24 +00:00
|
|
|
|
2023-12-09 17:09:24 +00:00
|
|
|
if diff := cmp.Diff(test.want, got, util.Comparers...); diff != "" {
|
2023-09-24 11:42:05 +00:00
|
|
|
t.Errorf("listNodesInUser() = (-want +got):\n%s", diff)
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_expandAlias(t *testing.T) {
|
2023-04-28 14:11:02 +00:00
|
|
|
set := func(ips []string, prefixes []string) *netipx.IPSet {
|
|
|
|
var builder netipx.IPSetBuilder
|
|
|
|
|
|
|
|
for _, ip := range ips {
|
|
|
|
builder.Add(netip.MustParseAddr(ip))
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, pre := range prefixes {
|
|
|
|
builder.AddPrefix(netip.MustParsePrefix(pre))
|
|
|
|
}
|
|
|
|
|
|
|
|
s, _ := builder.IPSet()
|
|
|
|
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
2023-04-26 08:58:26 +00:00
|
|
|
type field struct {
|
|
|
|
pol ACLPolicy
|
|
|
|
}
|
2022-02-07 15:12:05 +00:00
|
|
|
type args struct {
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes types.Nodes
|
2023-06-12 13:29:34 +00:00
|
|
|
aclPolicy ACLPolicy
|
|
|
|
alias string
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
2023-04-26 08:58:26 +00:00
|
|
|
field field
|
2022-02-07 15:12:05 +00:00
|
|
|
args args
|
2023-04-28 14:11:02 +00:00
|
|
|
want *netipx.IPSet
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "wildcard",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
|
|
|
alias: "*",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-07-26 09:53:42 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.78.84.227"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{}, []string{
|
|
|
|
"0.0.0.0/0",
|
|
|
|
"::/0",
|
|
|
|
}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "simple group",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
Groups: Groups{"group:accountant": []string{"joe", "marc"}},
|
|
|
|
},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2022-02-20 22:00:31 +00:00
|
|
|
alias: "group:accountant",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{
|
|
|
|
"100.64.0.1", "100.64.0.2", "100.64.0.3",
|
|
|
|
}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "wrong group",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
Groups: Groups{"group:accountant": []string{"joe", "marc"}},
|
|
|
|
},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2022-02-20 22:00:31 +00:00
|
|
|
alias: "group:hr",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "simple ipaddress",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
alias: "10.0.0.3",
|
|
|
|
nodes: types.Nodes{},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{
|
|
|
|
"10.0.0.3",
|
|
|
|
}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
2023-04-16 10:26:35 +00:00
|
|
|
name: "simple host by ip passed through",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
alias: "10.0.0.1",
|
|
|
|
nodes: types.Nodes{},
|
2023-04-16 10:26:35 +00:00
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{
|
|
|
|
"10.0.0.1",
|
|
|
|
}, []string{}),
|
2023-04-16 10:26:35 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "simple host by ipv4 single ipv4",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2023-04-16 10:26:35 +00:00
|
|
|
args: args{
|
|
|
|
alias: "10.0.0.1",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-04-16 10:26:35 +00:00
|
|
|
netip.MustParseAddr("10.0.0.1"),
|
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{
|
|
|
|
"10.0.0.1",
|
|
|
|
}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
2023-04-16 10:26:35 +00:00
|
|
|
name: "simple host by ipv4 single dual stack",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-04-16 10:26:35 +00:00
|
|
|
alias: "10.0.0.1",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-04-16 10:26:35 +00:00
|
|
|
netip.MustParseAddr("10.0.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0:ab12:4843:2222:6273:2222"),
|
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2023-04-16 10:26:35 +00:00
|
|
|
},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{
|
|
|
|
"10.0.0.1", "fd7a:115c:a1e0:ab12:4843:2222:6273:2222",
|
|
|
|
}, []string{}),
|
2023-04-16 10:26:35 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "simple host by ipv6 single dual stack",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2023-04-16 10:26:35 +00:00
|
|
|
args: args{
|
|
|
|
alias: "fd7a:115c:a1e0:ab12:4843:2222:6273:2222",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-04-16 10:26:35 +00:00
|
|
|
netip.MustParseAddr("10.0.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0:ab12:4843:2222:6273:2222"),
|
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2023-04-16 10:26:35 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{
|
|
|
|
"fd7a:115c:a1e0:ab12:4843:2222:6273:2222", "10.0.0.1",
|
|
|
|
}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
2023-01-30 08:39:27 +00:00
|
|
|
{
|
|
|
|
name: "simple host by hostname alias",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
2023-01-30 08:39:27 +00:00
|
|
|
Hosts: Hosts{
|
|
|
|
"testy": netip.MustParsePrefix("10.0.0.132/32"),
|
|
|
|
},
|
|
|
|
},
|
2023-04-26 08:58:26 +00:00
|
|
|
},
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
alias: "testy",
|
|
|
|
nodes: types.Nodes{},
|
2023-01-30 08:39:27 +00:00
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{}, []string{"10.0.0.132/32"}),
|
2023-01-30 08:39:27 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
2023-04-16 10:26:35 +00:00
|
|
|
{
|
|
|
|
name: "private network",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
2023-04-16 10:26:35 +00:00
|
|
|
Hosts: Hosts{
|
|
|
|
"homeNetwork": netip.MustParsePrefix("192.168.1.0/24"),
|
|
|
|
},
|
|
|
|
},
|
2023-04-26 08:58:26 +00:00
|
|
|
},
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
alias: "homeNetwork",
|
|
|
|
nodes: types.Nodes{},
|
2023-04-16 10:26:35 +00:00
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{}, []string{"192.168.1.0/24"}),
|
2023-04-16 10:26:35 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
{
|
|
|
|
name: "simple CIDR",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2023-06-12 13:29:34 +00:00
|
|
|
alias: "10.0.0.0/16",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{},
|
2023-06-12 13:29:34 +00:00
|
|
|
aclPolicy: ACLPolicy{},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{}, []string{"10.0.0.0/16"}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "simple tag",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
TagOwners: TagOwners{"tag:hr-webserver": []string{"joe"}},
|
|
|
|
},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2022-02-20 22:00:31 +00:00
|
|
|
alias: "tag:hr-webserver",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{
|
|
|
|
"100.64.0.1", "100.64.0.2",
|
|
|
|
}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "No tag defined",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
Groups: Groups{"group:accountant": []string{"joe", "marc"}},
|
|
|
|
TagOwners: TagOwners{
|
|
|
|
"tag:accountant-webserver": []string{"group:accountant"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2022-02-20 22:00:31 +00:00
|
|
|
alias: "tag:hr-webserver",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: true,
|
|
|
|
},
|
2022-04-15 16:01:13 +00:00
|
|
|
{
|
|
|
|
name: "Forced tag defined",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{},
|
|
|
|
},
|
2022-04-15 16:01:13 +00:00
|
|
|
args: args{
|
|
|
|
alias: "tag:hr-webserver",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-04-15 16:01:13 +00:00
|
|
|
ForcedTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-04-15 16:01:13 +00:00
|
|
|
ForcedTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{"100.64.0.1", "100.64.0.2"}, []string{}),
|
2022-06-01 13:39:59 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "Forced tag with legitimate tagOwner",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
TagOwners: TagOwners{
|
|
|
|
"tag:hr-webserver": []string{"joe"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2022-06-01 13:39:59 +00:00
|
|
|
args: args{
|
|
|
|
alias: "tag:hr-webserver",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-06-01 13:39:59 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-06-01 13:39:59 +00:00
|
|
|
ForcedTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-06-01 13:39:59 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-06-01 13:39:59 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
2022-06-01 13:39:59 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
2022-06-01 13:39:59 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-06-01 13:39:59 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2022-06-01 13:39:59 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{"100.64.0.1", "100.64.0.2"}, []string{}),
|
2022-04-15 16:01:13 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
{
|
2023-01-17 16:43:44 +00:00
|
|
|
name: "list host in user without correctly tagged servers",
|
2023-04-26 08:58:26 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
TagOwners: TagOwners{"tag:accountant-webserver": []string{"joe"}},
|
|
|
|
},
|
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
args: args{
|
2022-02-20 22:00:31 +00:00
|
|
|
alias: "joe",
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:accountant-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:accountant-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
2023-04-28 14:11:02 +00:00
|
|
|
want: set([]string{"100.64.0.4"}, []string{}),
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
}
|
2022-02-14 14:26:54 +00:00
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
2023-05-21 16:37:59 +00:00
|
|
|
got, err := test.field.pol.ExpandAlias(
|
2023-09-24 11:42:05 +00:00
|
|
|
test.args.nodes,
|
2022-02-14 14:54:51 +00:00
|
|
|
test.args.alias,
|
|
|
|
)
|
2022-02-14 14:26:54 +00:00
|
|
|
if (err != nil) != test.wantErr {
|
|
|
|
t.Errorf("expandAlias() error = %v, wantErr %v", err, test.wantErr)
|
|
|
|
|
2022-02-07 15:12:05 +00:00
|
|
|
return
|
|
|
|
}
|
2023-04-28 14:11:02 +00:00
|
|
|
if diff := cmp.Diff(test.want, got); diff != "" {
|
2023-05-31 07:59:37 +00:00
|
|
|
t.Errorf("expandAlias() unexpected result (-want +got):\n%s", diff)
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_excludeCorrectlyTaggedNodes(t *testing.T) {
|
|
|
|
type args struct {
|
2023-06-12 13:29:34 +00:00
|
|
|
aclPolicy *ACLPolicy
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes types.Nodes
|
2023-06-12 13:29:34 +00:00
|
|
|
user string
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
2023-09-24 11:42:05 +00:00
|
|
|
want types.Nodes
|
2022-02-07 15:12:05 +00:00
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "exclude nodes with valid tags",
|
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-02-20 22:00:31 +00:00
|
|
|
TagOwners: TagOwners{"tag:accountant-webserver": []string{"joe"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:accountant-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:accountant-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
user: "joe",
|
2022-08-04 08:42:47 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.4")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-08-04 08:42:47 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "exclude nodes with valid tags, and owner is in a group",
|
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-08-04 08:42:47 +00:00
|
|
|
Groups: Groups{
|
|
|
|
"group:accountant": []string{"joe", "bar"},
|
|
|
|
},
|
2022-08-04 08:47:00 +00:00
|
|
|
TagOwners: TagOwners{
|
|
|
|
"tag:accountant-webserver": []string{"group:accountant"},
|
|
|
|
},
|
2022-08-04 08:42:47 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-08-04 08:42:47 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-08-04 08:42:47 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:accountant-webserver"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-08-04 08:42:47 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-08-04 08:42:47 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:accountant-webserver"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-08-04 08:42:47 +00:00
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-08-04 08:42:47 +00:00
|
|
|
},
|
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
user: "joe",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.4")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "exclude nodes with valid tags and with forced tags",
|
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-04-15 16:01:13 +00:00
|
|
|
TagOwners: TagOwners{"tag:accountant-webserver": []string{"joe"}},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-04-15 16:01:13 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "foo",
|
|
|
|
RequestTags: []string{"tag:accountant-webserver"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2022-04-15 16:01:13 +00:00
|
|
|
ForcedTags: []string{"tag:accountant-webserver"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
user: "joe",
|
2022-04-15 16:01:13 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.4")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "all nodes have invalid tags, don't exclude them",
|
|
|
|
args: args{
|
2023-04-26 08:58:26 +00:00
|
|
|
aclPolicy: &ACLPolicy{
|
2022-02-20 22:00:31 +00:00
|
|
|
TagOwners: TagOwners{"tag:accountant-webserver": []string{"joe"}},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "hr-web1",
|
|
|
|
RequestTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "hr-web2",
|
|
|
|
RequestTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-06-12 13:29:34 +00:00
|
|
|
user: "joe",
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
2022-02-20 22:00:31 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "hr-web1",
|
|
|
|
RequestTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2022-02-20 22:00:31 +00:00
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2022-03-01 16:34:35 +00:00
|
|
|
OS: "centos",
|
|
|
|
Hostname: "hr-web2",
|
|
|
|
RequestTags: []string{"tag:hr-webserver"},
|
|
|
|
},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2022-09-01 22:05:43 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
2022-02-20 22:00:31 +00:00
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{},
|
2022-02-14 14:54:51 +00:00
|
|
|
},
|
2022-02-07 15:12:05 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2022-02-14 14:26:54 +00:00
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
2022-03-02 08:15:14 +00:00
|
|
|
got := excludeCorrectlyTaggedNodes(
|
2022-02-14 14:54:51 +00:00
|
|
|
test.args.aclPolicy,
|
|
|
|
test.args.nodes,
|
2023-01-17 16:43:44 +00:00
|
|
|
test.args.user,
|
2022-02-14 14:54:51 +00:00
|
|
|
)
|
2023-12-09 17:09:24 +00:00
|
|
|
if diff := cmp.Diff(test.want, got, util.Comparers...); diff != "" {
|
2023-06-21 06:12:24 +00:00
|
|
|
t.Errorf("excludeCorrectlyTaggedNodes() (-want +got):\n%s", diff)
|
2022-02-07 15:12:05 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2023-04-03 08:08:48 +00:00
|
|
|
|
2023-04-26 12:04:12 +00:00
|
|
|
func TestACLPolicy_generateFilterRules(t *testing.T) {
|
|
|
|
type field struct {
|
|
|
|
pol ACLPolicy
|
|
|
|
}
|
|
|
|
type args struct {
|
2023-09-24 11:42:05 +00:00
|
|
|
node *types.Node
|
|
|
|
peers types.Nodes
|
2023-04-26 12:04:12 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
field field
|
|
|
|
args args
|
|
|
|
want []tailcfg.FilterRule
|
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "no-policy",
|
|
|
|
field: field{},
|
|
|
|
args: args{},
|
|
|
|
want: []tailcfg.FilterRule{},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
2023-04-26 15:27:51 +00:00
|
|
|
name: "allow-all",
|
2023-04-26 12:04:12 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-06-13 08:03:22 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0:ab12:4843:2222:6273:2221"),
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
peers: types.Nodes{},
|
2023-04-26 12:04:12 +00:00
|
|
|
},
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
2023-04-28 14:11:02 +00:00
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
2023-04-26 12:04:12 +00:00
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{
|
2023-04-28 14:11:02 +00:00
|
|
|
IP: "0.0.0.0/0",
|
|
|
|
Ports: tailcfg.PortRange{
|
|
|
|
First: 0,
|
|
|
|
Last: 65535,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
IP: "::/0",
|
2023-04-26 12:04:12 +00:00
|
|
|
Ports: tailcfg.PortRange{
|
|
|
|
First: 0,
|
|
|
|
Last: 65535,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
2023-06-16 14:42:30 +00:00
|
|
|
name: "host1-can-reach-host2-full",
|
2023-04-26 12:04:12 +00:00
|
|
|
field: field{
|
|
|
|
pol: ACLPolicy{
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
2023-06-13 08:03:22 +00:00
|
|
|
Sources: []string{"100.64.0.2"},
|
|
|
|
Destinations: []string{"100.64.0.1:*"},
|
2023-04-26 12:04:12 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-06-12 09:18:58 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0:ab12:4843:2222:6273:2221"),
|
2023-04-26 12:04:12 +00:00
|
|
|
},
|
2023-06-12 09:18:58 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
peers: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-04-26 15:27:51 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
2023-04-26 12:04:12 +00:00
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0:ab12:4843:2222:6273:2222"),
|
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
2023-04-26 12:04:12 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
2023-05-21 16:37:59 +00:00
|
|
|
SrcIPs: []string{
|
2023-06-13 08:03:22 +00:00
|
|
|
"100.64.0.2/32",
|
|
|
|
"fd7a:115c:a1e0:ab12:4843:2222:6273:2222/128",
|
2023-05-21 16:37:59 +00:00
|
|
|
},
|
2023-04-26 12:04:12 +00:00
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{
|
2023-06-13 08:03:22 +00:00
|
|
|
IP: "100.64.0.1/32",
|
2023-04-26 12:04:12 +00:00
|
|
|
Ports: tailcfg.PortRange{
|
|
|
|
First: 0,
|
|
|
|
Last: 65535,
|
|
|
|
},
|
|
|
|
},
|
2023-04-26 15:27:51 +00:00
|
|
|
{
|
2023-06-13 08:03:22 +00:00
|
|
|
IP: "fd7a:115c:a1e0:ab12:4843:2222:6273:2221/128",
|
2023-04-26 15:27:51 +00:00
|
|
|
Ports: tailcfg.PortRange{
|
|
|
|
First: 0,
|
|
|
|
Last: 65535,
|
|
|
|
},
|
|
|
|
},
|
2023-04-26 12:04:12 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
2023-06-16 14:42:30 +00:00
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got, err := tt.field.pol.generateFilterRules(
|
2023-09-24 11:42:05 +00:00
|
|
|
tt.args.node,
|
2023-06-16 14:42:30 +00:00
|
|
|
tt.args.peers,
|
|
|
|
)
|
|
|
|
if (err != nil) != tt.wantErr {
|
|
|
|
t.Errorf("ACLgenerateFilterRules() error = %v, wantErr %v", err, tt.wantErr)
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if diff := cmp.Diff(tt.want, got); diff != "" {
|
|
|
|
log.Trace().Interface("got", got).Msg("result")
|
|
|
|
t.Errorf("ACLgenerateFilterRules() unexpected result (-want +got):\n%s", diff)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestReduceFilterRules(t *testing.T) {
|
|
|
|
tests := []struct {
|
2023-09-24 11:42:05 +00:00
|
|
|
name string
|
|
|
|
node *types.Node
|
|
|
|
peers types.Nodes
|
|
|
|
pol ACLPolicy
|
|
|
|
want []tailcfg.FilterRule
|
2023-06-16 14:42:30 +00:00
|
|
|
}{
|
2023-06-13 08:03:22 +00:00
|
|
|
{
|
|
|
|
name: "host1-can-reach-host2-no-rules",
|
2023-06-16 14:42:30 +00:00
|
|
|
pol: ACLPolicy{
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"100.64.0.1"},
|
|
|
|
Destinations: []string{"100.64.0.2:*"},
|
2023-06-13 08:03:22 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-06-16 14:42:30 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0:ab12:4843:2222:6273:2221"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
peers: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-06-16 14:42:30 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0:ab12:4843:2222:6273:2222"),
|
2023-06-13 08:03:22 +00:00
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
2023-06-16 14:42:30 +00:00
|
|
|
want: []tailcfg.FilterRule{},
|
2023-06-13 08:03:22 +00:00
|
|
|
},
|
2024-01-18 16:30:25 +00:00
|
|
|
{
|
|
|
|
name: "1604-subnet-routers-are-preserved",
|
|
|
|
pol: ACLPolicy{
|
|
|
|
Groups: Groups{
|
|
|
|
"group:admins": {"user1"},
|
|
|
|
},
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"group:admins"},
|
|
|
|
Destinations: []string{"group:admins:*"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"group:admins"},
|
|
|
|
Destinations: []string{"10.33.0.0/16:*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
node: &types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
|
|
|
RoutableIPs: []netip.Prefix{
|
|
|
|
netip.MustParsePrefix("10.33.0.0/16"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
peers: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
IPAddresses: types.NodeAddresses{
|
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{
|
|
|
|
"100.64.0.1/32",
|
|
|
|
"100.64.0.2/32",
|
|
|
|
"fd7a:115c:a1e0::1/128",
|
|
|
|
"fd7a:115c:a1e0::2/128",
|
|
|
|
},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{
|
|
|
|
IP: "100.64.0.1/32",
|
|
|
|
Ports: tailcfg.PortRangeAny,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
IP: "fd7a:115c:a1e0::1/128",
|
|
|
|
Ports: tailcfg.PortRangeAny,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
SrcIPs: []string{
|
|
|
|
"100.64.0.1/32",
|
|
|
|
"100.64.0.2/32",
|
|
|
|
"fd7a:115c:a1e0::1/128",
|
|
|
|
"fd7a:115c:a1e0::2/128",
|
|
|
|
},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{
|
|
|
|
IP: "10.33.0.0/16",
|
|
|
|
Ports: tailcfg.PortRangeAny,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-04-26 12:04:12 +00:00
|
|
|
}
|
2023-06-16 14:42:30 +00:00
|
|
|
|
2023-04-26 12:04:12 +00:00
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
2023-06-16 14:42:30 +00:00
|
|
|
rules, _ := tt.pol.generateFilterRules(
|
2023-09-24 11:42:05 +00:00
|
|
|
tt.node,
|
2023-06-16 14:42:30 +00:00
|
|
|
tt.peers,
|
2023-04-26 12:04:12 +00:00
|
|
|
)
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
got := ReduceFilterRules(tt.node, rules)
|
2023-04-26 12:04:12 +00:00
|
|
|
|
|
|
|
if diff := cmp.Diff(tt.want, got); diff != "" {
|
|
|
|
log.Trace().Interface("got", got).Msg("result")
|
2023-06-16 14:42:30 +00:00
|
|
|
t.Errorf("TestReduceFilterRules() unexpected result (-want +got):\n%s", diff)
|
2023-05-21 16:37:59 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_getTags(t *testing.T) {
|
|
|
|
type args struct {
|
2023-06-12 13:29:34 +00:00
|
|
|
aclPolicy *ACLPolicy
|
2023-09-24 11:42:05 +00:00
|
|
|
node *types.Node
|
2023-05-21 16:37:59 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
wantInvalid []string
|
|
|
|
wantValid []string
|
|
|
|
}{
|
|
|
|
{
|
2023-09-24 11:42:05 +00:00
|
|
|
name: "valid tag one nodes",
|
2023-05-21 16:37:59 +00:00
|
|
|
args: args{
|
|
|
|
aclPolicy: &ACLPolicy{
|
|
|
|
TagOwners: TagOwners{
|
|
|
|
"tag:valid": []string{"joe"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{
|
|
|
|
Name: "joe",
|
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2023-05-21 16:37:59 +00:00
|
|
|
RequestTags: []string{"tag:valid"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantValid: []string{"tag:valid"},
|
|
|
|
wantInvalid: nil,
|
|
|
|
},
|
|
|
|
{
|
2023-09-24 11:42:05 +00:00
|
|
|
name: "invalid tag and valid tag one nodes",
|
2023-05-21 16:37:59 +00:00
|
|
|
args: args{
|
|
|
|
aclPolicy: &ACLPolicy{
|
|
|
|
TagOwners: TagOwners{
|
|
|
|
"tag:valid": []string{"joe"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{
|
|
|
|
Name: "joe",
|
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2023-05-21 16:37:59 +00:00
|
|
|
RequestTags: []string{"tag:valid", "tag:invalid"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantValid: []string{"tag:valid"},
|
|
|
|
wantInvalid: []string{"tag:invalid"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "multiple invalid and identical tags, should return only one invalid tag",
|
|
|
|
args: args{
|
|
|
|
aclPolicy: &ACLPolicy{
|
|
|
|
TagOwners: TagOwners{
|
|
|
|
"tag:valid": []string{"joe"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{
|
|
|
|
Name: "joe",
|
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2023-05-21 16:37:59 +00:00
|
|
|
RequestTags: []string{
|
|
|
|
"tag:invalid",
|
|
|
|
"tag:valid",
|
|
|
|
"tag:invalid",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantValid: []string{"tag:valid"},
|
|
|
|
wantInvalid: []string{"tag:invalid"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "only invalid tags",
|
|
|
|
args: args{
|
|
|
|
aclPolicy: &ACLPolicy{
|
|
|
|
TagOwners: TagOwners{
|
|
|
|
"tag:valid": []string{"joe"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{
|
|
|
|
Name: "joe",
|
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2023-05-21 16:37:59 +00:00
|
|
|
RequestTags: []string{"tag:invalid", "very-invalid"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantValid: nil,
|
|
|
|
wantInvalid: []string{"tag:invalid", "very-invalid"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "empty ACLPolicy should return empty tags and should not panic",
|
|
|
|
args: args{
|
|
|
|
aclPolicy: &ACLPolicy{},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{
|
|
|
|
Name: "joe",
|
|
|
|
},
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &tailcfg.Hostinfo{
|
2023-05-21 16:37:59 +00:00
|
|
|
RequestTags: []string{"tag:invalid", "very-invalid"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantValid: nil,
|
|
|
|
wantInvalid: []string{"tag:invalid", "very-invalid"},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.name, func(t *testing.T) {
|
2023-09-24 11:42:05 +00:00
|
|
|
gotValid, gotInvalid := test.args.aclPolicy.TagsOfNode(
|
|
|
|
test.args.node,
|
2023-05-21 16:37:59 +00:00
|
|
|
)
|
|
|
|
for _, valid := range gotValid {
|
|
|
|
if !util.StringOrPrefixListContains(test.wantValid, valid) {
|
|
|
|
t.Errorf(
|
|
|
|
"valids: getTags() = %v, want %v",
|
|
|
|
gotValid,
|
|
|
|
test.wantValid,
|
|
|
|
)
|
|
|
|
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for _, invalid := range gotInvalid {
|
|
|
|
if !util.StringOrPrefixListContains(test.wantInvalid, invalid) {
|
|
|
|
t.Errorf(
|
|
|
|
"invalids: getTags() = %v, want %v",
|
|
|
|
gotInvalid,
|
|
|
|
test.wantInvalid,
|
|
|
|
)
|
|
|
|
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_getFilteredByACLPeers(t *testing.T) {
|
|
|
|
type args struct {
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes types.Nodes
|
|
|
|
rules []tailcfg.FilterRule
|
|
|
|
node *types.Node
|
2023-05-21 16:37:59 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
2023-09-24 11:42:05 +00:00
|
|
|
want types.Nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "all hosts can talk to each other",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ // list of all nodess in the database
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.64.0.1", "100.64.0.2", "100.64.0.3"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.2")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.3")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "One host can talk to another, but not all hosts",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ // list of all nodess in the database
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.64.0.1", "100.64.0.2", "100.64.0.3"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.2")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "host cannot directly talk to destination, but return path is authorized",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ // list of all nodess in the database
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.64.0.3"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.2")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.3")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rules allows all hosts to reach one destination",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ // list of all nodess in the database
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"*"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rules allows all hosts to reach one destination, destination can reach all hosts",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ // list of all nodess in the database
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"*"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "rule allows all hosts to reach all destinations",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ // list of all nodess in the database
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"*"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.2")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.3")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "without rule all communications are forbidden",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ // list of all nodess in the database
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "joe"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "mickael"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.2")},
|
2023-05-21 16:37:59 +00:00
|
|
|
User: types.User{Name: "marc"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{},
|
2023-05-21 16:37:59 +00:00
|
|
|
},
|
|
|
|
{
|
|
|
|
// Investigating 699
|
2023-09-24 11:42:05 +00:00
|
|
|
// Found some nodes: [ts-head-8w6paa ts-unstable-lys2ib ts-head-upcrmb ts-unstable-rlwpvr] nodes=ts-head-8w6paa
|
2023-05-21 16:37:59 +00:00
|
|
|
// ACL rules generated ACL=[{"DstPorts":[{"Bits":null,"IP":"*","Ports":{"First":0,"Last":65535}}],"SrcIPs":["fd7a:115c:a1e0::3","100.64.0.3","fd7a:115c:a1e0::4","100.64.0.4"]}]
|
|
|
|
// ACL Cache Map={"100.64.0.3":{"*":{}},"100.64.0.4":{"*":{}},"fd7a:115c:a1e0::3":{"*":{}},"fd7a:115c:a1e0::4":{"*":{}}}
|
|
|
|
name: "issue-699-broken-star",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: types.Nodes{ //
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
|
|
|
Hostname: "ts-head-upcrmb",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
|
|
|
Hostname: "ts-unstable-rlwpvr",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::4"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
|
|
|
Hostname: "ts-head-8w6paa",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 4,
|
|
|
|
Hostname: "ts-unstable-lys2ib",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.2"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::2"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{ // list of all ACLRules registered
|
|
|
|
{
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{
|
|
|
|
IP: "*",
|
|
|
|
Ports: tailcfg.PortRange{First: 0, Last: 65535},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
SrcIPs: []string{
|
|
|
|
"fd7a:115c:a1e0::3", "100.64.0.3",
|
|
|
|
"fd7a:115c:a1e0::4", "100.64.0.4",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{ // current nodes
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 3,
|
|
|
|
Hostname: "ts-head-8w6paa",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.1"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::1"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: types.Nodes{
|
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 1,
|
|
|
|
Hostname: "ts-head-upcrmb",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.3"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::3"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
&types.Node{
|
2023-05-21 16:37:59 +00:00
|
|
|
ID: 2,
|
|
|
|
Hostname: "ts-unstable-rlwpvr",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{
|
2023-05-21 16:37:59 +00:00
|
|
|
netip.MustParseAddr("100.64.0.4"),
|
|
|
|
netip.MustParseAddr("fd7a:115c:a1e0::4"),
|
|
|
|
},
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-05-31 16:45:04 +00:00
|
|
|
{
|
|
|
|
name: "failing-edge-case-during-p3-refactor",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: []*types.Node{
|
2023-05-31 16:45:04 +00:00
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.2")},
|
|
|
|
Hostname: "peer1",
|
|
|
|
User: types.User{Name: "mini"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 2,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.3")},
|
|
|
|
Hostname: "peer2",
|
|
|
|
User: types.User{Name: "peer2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.64.0.1/32"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.3/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "::/0", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-05-31 16:45:04 +00:00
|
|
|
ID: 0,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.1")},
|
|
|
|
Hostname: "mini",
|
|
|
|
User: types.User{Name: "mini"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: []*types.Node{
|
2023-05-31 16:45:04 +00:00
|
|
|
{
|
|
|
|
ID: 2,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.3")},
|
|
|
|
Hostname: "peer2",
|
|
|
|
User: types.User{Name: "peer2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-06-16 14:42:30 +00:00
|
|
|
{
|
|
|
|
name: "p4-host-in-netmap-user2-dest-bug",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: []*types.Node{
|
2023-06-16 14:42:30 +00:00
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.2")},
|
|
|
|
Hostname: "user1-2",
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 0,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.1")},
|
|
|
|
Hostname: "user1-1",
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 3,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.4")},
|
|
|
|
Hostname: "user2-2",
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{
|
|
|
|
"100.64.0.3/32",
|
|
|
|
"100.64.0.4/32",
|
|
|
|
"fd7a:115c:a1e0::3/128",
|
|
|
|
"fd7a:115c:a1e0::4/128",
|
|
|
|
},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.3/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "100.64.0.4/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::3/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::4/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
SrcIPs: []string{
|
|
|
|
"100.64.0.1/32",
|
|
|
|
"100.64.0.2/32",
|
|
|
|
"fd7a:115c:a1e0::1/128",
|
|
|
|
"fd7a:115c:a1e0::2/128",
|
|
|
|
},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.3/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "100.64.0.4/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::3/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::4/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-06-16 14:42:30 +00:00
|
|
|
ID: 2,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.3")},
|
|
|
|
Hostname: "user-2-1",
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: []*types.Node{
|
2023-06-16 14:42:30 +00:00
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.2")},
|
|
|
|
Hostname: "user1-2",
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 0,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.1")},
|
|
|
|
Hostname: "user1-1",
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 3,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.4")},
|
|
|
|
Hostname: "user2-2",
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "p4-host-in-netmap-user1-dest-bug",
|
|
|
|
args: args{
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes: []*types.Node{
|
2023-06-16 14:42:30 +00:00
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.2")},
|
|
|
|
Hostname: "user1-2",
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 2,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.3")},
|
|
|
|
Hostname: "user-2-1",
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 3,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.4")},
|
|
|
|
Hostname: "user2-2",
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
rules: []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{
|
|
|
|
"100.64.0.1/32",
|
|
|
|
"100.64.0.2/32",
|
|
|
|
"fd7a:115c:a1e0::1/128",
|
|
|
|
"fd7a:115c:a1e0::2/128",
|
|
|
|
},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.1/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "100.64.0.2/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::1/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::2/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
SrcIPs: []string{
|
|
|
|
"100.64.0.1/32",
|
|
|
|
"100.64.0.2/32",
|
|
|
|
"fd7a:115c:a1e0::1/128",
|
|
|
|
"fd7a:115c:a1e0::2/128",
|
|
|
|
},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.3/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "100.64.0.4/32", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::3/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
{IP: "fd7a:115c:a1e0::4/128", Ports: tailcfg.PortRangeAny},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
node: &types.Node{
|
2023-06-16 14:42:30 +00:00
|
|
|
ID: 0,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.1")},
|
|
|
|
Hostname: "user1-1",
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
want: []*types.Node{
|
2023-06-16 14:42:30 +00:00
|
|
|
{
|
|
|
|
ID: 1,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.2")},
|
|
|
|
Hostname: "user1-2",
|
|
|
|
User: types.User{Name: "user1"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 2,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.3")},
|
|
|
|
Hostname: "user-2-1",
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
ID: 3,
|
|
|
|
IPAddresses: []netip.Addr{netip.MustParseAddr("100.64.0.4")},
|
|
|
|
Hostname: "user2-2",
|
|
|
|
User: types.User{Name: "user2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2023-05-21 16:37:59 +00:00
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
2023-09-24 11:42:05 +00:00
|
|
|
got := FilterNodesByACL(
|
|
|
|
tt.args.node,
|
|
|
|
tt.args.nodes,
|
2023-05-21 16:37:59 +00:00
|
|
|
tt.args.rules,
|
|
|
|
)
|
2023-12-09 17:09:24 +00:00
|
|
|
if diff := cmp.Diff(tt.want, got, util.Comparers...); diff != "" {
|
2023-09-24 11:42:05 +00:00
|
|
|
t.Errorf("FilterNodesByACL() unexpected result (-want +got):\n%s", diff)
|
2023-04-26 12:04:12 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2023-06-08 17:50:59 +00:00
|
|
|
|
|
|
|
func TestSSHRules(t *testing.T) {
|
|
|
|
tests := []struct {
|
2023-09-24 11:42:05 +00:00
|
|
|
name string
|
|
|
|
node types.Node
|
|
|
|
peers types.Nodes
|
|
|
|
pol ACLPolicy
|
|
|
|
want []*tailcfg.SSHRule
|
2023-06-08 17:50:59 +00:00
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "peers-can-connect",
|
2023-09-24 11:42:05 +00:00
|
|
|
node: types.Node{
|
|
|
|
Hostname: "testnodes",
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.99.42")},
|
2023-06-08 17:50:59 +00:00
|
|
|
UserID: 0,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
peers: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
Hostname: "testnodes2",
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-06-08 17:50:59 +00:00
|
|
|
UserID: 0,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
pol: ACLPolicy{
|
|
|
|
Groups: Groups{
|
|
|
|
"group:test": []string{"user1"},
|
|
|
|
},
|
|
|
|
Hosts: Hosts{
|
|
|
|
"client": netip.PrefixFrom(netip.MustParseAddr("100.64.99.42"), 32),
|
|
|
|
},
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
SSHs: []SSH{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"group:test"},
|
|
|
|
Destinations: []string{"client"},
|
|
|
|
Users: []string{"autogroup:nonroot"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"client"},
|
|
|
|
Users: []string{"autogroup:nonroot"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"group:test"},
|
|
|
|
Destinations: []string{"100.64.99.42"},
|
|
|
|
Users: []string{"autogroup:nonroot"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"100.64.99.42"},
|
|
|
|
Users: []string{"autogroup:nonroot"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: []*tailcfg.SSHRule{
|
|
|
|
{
|
|
|
|
Principals: []*tailcfg.SSHPrincipal{
|
|
|
|
{
|
|
|
|
UserLogin: "user1",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
SSHUsers: map[string]string{
|
|
|
|
"autogroup:nonroot": "=",
|
|
|
|
},
|
|
|
|
Action: &tailcfg.SSHAction{Accept: true, AllowLocalPortForwarding: true},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
SSHUsers: map[string]string{
|
|
|
|
"autogroup:nonroot": "=",
|
|
|
|
},
|
|
|
|
Principals: []*tailcfg.SSHPrincipal{
|
|
|
|
{
|
|
|
|
Any: true,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Action: &tailcfg.SSHAction{Accept: true, AllowLocalPortForwarding: true},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Principals: []*tailcfg.SSHPrincipal{
|
|
|
|
{
|
|
|
|
UserLogin: "user1",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
SSHUsers: map[string]string{
|
|
|
|
"autogroup:nonroot": "=",
|
|
|
|
},
|
|
|
|
Action: &tailcfg.SSHAction{Accept: true, AllowLocalPortForwarding: true},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
SSHUsers: map[string]string{
|
|
|
|
"autogroup:nonroot": "=",
|
|
|
|
},
|
|
|
|
Principals: []*tailcfg.SSHPrincipal{
|
|
|
|
{
|
|
|
|
Any: true,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Action: &tailcfg.SSHAction{Accept: true, AllowLocalPortForwarding: true},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "peers-cannot-connect",
|
2023-09-24 11:42:05 +00:00
|
|
|
node: types.Node{
|
|
|
|
Hostname: "testnodes",
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-06-08 17:50:59 +00:00
|
|
|
UserID: 0,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
},
|
2023-09-24 11:42:05 +00:00
|
|
|
peers: types.Nodes{
|
|
|
|
&types.Node{
|
|
|
|
Hostname: "testnodes2",
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.99.42")},
|
2023-06-08 17:50:59 +00:00
|
|
|
UserID: 0,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
pol: ACLPolicy{
|
|
|
|
Groups: Groups{
|
|
|
|
"group:test": []string{"user1"},
|
|
|
|
},
|
|
|
|
Hosts: Hosts{
|
|
|
|
"client": netip.PrefixFrom(netip.MustParseAddr("100.64.99.42"), 32),
|
|
|
|
},
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
SSHs: []SSH{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"group:test"},
|
|
|
|
Destinations: []string{"100.64.99.42"},
|
|
|
|
Users: []string{"autogroup:nonroot"},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"100.64.99.42"},
|
|
|
|
Users: []string{"autogroup:nonroot"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: []*tailcfg.SSHRule{},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
2023-09-24 11:42:05 +00:00
|
|
|
got, err := tt.pol.generateSSHRules(&tt.node, tt.peers)
|
2023-06-08 17:50:59 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
if diff := cmp.Diff(tt.want, got); diff != "" {
|
|
|
|
t.Errorf("TestSSHRules() unexpected result (-want +got):\n%s", diff)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2023-06-12 13:59:05 +00:00
|
|
|
|
|
|
|
func TestParseDestination(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
dest string
|
|
|
|
wantAlias string
|
|
|
|
wantPort string
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
dest: "git-server:*",
|
|
|
|
wantAlias: "git-server",
|
|
|
|
wantPort: "*",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
dest: "192.168.1.0/24:22",
|
|
|
|
wantAlias: "192.168.1.0/24",
|
|
|
|
wantPort: "22",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
dest: "192.168.1.1:22",
|
|
|
|
wantAlias: "192.168.1.1",
|
|
|
|
wantPort: "22",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
dest: "fd7a:115c:a1e0::2:22",
|
|
|
|
wantAlias: "fd7a:115c:a1e0::2",
|
|
|
|
wantPort: "22",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
dest: "fd7a:115c:a1e0::2/128:22",
|
|
|
|
wantAlias: "fd7a:115c:a1e0::2/128",
|
|
|
|
wantPort: "22",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
dest: "tag:montreal-webserver:80,443",
|
|
|
|
wantAlias: "tag:montreal-webserver",
|
|
|
|
wantPort: "80,443",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
dest: "tag:api-server:443",
|
|
|
|
wantAlias: "tag:api-server",
|
|
|
|
wantPort: "443",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
dest: "example-host-1:*",
|
|
|
|
wantAlias: "example-host-1",
|
|
|
|
wantPort: "*",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.dest, func(t *testing.T) {
|
|
|
|
alias, port, _ := parseDestination(tt.dest)
|
|
|
|
|
|
|
|
if alias != tt.wantAlias {
|
|
|
|
t.Errorf("unexpected alias: want(%s) != got(%s)", tt.wantAlias, alias)
|
|
|
|
}
|
|
|
|
|
|
|
|
if port != tt.wantPort {
|
|
|
|
t.Errorf("unexpected port: want(%s) != got(%s)", tt.wantPort, port)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2023-06-19 06:48:49 +00:00
|
|
|
|
|
|
|
// this test should validate that we can expand a group in a TagOWner section and
|
|
|
|
// match properly the IP's of the related hosts. The owner is valid and the tag is also valid.
|
|
|
|
// the tag is matched in the Sources section.
|
|
|
|
func TestValidExpandTagOwnersInSources(t *testing.T) {
|
|
|
|
hostInfo := tailcfg.Hostinfo{
|
|
|
|
OS: "centos",
|
2023-09-24 11:42:05 +00:00
|
|
|
Hostname: "testnodes",
|
2023-06-19 06:48:49 +00:00
|
|
|
RequestTags: []string{"tag:test"},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
node := &types.Node{
|
2023-06-19 06:48:49 +00:00
|
|
|
ID: 0,
|
2023-09-24 11:42:05 +00:00
|
|
|
Hostname: "testnodes",
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-06-19 06:48:49 +00:00
|
|
|
UserID: 0,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
RegisterMethod: util.RegisterMethodAuthKey,
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &hostInfo,
|
2023-06-19 06:48:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pol := &ACLPolicy{
|
|
|
|
Groups: Groups{"group:test": []string{"user1", "user2"}},
|
|
|
|
TagOwners: TagOwners{"tag:test": []string{"user3", "group:test"}},
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"tag:test"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
got, _, err := GenerateFilterAndSSHRules(pol, node, types.Nodes{})
|
2023-06-19 06:48:49 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
want := []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.64.0.1/32"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "0.0.0.0/0", Ports: tailcfg.PortRange{Last: 65535}},
|
|
|
|
{IP: "::/0", Ports: tailcfg.PortRange{Last: 65535}},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
if diff := cmp.Diff(want, got); diff != "" {
|
|
|
|
t.Errorf("TestValidExpandTagOwnersInSources() unexpected result (-want +got):\n%s", diff)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// need a test with:
|
|
|
|
// tag on a host that isn't owned by a tag owners. So the user
|
|
|
|
// of the host should be valid.
|
|
|
|
func TestInvalidTagValidUser(t *testing.T) {
|
|
|
|
hostInfo := tailcfg.Hostinfo{
|
|
|
|
OS: "centos",
|
2023-09-24 11:42:05 +00:00
|
|
|
Hostname: "testnodes",
|
2023-06-19 06:48:49 +00:00
|
|
|
RequestTags: []string{"tag:foo"},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
node := &types.Node{
|
2023-06-19 06:48:49 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
Hostname: "testnodes",
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-06-19 06:48:49 +00:00
|
|
|
UserID: 1,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
RegisterMethod: util.RegisterMethodAuthKey,
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &hostInfo,
|
2023-06-19 06:48:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pol := &ACLPolicy{
|
|
|
|
TagOwners: TagOwners{"tag:test": []string{"user1"}},
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"user1"},
|
|
|
|
Destinations: []string{"*:*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
got, _, err := GenerateFilterAndSSHRules(pol, node, types.Nodes{})
|
2023-06-19 06:48:49 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
want := []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.64.0.1/32"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "0.0.0.0/0", Ports: tailcfg.PortRange{Last: 65535}},
|
|
|
|
{IP: "::/0", Ports: tailcfg.PortRange{Last: 65535}},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
if diff := cmp.Diff(want, got); diff != "" {
|
|
|
|
t.Errorf("TestInvalidTagValidUser() unexpected result (-want +got):\n%s", diff)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// this test should validate that we can expand a group in a TagOWner section and
|
|
|
|
// match properly the IP's of the related hosts. The owner is valid and the tag is also valid.
|
|
|
|
// the tag is matched in the Destinations section.
|
|
|
|
func TestValidExpandTagOwnersInDestinations(t *testing.T) {
|
|
|
|
hostInfo := tailcfg.Hostinfo{
|
|
|
|
OS: "centos",
|
2023-09-24 11:42:05 +00:00
|
|
|
Hostname: "testnodes",
|
2023-06-19 06:48:49 +00:00
|
|
|
RequestTags: []string{"tag:test"},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
node := &types.Node{
|
2023-06-19 06:48:49 +00:00
|
|
|
ID: 1,
|
2023-09-24 11:42:05 +00:00
|
|
|
Hostname: "testnodes",
|
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-06-19 06:48:49 +00:00
|
|
|
UserID: 1,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
RegisterMethod: util.RegisterMethodAuthKey,
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &hostInfo,
|
2023-06-19 06:48:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pol := &ACLPolicy{
|
|
|
|
Groups: Groups{"group:test": []string{"user1", "user2"}},
|
|
|
|
TagOwners: TagOwners{"tag:test": []string{"user3", "group:test"}},
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"*"},
|
|
|
|
Destinations: []string{"tag:test:*"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
// rules, _, err := GenerateFilterRules(pol, &node, peers, false)
|
2023-06-19 06:48:49 +00:00
|
|
|
// c.Assert(err, check.IsNil)
|
|
|
|
//
|
|
|
|
// c.Assert(rules, check.HasLen, 1)
|
|
|
|
// c.Assert(rules[0].DstPorts, check.HasLen, 1)
|
|
|
|
// c.Assert(rules[0].DstPorts[0].IP, check.Equals, "100.64.0.1/32")
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
got, _, err := GenerateFilterAndSSHRules(pol, node, types.Nodes{})
|
2023-06-19 06:48:49 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
want := []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"0.0.0.0/0", "::/0"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.1/32", Ports: tailcfg.PortRange{Last: 65535}},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
if diff := cmp.Diff(want, got); diff != "" {
|
|
|
|
t.Errorf(
|
|
|
|
"TestValidExpandTagOwnersInDestinations() unexpected result (-want +got):\n%s",
|
|
|
|
diff,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// tag on a host is owned by a tag owner, the tag is valid.
|
|
|
|
// an ACL rule is matching the tag to a user. It should not be valid since the
|
|
|
|
// host should be tied to the tag now.
|
|
|
|
func TestValidTagInvalidUser(t *testing.T) {
|
|
|
|
hostInfo := tailcfg.Hostinfo{
|
|
|
|
OS: "centos",
|
|
|
|
Hostname: "webserver",
|
|
|
|
RequestTags: []string{"tag:webapp"},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
node := &types.Node{
|
2023-06-19 06:48:49 +00:00
|
|
|
ID: 1,
|
|
|
|
Hostname: "webserver",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.1")},
|
2023-06-19 06:48:49 +00:00
|
|
|
UserID: 1,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
RegisterMethod: util.RegisterMethodAuthKey,
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &hostInfo,
|
2023-06-19 06:48:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
hostInfo2 := tailcfg.Hostinfo{
|
|
|
|
OS: "debian",
|
|
|
|
Hostname: "Hostname",
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
nodes2 := &types.Node{
|
2023-06-19 06:48:49 +00:00
|
|
|
ID: 2,
|
|
|
|
Hostname: "user",
|
2023-09-24 11:42:05 +00:00
|
|
|
IPAddresses: types.NodeAddresses{netip.MustParseAddr("100.64.0.2")},
|
2023-06-19 06:48:49 +00:00
|
|
|
UserID: 1,
|
|
|
|
User: types.User{
|
|
|
|
Name: "user1",
|
|
|
|
},
|
|
|
|
RegisterMethod: util.RegisterMethodAuthKey,
|
2023-11-21 17:20:06 +00:00
|
|
|
Hostinfo: &hostInfo2,
|
2023-06-19 06:48:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pol := &ACLPolicy{
|
|
|
|
TagOwners: TagOwners{"tag:webapp": []string{"user1"}},
|
|
|
|
ACLs: []ACL{
|
|
|
|
{
|
|
|
|
Action: "accept",
|
|
|
|
Sources: []string{"user1"},
|
|
|
|
Destinations: []string{"tag:webapp:80,443"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2023-09-24 11:42:05 +00:00
|
|
|
got, _, err := GenerateFilterAndSSHRules(pol, node, types.Nodes{nodes2})
|
2023-06-19 06:48:49 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
want := []tailcfg.FilterRule{
|
|
|
|
{
|
|
|
|
SrcIPs: []string{"100.64.0.2/32"},
|
|
|
|
DstPorts: []tailcfg.NetPortRange{
|
|
|
|
{IP: "100.64.0.1/32", Ports: tailcfg.PortRange{First: 80, Last: 80}},
|
|
|
|
{IP: "100.64.0.1/32", Ports: tailcfg.PortRange{First: 443, Last: 443}},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
if diff := cmp.Diff(want, got); diff != "" {
|
|
|
|
t.Errorf("TestValidTagInvalidUser() unexpected result (-want +got):\n%s", diff)
|
|
|
|
}
|
|
|
|
}
|