|
|
// Copyright 2022, 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 innertest
|
|
|
|
|
|
import (
|
|
|
"github.com/q191201771/lal/pkg/base"
|
|
|
"github.com/q191201771/lal/pkg/httpflv"
|
|
|
"github.com/q191201771/lal/pkg/remux"
|
|
|
"github.com/q191201771/lal/pkg/rtprtcp"
|
|
|
"github.com/q191201771/lal/pkg/rtsp"
|
|
|
"github.com/q191201771/lal/pkg/sdp"
|
|
|
"github.com/q191201771/naza/pkg/nazalog"
|
|
|
"io"
|
|
|
"testing"
|
|
|
)
|
|
|
|
|
|
// TestDump_Rtsp
|
|
|
//
|
|
|
// 重放业务方的rtsp流。
|
|
|
//
|
|
|
// 本测试函数模拟客户端，读取业务方对的dumpfile，解析为rtp，合帧，写flv文件
|
|
|
//
|
|
|
// 步骤：
|
|
|
//
|
|
|
// 1. 让业务方提供lalserver录制下来的dumpfile文件
|
|
|
// 2. 将dumpfile存放在下面filename变量处，或者修改下面filename变量值
|
|
|
// 3. 执行该测试
|
|
|
// go test -test.run TestDump_Rtsp
|
|
|
func TestDump_Rtsp(t *testing.T) {
|
|
|
// TODO(chef): [test] 合帧测试，只有音频部分，没有视频部分 202211
|
|
|
|
|
|
filename := "/tmp/outpullrtsp.laldump"
|
|
|
outFlvFilename := "/tmp/outtestdumprtsp.flv"
|
|
|
|
|
|
// 初始化输出的flv文件
|
|
|
var fileWriter httpflv.FlvFileWriter
|
|
|
err := fileWriter.Open(outFlvFilename)
|
|
|
nazalog.Assert(nil, err)
|
|
|
defer fileWriter.Dispose()
|
|
|
err = fileWriter.WriteRaw(httpflv.FlvHeader)
|
|
|
nazalog.Assert(nil, err)
|
|
|
|
|
|
// 初始化remuxer
|
|
|
remuxer := remux.NewAvPacket2RtmpRemuxer().WithOnRtmpMsg(func(msg base.RtmpMsg) {
|
|
|
nazalog.Debugf("< remuxer. %s", msg.DebugString())
|
|
|
err = fileWriter.WriteTag(*remux.RtmpMsg2FlvTag(msg))
|
|
|
nazalog.Assert(nil, err)
|
|
|
})
|
|
|
|
|
|
var ctx sdp.LogicContext
|
|
|
var unpacker rtprtcp.IRtpUnpacker
|
|
|
var unpackerVideo rtprtcp.IRtpUnpacker
|
|
|
var q *rtsp.AvPacketQueue
|
|
|
|
|
|
df := base.NewDumpFile()
|
|
|
err = df.OpenToRead(filename)
|
|
|
nazalog.Assert(nil, err)
|
|
|
|
|
|
if rtsp.BaseInSessionTimestampFilterFlag {
|
|
|
q = rtsp.NewAvPacketQueue(func(pkt base.AvPacket) {
|
|
|
remuxer.FeedAvPacket(pkt)
|
|
|
})
|
|
|
}
|
|
|
|
|
|
for {
|
|
|
m, err := df.ReadOneMessage()
|
|
|
nazalog.Debugf("< ReadOneMessage. %+v, %+v", m, err)
|
|
|
if err == io.EOF {
|
|
|
return
|
|
|
}
|
|
|
nazalog.Assert(nil, err)
|
|
|
|
|
|
if m.Typ == base.DumpTypeInnerFileHeaderData {
|
|
|
continue
|
|
|
}
|
|
|
|
|
|
if m.Typ != base.DumpTypeRtspRtpData && m.Typ != base.DumpTypeRtspSdpData {
|
|
|
nazalog.Errorf("unknown type. typ=%d", m.Typ)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
if m.Typ == base.DumpTypeRtspSdpData {
|
|
|
ctx, err = sdp.ParseSdp2LogicContext([]byte(m.Body))
|
|
|
nazalog.Debugf("parse sdp, %+v, %+v", ctx, err)
|
|
|
|
|
|
remuxer.OnSdp(ctx)
|
|
|
unpacker = rtprtcp.DefaultRtpUnpackerFactory(ctx.GetAudioPayloadTypeBase(), ctx.AudioClockRate, 1024, func(pkt base.AvPacket) {
|
|
|
nazalog.Debugf("audio avpacket. %s", pkt.DebugString())
|
|
|
if rtsp.BaseInSessionTimestampFilterFlag {
|
|
|
q.Feed(pkt)
|
|
|
} else {
|
|
|
remuxer.OnAvPacket(pkt)
|
|
|
}
|
|
|
})
|
|
|
unpackerVideo = rtprtcp.DefaultRtpUnpackerFactory(ctx.GetVideoPayloadTypeBase(), ctx.VideoClockRate, 1024, func(pkt base.AvPacket) {
|
|
|
nazalog.Debugf("video avpacket. %s", pkt.DebugString())
|
|
|
if rtsp.BaseInSessionTimestampFilterFlag {
|
|
|
q.Feed(pkt)
|
|
|
} else {
|
|
|
remuxer.OnAvPacket(pkt)
|
|
|
}
|
|
|
})
|
|
|
continue
|
|
|
}
|
|
|
|
|
|
pkt, err := rtprtcp.ParseRtpPacket(m.Body)
|
|
|
nazalog.Assert(nil, err)
|
|
|
nazalog.Debugf("< ParseRtpPacket. %s", pkt.DebugString())
|
|
|
if ctx.IsAudioPayloadTypeOrigin(int(pkt.Header.PacketType)) {
|
|
|
unpacker.Feed(pkt)
|
|
|
} else if ctx.IsVideoPayloadTypeOrigin(int(pkt.Header.PacketType)) {
|
|
|
unpackerVideo.Feed(pkt)
|
|
|
} else {
|
|
|
nazalog.Errorf("unknown payload type. pt=%d", pkt.Header.PacketType)
|
|
|
}
|
|
|
}
|
|
|
}
|