update

isayme · May 10, 2024 · e0e6e2a · e0e6e2a
1 parent 8c0ab3c
commit e0e6e2a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/tunnel/tunnel.go b/tunnel/tunnel.go
@@ -35,7 +35,7 @@ func NewClient(tunnel string, password string) (Client, error) {
 	case "http2", "h2":
 		return h2.NewClient(tunnel, password)
 	case "ws", "wss":
-		return websocket.NewClient(tunnel)
+		return websocket.NewClient(tunnel, password)
 	}
 
 	return nil, fmt.Errorf("not supported schema: %s", URL.Scheme)
@@ -51,9 +51,9 @@ func NewServer(tunnel string, password string) (Server, error) {
 	case "grpc", "grpcs":
 		return grpc.NewServer(tunnel, password)
 	case "http2", "h2":
-		return h2.NewServer(tunnel)
+		return h2.NewServer(tunnel, password)
 	case "ws", "wss":
-		return websocket.NewServer(tunnel)
+		return websocket.NewServer(tunnel, password)
 	}
 	return nil, fmt.Errorf("not supported schema: %s", URL.Scheme)
 }
diff --git a/tunnel/websocket/client.go b/tunnel/websocket/client.go
@@ -14,7 +14,7 @@ type Client struct {
 	origin string
 }
 
-func NewClient(tunnel string) (*Client, error) {
+func NewClient(tunnel string, password string) (*Client, error) {
 	URL, err := url.Parse(tunnel)
 	if err != nil {
 		return nil, err

diff --git a/tunnel/websocket/server.go b/tunnel/websocket/server.go
@@ -14,7 +14,7 @@ type Server struct {
 	handler func(io.ReadWriter)
 }
 
-func NewServer(tunnel string) (*Server, error) {
+func NewServer(tunnel string, password string) (*Server, error) {
 	return &Server{
 		tunnel: tunnel,
 	}, nil