From cc878be23c96de1c4cb80b53019f6d97a073dd33 Mon Sep 17 00:00:00 2001
From: RPRX <63339210+RPRX@users.noreply.github.com>
Date: Tue, 26 May 2026 08:44:21 +0000
Subject: [PATCH] Update temp_udp_listen.go

---
 proxy/socks/temp_udp_listen.go | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/proxy/socks/temp_udp_listen.go b/proxy/socks/temp_udp_listen.go
index 3c3769d5..d7aa2696 100644
--- a/proxy/socks/temp_udp_listen.go
+++ b/proxy/socks/temp_udp_listen.go
@@ -38,14 +38,11 @@ func (c *TempUDPConn) Read(b []byte) (n int, err error) {
 			break
 		}
 		if c.remote.Load() == nil {
-			udpRemote, _, _ := net.SplitHostPort(remote.String())
-			if c.ExpectedRemoteIP != udpRemote {
-				continue
-			} else {
+			if remoteIP, _, _ := net.SplitHostPort(remote.String()); remoteIP == c.ExpectedRemoteIP {
 				c.remote.CompareAndSwap(nil, &remote)
+				break
 			}
-		}
-		if remote.String() == c.remote.Load().String() {
+		} else if remote.String() == (*c.remote.Load()).String() {
 			break
 		}
 	}