mediamtx/internal/rtmp/message/set_peer_bandwidth.go

47 lines
1.0 KiB
Go

package message //nolint:dupl
import (
"fmt"
"github.com/aler9/mediamtx/internal/rtmp/rawmessage"
)
// SetPeerBandwidth is a set peer bandwidth message.
type SetPeerBandwidth struct {
Value uint32
Type byte
}
// Unmarshal implements Message.
func (m *SetPeerBandwidth) Unmarshal(raw *rawmessage.Message) error {
if raw.ChunkStreamID != ControlChunkStreamID {
return fmt.Errorf("unexpected chunk stream ID")
}
if len(raw.Body) != 5 {
return fmt.Errorf("invalid body size")
}
m.Value = uint32(raw.Body[0])<<24 | uint32(raw.Body[1])<<16 | uint32(raw.Body[2])<<8 | uint32(raw.Body[3])
m.Type = raw.Body[4]
return nil
}
// Marshal implements Message.
func (m *SetPeerBandwidth) Marshal() (*rawmessage.Message, error) {
buf := make([]byte, 5)
buf[0] = byte(m.Value >> 24)
buf[1] = byte(m.Value >> 16)
buf[2] = byte(m.Value >> 8)
buf[3] = byte(m.Value)
buf[4] = m.Type
return &rawmessage.Message{
ChunkStreamID: ControlChunkStreamID,
Type: uint8(TypeSetPeerBandwidth),
Body: buf,
}, nil
}