add request transport base

This commit is contained in:
Shelikhoo 2022-09-06 20:23:27 +01:00
parent 69ab87239a
commit f68115c37e
No known key found for this signature in database
GPG Key ID: C4D5E79D22B25316
25 changed files with 1844 additions and 0 deletions

View File

@ -65,6 +65,12 @@ import (
_ "github.com/v2fly/v2ray-core/v5/transport/internet/udp"
_ "github.com/v2fly/v2ray-core/v5/transport/internet/websocket"
// Developer preview transports
_ "github.com/v2fly/v2ray-core/v5/transport/internet/request/assembly"
_ "github.com/v2fly/v2ray-core/v5/transport/internet/request/assembler/simple"
_ "github.com/v2fly/v2ray-core/v5/transport/internet/request/roundtripper/httprt"
// Transport headers
_ "github.com/v2fly/v2ray-core/v5/transport/internet/headers/http"
_ "github.com/v2fly/v2ray-core/v5/transport/internet/headers/noop"

View File

@ -0,0 +1,22 @@
package request
import (
"io"
)
type SessionAssemblerClient interface {
SessionCreator
TransportClientAssemblyReceiver
}
type SessionAssemblerServer interface {
TripperReceiver
TransportServerAssemblyReceiver
}
type SessionOption interface {
RoundTripperOption()
}
type Session interface {
io.ReadWriteCloser
}

View File

@ -0,0 +1,173 @@
package simple
import (
"bytes"
"context"
"crypto/rand"
"io"
"time"
"github.com/v2fly/v2ray-core/v5/common"
"github.com/v2fly/v2ray-core/v5/transport/internet/request"
)
func newClient(config *ClientConfig) request.SessionAssemblerClient {
return &simpleAssemblerClient{config: config}
}
type simpleAssemblerClient struct {
assembly request.TransportClientAssembly
config *ClientConfig
}
func (s *simpleAssemblerClient) OnTransportClientAssemblyReady(assembly request.TransportClientAssembly) {
s.assembly = assembly
}
func (s *simpleAssemblerClient) NewSession(ctx context.Context, opts ...request.SessionOption) (request.Session, error) {
sessionID := make([]byte, 16)
_, err := io.ReadFull(rand.Reader, sessionID)
if err != nil {
return nil, err
}
sessionContext, finish := context.WithCancel(ctx)
session := &simpleAssemblerClientSession{
sessionID: sessionID, tripper: s.assembly.Tripper(), readBuffer: bytes.NewBuffer(nil),
ctx: sessionContext, finish: finish, writerChan: make(chan []byte), readerChan: make(chan []byte, 16), assembler: s}
go session.keepRunning()
return session, nil
}
type simpleAssemblerClientSession struct {
sessionID []byte
currentWriteWait int
assembler *simpleAssemblerClient
tripper request.Tripper
readBuffer *bytes.Buffer
writerChan chan []byte
readerChan chan []byte
ctx context.Context
finish func()
}
func (s *simpleAssemblerClientSession) keepRunning() {
s.currentWriteWait = int(s.assembler.config.InitialPollingIntervalMs)
for s.ctx.Err() == nil {
s.runOnce()
}
}
func (s *simpleAssemblerClientSession) runOnce() {
sendBuffer := bytes.NewBuffer(nil)
if s.currentWriteWait != 0 {
waitTimer := time.NewTimer(time.Millisecond * time.Duration(s.currentWriteWait))
waitForFirstWrite := true
copyFromWriterLoop:
for {
select {
case <-s.ctx.Done():
return
case data := <-s.writerChan:
sendBuffer.Write(data)
if sendBuffer.Len() >= int(s.assembler.config.MaxWriteSize) {
break copyFromWriterLoop
}
if waitForFirstWrite {
waitForFirstWrite = false
waitTimer.Reset(time.Millisecond * time.Duration(s.assembler.config.WaitSubsequentWriteMs))
}
case <-waitTimer.C:
break copyFromWriterLoop
}
}
waitTimer.Stop()
}
firstRound := true
pollConnection := true
for sendBuffer.Len() != 0 || firstRound {
firstRound = false
sendAmount := sendBuffer.Len()
if sendAmount > int(s.assembler.config.MaxWriteSize) {
sendAmount = int(s.assembler.config.MaxWriteSize)
}
data := sendBuffer.Next(sendAmount)
if len(data) != 0 {
pollConnection = false
}
for {
resp, err := s.tripper.RoundTrip(s.ctx, request.Request{Data: data, ConnectionTag: s.sessionID})
if err != nil {
newError("failed to send data").Base(err).WriteToLog()
if s.ctx.Err() != nil {
return
}
time.Sleep(time.Millisecond * time.Duration(s.assembler.config.FailedRetryIntervalMs))
continue
}
if len(resp.Data) != 0 {
s.readerChan <- resp.Data
}
if len(resp.Data) != 0 {
pollConnection = false
}
break
}
}
if pollConnection {
s.currentWriteWait = int(s.assembler.config.BackoffFactor * float32(s.currentWriteWait))
if s.currentWriteWait > int(s.assembler.config.MaxPollingIntervalMs) {
s.currentWriteWait = int(s.assembler.config.MaxPollingIntervalMs)
}
if s.currentWriteWait < int(s.assembler.config.MinPollingIntervalMs) {
s.currentWriteWait = int(s.assembler.config.MinPollingIntervalMs)
}
} else {
s.currentWriteWait = int(0)
}
}
func (s *simpleAssemblerClientSession) Read(p []byte) (n int, err error) {
if s.readBuffer.Len() == 0 {
select {
case <-s.ctx.Done():
return 0, s.ctx.Err()
case data := <-s.readerChan:
s.readBuffer.Write(data)
}
}
n, err = s.readBuffer.Read(p)
if err == io.EOF {
s.readBuffer.Reset()
return 0, nil
}
return
}
func (s *simpleAssemblerClientSession) Write(p []byte) (n int, err error) {
buf := make([]byte, len(p))
copy(buf, p)
select {
case <-s.ctx.Done():
return 0, s.ctx.Err()
case s.writerChan <- buf:
return len(p), nil
}
}
func (s *simpleAssemblerClientSession) Close() error {
s.finish()
return nil
}
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 newClient(clientConfig), nil
}))
}

