fix: tun udp with 4in6 ip

xishang0128 · Feb 13, 2023 · d00d83a · d00d83a
1 parent e6d16e4
commit d00d83a
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tunnel/connection.go b/tunnel/connection.go
@@ -44,11 +44,12 @@ func handleUDPToLocal(packet C.UDPPacket, pc net.PacketConn, key string, oAddr,
 		}
 
 		fromUDPAddr := from.(*net.UDPAddr)
-		if fAddr.IsValid() {
-			fromAddr, _ := netip.AddrFromSlice(fromUDPAddr.IP)
-			fromAddr.Unmap()
-			if oAddr == fromAddr {
-				fromUDPAddr.IP = fAddr.AsSlice()
+		fromUDPAddr = &(*fromUDPAddr) // make a copy
+		if fromAddr, ok := netip.AddrFromSlice(fromUDPAddr.IP); ok {
+			if fAddr.IsValid() && (oAddr.Unmap() == fromAddr.Unmap()) {
+				fromUDPAddr.IP = fAddr.Unmap().AsSlice()
+			} else {
+				fromUDPAddr.IP = fromAddr.Unmap().AsSlice()
 			}
 		}