1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-11-14 16:25:57 -05:00
v2fly/point.go

95 lines
2.6 KiB
Go
Raw Normal View History

2015-09-05 11:48:38 -04:00
package core
import (
2015-09-12 05:51:42 -04:00
"github.com/v2ray/v2ray-core/log"
2015-09-10 18:24:18 -04:00
v2net "github.com/v2ray/v2ray-core/net"
2015-09-05 11:48:38 -04:00
)
2015-09-12 05:51:42 -04:00
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
}
2015-09-12 16:11:54 -04:00
// Point is an single server in V2Ray system.
type Point struct {
2015-09-12 05:51:42 -04:00
port uint16
2015-09-09 18:50:21 -04:00
ichFactory InboundConnectionHandlerFactory
2015-09-12 05:51:42 -04:00
ichConfig []byte
2015-09-09 18:50:21 -04:00
ochFactory OutboundConnectionHandlerFactory
2015-09-12 05:51:42 -04:00
ochConfig []byte
2015-09-05 11:48:38 -04:00
}
2015-09-12 16:11:54 -04:00
// NewPoint returns a new Point server based on given configuration.
2015-09-07 06:00:46 -04:00
// The server is not started at this point.
func NewPoint(config PointConfig) (*Point, error) {
2015-09-12 16:11:54 -04:00
var vpoint = new(Point)
vpoint.port = config.Port()
2015-09-12 05:51:42 -04:00
ichFactory, ok := inboundFactories[config.InboundConfig().Protocol()]
2015-09-12 05:51:42 -04:00
if !ok {
panic(log.Error("Unknown inbound connection handler factory %s", config.InboundConfig().Protocol()))
2015-09-12 05:51:42 -04:00
}
2015-09-11 08:12:26 -04:00
vpoint.ichFactory = ichFactory
2015-09-19 09:35:20 -04:00
vpoint.ichConfig = config.InboundConfig().Content()
2015-09-12 05:51:42 -04:00
ochFactory, ok := outboundFactories[config.OutboundConfig().Protocol()]
2015-09-12 05:51:42 -04:00
if !ok {
panic(log.Error("Unknown outbound connection handler factory %s", config.OutboundConfig().Protocol))
2015-09-12 05:51:42 -04:00
}
2015-09-11 08:12:26 -04:00
vpoint.ochFactory = ochFactory
2015-09-19 09:35:20 -04:00
vpoint.ochConfig = config.OutboundConfig().Content()
2015-09-10 18:24:18 -04:00
2015-09-06 16:10:42 -04:00
return vpoint, nil
2015-09-05 11:48:38 -04:00
}
2015-09-09 18:50:21 -04:00
type InboundConnectionHandlerFactory interface {
2015-09-12 16:11:54 -04:00
Create(vp *Point, config []byte) (InboundConnectionHandler, error)
2015-09-09 18:50:21 -04:00
}
type InboundConnectionHandler interface {
2015-09-06 16:10:42 -04:00
Listen(port uint16) error
2015-09-05 11:48:38 -04:00
}
2015-09-09 18:50:21 -04:00
type OutboundConnectionHandlerFactory interface {
2015-09-12 16:11:54 -04:00
Create(VP *Point, config []byte, dest v2net.Address) (OutboundConnectionHandler, error)
2015-09-09 18:50:21 -04:00
}
type OutboundConnectionHandler interface {
2015-09-12 16:11:54 -04:00
Start(ray OutboundRay) error
2015-09-09 18:50:21 -04:00
}
2015-09-12 16:11:54 -04:00
// Start starts the Point server, and return any error during the process.
2015-09-07 06:00:46 -04:00
// In the case of any errors, the state of the server is unpredicatable.
2015-09-12 16:11:54 -04:00
func (vp *Point) Start() error {
2015-09-12 05:51:42 -04:00
if vp.port <= 0 {
return log.Error("Invalid port %d", vp.port)
2015-09-06 16:10:42 -04:00
}
2015-09-12 05:51:42 -04:00
inboundConnectionHandler, err := vp.ichFactory.Create(vp, vp.ichConfig)
2015-09-09 18:50:21 -04:00
if err != nil {
return err
}
2015-09-12 05:51:42 -04:00
err = inboundConnectionHandler.Listen(vp.port)
2015-09-06 16:10:42 -04:00
return nil
}
2015-09-10 18:24:18 -04:00
2015-09-12 16:11:54 -04:00
func (vp *Point) NewInboundConnectionAccepted(destination v2net.Address) InboundRay {
ray := NewRay()
2015-09-10 18:24:18 -04:00
// TODO: handle error
2015-09-12 05:51:42 -04:00
och, _ := vp.ochFactory.Create(vp, vp.ochConfig, destination)
2015-09-10 18:24:18 -04:00
_ = och.Start(ray)
return ray
}