View File

@ -0,0 +1,295 @@
package simple
import (
_ "github.com/v2fly/v2ray-core/v5/common/protoext"
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
reflect "reflect"
sync "sync"
)
const (
// Verify that this generated code is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
// Verify that runtime/protoimpl is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)
type ClientConfig struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
MaxWriteSize int32 `protobuf:"varint,1,opt,name=max_write_size,json=maxWriteSize,proto3" json:"max_write_size,omitempty"`
WaitSubsequentWriteMs int32 `protobuf:"varint,2,opt,name=wait_subsequent_write_ms,json=waitSubsequentWriteMs,proto3" json:"wait_subsequent_write_ms,omitempty"`
InitialPollingIntervalMs int32 `protobuf:"varint,3,opt,name=initial_polling_interval_ms,json=initialPollingIntervalMs,proto3" json:"initial_polling_interval_ms,omitempty"`
MaxPollingIntervalMs int32 `protobuf:"varint,4,opt,name=max_polling_interval_ms,json=maxPollingIntervalMs,proto3" json:"max_polling_interval_ms,omitempty"`
MinPollingIntervalMs int32 `protobuf:"varint,5,opt,name=min_polling_interval_ms,json=minPollingIntervalMs,proto3" json:"min_polling_interval_ms,omitempty"`
BackoffFactor float32 `protobuf:"fixed32,6,opt,name=backoff_factor,json=backoffFactor,proto3" json:"backoff_factor,omitempty"`
FailedRetryIntervalMs int32 `protobuf:"varint,7,opt,name=failed_retry_interval_ms,json=failedRetryIntervalMs,proto3" json:"failed_retry_interval_ms,omitempty"`
}
func (x *ClientConfig) Reset() {
*x = ClientConfig{}
if protoimpl.UnsafeEnabled {
mi := &file_transport_internet_request_assembler_simple_config_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ClientConfig) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ClientConfig) ProtoMessage() {}
func (x *ClientConfig) ProtoReflect() protoreflect.Message {
mi := &file_transport_internet_request_assembler_simple_config_proto_msgTypes[0]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use ClientConfig.ProtoReflect.Descriptor instead.
func (*ClientConfig) Descriptor() ([]byte, []int) {
return file_transport_internet_request_assembler_simple_config_proto_rawDescGZIP(), []int{0}
}
func (x *ClientConfig) GetMaxWriteSize() int32 {
if x != nil {
return x.MaxWriteSize
}
return 0
}
func (x *ClientConfig) GetWaitSubsequentWriteMs() int32 {
if x != nil {
return x.WaitSubsequentWriteMs
}
return 0
}
func (x *ClientConfig) GetInitialPollingIntervalMs() int32 {
if x != nil {
return x.InitialPollingIntervalMs
}
return 0
}
func (x *ClientConfig) GetMaxPollingIntervalMs() int32 {
if x != nil {
return x.MaxPollingIntervalMs
}
return 0
}
func (x *ClientConfig) GetMinPollingIntervalMs() int32 {
if x != nil {
return x.MinPollingIntervalMs
}
return 0
}
func (x *ClientConfig) GetBackoffFactor() float32 {
if x != nil {
return x.BackoffFactor
}
return 0
}
func (x *ClientConfig) GetFailedRetryIntervalMs() int32 {
if x != nil {
return x.FailedRetryIntervalMs
}
return 0
}
type ServerConfig struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
MaxWriteSize int32 `protobuf:"varint,1,opt,name=max_write_size,json=maxWriteSize,proto3" json:"max_write_size,omitempty"`
}
func (x *ServerConfig) Reset() {
*x = ServerConfig{}
if protoimpl.UnsafeEnabled {
mi := &file_transport_internet_request_assembler_simple_config_proto_msgTypes[1]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ServerConfig) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ServerConfig) ProtoMessage() {}
func (x *ServerConfig) ProtoReflect() protoreflect.Message {
mi := &file_transport_internet_request_assembler_simple_config_proto_msgTypes[1]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use ServerConfig.ProtoReflect.Descriptor instead.
func (*ServerConfig) Descriptor() ([]byte, []int) {
return file_transport_internet_request_assembler_simple_config_proto_rawDescGZIP(), []int{1}
}
func (x *ServerConfig) GetMaxWriteSize() int32 {
if x != nil {
return x.MaxWriteSize
}
return 0
}
var File_transport_internet_request_assembler_simple_config_proto protoreflect.FileDescriptor
var file_transport_internet_request_assembler_simple_config_proto_rawDesc = []byte{
0x0a, 0x38, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2f, 0x69, 0x6e, 0x74, 0x65,
0x72, 0x6e, 0x65, 0x74, 0x2f, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2f, 0x61, 0x73, 0x73,
0x65, 0x6d, 0x62, 0x6c, 0x65, 0x72, 0x2f, 0x73, 0x69, 0x6d, 0x70, 0x6c, 0x65, 0x2f, 0x63, 0x6f,
0x6e, 0x66, 0x69, 0x67, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x36, 0x76, 0x32, 0x72, 0x61,
0x79, 0x2e, 0x63, 0x6f, 0x72, 0x65, 0x2e, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74,
0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65, 0x74, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73,
0x74, 0x2e, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x65, 0x72, 0x2e, 0x73, 0x69, 0x6d, 0x70,
0x6c, 0x65, 0x1a, 0x20, 0x63, 0x6f, 0x6d, 0x6d, 0x6f, 0x6e, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f,
0x65, 0x78, 0x74, 0x2f, 0x65, 0x78, 0x74, 0x65, 0x6e, 0x73, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70,
0x72, 0x6f, 0x74, 0x6f, 0x22, 0xac, 0x03, 0x0a, 0x0c, 0x43, 0x6c, 0x69, 0x65, 0x6e, 0x74, 0x43,
0x6f, 0x6e, 0x66, 0x69, 0x67, 0x12, 0x24, 0x0a, 0x0e, 0x6d, 0x61, 0x78, 0x5f, 0x77, 0x72, 0x69,
0x74, 0x65, 0x5f, 0x73, 0x69, 0x7a, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x05, 0x52, 0x0c, 0x6d,
0x61, 0x78, 0x57, 0x72, 0x69, 0x74, 0x65, 0x53, 0x69, 0x7a, 0x65, 0x12, 0x37, 0x0a, 0x18, 0x77,
0x61, 0x69, 0x74, 0x5f, 0x73, 0x75, 0x62, 0x73, 0x65, 0x71, 0x75, 0x65, 0x6e, 0x74, 0x5f, 0x77,
0x72, 0x69, 0x74, 0x65, 0x5f, 0x6d, 0x73, 0x18, 0x02, 0x20, 0x01, 0x28, 0x05, 0x52, 0x15, 0x77,
0x61, 0x69, 0x74, 0x53, 0x75, 0x62, 0x73, 0x65, 0x71, 0x75, 0x65, 0x6e, 0x74, 0x57, 0x72, 0x69,
0x74, 0x65, 0x4d, 0x73, 0x12, 0x3d, 0x0a, 0x1b, 0x69, 0x6e, 0x69, 0x74, 0x69, 0x61, 0x6c, 0x5f,
0x70, 0x6f, 0x6c, 0x6c, 0x69, 0x6e, 0x67, 0x5f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61, 0x6c,
0x5f, 0x6d, 0x73, 0x18, 0x03, 0x20, 0x01, 0x28, 0x05, 0x52, 0x18, 0x69, 0x6e, 0x69, 0x74, 0x69,
0x61, 0x6c, 0x50, 0x6f, 0x6c, 0x6c, 0x69, 0x6e, 0x67, 0x49, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61,
0x6c, 0x4d, 0x73, 0x12, 0x35, 0x0a, 0x17, 0x6d, 0x61, 0x78, 0x5f, 0x70, 0x6f, 0x6c, 0x6c, 0x69,
0x6e, 0x67, 0x5f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61, 0x6c, 0x5f, 0x6d, 0x73, 0x18, 0x04,
0x20, 0x01, 0x28, 0x05, 0x52, 0x14, 0x6d, 0x61, 0x78, 0x50, 0x6f, 0x6c, 0x6c, 0x69, 0x6e, 0x67,
0x49, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61, 0x6c, 0x4d, 0x73, 0x12, 0x35, 0x0a, 0x17, 0x6d, 0x69,
0x6e, 0x5f, 0x70, 0x6f, 0x6c, 0x6c, 0x69, 0x6e, 0x67, 0x5f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x76,
0x61, 0x6c, 0x5f, 0x6d, 0x73, 0x18, 0x05, 0x20, 0x01, 0x28, 0x05, 0x52, 0x14, 0x6d, 0x69, 0x6e,
0x50, 0x6f, 0x6c, 0x6c, 0x69, 0x6e, 0x67, 0x49, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61, 0x6c, 0x4d,
0x73, 0x12, 0x25, 0x0a, 0x0e, 0x62, 0x61, 0x63, 0x6b, 0x6f, 0x66, 0x66, 0x5f, 0x66, 0x61, 0x63,
0x74, 0x6f, 0x72, 0x18, 0x06, 0x20, 0x01, 0x28, 0x02, 0x52, 0x0d, 0x62, 0x61, 0x63, 0x6b, 0x6f,
0x66, 0x66, 0x46, 0x61, 0x63, 0x74, 0x6f, 0x72, 0x12, 0x37, 0x0a, 0x18, 0x66, 0x61, 0x69, 0x6c,
0x65, 0x64, 0x5f, 0x72, 0x65, 0x74, 0x72, 0x79, 0x5f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61,
0x6c, 0x5f, 0x6d, 0x73, 0x18, 0x07, 0x20, 0x01, 0x28, 0x05, 0x52, 0x15, 0x66, 0x61, 0x69, 0x6c,
0x65, 0x64, 0x52, 0x65, 0x74, 0x72, 0x79, 0x49, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61, 0x6c, 0x4d,
0x73, 0x3a, 0x30, 0x82, 0xb5, 0x18, 0x2c, 0x0a, 0x22, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f,
0x72, 0x74, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x61, 0x73, 0x73, 0x65, 0x6d,
0x62, 0x6c, 0x65, 0x72, 0x2e, 0x63, 0x6c, 0x69, 0x65, 0x6e, 0x74, 0x12, 0x06, 0x73, 0x69, 0x6d,
0x70, 0x6c, 0x65, 0x22, 0x66, 0x0a, 0x0c, 0x53, 0x65, 0x72, 0x76, 0x65, 0x72, 0x43, 0x6f, 0x6e,
0x66, 0x69, 0x67, 0x12, 0x24, 0x0a, 0x0e, 0x6d, 0x61, 0x78, 0x5f, 0x77, 0x72, 0x69, 0x74, 0x65,
0x5f, 0x73, 0x69, 0x7a, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x05, 0x52, 0x0c, 0x6d, 0x61, 0x78,
0x57, 0x72, 0x69, 0x74, 0x65, 0x53, 0x69, 0x7a, 0x65, 0x3a, 0x30, 0x82, 0xb5, 0x18, 0x2c, 0x0a,
0x22, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65,
0x73, 0x74, 0x2e, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x65, 0x72, 0x2e, 0x73, 0x65, 0x72,
0x76, 0x65, 0x72, 0x12, 0x06, 0x73, 0x69, 0x6d, 0x70, 0x6c, 0x65, 0x42, 0xc3, 0x01, 0x0a, 0x3a,
0x63, 0x6f, 0x6d, 0x2e, 0x76, 0x32, 0x72, 0x61, 0x79, 0x2e, 0x63, 0x6f, 0x72, 0x65, 0x2e, 0x74,
0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65,
0x74, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62,
0x6c, 0x65, 0x72, 0x2e, 0x73, 0x69, 0x6d, 0x70, 0x6c, 0x65, 0x50, 0x01, 0x5a, 0x4a, 0x67, 0x69,
0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x76, 0x32, 0x66, 0x6c, 0x79, 0x2f, 0x76,
0x32, 0x72, 0x61, 0x79, 0x2d, 0x63, 0x6f, 0x72, 0x65, 0x2f, 0x76, 0x35, 0x2f, 0x74, 0x72, 0x61,
0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65, 0x74, 0x2f,
0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2f, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x65,
0x72, 0x2f, 0x73, 0x69, 0x6d, 0x70, 0x6c, 0x65, 0xaa, 0x02, 0x36, 0x56, 0x32, 0x52, 0x61, 0x79,
0x2e, 0x43, 0x6f, 0x72, 0x65, 0x2e, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2e,
0x49, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65, 0x74, 0x2e, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74,
0x2e, 0x41, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x65, 0x72, 0x2e, 0x53, 0x69, 0x6d, 0x70, 0x6c,
0x65, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_transport_internet_request_assembler_simple_config_proto_rawDescOnce sync.Once
file_transport_internet_request_assembler_simple_config_proto_rawDescData = file_transport_internet_request_assembler_simple_config_proto_rawDesc
)
func file_transport_internet_request_assembler_simple_config_proto_rawDescGZIP() []byte {
file_transport_internet_request_assembler_simple_config_proto_rawDescOnce.Do(func() {
file_transport_internet_request_assembler_simple_config_proto_rawDescData = protoimpl.X.CompressGZIP(file_transport_internet_request_assembler_simple_config_proto_rawDescData)
})
return file_transport_internet_request_assembler_simple_config_proto_rawDescData
}
var file_transport_internet_request_assembler_simple_config_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
var file_transport_internet_request_assembler_simple_config_proto_goTypes = []interface{}{
(*ClientConfig)(nil), // 0: v2ray.core.transport.internet.request.assembler.simple.ClientConfig
(*ServerConfig)(nil), // 1: v2ray.core.transport.internet.request.assembler.simple.ServerConfig
}
var file_transport_internet_request_assembler_simple_config_proto_depIdxs = []int32{
0, // [0:0] is the sub-list for method output_type
0, // [0:0] is the sub-list for method input_type
0, // [0:0] is the sub-list for extension type_name
0, // [0:0] is the sub-list for extension extendee
0, // [0:0] is the sub-list for field type_name
}
func init() { file_transport_internet_request_assembler_simple_config_proto_init() }
func file_transport_internet_request_assembler_simple_config_proto_init() {
if File_transport_internet_request_assembler_simple_config_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_transport_internet_request_assembler_simple_config_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*ClientConfig); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_transport_internet_request_assembler_simple_config_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*ServerConfig); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
type x struct{}
out := protoimpl.TypeBuilder{
File: protoimpl.DescBuilder{
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_transport_internet_request_assembler_simple_config_proto_rawDesc,
NumEnums: 0,
NumMessages: 2,
NumExtensions: 0,
NumServices: 0,
},
GoTypes: file_transport_internet_request_assembler_simple_config_proto_goTypes,
DependencyIndexes: file_transport_internet_request_assembler_simple_config_proto_depIdxs,
MessageInfos: file_transport_internet_request_assembler_simple_config_proto_msgTypes,
}.Build()
File_transport_internet_request_assembler_simple_config_proto = out.File
file_transport_internet_request_assembler_simple_config_proto_rawDesc = nil
file_transport_internet_request_assembler_simple_config_proto_goTypes = nil
file_transport_internet_request_assembler_simple_config_proto_depIdxs = nil
}

