shuguang/nezha
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

♻️ 优化 IPv4 / IPv6 地址判断

Browse Source
This commit is contained in:
naiba 2021-06-22 17:28:28 +08:00
parent 365148e0ac
commit d20f33af5f
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/utils/http.go
View File

@ -71,14 +71,15 @@ func resolveIP(addr string, ipv6 bool) (string, error) {
var ipv4Resolved, ipv6Resolved bool var ipv4Resolved, ipv6Resolved bool
for i := 0; i < len(res); i++ { for i := 0; i < len(res); i++ {
if ip4 := res[i].To4(); ip4 != nil && !ipv6 { ip := res[i].String()
if strings.Contains(ip, ".") && !ipv6 {
ipv4Resolved = true ipv4Resolved = true
url[0] = ip4.String() url[0] = ip
break break
} }
if ip6 := res[i].To16(); ip6 != nil && ipv6 { if strings.Contains(ip, ":") && ipv6 {
ipv6Resolved = true ipv6Resolved = true
url[0] = "[" + ip6.String() + "]" url[0] = "[" + ip + "]"
break break
} }
} }