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

rename pkg/formats into pkt/format

Browse Source
This commit is contained in:
aler9
2023-08-14 20:47:28 +02:00
parent e2c526b3b8
commit 5e5a21240d
228 changed files with 266 additions and 266 deletions
Download Patch File Download Diff File Expand all files Collapse all files

244
pkg/format/rtph264/decoder_test.go Normal file
View File

@@ -0,0 +1,244 @@
package rtph264
import (
"bytes"
"testing"
"github.com/bluenviron/mediacommon/pkg/codecs/h264"
"github.com/pion/rtp"
"github.com/stretchr/testify/require"
)
func TestDecode(t *testing.T) {
for _, ca := range cases {
t.Run(ca.name, func(t *testing.T) {
d := &Decoder{}
err := d.Init()
require.NoError(t, err)
var nalus [][]byte
for _, pkt := range ca.pkts {
clone := pkt.Clone()
addNALUs, _, err := d.Decode(pkt)
// test input integrity
require.Equal(t, clone, pkt)
if err == ErrMorePacketsNeeded {
continue
}
require.NoError(t, err)
nalus = append(nalus, addNALUs...)
}
require.Equal(t, ca.nalus, nalus)
})
}
}
func TestDecodeCorruptedFragment(t *testing.T) {
d := &Decoder{}
err := d.Init()
require.NoError(t, err)
_, _, err = d.Decode(&rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 96,
SequenceNumber: 17645,
Timestamp: 2289527317,
SSRC: 0x9dbb7812,
},
Payload: mergeBytes(
[]byte{
0x1c, 0x85,
},
bytes.Repeat([]byte{0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07}, 182),
[]byte{0x00, 0x01},
),
})
require.Equal(t, ErrMorePacketsNeeded, err)
nalus, _, err := d.Decode(&rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 96,
SequenceNumber: 17646,
Timestamp: 2289527317,
SSRC: 0x9dbb7812,
},
Payload: []byte{0x01, 0x00},
})
require.NoError(t, err)
require.Equal(t, [][]byte{{0x01, 0x00}}, nalus)
}
func TestDecodeSTAPAWithPadding(t *testing.T) {
d := &Decoder{}
err := d.Init()
require.NoError(t, err)
pkt := rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 96,
SequenceNumber: 17645,
Timestamp: 2289526357,
SSRC: 0x9dbb7812,
},
Payload: []byte{
0x18, 0x00, 0x02, 0xaa,
0xbb, 0x00, 0x02, 0xcc, 0xdd, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
},
}
nalus, _, err := d.Decode(&pkt)
require.NoError(t, err)
require.Equal(t, [][]byte{
{0xaa, 0xbb},
{0xcc, 0xdd},
}, nalus)
}
func TestDecodeAnnexB(t *testing.T) {
d := &Decoder{}
err := d.Init()
require.NoError(t, err)
nalus, _, err := d.Decode(&rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 96,
SequenceNumber: 17647,
Timestamp: 2289531307,
SSRC: 0x9dbb7812,
},
Payload: mergeBytes(
[]byte{0x01, 0x02, 0x03, 0x04},
),
})
require.NoError(t, err)
require.Equal(t, [][]byte{
{0x01, 0x02, 0x03, 0x04},
}, nalus)
for i := 0; i < 2; i++ {
nalus, _, err := d.Decode(&rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 96,
SequenceNumber: 17647,
Timestamp: 2289531307,
SSRC: 0x9dbb7812,
},
Payload: mergeBytes(
[]byte{0x00, 0x00, 0x00, 0x01},
[]byte{0x01, 0x02, 0x03, 0x04},
[]byte{0x00, 0x00, 0x00, 0x01},
[]byte{0x01, 0x02, 0x03, 0x04},
),
})
require.NoError(t, err)
require.Equal(t, [][]byte{
{0x01, 0x02, 0x03, 0x04},
{0x01, 0x02, 0x03, 0x04},
}, nalus)
}
}
func TestDecodeAccessUnit(t *testing.T) {
d := &Decoder{}
err := d.Init()
require.NoError(t, err)
nalus, _, err := d.Decode(&rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: false,
PayloadType: 96,
SequenceNumber: 17647,
Timestamp: 2289531307,
SSRC: 0x9dbb7812,
},
Payload: []byte{0x01, 0x02},
})
require.Equal(t, ErrMorePacketsNeeded, err)
require.Equal(t, [][]byte(nil), nalus)
nalus, _, err = d.Decode(&rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 96,
SequenceNumber: 17647,
Timestamp: 2289531307,
SSRC: 0x9dbb7812,
},
Payload: []byte{0x01, 0x02},
})
require.NoError(t, err)
require.Equal(t, [][]byte{{0x01, 0x02}, {0x01, 0x02}}, nalus)
}
func TestDecoderErrorLimit(t *testing.T) {
d := &Decoder{}
err := d.Init()
require.NoError(t, err)
for i := 0; i <= h264.MaxNALUsPerAccessUnit; i++ {
_, _, err = d.Decode(&rtp.Packet{
Header: rtp.Header{
Version: 2,
Marker: false,
PayloadType: 96,
SequenceNumber: 17645,
Timestamp: 2289527317,
SSRC: 0x9dbb7812,
},
Payload: []byte{1, 2, 3, 4},
})
}
require.EqualError(t, err, "NALU count exceeds maximum allowed (20)")
}
func FuzzDecoder(f *testing.F) {
f.Fuzz(func(t *testing.T, a []byte, b []byte) {
d := &Decoder{}
d.Init() //nolint:errcheck
d.Decode(&rtp.Packet{ //nolint:errcheck
Header: rtp.Header{
Version: 2,
Marker: false,
PayloadType: 96,
SequenceNumber: 17645,
Timestamp: 2289527317,
SSRC: 0x9dbb7812,
},
Payload: a,
})
d.Decode(&rtp.Packet{ //nolint:errcheck
Header: rtp.Header{
Version: 2,
Marker: false,
PayloadType: 96,
SequenceNumber: 17645,
Timestamp: 2289527317,
SSRC: 0x9dbb7812,
},
Payload: b,
})
})
}