Method dialUpstreamAndBridgeWS

proxy/client.go:266–326

(downstreamConn net.Conn, host string, resp []byte, extra, dialStyle byte)

}

func (proxy *ProxyClient) dialUpstreamAndBridgeWS(downstreamConn net.Conn, host string, resp []byte, extra, dialStyle byte) net.Conn {
	upstreamConn, err := proxy.dialUpstream(dialStyle)
	if err != nil {
		proxy.Logger.E("Dial", "Error", err)
		if downstreamConn != nil {
			downstreamConn.Close()
		}
		return nil
	}

	r := proxy.Cipher.newRequest()
	r.Opt = Options(doConnect | doWebSocket | extra)
	r.Auth = proxy.UserAuth
	if proxy.Partial {
		r.Opt.Set(doPartial)
	}

	var pl buffer
	if proxy.URLHeader == "" {
		pl.Writes("GET /", proxy.encryptHost(host, r), " HTTP/1.1\r\nHost: ", proxy.genHost(), "\r\n")
	} else {
		pl.Writes("GET http://", proxy.Upstream, "/ HTTP/1.1\r\nHost: ", proxy.Upstream, "\r\n",
			proxy.URLHeader, ": http://", proxy.genHost(), "/", proxy.encryptHost(host, r), "\r\n")
	}

	wsKey := [20]byte{}
	proxy.Cipher.Rand.Read(wsKey[:])

	pl.Writes("Upgrade: websocket\r\nConnection: Upgrade\r\n",
		"Sec-WebSocket-Key: ", base64.StdEncoding.EncodeToString(wsKey[:]), "\r\nSec-WebSocket-Version: 13\r\n\r\n")

	upstreamConn.Write(pl.Bytes())

	buf, err := readUntil(upstreamConn, "\r\n\r\n")
	if err != nil || !strings.HasPrefix(string(buf), "HTTP/1.1 101 Switching Protocols") {
		if err != nil {
			proxy.Logger.E("Client", "Invalid response", host, err)
		}

		upstreamConn.Close()
		if downstreamConn != nil {
			downstreamConn.Close()
		}
		return nil
	}

	if extra == doMuxWS {
		// we return here and handle the connection to tcpmux
		return upstreamConn
	}

	if resp != nil {
		downstreamConn.Write(resp)
	}

	go proxy.Cipher.IO.Bridge(downstreamConn, upstreamConn, &r.iv, IOConfig{
		Partial: proxy.Partial,
		WSCtrl:  wsClient,