1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2026-04-18 19:49:13 -04:00

open ReadBuffer for quic connections

This commit is contained in:
Darien Raymond
2018-12-03 16:01:14 +01:00
parent 3340f81d03
commit 0f63be6340
3 changed files with 10 additions and 12 deletions

View File

@@ -24,7 +24,8 @@ func readOneUDP(r io.Reader) (*Buffer, error) {
return nil, newError("Reader returns too many empty payloads.")
}
func readOne(r io.Reader) (*Buffer, error) {
// ReadBuffer reads a Buffer from the given reader, without allocating large buffer in advance.
func ReadBuffer(r io.Reader) (*Buffer, error) {
// Use an one-byte buffer to wait for incoming payload.
var firstByte [1]byte
nBytes, err := r.Read(firstByte[:])
@@ -163,7 +164,7 @@ type SingleReader struct {
// ReadMultiBuffer implements Reader.
func (r *SingleReader) ReadMultiBuffer() (MultiBuffer, error) {
b, err := readOne(r.Reader)
b, err := ReadBuffer(r.Reader)
if err != nil {
return nil, err
}