support publishing AV1/H265 with OBS 30 (#2217) (#2234)

internal/rtmp/message/extended_sequence_start.go

@@ -1,6 +1,8 @@
 package message
 
 import (
+	"fmt"
+
 	"github.com/bluenviron/mediamtx/internal/rtmp/rawmessage"
 )
 
@@ -8,26 +10,37 @@ import (
 type ExtendedSequenceStart struct {
 	ChunkStreamID   byte
 	MessageStreamID uint32
-	FourCC          [4]byte
+	FourCC          FourCC
 	Config          []byte
 }
 
 // Unmarshal implements Message.
 func (m *ExtendedSequenceStart) Unmarshal(raw *rawmessage.Message) error {
+	if len(raw.Body) < 6 {
+		return fmt.Errorf("not enough bytes")
+	}
+
 	m.ChunkStreamID = raw.ChunkStreamID
 	m.MessageStreamID = raw.MessageStreamID
-	copy(m.FourCC[:], raw.Body[1:5])
+	m.FourCC = FourCC(raw.Body[1])<<24 | FourCC(raw.Body[2])<<16 | FourCC(raw.Body[3])<<8 | FourCC(raw.Body[4])
 	m.Config = raw.Body[5:]
 
 	return nil
 }
 
+func (m ExtendedSequenceStart) marshalBodySize() int {
+	return 5 + len(m.Config)
+}
+
 // Marshal implements Message.
 func (m ExtendedSequenceStart) Marshal() (*rawmessage.Message, error) {
-	body := make([]byte, 5+len(m.Config))
+	body := make([]byte, m.marshalBodySize())
 
 	body[0] = 0b10000000 | byte(ExtendedTypeSequenceStart)
-	copy(body[1:5], m.FourCC[:])
+	body[1] = uint8(m.FourCC >> 24)
+	body[2] = uint8(m.FourCC >> 16)
+	body[3] = uint8(m.FourCC >> 8)
+	body[4] = uint8(m.FourCC)
 	copy(body[5:], m.Config)
 
 	return &rawmessage.Message{