mirror of https://github.com/v2ray/v2ray-core
90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package inbound
|
|
|
|
import (
|
|
"context"
|
|
|
|
"v2ray.com/core/app/log"
|
|
"v2ray.com/core/app/proxyman"
|
|
"v2ray.com/core/app/proxyman/mux"
|
|
"v2ray.com/core/common/dice"
|
|
"v2ray.com/core/common/net"
|
|
"v2ray.com/core/proxy"
|
|
)
|
|
|
|
type AlwaysOnInboundHandler struct {
|
|
proxy proxy.Inbound
|
|
workers []worker
|
|
mux *mux.Server
|
|
}
|
|
|
|
func NewAlwaysOnInboundHandler(ctx context.Context, tag string, receiverConfig *proxyman.ReceiverConfig, proxyConfig interface{}) (*AlwaysOnInboundHandler, error) {
|
|
p, err := proxy.CreateInboundHandler(ctx, proxyConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
h := &AlwaysOnInboundHandler{
|
|
proxy: p,
|
|
mux: mux.NewServer(ctx),
|
|
}
|
|
|
|
nl := p.Network()
|
|
pr := receiverConfig.PortRange
|
|
address := receiverConfig.Listen.AsAddress()
|
|
if address == nil {
|
|
address = net.AnyIP
|
|
}
|
|
for port := pr.From; port <= pr.To; port++ {
|
|
if nl.HasNetwork(net.Network_TCP) {
|
|
log.Trace(newError("creating tcp worker on ", address, ":", port).AtDebug())
|
|
worker := &tcpWorker{
|
|
address: address,
|
|
port: net.Port(port),
|
|
proxy: p,
|
|
stream: receiverConfig.StreamSettings,
|
|
recvOrigDest: receiverConfig.ReceiveOriginalDestination,
|
|
tag: tag,
|
|
dispatcher: h.mux,
|
|
}
|
|
h.workers = append(h.workers, worker)
|
|
}
|
|
|
|
if nl.HasNetwork(net.Network_UDP) {
|
|
worker := &udpWorker{
|
|
tag: tag,
|
|
proxy: p,
|
|
address: address,
|
|
port: net.Port(port),
|
|
recvOrigDest: receiverConfig.ReceiveOriginalDestination,
|
|
dispatcher: h.mux,
|
|
}
|
|
h.workers = append(h.workers, worker)
|
|
}
|
|
}
|
|
|
|
return h, nil
|
|
}
|
|
|
|
func (h *AlwaysOnInboundHandler) Start() error {
|
|
for _, worker := range h.workers {
|
|
if err := worker.Start(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (h *AlwaysOnInboundHandler) Close() {
|
|
for _, worker := range h.workers {
|
|
worker.Close()
|
|
}
|
|
}
|
|
|
|
func (h *AlwaysOnInboundHandler) GetRandomInboundProxy() (proxy.Inbound, net.Port, int) {
|
|
if len(h.workers) == 0 {
|
|
return nil, 0, 0
|
|
}
|
|
w := h.workers[dice.Roll(len(h.workers))]
|
|
return w.Proxy(), w.Port(), 9999
|
|
}
|