webrtc/pkg/rtcp/reader_test.go

package rtcp

import (
	"bytes"
	"io"
	"reflect"
	"testing"

	"github.com/stretchr/testify/assert"
)

// An RTCP packet from a packet dump
var realPacket = []byte{
	// Receiver Report (offset=0)
	// v=2, p=0, count=1, RR, len=7
	0x81, 0xc9, 0x0, 0x7,
	// ssrc=0x902f9e2e
	0x90, 0x2f, 0x9e, 0x2e,
	// ssrc=0xbc5e9a40
	0xbc, 0x5e, 0x9a, 0x40,
	// fracLost=0, totalLost=0
	0x0, 0x0, 0x0, 0x0,
	// lastSeq=0x46e1
	0x0, 0x0, 0x46, 0xe1,
	// jitter=273
	0x0, 0x0, 0x1, 0x11,
	// lsr=0x9f36432
	0x9, 0xf3, 0x64, 0x32,
	// delay=150137
	0x0, 0x2, 0x4a, 0x79,

	// Source Description (offset=32)
	// v=2, p=0, count=1, SDES, len=12
	0x81, 0xca, 0x0, 0xc,
	// ssrc=0x902f9e2e
	0x90, 0x2f, 0x9e, 0x2e,
	// CNAME, len=38
	0x1, 0x26,
	// text="{9c00eb92-1afb-9d49-a47d-91f64eee69f5}"
	0x7b, 0x39, 0x63, 0x30,
	0x30, 0x65, 0x62, 0x39,
	0x32, 0x2d, 0x31, 0x61,
	0x66, 0x62, 0x2d, 0x39,
	0x64, 0x34, 0x39, 0x2d,
	0x61, 0x34, 0x37, 0x64,
	0x2d, 0x39, 0x31, 0x66,
	0x36, 0x34, 0x65, 0x65,
	0x65, 0x36, 0x39, 0x66,
	0x35, 0x7d,
	// END + padding
	0x0, 0x0, 0x0, 0x0,

	// Goodbye (offset=84)
	// v=2, p=0, count=1, BYE, len=1
	0x81, 0xcb, 0x0, 0x1,
	// source=0x902f9e2e
	0x90, 0x2f, 0x9e, 0x2e,

	// Picture Loss Indication (offset=92)
	0x81, 0xce, 0x0, 0x2,
	// sender=0x902f9e2e
	0x90, 0x2f, 0x9e, 0x2e,
	// media=0x902f9e2e
	0x90, 0x2f, 0x9e, 0x2e,

	// RapidResynchronizationRequest (offset=104)
	0x85, 0xcd, 0x0, 0x2,
	// sender=0x902f9e2e
	0x90, 0x2f, 0x9e, 0x2e,
	// media=0x902f9e2e
	0x90, 0x2f, 0x9e, 0x2e,
}

func TestUnmarshal(t *testing.T) {
	r := NewReader(bytes.NewReader(realPacket))

	// ReceiverReport
	_, packet, err := r.ReadPacket()
	if err != nil {
		t.Fatalf("Read rr: %v", err)
	}
	var parsed Packet
	if parsed, _, err = Unmarshal(packet); err != nil {
		t.Errorf("Unmarshal parsed: %v", err)
	}
	assert.IsType(t, parsed, (*ReceiverReport)(nil), "Unmarshalled to incorrect type")

	wantRR := &ReceiverReport{
		SSRC: 0x902f9e2e,
		Reports: []ReceptionReport{{
			SSRC:               0xbc5e9a40,
			FractionLost:       0,
			TotalLost:          0,
			LastSequenceNumber: 0x46e1,
			Jitter:             273,
			LastSenderReport:   0x9f36432,
			Delay:              150137,
		}},
		ProfileExtensions: []uint8{},
	}
	if got, want := wantRR, parsed; !reflect.DeepEqual(got, want) {
		t.Errorf("Unmarshal rr: got %#v, want %#v", got, want)
	}

	// SourceDescription
	_, packet, err = r.ReadPacket()
	if err != nil {
		t.Fatalf("Read sdes: %v", err)
	}

	if parsed, _, err = Unmarshal(packet); err != nil {
		t.Errorf("Unmarshal parsed: %v", err)
	}
	assert.IsType(t, parsed, (*SourceDescription)(nil), "Unmarshalled to incorrect type")

	wantSdes := &SourceDescription{
		Chunks: []SourceDescriptionChunk{
			{
				Source: 0x902f9e2e,
				Items: []SourceDescriptionItem{
					{
						Type: SDESCNAME,
						Text: "{9c00eb92-1afb-9d49-a47d-91f64eee69f5}",
					},
				},
			},
		},
	}

	if got, want := parsed, wantSdes; !reflect.DeepEqual(got, want) {
		t.Errorf("Unmarshal sdes: got %#v, want %#v", got, want)
	}

	// Goodbye
	_, packet, err = r.ReadPacket()
	if err != nil {
		t.Fatalf("Read bye: %v", err)
	}

	if parsed, _, err = Unmarshal(packet); err != nil {
		t.Errorf("Unmarshal parsed: %v", err)
	}

	assert.IsType(t, parsed, (*Goodbye)(nil), "Unmarshalled to incorrect type")

	wantBye := &Goodbye{
		Sources: []uint32{0x902f9e2e},
	}
	if got, want := parsed, wantBye; !reflect.DeepEqual(got, want) {
		t.Errorf("Unmarshal bye: got %#v, want %#v", got, want)
	}

	// PictureLossIndication
	_, packet, err = r.ReadPacket()
	if err != nil {
		t.Fatalf("Read pli: %v", err)
	}

	if parsed, _, err = Unmarshal(packet); err != nil {
		t.Errorf("Unmarshal parsed: %v", err)
	}

	assert.IsType(t, parsed, (*PictureLossIndication)(nil), "Unmarshalled to incorrect type")

	wantPli := &PictureLossIndication{
		SenderSSRC: 0x902f9e2e,
		MediaSSRC:  0x902f9e2e,
	}
	if got, want := parsed, wantPli; !reflect.DeepEqual(got, want) {
		t.Errorf("Unmarshal pli: got %#v, want %#v", got, want)
	}

	// RapidResynchronizationRequest
	_, packet, err = r.ReadPacket()
	if err != nil {
		t.Fatalf("Read rrr: %v", err)
	}

	if parsed, _, err = Unmarshal(packet); err != nil {
		t.Errorf("Unmarshal parsed: %v", err)
	}

	assert.IsType(t, parsed, (*RapidResynchronizationRequest)(nil), "Unmarshalled to incorrect type")

	wantRrr := &RapidResynchronizationRequest{
		SenderSSRC: 0x902f9e2e,
		MediaSSRC:  0x902f9e2e,
	}
	if got, want := parsed, wantRrr; !reflect.DeepEqual(got, want) {
		t.Errorf("Unmarshal rrr: got %#v, want %#v", got, want)
	}
}

func TestReadEOF(t *testing.T) {
	shortHeader := []byte{
		0x81, 0xc9, // missing type & len
	}

	r := NewReader(bytes.NewReader(shortHeader))
	_, _, err := r.ReadPacket()
	if got, want := err, io.ErrUnexpectedEOF; got != want {
		t.Fatalf("read short header: got err = %v, want %v", got, want)
	}
}