mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-16 09:26:21 -05:00
97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
package core
|
|
|
|
import (
|
|
"github.com/v2ray/v2ray-core/common/log"
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
"github.com/v2ray/v2ray-core/config"
|
|
)
|
|
|
|
var (
|
|
inboundFactories = make(map[string]InboundConnectionHandlerFactory)
|
|
outboundFactories = make(map[string]OutboundConnectionHandlerFactory)
|
|
)
|
|
|
|
func RegisterInboundConnectionHandlerFactory(name string, factory InboundConnectionHandlerFactory) error {
|
|
// TODO check name
|
|
inboundFactories[name] = factory
|
|
return nil
|
|
}
|
|
|
|
func RegisterOutboundConnectionHandlerFactory(name string, factory OutboundConnectionHandlerFactory) error {
|
|
// TODO check name
|
|
outboundFactories[name] = factory
|
|
return nil
|
|
}
|
|
|
|
// Point is an single server in V2Ray system.
|
|
type Point struct {
|
|
port uint16
|
|
ichFactory InboundConnectionHandlerFactory
|
|
ichConfig interface{}
|
|
ochFactory OutboundConnectionHandlerFactory
|
|
ochConfig interface{}
|
|
}
|
|
|
|
// NewPoint returns a new Point server based on given configuration.
|
|
// The server is not started at this point.
|
|
func NewPoint(pConfig config.PointConfig) (*Point, error) {
|
|
var vpoint = new(Point)
|
|
vpoint.port = pConfig.Port()
|
|
|
|
ichFactory, ok := inboundFactories[pConfig.InboundConfig().Protocol()]
|
|
if !ok {
|
|
panic(log.Error("Unknown inbound connection handler factory %s", pConfig.InboundConfig().Protocol()))
|
|
}
|
|
vpoint.ichFactory = ichFactory
|
|
vpoint.ichConfig = pConfig.InboundConfig().Settings(config.TypeInbound)
|
|
|
|
ochFactory, ok := outboundFactories[pConfig.OutboundConfig().Protocol()]
|
|
if !ok {
|
|
panic(log.Error("Unknown outbound connection handler factory %s", pConfig.OutboundConfig().Protocol))
|
|
}
|
|
|
|
vpoint.ochFactory = ochFactory
|
|
vpoint.ochConfig = pConfig.OutboundConfig().Settings(config.TypeOutbound)
|
|
|
|
return vpoint, nil
|
|
}
|
|
|
|
type InboundConnectionHandlerFactory interface {
|
|
Create(vp *Point, config interface{}) (InboundConnectionHandler, error)
|
|
}
|
|
|
|
type InboundConnectionHandler interface {
|
|
Listen(port uint16) error
|
|
}
|
|
|
|
type OutboundConnectionHandlerFactory interface {
|
|
Create(VP *Point, config interface{}, firstPacket v2net.Packet) (OutboundConnectionHandler, error)
|
|
}
|
|
|
|
type OutboundConnectionHandler interface {
|
|
Start(ray OutboundRay) error
|
|
}
|
|
|
|
// Start starts the Point server, and return any error during the process.
|
|
// In the case of any errors, the state of the server is unpredicatable.
|
|
func (vp *Point) Start() error {
|
|
if vp.port <= 0 {
|
|
return log.Error("Invalid port %d", vp.port)
|
|
}
|
|
|
|
inboundConnectionHandler, err := vp.ichFactory.Create(vp, vp.ichConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = inboundConnectionHandler.Listen(vp.port)
|
|
return nil
|
|
}
|
|
|
|
func (p *Point) DispatchToOutbound(packet v2net.Packet) InboundRay {
|
|
ray := NewRay()
|
|
// TODO: handle error
|
|
och, _ := p.ochFactory.Create(p, p.ochConfig, packet)
|
|
_ = och.Start(ray)
|
|
return ray
|
|
}
|