diff --git a/app/proxyman/config.pb.go b/app/proxyman/config.pb.go index 8b745a95..f25206d9 100644 --- a/app/proxyman/config.pb.go +++ b/app/proxyman/config.pb.go @@ -411,8 +411,10 @@ type MultiplexingConfig struct { XudpConcurrency int32 `protobuf:"varint,3,opt,name=xudpConcurrency,proto3" json:"xudpConcurrency,omitempty"` // "reject" (default), "allow" or "skip". XudpProxyUDP443 string `protobuf:"bytes,4,opt,name=xudpProxyUDP443,proto3" json:"xudpProxyUDP443,omitempty"` - unknownFields protoimpl.UnknownFields - sizeCache protoimpl.SizeCache + // MaxReuseTimes for an connection + MaxReuseTimes int32 `protobuf:"varint,5,opt,name=maxReuseTimes,proto3" json:"maxReuseTimes,omitempty"` + unknownFields protoimpl.UnknownFields + sizeCache protoimpl.SizeCache } func (x *MultiplexingConfig) Reset() { @@ -473,6 +475,13 @@ func (x *MultiplexingConfig) GetXudpProxyUDP443() string { return "" } +func (x *MultiplexingConfig) GetMaxReuseTimes() int32 { + if x != nil { + return x.MaxReuseTimes + } + return 0 +} + var File_app_proxyman_config_proto protoreflect.FileDescriptor const file_app_proxyman_config_proto_rawDesc = "" + @@ -503,12 +512,13 @@ const file_app_proxyman_config_proto_rawDesc = "" + "\x0eproxy_settings\x18\x03 \x01(\v2$.xray.transport.internet.ProxyConfigR\rproxySettings\x12T\n" + "\x12multiplex_settings\x18\x04 \x01(\v2%.xray.app.proxyman.MultiplexingConfigR\x11multiplexSettings\x12\x19\n" + "\bvia_cidr\x18\x05 \x01(\tR\aviaCidr\x12P\n" + - "\x0ftarget_strategy\x18\x06 \x01(\x0e2'.xray.transport.internet.DomainStrategyR\x0etargetStrategy\"\xa4\x01\n" + + "\x0ftarget_strategy\x18\x06 \x01(\x0e2'.xray.transport.internet.DomainStrategyR\x0etargetStrategy\"\xca\x01\n" + "\x12MultiplexingConfig\x12\x18\n" + "\aenabled\x18\x01 \x01(\bR\aenabled\x12 \n" + "\vconcurrency\x18\x02 \x01(\x05R\vconcurrency\x12(\n" + "\x0fxudpConcurrency\x18\x03 \x01(\x05R\x0fxudpConcurrency\x12(\n" + - "\x0fxudpProxyUDP443\x18\x04 \x01(\tR\x0fxudpProxyUDP443BU\n" + + "\x0fxudpProxyUDP443\x18\x04 \x01(\tR\x0fxudpProxyUDP443\x12$\n" + + "\rmaxReuseTimes\x18\x05 \x01(\x05R\rmaxReuseTimesBU\n" + "\x15com.xray.app.proxymanP\x01Z&github.com/xtls/xray-core/app/proxyman\xaa\x02\x11Xray.App.Proxymanb\x06proto3" var ( diff --git a/app/proxyman/config.proto b/app/proxyman/config.proto index 4f1298b9..6b3f0166 100644 --- a/app/proxyman/config.proto +++ b/app/proxyman/config.proto @@ -68,4 +68,6 @@ message MultiplexingConfig { int32 xudpConcurrency = 3; // "reject" (default), "allow" or "skip". string xudpProxyUDP443 = 4; + // MaxReuseTimes for an connection + int32 maxReuseTimes = 5; } diff --git a/app/proxyman/outbound/handler.go b/app/proxyman/outbound/handler.go index 62902c60..035cbb3a 100644 --- a/app/proxyman/outbound/handler.go +++ b/app/proxyman/outbound/handler.go @@ -121,6 +121,12 @@ func NewHandler(ctx context.Context, config *core.OutboundHandlerConfig) (outbou if h.senderSettings != nil && h.senderSettings.MultiplexSettings != nil { if config := h.senderSettings.MultiplexSettings; config.Enabled { + // MaxReuseTimes use 60000 as default, and it also means the upper limit of MaxReuseTimes + // In mux cool spec, connection ID is 2 bytes, so physical limit is 65535, bu we reserve some IDs for future use + MaxReuseTimes := uint32(60000) + if config.MaxReuseTimes != 0 && config.MaxReuseTimes < 60000 { + MaxReuseTimes = uint32(config.MaxReuseTimes) + } if config.Concurrency < 0 { h.mux = &mux.ClientManager{Enabled: false} } @@ -136,7 +142,7 @@ func NewHandler(ctx context.Context, config *core.OutboundHandlerConfig) (outbou Dialer: h, Strategy: mux.ClientStrategy{ MaxConcurrency: uint32(config.Concurrency), - MaxConnection: 128, + MaxReuseTimes: MaxReuseTimes, }, }, }, @@ -157,7 +163,7 @@ func NewHandler(ctx context.Context, config *core.OutboundHandlerConfig) (outbou Dialer: h, Strategy: mux.ClientStrategy{ MaxConcurrency: uint32(config.XudpConcurrency), - MaxConnection: 128, + MaxReuseTimes: 128, }, }, }, diff --git a/common/mux/client.go b/common/mux/client.go index 47943a2b..5b90991d 100644 --- a/common/mux/client.go +++ b/common/mux/client.go @@ -170,7 +170,7 @@ func (f *DialingWorkerFactory) Create() (*ClientWorker, error) { type ClientStrategy struct { MaxConcurrency uint32 - MaxConnection uint32 + MaxReuseTimes uint32 } type ClientWorker struct { @@ -179,6 +179,7 @@ type ClientWorker struct { done *done.Instance timer *time.Ticker strategy ClientStrategy + timeCretaed time.Time } var ( @@ -194,6 +195,7 @@ func NewClientWorker(stream transport.Link, s ClientStrategy) (*ClientWorker, er done: done.New(), timer: time.NewTicker(time.Second * 16), strategy: s, + timeCretaed: time.Now(), } go c.fetchOutput() @@ -288,7 +290,7 @@ func fetchInput(ctx context.Context, s *Session, output buf.Writer) { func (m *ClientWorker) IsClosing() bool { sm := m.sessionManager - if m.strategy.MaxConnection > 0 && sm.Count() >= int(m.strategy.MaxConnection) { + if m.strategy.MaxReuseTimes > 0 && sm.Count() >= int(m.strategy.MaxReuseTimes) { return true } return false @@ -318,6 +320,7 @@ func (m *ClientWorker) Dispatch(ctx context.Context, link *transport.Link) bool if s == nil { return false } + errors.LogInfo(ctx, "Allocated mux.cool sub connection ID: ", s.ID, "/", m.strategy.MaxReuseTimes, " living: ", m.ActiveConnections(), "/", m.strategy.MaxConcurrency, " age: ", time.Since(m.timeCretaed).Truncate(time.Second)) s.input = link.Reader s.output = link.Writer go fetchInput(ctx, s, m.link.Writer) diff --git a/common/mux/client_test.go b/common/mux/client_test.go index 9626e2a2..d0238e96 100644 --- a/common/mux/client_test.go +++ b/common/mux/client_test.go @@ -58,7 +58,7 @@ func TestClientWorkerClose(t *testing.T) { Writer: w1, }, mux.ClientStrategy{ MaxConcurrency: 4, - MaxConnection: 4, + MaxReuseTimes: 4, }) common.Must(err) @@ -68,7 +68,7 @@ func TestClientWorkerClose(t *testing.T) { Writer: w2, }, mux.ClientStrategy{ MaxConcurrency: 4, - MaxConnection: 4, + MaxReuseTimes: 4, }) common.Must(err) diff --git a/common/mux/session.go b/common/mux/session.go index 66b9674c..2dc496e0 100644 --- a/common/mux/session.go +++ b/common/mux/session.go @@ -56,7 +56,7 @@ func (m *SessionManager) Allocate(Strategy *ClientStrategy) *Session { defer m.Unlock() MaxConcurrency := int(Strategy.MaxConcurrency) - MaxConnection := uint16(Strategy.MaxConnection) + MaxConnection := uint16(Strategy.MaxReuseTimes) if m.closed || (MaxConcurrency > 0 && len(m.sessions) >= MaxConcurrency) || (MaxConnection > 0 && m.count >= MaxConnection) { return nil diff --git a/infra/conf/xray.go b/infra/conf/xray.go index 39a1f763..c32e9ece 100644 --- a/infra/conf/xray.go +++ b/infra/conf/xray.go @@ -101,6 +101,7 @@ func (c *SniffingConfig) Build() (*proxyman.SniffingConfig, error) { type MuxConfig struct { Enabled bool `json:"enabled"` Concurrency int16 `json:"concurrency"` + MaxReuseTimes int32 `json:"maxReuseTimes"` XudpConcurrency int16 `json:"xudpConcurrency"` XudpProxyUDP443 string `json:"xudpProxyUDP443"` } @@ -117,6 +118,7 @@ func (m *MuxConfig) Build() (*proxyman.MultiplexingConfig, error) { return &proxyman.MultiplexingConfig{ Enabled: m.Enabled, Concurrency: int32(m.Concurrency), + MaxReuseTimes: m.MaxReuseTimes, XudpConcurrency: int32(m.XudpConcurrency), XudpProxyUDP443: m.XudpProxyUDP443, }, nil