View File

@ -0,0 +1,29 @@
syntax = "proto3";
package v2ray.core.transport.internet.request.assembler.simple;
option csharp_namespace = "V2Ray.Core.Transport.Internet.Request.Assembler.Simple";
option go_package = "github.com/v2fly/v2ray-core/v5/transport/internet/request/assembler/simple";
option java_package = "com.v2ray.core.transport.internet.request.assembler.simple";
option java_multiple_files = true;
import "common/protoext/extensions.proto";
message ClientConfig {
option (v2ray.core.common.protoext.message_opt).type = "transport.request.assembler.client";
option (v2ray.core.common.protoext.message_opt).short_name = "simple";
int32 max_write_size = 1;
int32 wait_subsequent_write_ms = 2;
int32 initial_polling_interval_ms = 3;
int32 max_polling_interval_ms = 4;
int32 min_polling_interval_ms = 5;
float backoff_factor = 6;
int32 failed_retry_interval_ms = 7;
}
message ServerConfig {
option (v2ray.core.common.protoext.message_opt).type = "transport.request.assembler.server";
option (v2ray.core.common.protoext.message_opt).short_name = "simple";
int32 max_write_size = 1;
}

View File

@ -0,0 +1,9 @@
package simple
import "github.com/v2fly/v2ray-core/v5/common/errors"
type errPathObjHolder struct{}
func newError(values ...interface{}) *errors.Error {
return errors.New(values...).WithPathObj(errPathObjHolder{})
}

