|
|
// Copyright 2019, 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"
|
|
|
"os"
|
|
|
|
|
|
"github.com/q191201771/lal/pkg/base"
|
|
|
"github.com/q191201771/lal/pkg/httpflv"
|
|
|
"github.com/q191201771/lal/pkg/rtmp"
|
|
|
"github.com/q191201771/naza/pkg/bele"
|
|
|
"github.com/q191201771/naza/pkg/nazalog"
|
|
|
)
|
|
|
|
|
|
// 拉取http-flv的流并存储为flv文件
|
|
|
|
|
|
func main() {
|
|
|
_ = nazalog.Init(func(option *nazalog.Option) {
|
|
|
option.AssertBehavior = nazalog.AssertFatal
|
|
|
})
|
|
|
defer nazalog.Sync()
|
|
|
base.LogoutStartInfo()
|
|
|
|
|
|
url, flvname := parseFlag()
|
|
|
flvfile, err := os.Create(flvname)
|
|
|
if err != nil {
|
|
|
nazalog.Errorf("create flv file failed, err=%+v", err)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
defer flvfile.Close()
|
|
|
|
|
|
session := httpflv.NewPullSession()
|
|
|
err = session.Pull(url, func(tag httpflv.Tag) {
|
|
|
if tag.Header.Type == httpflv.TagTypeMetadata {
|
|
|
// TODO(chef): httpflv.PullSession支持返回flv header，可供业务方选择使用 202210
|
|
|
// 根据metadata填写flv头
|
|
|
opa, err := rtmp.ParseMetadata(tag.Payload())
|
|
|
if err != nil {
|
|
|
nazalog.Errorf("ParseMetadata failed, err=%+v", err)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
b := make([]byte, 13)
|
|
|
var flags uint8
|
|
|
|
|
|
audiocodecid := opa.Find("audiocodecid")
|
|
|
videocodecid := opa.Find("videocodecid")
|
|
|
if audiocodecid != 0 {
|
|
|
flags |= 0x04
|
|
|
}
|
|
|
|
|
|
if videocodecid != 0 {
|
|
|
flags |= 0x01
|
|
|
}
|
|
|
|
|
|
writeFlvHeader(b, flags)
|
|
|
flvfile.Write(b)
|
|
|
}
|
|
|
|
|
|
nazalog.Infof("tag Type:%d, tag Size:%d", tag.Header.Type, tag.Header.DataSize)
|
|
|
|
|
|
flvfile.Write(tag.Raw)
|
|
|
})
|
|
|
nazalog.Assert(nil, err)
|
|
|
err = <-session.WaitChan()
|
|
|
nazalog.Assert(nil, err)
|
|
|
}
|
|
|
|
|
|
func parseFlag() (url, flvfile string) {
|
|
|
i := flag.String("i", "", "specify http-flv url")
|
|
|
o := flag.String("o", "", "specify output flv file")
|
|
|
flag.Parse()
|
|
|
if *i == "" || *o == "" {
|
|
|
flag.Usage()
|
|
|
base.OsExitAndWaitPressIfWindows(1)
|
|
|
}
|
|
|
return *i, *o
|
|
|
}
|
|
|
|
|
|
func writeFlvHeader(b []byte, flags uint8) {
|
|
|
|
|
|
// 'FLV', version 1
|
|
|
bele.BePutUint32(b, 0x464c5601)
|
|
|
b[4] = flags
|
|
|
|
|
|
// DataOffset: UI32 Offset in bytes from start of file to start of body (that is, size of header)
|
|
|
// The DataOffset field usually has a value of 9 for FLV version 1.
|
|
|
bele.BePutUint32(b[5:9], 9)
|
|
|
|
|
|
// PreviousTagSize0: UI32 Always 0
|
|
|
bele.BePutUint32(b[9:13], 0)
|
|
|
|
|
|
return
|
|
|
}
|