93 lines
3.4 KiB
Go
93 lines
3.4 KiB
Go
package fetchguard
|
|
|
|
import (
|
|
"net/netip"
|
|
"testing"
|
|
)
|
|
|
|
func TestIsReservedOrPrivate(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
ip string
|
|
want bool
|
|
}{
|
|
{name: "loopback v4", ip: "127.0.0.1", want: true},
|
|
{name: "rfc1918 10/8", ip: "10.1.2.3", want: true},
|
|
{name: "rfc1918 172.16/12 start", ip: "172.16.0.1", want: true},
|
|
{name: "rfc1918 172.16/12 end", ip: "172.31.255.255", want: true},
|
|
{name: "rfc1918 192.168/16", ip: "192.168.1.1", want: true},
|
|
{name: "cloud metadata link-local", ip: "169.254.169.254", want: true},
|
|
{name: "cgnat 100.64/10", ip: "100.64.0.1", want: true},
|
|
{name: "this-network 0.0.0.0/8", ip: "0.0.0.1", want: true},
|
|
{name: "public 8.8.8.8", ip: "8.8.8.8", want: false},
|
|
{name: "public 1.1.1.1", ip: "1.1.1.1", want: false},
|
|
{name: "public 93.184.216.34", ip: "93.184.216.34", want: false},
|
|
{name: "loopback v6", ip: "::1", want: true},
|
|
{name: "link-local v6", ip: "fe80::1", want: true},
|
|
{name: "unique-local v6", ip: "fc00::1", want: true},
|
|
{name: "public v6", ip: "2606:4700:4700::1111", want: false},
|
|
{name: "v4-mapped metadata", ip: "::ffff:169.254.169.254", want: true},
|
|
{name: "multicast v4", ip: "224.0.0.1", want: true},
|
|
{name: "unspecified v4", ip: "0.0.0.0", want: true},
|
|
{name: "just below 172.16.0.0/12", ip: "172.15.255.255", want: false},
|
|
{name: "just above 172.16.0.0/12", ip: "172.32.0.0", want: false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
addr, err := netip.ParseAddr(tt.ip)
|
|
if err != nil {
|
|
t.Fatalf("ParseAddr(%q): %v", tt.ip, err)
|
|
}
|
|
got := isReservedOrPrivate(addr)
|
|
if got != tt.want {
|
|
t.Fatalf("isReservedOrPrivate(%s) = %v, want %v", addr, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestIsReservedOrPrivateIPv6Transitions(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
ip string
|
|
want bool
|
|
}{
|
|
{name: "nat64 well-known loopback", ip: "64:ff9b::7f00:1", want: true},
|
|
{name: "nat64 well-known rfc1918", ip: "64:ff9b::a00:1", want: true},
|
|
{name: "nat64 well-known metadata", ip: "64:ff9b::a9fe:a9fe", want: true},
|
|
{name: "nat64 well-known public", ip: "64:ff9b::808:808", want: false},
|
|
{name: "nat64 local-use loopback", ip: "64:ff9b:1:7f00:0:100::", want: true},
|
|
{name: "nat64 local-use rfc1918", ip: "64:ff9b:1:a00:0:100::", want: true},
|
|
{name: "nat64 local-use metadata", ip: "64:ff9b:1:a9fe:a9:fe00::", want: true},
|
|
{name: "nat64 local-use public", ip: "64:ff9b:1:808:8:800::", want: false},
|
|
{name: "nat64 local-use non-zero u octet", ip: "64:ff9b:1:7f00:100:100::", want: true},
|
|
{name: "6to4 loopback", ip: "2002:7f00:1::", want: true},
|
|
{name: "6to4 rfc1918", ip: "2002:a00:1::", want: true},
|
|
{name: "6to4 metadata", ip: "2002:a9fe:a9fe::", want: true},
|
|
{name: "6to4 public", ip: "2002:808:808::", want: false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
addr := netip.MustParseAddr(tt.ip)
|
|
if got := isReservedOrPrivate(addr); got != tt.want {
|
|
t.Fatalf("isReservedOrPrivate(%s) = %v, want %v", addr, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestIsReservedOrPrivateSpecialUseSmoke(t *testing.T) {
|
|
for _, s := range []string{"198.18.0.1", "192.0.0.1", "240.0.0.1", "255.255.255.255", "2001:db8::1", "fec0::1", "fe80::1%eth0"} {
|
|
if !isReservedOrPrivate(netip.MustParseAddr(s)) {
|
|
t.Errorf("%s should be non-public", s)
|
|
}
|
|
}
|
|
for _, s := range []string{"8.8.8.8", "1.1.1.1", "2606:4700:4700::1111"} {
|
|
if isReservedOrPrivate(netip.MustParseAddr(s)) {
|
|
t.Errorf("%s should be public", s)
|
|
}
|
|
}
|
|
}
|