Update main.go

main.go

@@ -14,20 +14,26 @@ func main() {
 	//modules.ImportDatabaseToHex()
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		//测试用ip地址
-		//ip := "103.47.100.182"
+		//ip := "192.168.2.134"
 		//获取访问浏览器ip
 		ip := strings.Split(r.RemoteAddr, ":")[0]
 		//fmt.Println("客户端IP地址:", ip)
-
 		if ip == "" {
 			w.Write([]byte("Invalid IP"))
+			return
 		}
-		IP_info = modules.GetIPInformation(ip)
-		log.Println("客户端ip：", ip)
-
+		if ip != "" {
+			IPinfo := modules.GetIPInformation(ip)
+			fmt.Println("IP", IPinfo.IP)
+			fmt.Println("IsPrivate", IPinfo.IsPrivate)
+			fmt.Println("PrivateCIDR", IPinfo.PrivateCIDR)
+			if IPinfo.IsPrivate == true {
+				w.Write([]byte("Invalid IP"))
+				return
+			}
+			if IPinfo.IsPrivate != true {
 				//使用mmdb查询ip
-		IpIformation := modules.MmdbQueryIP(ip)
-
+				IpIformation := modules.MmdbQueryIP(IPinfo.IP)
 				//使用MySql查询ip
 				//dsn := modules.Readconfig()
 				//IpIformation := modules.MysqlQuery(ip, dsn)
@@ -43,6 +49,8 @@ func main() {
 				// 返回 JSON 数据
 				//log.Println(jsonData)
 				w.Write(jsonData)
+			}
+		}
 	})
 	http.ListenAndServe(":8341", nil)
 }