1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-10 09:50:43 +00:00
v2fly/transport/internet/tcp_hub.go

121 lines
2.6 KiB
Go
Raw Normal View History

2016-06-14 20:54:08 +00:00
package internet
import (
"net"
2017-01-03 13:53:59 +00:00
2017-02-24 00:05:16 +00:00
"context"
2017-02-21 22:14:07 +00:00
"v2ray.com/core/app/log"
2016-12-04 08:10:47 +00:00
"v2ray.com/core/common/errors"
2016-08-20 18:55:45 +00:00
v2net "v2ray.com/core/common/net"
2016-11-20 20:47:51 +00:00
"v2ray.com/core/common/retry"
2016-06-14 20:54:08 +00:00
)
var (
transportListenerCache = make(map[TransportProtocol]ListenFunc)
2016-06-14 20:54:08 +00:00
)
func RegisterTransportListener(protocol TransportProtocol, listener ListenFunc) error {
if _, found := transportListenerCache[protocol]; found {
return errors.New("Internet|TCPHub: ", protocol, " listener already registered.")
2017-01-03 14:16:48 +00:00
}
transportListenerCache[protocol] = listener
2017-01-03 14:16:48 +00:00
return nil
}
2017-02-24 00:05:16 +00:00
type ListenFunc func(ctx context.Context, address v2net.Address, port v2net.Port) (Listener, error)
2016-09-30 14:53:40 +00:00
2016-06-14 20:54:08 +00:00
type Listener interface {
Accept() (Connection, error)
Close() error
Addr() net.Addr
}
type TCPHub struct {
listener Listener
connCallback ConnectionHandler
2017-02-19 21:29:31 +00:00
closed chan bool
2016-06-14 20:54:08 +00:00
}
2016-10-02 21:43:58 +00:00
func ListenTCP(address v2net.Address, port v2net.Port, callback ConnectionHandler, settings *StreamConfig) (*TCPHub, error) {
2017-02-24 00:05:16 +00:00
ctx := context.Background()
protocol := settings.GetEffectiveProtocol()
2017-02-24 00:05:16 +00:00
transportSettings, err := settings.GetEffectiveTransportSettings()
if err != nil {
return nil, err
}
ctx = ContextWithTransportSettings(ctx, transportSettings)
if settings != nil && settings.HasSecuritySettings() {
securitySettings, err := settings.GetEffectiveSecuritySettings()
if err != nil {
return nil, err
}
ctx = ContextWithSecuritySettings(ctx, securitySettings)
}
listenFunc := transportListenerCache[protocol]
2017-01-03 14:16:48 +00:00
if listenFunc == nil {
return nil, errors.New("Internet|TCPHub: ", protocol, " listener not registered.")
2016-06-14 20:54:08 +00:00
}
2017-02-24 00:05:16 +00:00
listener, err := listenFunc(ctx, address, port)
2016-06-14 20:54:08 +00:00
if err != nil {
2017-02-10 15:50:45 +00:00
return nil, errors.Base(err).Message("Internet|TCPHub: Failed to listen on address: ", address, ":", port)
2016-06-14 20:54:08 +00:00
}
hub := &TCPHub{
listener: listener,
connCallback: callback,
}
go hub.start()
return hub, nil
}
2016-11-27 20:39:09 +00:00
func (v *TCPHub) Close() {
2017-02-21 22:18:25 +00:00
defer func() {
recover()
}()
2017-02-19 21:29:31 +00:00
select {
case <-v.closed:
return
default:
v.listener.Close()
2017-02-21 22:18:25 +00:00
close(v.closed)
2017-02-19 21:29:31 +00:00
}
2016-06-14 20:54:08 +00:00
}
2016-11-27 20:39:09 +00:00
func (v *TCPHub) start() {
2017-02-19 21:29:31 +00:00
for {
select {
case <-v.closed:
return
default:
}
2016-11-20 20:47:51 +00:00
var newConn Connection
2017-02-10 10:41:50 +00:00
err := retry.ExponentialBackoff(10, 500).On(func() error {
2017-02-19 21:29:31 +00:00
select {
case <-v.closed:
return nil
default:
conn, err := v.listener.Accept()
if err != nil {
2017-02-21 22:14:07 +00:00
return errors.Base(err).RequireUserAction().Message("Internet|Listener: Failed to accept new TCP connection.")
2017-02-19 21:29:31 +00:00
}
newConn = conn
2016-11-20 20:47:51 +00:00
return nil
}
})
2017-02-21 22:14:07 +00:00
if err != nil {
if errors.IsActionRequired(err) {
log.Warning(err)
} else {
log.Info(err)
}
continue
}
if newConn != nil {
2016-11-27 20:39:09 +00:00
go v.connCallback(newConn)
2016-06-14 20:54:08 +00:00
}
}
}