为ipv6直连和tcp直连做准备

utils/net.go

@@ -1,23 +1,68 @@
 package utils
 
 import (
+	"crypto/tls"
 	"fmt"
+	"io"
 	"net"
+	"net/http"
+	"strconv"
+	"strings"
+	"time"
 )
 
-func GetListenUDP() *net.UDPConn {
-	conn, err := net.ListenUDP("udp4", nil)
+func GetListenUDP(level string) *net.UDPConn {
+	conn, err := net.ListenUDP(level, nil)
 	if err != nil {
 		Log().ErrorF("绑定端口失败: %v", err)
 	}
 	return conn
 }
 
-func GetListenUDP2(port int) *net.UDPConn {
-	if addr, err := net.ResolveUDPAddr("udp4", fmt.Sprintf(":%d", port)); addr != nil && err == nil {
-		if conn, err := net.ListenUDP("udp4", addr); conn != nil && err == nil {
+func GetListenUDP2(level string, port int) *net.UDPConn {
+	if addr, err := net.ResolveUDPAddr(level, fmt.Sprintf(":%d", port)); addr != nil && err == nil {
+		if conn, err := net.ListenUDP(level, addr); conn != nil && err == nil {
 			return conn
 		}
 	}
 	return nil
 }
+
+func GetUDPLocalIPPort(level string) (string, int) {
+	conn, err := net.Dial(level, "ifconfig.co:80")
+	if err != nil {
+		return "", 0
+	}
+	defer conn.Close()
+
+	return conn.LocalAddr().(*net.UDPAddr).IP.String(), conn.LocalAddr().(*net.UDPAddr).Port
+}
+
+func GetTCPWanIPv4Port() (string, int) {
+	var res []byte
+	var err error
+	var resp *http.Response
+
+	client := &http.Client{
+		Transport: &http.Transport{
+			TLSClientConfig: &tls.Config{
+				InsecureSkipVerify: true, // 跳过证书验证
+			},
+		},
+		Timeout: 3 * time.Second,
+	}
+	if resp, err = client.Get("https://goodlink.kony.vip/ip"); err != nil {
+		return "", 0
+	}
+	defer resp.Body.Close()
+
+	if res, err = io.ReadAll(resp.Body); err != nil {
+		return "", 0
+	}
+
+	localAddr := string(res)
+
+	ip := strings.Split(localAddr, ":")[0]
+	port, _ := strconv.Atoi(strings.Split(localAddr, ":")[1])
+	return ip, port
+}

utils/tools.go

@@ -3,7 +3,6 @@ package utils
 import (
 	"crypto/rand"
 	"math/big"
-	"net"
 )
 
 func RandomBytes(length int) []byte {
@@ -19,14 +18,3 @@ func RandomBytes(length int) []byte {
 	}
 	return bytes
 }
-
-// 检测未使用的端口
-func GetFreeLocalAddr() string {
-	listener, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		return ""
-	}
-	defer listener.Close() // 确保在函数退出时关闭监听器
-
-	return listener.Addr().String()
-}