Merge pull request #331 from jaesung9507/panic

[fix] rtmp: panic in SetChunkSize msg while the length is too short
q191201771 · Nov 14, 2023 · 92252c8 · 92252c8
2 parents 181e594 + 5f32142
commit 92252c8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pkg/rtmp/chunk_composer.go b/pkg/rtmp/chunk_composer.go
@@ -183,8 +183,10 @@ func (c *ChunkComposer) RunLoop(reader io.Reader, cb OnCompleteMessage) error {
 		if stream.msg.Len() == stream.header.MsgLen {
 			// 对端设置了chunk size
 			if stream.header.MsgTypeId == base.RtmpTypeIdSetChunkSize {
-				val := bele.BeUint32(stream.msg.buff.Bytes())
-				c.SetPeerChunkSize(val)
+				if buf := stream.msg.buff.Bytes(); len(buf) >= 4 {
+					val := bele.BeUint32(buf)
+					c.SetPeerChunkSize(val)
+				}
 			}
 
 			stream.header.Csid = csid