package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-06 07:37:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix crash in client-read-h264-save-to-disk

Browse Source
This commit is contained in:
aler9
2022-06-15 10:37:15 +02:00
parent aa08f973dd
commit b4925dbc24
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/client-read-h264-save-to-disk/mpegtsencoder.go
View File

@@ -61,6 +61,7 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
{byte(h264.NALUTypeAccessUnitDelimiter), 240}, {byte(h264.NALUTypeAccessUnitDelimiter), 240},
} }
nonIDRPresent := false
idrPresent := false idrPresent := false
for _, nalu := range nalus { for _, nalu := range nalus {
@@ -84,11 +85,18 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
if e.sps != nil && e.pps != nil { if e.sps != nil && e.pps != nil {
filteredNALUs = append(filteredNALUs, e.sps, e.pps) filteredNALUs = append(filteredNALUs, e.sps, e.pps)
} }
case h264.NALUTypeNonIDR:
nonIDRPresent = true
} }
filteredNALUs = append(filteredNALUs, nalu) filteredNALUs = append(filteredNALUs, nalu)
} }
if !nonIDRPresent && !idrPresent {
return nil
}
var dts time.Duration var dts time.Duration
if !e.firstIDRReceived { if !e.firstIDRReceived {