From dcdb0cf13fb11f7fdd7983790ac4cd315ac70086 Mon Sep 17 00:00:00 2001
From: aler9 <46489434+aler9@users.noreply.github.com>
Date: Sat, 14 May 2022 00:19:43 +0200
Subject: [PATCH] rtmp: fix timestamp in MessageWriter

---
 internal/rtmp/base/messagewriter.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/internal/rtmp/base/messagewriter.go b/internal/rtmp/base/messagewriter.go
index d527b374..b63e4420 100644
--- a/internal/rtmp/base/messagewriter.go
+++ b/internal/rtmp/base/messagewriter.go
@@ -37,6 +37,7 @@ func (wc *messageWriterChunkStream) write(msg *Message) error {
 			case wc.lastMessageStreamID == nil || *wc.lastMessageStreamID != msg.MessageStreamID:
 				err := Chunk0{
 					ChunkStreamID:   msg.ChunkStreamID,
+					Timestamp:       msg.Timestamp,
 					Type:            msg.Type,
 					MessageStreamID: msg.MessageStreamID,
 					BodyLen:         uint32(bodyLen),
@@ -46,7 +47,7 @@ func (wc *messageWriterChunkStream) write(msg *Message) error {
 					return err
 				}
 
-			case wc.lastTimestampDelta == nil || *wc.lastType != msg.Type || *wc.lastBodyLen != bodyLen:
+			case *wc.lastType != msg.Type || *wc.lastBodyLen != bodyLen:
 				err := Chunk1{
 					ChunkStreamID:  msg.ChunkStreamID,
 					TimestampDelta: *timestampDelta,
@@ -58,7 +59,7 @@ func (wc *messageWriterChunkStream) write(msg *Message) error {
 					return err
 				}
 
-			case *wc.lastTimestampDelta != *timestampDelta:
+			case wc.lastTimestampDelta == nil || *wc.lastTimestampDelta != *timestampDelta:
 				err := Chunk2{
 					ChunkStreamID:  msg.ChunkStreamID,
 					TimestampDelta: *timestampDelta,