Fix host if we already had the port

elazarl · Dec 27, 2024 · 606639f · 606639f
1 parent 0d842e7
commit 606639f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/websocket.go b/websocket.go
@@ -38,6 +38,9 @@ func (proxy *ProxyHttpServer) serveWebsocketTLS(
 	host, port, err := net.SplitHostPort(req.URL.Host)
 	if err != nil || port == "" {
 		host = net.JoinHostPort(req.URL.Host, "443")
+	} else {
+		// We already had a port, just use it
+		host = req.URL.Host
 	}
 	targetURL := url.URL{Scheme: "wss", Host: host, Path: req.URL.Path}
 
@@ -70,6 +73,9 @@ func (proxy *ProxyHttpServer) serveWebsocketHttpOverTLS(
 	host, port, err := net.SplitHostPort(req.URL.Host)
 	if err != nil || port == "" {
 		host = net.JoinHostPort(req.URL.Host, "80")
+	} else {
+		// We already had a port, just use it
+		host = req.URL.Host
 	}
 	targetURL := url.URL{Scheme: "ws", Host: host, Path: req.URL.Path}