client short circuit replace support ListenerConfig too

client/client.go

@@ -58,6 +58,15 @@ func (c *client) SetClientImpl(impl common.ClientImpl) {
 	c.ClientImpl = impl
 }
 
+func (c *client) GetListenerConfig() any {
+	return c.listenerConfig
+}
+
+func (c *client) SetListenerConfig(cfg any) {
+	c.listenerConfig = cfg.(listener.Config)
+	c.listenerConfig.IsWebSocketListener = false
+}
+
 func (c *client) GetServerWSPath() string {
 	return c.serverWSPath
 }
@@ -353,6 +362,8 @@ func StartClients() {
 						client.Addr(),
 						")")
 					newServer := _server.CloneWithNewAddress(client.Addr())
+					listenerConfig := client.GetListenerConfig()
+					newServer.SetListenerConfig(listenerConfig)
 					common.PortToServer[clientPort] = newServer
 					delete(common.PortToClient, clientPort) //It is safe in Golang!!!
 					continue

common/common.go

@@ -10,6 +10,7 @@ import (
 var PortToServer = make(map[string]Server)
 
 type Server interface {
+	HasListenerConfig
 	Start()
 	Addr() string
 	CloneWithNewAddress(bindAddress string) Server
@@ -19,6 +20,7 @@ var PortToClient = make(map[string]Client)
 
 type Client interface {
 	ClientImpl
+	HasListenerConfig
 	Start()
 	Addr() string
 	GetClientImpl() ClientImpl
@@ -38,3 +40,8 @@ type ClientConn interface {
 	TunnelTcp(tcp net.Conn)
 	TunnelWs(ws *websocket.Conn)
 }
+
+type HasListenerConfig interface {
+	GetListenerConfig() any
+	SetListenerConfig(cfg any)
+}

server/server.go

@@ -52,6 +52,15 @@ func (s *server) CloneWithNewAddress(bindAddress string) common.Server {
 	return &ns
 }
 
+func (s *server) GetListenerConfig() any {
+	return s.listenerConfig
+}
+
+func (s *server) SetListenerConfig(cfg any) {
+	s.listenerConfig = cfg.(listener.Config)
+	s.listenerConfig.IsWebSocketListener = true
+}
+
 type ServerHandler http.Handler
 
 type serverHandler struct {