mirror of https://github.com/v2ray/v2ray-core
64 lines
1.1 KiB
Go
64 lines
1.1 KiB
Go
package hub
|
|
|
|
import (
|
|
"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.TCPListener
|
|
connCallback ConnectionHandler
|
|
accepting bool
|
|
}
|
|
|
|
func ListenTCP(port v2net.Port, callback ConnectionHandler) (*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
|
|
}
|
|
tcpListener := &TCPHub{
|
|
listener: listener,
|
|
connCallback: callback,
|
|
}
|
|
go tcpListener.start()
|
|
return tcpListener, 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.AcceptTCP()
|
|
if err != nil {
|
|
if this.accepting {
|
|
log.Warning("Listener: Failed to accept new TCP connection: ", err)
|
|
}
|
|
continue
|
|
}
|
|
go this.connCallback(&TCPConnection{
|
|
conn: conn,
|
|
listener: this,
|
|
})
|
|
}
|
|
}
|
|
|
|
func (this *TCPHub) recycle(conn *net.TCPConn) {
|
|
|
|
}
|