Fix: hot recreate socks should ignore when receive same address

Mryange · Oct 13, 2019 · 521a190 · 521a190
1 parent e22ff74
commit 521a190
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/proxy/listener.go b/proxy/listener.go
@@ -74,20 +74,31 @@ func ReCreateHTTP(port int) error {
 func ReCreateSocks(port int) error {
 	addr := genAddr(bindAddress, port, allowLan)
 
+	shouldTCPIgnore := false
+	shouldUDPIgnore := false
+
 	if socksListener != nil {
 		if socksListener.Address() != addr {
 			socksListener.Close()
 			socksListener = nil
+		} else {
+			shouldTCPIgnore = true
 		}
 	}
 
 	if socksUDPListener != nil {
 		if socksUDPListener.Address() != addr {
 			socksUDPListener.Close()
 			socksUDPListener = nil
+		} else {
+			shouldUDPIgnore = true
 		}
 	}
 
+	if shouldTCPIgnore && shouldUDPIgnore {
+		return nil
+	}
+
 	if portIsZero(addr) {
 		return nil
 	}