-
Notifications
You must be signed in to change notification settings - Fork 2
/
is_private_ip_test.go
63 lines (56 loc) · 3.29 KB
/
is_private_ip_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package realip
import (
"net"
"testing"
"github.com/stretchr/testify/assert"
)
func TestIsPrivateIpV4(t *testing.T) {
cases := []struct {
ip net.IP
isPrivate bool
msg string
}{
{ip: net.ParseIP("127.0.0.1"), isPrivate: true, msg: "127.0.0.1 should be private"},
{ip: net.ParseIP("192.168.254.254"), isPrivate: true, msg: "192.168.254.254 should be private"},
{ip: net.ParseIP("10.255.0.3"), isPrivate: true, msg: "10.255.0.3 should be private"},
{ip: net.ParseIP("172.16.255.255"), isPrivate: true, msg: "172.16.255.255 should be private"},
{ip: net.ParseIP("172.31.255.255"), isPrivate: true, msg: "172.31.255.255 should be private"},
{ip: net.ParseIP("192.169.255.255"), isPrivate: false, msg: "192.169.255.255 should not be private"},
{ip: net.ParseIP("9.255.0.255"), isPrivate: false, msg: "9.255.0.255 should not be private"},
{ip: net.ParseIP("67.22.74.181"), isPrivate: false, msg: "67.22.74.181 should not be private"},
{ip: net.ParseIP("109.86.51.55"), isPrivate: false, msg: "109.86.51.55 should not be private"},
{ip: net.ParseIP("76.233.86.21"), isPrivate: false, msg: "76.233.86.21 should not be private"},
{ip: net.ParseIP("109.42.112.5"), isPrivate: false, msg: "109.42.112.5 should not be private"},
{ip: net.ParseIP("81.185.175.46"), isPrivate: false, msg: "81.185.175.46 should not be private"},
}
for _, ts := range cases {
assert.Equal(t, ts.isPrivate, IsPrivateIp(ts.ip))
}
}
func TestIsPrivateIpV6(t *testing.T) {
cases := []struct {
ip net.IP
isPrivate bool
msg string
}{
{ip: net.ParseIP("::0"), isPrivate: true, msg: "::0 should be private"},
{ip: net.ParseIP("::1"), isPrivate: true, msg: "::1 should be private"},
{ip: net.ParseIP("fe80::1"), isPrivate: true, msg: "fe80::1 should be private"},
{ip: net.ParseIP("febf::1"), isPrivate: true, msg: "febf::1 should be private"},
{ip: net.ParseIP("ff00::1"), isPrivate: true, msg: "ff00::1 should be private"},
{ip: net.ParseIP("ff10::1"), isPrivate: true, msg: "ff10::1 should be private"},
{ip: net.ParseIP("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff"), isPrivate: true, msg: "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff should be private"},
{ip: net.ParseIP("2002::"), isPrivate: true, msg: "2002:: should be private"},
{ip: net.ParseIP("2002:ffff:ffff:ffff:ffff:ffff:ffff:ffff"), isPrivate: true, msg: "2002:ffff:ffff:ffff:ffff:ffff:ffff:ffff should be private"},
{ip: net.ParseIP("::2"), isPrivate: false, msg: "::2 should not be private"},
{ip: net.ParseIP("fec0::1"), isPrivate: false, msg: "fec0::1 should not be private"},
{ip: net.ParseIP("feff::1"), isPrivate: false, msg: "feff::1 should not be private"},
{ip: net.ParseIP("2409:4073:f:11b9:817a:81ea:a128:1955"), isPrivate: false, msg: "2409:4073:f:11b9:817a:81ea:a128:1955 should not be private"},
{ip: net.ParseIP("2401:4900:1a8d:3eaf:84b5:5538:efd:5586"), isPrivate: false, msg: "2401:4900:1a8d:3eaf:84b5:5538:efd:5586 should not be private"},
{ip: net.ParseIP("2409:4063:230c:e172:dd1a:4c74:de7c:cc21"), isPrivate: false, msg: "2409:4063:230c:e172:dd1a:4c74:de7c:cc21 should not be private"},
{ip: net.ParseIP("2601:240:a:1bc6:9c39:66f0:ced8:d643"), isPrivate: false, msg: "2601:240:a:1bc6:9c39:66f0:ced8:d643 should not be private"},
}
for _, ts := range cases {
assert.Equal(t, ts.isPrivate, IsPrivateIp(ts.ip))
}
}