View File

@ -0,0 +1,137 @@
package simple
import (
"bytes"
"context"
"sync"
"github.com/v2fly/v2ray-core/v5/common"
"github.com/v2fly/v2ray-core/v5/transport/internet/request"
)
func newServer(config *ServerConfig) request.SessionAssemblerServer {
return &simpleAssemblerServer{}
}
type simpleAssemblerServer struct {
sessions sync.Map
assembly request.TransportServerAssembly
}
func (s *simpleAssemblerServer) OnTransportServerAssemblyReady(assembly request.TransportServerAssembly) {
s.assembly = assembly
}
func (s *simpleAssemblerServer) OnRoundTrip(ctx context.Context, req request.Request, opts ...request.RoundTripperOption,
) (resp request.Response, err error) {
connectionID := req.ConnectionTag
session := newSimpleAssemblerServerSession(ctx)
loadedSession, loaded := s.sessions.LoadOrStore(string(connectionID), session)
if loaded {
session = loadedSession.(*simpleAssemblerServerSession)
} else {
if err := s.assembly.SessionReceiver().OnNewSession(ctx, session); err != nil {
return request.Response{}, newError("failed to create new session").Base(err)
}
}
return session.OnRoundTrip(ctx, req, opts...)
}
func newSimpleAssemblerServerSession(ctx context.Context) *simpleAssemblerServerSession {
sessionCtx, finish := context.WithCancel(ctx)
return &simpleAssemblerServerSession{
readBuffer: bytes.NewBuffer(nil),
readChan: make(chan []byte, 16),
requestProcessed: make(chan struct{}),
writeLock: new(sync.Mutex),
writeBuffer: bytes.NewBuffer(nil),
maxWriteSize: 4096,
ctx: sessionCtx,
finish: finish,
}
}
type simpleAssemblerServerSession struct {
maxWriteSize int
readBuffer *bytes.Buffer
readChan chan []byte
requestProcessed chan struct{}
writeLock *sync.Mutex
writeBuffer *bytes.Buffer
ctx context.Context
finish func()
}
func (s *simpleAssemblerServerSession) Read(p []byte) (n int, err error) {
if s.readBuffer.Len() == 0 {
select {
case <-s.ctx.Done():
return 0, s.ctx.Err()
case data := <-s.readChan:
s.readBuffer.Write(data)
}
}
return s.readBuffer.Read(p)
}
func (s *simpleAssemblerServerSession) Write(p []byte) (n int, err error) {
s.writeLock.Lock()
n, err = s.writeBuffer.Write(p)
length := s.writeBuffer.Len()
s.writeLock.Unlock()
if err != nil {
return 0, err
}
if length > s.maxWriteSize {
select {
case <-s.requestProcessed:
case <-s.ctx.Done():
return 0, s.ctx.Err()
}
}
return
}
func (s *simpleAssemblerServerSession) Close() error {
s.finish()
return nil
}
func (s *simpleAssemblerServerSession) OnRoundTrip(ctx context.Context, req request.Request, opts ...request.RoundTripperOption,
) (resp request.Response, err error) {
if req.Data != nil && len(req.Data) > 0 {
select {
case <-s.ctx.Done():
return request.Response{}, s.ctx.Err()
case s.readChan <- req.Data:
}
}
s.writeLock.Lock()
nextWrite := s.writeBuffer.Next(s.maxWriteSize)
data := make([]byte, len(nextWrite))
copy(data, nextWrite)
s.writeLock.Unlock()
select {
case s.requestProcessed <- struct{}{}:
case <-s.ctx.Done():
return request.Response{}, s.ctx.Err()
default:
}
return request.Response{Data: data}, nil
}
func init() {
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 SimpleServerConfig")
}
return newServer(serverConfig), nil
}))
}

