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.
107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
4 years ago
|
// 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 main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/q191201771/lal/pkg/rtmp"
|
||
|
|
|
||
|
|
"github.com/q191201771/lal/pkg/base"
|
||
|
|
"github.com/q191201771/lal/pkg/remux"
|
||
|
|
"github.com/q191201771/lal/pkg/rtprtcp"
|
||
|
|
"github.com/q191201771/lal/pkg/rtsp"
|
||
|
|
"github.com/q191201771/naza/pkg/nazalog"
|
||
|
|
)
|
||
|
|
|
||
|
|
var rtmpPushSession *rtmp.PushSession
|
||
|
|
|
||
|
|
type Observer struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *Observer) OnRTPPacket(pkt rtprtcp.RTPPacket) {
|
||
|
|
// noop
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *Observer) OnAVConfig(asc, vps, sps, pps []byte) {
|
||
|
|
metadata, vsh, ash, err := remux.AVConfig2RTMPMsg(asc, vps, sps, pps)
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
err = rtmpPushSession.AsyncWrite(rtmp.Message2Chunks(metadata.Payload, &metadata.Header))
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
err = rtmpPushSession.AsyncWrite(rtmp.Message2Chunks(vsh.Payload, &vsh.Header))
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
err = rtmpPushSession.AsyncWrite(rtmp.Message2Chunks(ash.Payload, &ash.Header))
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *Observer) OnAVPacket(pkt base.AVPacket) {
|
||
|
|
msg, err := remux.AVPacket2RTMPMsg(pkt)
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
err = rtmpPushSession.AsyncWrite(rtmp.Message2Chunks(msg.Payload, &msg.Header))
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
_ = nazalog.Init(func(option *nazalog.Option) {
|
||
|
|
option.AssertBehavior = nazalog.AssertFatal
|
||
|
|
})
|
||
|
|
|
||
|
|
inURL, outURL := parseFlag()
|
||
|
|
|
||
|
|
rtmpPushSession = rtmp.NewPushSession(func(option *rtmp.PushSessionOption) {
|
||
|
|
option.ConnectTimeoutMS = 5000
|
||
|
|
option.PushTimeoutMS = 5000
|
||
|
|
option.WriteAVTimeoutMS = 5000
|
||
|
|
})
|
||
|
|
|
||
|
|
err := rtmpPushSession.Push(outURL)
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
err := <-rtmpPushSession.Done()
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
}()
|
||
|
|
|
||
|
|
o := &Observer{}
|
||
|
|
rtspPullSession := rtsp.NewPullSession(o, func(option *rtsp.PullSessionOption) {
|
||
|
|
option.PullTimeoutMS = 5000
|
||
|
|
})
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
for {
|
||
|
|
rtspPullSession.UpdateStat(1)
|
||
|
|
rtspStat := rtspPullSession.GetStat()
|
||
|
|
rtmpPushSession.UpdateStat(1)
|
||
|
|
rtmpStat := rtmpPushSession.GetStat()
|
||
|
|
nazalog.Debugf("bitrate. rtsp pull=%dkbit/s, rtmp push=%dkbit/s", rtspStat.Bitrate, rtmpStat.Bitrate)
|
||
|
|
time.Sleep(1 * time.Second)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
err = rtspPullSession.Pull(inURL)
|
||
|
|
nazalog.Assert(nil, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func parseFlag() (inURL string, outFilename string) {
|
||
|
|
i := flag.String("i", "", "specify pull rtsp url")
|
||
|
|
o := flag.String("o", "", "specify push rtmp url")
|
||
|
|
flag.Parse()
|
||
|
|
if *i == "" || *o == "" {
|
||
|
|
flag.Usage()
|
||
|
|
_, _ = fmt.Fprintf(os.Stderr, `Example:
|
||
|
|
./bin/pullrtsp -i rtsp://localhost:5544/live/test110 -o rtmp://localhost:19350/live/test220
|
||
|
|
`)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
return *i, *o
|
||
|
|
}
|