mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-03 02:04:34 -04:00
75 lines
1.3 KiB
Go
75 lines
1.3 KiB
Go
package hub
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"errors"
|
|
"net"
|
|
|
|
"github.com/v2ray/v2ray-core/common/log"
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
)
|
|
|
|
var (
|
|
ErrorClosedConnection = errors.New("Connection already closed.")
|
|
)
|
|
|
|
type TCPHub struct {
|
|
listener net.Listener
|
|
connCallback ConnectionHandler
|
|
accepting bool
|
|
}
|
|
|
|
func ListenTCP(port v2net.Port, callback ConnectionHandler, tlsConfig *tls.Config) (*TCPHub, error) {
|
|
listener, err := net.ListenTCP("tcp", &net.TCPAddr{
|
|
IP: []byte{0, 0, 0, 0},
|
|
Port: int(port),
|
|
Zone: "",
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var hub *TCPHub
|
|
if tlsConfig != nil {
|
|
tlsListener := tls.NewListener(listener, tlsConfig)
|
|
hub = &TCPHub{
|
|
listener: tlsListener,
|
|
connCallback: callback,
|
|
}
|
|
} else {
|
|
hub = &TCPHub{
|
|
listener: listener,
|
|
connCallback: callback,
|
|
}
|
|
}
|
|
|
|
go hub.start()
|
|
return hub, nil
|
|
}
|
|
|
|
func (this *TCPHub) Close() {
|
|
this.accepting = false
|
|
this.listener.Close()
|
|
this.listener = nil
|
|
}
|
|
|
|
func (this *TCPHub) start() {
|
|
this.accepting = true
|
|
for this.accepting {
|
|
conn, err := this.listener.Accept()
|
|
if err != nil {
|
|
if this.accepting {
|
|
log.Warning("Listener: Failed to accept new TCP connection: ", err)
|
|
}
|
|
continue
|
|
}
|
|
go this.connCallback(&Connection{
|
|
conn: conn,
|
|
listener: this,
|
|
})
|
|
}
|
|
}
|
|
|
|
func (this *TCPHub) recycle(conn *net.TCPConn) {
|
|
|
|
}
|