mirror of
https://github.com/bluenviron/mediamtx
synced 2024-12-13 18:24:58 +00:00
9e6abc6e9f
* rtmp: improve MsgCommandAMF0 * rtmp: fix MsgSetPeerBandwidth * rtmp: add message tests * rtmp: replace implementation with new one * rtmp: rename handshake functions * rtmp: avoid calling useless function * rtmp: use time.Duration for PTSDelta * rtmp: fix decoding chunks with relevant size * rtmp: rewrite implementation of rtmp connection * rtmp: fix tests * rtmp: improve error message * rtmp: replace h264 config implementation * link against github.com/notedit/rtmp * normalize MessageStreamID * rtmp: make acknowledge optional * rtmp: fix decoding of chunk2 + chunk3 * avoid using encoding/binary
53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
package message //nolint:dupl
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/chunk"
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/rawmessage"
|
|
)
|
|
|
|
// MsgUserControlSetBufferLength is a user control message.
|
|
type MsgUserControlSetBufferLength struct {
|
|
StreamID uint32
|
|
BufferLength uint32
|
|
}
|
|
|
|
// Unmarshal implements Message.
|
|
func (m *MsgUserControlSetBufferLength) Unmarshal(raw *rawmessage.Message) error {
|
|
if raw.ChunkStreamID != ControlChunkStreamID {
|
|
return fmt.Errorf("unexpected chunk stream ID")
|
|
}
|
|
|
|
if len(raw.Body) != 10 {
|
|
return fmt.Errorf("invalid body size")
|
|
}
|
|
|
|
m.StreamID = uint32(raw.Body[2])<<24 | uint32(raw.Body[3])<<16 | uint32(raw.Body[4])<<8 | uint32(raw.Body[5])
|
|
m.BufferLength = uint32(raw.Body[6])<<24 | uint32(raw.Body[7])<<16 | uint32(raw.Body[8])<<8 | uint32(raw.Body[9])
|
|
|
|
return nil
|
|
}
|
|
|
|
// Marshal implements Message.
|
|
func (m MsgUserControlSetBufferLength) Marshal() (*rawmessage.Message, error) {
|
|
buf := make([]byte, 10)
|
|
|
|
buf[0] = byte(UserControlTypeSetBufferLength >> 8)
|
|
buf[1] = byte(UserControlTypeSetBufferLength)
|
|
buf[2] = byte(m.StreamID >> 24)
|
|
buf[3] = byte(m.StreamID >> 16)
|
|
buf[4] = byte(m.StreamID >> 8)
|
|
buf[5] = byte(m.StreamID)
|
|
buf[6] = byte(m.BufferLength >> 24)
|
|
buf[7] = byte(m.BufferLength >> 16)
|
|
buf[8] = byte(m.BufferLength >> 8)
|
|
buf[9] = byte(m.BufferLength)
|
|
|
|
return &rawmessage.Message{
|
|
ChunkStreamID: ControlChunkStreamID,
|
|
Type: chunk.MessageTypeUserControl,
|
|
Body: buf,
|
|
}, nil
|
|
}
|