Fix panic on websocket handshake failure

2026-09-23 08:20:26 +00:00 · 2026-08-30 17:23:03 +08:00
parent 56f91dfeab
commit bd602b6d11
1 changed files with 5 additions and 1 deletions
@@ -81,7 +81,11 @@ func (c *Client) DialContext(ctx context.Context) (net.Conn, error) {
 	if c.maxEarlyData > 0 {
 		return &EarlyWebsocketConn{Client: c, rawConn: conn, create: make(chan struct{})}, nil
 	}
-	return c.upgrade(conn, &c.requestURL, c.headers)
+	websocketConn, err := c.upgrade(conn, &c.requestURL, c.headers)
+	if err != nil {
+		return nil, err
+	}
+	return websocketConn, nil
 }

 func (c *Client) upgrade(conn net.Conn, requestURL *url.URL, headers http.Header) (*WebsocketConn, error) {