View File

@ -0,0 +1,3 @@
package simple
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen

View File

@ -0,0 +1,17 @@
package assembly
import (
"context"
"github.com/v2fly/v2ray-core/v5/common"
)
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen
const protocolName = "request"
func init() {
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
return nil, newError("request is a transport protocol.")
}))
}

View File

@ -0,0 +1,174 @@
package assembly
import (
_ "github.com/v2fly/v2ray-core/v5/common/protoext"
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
anypb "google.golang.org/protobuf/types/known/anypb"
reflect "reflect"
sync "sync"
)
const (
// Verify that this generated code is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
// Verify that runtime/protoimpl is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)
type Config struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Assembler *anypb.Any `protobuf:"bytes,1,opt,name=assembler,proto3" json:"assembler,omitempty"`
Roundtripper *anypb.Any `protobuf:"bytes,2,opt,name=roundtripper,proto3" json:"roundtripper,omitempty"`
}
func (x *Config) Reset() {
*x = Config{}
if protoimpl.UnsafeEnabled {
mi := &file_transport_internet_request_assembly_config_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *Config) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*Config) ProtoMessage() {}
func (x *Config) ProtoReflect() protoreflect.Message {
mi := &file_transport_internet_request_assembly_config_proto_msgTypes[0]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use Config.ProtoReflect.Descriptor instead.
func (*Config) Descriptor() ([]byte, []int) {
return file_transport_internet_request_assembly_config_proto_rawDescGZIP(), []int{0}
}
func (x *Config) GetAssembler() *anypb.Any {
if x != nil {
return x.Assembler
}
return nil
}
func (x *Config) GetRoundtripper() *anypb.Any {
if x != nil {
return x.Roundtripper
}
return nil
}
var File_transport_internet_request_assembly_config_proto protoreflect.FileDescriptor
var file_transport_internet_request_assembly_config_proto_rawDesc = []byte{
0x0a, 0x30, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2f, 0x69, 0x6e, 0x74, 0x65,
0x72, 0x6e, 0x65, 0x74, 0x2f, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2f, 0x61, 0x73, 0x73,
0x65, 0x6d, 0x62, 0x6c, 0x79, 0x2f, 0x63, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x2e, 0x70, 0x72, 0x6f,
0x74, 0x6f, 0x12, 0x2e, 0x76, 0x32, 0x72, 0x61, 0x79, 0x2e, 0x63, 0x6f, 0x72, 0x65, 0x2e, 0x74,
0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65,
0x74, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62,
0x6c, 0x79, 0x1a, 0x20, 0x63, 0x6f, 0x6d, 0x6d, 0x6f, 0x6e, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f,
0x65, 0x78, 0x74, 0x2f, 0x65, 0x78, 0x74, 0x65, 0x6e, 0x73, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70,
0x72, 0x6f, 0x74, 0x6f, 0x1a, 0x19, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f,
0x74, 0x6f, 0x62, 0x75, 0x66, 0x2f, 0x61, 0x6e, 0x79, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22,
0x90, 0x01, 0x0a, 0x06, 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x12, 0x32, 0x0a, 0x09, 0x61, 0x73,
0x73, 0x65, 0x6d, 0x62, 0x6c, 0x65, 0x72, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x14, 0x2e,
0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e,
0x41, 0x6e, 0x79, 0x52, 0x09, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x65, 0x72, 0x12, 0x38,
0x0a, 0x0c, 0x72, 0x6f, 0x75, 0x6e, 0x64, 0x74, 0x72, 0x69, 0x70, 0x70, 0x65, 0x72, 0x18, 0x02,
0x20, 0x01, 0x28, 0x0b, 0x32, 0x14, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72,
0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x41, 0x6e, 0x79, 0x52, 0x0c, 0x72, 0x6f, 0x75, 0x6e,
0x64, 0x74, 0x72, 0x69, 0x70, 0x70, 0x65, 0x72, 0x3a, 0x18, 0x82, 0xb5, 0x18, 0x14, 0x0a, 0x09,
0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x12, 0x07, 0x72, 0x65, 0x71, 0x75, 0x65,
0x73, 0x74, 0x42, 0xab, 0x01, 0x0a, 0x32, 0x63, 0x6f, 0x6d, 0x2e, 0x76, 0x32, 0x72, 0x61, 0x79,
0x2e, 0x63, 0x6f, 0x72, 0x65, 0x2e, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2e,
0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65, 0x74, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74,
0x2e, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x79, 0x50, 0x01, 0x5a, 0x42, 0x67, 0x69, 0x74,
0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x76, 0x32, 0x66, 0x6c, 0x79, 0x2f, 0x76, 0x32,
0x72, 0x61, 0x79, 0x2d, 0x63, 0x6f, 0x72, 0x65, 0x2f, 0x76, 0x35, 0x2f, 0x74, 0x72, 0x61, 0x6e,
0x73, 0x70, 0x6f, 0x72, 0x74, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65, 0x74, 0x2f, 0x72,
0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2f, 0x61, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x79, 0xaa,
0x02, 0x2e, 0x56, 0x32, 0x52, 0x61, 0x79, 0x2e, 0x43, 0x6f, 0x72, 0x65, 0x2e, 0x54, 0x72, 0x61,
0x6e, 0x73, 0x70, 0x6f, 0x72, 0x74, 0x2e, 0x49, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65, 0x74, 0x2e,
0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x41, 0x73, 0x73, 0x65, 0x6d, 0x62, 0x6c, 0x79,
0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_transport_internet_request_assembly_config_proto_rawDescOnce sync.Once
file_transport_internet_request_assembly_config_proto_rawDescData = file_transport_internet_request_assembly_config_proto_rawDesc
)
func file_transport_internet_request_assembly_config_proto_rawDescGZIP() []byte {
file_transport_internet_request_assembly_config_proto_rawDescOnce.Do(func() {
file_transport_internet_request_assembly_config_proto_rawDescData = protoimpl.X.CompressGZIP(file_transport_internet_request_assembly_config_proto_rawDescData)
})
return file_transport_internet_request_assembly_config_proto_rawDescData
}
var file_transport_internet_request_assembly_config_proto_msgTypes = make([]protoimpl.MessageInfo, 1)
var file_transport_internet_request_assembly_config_proto_goTypes = []interface{}{
(*Config)(nil), // 0: v2ray.core.transport.internet.request.assembly.Config
(*anypb.Any)(nil), // 1: google.protobuf.Any
}
var file_transport_internet_request_assembly_config_proto_depIdxs = []int32{
1, // 0: v2ray.core.transport.internet.request.assembly.Config.assembler:type_name -> google.protobuf.Any
1, // 1: v2ray.core.transport.internet.request.assembly.Config.roundtripper:type_name -> google.protobuf.Any
2, // [2:2] is the sub-list for method output_type
2, // [2:2] is the sub-list for method input_type
2, // [2:2] is the sub-list for extension type_name
2, // [2:2] is the sub-list for extension extendee
0, // [0:2] is the sub-list for field type_name
}
func init() { file_transport_internet_request_assembly_config_proto_init() }
func file_transport_internet_request_assembly_config_proto_init() {
if File_transport_internet_request_assembly_config_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_transport_internet_request_assembly_config_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*Config); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
type x struct{}
out := protoimpl.TypeBuilder{
File: protoimpl.DescBuilder{
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_transport_internet_request_assembly_config_proto_rawDesc,
NumEnums: 0,
NumMessages: 1,
NumExtensions: 0,
NumServices: 0,
},
GoTypes: file_transport_internet_request_assembly_config_proto_goTypes,
DependencyIndexes: file_transport_internet_request_assembly_config_proto_depIdxs,
MessageInfos: file_transport_internet_request_assembly_config_proto_msgTypes,
}.Build()
File_transport_internet_request_assembly_config_proto = out.File
file_transport_internet_request_assembly_config_proto_rawDesc = nil
file_transport_internet_request_assembly_config_proto_goTypes = nil
file_transport_internet_request_assembly_config_proto_depIdxs = nil
}

