mihomo/transport/vmess/http.go

package vmess

import (
	"bufio"
	"bytes"
	"io"
	"net"
	"net/textproto"
	"net/url"

	"github.com/metacubex/http"
	"github.com/metacubex/randv2"
)

type httpConn struct {
	net.Conn
	cfg        *HTTPConfig
	reader     *bufio.Reader
	whandshake bool
}

type HTTPConfig struct {
	Method  string
	Host    string
	Path    []string
	Headers map[string][]string
}

// Read implements net.Conn.Read()
func (hc *httpConn) Read(b []byte) (int, error) {
	if hc.reader != nil {
		n, err := hc.reader.Read(b)
		return n, err
	}

	reader := textproto.NewConn(hc.Conn)
	// First line: GET /index.html HTTP/1.0
	if _, err := reader.ReadLine(); err != nil {
		return 0, err
	}

	if _, err := reader.ReadMIMEHeader(); err != nil {
		return 0, err
	}

	hc.reader = reader.R
	return reader.R.Read(b)
}

// Write implements io.Writer.
func (hc *httpConn) Write(b []byte) (int, error) {
	if hc.whandshake {
		return hc.Conn.Write(b)
	}

	path := "/"
	if len(hc.cfg.Path) > 0 {
		path = hc.cfg.Path[randv2.IntN(len(hc.cfg.Path))]
	}

	host := hc.cfg.Host
	if header := hc.cfg.Headers["Host"]; len(header) != 0 {
		host = header[randv2.IntN(len(header))]
	}

	req := http.Request{
		Method: hc.cfg.Method, // default is GET
		Host:   host,
		URL:    &url.URL{Scheme: "http", Host: host, Path: path},
		Header: make(http.Header),
		Body:   io.NopCloser(bytes.NewReader(b)),
	}
	for key, list := range hc.cfg.Headers {
		if len(list) > 0 {
			req.Header.Set(key, list[randv2.IntN(len(list))])
		}
	}
	req.ContentLength = int64(len(b))
	if err := req.Write(hc.Conn); err != nil {
		return 0, err
	}
	hc.whandshake = true
	return len(b), nil
}

func (hc *httpConn) Close() error {
	return hc.Conn.Close()
}

func StreamHTTPConn(conn net.Conn, cfg *HTTPConfig) net.Conn {
	return &httpConn{
		Conn: conn,
		cfg:  cfg,
	}
}