1 /*
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "modules/rtp_rtcp/source/rtcp_packet/common_header.h"
12
13 #include "test/gtest.h"
14
15 using webrtc::rtcp::CommonHeader;
16
17 namespace webrtc {
18
TEST(RtcpCommonHeaderTest,TooSmallBuffer)19 TEST(RtcpCommonHeaderTest, TooSmallBuffer) {
20 uint8_t buffer[] = {0x80, 0x00, 0x00, 0x00};
21 CommonHeader header;
22 // Buffer needs to be able to hold the header.
23 EXPECT_FALSE(header.Parse(buffer, 0));
24 EXPECT_FALSE(header.Parse(buffer, 1));
25 EXPECT_FALSE(header.Parse(buffer, 2));
26 EXPECT_FALSE(header.Parse(buffer, 3));
27 EXPECT_TRUE(header.Parse(buffer, 4));
28 }
29
TEST(RtcpCommonHeaderTest,Version)30 TEST(RtcpCommonHeaderTest, Version) {
31 uint8_t buffer[] = {0x00, 0x00, 0x00, 0x00};
32 CommonHeader header;
33 // Version 2 is the only allowed.
34 buffer[0] = 0 << 6;
35 EXPECT_FALSE(header.Parse(buffer, sizeof(buffer)));
36 buffer[0] = 1 << 6;
37 EXPECT_FALSE(header.Parse(buffer, sizeof(buffer)));
38 buffer[0] = 2 << 6;
39 EXPECT_TRUE(header.Parse(buffer, sizeof(buffer)));
40 buffer[0] = 3 << 6;
41 EXPECT_FALSE(header.Parse(buffer, sizeof(buffer)));
42 }
43
TEST(RtcpCommonHeaderTest,PacketSize)44 TEST(RtcpCommonHeaderTest, PacketSize) {
45 uint8_t buffer[] = {0x80, 0x00, 0x00, 0x02, 0x00, 0x00,
46 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
47 CommonHeader header;
48 EXPECT_FALSE(header.Parse(buffer, sizeof(buffer) - 1));
49 EXPECT_TRUE(header.Parse(buffer, sizeof(buffer)));
50 EXPECT_EQ(8u, header.payload_size_bytes());
51 EXPECT_EQ(buffer + sizeof(buffer), header.NextPacket());
52 EXPECT_EQ(sizeof(buffer), header.packet_size());
53 }
54
TEST(RtcpCommonHeaderTest,PaddingAndPayloadSize)55 TEST(RtcpCommonHeaderTest, PaddingAndPayloadSize) {
56 // Set v = 2, p = 1, but leave fmt, pt as 0.
57 uint8_t buffer[] = {0xa0, 0x00, 0x00, 0x00, 0x00, 0x00,
58 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
59 CommonHeader header;
60 // Padding bit set, but no byte for padding (can't specify padding length).
61 EXPECT_FALSE(header.Parse(buffer, 4));
62
63 buffer[3] = 2; // Set payload size to 2x32bit.
64 const size_t kPayloadSizeBytes = buffer[3] * 4;
65 const size_t kPaddingAddress =
66 CommonHeader::kHeaderSizeBytes + kPayloadSizeBytes - 1;
67
68 // Padding one byte larger than possible.
69 buffer[kPaddingAddress] = kPayloadSizeBytes + 1;
70 EXPECT_FALSE(header.Parse(buffer, sizeof(buffer)));
71
72 // Invalid zero padding size.
73 buffer[kPaddingAddress] = 0;
74 EXPECT_FALSE(header.Parse(buffer, sizeof(buffer)));
75
76 // Pure padding packet.
77 buffer[kPaddingAddress] = kPayloadSizeBytes;
78 EXPECT_TRUE(header.Parse(buffer, sizeof(buffer)));
79 EXPECT_EQ(0u, header.payload_size_bytes());
80 EXPECT_EQ(buffer + sizeof(buffer), header.NextPacket());
81 EXPECT_EQ(header.payload(), buffer + CommonHeader::kHeaderSizeBytes);
82 EXPECT_EQ(header.packet_size(), sizeof(buffer));
83
84 // Single byte of actual data.
85 buffer[kPaddingAddress] = kPayloadSizeBytes - 1;
86 EXPECT_TRUE(header.Parse(buffer, sizeof(buffer)));
87 EXPECT_EQ(1u, header.payload_size_bytes());
88 EXPECT_EQ(buffer + sizeof(buffer), header.NextPacket());
89 EXPECT_EQ(header.packet_size(), sizeof(buffer));
90 }
91
TEST(RtcpCommonHeaderTest,FormatAndPayloadType)92 TEST(RtcpCommonHeaderTest, FormatAndPayloadType) {
93 uint8_t buffer[] = {0x9e, 0xab, 0x00, 0x00};
94 CommonHeader header;
95 EXPECT_TRUE(header.Parse(buffer, sizeof(buffer)));
96
97 EXPECT_EQ(header.count(), 0x1e);
98 EXPECT_EQ(header.fmt(), 0x1e);
99 EXPECT_EQ(header.type(), 0xab);
100 EXPECT_EQ(header.payload_size_bytes(), 0u);
101 EXPECT_EQ(header.payload(), buffer + CommonHeader::kHeaderSizeBytes);
102 }
103 } // namespace webrtc
104