Add RTCP packet reader for output streams

Default interceptors need those.
giongto35 · Feb 13, 2024 · 6258f9a · 6258f9a
1 parent 61b4108
commit 6258f9a
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/pkg/network/webrtc/webrtc.go b/pkg/network/webrtc/webrtc.go
@@ -42,9 +42,20 @@ func (p *Peer) NewCall(vCodec, aCodec string, onICECandidate func(ice any)) (sdp
 	if err != nil {
 		return "", err
 	}
-	if _, err = p.conn.AddTrack(video); err != nil {
+	vs, err := p.conn.AddTrack(video)
+	if err != nil {
 		return "", err
 	}
+	// Read incoming RTCP packets
+	go func() {
+		rtcpBuf := make([]byte, 1500)
+		for {
+			_, _, rtcpErr := vs.Read(rtcpBuf)
+			if rtcpErr != nil {
+				return
+			}
+		}
+	}()
 	p.v = video
 	p.log.Debug().Msgf("Added [%s] track", video.Codec().MimeType)
 
@@ -53,9 +64,20 @@ func (p *Peer) NewCall(vCodec, aCodec string, onICECandidate func(ice any)) (sdp
 	if err != nil {
 		return "", err
 	}
-	if _, err = p.conn.AddTrack(audio); err != nil {
+	as, err := p.conn.AddTrack(audio)
+	if err != nil {
 		return "", err
 	}
+	// Read incoming RTCP packets
+	go func() {
+		rtcpBuf := make([]byte, 1500)
+		for {
+			_, _, rtcpErr := as.Read(rtcpBuf)
+			if rtcpErr != nil {
+				return
+			}
+		}
+	}()
 	p.log.Debug().Msgf("Added [%s] track", audio.Codec().MimeType)
 	p.a = audio