mirror of
https://github.com/aler9/rtsp-simple-server
synced 2025-10-05 23:56:54 +08:00
move protocol-related code into internal/protocols (#2572)
This commit is contained in:
45
internal/protocols/rtmp/message/user_control_ping_request.go
Normal file
45
internal/protocols/rtmp/message/user_control_ping_request.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package message //nolint:dupl
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/rawmessage"
|
||||
)
|
||||
|
||||
// UserControlPingRequest is a user control message.
|
||||
type UserControlPingRequest struct {
|
||||
ServerTime uint32
|
||||
}
|
||||
|
||||
// Unmarshal implements Message.
|
||||
func (m *UserControlPingRequest) Unmarshal(raw *rawmessage.Message) error {
|
||||
if raw.ChunkStreamID != ControlChunkStreamID {
|
||||
return fmt.Errorf("unexpected chunk stream ID")
|
||||
}
|
||||
|
||||
if len(raw.Body) != 6 {
|
||||
return fmt.Errorf("invalid body size")
|
||||
}
|
||||
|
||||
m.ServerTime = uint32(raw.Body[2])<<24 | uint32(raw.Body[3])<<16 | uint32(raw.Body[4])<<8 | uint32(raw.Body[5])
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Marshal implements Message.
|
||||
func (m UserControlPingRequest) Marshal() (*rawmessage.Message, error) {
|
||||
buf := make([]byte, 6)
|
||||
|
||||
buf[0] = byte(UserControlTypePingRequest >> 8)
|
||||
buf[1] = byte(UserControlTypePingRequest)
|
||||
buf[2] = byte(m.ServerTime >> 24)
|
||||
buf[3] = byte(m.ServerTime >> 16)
|
||||
buf[4] = byte(m.ServerTime >> 8)
|
||||
buf[5] = byte(m.ServerTime)
|
||||
|
||||
return &rawmessage.Message{
|
||||
ChunkStreamID: ControlChunkStreamID,
|
||||
Type: uint8(TypeUserControl),
|
||||
Body: buf,
|
||||
}, nil
|
||||
}
|
Reference in New Issue
Block a user