2015-10-04 14:22:52 -04:00
|
|
|
package tcp
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2017-04-16 15:31:28 -04:00
|
|
|
"io"
|
2015-10-04 14:22:52 -04:00
|
|
|
|
2018-07-02 09:26:42 -04:00
|
|
|
"v2ray.com/core/common/buf"
|
2017-08-29 08:32:54 -04:00
|
|
|
"v2ray.com/core/common/net"
|
2018-07-02 09:26:42 -04:00
|
|
|
"v2ray.com/core/common/task"
|
|
|
|
"v2ray.com/core/transport/pipe"
|
2015-10-04 14:22:52 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
type Server struct {
|
2017-08-29 08:32:54 -04:00
|
|
|
Port net.Port
|
2015-10-04 14:22:52 -04:00
|
|
|
MsgProcessor func(msg []byte) []byte
|
2018-04-18 17:57:28 -04:00
|
|
|
ShouldClose bool
|
2016-08-20 03:12:44 -04:00
|
|
|
SendFirst []byte
|
2017-08-29 08:32:54 -04:00
|
|
|
Listen net.Address
|
2016-02-03 06:18:52 -05:00
|
|
|
listener *net.TCPListener
|
2015-10-04 14:22:52 -04:00
|
|
|
}
|
|
|
|
|
2017-08-29 08:32:54 -04:00
|
|
|
func (server *Server) Start() (net.Destination, error) {
|
2017-02-07 18:26:40 -05:00
|
|
|
listenerAddr := server.Listen
|
|
|
|
if listenerAddr == nil {
|
2017-08-29 08:32:54 -04:00
|
|
|
listenerAddr = net.LocalHostIP
|
2017-02-07 18:26:40 -05:00
|
|
|
}
|
2015-10-04 14:22:52 -04:00
|
|
|
listener, err := net.ListenTCP("tcp", &net.TCPAddr{
|
2017-02-07 18:26:40 -05:00
|
|
|
IP: listenerAddr.IP(),
|
2015-10-04 14:22:52 -04:00
|
|
|
Port: int(server.Port),
|
|
|
|
Zone: "",
|
|
|
|
})
|
|
|
|
if err != nil {
|
2017-08-29 08:32:54 -04:00
|
|
|
return net.Destination{}, err
|
2015-10-04 14:22:52 -04:00
|
|
|
}
|
2017-08-29 08:32:54 -04:00
|
|
|
server.Port = net.Port(listener.Addr().(*net.TCPAddr).Port)
|
2016-02-02 10:03:56 -05:00
|
|
|
server.listener = listener
|
2015-10-04 14:22:52 -04:00
|
|
|
go server.acceptConnections(listener)
|
2017-08-29 08:32:54 -04:00
|
|
|
localAddr := listener.Addr().(*net.TCPAddr)
|
|
|
|
return net.TCPDestination(net.IPAddress(localAddr.IP), net.Port(localAddr.Port)), nil
|
2015-10-04 14:22:52 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func (server *Server) acceptConnections(listener *net.TCPListener) {
|
2017-10-20 17:36:14 -04:00
|
|
|
for {
|
2015-10-04 14:22:52 -04:00
|
|
|
conn, err := listener.Accept()
|
|
|
|
if err != nil {
|
2016-11-10 17:41:28 -05:00
|
|
|
fmt.Printf("Failed accept TCP connection: %v\n", err)
|
2017-10-20 17:36:14 -04:00
|
|
|
return
|
2015-10-04 14:22:52 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
go server.handleConnection(conn)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (server *Server) handleConnection(conn net.Conn) {
|
2016-08-20 03:12:44 -04:00
|
|
|
if len(server.SendFirst) > 0 {
|
|
|
|
conn.Write(server.SendFirst)
|
|
|
|
}
|
2018-07-02 09:26:42 -04:00
|
|
|
|
|
|
|
pReader, pWriter := pipe.New(pipe.WithoutSizeLimit())
|
|
|
|
err := task.Run(task.Parallel(func() error {
|
2018-07-02 09:40:32 -04:00
|
|
|
defer pWriter.Close() // nolint: errcheck
|
|
|
|
|
2018-07-02 09:26:42 -04:00
|
|
|
for {
|
|
|
|
b := buf.New()
|
|
|
|
if err := b.AppendSupplier(buf.ReadFrom(conn)); err != nil {
|
|
|
|
if err == io.EOF {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
copy(b.Bytes(), server.MsgProcessor(b.Bytes()))
|
|
|
|
if err := pWriter.WriteMultiBuffer(buf.NewMultiBufferValue(b)); err != nil {
|
|
|
|
return err
|
2017-04-16 15:31:28 -04:00
|
|
|
}
|
|
|
|
}
|
2018-07-02 09:26:42 -04:00
|
|
|
}, func() error {
|
2018-07-02 09:40:32 -04:00
|
|
|
defer pReader.CloseError()
|
|
|
|
|
2018-07-02 09:26:42 -04:00
|
|
|
w := buf.NewWriter(conn)
|
|
|
|
for {
|
|
|
|
mb, err := pReader.ReadMultiBuffer()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if err := w.WriteMultiBuffer(mb); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2018-04-18 17:57:28 -04:00
|
|
|
}
|
2018-07-02 09:26:42 -04:00
|
|
|
}))()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println("failed to transfer data: ", err.Error())
|
2015-10-04 14:22:52 -04:00
|
|
|
}
|
2018-07-02 09:26:42 -04:00
|
|
|
|
|
|
|
conn.Close() // nolint: errcheck
|
2015-10-04 14:22:52 -04:00
|
|
|
}
|
2015-12-15 10:00:47 -05:00
|
|
|
|
2018-02-08 09:39:46 -05:00
|
|
|
func (server *Server) Close() error {
|
|
|
|
return server.listener.Close()
|
2015-12-15 10:00:47 -05:00
|
|
|
}
|