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/tmmbn.h" | 11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/tmmbn.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::Tmmbn; | 21 using webrtc::rtcp::Tmmbn; |
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[] = {0x84, 205, 0x00, 0x04, | 29 const uint8_t kPacket[] = {0x84, 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 ParseTmmbn(const uint8_t* buffer, size_t length, Tmmbn* tmmbn) { | |
37 RtcpCommonHeader header; | |
38 EXPECT_TRUE(RtcpParseCommonHeader(buffer, length, &header)); | |
39 EXPECT_EQ(length, header.BlockSize()); | |
40 return tmmbn->Parse(header, buffer + RtcpCommonHeader::kHeaderSizeBytes); | |
41 } | |
42 } // namespace | 34 } // namespace |
43 | 35 |
44 TEST(RtcpPacketTmmbnTest, Create) { | 36 TEST(RtcpPacketTmmbnTest, Create) { |
45 Tmmbn tmmbn; | 37 Tmmbn tmmbn; |
46 tmmbn.From(kSenderSsrc); | 38 tmmbn.From(kSenderSsrc); |
47 tmmbn.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); | 39 tmmbn.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); |
48 | 40 |
49 rtc::Buffer packet = tmmbn.Build(); | 41 rtc::Buffer packet = tmmbn.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(RtcpPacketTmmbnTest, Parse) { | 47 TEST(RtcpPacketTmmbnTest, Parse) { |
56 Tmmbn tmmbn; | 48 Tmmbn tmmbn; |
57 EXPECT_TRUE(ParseTmmbn(kPacket, sizeof(kPacket), &tmmbn)); | 49 EXPECT_TRUE(test::ParseSinglePacket(kPacket, &tmmbn)); |
58 | 50 |
59 const Tmmbn& parsed = tmmbn; | 51 const Tmmbn& parsed = tmmbn; |
60 | 52 |
61 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); | 53 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); |
62 ASSERT_EQ(1u, parsed.items().size()); | 54 ASSERT_EQ(1u, parsed.items().size()); |
63 EXPECT_EQ(kRemoteSsrc, parsed.items().front().ssrc()); | 55 EXPECT_EQ(kRemoteSsrc, parsed.items().front().ssrc()); |
64 EXPECT_EQ(kBitrateBps, parsed.items().front().bitrate_bps()); | 56 EXPECT_EQ(kBitrateBps, parsed.items().front().bitrate_bps()); |
65 EXPECT_EQ(kOverhead, parsed.items().front().packet_overhead()); | 57 EXPECT_EQ(kOverhead, parsed.items().front().packet_overhead()); |
66 } | 58 } |
67 | 59 |
68 TEST(RtcpPacketTmmbnTest, CreateAndParseWithoutItems) { | 60 TEST(RtcpPacketTmmbnTest, CreateAndParseWithoutItems) { |
69 Tmmbn tmmbn; | 61 Tmmbn tmmbn; |
70 tmmbn.From(kSenderSsrc); | 62 tmmbn.From(kSenderSsrc); |
71 | 63 |
72 rtc::Buffer packet = tmmbn.Build(); | 64 rtc::Buffer packet = tmmbn.Build(); |
73 Tmmbn parsed; | 65 Tmmbn parsed; |
74 EXPECT_TRUE(ParseTmmbn(packet.data(), packet.size(), &parsed)); | 66 EXPECT_TRUE(test::ParseSinglePacket(packet, &parsed)); |
75 | 67 |
76 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); | 68 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); |
77 EXPECT_THAT(parsed.items(), IsEmpty()); | 69 EXPECT_THAT(parsed.items(), IsEmpty()); |
78 } | 70 } |
79 | 71 |
80 TEST(RtcpPacketTmmbnTest, CreateAndParseWithTwoItems) { | 72 TEST(RtcpPacketTmmbnTest, CreateAndParseWithTwoItems) { |
81 Tmmbn tmmbn; | 73 Tmmbn tmmbn; |
82 tmmbn.From(kSenderSsrc); | 74 tmmbn.From(kSenderSsrc); |
83 tmmbn.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); | 75 tmmbn.WithTmmbr(TmmbItem(kRemoteSsrc, kBitrateBps, kOverhead)); |
84 tmmbn.WithTmmbr(TmmbItem(kRemoteSsrc + 1, 4 * kBitrateBps, 40)); | 76 tmmbn.WithTmmbr(TmmbItem(kRemoteSsrc + 1, 4 * kBitrateBps, 40)); |
85 | 77 |
86 rtc::Buffer packet = tmmbn.Build(); | 78 rtc::Buffer packet = tmmbn.Build(); |
87 Tmmbn parsed; | 79 Tmmbn parsed; |
88 EXPECT_TRUE(ParseTmmbn(packet.data(), packet.size(), &parsed)); | 80 EXPECT_TRUE(test::ParseSinglePacket(packet, &parsed)); |
89 | 81 |
90 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); | 82 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc()); |
91 EXPECT_EQ(2u, parsed.items().size()); | 83 EXPECT_EQ(2u, parsed.items().size()); |
92 EXPECT_EQ(kRemoteSsrc, parsed.items()[0].ssrc()); | 84 EXPECT_EQ(kRemoteSsrc, parsed.items()[0].ssrc()); |
93 EXPECT_EQ(kBitrateBps, parsed.items()[0].bitrate_bps()); | 85 EXPECT_EQ(kBitrateBps, parsed.items()[0].bitrate_bps()); |
94 EXPECT_EQ(kOverhead, parsed.items()[0].packet_overhead()); | 86 EXPECT_EQ(kOverhead, parsed.items()[0].packet_overhead()); |
95 EXPECT_EQ(kRemoteSsrc + 1, parsed.items()[1].ssrc()); | 87 EXPECT_EQ(kRemoteSsrc + 1, parsed.items()[1].ssrc()); |
96 EXPECT_EQ(4 * kBitrateBps, parsed.items()[1].bitrate_bps()); | 88 EXPECT_EQ(4 * kBitrateBps, parsed.items()[1].bitrate_bps()); |
97 EXPECT_EQ(40U, parsed.items()[1].packet_overhead()); | 89 EXPECT_EQ(40U, parsed.items()[1].packet_overhead()); |
98 } | 90 } |
99 | 91 |
100 TEST(RtcpPacketTmmbnTest, ParseFailsOnTooSmallPacket) { | 92 TEST(RtcpPacketTmmbnTest, ParseFailsOnTooSmallPacket) { |
101 const uint8_t kSmallPacket[] = {0x84, 205, 0x00, 0x01, | 93 const uint8_t kSmallPacket[] = {0x84, 205, 0x00, 0x01, |
102 0x12, 0x34, 0x56, 0x78}; | 94 0x12, 0x34, 0x56, 0x78}; |
103 Tmmbn tmmbn; | 95 Tmmbn tmmbn; |
104 EXPECT_FALSE(ParseTmmbn(kSmallPacket, sizeof(kSmallPacket), &tmmbn)); | 96 EXPECT_FALSE(test::ParseSinglePacket(kSmallPacket, &tmmbn)); |
105 } | 97 } |
106 | 98 |
107 TEST(RtcpPacketTmmbnTest, ParseFailsOnUnAlignedPacket) { | 99 TEST(RtcpPacketTmmbnTest, ParseFailsOnUnAlignedPacket) { |
108 const uint8_t kUnalignedPacket[] = {0x84, 205, 0x00, 0x03, | 100 const uint8_t kUnalignedPacket[] = {0x84, 205, 0x00, 0x03, |
109 0x12, 0x34, 0x56, 0x78, | 101 0x12, 0x34, 0x56, 0x78, |
110 0x00, 0x00, 0x00, 0x00, | 102 0x00, 0x00, 0x00, 0x00, |
111 0x23, 0x45, 0x67, 0x89}; | 103 0x23, 0x45, 0x67, 0x89}; |
112 | 104 |
113 Tmmbn tmmbn; | 105 Tmmbn tmmbn; |
114 EXPECT_FALSE(ParseTmmbn(kUnalignedPacket, sizeof(kUnalignedPacket), &tmmbn)); | 106 EXPECT_FALSE(test::ParseSinglePacket(kUnalignedPacket, &tmmbn)); |
115 } | 107 } |
116 } // namespace webrtc | 108 } // namespace webrtc |
OLD | NEW |