Update session.go

bluenviron · Feb 4, 2025 · 61a8614 · 61a8614
1 parent 36fad69
commit 61a8614
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/internal/servers/rtsp/session.go b/internal/servers/rtsp/session.go
@@ -271,19 +271,22 @@ func (s *session) onPlay(_ *gortsplib.ServerHandlerOnPlayCtx) (*base.Response, e
 		s.transport = s.rsession.SetuppedTransport()
 		s.mutex.Unlock()
 
-		for _, medi := range s.stream.Desc().Medias {
-			if medi.Type == description.MediaTypeVideo {
-				for _, u := range s.stream.CachedUnits {
-					for _, pkt := range u.GetRTPPackets() {
-						err := s.rsession.WritePacketRTP(medi, pkt)
-						if err != nil {
-							break
+		if len(s.stream.CachedUnits) > 0 {
+			lastTimestamp := s.stream.CachedUnits[len(s.stream.CachedUnits)-1].GetRTPPackets()[0].Timestamp
+			for _, medi := range s.stream.Desc().Medias {
+				if medi.Type == description.MediaTypeVideo {
+					for _, u := range s.stream.CachedUnits {
+						for _, pkt := range u.GetRTPPackets() {
+							pkt.Timestamp = lastTimestamp
+							err := s.rsession.WritePacketRTP(medi, pkt)
+							if err != nil {
+								break
+							}
 						}
 					}
 				}
 			}
 		}
-	}
 
 	return &base.Response{
 		StatusCode: base.StatusOK,