OLD | NEW |
---|---|
(Empty) | |
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 <string.h> | |
12 | |
13 #include <memory> | |
14 #include <utility> | |
15 | |
16 #include "testing/gtest/include/gtest/gtest.h" | |
17 #include "webrtc/base/basictypes.h" | |
18 #include "webrtc/base/checks.h" | |
19 #include "webrtc/base/random.h" | |
20 #include "webrtc/base/scoped_ref_ptr.h" | |
21 #include "webrtc/modules/rtp_rtcp/source/fec_test_helper.h" | |
22 #include "webrtc/modules/rtp_rtcp/source/forward_error_correction.h" | |
23 #include "webrtc/modules/rtp_rtcp/source/forward_error_correction_internal.h" | |
24 #include "webrtc/modules/rtp_rtcp/source/ulpfec_header_reader_writer.h" | |
25 | |
26 namespace webrtc { | |
27 | |
28 namespace { | |
29 // Transport header size in bytes. Assume UDP/IPv4 as a reasonable minimum. | |
30 constexpr size_t kTransportOverhead = 28; | |
31 | |
32 constexpr size_t kUlpfecHeaderSizeLBitClear = 14; | |
33 constexpr size_t kUlpfecHeaderSizeLBitSet = 18; | |
34 constexpr size_t kUlpfecPacketMaskOffset = 12; | |
35 // These are pulled in from forward_error_correction_internal.h | |
danilchap
2016/08/25 17:49:15
Might be better to remove those constants or forwa
brandtr
2016/08/26 11:22:22
Removed them.
| |
36 // constexpr size_t kUlpfecPacketMaskSizeLBitClear = 2; | |
37 // constexpr size_t kUlpfecPacketMaskSizeLBitSet = 6; | |
38 | |
39 constexpr size_t kFlexfecPacketMaskSize2K = 14; | |
danilchap
2016/08/25 17:49:15
what does 2K means?
is it kFlexfexMaxPacketMaskSiz
brandtr
2016/08/26 11:22:22
Yep. Added and renamed these in the next CL.
| |
40 | |
41 constexpr uint32_t kMediaSsrc = 1254983; | |
42 } // namespace | |
43 | |
44 using Packet = ForwardErrorCorrection::Packet; | |
danilchap
2016/08/25 17:49:15
prefer full names:
using Packet = ::webrtc::Forwar
brandtr
2016/08/26 11:22:22
Done.
| |
45 using PacketList = ForwardErrorCorrection::PacketList; | |
46 using ReceivedFecPacket = ForwardErrorCorrection::ReceivedFecPacket; | |
47 | |
48 class FecPacketizationTest : public ::testing::Test { | |
49 protected: | |
50 FecPacketizationTest(std::unique_ptr<FecHeaderReader> fec_header_reader, | |
51 std::unique_ptr<FecHeaderWriter> fec_header_writer, | |
52 size_t min_packet_size, | |
53 size_t max_packet_size) | |
54 : fec_header_reader_(std::move(fec_header_reader)), | |
55 fec_header_writer_(std::move(fec_header_writer)), | |
56 random_(0xabcdef123456), | |
57 media_packet_generator_(min_packet_size, | |
58 max_packet_size, | |
59 kMediaSsrc, | |
60 &random_) {} | |
61 | |
62 // Set the corresponding bits of the ULPFEC formatted packet mask. | |
63 void ClearBit(size_t index); | |
64 void SetBit(size_t index); | |
65 | |
66 // Randomly generate ULPFEC formatted packet mask of specified size. | |
67 void GenerateUlpfecFormattedPacketMask(size_t packet_mask_size); | |
68 | |
69 void ReadWriteAndVerifyHeaders(size_t ulpfec_formatted_packet_mask_size, | |
70 size_t expected_fec_header_size, | |
71 size_t expected_packet_mask_size); | |
72 | |
73 virtual void MaybeSetFecSsrcToMediaSsrc(uint32_t media_ssrc, | |
74 ReceivedFecPacket* fec_packet) = 0; | |
75 | |
76 virtual size_t ExpectedPacketMaskOffset() = 0; | |
77 | |
78 std::unique_ptr<FecHeaderReader> fec_header_reader_; | |
79 std::unique_ptr<FecHeaderWriter> fec_header_writer_; | |
80 | |
81 Random random_; | |
82 test::fec::MediaPacketGenerator media_packet_generator_; | |
83 PacketList* media_packets_; | |
84 uint8_t packet_mask_[kFlexfecPacketMaskSize2K]; // Only filled with ULPFEC | |
85 // packet mask (max 6 | |
86 // bytes), but needs to be | |
87 // 14 bytes in order to do | |
88 // memcmp with parsed | |
89 // FlexFEC mask. | |
90 }; | |
91 | |
92 class UlpfecPacketizationTest : public FecPacketizationTest { | |
danilchap
2016/08/25 17:49:15
any reason why this is in same file instead of own
brandtr
2016/08/26 11:22:22
Move and renamed.
| |
93 protected: | |
94 UlpfecPacketizationTest() | |
95 : FecPacketizationTest( | |
96 std::unique_ptr<FecHeaderReader>(new UlpfecHeaderReader()), | |
97 std::unique_ptr<FecHeaderWriter>(new UlpfecHeaderWriter()), | |
98 kRtpHeaderSize, | |
99 IP_PACKET_SIZE - kRtpHeaderSize - kTransportOverhead - | |
100 kUlpfecHeaderSizeLBitSet) {} | |
101 | |
102 void MaybeSetFecSsrcToMediaSsrc(uint32_t media_ssrc, | |
103 ReceivedFecPacket* fec_packet) override; | |
104 | |
105 size_t ExpectedPacketMaskOffset() override; | |
106 }; | |
107 | |
108 void FecPacketizationTest::ClearBit(size_t index) { | |
109 packet_mask_[index / 8] &= ~(1 << (7 - index % 8)); | |
110 } | |
111 | |
112 void FecPacketizationTest::SetBit(size_t index) { | |
113 packet_mask_[index / 8] |= (1 << (7 - index % 8)); | |
114 } | |
115 | |
116 void FecPacketizationTest::GenerateUlpfecFormattedPacketMask( | |
117 size_t packet_mask_size) { | |
118 RTC_DCHECK_LE(packet_mask_size, 6u); | |
danilchap
2016/08/25 17:49:15
kUlpfecMaxPacketMaskSize instead of 6
brandtr
2016/08/26 11:22:22
Good catch, thanks.
| |
119 memset(packet_mask_, 0u, 14); | |
danilchap
2016/08/25 17:49:15
kFlexfecPacketMaskSize2K instead of 14
brandtr
2016/08/26 11:22:22
Done.
| |
120 for (size_t i = 0; i < packet_mask_size; ++i) { | |
121 packet_mask_[i] = random_.Rand<uint8_t>(); | |
122 } | |
123 } | |
124 | |
125 void FecPacketizationTest::ReadWriteAndVerifyHeaders( | |
126 size_t ulpfec_formatted_packet_mask_size, | |
127 size_t expected_fec_header_size, | |
128 size_t expected_packet_mask_size) { | |
129 // Generate media. | |
130 constexpr int kNumMediaPackets = 1; | |
131 PacketList media_packets = | |
132 media_packet_generator_.ConstructMediaPackets(kNumMediaPackets); | |
133 uint16_t media_seq_num = | |
134 media_packet_generator_.GetFecSeqNum() - kNumMediaPackets; | |
danilchap
2016/08/25 17:49:15
it is not obvious why it is correct.
may be instea
brandtr
2016/08/26 11:22:22
Done. (See next patchset.)
| |
135 | |
136 // Set up FEC packet containers. | |
137 ReceivedFecPacket received_fec_packet; | |
138 received_fec_packet.pkt = rtc::scoped_refptr<Packet>(new Packet()); | |
139 Packet* sent_fec_packet = received_fec_packet.pkt.get(); | |
140 sent_fec_packet->length = media_packets.front()->length; | |
141 for (size_t i = 0; i < sent_fec_packet->length; ++i) { | |
142 sent_fec_packet->data[i] = random_.Rand<uint8_t>(); | |
143 } | |
144 | |
145 // Write FEC header. | |
146 fec_header_writer_->FinalizeFecHeader(media_packets, packet_mask_, | |
147 ulpfec_formatted_packet_mask_size, | |
148 sent_fec_packet); | |
149 | |
150 // Read FEC header. | |
151 MaybeSetFecSsrcToMediaSsrc(kMediaSsrc, &received_fec_packet); | |
152 EXPECT_TRUE(fec_header_reader_->ReadFecHeader(&received_fec_packet)); | |
153 | |
154 // Verify that written and read FEC headers are consistent. | |
155 EXPECT_EQ(expected_fec_header_size, received_fec_packet.fec_header_size); | |
156 EXPECT_EQ(received_fec_packet.pkt->length - expected_fec_header_size, | |
157 received_fec_packet.protection_length); | |
158 auto stream_packet_mask_info = | |
159 received_fec_packet.packet_mask_infos[kMediaSsrc]; | |
160 EXPECT_EQ(media_seq_num, stream_packet_mask_info.seq_num_base); | |
161 size_t offset = stream_packet_mask_info.offset; | |
162 EXPECT_EQ(ExpectedPacketMaskOffset(), offset); | |
163 size_t size = stream_packet_mask_info.size; | |
164 ASSERT_EQ(expected_packet_mask_size, size); | |
165 EXPECT_EQ(0, | |
166 memcmp(packet_mask_, &received_fec_packet.pkt->data[offset], size)); | |
167 EXPECT_EQ(0, memcmp(&sent_fec_packet->data[expected_fec_header_size], | |
168 &received_fec_packet.pkt->data[expected_fec_header_size], | |
169 sent_fec_packet->length - expected_fec_header_size)); | |
170 } | |
171 | |
172 void UlpfecPacketizationTest::MaybeSetFecSsrcToMediaSsrc( | |
173 uint32_t media_ssrc, | |
174 ReceivedFecPacket* fec_packet) { | |
175 fec_packet->rtp_ssrc = media_ssrc; | |
176 } | |
177 | |
178 size_t UlpfecPacketizationTest::ExpectedPacketMaskOffset() { | |
179 return kUlpfecPacketMaskOffset; | |
180 } | |
181 | |
182 TEST_F(UlpfecPacketizationTest, CalculateSmallHeaderSize) { | |
183 constexpr size_t kShortMask = kUlpfecPacketMaskSizeLBitClear; | |
184 GenerateUlpfecFormattedPacketMask(kShortMask); | |
185 EXPECT_EQ( | |
186 kUlpfecHeaderSizeLBitClear, | |
187 fec_header_writer_->FecHeaderSize( | |
188 fec_header_writer_->MinPacketMaskSize(packet_mask_, kShortMask))); | |
189 } | |
190 | |
191 TEST_F(UlpfecPacketizationTest, CalculateLargeHeaderSize) { | |
192 constexpr size_t kLongMask = kUlpfecPacketMaskSizeLBitSet; | |
193 GenerateUlpfecFormattedPacketMask(kLongMask); | |
194 EXPECT_EQ(kUlpfecHeaderSizeLBitSet, fec_header_writer_->FecHeaderSize( | |
195 fec_header_writer_->MinPacketMaskSize( | |
196 packet_mask_, kLongMask))); | |
197 } | |
198 | |
199 TEST_F(UlpfecPacketizationTest, WriteAndReadHeaderSmall) { | |
200 constexpr size_t kShortMask = kUlpfecPacketMaskSizeLBitClear; | |
201 GenerateUlpfecFormattedPacketMask(kShortMask); | |
202 ReadWriteAndVerifyHeaders(kShortMask, kUlpfecHeaderSizeLBitClear, | |
203 kUlpfecPacketMaskSizeLBitClear); | |
204 } | |
205 | |
206 TEST_F(UlpfecPacketizationTest, WriteAndReadHeaderLarge) { | |
207 constexpr size_t kLongMask = kUlpfecPacketMaskSizeLBitSet; | |
208 GenerateUlpfecFormattedPacketMask(kLongMask); | |
209 ReadWriteAndVerifyHeaders(kLongMask, kUlpfecHeaderSizeLBitSet, | |
210 kUlpfecPacketMaskSizeLBitSet); | |
211 } | |
212 | |
213 } // namespace webrtc | |
OLD | NEW |