mirror of https://github.com/v2ray/v2ray-core
72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
package point
|
|
|
|
import (
|
|
"github.com/v2ray/v2ray-core/app"
|
|
"github.com/v2ray/v2ray-core/common/dice"
|
|
"github.com/v2ray/v2ray-core/common/log"
|
|
"github.com/v2ray/v2ray-core/common/retry"
|
|
"github.com/v2ray/v2ray-core/proxy"
|
|
proxyregistry "github.com/v2ray/v2ray-core/proxy/registry"
|
|
)
|
|
|
|
// Handler for inbound detour connections.
|
|
type InboundDetourHandlerAlways struct {
|
|
space app.Space
|
|
config *InboundDetourConfig
|
|
ich []proxy.InboundHandler
|
|
}
|
|
|
|
func NewInboundDetourHandlerAlways(space app.Space, config *InboundDetourConfig) (*InboundDetourHandlerAlways, error) {
|
|
handler := &InboundDetourHandlerAlways{
|
|
space: space,
|
|
config: config,
|
|
}
|
|
ports := config.PortRange
|
|
handler.ich = make([]proxy.InboundHandler, 0, ports.To-ports.From+1)
|
|
for i := ports.From; i <= ports.To; i++ {
|
|
ichConfig := config.Settings
|
|
ich, err := proxyregistry.CreateInboundHandler(config.Protocol, space, ichConfig, &proxy.InboundHandlerMeta{
|
|
Address: config.ListenOn,
|
|
Port: i,
|
|
Tag: config.Tag,
|
|
StreamSettings: config.StreamSettings,
|
|
AllowPassiveConnection: config.AllowPassiveConnection,
|
|
})
|
|
if err != nil {
|
|
log.Error("Failed to create inbound connection handler: ", err)
|
|
return nil, err
|
|
}
|
|
handler.ich = append(handler.ich, ich)
|
|
}
|
|
return handler, nil
|
|
}
|
|
|
|
func (this *InboundDetourHandlerAlways) GetConnectionHandler() (proxy.InboundHandler, int) {
|
|
ich := this.ich[dice.Roll(len(this.ich))]
|
|
return ich, this.config.Allocation.Refresh
|
|
}
|
|
|
|
func (this *InboundDetourHandlerAlways) Close() {
|
|
for _, ich := range this.ich {
|
|
ich.Close()
|
|
}
|
|
}
|
|
|
|
// Starts the inbound connection handler.
|
|
func (this *InboundDetourHandlerAlways) Start() error {
|
|
for _, ich := range this.ich {
|
|
err := retry.Timed(100 /* times */, 100 /* ms */).On(func() error {
|
|
err := ich.Start()
|
|
if err != nil {
|
|
log.Error("Failed to start inbound detour:", err)
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|