v1.0.0

common/mux/reader.go

@@ -0,0 +1,52 @@
+package mux
+
+import (
+	"io"
+
+	"github.com/xtls/xray-core/v1/common/buf"
+	"github.com/xtls/xray-core/v1/common/crypto"
+	"github.com/xtls/xray-core/v1/common/serial"
+)
+
+// PacketReader is an io.Reader that reads whole chunk of Mux frames every time.
+type PacketReader struct {
+	reader io.Reader
+	eof    bool
+}
+
+// NewPacketReader creates a new PacketReader.
+func NewPacketReader(reader io.Reader) *PacketReader {
+	return &PacketReader{
+		reader: reader,
+		eof:    false,
+	}
+}
+
+// ReadMultiBuffer implements buf.Reader.
+func (r *PacketReader) ReadMultiBuffer() (buf.MultiBuffer, error) {
+	if r.eof {
+		return nil, io.EOF
+	}
+
+	size, err := serial.ReadUint16(r.reader)
+	if err != nil {
+		return nil, err
+	}
+
+	if size > buf.Size {
+		return nil, newError("packet size too large: ", size)
+	}
+
+	b := buf.New()
+	if _, err := b.ReadFullFrom(r.reader, int32(size)); err != nil {
+		b.Release()
+		return nil, err
+	}
+	r.eof = true
+	return buf.MultiBuffer{b}, nil
+}
+
+// NewStreamReader creates a new StreamReader.
+func NewStreamReader(reader *buf.BufferedReader) buf.Reader {
+	return crypto.NewChunkStreamReaderWithChunkCount(crypto.PlainChunkSizeParser{}, reader, 1)
+}