webrtc: return an error when proxying stream with no tracks

bluenviron · Feb 18, 2024 · 26f8e68 · 26f8e68
1 parent 6e72120
commit 26f8e68
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/protocols/webrtc/peer_connection.go b/internal/protocols/webrtc/peer_connection.go
@@ -227,7 +227,7 @@ outer:
 // GatherIncomingTracks gathers incoming tracks.
 func (co *PeerConnection) GatherIncomingTracks(
 	ctx context.Context,
-	count int,
+	maxCount int,
 ) ([]*IncomingTrack, error) {
 	var tracks []*IncomingTrack
 
@@ -237,7 +237,7 @@ func (co *PeerConnection) GatherIncomingTracks(
 	for {
 		select {
 		case <-t.C:
-			if count == 0 {
+			if maxCount == 0 && len(tracks) != 0 {
 				return tracks, nil
 			}
 			return nil, fmt.Errorf("deadline exceeded while waiting tracks")
@@ -249,7 +249,7 @@ func (co *PeerConnection) GatherIncomingTracks(
 			}
 			tracks = append(tracks, track)
 
-			if len(tracks) == count || len(tracks) >= 2 {
+			if len(tracks) == maxCount || len(tracks) >= 2 {
 				return tracks, nil
 			}