1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-23 02:26:49 -05:00
v2fly/app/point/point.go

106 lines
2.8 KiB
Go
Raw Normal View History

2015-10-14 08:51:19 -04:00
package point
2015-09-05 11:48:38 -04:00
import (
2015-10-30 17:50:45 -04:00
"github.com/v2ray/v2ray-core/app/point/config"
2015-09-19 18:50:21 -04:00
"github.com/v2ray/v2ray-core/common/log"
v2net "github.com/v2ray/v2ray-core/common/net"
2015-10-13 06:27:50 -04:00
"github.com/v2ray/v2ray-core/common/retry"
"github.com/v2ray/v2ray-core/proxy/common/connhandler"
2015-10-14 08:51:19 -04:00
"github.com/v2ray/v2ray-core/transport/ray"
2015-09-05 11:48:38 -04:00
)
2015-09-12 16:11:54 -04:00
// Point is an single server in V2Ray system.
type Point struct {
2015-10-06 18:30:44 -04:00
port uint16
ich connhandler.InboundConnectionHandler
och connhandler.OutboundConnectionHandler
2015-10-31 19:11:41 -04:00
idh []*InboundDetourHandler
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.
2015-10-06 17:11:08 -04:00
func NewPoint(pConfig config.PointConfig) (*Point, error) {
2015-09-12 16:11:54 -04:00
var vpoint = new(Point)
2015-10-06 17:11:08 -04:00
vpoint.port = pConfig.Port()
2015-09-12 05:51:42 -04:00
ichFactory := connhandler.GetInboundConnectionHandlerFactory(pConfig.InboundConfig().Protocol())
2015-10-14 08:51:19 -04:00
if ichFactory == nil {
2015-10-07 04:05:11 -04:00
log.Error("Unknown inbound connection handler factory %s", pConfig.InboundConfig().Protocol())
2015-10-13 07:55:06 -04:00
return nil, config.BadConfiguration
2015-09-12 05:51:42 -04:00
}
ichConfig := pConfig.InboundConfig().Settings()
2015-10-06 18:30:44 -04:00
ich, err := ichFactory.Create(vpoint, ichConfig)
if err != nil {
log.Error("Failed to create inbound connection handler: %v", err)
return nil, err
}
vpoint.ich = ich
2015-09-12 05:51:42 -04:00
ochFactory := connhandler.GetOutboundConnectionHandlerFactory(pConfig.OutboundConfig().Protocol())
2015-10-14 08:51:19 -04:00
if ochFactory == nil {
2015-10-07 04:05:11 -04:00
log.Error("Unknown outbound connection handler factory %s", pConfig.OutboundConfig().Protocol())
2015-10-13 07:55:06 -04:00
return nil, config.BadConfiguration
2015-09-12 05:51:42 -04:00
}
ochConfig := pConfig.OutboundConfig().Settings()
2015-10-14 08:51:19 -04:00
och, err := ochFactory.Create(ochConfig)
2015-10-06 18:30:44 -04:00
if err != nil {
log.Error("Failed to create outbound connection handler: %v", err)
return nil, err
}
vpoint.och = och
2015-09-10 18:24:18 -04:00
2015-10-31 19:11:41 -04:00
detours := pConfig.InboundDetours()
if len(detours) > 0 {
vpoint.idh = make([]*InboundDetourHandler, len(detours))
for idx, detourConfig := range detours {
detourHandler := &InboundDetourHandler{
point: vpoint,
config: detourConfig,
}
err := detourHandler.Initialize()
if err != nil {
return nil, err
}
vpoint.idh[idx] = detourHandler
}
}
2015-09-06 16:10:42 -04:00
return vpoint, nil
2015-09-05 11:48:38 -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 {
2015-10-07 04:05:11 -04:00
log.Error("Invalid port %d", vp.port)
2015-10-13 07:55:06 -04:00
return config.BadConfiguration
2015-09-06 16:10:42 -04:00
}
2015-09-22 12:11:55 -04:00
2015-10-31 19:11:41 -04:00
err := retry.Timed(100 /* times */, 100 /* ms */).On(func() error {
2015-10-13 18:08:52 -04:00
err := vp.ich.Listen(vp.port)
2015-10-31 19:11:41 -04:00
if err != nil {
return err
2015-10-13 19:15:29 -04:00
}
2015-10-31 19:11:41 -04:00
log.Warning("Point server started on port %d", vp.port)
return nil
2015-10-13 06:27:50 -04:00
})
2015-10-31 19:11:41 -04:00
if err != nil {
return err
}
for _, detourHandler := range vp.idh {
err := detourHandler.Start()
if err != nil {
return err
}
}
return nil
2015-09-06 16:10:42 -04:00
}
2015-09-10 18:24:18 -04:00
2015-10-14 08:51:19 -04:00
func (p *Point) DispatchToOutbound(packet v2net.Packet) ray.InboundRay {
direct := ray.NewRay()
go p.och.Dispatch(packet, direct)
return direct
2015-09-10 18:24:18 -04:00
}