View File

@ -0,0 +1,18 @@
syntax = "proto3";
package v2ray.core.transport.internet.request.assembly;
option csharp_namespace = "V2Ray.Core.Transport.Internet.Request.Assembly";
option go_package = "github.com/v2fly/v2ray-core/v5/transport/internet/request/assembly";
option java_package = "com.v2ray.core.transport.internet.request.assembly";
option java_multiple_files = true;
import "common/protoext/extensions.proto";
import "google/protobuf/any.proto";
message Config {
option (v2ray.core.common.protoext.message_opt).type = "transport";
option (v2ray.core.common.protoext.message_opt).short_name = "request";
google.protobuf.Any assembler = 1;
google.protobuf.Any roundtripper = 2;
}

View File

@ -0,0 +1,125 @@
package assembly
import (
"context"
gonet "net"
"time"
"github.com/v2fly/v2ray-core/v5/transport/internet/transportcommon"
"github.com/v2fly/v2ray-core/v5/common"
"github.com/v2fly/v2ray-core/v5/common/net"
"github.com/v2fly/v2ray-core/v5/common/serial"
"github.com/v2fly/v2ray-core/v5/common/session"
"github.com/v2fly/v2ray-core/v5/transport/internet"
"github.com/v2fly/v2ray-core/v5/transport/internet/request"
)
type client struct {
tripper request.RoundTripperClient
assembler request.SessionAssemblerClient
streamSettings *internet.MemoryStreamConfig
dest net.Destination
}
func (c client) Dial(ctx context.Context) (net.Conn, error) {
return transportcommon.DialWithSecuritySettings(ctx, c.dest, c.streamSettings)
}
func (c client) AutoImplDialer() request.Dialer {
return c
}
func (c client) Tripper() request.Tripper {
return c.tripper
}
func (c client) dialRequestSession(ctx context.Context) (net.Conn, error) {
session, err := c.assembler.NewSession(ctx)
if err != nil {
return nil, newError("failed to create new session").Base(err)
}
return clientConnection{session}, nil
}
type clientConnection struct {
request.Session
}
func (c clientConnection) LocalAddr() gonet.Addr {
return &net.UnixAddr{Name: "unimplemented"}
}
func (c clientConnection) RemoteAddr() gonet.Addr {
return &net.UnixAddr{Name: "unimplemented"}
}
func (c clientConnection) SetDeadline(t time.Time) error {
// Unimplemented
return nil
}
func (c clientConnection) SetReadDeadline(t time.Time) error {
// Unimplemented
return nil
}
func (c clientConnection) SetWriteDeadline(t time.Time) error {
// Unimplemented
return nil
}
func dialRequest(ctx context.Context, dest net.Destination, streamSettings *internet.MemoryStreamConfig) (net.Conn, error) {
clientAssembly := &client{}
transportConfiguration := streamSettings.ProtocolSettings.(*Config)
assemblerConfigInstance, err := serial.GetInstanceOf(transportConfiguration.Assembler)
if err != nil {
return nil, newError("failed to get config instance of assembler").Base(err)
}
assembler, err := common.CreateObject(ctx, assemblerConfigInstance)
if err != nil {
return nil, newError("failed to create assembler").Base(err)
}
if typedAssembler, ok := assembler.(request.SessionAssemblerClient); !ok {
return nil, newError("failed to type assert assembler to SessionAssemblerClient")
} else {
clientAssembly.assembler = typedAssembler
}
roundtripperConfigInstance, err := serial.GetInstanceOf(transportConfiguration.Roundtripper)
if err != nil {
return nil, newError("failed to get config instance of roundtripper").Base(err)
}
roundtripper, err := common.CreateObject(ctx, roundtripperConfigInstance)
if err != nil {
return nil, newError("failed to create roundtripper").Base(err)
}
if typedRoundtripper, ok := roundtripper.(request.RoundTripperClient); !ok {
return nil, newError("failed to type assert roundtripper to RoundTripperClient")
} else {
clientAssembly.tripper = typedRoundtripper
}
clientAssembly.streamSettings = streamSettings
clientAssembly.dest = dest
clientAssembly.assembler.OnTransportClientAssemblyReady(clientAssembly)
clientAssembly.tripper.OnTransportClientAssemblyReady(clientAssembly)
return clientAssembly.dialRequestSession(ctx)
}
func dial(ctx context.Context, dest net.Destination, streamSettings *internet.MemoryStreamConfig) (internet.Connection, error) {
newError("creating connection to ", dest).WriteToLog(session.ExportIDToError(ctx))
conn, err := dialRequest(ctx, dest, streamSettings)
if err != nil {
return nil, newError("failed to dial request to ", dest).Base(err)
}
return internet.Connection(conn), nil
}
func init() {
common.Must(internet.RegisterTransportDialer(protocolName, dial))
}

