unexport RTCPSender, RTCPReceiver, LossDetector, Reorderer (#667)

2025-10-20 05:54:47 +08:00 · 2024-12-24 21:21:11 +01:00
parent 93b012b3ca
commit 8c4a3ca018
16 changed files with 1186 additions and 505 deletions
--- a/internal/rtcpsender/rtcpsender_test.go
+++ b/internal/rtcpsender/rtcpsender_test.go
@@ -0,0 +1,100 @@
+package rtcpsender
+
+import (
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/pion/rtcp"
+	"github.com/pion/rtp"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRTCPSender(t *testing.T) {
+	var curTime time.Time
+	var mutex sync.Mutex
+
+	setCurTime := func(v time.Time) {
+		mutex.Lock()
+		defer mutex.Unlock()
+		curTime = v
+	}
+
+	sent := make(chan struct{})
+
+	rs := &RTCPSender{
+		ClockRate: 90000,
+		Period:    100 * time.Millisecond,
+		TimeNow: func() time.Time {
+			mutex.Lock()
+			defer mutex.Unlock()
+			return curTime
+		},
+		WritePacketRTCP: func(pkt rtcp.Packet) {
+			require.Equal(t, &rtcp.SenderReport{
+				SSRC: 0xba9da416,
+				NTPTime: func() uint64 {
+					// timeDiff = (24 - 22) = 2
+					// 21 + 2 = 23
+					d := time.Date(2008, 5, 20, 22, 15, 23, 0, time.UTC)
+					s := uint64(d.UnixNano()) + 2208988800*1000000000
+					return (s/1000000000)<<32 | (s % 1000000000)
+				}(),
+				RTPTime:     1287987768 + 2*90000,
+				PacketCount: 3,
+				OctetCount:  6,
+			}, pkt)
+			close(sent)
+		},
+	}
+	rs.Initialize()
+	defer rs.Close()
+
+	setCurTime(time.Date(2008, 5, 20, 22, 16, 20, 0, time.UTC))
+	rtpPkt := rtp.Packet{
+		Header: rtp.Header{
+			Version:        2,
+			Marker:         true,
+			PayloadType:    96,
+			SequenceNumber: 946,
+			Timestamp:      1287987768,
+			SSRC:           0xba9da416,
+		},
+		Payload: []byte("\x00\x00"),
+	}
+	ts := time.Date(2008, 0o5, 20, 22, 15, 20, 0, time.UTC)
+	rs.ProcessPacket(&rtpPkt, ts, true)
+
+	setCurTime(time.Date(2008, 5, 20, 22, 16, 22, 0, time.UTC))
+	rtpPkt = rtp.Packet{
+		Header: rtp.Header{
+			Version:        2,
+			Marker:         true,
+			PayloadType:    96,
+			SequenceNumber: 947,
+			Timestamp:      1287987768,
+			SSRC:           0xba9da416,
+		},
+		Payload: []byte("\x00\x00"),
+	}
+	ts = time.Date(2008, 0o5, 20, 22, 15, 21, 0, time.UTC)
+	rs.ProcessPacket(&rtpPkt, ts, true)
+
+	rtpPkt = rtp.Packet{
+		Header: rtp.Header{
+			Version:        2,
+			Marker:         true,
+			PayloadType:    96,
+			SequenceNumber: 948,
+			Timestamp:      1287987768,
+			SSRC:           0xba9da416,
+		},
+		Payload: []byte("\x00\x00"),
+	}
+	ts = time.Date(2008, 0o5, 20, 22, 15, 22, 0, time.UTC)
+	rs.ProcessPacket(&rtpPkt, ts, false)
+
+	setCurTime(time.Date(2008, 5, 20, 22, 16, 24, 0, time.UTC))
+
+	<-sent
+}