apps/plugin-hdl
1
0
Fork 0
mirror of https://github.com/Monibuca/plugin-hdl.git synced 2025-10-06 01:07:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

对远端拉流连接时增加flv头判断

Browse Source
This commit is contained in:
dexter
2022-12-18 14:29:17 +08:00
parent c5562764e5
commit a2b7c2781f
2 changed files with 30 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
pull.go
View File

@@ -1,7 +1,6 @@
package hdl
import (
"bufio"
"io"
"net/http"
"os"
@@ -13,6 +12,19 @@ import (
"m7s.live/engine/v4/util"
)
type HDLPuller struct {
Publisher
Puller
absTS uint32 //绝对时间戳
buf util.Buffer
}
func NewHDLPuller() *HDLPuller {
return &HDLPuller{
buf: util.Buffer(make([]byte, len(codec.FLVHeader))),
}
}
func (puller *HDLPuller) Connect() (err error) {
HDLPlugin.Info("connect", zap.String("remoteURL", puller.RemoteURL))
if strings.HasPrefix(puller.RemoteURL, "http") {
@@ -29,20 +41,20 @@ func (puller *HDLPuller) Connect() (err error) {
if err != nil {
HDLPlugin.Error("connect", zap.Error(err))
}
head := puller.buf.SubBuf(0, len(codec.FLVHeader))
_, err = io.ReadFull(puller, head)
if head[0] != 'F' || head[1] != 'L' || head[2] != 'V' {
err = codec.ErrInvalidFLV
}
return
}
func (puller *HDLPuller) Pull() {
head := util.Buffer(make([]byte, len(codec.FLVHeader)))
reader := bufio.NewReader(puller)
_, err := io.ReadFull(reader, head)
if err != nil {
return
}
head.Reset()
func (puller *HDLPuller) Pull() (err error) {
puller.buf.Reset()
var startTs uint32
for offsetTs := puller.absTS; err == nil && puller.Err() == nil; _, err = io.ReadFull(reader, head[:4]) {
tmp := head.SubBuf(0, 11)
_, err = io.ReadFull(reader, tmp)
for offsetTs := puller.absTS; err == nil && puller.Err() == nil; _, err = io.ReadFull(puller, puller.buf[:4]) {
tmp := puller.buf.SubBuf(0, 11)
_, err = io.ReadFull(puller, tmp)
if err != nil {
return
}
@@ -54,7 +66,7 @@ func (puller *HDLPuller) Pull() {
}
tmp.ReadUint24()
payload := make([]byte, dataSize)
_, err = io.ReadFull(reader, payload)
_, err = io.ReadFull(puller, payload)
if err != nil {
return
}
@@ -66,10 +78,5 @@ func (puller *HDLPuller) Pull() {
puller.WriteAVCCVideo(puller.absTS, payload)
}
}
}
type HDLPuller struct {
Publisher
Puller
absTS uint32 //绝对时间戳
return
}