-
Notifications
You must be signed in to change notification settings - Fork 23
/
ipaddr.go
83 lines (72 loc) · 1.51 KB
/
ipaddr.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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package ipaddr
import "fmt"
import "net"
import "strings"
var privateNetworks []*net.IPNet
func Init() {
privateNetworks = make([]*net.IPNet, 0)
for _, cidr := range []string{"192.168.0.0/16", "172.16.0.0/12", "10.0.0.0/8"} {
_, network, err := net.ParseCIDR(cidr)
if err != nil {
panic(err)
}
privateNetworks = append(privateNetworks, network)
}
}
func IsPrivate(ipString string) bool {
if privateNetworks == nil {
Init()
}
ip := net.ParseIP(ipString)
for _, net := range privateNetworks {
if net.Contains(ip) {
return true
}
}
return false
}
func ParseCIDROrIP(s string) *net.IPNet {
// first try as network
_, network, err := net.ParseCIDR(s)
if err == nil {
return network
}
// else try as IP
_, network, err = net.ParseCIDR(s + "/32")
if err == nil {
return network
} else {
return nil
}
}
func ParseNetworks(s string) ([]*net.IPNet, error) {
var networks []*net.IPNet
for _, part := range strings.Split(s, ",") {
part = strings.TrimSpace(part)
if part != "" {
network := ParseCIDROrIP(s)
if network != nil {
networks = append(networks, network)
} else {
return nil, fmt.Errorf("failed to parse \"%s\" as IP/CIDR", part)
}
}
}
return networks, nil
}
func MatchNetworks(netString string, ipString string) bool {
ip := net.ParseIP(ipString)
if ip == nil {
return false
}
networks, err := ParseNetworks(netString)
if err != nil {
return false
}
for _, network := range networks {
if network.Contains(ip) {
return true
}
}
return false
}