View File

@ -0,0 +1,9 @@
package assembly
import "github.com/v2fly/v2ray-core/v5/common/errors"
type errPathObjHolder struct{}
func newError(values ...interface{}) *errors.Error {
return errors.New(values...).WithPathObj(errPathObjHolder{})
}

View File

@ -0,0 +1,135 @@
package assembly
import (
"context"
gonet "net"
"time"
"github.com/v2fly/v2ray-core/v5/transport/internet/transportcommon"
"github.com/v2fly/v2ray-core/v5/common"
"github.com/v2fly/v2ray-core/v5/common/net"
"github.com/v2fly/v2ray-core/v5/common/serial"
"github.com/v2fly/v2ray-core/v5/transport/internet"
"github.com/v2fly/v2ray-core/v5/transport/internet/request"
)
type server struct {
tripper request.RoundTripperServer
assembler request.SessionAssemblerServer
addConn internet.ConnHandler
streamSettings *internet.MemoryStreamConfig
addr net.Address
port net.Port
}
func (s server) Listen(ctx context.Context) (net.Listener, error) {
return transportcommon.ListenWithSecuritySettings(ctx, s.addr, s.port, s.streamSettings)
}
func (s server) AutoImplListener() request.Listener {
return s
}
func (s server) Close() error {
if err := s.tripper.Close(); err != nil {
return newError("failed to close tripper").Base(err)
}
return nil
}
func (s server) Addr() net.Addr {
// Unimplemented
return nil
}
type serverConnection struct {
request.Session
}
func (s serverConnection) LocalAddr() gonet.Addr {
return &net.UnixAddr{Name: "unimplemented"}
}
func (s serverConnection) RemoteAddr() gonet.Addr {
return &net.UnixAddr{Name: "unimplemented"}
}
func (s serverConnection) SetDeadline(t time.Time) error {
// Unimplemented
return nil
}
func (s serverConnection) SetReadDeadline(t time.Time) error {
// Unimplemented
return nil
}
func (s serverConnection) SetWriteDeadline(t time.Time) error {
// Unimplemented
return nil
}
func (s server) OnNewSession(ctx context.Context, sess request.Session, opts ...request.SessionOption) error {
s.addConn(&serverConnection{sess})
return nil
}
func (s server) SessionReceiver() request.SessionReceiver {
return s
}
func (s server) TripperReceiver() request.TripperReceiver {
return s.assembler
}
func listenRequest(ctx context.Context, address net.Address, port net.Port, streamSettings *internet.MemoryStreamConfig, addConn internet.ConnHandler) (internet.Listener, error) {
transportConfiguration := streamSettings.ProtocolSettings.(*Config)
serverAssembly := &server{addConn: addConn}
assemblerConfigInstance, err := serial.GetInstanceOf(transportConfiguration.Assembler)
if err != nil {
return nil, newError("failed to get config instance of assembler").Base(err)
}
assembler, err := common.CreateObject(ctx, assemblerConfigInstance)
if err != nil {
return nil, newError("failed to create assembler").Base(err)
}
if typedAssembler, ok := assembler.(request.SessionAssemblerServer); !ok {
return nil, newError("failed to type assert assembler to SessionAssemblerServer")
} else {
serverAssembly.assembler = typedAssembler
}
roundtripperConfigInstance, err := serial.GetInstanceOf(transportConfiguration.Roundtripper)
if err != nil {
return nil, newError("failed to get config instance of roundtripper").Base(err)
}
roundtripper, err := common.CreateObject(ctx, roundtripperConfigInstance)
if err != nil {
return nil, newError("failed to create roundtripper").Base(err)
}
if typedRoundtripper, ok := roundtripper.(request.RoundTripperServer); !ok {
return nil, newError("failed to type assert roundtripper to RoundTripperServer")
} else {
serverAssembly.tripper = typedRoundtripper
}
serverAssembly.addr = address
serverAssembly.port = port
serverAssembly.streamSettings = streamSettings
serverAssembly.assembler.OnTransportServerAssemblyReady(serverAssembly)
serverAssembly.tripper.OnTransportServerAssemblyReady(serverAssembly)
if err := serverAssembly.tripper.Start(); err != nil {
return nil, newError("failed to start tripper").Base(err)
}
return serverAssembly, nil
}
func init() {
common.Must(internet.RegisterTransportListener(protocolName, listenRequest))
}

