Move common/net/connection.go into cnc folder

common/net/cnc/connection.go

@@ -0,0 +1,160 @@
+package cnc
+
+import (
+	"io"
+	"time"
+
+	"github.com/xtls/xray-core/common"
+	"github.com/xtls/xray-core/common/buf"
+	"github.com/xtls/xray-core/common/net"
+	"github.com/xtls/xray-core/common/signal/done"
+)
+
+type ConnectionOption func(*connection)
+
+func ConnectionLocalAddr(a net.Addr) ConnectionOption {
+	return func(c *connection) {
+		c.local = a
+	}
+}
+
+func ConnectionRemoteAddr(a net.Addr) ConnectionOption {
+	return func(c *connection) {
+		c.remote = a
+	}
+}
+
+func ConnectionInput(writer io.Writer) ConnectionOption {
+	return func(c *connection) {
+		c.writer = buf.NewWriter(writer)
+	}
+}
+
+func ConnectionInputMulti(writer buf.Writer) ConnectionOption {
+	return func(c *connection) {
+		c.writer = writer
+	}
+}
+
+func ConnectionOutput(reader io.Reader) ConnectionOption {
+	return func(c *connection) {
+		c.reader = &buf.BufferedReader{Reader: buf.NewReader(reader)}
+	}
+}
+
+func ConnectionOutputMulti(reader buf.Reader) ConnectionOption {
+	return func(c *connection) {
+		c.reader = &buf.BufferedReader{Reader: reader}
+	}
+}
+
+func ConnectionOutputMultiUDP(reader buf.Reader) ConnectionOption {
+	return func(c *connection) {
+		c.reader = &buf.BufferedReader{
+			Reader:  reader,
+			Spliter: buf.SplitFirstBytes,
+		}
+	}
+}
+
+func ConnectionOnClose(n io.Closer) ConnectionOption {
+	return func(c *connection) {
+		c.onClose = n
+	}
+}
+
+func NewConnection(opts ...ConnectionOption) net.Conn {
+	c := &connection{
+		done: done.New(),
+		local: &net.TCPAddr{
+			IP:   []byte{0, 0, 0, 0},
+			Port: 0,
+		},
+		remote: &net.TCPAddr{
+			IP:   []byte{0, 0, 0, 0},
+			Port: 0,
+		},
+	}
+
+	for _, opt := range opts {
+		opt(c)
+	}
+
+	return c
+}
+
+type connection struct {
+	reader  *buf.BufferedReader
+	writer  buf.Writer
+	done    *done.Instance
+	onClose io.Closer
+	local   net.Addr
+	remote  net.Addr
+}
+
+func (c *connection) Read(b []byte) (int, error) {
+	return c.reader.Read(b)
+}
+
+// ReadMultiBuffer implements buf.Reader.
+func (c *connection) ReadMultiBuffer() (buf.MultiBuffer, error) {
+	return c.reader.ReadMultiBuffer()
+}
+
+// Write implements net.Conn.Write().
+func (c *connection) Write(b []byte) (int, error) {
+	if c.done.Done() {
+		return 0, io.ErrClosedPipe
+	}
+
+	l := len(b)
+	mb := make(buf.MultiBuffer, 0, l/buf.Size+1)
+	mb = buf.MergeBytes(mb, b)
+	return l, c.writer.WriteMultiBuffer(mb)
+}
+
+func (c *connection) WriteMultiBuffer(mb buf.MultiBuffer) error {
+	if c.done.Done() {
+		buf.ReleaseMulti(mb)
+		return io.ErrClosedPipe
+	}
+
+	return c.writer.WriteMultiBuffer(mb)
+}
+
+// Close implements net.Conn.Close().
+func (c *connection) Close() error {
+	common.Must(c.done.Close())
+	common.Interrupt(c.reader)
+	common.Close(c.writer)
+	if c.onClose != nil {
+		return c.onClose.Close()
+	}
+
+	return nil
+}
+
+// LocalAddr implements net.Conn.LocalAddr().
+func (c *connection) LocalAddr() net.Addr {
+	return c.local
+}
+
+// RemoteAddr implements net.Conn.RemoteAddr().
+func (c *connection) RemoteAddr() net.Addr {
+	return c.remote
+}
+
+// SetDeadline implements net.Conn.SetDeadline().
+func (c *connection) SetDeadline(t time.Time) error {
+	return nil
+}
+
+// SetReadDeadline implements net.Conn.SetReadDeadline().
+func (c *connection) SetReadDeadline(t time.Time) error {
+	return nil
+}
+
+// SetWriteDeadline implements net.Conn.SetWriteDeadline().
+func (c *connection) SetWriteDeadline(t time.Time) error {
+	return nil
+}