fix: rtsp client read timeout

pkg/util/buf_reader_test.go

@@ -0,0 +1,157 @@
+package util
+
+import (
+	"bufio"
+	"bytes"
+	"encoding/binary"
+	"io"
+	"math/rand"
+	"net"
+	"testing"
+)
+
+type limitReader struct {
+	reader io.Reader
+	limit  int
+}
+
+func (l *limitReader) Read(p []byte) (n int, err error) {
+	if len(p) > l.limit {
+		p = p[:l.limit]
+	}
+	n, err = l.reader.Read(p)
+	return
+}
+
+func TestBufReader_Buffered(t *testing.T) {
+	var feeder = make(chan net.Buffers, 100)
+	testReader := NewBufReaderBuffersChan(feeder)
+	testReader.BufLen = 4
+	t.Run("feed", func(t *testing.T) {
+		feeder <- net.Buffers{[]byte{1, 2, 3, 4, 5}, []byte{6, 7, 8, 9, 10}}
+		feeder <- net.Buffers{[]byte{11, 12, 13, 14, 15}, []byte{16, 17, 18, 19, 20}}
+	})
+	t.Run("read", func(t *testing.T) {
+		var b = make([]byte, 10)
+		testReader.Read(b)
+		if !bytes.Equal(b, []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}) {
+			t.Error("read error")
+			return
+		}
+		testReader.Read(b)
+		if !bytes.Equal(b, []byte{11, 12, 13, 14, 15, 16, 17, 18, 19, 20}) {
+			t.Error("read error")
+			return
+		}
+	})
+}
+
+func TestBufRead(t *testing.T) {
+	t.Run(t.Name(), func(t *testing.T) {
+		var testData = []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20}
+		testReader := bytes.NewReader(testData)
+		limitReader := &limitReader{reader: testReader, limit: 2}
+		reader := NewBufReader(limitReader)
+		reader.BufLen = 5
+		b, err := reader.ReadByte()
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		if b != 1 {
+			t.Error("byte read error")
+			return
+		}
+		num, err := reader.ReadBE(4)
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		if num != 0x02030405 {
+			t.Error("read be error")
+			return
+		}
+		b, err = reader.ReadByte()
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		if b != 6 {
+			t.Error("byte read error")
+			return
+		}
+		mem, err := reader.ReadBytes(5)
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		reader2 := mem.NewReader()
+		b, err = reader2.ReadByte()
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		if b != 7 {
+			t.Error("byte read error")
+			return
+		}
+		b, err = reader.ReadByte()
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		if b != 12 {
+			t.Error("byte read error")
+			return
+		}
+	})
+}
+func BenchmarkIoReader(b *testing.B) {
+	b.ResetTimer()
+	b.RunParallel(func(pb *testing.PB) {
+		var testData = make([]byte, 10*1024*1024)
+		var err error
+		for pb.Next() {
+			rand.Read(testData)
+			testReader := bytes.NewReader(testData)
+			reader := bufio.NewReader(testReader)
+			var bb []byte
+			for err == nil {
+				r := rand.Intn(10)
+				if r < 4 {
+					_, err = reader.ReadByte()
+				} else if r < 7 {
+					bb = make([]byte, 4)
+					reader.Read(bb)
+					binary.BigEndian.Uint32(bb)
+				} else {
+					bb = make([]byte, rand.Intn(4096))
+					_, err = reader.Read(bb)
+				}
+			}
+		}
+	})
+}
+
+func BenchmarkBufRead(b *testing.B) {
+	b.ResetTimer()
+	b.RunParallel(func(pb *testing.PB) {
+		var testData = make([]byte, 10*1024*1024)
+		var err error
+		for pb.Next() {
+			rand.Read(testData)
+			testReader := bytes.NewReader(testData)
+			reader := NewBufReaderWithBufLen(testReader, 1024)
+			for err == nil {
+				r := rand.Intn(10)
+				if r < 4 {
+					_, err = reader.ReadByte()
+				} else if r < 7 {
+					_, err = reader.ReadBE(4)
+				} else {
+					_, err = reader.ReadBytes(rand.Intn(4096))
+				}
+			}
+		}
+	})
+}