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

add MPEG4Video.SafeParams(), MPEG4Video.SafeSetParams() (#395)

Browse Source
This commit is contained in:
Alessandro Ros
2023-09-01 18:36:21 +02:00
committed by GitHub
parent 8f18a0d83b
commit 6c413c03f5
7 changed files with 45 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/format/rtpmpeg4video/decoder.go
View File

@@ -5,15 +5,13 @@ import (
"fmt"
"github.com/pion/rtp"
"github.com/bluenviron/mediacommon/pkg/codecs/mpeg4video"
)
// ErrMorePacketsNeeded is returned when more packets are needed.
var ErrMorePacketsNeeded = errors.New("need more packets")
const (
maxFrameSize = 1 * 1024 * 1024
)
func joinFragments(fragments [][]byte, size int) []byte {
ret := make([]byte, size)
n := 0
@@ -49,9 +47,9 @@ func (d *Decoder) Decode(pkt *rtp.Packet) ([]byte, error) {
}
} else {
d.fragmentsSize += len(pkt.Payload)
if d.fragmentsSize > maxFrameSize {
if d.fragmentsSize > mpeg4video.MaxFrameSize {
d.fragments = d.fragments[:0] // discard pending fragments
return nil, fmt.Errorf("frame size (%d) is too big, maximum is %d", d.fragmentsSize, maxFrameSize)
return nil, fmt.Errorf("frame size (%d) is too big, maximum is %d", d.fragmentsSize, mpeg4video.MaxFrameSize)
}
d.fragments = append(d.fragments, pkt.Payload)