split host port

This commit is contained in:
Zack Scholl 2020-10-05 08:16:34 -07:00
parent 16fef9e9f7
commit e145f6c61a
2 changed files with 6 additions and 1 deletions

View File

@ -278,7 +278,8 @@ var PrivateIPNetworks = []net.IPNet{
}
func IsLocalIP(ipaddress string) bool {
ip := net.ParseIP(ipaddress)
host, _, _ := net.SplitHostPort(ipaddress)
ip := net.ParseIP(host)
for _, ipNet := range PrivateIPNetworks {
if ipNet.Contains(ip) {
return true

View File

@ -191,3 +191,7 @@ func TestFindOpenPorts(t *testing.T) {
openPorts := FindOpenPorts("localhost", 9009, 4)
assert.Equal(t, []int{9009, 9010, 9011, 9012}, openPorts)
}
func TestIsLocalIP(t *testing.T) {
assert.True(t, IsLocalIP("192.168.0.14:9009"))
}