From eceada8b31a93cd43a689f527a03d43e33b71e0d Mon Sep 17 00:00:00 2001 From: Raja Subramanian Date: Fri, 15 Nov 2024 03:13:53 +0530 Subject: [PATCH] use spatialLayer var (#3178) * use spatialLayer var * lower end check --- pkg/sfu/receiver.go | 4 ++-- pkg/sfu/streamtrackermanager.go | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pkg/sfu/receiver.go b/pkg/sfu/receiver.go index aa52454da..e2a453e80 100644 --- a/pkg/sfu/receiver.go +++ b/pkg/sfu/receiver.go @@ -749,9 +749,9 @@ func (w *WebRTCReceiver) forwardRTP(layer int32, buff *buffer.Buffer) { // track video layers if w.Kind() == webrtc.RTPCodecTypeVideo { if spatialTrackers[spatialLayer] == nil { - spatialTrackers[spatialLayer] = w.streamTrackerManager.GetTracker(pkt.Spatial) + spatialTrackers[spatialLayer] = w.streamTrackerManager.GetTracker(spatialLayer) if spatialTrackers[spatialLayer] == nil { - spatialTrackers[spatialLayer] = w.streamTrackerManager.AddTracker(pkt.Spatial) + spatialTrackers[spatialLayer] = w.streamTrackerManager.AddTracker(spatialLayer) } } if spatialTrackers[spatialLayer] != nil { diff --git a/pkg/sfu/streamtrackermanager.go b/pkg/sfu/streamtrackermanager.go index adacdfcf7..a0e35d65a 100644 --- a/pkg/sfu/streamtrackermanager.go +++ b/pkg/sfu/streamtrackermanager.go @@ -349,7 +349,7 @@ func (s *StreamTrackerManager) GetTracker(layer int32) streamtracker.StreamTrack s.lock.RLock() defer s.lock.RUnlock() - if int(layer) >= len(s.trackers) { + if layer < 0 || int(layer) >= len(s.trackers) { s.logger.Errorw("unexpected layer", nil, "layer", layer) return nil }