mirror of
https://github.com/aler9/rtsp-simple-server
synced 2025-10-06 08:06:58 +08: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
43 lines
983 B
Go
43 lines
983 B
Go
package chunk
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
// Chunk2 is a type 2 chunk.
|
|
// Neither the stream ID nor the
|
|
// message length is included; this chunk has the same stream ID and
|
|
// message length as the preceding chunk.
|
|
type Chunk2 struct {
|
|
ChunkStreamID byte
|
|
TimestampDelta uint32
|
|
Body []byte
|
|
}
|
|
|
|
// Read reads the chunk.
|
|
func (c *Chunk2) Read(r io.Reader, chunkBodyLen uint32) error {
|
|
header := make([]byte, 4)
|
|
_, err := io.ReadFull(r, header)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.ChunkStreamID = header[0] & 0x3F
|
|
c.TimestampDelta = uint32(header[1])<<16 | uint32(header[2])<<8 | uint32(header[3])
|
|
|
|
c.Body = make([]byte, chunkBodyLen)
|
|
_, err = io.ReadFull(r, c.Body)
|
|
return err
|
|
}
|
|
|
|
// Marshal writes the chunk.
|
|
func (c Chunk2) Marshal() ([]byte, error) {
|
|
buf := make([]byte, 4+len(c.Body))
|
|
buf[0] = 2<<6 | c.ChunkStreamID
|
|
buf[1] = byte(c.TimestampDelta >> 16)
|
|
buf[2] = byte(c.TimestampDelta >> 8)
|
|
buf[3] = byte(c.TimestampDelta)
|
|
copy(buf[4:], c.Body)
|
|
return buf, nil
|
|
}
|