1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-10 18:00:43 +00:00
v2fly/testing/servers/tcp/tcp.go

72 lines
1.5 KiB
Go
Raw Normal View History

2015-10-04 18:22:52 +00:00
package tcp
import (
"fmt"
"net"
2016-08-20 18:55:45 +00:00
v2net "v2ray.com/core/common/net"
2015-10-04 18:22:52 +00:00
)
type Server struct {
2015-12-02 20:44:01 +00:00
Port v2net.Port
2015-10-04 18:22:52 +00:00
MsgProcessor func(msg []byte) []byte
2016-08-20 07:12:44 +00:00
SendFirst []byte
2017-02-07 23:26:40 +00:00
Listen v2net.Address
2015-12-15 15:00:47 +00:00
accepting bool
2016-02-03 11:18:52 +00:00
listener *net.TCPListener
2015-10-04 18:22:52 +00:00
}
2015-12-16 22:53:38 +00:00
func (server *Server) Start() (v2net.Destination, error) {
2017-02-07 23:26:40 +00:00
listenerAddr := server.Listen
if listenerAddr == nil {
listenerAddr = v2net.LocalHostIP
}
2015-10-04 18:22:52 +00:00
listener, err := net.ListenTCP("tcp", &net.TCPAddr{
2017-02-07 23:26:40 +00:00
IP: listenerAddr.IP(),
2015-10-04 18:22:52 +00:00
Port: int(server.Port),
Zone: "",
})
if err != nil {
2016-09-20 09:53:05 +00:00
return v2net.Destination{}, err
2015-10-04 18:22:52 +00:00
}
2016-08-15 20:12:11 +00:00
server.Port = v2net.Port(listener.Addr().(*net.TCPAddr).Port)
2016-02-02 15:03:56 +00:00
server.listener = listener
2015-10-04 18:22:52 +00:00
go server.acceptConnections(listener)
localAddr := listener.Addr().(*net.TCPAddr)
2015-12-16 22:53:38 +00:00
return v2net.TCPDestination(v2net.IPAddress(localAddr.IP), v2net.Port(localAddr.Port)), nil
2015-10-04 18:22:52 +00:00
}
func (server *Server) acceptConnections(listener *net.TCPListener) {
2015-12-15 15:00:47 +00:00
server.accepting = true
for server.accepting {
2015-10-04 18:22:52 +00:00
conn, err := listener.Accept()
if err != nil {
2016-11-10 22:41:28 +00:00
fmt.Printf("Failed accept TCP connection: %v\n", err)
2015-10-04 18:22:52 +00:00
continue
}
go server.handleConnection(conn)
}
}
func (server *Server) handleConnection(conn net.Conn) {
2016-08-20 07:12:44 +00:00
if len(server.SendFirst) > 0 {
conn.Write(server.SendFirst)
}
2016-07-17 10:18:34 +00:00
request := make([]byte, 4096)
2015-12-15 15:00:47 +00:00
for true {
2016-07-17 10:18:34 +00:00
nBytes, err := conn.Read(request)
2015-12-15 15:00:47 +00:00
if err != nil {
break
}
2016-07-17 10:18:34 +00:00
response := server.MsgProcessor(request[:nBytes])
2015-12-15 15:00:47 +00:00
conn.Write(response)
2015-10-04 18:22:52 +00:00
}
conn.Close()
}
2015-12-15 15:00:47 +00:00
2016-11-27 20:39:09 +00:00
func (v *Server) Close() {
v.accepting = false
v.listener.Close()
2015-12-15 15:00:47 +00:00
}