go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-03-17 21:07:54 +02:00
Code Issues Releases Activity

host: optimize isPrivateIP (#817)

Browse Source
This commit is contained in:
Cluas 2021-04-08 17:45:11 +08:00 committed by GitHub
parent afd2a6b19a
commit e1b9dd5244
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 39 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
internal/host/host.go
View File

@ -5,20 +5,24 @@ import (
"strconv"
)
var (
privateAddrs = []string{"10.0.0.0/8", "172.16.0.0/12", "192.168.0.0/16", "100.64.0.0/10", "fd00::/8"}
)
func isPrivateIP(addr string) bool {
ipAddr := net.ParseIP(addr)
for _, privateAddr := range privateAddrs {
if _, priv, err := net.ParseCIDR(privateAddr); err == nil {
if priv.Contains(ipAddr) {
return true
}
}
ip := net.ParseIP(addr)
if ip4 := ip.To4(); ip4 != nil {
// Following RFC 4193, Section 3. Local IPv6 Unicast Addresses which says:
// The Internet Assigned Numbers Authority (IANA) has reserved the
// following three blocks of the IPv4 address space for private internets:
// 10.0.0.0 - 10.255.255.255 (10/8 prefix)
// 172.16.0.0 - 172.31.255.255 (172.16/12 prefix)
// 192.168.0.0 - 192.168.255.255 (192.168/16 prefix)
return ip4[0] == 10 ||
(ip4[0] == 172 && ip4[1]&0xf0 == 16) ||
(ip4[0] == 192 && ip4[1] == 168)
}
return false
// Following RFC 4193, Section 3. Private Address Space which says:
// The Internet Assigned Numbers Authority (IANA) has reserved the
// following block of the IPv6 address space for local internets:
// FC00:: - FDFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF (FC00::/7 prefix)
return len(ip) == net.IPv6len && ip[0]&0xfe == 0xfc
}
// Port return a real port.
@ -30,8 +34,8 @@ func Port(lis net.Listener) (int, bool) {
}
// Extract returns a private addr and port.
func Extract(hostport string, lis net.Listener) (string, error) {
addr, port, err := net.SplitHostPort(hostport)
func Extract(hostPort string, lis net.Listener) (string, error) {
addr, port, err := net.SplitHostPort(hostPort)
if err != nil {
return "", err
}

21
internal/host/host_test.go
View File

@ -14,6 +14,27 @@ func TestPrivateIP(t *testing.T) {
{"172.16.0.1", true},
{"192.168.1.1", true},
{"8.8.8.8", false},
{"1.1.1.1", false},
{"9.255.255.255", false},
{"10.0.0.0", true},
{"10.255.255.255", true},
{"11.0.0.0", false},
{"172.15.255.255", false},
{"172.16.0.0", true},
{"172.16.255.255", true},
{"172.23.18.255", true},
{"172.31.255.255", true},
{"172.31.0.0", true},
{"172.32.0.0", false},
{"192.167.255.255", false},
{"192.168.0.0", true},
{"192.168.255.255", true},
{"192.169.0.0", false},
{"fbff:ffff:ffff:ffff:ffff:ffff:ffff:ffff", false},
{"fc00::", true},
{"fcff:1200:0:44::", true},
{"fdff:ffff:ffff:ffff:ffff:ffff:ffff:ffff", true},
{"fe00::", false},
}
for _, test := range tests {
t.Run(test.addr, func(t *testing.T) {