2022-09-06 15:23:27 -04:00
|
|
|
package httprt
|
|
|
|
|
|
|
|
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"context"
|
|
|
|
"encoding/base64"
|
|
|
|
"io"
|
2023-05-29 19:42:50 -04:00
|
|
|
gonet "net"
|
2022-09-06 15:23:27 -04:00
|
|
|
"net/http"
|
|
|
|
|
2023-05-29 19:42:50 -04:00
|
|
|
"github.com/v2fly/v2ray-core/v5/transport/internet/transportcommon"
|
2022-09-06 15:23:27 -04:00
|
|
|
|
2023-05-29 19:42:50 -04:00
|
|
|
"github.com/v2fly/v2ray-core/v5/common"
|
2022-09-06 15:23:27 -04:00
|
|
|
"github.com/v2fly/v2ray-core/v5/common/net"
|
|
|
|
"github.com/v2fly/v2ray-core/v5/transport/internet/request"
|
|
|
|
)
|
|
|
|
|
|
|
|
func newHTTPRoundTripperClient(ctx context.Context, config *ClientConfig) request.RoundTripperClient {
|
|
|
|
return &httpTripperClient{config: config}
|
|
|
|
}
|
|
|
|
|
|
|
|
type httpTripperClient struct {
|
|
|
|
httpRTT http.RoundTripper
|
|
|
|
config *ClientConfig
|
|
|
|
assembly request.TransportClientAssembly
|
|
|
|
}
|
|
|
|
|
2023-05-29 19:42:50 -04:00
|
|
|
type unimplementedBackDrop struct {
|
|
|
|
}
|
|
|
|
|
|
|
|
func (u unimplementedBackDrop) RoundTrip(r *http.Request) (*http.Response, error) {
|
|
|
|
return nil, newError("unimplemented")
|
|
|
|
}
|
|
|
|
|
2022-09-06 15:23:27 -04:00
|
|
|
func (h *httpTripperClient) OnTransportClientAssemblyReady(assembly request.TransportClientAssembly) {
|
|
|
|
h.assembly = assembly
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTripperClient) RoundTrip(ctx context.Context, req request.Request, opts ...request.RoundTripperOption) (resp request.Response, err error) {
|
|
|
|
if h.httpRTT == nil {
|
2023-05-29 19:42:50 -04:00
|
|
|
h.httpRTT = transportcommon.NewALPNAwareHTTPRoundTripper(ctx, func(ctx context.Context, addr string) (gonet.Conn, error) {
|
|
|
|
return h.assembly.AutoImplDialer().Dial(ctx)
|
|
|
|
}, unimplementedBackDrop{})
|
2022-09-06 15:23:27 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
connectionTagStr := base64.RawURLEncoding.EncodeToString(req.ConnectionTag)
|
|
|
|
|
|
|
|
httpRequest, err := http.NewRequest("POST", h.config.Http.UrlPrefix+h.config.Http.Path, bytes.NewReader(req.Data))
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
httpRequest.Header.Set("X-Session-ID", connectionTagStr)
|
|
|
|
|
|
|
|
httpResp, err := h.httpRTT.RoundTrip(httpRequest)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
defer httpResp.Body.Close()
|
|
|
|
result, err := io.ReadAll(httpResp.Body)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
return request.Response{Data: result}, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func newHTTPRoundTripperServer(ctx context.Context, config *ServerConfig) request.RoundTripperServer {
|
|
|
|
return &httpTripperServer{ctx: ctx, config: config}
|
|
|
|
}
|
|
|
|
|
|
|
|
type httpTripperServer struct {
|
|
|
|
ctx context.Context
|
|
|
|
listener net.Listener
|
|
|
|
assembly request.TransportServerAssembly
|
|
|
|
|
|
|
|
listenAddress net.Addr
|
|
|
|
config *ServerConfig
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTripperServer) OnTransportServerAssemblyReady(assembly request.TransportServerAssembly) {
|
|
|
|
h.assembly = assembly
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTripperServer) ServeHTTP(writer http.ResponseWriter, r *http.Request) {
|
|
|
|
h.onRequest(writer, r)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTripperServer) onRequest(resp http.ResponseWriter, req *http.Request) {
|
|
|
|
tail := req.Header.Get("X-Session-ID")
|
|
|
|
data := []byte(tail)
|
|
|
|
if !h.config.NoDecodingSessionTag {
|
|
|
|
decodedData, err := base64.RawURLEncoding.DecodeString(tail)
|
|
|
|
if err != nil {
|
|
|
|
newError("unable to decode tag").Base(err).AtInfo().WriteToLog()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
data = decodedData
|
|
|
|
}
|
|
|
|
body, err := io.ReadAll(req.Body)
|
|
|
|
req.Body.Close()
|
|
|
|
if err != nil {
|
|
|
|
newError("unable to read body").Base(err).AtInfo().WriteToLog()
|
|
|
|
}
|
|
|
|
recvResp, err := h.assembly.TripperReceiver().OnRoundTrip(h.ctx, request.Request{Data: body, ConnectionTag: data})
|
|
|
|
if err != nil {
|
|
|
|
newError("unable to process roundtrip").Base(err).AtInfo().WriteToLog()
|
|
|
|
}
|
|
|
|
_, err = io.Copy(resp, bytes.NewReader(recvResp.Data))
|
|
|
|
if err != nil {
|
|
|
|
newError("unable to send response").Base(err).AtInfo().WriteToLog()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTripperServer) Start() error {
|
|
|
|
listener, err := h.assembly.AutoImplListener().Listen(h.ctx)
|
|
|
|
if err != nil {
|
|
|
|
return newError("unable to create a listener for http tripper server").Base(err)
|
|
|
|
}
|
|
|
|
h.listener = listener
|
|
|
|
go func() {
|
|
|
|
err := http.Serve(listener, h)
|
|
|
|
if err != nil {
|
|
|
|
newError("unable to serve listener for http tripper server").Base(err).WriteToLog()
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTripperServer) Close() error {
|
|
|
|
return h.listener.Close()
|
|
|
|
}
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
common.Must(common.RegisterConfig((*ClientConfig)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
|
|
|
|
clientConfig, ok := config.(*ClientConfig)
|
|
|
|
if !ok {
|
|
|
|
return nil, newError("not a ClientConfig")
|
|
|
|
}
|
|
|
|
return newHTTPRoundTripperClient(ctx, clientConfig), nil
|
|
|
|
}))
|
|
|
|
common.Must(common.RegisterConfig((*ServerConfig)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
|
|
|
|
serverConfig, ok := config.(*ServerConfig)
|
|
|
|
if !ok {
|
|
|
|
return nil, newError("not a ServerConfig")
|
|
|
|
}
|
|
|
|
return newHTTPRoundTripperServer(ctx, serverConfig), nil
|
|
|
|
}))
|
|
|
|
}
|