View File

@ -0,0 +1 @@
package request

View File

@ -0,0 +1,42 @@
package request
import (
"context"
"github.com/v2fly/v2ray-core/v5/common/net"
)
type TransportClientAssembly interface {
Tripper() Tripper
AutoImplDialer() Dialer
}
type TransportClientAssemblyReceiver interface {
OnTransportClientAssemblyReady(TransportClientAssembly)
}
type TransportServerAssembly interface {
TripperReceiver() TripperReceiver
SessionReceiver() SessionReceiver
AutoImplListener() Listener
}
type TransportServerAssemblyReceiver interface {
OnTransportServerAssemblyReady(TransportServerAssembly)
}
type SessionCreator interface {
NewSession(ctx context.Context, opts ...SessionOption) (Session, error)
}
type SessionReceiver interface {
OnNewSession(ctx context.Context, sess Session, opts ...SessionOption) error
}
type Dialer interface {
Dial(ctx context.Context) (net.Conn, error)
}
type Listener interface {
Listen(ctx context.Context) (net.Listener, error)
}

View File

@ -0,0 +1,38 @@
package request
import (
"context"
"github.com/v2fly/v2ray-core/v5/common"
)
type RoundTripperClient interface {
Tripper
TransportClientAssemblyReceiver
}
type RoundTripperServer interface {
common.Runnable
TransportServerAssemblyReceiver
}
type Tripper interface {
RoundTrip(ctx context.Context, req Request, opts ...RoundTripperOption) (resp Response, err error)
}
type TripperReceiver interface {
OnRoundTrip(ctx context.Context, req Request, opts ...RoundTripperOption) (resp Response, err error)
}
type RoundTripperOption interface {
RoundTripperOption()
}
type Request struct {
Data []byte
ConnectionTag []byte
}
type Response struct {
Data []byte
}

View File

@ -0,0 +1,320 @@
package httprt
import (
_ "github.com/v2fly/v2ray-core/v5/common/protoext"
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
reflect "reflect"
sync "sync"
)
const (
// Verify that this generated code is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
// Verify that runtime/protoimpl is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)
type ClientConfig struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Http *HTTPConfig `protobuf:"bytes,1,opt,name=http,proto3" json:"http,omitempty"`
}
func (x *ClientConfig) Reset() {
*x = ClientConfig{}
if protoimpl.UnsafeEnabled {
mi := &file_transport_internet_request_roundtripper_httprt_config_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ClientConfig) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ClientConfig) ProtoMessage() {}
func (x *ClientConfig) ProtoReflect() protoreflect.Message {
mi := &file_transport_internet_request_roundtripper_httprt_config_proto_msgTypes[0]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use ClientConfig.ProtoReflect.Descriptor instead.
func (*ClientConfig) Descriptor() ([]byte, []int) {
return file_transport_internet_request_roundtripper_httprt_config_proto_rawDescGZIP(), []int{0}
}
func (x *ClientConfig) GetHttp() *HTTPConfig {
if x != nil {
return x.Http
}
return nil
}
type ServerConfig struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Http *HTTPConfig `protobuf:"bytes,1,opt,name=http,proto3" json:"http,omitempty"`
NoDecodingSessionTag bool `protobuf:"varint,2,opt,name=no_decoding_session_tag,json=noDecodingSessionTag,proto3" json:"no_decoding_session_tag,omitempty"`
}
func (x *ServerConfig) Reset() {
*x = ServerConfig{}
if protoimpl.UnsafeEnabled {
mi := &file_transport_internet_request_roundtripper_httprt_config_proto_msgTypes[1]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ServerConfig) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ServerConfig) ProtoMessage() {}
func (x *ServerConfig) ProtoReflect() protoreflect.Message {
mi := &file_transport_internet_request_roundtripper_httprt_config_proto_msgTypes[1]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use ServerConfig.ProtoReflect.Descriptor instead.
func (*ServerConfig) Descriptor() ([]byte, []int) {
return file_transport_internet_request_roundtripper_httprt_config_proto_rawDescGZIP(), []int{1}
}
func (x *ServerConfig) GetHttp() *HTTPConfig {
if x != nil {
return x.Http
}
return nil
}
func (x *ServerConfig) GetNoDecodingSessionTag() bool {
if x != nil {
return x.NoDecodingSessionTag
}
return false
}
type HTTPConfig struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Path string `protobuf:"bytes,1,opt,name=path,proto3" json:"path,omitempty"`
UrlPrefix string `protobuf:"bytes,2,opt,name=urlPrefix,proto3" json:"urlPrefix,omitempty"`
}
func (x *HTTPConfig) Reset() {
*x = HTTPConfig{}
if protoimpl.UnsafeEnabled {
mi := &file_transport_internet_request_roundtripper_httprt_config_proto_msgTypes[2]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *HTTPConfig) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*HTTPConfig) ProtoMessage() {}
func (x *HTTPConfig) ProtoReflect() protoreflect.Message {
mi := &file_transport_internet_request_roundtripper_httprt_config_proto_msgTypes[2]