Merge pull request #94 from DOCCA0/main

ipv6本地地址上传图片报错
rocboss · Jun 12, 2022 · 087c6d2 · 087c6d2
2 parents 6b23c6e + 9cf6560
commit 087c6d2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/util/iploc/iploc.go b/pkg/util/iploc/iploc.go
@@ -21,7 +21,12 @@ const (
 
 // Find get country and city base ip
 func Find(ip string) (string, string) {
-	offset := searchIndex(binary.BigEndian.Uint32(net.ParseIP(ip).To4()))
+	// If ip is "::1", To4 returns nil.
+	to4 := net.ParseIP(ip).To4()
+	if to4 == nil {
+		to4 = net.ParseIP("127.0.0.1").To4()
+	}
+	offset := searchIndex(binary.BigEndian.Uint32(to4))
 	if offset <= 0 {
 		return "", ""
 	}