mirror of https://github.com/q191201771/lal.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
// Copyright 2020, Chef. All rights reserved.
|
|
// https://github.com/q191201771/lal
|
|
//
|
|
// Use of this source code is governed by a MIT-style license
|
|
// that can be found in the License file.
|
|
//
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
package rtsp
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
|
|
"github.com/q191201771/naza/pkg/bele"
|
|
)
|
|
|
|
// rfc2326 10.12 Embedded (Interleaved) Binary Data
|
|
|
|
func readInterleaved(r *bufio.Reader) (isInterleaved bool, packet []byte, channel uint8, err error) {
|
|
flag, err := r.ReadByte()
|
|
if err != nil {
|
|
return false, nil, 0, err
|
|
}
|
|
|
|
if flag != Interleaved {
|
|
_ = r.UnreadByte()
|
|
return false, nil, 0, nil
|
|
}
|
|
|
|
channel, err = r.ReadByte()
|
|
if err != nil {
|
|
return false, nil, 0, err
|
|
}
|
|
rtpLenBuf := make([]byte, 2)
|
|
_, err = io.ReadFull(r, rtpLenBuf)
|
|
if err != nil {
|
|
return false, nil, 0, err
|
|
}
|
|
rtpLen := int(bele.BEUint16(rtpLenBuf))
|
|
// TODO chef: 这里为了安全性，应该检查大小
|
|
rtpBuf := make([]byte, rtpLen)
|
|
_, err = io.ReadFull(r, rtpBuf)
|
|
if err != nil {
|
|
return false, nil, 0, err
|
|
}
|
|
|
|
return true, rtpBuf, channel, nil
|
|
}
|
|
|
|
func packInterleaved(channel int, rtpPacket []byte) []byte {
|
|
ret := make([]byte, 4+len(rtpPacket))
|
|
ret[0] = Interleaved
|
|
ret[1] = uint8(channel)
|
|
bele.BEPutUint16(ret[2:], uint16(len(rtpPacket)))
|
|
copy(ret[4:], rtpPacket)
|
|
return ret
|
|
}
|