1 /*
2 * Copyright (c) 2015 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/nack.h"
12
13 #include "test/gmock.h"
14 #include "test/gtest.h"
15 #include "test/rtcp_packet_parser.h"
16
17 namespace webrtc {
18 namespace {
19
20 using ::testing::_;
21 using ::testing::ElementsAre;
22 using ::testing::ElementsAreArray;
23 using ::testing::Invoke;
24 using ::testing::make_tuple;
25 using ::testing::MockFunction;
26 using ::testing::UnorderedElementsAreArray;
27 using ::webrtc::rtcp::Nack;
28
29 constexpr uint32_t kSenderSsrc = 0x12345678;
30 constexpr uint32_t kRemoteSsrc = 0x23456789;
31
32 constexpr uint16_t kList[] = {0, 1, 3, 8, 16};
33 constexpr size_t kListLength = sizeof(kList) / sizeof(kList[0]);
34 constexpr uint8_t kVersionBits = 2 << 6;
35 // clang-format off
36 constexpr uint8_t kPacket[] = {
37 kVersionBits | Nack::kFeedbackMessageType, Nack::kPacketType, 0, 3,
38 0x12, 0x34, 0x56, 0x78,
39 0x23, 0x45, 0x67, 0x89,
40 0x00, 0x00, 0x80, 0x85};
41
42 constexpr uint16_t kWrapList[] = {0xffdc, 0xffec, 0xfffe, 0xffff, 0x0000,
43 0x0001, 0x0003, 0x0014, 0x0064};
44 constexpr size_t kWrapListLength = sizeof(kWrapList) / sizeof(kWrapList[0]);
45 constexpr uint8_t kWrapPacket[] = {
46 kVersionBits | Nack::kFeedbackMessageType, Nack::kPacketType, 0, 6,
47 0x12, 0x34, 0x56, 0x78,
48 0x23, 0x45, 0x67, 0x89,
49 0xff, 0xdc, 0x80, 0x00,
50 0xff, 0xfe, 0x00, 0x17,
51 0x00, 0x14, 0x00, 0x00,
52 0x00, 0x64, 0x00, 0x00};
53 constexpr uint8_t kTooSmallPacket[] = {
54 kVersionBits | Nack::kFeedbackMessageType, Nack::kPacketType, 0, 2,
55 0x12, 0x34, 0x56, 0x78,
56 0x23, 0x45, 0x67, 0x89};
57 // clang-format on
58 } // namespace
59
TEST(RtcpPacketNackTest,Create)60 TEST(RtcpPacketNackTest, Create) {
61 Nack nack;
62 nack.SetSenderSsrc(kSenderSsrc);
63 nack.SetMediaSsrc(kRemoteSsrc);
64 nack.SetPacketIds(kList, kListLength);
65
66 rtc::Buffer packet = nack.Build();
67
68 EXPECT_THAT(make_tuple(packet.data(), packet.size()),
69 ElementsAreArray(kPacket));
70 }
71
TEST(RtcpPacketNackTest,Parse)72 TEST(RtcpPacketNackTest, Parse) {
73 Nack parsed;
74 EXPECT_TRUE(test::ParseSinglePacket(kPacket, &parsed));
75 const Nack& const_parsed = parsed;
76
77 EXPECT_EQ(kSenderSsrc, const_parsed.sender_ssrc());
78 EXPECT_EQ(kRemoteSsrc, const_parsed.media_ssrc());
79 EXPECT_THAT(const_parsed.packet_ids(), ElementsAreArray(kList));
80 }
81
TEST(RtcpPacketNackTest,CreateWrap)82 TEST(RtcpPacketNackTest, CreateWrap) {
83 Nack nack;
84 nack.SetSenderSsrc(kSenderSsrc);
85 nack.SetMediaSsrc(kRemoteSsrc);
86 nack.SetPacketIds(kWrapList, kWrapListLength);
87
88 rtc::Buffer packet = nack.Build();
89
90 EXPECT_THAT(make_tuple(packet.data(), packet.size()),
91 ElementsAreArray(kWrapPacket));
92 }
93
TEST(RtcpPacketNackTest,ParseWrap)94 TEST(RtcpPacketNackTest, ParseWrap) {
95 Nack parsed;
96 EXPECT_TRUE(test::ParseSinglePacket(kWrapPacket, &parsed));
97
98 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc());
99 EXPECT_EQ(kRemoteSsrc, parsed.media_ssrc());
100 EXPECT_THAT(parsed.packet_ids(), ElementsAreArray(kWrapList));
101 }
102
TEST(RtcpPacketNackTest,BadOrder)103 TEST(RtcpPacketNackTest, BadOrder) {
104 // Does not guarantee optimal packing, but should guarantee correctness.
105 const uint16_t kUnorderedList[] = {1, 25, 13, 12, 9, 27, 29};
106 const size_t kUnorderedListLength =
107 sizeof(kUnorderedList) / sizeof(kUnorderedList[0]);
108 Nack nack;
109 nack.SetSenderSsrc(kSenderSsrc);
110 nack.SetMediaSsrc(kRemoteSsrc);
111 nack.SetPacketIds(kUnorderedList, kUnorderedListLength);
112
113 rtc::Buffer packet = nack.Build();
114
115 Nack parsed;
116 EXPECT_TRUE(test::ParseSinglePacket(packet, &parsed));
117
118 EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc());
119 EXPECT_EQ(kRemoteSsrc, parsed.media_ssrc());
120 EXPECT_THAT(parsed.packet_ids(), UnorderedElementsAreArray(kUnorderedList));
121 }
122
TEST(RtcpPacketNackTest,CreateFragmented)123 TEST(RtcpPacketNackTest, CreateFragmented) {
124 Nack nack;
125 const uint16_t kList[] = {1, 100, 200, 300, 400};
126 const uint16_t kListLength = sizeof(kList) / sizeof(kList[0]);
127 nack.SetSenderSsrc(kSenderSsrc);
128 nack.SetMediaSsrc(kRemoteSsrc);
129 nack.SetPacketIds(kList, kListLength);
130
131 const size_t kBufferSize = 12 + (3 * 4); // Fits common header + 3 nack items
132
133 MockFunction<void(rtc::ArrayView<const uint8_t>)> callback;
134 EXPECT_CALL(callback, Call(_))
135 .WillOnce(Invoke([&](rtc::ArrayView<const uint8_t> packet) {
136 Nack nack;
137 EXPECT_TRUE(test::ParseSinglePacket(packet, &nack));
138 EXPECT_EQ(kSenderSsrc, nack.sender_ssrc());
139 EXPECT_EQ(kRemoteSsrc, nack.media_ssrc());
140 EXPECT_THAT(nack.packet_ids(), ElementsAre(1, 100, 200));
141 }))
142 .WillOnce(Invoke([&](rtc::ArrayView<const uint8_t> packet) {
143 Nack nack;
144 EXPECT_TRUE(test::ParseSinglePacket(packet, &nack));
145 EXPECT_EQ(kSenderSsrc, nack.sender_ssrc());
146 EXPECT_EQ(kRemoteSsrc, nack.media_ssrc());
147 EXPECT_THAT(nack.packet_ids(), ElementsAre(300, 400));
148 }));
149
150 EXPECT_TRUE(nack.Build(kBufferSize, callback.AsStdFunction()));
151 }
152
TEST(RtcpPacketNackTest,CreateFailsWithTooSmallBuffer)153 TEST(RtcpPacketNackTest, CreateFailsWithTooSmallBuffer) {
154 const uint16_t kList[] = {1};
155 const size_t kMinNackBlockSize = 16;
156 Nack nack;
157 nack.SetSenderSsrc(kSenderSsrc);
158 nack.SetMediaSsrc(kRemoteSsrc);
159 nack.SetPacketIds(kList, 1);
160
161 MockFunction<void(rtc::ArrayView<const uint8_t>)> callback;
162 EXPECT_CALL(callback, Call(_)).Times(0);
163 EXPECT_FALSE(nack.Build(kMinNackBlockSize - 1, callback.AsStdFunction()));
164 }
165
TEST(RtcpPacketNackTest,ParseFailsWithTooSmallBuffer)166 TEST(RtcpPacketNackTest, ParseFailsWithTooSmallBuffer) {
167 Nack parsed;
168 EXPECT_FALSE(test::ParseSinglePacket(kTooSmallPacket, &parsed));
169 }
170
171 } // namespace webrtc
172