OLD | NEW |
---|---|
(Empty) | |
1 /* | |
danilchap
2016/08/22 13:00:41
make unittest filename match source file it is tes
brandtr
2016/08/23 08:19:10
This file tests the Ulpfec stuff too though. I cou
| |
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 <string.h> | |
12 | |
13 #include <memory> | |
14 #include <utility> | |
15 | |
16 #include "testing/gtest/include/gtest/gtest.h" | |
17 #include "webrtc/base/checks.h" | |
18 #include "webrtc/base/random.h" | |
19 #include "webrtc/modules/rtp_rtcp/source/fec_test_helper.h" | |
20 #include "webrtc/modules/rtp_rtcp/source/flexfec_header_reader_writer.h" | |
21 #include "webrtc/modules/rtp_rtcp/source/forward_error_correction.h" | |
22 #include "webrtc/modules/rtp_rtcp/source/ulpfec_header_reader_writer.h" | |
23 | |
24 namespace webrtc { | |
25 | |
26 namespace { | |
27 // Transport header size in bytes. Assume UDP/IPv4 as a reasonable minimum. | |
28 constexpr size_t kTransportOverhead = 28; | |
29 | |
30 // ULPFEC header: largest packet mask. | |
31 constexpr size_t kUlpfecMaxHeaderSize = 18; | |
32 | |
33 // FlexFEC header: single protected SSRC with largest packet mask. | |
34 constexpr size_t kFlexfecMaxHeaderSize = 32; | |
35 } // namespace | |
36 | |
37 class FecPacketizationTest : public ::testing::Test { | |
38 protected: | |
39 FecPacketizationTest(std::unique_ptr<FecHeaderReader> fec_header_reader, | |
40 std::unique_ptr<FecHeaderWriter> fec_header_writer, | |
41 size_t min_packet_size, | |
42 size_t max_packet_size) | |
43 : fec_header_reader_(std::move(fec_header_reader)), | |
44 fec_header_writer_(std::move(fec_header_writer)), | |
45 random_(0xabcdef123456), | |
46 media_packet_generator_(min_packet_size, | |
47 max_packet_size, | |
48 &random_), | |
49 packet_mask_() {} | |
50 | |
51 // Set the corresponding bits of the ULPFEC packet mask. | |
52 void Bit15Clear(); | |
53 void Bit15Set(); | |
54 void Bit46ClearBit47Clear(); | |
55 void Bit46SetBit47Clear(); | |
56 void Bit46ClearBit47Set(); | |
57 void Bit46SetBit47Set(); | |
58 | |
59 // Randomly generate ULPFEC packet mask of specified size. | |
60 void GenerateUlpfecPacketMask(size_t packet_mask_size); | |
61 | |
62 void ReadWriteAndVerifyHeaders(size_t ulpfec_packet_mask_size, | |
63 size_t expected_fec_header_size, | |
64 size_t expected_packet_mask_size); | |
65 | |
66 virtual void MaybeSetFecSsrcToMediaSsrc( | |
67 uint32_t media_ssrc, | |
68 ForwardErrorCorrection::ReceivedFecPacket* fec_packet) = 0; | |
69 | |
70 virtual size_t ExpectedPacketMaskOffset() = 0; | |
71 | |
72 std::unique_ptr<FecHeaderReader> fec_header_reader_; | |
73 std::unique_ptr<FecHeaderWriter> fec_header_writer_; | |
74 | |
75 Random random_; | |
76 test::fec::MediaPacketGenerator media_packet_generator_; | |
77 ForwardErrorCorrection::PacketList* media_packets_; | |
78 uint8_t packet_mask_[14]; // Only filled with ULPFEC packet mask (6 bytes), | |
79 // but needs to be 14 bytes in order to do | |
80 // memcmp with parsed FlexFEC mask. | |
81 }; | |
82 | |
83 class UlpfecPacketizationTest : public FecPacketizationTest { | |
84 protected: | |
85 UlpfecPacketizationTest() | |
86 : FecPacketizationTest( | |
87 std::unique_ptr<FecHeaderReader>(new UlpfecHeaderReader()), | |
88 std::unique_ptr<FecHeaderWriter>(new UlpfecHeaderWriter()), | |
89 kRtpHeaderSize, | |
90 IP_PACKET_SIZE - kRtpHeaderSize - kTransportOverhead - | |
91 kUlpfecMaxHeaderSize) {} | |
92 | |
93 void MaybeSetFecSsrcToMediaSsrc( | |
94 uint32_t media_ssrc, | |
95 ForwardErrorCorrection::ReceivedFecPacket* fec_packet) { | |
96 fec_packet->rtp_ssrc = media_ssrc; | |
97 } | |
98 | |
99 size_t ExpectedPacketMaskOffset() { return 12u; } | |
100 }; | |
101 | |
102 class FlexfecPacketizationTest : public FecPacketizationTest { | |
103 protected: | |
104 FlexfecPacketizationTest() | |
105 : FecPacketizationTest( | |
106 std::unique_ptr<FecHeaderReader>(new FlexfecHeaderReader()), | |
107 std::unique_ptr<FecHeaderWriter>(new FlexfecHeaderWriter()), | |
108 kRtpHeaderSize, | |
109 IP_PACKET_SIZE - kRtpHeaderSize - kTransportOverhead - | |
110 kFlexfecMaxHeaderSize) {} | |
111 | |
112 void MaybeSetFecSsrcToMediaSsrc( | |
113 uint32_t media_ssrc, | |
114 ForwardErrorCorrection::ReceivedFecPacket* fec_packet) {} | |
115 | |
116 size_t ExpectedPacketMaskOffset() { return 18u; } | |
117 }; | |
118 | |
119 // Verifies that the header size is correctly calculated. | |
120 TEST_F(UlpfecPacketizationTest, HeaderSizeTest) { | |
121 constexpr size_t kShortMask = kUlpfecPacketMaskSizeLBitClear; | |
122 GenerateUlpfecPacketMask(kShortMask); | |
123 EXPECT_EQ(14u, fec_header_writer_->FecHeaderSize( | |
124 fec_header_writer_->MinPacketMaskSize(packet_mask_, | |
125 kShortMask))); | |
126 | |
127 constexpr size_t kLongMask = kUlpfecPacketMaskSizeLBitSet; | |
128 GenerateUlpfecPacketMask(kLongMask); | |
129 EXPECT_EQ( | |
130 18u, fec_header_writer_->FecHeaderSize( | |
131 fec_header_writer_->MinPacketMaskSize(packet_mask_, kLongMask))); | |
132 } | |
133 | |
134 // Verifies that the reading and writing operations are consistent. | |
135 TEST_F(UlpfecPacketizationTest, ReaderWriterTest) { | |
136 SCOPED_TRACE(""); // To get call site line number, if failing. | |
137 | |
138 constexpr size_t kShortMask = kUlpfecPacketMaskSizeLBitClear; | |
139 GenerateUlpfecPacketMask(kShortMask); | |
140 ReadWriteAndVerifyHeaders(kShortMask, 14u, 2u); | |
141 | |
142 constexpr size_t kLongMask = kUlpfecPacketMaskSizeLBitSet; | |
143 GenerateUlpfecPacketMask(kLongMask); | |
144 ReadWriteAndVerifyHeaders(kLongMask, 18u, 6u); | |
145 } | |
146 | |
147 // Verifies that a correct FlexFEC packet mask size is | |
148 // calculated, given an ULPFEC packet mask size. | |
149 TEST_F(FlexfecPacketizationTest, HeaderSizeExpansionTest) { | |
150 constexpr size_t kShortMask = kUlpfecPacketMaskSizeLBitClear; | |
151 GenerateUlpfecPacketMask(kShortMask); | |
152 Bit15Clear(); | |
153 EXPECT_EQ(20u, fec_header_writer_->FecHeaderSize( | |
154 fec_header_writer_->MinPacketMaskSize(packet_mask_, | |
155 kShortMask))); | |
156 GenerateUlpfecPacketMask(kShortMask); | |
157 Bit15Set(); | |
158 EXPECT_EQ(24u, fec_header_writer_->FecHeaderSize( | |
159 fec_header_writer_->MinPacketMaskSize(packet_mask_, | |
160 kShortMask))); | |
161 | |
162 constexpr size_t kLongMask = kUlpfecPacketMaskSizeLBitSet; | |
163 GenerateUlpfecPacketMask(kLongMask); | |
164 Bit46ClearBit47Clear(); | |
165 EXPECT_EQ( | |
166 24u, fec_header_writer_->FecHeaderSize( | |
167 fec_header_writer_->MinPacketMaskSize(packet_mask_, kLongMask))); | |
168 GenerateUlpfecPacketMask(kLongMask); | |
169 Bit46SetBit47Clear(); | |
170 EXPECT_EQ( | |
171 32u, fec_header_writer_->FecHeaderSize( | |
172 fec_header_writer_->MinPacketMaskSize(packet_mask_, kLongMask))); | |
173 GenerateUlpfecPacketMask(kLongMask); | |
174 Bit46ClearBit47Set(); | |
175 EXPECT_EQ( | |
176 32u, fec_header_writer_->FecHeaderSize( | |
177 fec_header_writer_->MinPacketMaskSize(packet_mask_, kLongMask))); | |
178 GenerateUlpfecPacketMask(kLongMask); | |
179 Bit46SetBit47Set(); | |
180 EXPECT_EQ( | |
181 32u, fec_header_writer_->FecHeaderSize( | |
182 fec_header_writer_->MinPacketMaskSize(packet_mask_, kLongMask))); | |
183 } | |
184 | |
185 // Verifies that the reading and writing operations are consistent. | |
186 TEST_F(FlexfecPacketizationTest, ReaderWriterTest) { | |
187 SCOPED_TRACE(""); // To get call site line number, if failing. | |
188 | |
189 constexpr size_t kShortMask = kUlpfecPacketMaskSizeLBitClear; | |
190 GenerateUlpfecPacketMask(kShortMask); | |
191 Bit15Clear(); | |
192 ReadWriteAndVerifyHeaders(kShortMask, 20u, 2u); | |
193 Bit15Set(); | |
194 ReadWriteAndVerifyHeaders(kShortMask, 24u, 6u); | |
195 | |
196 constexpr size_t kLongMask = kUlpfecPacketMaskSizeLBitSet; | |
197 GenerateUlpfecPacketMask(kLongMask); | |
198 Bit46ClearBit47Clear(); | |
199 ReadWriteAndVerifyHeaders(kLongMask, 24u, 6u); | |
200 Bit46SetBit47Clear(); | |
201 ReadWriteAndVerifyHeaders(kLongMask, 32u, 14u); | |
202 Bit46ClearBit47Set(); | |
203 ReadWriteAndVerifyHeaders(kLongMask, 32u, 14u); | |
204 Bit46SetBit47Set(); | |
205 ReadWriteAndVerifyHeaders(kLongMask, 32u, 14u); | |
206 } | |
207 | |
208 void FecPacketizationTest::Bit15Clear() { | |
209 packet_mask_[1] &= 0xfe; | |
210 } | |
211 | |
212 void FecPacketizationTest::Bit15Set() { | |
213 packet_mask_[1] |= 0x01; | |
214 } | |
215 | |
216 void FecPacketizationTest::Bit46ClearBit47Clear() { | |
217 packet_mask_[5] &= 0xfc; | |
218 } | |
219 | |
220 void FecPacketizationTest::Bit46SetBit47Clear() { | |
221 Bit46ClearBit47Clear(); | |
222 packet_mask_[5] ^= 0x02; | |
223 } | |
224 | |
225 void FecPacketizationTest::Bit46ClearBit47Set() { | |
226 Bit46ClearBit47Clear(); | |
227 packet_mask_[5] ^= 0x01; | |
228 } | |
229 | |
230 void FecPacketizationTest::Bit46SetBit47Set() { | |
231 packet_mask_[5] |= 0x03; | |
232 } | |
233 | |
234 void FecPacketizationTest::GenerateUlpfecPacketMask(size_t packet_mask_size) { | |
235 RTC_DCHECK_LE(packet_mask_size, 6u); | |
236 memset(packet_mask_, 0u, 14); | |
237 for (size_t i = 0; i < packet_mask_size; ++i) { | |
238 packet_mask_[i] = random_.Rand<uint8_t>(); | |
239 } | |
240 } | |
241 | |
242 void FecPacketizationTest::ReadWriteAndVerifyHeaders( | |
243 size_t ulpfec_packet_mask_size, | |
244 size_t expected_fec_header_size, | |
245 size_t expected_packet_mask_size) { | |
246 // Generate media. | |
247 constexpr int kNumMediaPackets = 1; | |
248 media_packet_generator_.ConstructMediaPackets(kNumMediaPackets); | |
249 ForwardErrorCorrection::PacketList* media_packets = | |
250 media_packet_generator_.GetMediaPackets(); | |
251 uint32_t media_ssrc = media_packet_generator_.GetSsrc(); | |
252 uint16_t media_seq_num = | |
253 media_packet_generator_.GetFecSeqNum() - kNumMediaPackets; | |
254 | |
255 // Set up FEC packet containers. | |
256 ForwardErrorCorrection::Packet* sent_fec_packet = | |
257 new ForwardErrorCorrection::Packet(); // deleted by scoped_refptr in | |
258 // |received_fec_packet| | |
259 sent_fec_packet->length = media_packets->front()->length; | |
260 for (size_t i = 0; i < sent_fec_packet->length; ++i) { | |
261 sent_fec_packet->data[i] = random_.Rand<uint8_t>(); | |
262 } | |
263 ForwardErrorCorrection::ReceivedFecPacket received_fec_packet; | |
264 received_fec_packet.pkt = sent_fec_packet; | |
265 | |
266 // Write FEC header. | |
267 fec_header_writer_->FinalizeFecHeader( | |
268 *media_packets, packet_mask_, ulpfec_packet_mask_size, sent_fec_packet); | |
269 | |
270 // Read FEC header. | |
271 MaybeSetFecSsrcToMediaSsrc(media_ssrc, &received_fec_packet); | |
272 EXPECT_TRUE(fec_header_reader_->ReadFecHeader(&received_fec_packet)); | |
273 | |
274 // Verify that written and read FEC headers are consistent. | |
275 EXPECT_EQ(expected_fec_header_size, received_fec_packet.fec_header_size); | |
276 EXPECT_EQ(received_fec_packet.pkt->length - expected_fec_header_size, | |
277 received_fec_packet.protection_length); | |
278 auto packet_mask_info = received_fec_packet.packet_mask_info[media_ssrc]; | |
279 uint16_t seq_num_base = std::get<0>(packet_mask_info); | |
280 EXPECT_EQ(media_seq_num, seq_num_base); | |
281 size_t packet_mask_offset = std::get<1>(packet_mask_info); | |
282 EXPECT_EQ(ExpectedPacketMaskOffset(), packet_mask_offset); | |
283 size_t packet_mask_size = std::get<2>(packet_mask_info); | |
284 ASSERT_EQ(expected_packet_mask_size, packet_mask_size); | |
285 EXPECT_EQ(0, memcmp(packet_mask_, | |
286 &received_fec_packet.pkt->data[packet_mask_offset], | |
287 packet_mask_size)); | |
288 EXPECT_EQ(0, memcmp(&sent_fec_packet->data[expected_fec_header_size], | |
289 &received_fec_packet.pkt->data[expected_fec_header_size], | |
290 sent_fec_packet->length - expected_fec_header_size)); | |
291 } | |
292 | |
293 } // namespace webrtc | |
OLD | NEW |