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

根据是否含有音视频写FLV头

Browse Source
This commit is contained in:
dexter
2022-03-28 19:02:38 +08:00
parent cea276b5b1
commit f2d09a17fc
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.go
View File

@@ -108,7 +108,7 @@ func HDLHandler(w http.ResponseWriter, r *http.Request) {
} }
w.Header().Set("Transfer-Encoding", "chunked") w.Header().Set("Transfer-Encoding", "chunked")
w.Header().Set("Content-Type", "video/x-flv") w.Header().Set("Content-Type", "video/x-flv")
w.Write(codec.FLVHeader)
sub := Subscriber{ID: r.RemoteAddr, Type: "FLV", Ctx2: r.Context()} sub := Subscriber{ID: r.RemoteAddr, Type: "FLV", Ctx2: r.Context()}
if err := sub.Subscribe(stringPath); err == nil { if err := sub.Subscribe(stringPath); err == nil {
vt, at := sub.WaitVideoTrack(), sub.WaitAudioTrack() vt, at := sub.WaitVideoTrack(), sub.WaitAudioTrack()
@@ -128,7 +128,9 @@ func HDLHandler(w http.ResponseWriter, r *http.Request) {
"videodatarate": 0, "videodatarate": 0,
"filesize": 0, "filesize": 0,
} }
var flags byte
if vt != nil { if vt != nil {
flags |= 1
metaData["videocodecid"] = int(vt.CodecID) metaData["videocodecid"] = int(vt.CodecID)
metaData["width"] = vt.SPSInfo.Width metaData["width"] = vt.SPSInfo.Width
metaData["height"] = vt.SPSInfo.Height metaData["height"] = vt.SPSInfo.Height
@@ -138,6 +140,7 @@ func HDLHandler(w http.ResponseWriter, r *http.Request) {
} }
} }
if at != nil { if at != nil {
flags |= (1 << 2)
metaData["audiocodecid"] = int(at.CodecID) metaData["audiocodecid"] = int(at.CodecID)
metaData["audiosamplerate"] = at.SoundRate metaData["audiosamplerate"] = at.SoundRate
metaData["audiosamplesize"] = int(at.SoundSize) metaData["audiosamplesize"] = int(at.SoundSize)
@@ -152,6 +155,7 @@ func HDLHandler(w http.ResponseWriter, r *http.Request) {
if _, err := WriteEcmaArray(&buffer, metaData); err != nil { if _, err := WriteEcmaArray(&buffer, metaData); err != nil {
return return
} }
w.Write([]byte{'F', 'L', 'V', 0x01, flags, 0, 0, 0, 9, 0, 0, 0, 0})
codec.WriteFLVTag(w, codec.FLV_TAG_TYPE_SCRIPT, 0, buffer.Bytes()) codec.WriteFLVTag(w, codec.FLV_TAG_TYPE_SCRIPT, 0, buffer.Bytes())
sub.Play(at, vt) sub.Play(at, vt)
} }