Fix parsing UDP transport

SagerNet · Feb 22, 2024 · 498f63b · 498f63b
1 parent 9a328d8
commit 498f63b
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/transport_udp.go b/transport_udp.go
@@ -28,15 +28,12 @@ type UDPTransport struct {
 }
 
 func NewUDPTransport(options TransportOptions) (*UDPTransport, error) {
-	serverURL, err := url.Parse(options.Address)
-	if err != nil {
-		return nil, err
-	}
-	if serverURL.Scheme == "" {
-		serverURL.Host = serverURL.Path
-		serverURL.Path = ""
+	var serverAddr M.Socksaddr
+	if serverURL, err := url.Parse(options.Address); err != nil || serverURL.Scheme == "" {
+		serverAddr = M.ParseSocksaddr(options.Address)
+	} else {
+		serverAddr = M.ParseSocksaddr(serverURL.Host)
 	}
-	serverAddr := M.ParseSocksaddr(serverURL.Host)
 	if !serverAddr.IsValid() {
 		return nil, E.New("invalid server address")
 	}