kunmeng/vdk
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

testing

Browse Source
This commit is contained in:
Andrey Semochkin 2024-12-18 10:46:02 +03:00
parent b6b1f4a437
commit 5438b5fbf7
1 changed files with 41 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
format/mp4/mp4io/atoms.go
View File

@ -1,6 +1,7 @@
package mp4io package mp4io
import ( import (
"log"
"time" "time"
"github.com/deepch/vdk/utils/bits/pio" "github.com/deepch/vdk/utils/bits/pio"
@ -32,6 +33,7 @@ func (self HV1Desc) Tag() Tag {
} }
// const HVC1 = Tag(0x68766331) // const HVC1 = Tag(0x68766331)
//
// func (self HVC1Desc) Tag() Tag { // func (self HVC1Desc) Tag() Tag {
// return HVC1 // return HVC1
// } // }
@ -571,6 +573,11 @@ func (self *Track) Unmarshal(b []byte, offset int) (n int, err error) {
for n+8 < len(b) { for n+8 < len(b) {
tag := Tag(pio.U32BE(b[n+4:])) tag := Tag(pio.U32BE(b[n+4:]))
size := int(pio.U32BE(b[n:])) size := int(pio.U32BE(b[n:]))
if size > 5242880 {
log.Println("track size", size)
err = parseErr("_len_Entries", 5242880, err)
return
}
if len(b) < n+size { if len(b) < n+size {
err = parseErr("TagSizeInvalid", n+offset, err) err = parseErr("TagSizeInvalid", n+offset, err)
return return
@ -2552,6 +2559,10 @@ func (self *TimeToSample) Unmarshal(b []byte, offset int) (n int, err error) {
var _len_Entries uint32 var _len_Entries uint32
_len_Entries = pio.U32BE(b[n:]) _len_Entries = pio.U32BE(b[n:])
n += 4 n += 4
if _len_Entries > 5242880 {
err = parseErr("_len_Entries", 5242880, err)
return
}
self.Entries = make([]TimeToSampleEntry, _len_Entries) self.Entries = make([]TimeToSampleEntry, _len_Entries)
if len(b) < n+LenTimeToSampleEntry*len(self.Entries) { if len(b) < n+LenTimeToSampleEntry*len(self.Entries) {
err = parseErr("TimeToSampleEntry", n+offset, err) err = parseErr("TimeToSampleEntry", n+offset, err)
@ -2636,6 +2647,10 @@ func (self *SampleToChunk) Unmarshal(b []byte, offset int) (n int, err error) {
var _len_Entries uint32 var _len_Entries uint32
_len_Entries = pio.U32BE(b[n:]) _len_Entries = pio.U32BE(b[n:])
n += 4 n += 4
if _len_Entries > 5242880 {
err = parseErr("_len_Entries", 5242880, err)
return
}
self.Entries = make([]SampleToChunkEntry, _len_Entries) self.Entries = make([]SampleToChunkEntry, _len_Entries)
if len(b) < n+LenSampleToChunkEntry*len(self.Entries) { if len(b) < n+LenSampleToChunkEntry*len(self.Entries) {
err = parseErr("SampleToChunkEntry", n+offset, err) err = parseErr("SampleToChunkEntry", n+offset, err)
@ -2723,6 +2738,10 @@ func (self *CompositionOffset) Unmarshal(b []byte, offset int) (n int, err error
var _len_Entries uint32 var _len_Entries uint32
_len_Entries = pio.U32BE(b[n:]) _len_Entries = pio.U32BE(b[n:])
n += 4 n += 4
if _len_Entries > 5242880 {
err = parseErr("_len_Entries", 5242880, err)
return
}
self.Entries = make([]CompositionOffsetEntry, _len_Entries) self.Entries = make([]CompositionOffsetEntry, _len_Entries)
if len(b) < n+LenCompositionOffsetEntry*len(self.Entries) { if len(b) < n+LenCompositionOffsetEntry*len(self.Entries) {
err = parseErr("CompositionOffsetEntry", n+offset, err) err = parseErr("CompositionOffsetEntry", n+offset, err)
@ -2807,6 +2826,10 @@ func (self *SyncSample) Unmarshal(b []byte, offset int) (n int, err error) {
var _len_Entries uint32 var _len_Entries uint32
_len_Entries = pio.U32BE(b[n:]) _len_Entries = pio.U32BE(b[n:])
n += 4 n += 4
if _len_Entries > 5242880 {
err = parseErr("_len_Entries", 5242880, err)
return
}
self.Entries = make([]uint32, _len_Entries) self.Entries = make([]uint32, _len_Entries)
if len(b) < n+4*len(self.Entries) { if len(b) < n+4*len(self.Entries) {
err = parseErr("uint32", n+offset, err) err = parseErr("uint32", n+offset, err)
@ -2874,6 +2897,10 @@ func (self *ChunkOffset) Unmarshal(b []byte, offset int) (n int, err error) {
var _len_Entries uint32 var _len_Entries uint32
_len_Entries = pio.U32BE(b[n:]) _len_Entries = pio.U32BE(b[n:])
n += 4 n += 4
if _len_Entries > 5242880 {
err = parseErr("_len_Entries", 5242880, err)
return
}
self.Entries = make([]uint32, _len_Entries) self.Entries = make([]uint32, _len_Entries)
if len(b) < n+4*len(self.Entries) { if len(b) < n+4*len(self.Entries) {
err = parseErr("uint32", n+offset, err) err = parseErr("uint32", n+offset, err)
@ -3388,6 +3415,11 @@ func (self *SampleSize) Unmarshal(b []byte, offset int) (n int, err error) {
var _len_Entries uint32 var _len_Entries uint32
_len_Entries = pio.U32BE(b[n:]) _len_Entries = pio.U32BE(b[n:])
n += 4 n += 4
if _len_Entries > 5242880 {
err = parseErr("_len_Entries", 5242880, err)
log.Println("_len_Entries", _len_Entries)
return
}
self.Entries = make([]uint32, _len_Entries) self.Entries = make([]uint32, _len_Entries)
if len(b) < n+4*len(self.Entries) { if len(b) < n+4*len(self.Entries) {
err = parseErr("uint32", n+offset, err) err = parseErr("uint32", n+offset, err)
@ -3520,6 +3552,10 @@ func (self *TrackFragRun) Unmarshal(b []byte, offset int) (n int, err error) {
var _len_Entries uint32 var _len_Entries uint32
_len_Entries = pio.U32BE(b[n:]) _len_Entries = pio.U32BE(b[n:])
n += 4 n += 4
if _len_Entries > 5242880 {
err = parseErr("_len_Entries", 5242880, err)
return
}
self.Entries = make([]TrackFragRunEntry, _len_Entries) self.Entries = make([]TrackFragRunEntry, _len_Entries)
if self.Flags&TRUN_DATA_OFFSET != 0 { if self.Flags&TRUN_DATA_OFFSET != 0 {
{ {