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

prevent decoding formats with zero clock rate or channels (#770)

Browse Source
This commit is contained in:
Alessandro Ros
2025-05-01 17:06:55 +02:00
committed by GitHub
parent 18307140ec
commit bfacd10d35
16 changed files with 76 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/format/ac3.go
View File

@@ -1,6 +1,7 @@
package format
import (
"fmt"
"strconv"
"strings"
@@ -23,15 +24,15 @@ func (f *AC3) unmarshal(ctx *unmarshalContext) error {
tmp := strings.SplitN(ctx.clock, "/", 2)
tmp1, err := strconv.ParseUint(tmp[0], 10, 31)
if err != nil {
return err
if err != nil || tmp1 == 0 {
return fmt.Errorf("invalid sample rate: '%s'", tmp[0])
}
f.SampleRate = int(tmp1)
if len(tmp) >= 2 {
tmp1, err := strconv.ParseUint(tmp[1], 10, 31)
if err != nil {
return err
if err != nil || tmp1 == 0 {
return fmt.Errorf("invalid channel count: '%s'", tmp[1])
}
f.ChannelCount = int(tmp1)
} else {

46
pkg/format/format_test.go
View File

@@ -1267,19 +1267,45 @@ func FuzzUnmarshal(f *testing.F) {
f.Add(ca.in)
}
f.Fuzz(func(_ *testing.T, in string) {
f.Fuzz(func(t *testing.T, in string) {
var desc sdp.SessionDescription
err := desc.Unmarshal([]byte(in))
if err != nil || len(desc.MediaDescriptions) == 0 || len(desc.MediaDescriptions[0].MediaName.Formats) == 0 {
return
}
if err == nil && len(desc.MediaDescriptions) >= 1 && len(desc.MediaDescriptions[0].MediaName.Formats) >= 1 {
f, err := Unmarshal(desc.MediaDescriptions[0], desc.MediaDescriptions[0].MediaName.Formats[0])
if err == nil {
f.Codec()
f.ClockRate()
f.PayloadType()
f.RTPMap()
f.FMTP()
}
f, err := Unmarshal(desc.MediaDescriptions[0], desc.MediaDescriptions[0].MediaName.Formats[0])
if err != nil {
return
}
// only Generic can return zero ClockRate
if _, ok := f.(*Generic); !ok {
require.NotZero(t, f.ClockRate())
} else {
f.ClockRate()
}
f.Codec()
f.PayloadType()
f.RTPMap()
f.FMTP()
switch f := f.(type) {
case *AC3:
require.NotZero(t, f.ChannelCount)
case *G711:
require.NotZero(t, f.ChannelCount)
case *LPCM:
require.NotZero(t, f.ChannelCount)
case *Opus:
require.NotZero(t, f.ChannelCount)
case *Vorbis:
require.NotZero(t, f.ChannelCount)
}
})
}

9
pkg/format/g711.go
View File

@@ -1,6 +1,7 @@
package format
import (
"fmt"
"strconv"
"strings"
@@ -40,15 +41,15 @@ func (f *G711) unmarshal(ctx *unmarshalContext) error {
tmp := strings.SplitN(ctx.clock, "/", 2)
tmp1, err := strconv.ParseUint(tmp[0], 10, 31)
if err != nil {
return err
if err != nil || tmp1 == 0 {
return fmt.Errorf("invalid sample rate: '%s'", tmp[0])
}
f.SampleRate = int(tmp1)
if len(tmp) >= 2 {
tmp1, err := strconv.ParseUint(tmp[1], 10, 31)
if err != nil {
return err
if err != nil || tmp1 == 0 {
return fmt.Errorf("invalid channel count: '%s'", tmp[1])
}
f.ChannelCount = int(tmp1)
} else {

9
pkg/format/lpcm.go
View File

@@ -1,6 +1,7 @@
package format
import (
"fmt"
"strconv"
"strings"
@@ -50,15 +51,15 @@ func (f *LPCM) unmarshal(ctx *unmarshalContext) error {
tmp := strings.SplitN(ctx.clock, "/", 2)
tmp1, err := strconv.ParseUint(tmp[0], 10, 31)
if err != nil {
return err
if err != nil || tmp1 == 0 {
return fmt.Errorf("invalid sample rate: '%s'", tmp[0])
}
f.SampleRate = int(tmp1)
if len(tmp) >= 2 {
tmp1, err := strconv.ParseUint(tmp[1], 10, 31)
if err != nil {
return err
if err != nil || tmp1 == 0 {
return fmt.Errorf("invalid channel count: '%s'", tmp[1])
}
f.ChannelCount = int(tmp1)
} else {

3
pkg/format/opus.go
View File

@@ -65,10 +65,9 @@ func (f *Opus) unmarshal(ctx *unmarshalContext) error {
}
channelCount, err := strconv.ParseUint(tmp[1], 10, 31)
if err != nil {
if err != nil || channelCount == 0 {
return fmt.Errorf("invalid channel count: '%s'", tmp[1])
}
f.ChannelCount = int(channelCount)
}

4
pkg/format/speex.go
View File

@@ -19,8 +19,8 @@ func (f *Speex) unmarshal(ctx *unmarshalContext) error {
f.PayloadTyp = ctx.payloadType
sampleRate, err := strconv.ParseUint(ctx.clock, 10, 31)
if err != nil {
return err
if err != nil || sampleRate == 0 {
return fmt.Errorf("invalid sample rate: '%s'", ctx.clock)
}
f.SampleRate = int(sampleRate)

2
pkg/format/testdata/fuzz/FuzzUnmarshal/2910e512f655ed6b vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 96\na=rtpmap:96 VORBIS/0/0\na=fmtp:96 ConfigurAtion=")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/2beaf42d3ab9327e vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 97\na=rtpmap:97 L8/1/0")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/2c9087410b17869e vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 96\na=rtpmap:96 PCMA/1/0")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/5c81ab28bdb170e5 vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 97\na=rtpmap:97 AC3/0")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/6d4ccd096ae9afd2 vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 96\na=rtpmap:96 AC3/1/0")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/72052247f0539111 vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 96\na=rtpmap:96 multiopus/48000/0")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/930d7278501087f2 vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 96\na=rtpmap:96 VORBIS/1/0\na=fmtp:96 ConfigurAtion=")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/b1a3816de12e306d vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 96\na=rtpmap:96 speeX/0")

2
pkg/format/testdata/fuzz/FuzzUnmarshal/c3f02e1806e34919 vendored Normal file
View File

@@ -0,0 +1,2 @@
go test fuzz v1
string("m=audio 0 AVP 97\na=rtpmap:97 L16/0")

8
pkg/format/vorbis.go
View File

@@ -27,14 +27,14 @@ func (f *Vorbis) unmarshal(ctx *unmarshalContext) error {
}
sampleRate, err := strconv.ParseUint(tmp[0], 10, 31)
if err != nil {
return err
if err != nil || sampleRate == 0 {
return fmt.Errorf("invalid sample rate: '%s'", tmp[0])
}
f.SampleRate = int(sampleRate)
channelCount, err := strconv.ParseUint(tmp[1], 10, 31)
if err != nil {
return err
if err != nil || channelCount == 0 {
return fmt.Errorf("invalid channel count: '%s'", tmp[1])
}
f.ChannelCount = int(channelCount)