Method embedHandleHTTP

pkg/http/server.go:183–238

(conn net.Conn, req *http.Request, isConnectMethod bool)

}

func (s *Server) embedHandleHTTP(conn net.Conn, req *http.Request, isConnectMethod bool) error {
	defer func() {
		_ = conn.Close()
	}()

	targetAddr := req.URL.Host
	host, portStr, err := net.SplitHostPort(targetAddr)
	if err != nil {
		host = targetAddr
		if req.URL.Scheme == "https" || isConnectMethod {
			portStr = "443"
		} else {
			portStr = "80"
		}
		targetAddr = net.JoinHostPort(host, portStr)
	}

	target, err := s.ProxyDial(s.Context, "tcp", targetAddr)
	if err != nil {
		http.Error(
			NewHTTPResponseWriter(conn),
			err.Error(),
			http.StatusServiceUnavailable,
		)
		return err
	}
	defer func() {
		_ = target.Close()
	}()

	if isConnectMethod {
		_, err = conn.Write([]byte("HTTP/1.1 200 Connection Established\r\n\r\n"))
		if err != nil {
			return err
		}
	} else {
		err = req.Write(target)
		if err != nil {
			return err
		}
	}

	var buf1, buf2 []byte
	if s.BytesPool != nil {
		buf1 = s.BytesPool.Get()
		buf2 = s.BytesPool.Get()
		defer func() {
			s.BytesPool.Put(buf1)
			s.BytesPool.Put(buf2)
		}()
	} else {
		buf1 = make([]byte, 32*1024)
		buf2 = make([]byte, 32*1024)
	}
	return statute.Tunnel(s.Context, target, conn, buf1, buf2)
}