random tunnel path for ws/wss connections

Musixal · Nov 28, 2024 · 137d699 · 137d699
1 parent 75046d4
commit 137d699
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/internal/server/transport/ws.go b/internal/server/transport/ws.go
@@ -239,7 +239,6 @@ func (s *WsTransport) tunnelListener() {
 					go s.Restart()
 					return
 				}
-
 				s.controlChannel = conn
 
 				s.logger.Info("control channel established successfully")
@@ -260,7 +259,7 @@ func (s *WsTransport) tunnelListener() {
 
 				s.config.TunnelStatus = fmt.Sprintf("Connected (%s)", s.config.Mode)
 
-			} else if r.URL.Path == "/tunnel" {
+			} else if strings.HasPrefix(r.URL.Path, "/tunnel") {
 				wsConn := TunnelChannel{
 					conn: conn,
 					ping: make(chan struct{}),

diff --git a/internal/server/transport/wsmux.go b/internal/server/transport/wsmux.go
@@ -284,7 +284,7 @@ func (s *WsMuxTransport) tunnelListener() {
 
 				s.config.TunnelStatus = fmt.Sprintf("Connected (%s)", s.config.Mode)
 
-			} else if r.URL.Path == "/tunnel" {
+			} else if strings.HasPrefix(r.URL.Path, "/tunnel") {
 				session, err := smux.Client(conn.NetConn(), s.smuxConfig)
 				if err != nil {
 					s.logger.Errorf("failed to create MUX session for connection %s: %v", conn.RemoteAddr().String(), err)