Fix: ip version check

This commit is contained in:
Dreamacro
2019-09-27 15:26:07 +08:00
parent 045c3a3ad4
commit c38469330d
2 changed files with 6 additions and 5 deletions

View File

@ -37,8 +37,8 @@ func ResolveIPv4(host string) (net.IP, error) {
}
for _, ip := range ipAddrs {
if len(ip) == net.IPv4len {
return ip, nil
if ip4 := ip.To4(); ip4 != nil {
return ip4, nil
}
}
@ -71,7 +71,7 @@ func ResolveIPv6(host string) (net.IP, error) {
}
for _, ip := range ipAddrs {
if len(ip) == net.IPv6len {
if ip.To4() == nil {
return ip, nil
}
}