gortsplib/pkg/format/mpeg4_audio_test.go

package format

import (
	"testing"

	"github.com/pion/rtp"
	"github.com/stretchr/testify/require"

	"github.com/bluenviron/mediacommon/pkg/codecs/mpeg4audio"
)

func TestMPEG4AudioAttributes(t *testing.T) {
	t.Run("generic", func(t *testing.T) {
		format := &MPEG4Audio{
			PayloadTyp: 96,
			Config: &mpeg4audio.Config{
				Type:         mpeg4audio.ObjectTypeAACLC,
				SampleRate:   48000,
				ChannelCount: 2,
			},
			SizeLength:       13,
			IndexLength:      3,
			IndexDeltaLength: 3,
		}
		require.Equal(t, "MPEG-4 Audio", format.Codec())
		require.Equal(t, 48000, format.ClockRate())
		require.Equal(t, true, format.PTSEqualsDTS(&rtp.Packet{}))
		require.Equal(t, &mpeg4audio.Config{
			Type:         mpeg4audio.ObjectTypeAACLC,
			SampleRate:   48000,
			ChannelCount: 2,
		}, format.GetConfig())
	})

	t.Run("latm", func(t *testing.T) {
		format := &MPEG4Audio{
			LATM:           true,
			PayloadTyp:     96,
			ProfileLevelID: 1,
			StreamMuxConfig: &mpeg4audio.StreamMuxConfig{
				Programs: []*mpeg4audio.StreamMuxConfigProgram{{
					Layers: []*mpeg4audio.StreamMuxConfigLayer{{
						AudioSpecificConfig: &mpeg4audio.Config{
							Type:         2,
							SampleRate:   44100,
							ChannelCount: 2,
						},
						LatmBufferFullness: 255,
					}},
				}},
			},
		}
		require.Equal(t, "MPEG-4 Audio", format.Codec())
		require.Equal(t, 44100, format.ClockRate())
		require.Equal(t, true, format.PTSEqualsDTS(&rtp.Packet{}))
		require.Equal(t, &mpeg4audio.Config{
			Type:         2,
			SampleRate:   44100,
			ChannelCount: 2,
		}, format.GetConfig())
	})
}

func TestMPEG4AudioDecEncoder(t *testing.T) {
	t.Run("generic", func(t *testing.T) {
		format := &MPEG4Audio{
			PayloadTyp: 96,
			Config: &mpeg4audio.Config{
				Type:         mpeg4audio.ObjectTypeAACLC,
				SampleRate:   48000,
				ChannelCount: 2,
			},
			SizeLength:       13,
			IndexLength:      3,
			IndexDeltaLength: 3,
		}

		enc, err := format.CreateEncoder()
		require.NoError(t, err)

		pkts, err := enc.Encode([][]byte{{0x01, 0x02, 0x03, 0x04}})
		require.NoError(t, err)
		require.Equal(t, format.PayloadType(), pkts[0].PayloadType)

		dec, err := format.CreateDecoder()
		require.NoError(t, err)

		byts, err := dec.Decode(pkts[0])
		require.NoError(t, err)
		require.Equal(t, [][]byte{{0x01, 0x02, 0x03, 0x04}}, byts)
	})

	t.Run("latm", func(t *testing.T) {
		format := &MPEG4Audio{
			LATM:           true,
			PayloadTyp:     96,
			ProfileLevelID: 1,
			StreamMuxConfig: &mpeg4audio.StreamMuxConfig{
				Programs: []*mpeg4audio.StreamMuxConfigProgram{{
					Layers: []*mpeg4audio.StreamMuxConfigLayer{{
						AudioSpecificConfig: &mpeg4audio.Config{
							Type:         2,
							SampleRate:   48000,
							ChannelCount: 2,
						},
						LatmBufferFullness: 255,
					}},
				}},
			},
		}

		enc, err := format.CreateEncoder()
		require.NoError(t, err)

		pkts, err := enc.Encode([][]byte{{0x01, 0x02, 0x03, 0x04}})
		require.NoError(t, err)
		require.Equal(t, format.PayloadType(), pkts[0].PayloadType)

		dec, err := format.CreateDecoder()
		require.NoError(t, err)

		byts, err := dec.Decode(pkts[0])
		require.NoError(t, err)
		require.Equal(t, [][]byte{{0x01, 0x02, 0x03, 0x04}}, byts)
	})
}

func FuzzUnmarshalMPEG4AudioGeneric(f *testing.F) {
	f.Fuzz(func(
		_ *testing.T,
		a bool,
		b string,
		c bool,
		d string,
		e bool,
		f string,
		g bool,
		h string,
		i bool,
		j string,
		k bool,
		l string,
		m bool,
		n string,
	) {
		ma := map[string]string{}

		if a {
			ma["streamtype"] = b
		}

		if c {
			ma["mode"] = d
		}

		if e {
			ma["profile-level-id"] = f
		}

		if g {
			ma["config"] = h
		}

		if i {
			ma["sizelength"] = j
		}

		if k {
			ma["indexlength"] = l
		}

		if m {
			ma["indexdeltalength"] = n
		}

		fo, err := Unmarshal("audio", 96, "MPEG4-generic/48000/2", ma) //nolint:errcheck
		if err == nil {
			fo.(*MPEG4Audio).GetConfig()
		}
	})
}

func FuzzUnmarshalMPEG4AudioLATM(f *testing.F) {
	f.Fuzz(func(
		_ *testing.T,
		a bool,
		b string,
		c bool,
		d string,
		e bool,
		f string,
		g bool,
		h string,
		i bool,
		j string,
		k bool,
		l string,
	) {
		ma := map[string]string{}

		if a {
			ma["profile-level-id"] = b
		}

		if c {
			ma["bitrate"] = d
		}

		if e {
			ma["cpresent"] = f
		}

		if g {
			ma["config"] = h
		}

		if i {
			ma["object"] = j
		}

		if k {
			ma["sbr-enabled"] = l
		}

		fo, err := Unmarshal("audio", 96, "MP4A-LATM/48000/2", ma)
		if err == nil {
			fo.(*MPEG4Audio).RTPMap()
			fo.(*MPEG4Audio).FMTP()
			fo.(*MPEG4Audio).GetConfig()
		}
	})
}