mirror of https://github.com/q191201771/lal.git
* PullSession和SubSession的主动关闭、被动关闭
* 引入日志库seelog，添加一些日志 * 缓存gop * 部分解析avc seq headerpull/200/head
q191201771
6 years ago
parent
5a2ea4fadc
commit
bf6b9502df
@ -1,7 +1,10 @@
|
||||
/.idea
|
||||
/rtmp
|
||||
/demo
|
||||
/conf/lal.conf.json
|
||||
/demo/httpflvpull/httpflvpull.go
|
||||
/demo/
|
||||
/conf/self.conf.json
|
||||
|
||||
/rtmp/
|
||||
/.idea/
|
||||
/logs/
|
||||
/TODO.md
|
||||
/lal
|
||||
/lal_linux
|
||||
/build_linux.sh
|
||||
|
||||
@ -0,0 +1,11 @@
|
||||
{
|
||||
"httpflv": {
|
||||
"sub_listen_addr": ":8080",
|
||||
"pull_addr": "pull.xxx.com",
|
||||
"pull_connect_timeout": 2,
|
||||
"pull_read_timeout": 20,
|
||||
"sub_idle_timeout": 10,
|
||||
"stop_pull_while_no_sub_timeout": 5,
|
||||
"gop_cache_num": 2
|
||||
}
|
||||
}
|
||||
@ -1,10 +0,0 @@
|
||||
{
|
||||
"httpflv": {
|
||||
"sub_listen_addr": ":8080",
|
||||
"pull_addr": "pull.x.com",
|
||||
"pull_connect_timeout": 2,
|
||||
"pull_read_timeout": 10,
|
||||
"sub_idle_timeout": 10,
|
||||
"stop_pull_while_no_sub_timeout": 30
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,35 @@
|
||||
<seelog minlevel="trace">
|
||||
<outputs formatid="file_common">
|
||||
<filter levels="trace">
|
||||
<console formatid="console_trace" />
|
||||
</filter>
|
||||
<filter levels="debug">
|
||||
<console formatid="console_debug" />
|
||||
</filter>
|
||||
<filter levels="info">
|
||||
<console formatid="console_info" />
|
||||
</filter>
|
||||
<filter levels="warn">
|
||||
<console formatid="console_warn" />
|
||||
</filter>
|
||||
<filter levels="error">
|
||||
<console formatid="console_error" />
|
||||
</filter>
|
||||
<filter levels="critical">
|
||||
<console formatid="console_critical" />
|
||||
</filter>
|
||||
<file path="./logs/lal.log" />
|
||||
<filter levels="warn,error,critical">
|
||||
<file path="./logs/lal_error.log" />
|
||||
</filter>
|
||||
</outputs>
|
||||
<formats>
|
||||
<format id="file_common" format="%Date(2006-01-02 15:04:05.000) %LEV %Msg - %File:%Line%n" />
|
||||
<format id="console_trace" format="%Date(2006-01-02 15:04:05.000) %EscM(37)%LEV%EscM(49)%EscM(0) %Msg - %File:%Line%n" />
|
||||
<format id="console_debug" format="%Date(2006-01-02 15:04:05.000) %EscM(37)%LEV%EscM(49)%EscM(0) %Msg - %File:%Line%n" />
|
||||
<format id="console_info" format="%Date(2006-01-02 15:04:05.000) %EscM(36)%LEV%EscM(49)%EscM(0) %Msg - %File:%Line%n" />
|
||||
<format id="console_warn" format="%Date(2006-01-02 15:04:05.000) %EscM(33)%LEV%EscM(49)%EscM(0) %Msg - %File:%Line%n" />
|
||||
<format id="console_error" format="%Date(2006-01-02 15:04:05.000) %EscM(31)%LEV%EscM(49)%EscM(0) %Msg - %File:%Line%n" />
|
||||
<format id="console_critical" format="%Date(2006-01-02 15:04:05.000) %EscM(31)%LEV%EscM(49)%EscM(0) %Msg - %File:%Line%n" />
|
||||
</formats>
|
||||
</seelog>
|
||||
@ -0,0 +1,11 @@
|
||||
<seelog minlevel="info">
|
||||
<outputs formatid="common">
|
||||
<rollingfile type="date" filename="./logs/lal.log" datepattern="2006-01-02" />
|
||||
<filter levels="warn,error,critical">
|
||||
<rollingfile type="date" filename="./logs/lal_error.log" datepattern="2006-01-02" />
|
||||
</filter>
|
||||
</outputs>
|
||||
<formats>
|
||||
<format id="common" format="%Date(2006-01-02 15:04:05) %LEV %Msg%n" />
|
||||
</formats>
|
||||
</seelog>
|
||||
@ -0,0 +1,52 @@
|
||||
package httpflv
|
||||
|
||||
import (
|
||||
"github.com/q191201771/lal/bele"
|
||||
"github.com/q191201771/lal/log"
|
||||
)
|
||||
|
||||
// TODO chef: move me to other packet
|
||||
|
||||
// H.264-AVC-ISO_IEC_14496-15.pdf
|
||||
// 5.2.4 Decoder configuration information
|
||||
|
||||
// <buf> body of tag
|
||||
func parseAvcSeqHeader(buf []byte) (sps, pps []byte, err error) {
|
||||
// TODO chef: check if read out of <buf> range
|
||||
|
||||
if buf[0] != AvcKey || buf[1] != AvcPacketTypeSeqHeader || buf[2] != 0 || buf[3] != 0 || buf[4] != 0 {
|
||||
log.Error("parse avc seq header failed.")
|
||||
err = fxxkErr
|
||||
return
|
||||
}
|
||||
|
||||
//configurationVersion := buf[5]
|
||||
//avcProfileIndication := buf[6]
|
||||
//profileCompatibility := buf[7]
|
||||
//avcLevelIndication := buf[8]
|
||||
//lengthSizeMinusOne := buf[9] & 0x03
|
||||
|
||||
index := 10
|
||||
|
||||
numOfSps := int(buf[index] & 0x1F)
|
||||
index++
|
||||
// TODO chef: if the situation of multi sps exist?
|
||||
// only take the last one.
|
||||
for i := 0; i < numOfSps; i++ {
|
||||
lenOfSps := int(bele.BeUInt16(buf[index:]))
|
||||
index += 2
|
||||
sps = append(sps, buf[index:index+lenOfSps]...)
|
||||
index += lenOfSps
|
||||
}
|
||||
|
||||
numOfPps := int(buf[index] & 0x1F)
|
||||
index++
|
||||
for i := 0; i < numOfPps; i++ {
|
||||
lenOfPps := int(bele.BeUInt16(buf[index:]))
|
||||
index += 2
|
||||
pps = append(pps, buf[index:index+lenOfPps]...)
|
||||
index += lenOfPps
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
@ -0,0 +1,206 @@
|
||||
package httpflv
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"github.com/q191201771/lal/log"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type Gop struct {
|
||||
raw []byte
|
||||
firstTimestamp uint32
|
||||
}
|
||||
|
||||
type GopCache struct {
|
||||
gopNum int
|
||||
|
||||
metadata *Tag
|
||||
avcSeqHeader *Tag
|
||||
aacSeqHeader *Tag
|
||||
gops []*Gop // TODO chef: maybe use other container to mock a queue
|
||||
mutex sync.Mutex
|
||||
}
|
||||
|
||||
// gopNum: 0 means only cache metadata, avc seq header, aac seq header
|
||||
func NewGopCache(gopNum int) *GopCache {
|
||||
return &GopCache{
|
||||
gopNum: gopNum,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *GopCache) Push(tag *Tag) {
|
||||
c.mutex.Lock()
|
||||
defer c.mutex.Unlock()
|
||||
|
||||
if tag.isMetaData() {
|
||||
// TODO chef: will this happen?
|
||||
if c.metadata != nil {
|
||||
log.Debugf("updating metadata.")
|
||||
log.Debug(tag.Header, tag.Raw[tagHeaderSize:])
|
||||
log.Debug(c.metadata.Header, c.metadata.Raw[tagHeaderSize:])
|
||||
c.clearGop()
|
||||
}
|
||||
c.metadata = tag
|
||||
}
|
||||
if tag.isAvcKeySeqHeader() {
|
||||
//log.Debug(parseAvcSeqHeader(tag.Raw[tagHeaderSize:]))
|
||||
if c.avcSeqHeader == nil {
|
||||
c.avcSeqHeader = tag
|
||||
} else {
|
||||
// TODO chef: compare nessary? if other way to update seq header and handle cache stuff?
|
||||
if bytes.Compare(tag.Raw[tagHeaderSize:], c.avcSeqHeader.Raw[tagHeaderSize:]) == 0 {
|
||||
// noop
|
||||
} else {
|
||||
log.Debugf("updating avc seq header.")
|
||||
log.Debug(tag.Header, tag.Raw[tagHeaderSize:])
|
||||
log.Debug(c.avcSeqHeader.Header, c.avcSeqHeader.Raw[tagHeaderSize:])
|
||||
c.clearGop()
|
||||
c.avcSeqHeader = tag
|
||||
}
|
||||
}
|
||||
}
|
||||
if tag.isAacSeqHeader() {
|
||||
if c.aacSeqHeader == nil {
|
||||
c.aacSeqHeader = tag
|
||||
} else {
|
||||
if bytes.Compare(tag.Raw[tagHeaderSize:], c.aacSeqHeader.Raw[tagHeaderSize:]) == 0 {
|
||||
// noop
|
||||
} else {
|
||||
log.Debugf("updating aac seq header.")
|
||||
c.clearGop()
|
||||
c.aacSeqHeader = tag
|
||||
}
|
||||
}
|
||||
c.aacSeqHeader = tag
|
||||
}
|
||||
|
||||
if c.gopNum == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
if len(c.gops) == 0 {
|
||||
if tag.isAvcKeyNalu() {
|
||||
gop := &Gop{}
|
||||
gop.firstTimestamp = tag.Header.Timestamp
|
||||
gop.raw = append(gop.raw, tag.Raw...)
|
||||
c.gops = append(c.gops, gop)
|
||||
c.syncOldestKeyNaluTimestampToSeqHeader()
|
||||
}
|
||||
} else {
|
||||
if tag.isAvcKeyNalu() {
|
||||
gop := &Gop{}
|
||||
gop.firstTimestamp = tag.Header.Timestamp
|
||||
gop.raw = append(gop.raw, tag.Raw...)
|
||||
c.gops = append(c.gops, gop)
|
||||
if len(c.gops) > c.gopNum+1 {
|
||||
c.gops = c.gops[1:]
|
||||
c.syncOldestKeyNaluTimestampToSeqHeader()
|
||||
}
|
||||
} else {
|
||||
c.gops[len(c.gops)-1].raw = append(c.gops[len(c.gops)-1].raw, tag.Raw...)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (c *GopCache) GetWholeThings() (hasKeyFrame bool, res []byte) {
|
||||
if tag := c.getMetadata(); tag != nil {
|
||||
res = append(res, tag.Raw...)
|
||||
}
|
||||
|
||||
avc := c.getAvcSeqHeader()
|
||||
aac := c.getAacSeqHeader()
|
||||
// TODO chef: if nessary to sort them by timestamp
|
||||
if avc != nil && aac != nil {
|
||||
if avc.Header.Timestamp <= aac.Header.Timestamp {
|
||||
res = append(res, avc.Raw...)
|
||||
res = append(res, aac.Raw...)
|
||||
} else {
|
||||
res = append(res, aac.Raw...)
|
||||
res = append(res, avc.Raw...)
|
||||
}
|
||||
} else if avc != nil && aac == nil {
|
||||
res = append(res, avc.Raw...)
|
||||
} else if avc == nil && aac != nil {
|
||||
res = append(res, aac.Raw...)
|
||||
}
|
||||
|
||||
if gops := c.getGops(false); gops != nil {
|
||||
res = append(res, gops...)
|
||||
log.Debug("cache match.")
|
||||
hasKeyFrame = true
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (c *GopCache) ClearAll() {
|
||||
c.mutex.Lock()
|
||||
defer c.mutex.Unlock()
|
||||
c.metadata = nil
|
||||
c.avcSeqHeader = nil
|
||||
c.aacSeqHeader = nil
|
||||
c.gops = nil
|
||||
}
|
||||
|
||||
func (c *GopCache) getGops(mustCompleted bool) []byte {
|
||||
c.mutex.Lock()
|
||||
defer c.mutex.Unlock()
|
||||
|
||||
neededLen := len(c.gops)
|
||||
if mustCompleted {
|
||||
neededLen--
|
||||
}
|
||||
if neededLen <= 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
var res []byte
|
||||
for i := 0; i != neededLen; i++ {
|
||||
res = append(res, c.gops[i].raw...)
|
||||
}
|
||||
return res
|
||||
}
|
||||
|
||||
func (c *GopCache) getMetadata() (res *Tag) {
|
||||
c.mutex.Lock()
|
||||
defer c.mutex.Unlock()
|
||||
if c.metadata != nil {
|
||||
res = c.metadata.cloneTag()
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (c *GopCache) getAvcSeqHeader() (res *Tag) {
|
||||
c.mutex.Lock()
|
||||
defer c.mutex.Unlock()
|
||||
|
||||
if c.avcSeqHeader != nil {
|
||||
res = c.avcSeqHeader.cloneTag()
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (c *GopCache) getAacSeqHeader() (res *Tag) {
|
||||
c.mutex.Lock()
|
||||
defer c.mutex.Unlock()
|
||||
|
||||
if c.aacSeqHeader != nil {
|
||||
res = c.aacSeqHeader.cloneTag()
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (c *GopCache) clearGop() {
|
||||
log.Debug("clearGop")
|
||||
c.gops = nil
|
||||
}
|
||||
|
||||
// TODO chef: if nessary
|
||||
func (c *GopCache) syncOldestKeyNaluTimestampToSeqHeader() {
|
||||
ts := c.gops[0].firstTimestamp
|
||||
if c.avcSeqHeader != nil {
|
||||
c.avcSeqHeader.Header.Timestamp = ts
|
||||
}
|
||||
if c.aacSeqHeader != nil {
|
||||
c.aacSeqHeader.Header.Timestamp = ts
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,47 @@
|
||||
package log
|
||||
|
||||
import "github.com/cihub/seelog"
|
||||
|
||||
var log seelog.LoggerInterface
|
||||
|
||||
func Initial(configFileName string) error {
|
||||
var err error
|
||||
log, err = seelog.LoggerFromConfigAsFile(configFileName)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = log.SetAdditionalStackDepth(1)
|
||||
return err
|
||||
}
|
||||
|
||||
func Debugf(format string, params ...interface{}) {
|
||||
log.Debugf(format, params...)
|
||||
}
|
||||
|
||||
func Infof(format string, params ...interface{}) {
|
||||
log.Infof(format, params...)
|
||||
}
|
||||
|
||||
func Warnf(format string, params ...interface{}) {
|
||||
log.Warnf(format, params...)
|
||||
}
|
||||
|
||||
func Errorf(format string, params ...interface{}) {
|
||||
log.Errorf(format, params...)
|
||||
}
|
||||
|
||||
func Debug(v ...interface{}) {
|
||||
log.Debug(v...)
|
||||
}
|
||||
|
||||
func Info(v ...interface{}) {
|
||||
log.Info(v...)
|
||||
}
|
||||
|
||||
func Warn(v ...interface{}) {
|
||||
log.Warn(v...)
|
||||
}
|
||||
|
||||
func Error(v ...interface{}) {
|
||||
log.Error(v...)
|
||||
}
|
||||
@ -0,0 +1,16 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sync/atomic"
|
||||
)
|
||||
|
||||
var globalId uint64
|
||||
|
||||
func GenUniqueId() uint64 {
|
||||
return atomic.AddUint64(&globalId, 1)
|
||||
}
|
||||
|
||||
func GenUniqueKey(prefix string) string {
|
||||
return fmt.Sprintf("%s%d", prefix, GenUniqueId())
|
||||
}
|
||||
Loading…
Reference in New Issue