OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/tmmbr.h" | 11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/tmmbr.h" |
12 | 12 |
13 #include "testing/gmock/include/gmock/gmock.h" | 13 #include "testing/gmock/include/gmock/gmock.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 #include "webrtc/test/rtcp_packet_parser.h" |
15 | 16 |
16 using testing::ElementsAreArray; | 17 using testing::ElementsAreArray; |
17 using testing::IsEmpty; | 18 using testing::IsEmpty; |
18 using testing::make_tuple; | 19 using testing::make_tuple; |
19 using webrtc::rtcp::TmmbItem; | 20 using webrtc::rtcp::TmmbItem; |
20 using webrtc::rtcp::Tmmbr; | 21 using webrtc::rtcp::Tmmbr; |
21 using webrtc::RTCPUtility::RtcpCommonHeader; | |
22 using webrtc::RTCPUtility::RtcpParseCommonHeader; | |
23 | 22 |
24 namespace webrtc { | 23 namespace webrtc { |
25 namespace { | 24 namespace { |
26 const uint32_t kSenderSsrc = 0x12345678; | 25 const uint32_t kSenderSsrc = 0x12345678; |
27 const uint32_t kRemoteSsrc = 0x23456789; | 26 const uint32_t kRemoteSsrc = 0x23456789; |
28 const uint32_t kBitrateBps = 312000; | 27 const uint32_t kBitrateBps = 312000; |
29 const uint16_t kOverhead = 0x1fe; | 28 const uint16_t kOverhead = 0x1fe; |
30 const uint8_t kPacket[] = {0x83, 205, 0x00, 0x04, | 29 const uint8_t kPacket[] = {0x83, 205, 0x00, 0x04, |
31 0x12, 0x34, 0x56, 0x78, | 30 0x12, 0x34, 0x56, 0x78, |
32 0x00, 0x00, 0x00, 0x00, | 31 0x00, 0x00, 0x00, 0x00, |
33 0x23, 0x45, 0x67, 0x89, | 32 0x23, 0x45, 0x67, 0x89, |
34 0x0a, 0x61, 0x61, 0xfe}; | 33 0x0a, 0x61, 0x61, 0xfe}; |
35 | |
36 bool ParseTmmbr(const uint8_t* buffer, size_t length, Tmmbr* tmmbr) { | |
37 RtcpCommonHeader header; | |
38 EXPECT_TRUE(RtcpParseCommonHeader(buffer, length, &header)); | |
39 EXPECT_EQ(length, header.BlockSize()); | |
40 return tmmbr->Parse(header, buffer + RtcpCommonHeader::kHeaderSizeBytes); | |
41 } | |
42 } // namespace | 34 } // namespace |
43 | 35 |
44 TEST(RtcpPacketTmmbrTest, Create) { | 36 TEST(RtcpPacketTmmbrTest, Create) { |
45 Tmmbr tmmbr; | 37 Tmmbr tmmbr; |
46 tmmbr.From(kSenderSsrc); | 38 tmmbr.From(kSenderSsrc); |
47 tmmbr.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); | 39 tmmbr.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); |
48 | 40 |
49 rtc::Buffer packet = tmmbr.Build(); | 41 rtc::Buffer packet = tmmbr.Build(); |
50 | 42 |
51 EXPECT_THAT(make_tuple(packet.data(), packet.size()), | 43 EXPECT_THAT(make_tuple(packet.data(), packet.size()), |
52 ElementsAreArray(kPacket)); | 44 ElementsAreArray(kPacket)); |
53 } | 45 } |
54 | 46 |
55 TEST(RtcpPacketTmmbrTest, Parse) { | 47 TEST(RtcpPacketTmmbrTest, Parse) { |
56 Tmmbr tmmbr; | 48 Tmmbr tmmbr; |
57 EXPECT_TRUE(ParseTmmbr(kPacket, sizeof(kPacket), &tmmbr)); | 49 EXPECT_TRUE(test::ParseSinglePacket(kPacket, &tmmbr)); |
58 const Tmmbr& parsed = tmmbr; | 50 const Tmmbr& parsed = tmmbr; |
59 | 51 |
60 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); | 52 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); |
61 ASSERT_EQ(1u, parsed.requests().size()); | 53 ASSERT_EQ(1u, parsed.requests().size()); |
62 EXPECT_EQ(kRemoteSsrc, parsed.requests().front().ssrc()); | 54 EXPECT_EQ(kRemoteSsrc, parsed.requests().front().ssrc()); |
63 EXPECT_EQ(kBitrateBps, parsed.requests().front().bitrate_bps()); | 55 EXPECT_EQ(kBitrateBps, parsed.requests().front().bitrate_bps()); |
64 EXPECT_EQ(kOverhead, parsed.requests().front().packet_overhead()); | 56 EXPECT_EQ(kOverhead, parsed.requests().front().packet_overhead()); |
65 } | 57 } |
66 | 58 |
67 TEST(RtcpPacketTmmbrTest, CreateAndParseWithTwoEntries) { | 59 TEST(RtcpPacketTmmbrTest, CreateAndParseWithTwoEntries) { |
68 Tmmbr tmmbr; | 60 Tmmbr tmmbr; |
69 tmmbr.From(kSenderSsrc); | 61 tmmbr.From(kSenderSsrc); |
70 tmmbr.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); | 62 tmmbr.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); |
71 tmmbr.WithTmmbr(TmmbItem(kRemoteSsrc + 1, 4 * kBitrateBps, kOverhead + 1)); | 63 tmmbr.WithTmmbr(TmmbItem(kRemoteSsrc + 1, 4 * kBitrateBps, kOverhead + 1)); |
72 | 64 |
73 rtc::Buffer packet = tmmbr.Build(); | 65 rtc::Buffer packet = tmmbr.Build(); |
74 | 66 |
75 Tmmbr parsed; | 67 Tmmbr parsed; |
76 EXPECT_TRUE(ParseTmmbr(packet.data(), packet.size(), &parsed)); | 68 EXPECT_TRUE(test::ParseSinglePacket(packet, &parsed)); |
77 | 69 |
78 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); | 70 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); |
79 EXPECT_EQ(2u, parsed.requests().size()); | 71 EXPECT_EQ(2u, parsed.requests().size()); |
80 EXPECT_EQ(kRemoteSsrc, parsed.requests()[0].ssrc()); | 72 EXPECT_EQ(kRemoteSsrc, parsed.requests()[0].ssrc()); |
81 EXPECT_EQ(kRemoteSsrc + 1, parsed.requests()[1].ssrc()); | 73 EXPECT_EQ(kRemoteSsrc + 1, parsed.requests()[1].ssrc()); |
82 } | 74 } |
83 | 75 |
84 TEST(RtcpPacketTmmbrTest, ParseFailsWithoutItems) { | 76 TEST(RtcpPacketTmmbrTest, ParseFailsWithoutItems) { |
85 const uint8_t kZeroItemsPacket[] = {0x83, 205, 0x00, 0x02, | 77 const uint8_t kZeroItemsPacket[] = {0x83, 205, 0x00, 0x02, |
86 0x12, 0x34, 0x56, 0x78, | 78 0x12, 0x34, 0x56, 0x78, |
87 0x00, 0x00, 0x00, 0x00}; | 79 0x00, 0x00, 0x00, 0x00}; |
88 | 80 |
89 Tmmbr tmmbr; | 81 Tmmbr tmmbr; |
90 EXPECT_FALSE(ParseTmmbr(kZeroItemsPacket, sizeof(kZeroItemsPacket), &tmmbr)); | 82 EXPECT_FALSE(test::ParseSinglePacket(kZeroItemsPacket, &tmmbr)); |
91 } | 83 } |
92 | 84 |
93 TEST(RtcpPacketTmmbrTest, ParseFailsOnUnAlignedPacket) { | 85 TEST(RtcpPacketTmmbrTest, ParseFailsOnUnAlignedPacket) { |
94 const uint8_t kUnalignedPacket[] = {0x83, 205, 0x00, 0x05, | 86 const uint8_t kUnalignedPacket[] = {0x83, 205, 0x00, 0x05, |
95 0x12, 0x34, 0x56, 0x78, | 87 0x12, 0x34, 0x56, 0x78, |
96 0x00, 0x00, 0x00, 0x00, | 88 0x00, 0x00, 0x00, 0x00, |
97 0x23, 0x45, 0x67, 0x89, | 89 0x23, 0x45, 0x67, 0x89, |
98 0x0a, 0x61, 0x61, 0xfe, | 90 0x0a, 0x61, 0x61, 0xfe, |
99 0x34, 0x56, 0x78, 0x9a}; | 91 0x34, 0x56, 0x78, 0x9a}; |
100 | 92 |
101 Tmmbr tmmbr; | 93 Tmmbr tmmbr; |
102 EXPECT_FALSE(ParseTmmbr(kUnalignedPacket, sizeof(kUnalignedPacket), &tmmbr)); | 94 EXPECT_FALSE(test::ParseSinglePacket(kUnalignedPacket, &tmmbr)); |
103 } | 95 } |
104 } // namespace webrtc | 96 } // namespace webrtc |
OLD | NEW |