From 1107d353e3f984d0e89679879f5375e6d9a89576 Mon Sep 17 00:00:00 2001
From: xiaoqidun <xiaoqidun@gmail.com>
Date: Thu, 22 Jan 2026 20:10:42 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E5=BC=BA=E5=8C=96=E6=A3=80=E6=9F=A5):=20?=
 =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E6=9B=B4=E5=A4=9A=E7=9A=84=E6=A3=80=E6=9F=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 probe_nat.go | 51 ++++++++++++++++++++++++++++++++++++---------------
 1 file changed, 36 insertions(+), 15 deletions(-)

diff --git a/probe_nat.go b/probe_nat.go
index cdd84b4..acd4c5e 100644
--- a/probe_nat.go
+++ b/probe_nat.go
@@ -195,28 +195,49 @@ func DetectNAT(conn net.PacketConn, primarySTUN, secondarySTUN, network string,
 			}
 		}
 	}
-	var mappedAddr3 *net.UDPAddr
+	var mappedAddr2 *net.UDPAddr
 	if targetSTUN2 != "" {
-		resp3, _, err := performTest(conn, targetSTUN2, network, timeout, false, false)
-		if err == nil && resp3 != nil {
-			mappedAddr3 = resp3.GetMappedAddress()
+		resp2, _, err := performTest(conn, targetSTUN2, network, timeout, false, false)
+		if err == nil && resp2 != nil {
+			mappedAddr2 = resp2.GetMappedAddress()
 		}
 	}
-	if mappedAddr3 != nil {
-		if mappedAddr1.String() == mappedAddr3.String() {
-			res.Mapping = MappingEndpointIndependent
-		} else {
-			res.Mapping = MappingAddressPortDependent
-		}
-	} else {
+	if mappedAddr2 == nil {
 		res.Mapping = MappingUnknown
+	} else if mappedAddr1.String() == mappedAddr2.String() {
+		res.Mapping = MappingEndpointIndependent
+	} else {
+		res.Mapping = MappingAddressPortDependent
+		changedAddr := resp1.GetChangedAddress()
+		if changedAddr != nil {
+			host, _, _ := net.SplitHostPort(primarySTUN)
+			primaryIP := net.ParseIP(host)
+			if primaryIP == nil {
+				if ipAddr, err := net.ResolveIPAddr("ip", host); err == nil {
+					primaryIP = ipAddr.IP
+				}
+			}
+			if primaryIP != nil && changedAddr.Port != 0 {
+				altPortSTUN := net.JoinHostPort(primaryIP.String(), strconv.Itoa(changedAddr.Port))
+				if altPortSTUN != primarySTUN {
+					resp3, _, err := performTest(conn, altPortSTUN, network, timeout, false, false)
+					if err == nil && resp3 != nil {
+						if mappedAddr3 := resp3.GetMappedAddress(); mappedAddr3 != nil {
+							if mappedAddr1.String() == mappedAddr3.String() {
+								res.Mapping = MappingAddressDependent
+							}
+						}
+					}
+				}
+			}
+		}
 	}
-	resp2, _, _ := performTest(conn, primarySTUN, network, timeout, true, true)
-	if resp2 != nil {
+	respF1, _, _ := performTest(conn, primarySTUN, network, timeout, true, true)
+	if respF1 != nil {
 		res.Filtering = FilteringEndpointIndependent
 	} else {
-		resp4, _, _ := performTest(conn, primarySTUN, network, timeout, false, true)
-		if resp4 != nil {
+		respF2, _, _ := performTest(conn, primarySTUN, network, timeout, false, true)
+		if respF2 != nil {
 			res.Filtering = FilteringAddressDependent
 		} else {
 			res.Filtering = FilteringAddressPortDependent