mirror of
https://github.com/Monibuca/engine.git
synced 2025-10-08 01:50:03 +08:00
90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package track
|
||
|
||
import (
|
||
"github.com/pion/rtp"
|
||
"go.uber.org/zap"
|
||
. "m7s.live/engine/v4/common"
|
||
"m7s.live/engine/v4/config"
|
||
"m7s.live/engine/v4/util"
|
||
)
|
||
|
||
type RTPWriter interface {
|
||
writeRTPFrame(frame *RTPFrame)
|
||
}
|
||
|
||
func (av *Media[T]) UnmarshalRTPPacket(p *rtp.Packet) (frame *RTPFrame) {
|
||
if av.DecoderConfiguration.PayloadType != p.PayloadType {
|
||
av.Stream.Warn("RTP PayloadType error", zap.Uint8("want", av.DecoderConfiguration.PayloadType), zap.Uint8("got", p.PayloadType))
|
||
return
|
||
}
|
||
frame = &RTPFrame{
|
||
Packet: *p,
|
||
}
|
||
av.Value.BytesIn += len(p.Payload) + 12
|
||
return av.recorderRTP(frame)
|
||
}
|
||
|
||
func (av *Media[T]) UnmarshalRTP(raw []byte) (frame *RTPFrame) {
|
||
var p rtp.Packet
|
||
err := p.Unmarshal(raw)
|
||
if err != nil {
|
||
av.Stream.Warn("RTP Unmarshal error", zap.Error(err))
|
||
return
|
||
}
|
||
return av.UnmarshalRTPPacket(&p)
|
||
}
|
||
|
||
// WriteRTPPack 写入已反序列化的RTP包
|
||
func (av *Media[T]) WriteRTPPack(p *rtp.Packet) {
|
||
for frame := av.UnmarshalRTPPacket(p); frame != nil; frame = av.nextRTPFrame() {
|
||
av.writeRTPFrame(frame)
|
||
}
|
||
}
|
||
|
||
// WriteRTP 写入未反序列化的RTP包
|
||
func (av *Media[T]) WriteRTP(raw []byte) {
|
||
for frame := av.UnmarshalRTP(raw); frame != nil; frame = av.nextRTPFrame() {
|
||
av.writeRTPFrame(frame)
|
||
}
|
||
}
|
||
|
||
type RTPDemuxer struct {
|
||
lastSeq uint16 //上一个收到的序号，用于乱序重排
|
||
lastSeq2 uint16 //记录上上一个收到的序列号
|
||
乱序重排 util.RTPReorder[*RTPFrame]
|
||
RTPWriter `json:"-"`
|
||
}
|
||
|
||
// 获取缓存中下一个rtpFrame
|
||
func (av *RTPDemuxer) nextRTPFrame() (frame *RTPFrame) {
|
||
if config.Global.RTPReorder {
|
||
return av.乱序重排.Pop()
|
||
}
|
||
return
|
||
}
|
||
|
||
// 对RTP包乱序重排
|
||
func (av *RTPDemuxer) recorderRTP(frame *RTPFrame) *RTPFrame {
|
||
if config.Global.RTPReorder {
|
||
return av.乱序重排.Push(frame.SequenceNumber, frame)
|
||
} else {
|
||
if av.lastSeq == 0 {
|
||
av.lastSeq = frame.SequenceNumber
|
||
} else if frame.SequenceNumber == av.lastSeq2+1 { // 本次序号是上上次的序号+1 说明中间隔了一个错误序号（某些rtsp流中的rtcp包写成了rtp包导致的）
|
||
av.lastSeq = frame.SequenceNumber
|
||
} else {
|
||
av.lastSeq2 = av.lastSeq
|
||
av.lastSeq = frame.SequenceNumber
|
||
if av.lastSeq != av.lastSeq2+1 { //序号不连续
|
||
// av.Stream.Warn("RTP SequenceNumber error", av.lastSeq2, av.lastSeq)
|
||
return frame
|
||
}
|
||
}
|
||
return frame
|
||
}
|
||
}
|
||
|
||
type RTPMuxer struct {
|
||
rtpSequence uint16 //用于生成下一个rtp包的序号
|
||
}
|