mirror of
https://github.com/bluenviron/mediamtx
synced 2025-02-02 20:53:02 +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
113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package message
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/bytecounter"
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/chunk"
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/rawmessage"
|
|
)
|
|
|
|
func allocateMessage(raw *rawmessage.Message) (Message, error) {
|
|
switch raw.Type {
|
|
case chunk.MessageTypeSetChunkSize:
|
|
return &MsgSetChunkSize{}, nil
|
|
|
|
case chunk.MessageTypeAcknowledge:
|
|
return &MsgAcknowledge{}, nil
|
|
|
|
case chunk.MessageTypeSetWindowAckSize:
|
|
return &MsgSetWindowAckSize{}, nil
|
|
|
|
case chunk.MessageTypeSetPeerBandwidth:
|
|
return &MsgSetPeerBandwidth{}, nil
|
|
|
|
case chunk.MessageTypeUserControl:
|
|
if len(raw.Body) < 2 {
|
|
return nil, fmt.Errorf("invalid body size")
|
|
}
|
|
|
|
subType := uint16(raw.Body[0])<<8 | uint16(raw.Body[1])
|
|
switch subType {
|
|
case UserControlTypeStreamBegin:
|
|
return &MsgUserControlStreamBegin{}, nil
|
|
|
|
case UserControlTypeStreamEOF:
|
|
return &MsgUserControlStreamEOF{}, nil
|
|
|
|
case UserControlTypeStreamDry:
|
|
return &MsgUserControlStreamDry{}, nil
|
|
|
|
case UserControlTypeSetBufferLength:
|
|
return &MsgUserControlSetBufferLength{}, nil
|
|
|
|
case UserControlTypeStreamIsRecorded:
|
|
return &MsgUserControlStreamIsRecorded{}, nil
|
|
|
|
case UserControlTypePingRequest:
|
|
return &MsgUserControlPingRequest{}, nil
|
|
|
|
case UserControlTypePingResponse:
|
|
return &MsgUserControlPingResponse{}, nil
|
|
|
|
default:
|
|
return nil, fmt.Errorf("invalid user control type")
|
|
}
|
|
|
|
case chunk.MessageTypeCommandAMF0:
|
|
return &MsgCommandAMF0{}, nil
|
|
|
|
case chunk.MessageTypeDataAMF0:
|
|
return &MsgDataAMF0{}, nil
|
|
|
|
case chunk.MessageTypeAudio:
|
|
return &MsgAudio{}, nil
|
|
|
|
case chunk.MessageTypeVideo:
|
|
return &MsgVideo{}, nil
|
|
|
|
default:
|
|
return nil, fmt.Errorf("unhandled message type (%v)", raw.Type)
|
|
}
|
|
}
|
|
|
|
// Reader is a message reader.
|
|
type Reader struct {
|
|
r *rawmessage.Reader
|
|
}
|
|
|
|
// NewReader allocates a Reader.
|
|
func NewReader(r *bytecounter.Reader, onAckNeeded func(uint32) error) *Reader {
|
|
return &Reader{
|
|
r: rawmessage.NewReader(r, onAckNeeded),
|
|
}
|
|
}
|
|
|
|
// Read reads a Message.
|
|
func (r *Reader) Read() (Message, error) {
|
|
raw, err := r.r.Read()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
msg, err := allocateMessage(raw)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = msg.Unmarshal(raw)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch tmsg := msg.(type) {
|
|
case *MsgSetChunkSize:
|
|
r.r.SetChunkSize(tmsg.Value)
|
|
|
|
case *MsgSetWindowAckSize:
|
|
r.r.SetWindowAckSize(tmsg.Value)
|
|
}
|
|
|
|
return msg, nil
|
|
}
|