mirror of https://github.com/v2ray/v2ray-core
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package mux
|
|
|
|
import "io"
|
|
import "v2ray.com/core/common/buf"
|
|
import "v2ray.com/core/common/serial"
|
|
|
|
type muxReader struct {
|
|
reader io.Reader
|
|
remainingLength int
|
|
buffer *buf.Buffer
|
|
}
|
|
|
|
func NewReader(reader buf.Reader) *muxReader {
|
|
return &muxReader{
|
|
reader: buf.ToBytesReader(reader),
|
|
buffer: buf.NewLocal(1024),
|
|
}
|
|
}
|
|
|
|
func (r *muxReader) ReadMetadata() (*FrameMetadata, error) {
|
|
b := r.buffer
|
|
b.Clear()
|
|
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(r.reader, 2)); err != nil {
|
|
return nil, err
|
|
}
|
|
metaLen := serial.BytesToUint16(b.Bytes())
|
|
b.Clear()
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(r.reader, int(metaLen))); err != nil {
|
|
return nil, err
|
|
}
|
|
return ReadFrameFrom(b.Bytes())
|
|
}
|
|
|
|
func (r *muxReader) Read() (*buf.Buffer, bool, error) {
|
|
b := buf.New()
|
|
var dataLen int
|
|
if r.remainingLength > 0 {
|
|
dataLen = r.remainingLength
|
|
r.remainingLength = 0
|
|
} else {
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(r.reader, 2)); err != nil {
|
|
return nil, false, err
|
|
}
|
|
dataLen = int(serial.BytesToUint16(b.Bytes()))
|
|
b.Clear()
|
|
}
|
|
|
|
if dataLen > buf.Size {
|
|
r.remainingLength = dataLen - buf.Size
|
|
dataLen = buf.Size
|
|
}
|
|
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(r.reader, dataLen)); err != nil {
|
|
return nil, false, err
|
|
}
|
|
|
|
return b, (r.remainingLength > 0), nil
|
|
}
|