blob: 536b9972276905becb81ec0cd353dc2b2b3be913 [file] [log] [blame]
Bjorn Terelius36411852015-07-30 12:45:18 +02001/*
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#ifdef ENABLE_RTC_EVENT_LOG
12
13#include <stdio.h>
14#include <string>
15#include <vector>
16
17#include "testing/gtest/include/gtest/gtest.h"
terelius2f9fd5d2015-09-04 03:39:42 -070018#include "webrtc/base/buffer.h"
Bjorn Terelius36411852015-07-30 12:45:18 +020019#include "webrtc/base/checks.h"
20#include "webrtc/base/scoped_ptr.h"
21#include "webrtc/call.h"
Peter Boström5c389d32015-09-25 13:58:30 +020022#include "webrtc/call/rtc_event_log.h"
terelius2f9fd5d2015-09-04 03:39:42 -070023#include "webrtc/modules/rtp_rtcp/source/rtp_sender.h"
Bjorn Terelius36411852015-07-30 12:45:18 +020024#include "webrtc/system_wrappers/interface/clock.h"
25#include "webrtc/test/test_suite.h"
26#include "webrtc/test/testsupport/fileutils.h"
27#include "webrtc/test/testsupport/gtest_disable.h"
Bjorn Terelius36411852015-07-30 12:45:18 +020028
29// Files generated at build-time by the protobuf compiler.
30#ifdef WEBRTC_ANDROID_PLATFORM_BUILD
Peter Boström5c389d32015-09-25 13:58:30 +020031#include "external/webrtc/webrtc/call/rtc_event_log.pb.h"
Bjorn Terelius36411852015-07-30 12:45:18 +020032#else
Peter Boström5c389d32015-09-25 13:58:30 +020033#include "webrtc/call/rtc_event_log.pb.h"
Bjorn Terelius36411852015-07-30 12:45:18 +020034#endif
35
36namespace webrtc {
37
terelius2f9fd5d2015-09-04 03:39:42 -070038namespace {
39
40const RTPExtensionType kExtensionTypes[] = {
41 RTPExtensionType::kRtpExtensionTransmissionTimeOffset,
42 RTPExtensionType::kRtpExtensionAudioLevel,
43 RTPExtensionType::kRtpExtensionAbsoluteSendTime,
44 RTPExtensionType::kRtpExtensionVideoRotation,
45 RTPExtensionType::kRtpExtensionTransportSequenceNumber};
46const char* kExtensionNames[] = {RtpExtension::kTOffset,
47 RtpExtension::kAudioLevel,
48 RtpExtension::kAbsSendTime,
49 RtpExtension::kVideoRotation,
50 RtpExtension::kTransportSequenceNumber};
51const size_t kNumExtensions = 5;
52
Peter Boström5c389d32015-09-25 13:58:30 +020053} // namespace
terelius2f9fd5d2015-09-04 03:39:42 -070054
Bjorn Terelius36411852015-07-30 12:45:18 +020055// TODO(terelius): Place this definition with other parsing functions?
56MediaType GetRuntimeMediaType(rtclog::MediaType media_type) {
57 switch (media_type) {
58 case rtclog::MediaType::ANY:
59 return MediaType::ANY;
60 case rtclog::MediaType::AUDIO:
61 return MediaType::AUDIO;
62 case rtclog::MediaType::VIDEO:
63 return MediaType::VIDEO;
64 case rtclog::MediaType::DATA:
65 return MediaType::DATA;
66 }
67 RTC_NOTREACHED();
68 return MediaType::ANY;
69}
70
71// Checks that the event has a timestamp, a type and exactly the data field
72// corresponding to the type.
73::testing::AssertionResult IsValidBasicEvent(const rtclog::Event& event) {
74 if (!event.has_timestamp_us())
75 return ::testing::AssertionFailure() << "Event has no timestamp";
76 if (!event.has_type())
77 return ::testing::AssertionFailure() << "Event has no event type";
78 rtclog::Event_EventType type = event.type();
79 if ((type == rtclog::Event::RTP_EVENT) != event.has_rtp_packet())
80 return ::testing::AssertionFailure()
81 << "Event of type " << type << " has "
82 << (event.has_rtp_packet() ? "" : "no ") << "RTP packet";
83 if ((type == rtclog::Event::RTCP_EVENT) != event.has_rtcp_packet())
84 return ::testing::AssertionFailure()
85 << "Event of type " << type << " has "
86 << (event.has_rtcp_packet() ? "" : "no ") << "RTCP packet";
Ivo Creusen301aaed2015-10-08 18:07:41 +020087 if ((type == rtclog::Event::AUDIO_PLAYOUT_EVENT) !=
88 event.has_audio_playout_event())
Bjorn Terelius36411852015-07-30 12:45:18 +020089 return ::testing::AssertionFailure()
90 << "Event of type " << type << " has "
Ivo Creusen301aaed2015-10-08 18:07:41 +020091 << (event.has_audio_playout_event() ? "" : "no ")
92 << "audio_playout event";
Bjorn Terelius36411852015-07-30 12:45:18 +020093 if ((type == rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT) !=
94 event.has_video_receiver_config())
95 return ::testing::AssertionFailure()
96 << "Event of type " << type << " has "
97 << (event.has_video_receiver_config() ? "" : "no ")
98 << "receiver config";
99 if ((type == rtclog::Event::VIDEO_SENDER_CONFIG_EVENT) !=
100 event.has_video_sender_config())
101 return ::testing::AssertionFailure()
102 << "Event of type " << type << " has "
103 << (event.has_video_sender_config() ? "" : "no ") << "sender config";
104 if ((type == rtclog::Event::AUDIO_RECEIVER_CONFIG_EVENT) !=
105 event.has_audio_receiver_config()) {
106 return ::testing::AssertionFailure()
107 << "Event of type " << type << " has "
108 << (event.has_audio_receiver_config() ? "" : "no ")
109 << "audio receiver config";
110 }
111 if ((type == rtclog::Event::AUDIO_SENDER_CONFIG_EVENT) !=
112 event.has_audio_sender_config()) {
113 return ::testing::AssertionFailure()
114 << "Event of type " << type << " has "
115 << (event.has_audio_sender_config() ? "" : "no ")
116 << "audio sender config";
117 }
118 return ::testing::AssertionSuccess();
119}
120
121void VerifyReceiveStreamConfig(const rtclog::Event& event,
122 const VideoReceiveStream::Config& config) {
123 ASSERT_TRUE(IsValidBasicEvent(event));
124 ASSERT_EQ(rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT, event.type());
125 const rtclog::VideoReceiveConfig& receiver_config =
126 event.video_receiver_config();
127 // Check SSRCs.
128 ASSERT_TRUE(receiver_config.has_remote_ssrc());
129 EXPECT_EQ(config.rtp.remote_ssrc, receiver_config.remote_ssrc());
130 ASSERT_TRUE(receiver_config.has_local_ssrc());
131 EXPECT_EQ(config.rtp.local_ssrc, receiver_config.local_ssrc());
132 // Check RTCP settings.
133 ASSERT_TRUE(receiver_config.has_rtcp_mode());
pbosda903ea2015-10-02 02:36:56 -0700134 if (config.rtp.rtcp_mode == RtcpMode::kCompound)
Bjorn Terelius36411852015-07-30 12:45:18 +0200135 EXPECT_EQ(rtclog::VideoReceiveConfig::RTCP_COMPOUND,
136 receiver_config.rtcp_mode());
137 else
138 EXPECT_EQ(rtclog::VideoReceiveConfig::RTCP_REDUCEDSIZE,
139 receiver_config.rtcp_mode());
140 ASSERT_TRUE(receiver_config.has_receiver_reference_time_report());
141 EXPECT_EQ(config.rtp.rtcp_xr.receiver_reference_time_report,
142 receiver_config.receiver_reference_time_report());
143 ASSERT_TRUE(receiver_config.has_remb());
144 EXPECT_EQ(config.rtp.remb, receiver_config.remb());
145 // Check RTX map.
146 ASSERT_EQ(static_cast<int>(config.rtp.rtx.size()),
147 receiver_config.rtx_map_size());
148 for (const rtclog::RtxMap& rtx_map : receiver_config.rtx_map()) {
149 ASSERT_TRUE(rtx_map.has_payload_type());
150 ASSERT_TRUE(rtx_map.has_config());
151 EXPECT_EQ(1u, config.rtp.rtx.count(rtx_map.payload_type()));
152 const rtclog::RtxConfig& rtx_config = rtx_map.config();
153 const VideoReceiveStream::Config::Rtp::Rtx& rtx =
154 config.rtp.rtx.at(rtx_map.payload_type());
155 ASSERT_TRUE(rtx_config.has_rtx_ssrc());
156 ASSERT_TRUE(rtx_config.has_rtx_payload_type());
157 EXPECT_EQ(rtx.ssrc, rtx_config.rtx_ssrc());
158 EXPECT_EQ(rtx.payload_type, rtx_config.rtx_payload_type());
159 }
160 // Check header extensions.
161 ASSERT_EQ(static_cast<int>(config.rtp.extensions.size()),
162 receiver_config.header_extensions_size());
163 for (int i = 0; i < receiver_config.header_extensions_size(); i++) {
164 ASSERT_TRUE(receiver_config.header_extensions(i).has_name());
165 ASSERT_TRUE(receiver_config.header_extensions(i).has_id());
166 const std::string& name = receiver_config.header_extensions(i).name();
167 int id = receiver_config.header_extensions(i).id();
168 EXPECT_EQ(config.rtp.extensions[i].id, id);
169 EXPECT_EQ(config.rtp.extensions[i].name, name);
170 }
171 // Check decoders.
172 ASSERT_EQ(static_cast<int>(config.decoders.size()),
173 receiver_config.decoders_size());
174 for (int i = 0; i < receiver_config.decoders_size(); i++) {
175 ASSERT_TRUE(receiver_config.decoders(i).has_name());
176 ASSERT_TRUE(receiver_config.decoders(i).has_payload_type());
177 const std::string& decoder_name = receiver_config.decoders(i).name();
178 int decoder_type = receiver_config.decoders(i).payload_type();
179 EXPECT_EQ(config.decoders[i].payload_name, decoder_name);
180 EXPECT_EQ(config.decoders[i].payload_type, decoder_type);
181 }
182}
183
184void VerifySendStreamConfig(const rtclog::Event& event,
185 const VideoSendStream::Config& config) {
186 ASSERT_TRUE(IsValidBasicEvent(event));
187 ASSERT_EQ(rtclog::Event::VIDEO_SENDER_CONFIG_EVENT, event.type());
188 const rtclog::VideoSendConfig& sender_config = event.video_sender_config();
189 // Check SSRCs.
190 ASSERT_EQ(static_cast<int>(config.rtp.ssrcs.size()),
191 sender_config.ssrcs_size());
192 for (int i = 0; i < sender_config.ssrcs_size(); i++) {
193 EXPECT_EQ(config.rtp.ssrcs[i], sender_config.ssrcs(i));
194 }
195 // Check header extensions.
196 ASSERT_EQ(static_cast<int>(config.rtp.extensions.size()),
197 sender_config.header_extensions_size());
198 for (int i = 0; i < sender_config.header_extensions_size(); i++) {
199 ASSERT_TRUE(sender_config.header_extensions(i).has_name());
200 ASSERT_TRUE(sender_config.header_extensions(i).has_id());
201 const std::string& name = sender_config.header_extensions(i).name();
202 int id = sender_config.header_extensions(i).id();
203 EXPECT_EQ(config.rtp.extensions[i].id, id);
204 EXPECT_EQ(config.rtp.extensions[i].name, name);
205 }
206 // Check RTX settings.
207 ASSERT_EQ(static_cast<int>(config.rtp.rtx.ssrcs.size()),
208 sender_config.rtx_ssrcs_size());
209 for (int i = 0; i < sender_config.rtx_ssrcs_size(); i++) {
210 EXPECT_EQ(config.rtp.rtx.ssrcs[i], sender_config.rtx_ssrcs(i));
211 }
212 if (sender_config.rtx_ssrcs_size() > 0) {
213 ASSERT_TRUE(sender_config.has_rtx_payload_type());
214 EXPECT_EQ(config.rtp.rtx.payload_type, sender_config.rtx_payload_type());
215 }
216 // Check CNAME.
217 ASSERT_TRUE(sender_config.has_c_name());
218 EXPECT_EQ(config.rtp.c_name, sender_config.c_name());
219 // Check encoder.
220 ASSERT_TRUE(sender_config.has_encoder());
221 ASSERT_TRUE(sender_config.encoder().has_name());
222 ASSERT_TRUE(sender_config.encoder().has_payload_type());
223 EXPECT_EQ(config.encoder_settings.payload_name,
224 sender_config.encoder().name());
225 EXPECT_EQ(config.encoder_settings.payload_type,
226 sender_config.encoder().payload_type());
227}
228
229void VerifyRtpEvent(const rtclog::Event& event,
230 bool incoming,
231 MediaType media_type,
232 uint8_t* header,
233 size_t header_size,
234 size_t total_size) {
235 ASSERT_TRUE(IsValidBasicEvent(event));
236 ASSERT_EQ(rtclog::Event::RTP_EVENT, event.type());
237 const rtclog::RtpPacket& rtp_packet = event.rtp_packet();
238 ASSERT_TRUE(rtp_packet.has_incoming());
239 EXPECT_EQ(incoming, rtp_packet.incoming());
240 ASSERT_TRUE(rtp_packet.has_type());
241 EXPECT_EQ(media_type, GetRuntimeMediaType(rtp_packet.type()));
242 ASSERT_TRUE(rtp_packet.has_packet_length());
243 EXPECT_EQ(total_size, rtp_packet.packet_length());
244 ASSERT_TRUE(rtp_packet.has_header());
245 ASSERT_EQ(header_size, rtp_packet.header().size());
246 for (size_t i = 0; i < header_size; i++) {
247 EXPECT_EQ(header[i], static_cast<uint8_t>(rtp_packet.header()[i]));
248 }
249}
250
251void VerifyRtcpEvent(const rtclog::Event& event,
252 bool incoming,
253 MediaType media_type,
254 uint8_t* packet,
255 size_t total_size) {
256 ASSERT_TRUE(IsValidBasicEvent(event));
257 ASSERT_EQ(rtclog::Event::RTCP_EVENT, event.type());
258 const rtclog::RtcpPacket& rtcp_packet = event.rtcp_packet();
259 ASSERT_TRUE(rtcp_packet.has_incoming());
260 EXPECT_EQ(incoming, rtcp_packet.incoming());
261 ASSERT_TRUE(rtcp_packet.has_type());
262 EXPECT_EQ(media_type, GetRuntimeMediaType(rtcp_packet.type()));
263 ASSERT_TRUE(rtcp_packet.has_packet_data());
264 ASSERT_EQ(total_size, rtcp_packet.packet_data().size());
265 for (size_t i = 0; i < total_size; i++) {
266 EXPECT_EQ(packet[i], static_cast<uint8_t>(rtcp_packet.packet_data()[i]));
267 }
268}
269
Ivo Creusenae856f22015-09-17 16:30:16 +0200270void VerifyPlayoutEvent(const rtclog::Event& event, uint32_t ssrc) {
terelius2f9fd5d2015-09-04 03:39:42 -0700271 ASSERT_TRUE(IsValidBasicEvent(event));
Ivo Creusen301aaed2015-10-08 18:07:41 +0200272 ASSERT_EQ(rtclog::Event::AUDIO_PLAYOUT_EVENT, event.type());
273 const rtclog::AudioPlayoutEvent& playout_event = event.audio_playout_event();
274 ASSERT_TRUE(playout_event.has_local_ssrc());
275 EXPECT_EQ(ssrc, playout_event.local_ssrc());
terelius2f9fd5d2015-09-04 03:39:42 -0700276}
277
Bjorn Terelius36411852015-07-30 12:45:18 +0200278void VerifyLogStartEvent(const rtclog::Event& event) {
279 ASSERT_TRUE(IsValidBasicEvent(event));
Ivo Creusen301aaed2015-10-08 18:07:41 +0200280 EXPECT_EQ(rtclog::Event::LOG_START, event.type());
Bjorn Terelius36411852015-07-30 12:45:18 +0200281}
282
terelius2f9fd5d2015-09-04 03:39:42 -0700283/*
284 * Bit number i of extension_bitvector is set to indicate the
285 * presence of extension number i from kExtensionTypes / kExtensionNames.
286 * The least significant bit extension_bitvector has number 0.
287 */
288size_t GenerateRtpPacket(uint32_t extensions_bitvector,
289 uint32_t csrcs_count,
290 uint8_t* packet,
291 size_t packet_size) {
henrikg91d6ede2015-09-17 00:24:34 -0700292 RTC_CHECK_GE(packet_size, 16 + 4 * csrcs_count + 4 * kNumExtensions);
terelius2f9fd5d2015-09-04 03:39:42 -0700293 Clock* clock = Clock::GetRealTimeClock();
294
Peter Boströmac547a62015-09-17 23:03:57 +0200295 RTPSender rtp_sender(false, // bool audio
terelius2f9fd5d2015-09-04 03:39:42 -0700296 clock, // Clock* clock
297 nullptr, // Transport*
298 nullptr, // RtpAudioFeedback*
299 nullptr, // PacedSender*
300 nullptr, // PacketRouter*
301 nullptr, // SendTimeObserver*
302 nullptr, // BitrateStatisticsObserver*
303 nullptr, // FrameCountObserver*
304 nullptr); // SendSideDelayObserver*
305
306 std::vector<uint32_t> csrcs;
307 for (unsigned i = 0; i < csrcs_count; i++) {
308 csrcs.push_back(rand());
309 }
310 rtp_sender.SetCsrcs(csrcs);
311 rtp_sender.SetSSRC(rand());
312 rtp_sender.SetStartTimestamp(rand(), true);
313 rtp_sender.SetSequenceNumber(rand());
314
315 for (unsigned i = 0; i < kNumExtensions; i++) {
316 if (extensions_bitvector & (1u << i)) {
317 rtp_sender.RegisterRtpHeaderExtension(kExtensionTypes[i], i + 1);
318 }
319 }
320
321 int8_t payload_type = rand() % 128;
322 bool marker_bit = (rand() % 2 == 1);
323 uint32_t capture_timestamp = rand();
324 int64_t capture_time_ms = rand();
325 bool timestamp_provided = (rand() % 2 == 1);
326 bool inc_sequence_number = (rand() % 2 == 1);
327
328 size_t header_size = rtp_sender.BuildRTPheader(
329 packet, payload_type, marker_bit, capture_timestamp, capture_time_ms,
330 timestamp_provided, inc_sequence_number);
331
332 for (size_t i = header_size; i < packet_size; i++) {
333 packet[i] = rand();
334 }
335
336 return header_size;
337}
338
339void GenerateRtcpPacket(uint8_t* packet, size_t packet_size) {
340 for (size_t i = 0; i < packet_size; i++) {
341 packet[i] = rand();
342 }
343}
344
345void GenerateVideoReceiveConfig(uint32_t extensions_bitvector,
346 VideoReceiveStream::Config* config) {
Bjorn Terelius36411852015-07-30 12:45:18 +0200347 // Create a map from a payload type to an encoder name.
348 VideoReceiveStream::Decoder decoder;
349 decoder.payload_type = rand();
350 decoder.payload_name = (rand() % 2 ? "VP8" : "H264");
351 config->decoders.push_back(decoder);
352 // Add SSRCs for the stream.
353 config->rtp.remote_ssrc = rand();
354 config->rtp.local_ssrc = rand();
355 // Add extensions and settings for RTCP.
pbosda903ea2015-10-02 02:36:56 -0700356 config->rtp.rtcp_mode =
357 rand() % 2 ? RtcpMode::kCompound : RtcpMode::kReducedSize;
terelius2f9fd5d2015-09-04 03:39:42 -0700358 config->rtp.rtcp_xr.receiver_reference_time_report = (rand() % 2 == 1);
359 config->rtp.remb = (rand() % 2 == 1);
Bjorn Terelius36411852015-07-30 12:45:18 +0200360 // Add a map from a payload type to a new ssrc and a new payload type for RTX.
361 VideoReceiveStream::Config::Rtp::Rtx rtx_pair;
362 rtx_pair.ssrc = rand();
363 rtx_pair.payload_type = rand();
364 config->rtp.rtx.insert(std::make_pair(rand(), rtx_pair));
terelius2f9fd5d2015-09-04 03:39:42 -0700365 // Add header extensions.
366 for (unsigned i = 0; i < kNumExtensions; i++) {
367 if (extensions_bitvector & (1u << i)) {
368 config->rtp.extensions.push_back(
369 RtpExtension(kExtensionNames[i], rand()));
370 }
371 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200372}
373
terelius2f9fd5d2015-09-04 03:39:42 -0700374void GenerateVideoSendConfig(uint32_t extensions_bitvector,
375 VideoSendStream::Config* config) {
Bjorn Terelius36411852015-07-30 12:45:18 +0200376 // Create a map from a payload type to an encoder name.
377 config->encoder_settings.payload_type = rand();
378 config->encoder_settings.payload_name = (rand() % 2 ? "VP8" : "H264");
379 // Add SSRCs for the stream.
380 config->rtp.ssrcs.push_back(rand());
381 // Add a map from a payload type to new ssrcs and a new payload type for RTX.
382 config->rtp.rtx.ssrcs.push_back(rand());
383 config->rtp.rtx.payload_type = rand();
384 // Add a CNAME.
385 config->rtp.c_name = "some.user@some.host";
terelius2f9fd5d2015-09-04 03:39:42 -0700386 // Add header extensions.
387 for (unsigned i = 0; i < kNumExtensions; i++) {
388 if (extensions_bitvector & (1u << i)) {
389 config->rtp.extensions.push_back(
390 RtpExtension(kExtensionNames[i], rand()));
391 }
392 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200393}
394
395// Test for the RtcEventLog class. Dumps some RTP packets to disk, then reads
396// them back to see if they match.
terelius2f9fd5d2015-09-04 03:39:42 -0700397void LogSessionAndReadBack(size_t rtp_count,
398 size_t rtcp_count,
Ivo Creusen301aaed2015-10-08 18:07:41 +0200399 size_t playout_count,
terelius2f9fd5d2015-09-04 03:39:42 -0700400 uint32_t extensions_bitvector,
401 uint32_t csrcs_count,
402 unsigned random_seed) {
403 ASSERT_LE(rtcp_count, rtp_count);
Ivo Creusen301aaed2015-10-08 18:07:41 +0200404 ASSERT_LE(playout_count, rtp_count);
terelius2f9fd5d2015-09-04 03:39:42 -0700405 std::vector<rtc::Buffer> rtp_packets;
406 std::vector<rtc::Buffer> rtcp_packets;
407 std::vector<size_t> rtp_header_sizes;
Ivo Creusenae856f22015-09-17 16:30:16 +0200408 std::vector<uint32_t> playout_ssrcs;
Bjorn Terelius36411852015-07-30 12:45:18 +0200409
solenberg4fbae2b2015-08-28 04:07:10 -0700410 VideoReceiveStream::Config receiver_config(nullptr);
411 VideoSendStream::Config sender_config(nullptr);
Bjorn Terelius36411852015-07-30 12:45:18 +0200412
413 srand(random_seed);
414
415 // Create rtp_count RTP packets containing random data.
Bjorn Terelius36411852015-07-30 12:45:18 +0200416 for (size_t i = 0; i < rtp_count; i++) {
terelius2f9fd5d2015-09-04 03:39:42 -0700417 size_t packet_size = 1000 + rand() % 64;
418 rtp_packets.push_back(rtc::Buffer(packet_size));
419 size_t header_size = GenerateRtpPacket(extensions_bitvector, csrcs_count,
420 rtp_packets[i].data(), packet_size);
421 rtp_header_sizes.push_back(header_size);
Bjorn Terelius36411852015-07-30 12:45:18 +0200422 }
terelius2f9fd5d2015-09-04 03:39:42 -0700423 // Create rtcp_count RTCP packets containing random data.
424 for (size_t i = 0; i < rtcp_count; i++) {
425 size_t packet_size = 1000 + rand() % 64;
426 rtcp_packets.push_back(rtc::Buffer(packet_size));
427 GenerateRtcpPacket(rtcp_packets[i].data(), packet_size);
Bjorn Terelius36411852015-07-30 12:45:18 +0200428 }
Ivo Creusen301aaed2015-10-08 18:07:41 +0200429 // Create playout_count random SSRCs to use when logging AudioPlayout events.
430 for (size_t i = 0; i < playout_count; i++) {
Ivo Creusenae856f22015-09-17 16:30:16 +0200431 playout_ssrcs.push_back(static_cast<uint32_t>(rand()));
432 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200433 // Create configurations for the video streams.
terelius2f9fd5d2015-09-04 03:39:42 -0700434 GenerateVideoReceiveConfig(extensions_bitvector, &receiver_config);
435 GenerateVideoSendConfig(extensions_bitvector, &sender_config);
436 const int config_count = 2;
Bjorn Terelius36411852015-07-30 12:45:18 +0200437
438 // Find the name of the current test, in order to use it as a temporary
439 // filename.
440 auto test_info = ::testing::UnitTest::GetInstance()->current_test_info();
441 const std::string temp_filename =
442 test::OutputPath() + test_info->test_case_name() + test_info->name();
443
444 // When log_dumper goes out of scope, it causes the log file to be flushed
445 // to disk.
446 {
447 rtc::scoped_ptr<RtcEventLog> log_dumper(RtcEventLog::Create());
448 log_dumper->LogVideoReceiveStreamConfig(receiver_config);
449 log_dumper->LogVideoSendStreamConfig(sender_config);
Ivo Creusen301aaed2015-10-08 18:07:41 +0200450 size_t rtcp_index = 1, playout_index = 1;
terelius2f9fd5d2015-09-04 03:39:42 -0700451 for (size_t i = 1; i <= rtp_count; i++) {
Bjorn Terelius36411852015-07-30 12:45:18 +0200452 log_dumper->LogRtpHeader(
453 (i % 2 == 0), // Every second packet is incoming.
454 (i % 3 == 0) ? MediaType::AUDIO : MediaType::VIDEO,
terelius2f9fd5d2015-09-04 03:39:42 -0700455 rtp_packets[i - 1].data(), rtp_packets[i - 1].size());
456 if (i * rtcp_count >= rtcp_index * rtp_count) {
457 log_dumper->LogRtcpPacket(
458 rtcp_index % 2 == 0, // Every second packet is incoming
459 rtcp_index % 3 == 0 ? MediaType::AUDIO : MediaType::VIDEO,
460 rtcp_packets[rtcp_index - 1].data(),
461 rtcp_packets[rtcp_index - 1].size());
462 rtcp_index++;
463 }
Ivo Creusen301aaed2015-10-08 18:07:41 +0200464 if (i * playout_count >= playout_index * rtp_count) {
465 log_dumper->LogAudioPlayout(playout_ssrcs[playout_index - 1]);
466 playout_index++;
terelius2f9fd5d2015-09-04 03:39:42 -0700467 }
468 if (i == rtp_count / 2) {
469 log_dumper->StartLogging(temp_filename, 10000000);
470 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200471 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200472 }
473
Bjorn Terelius36411852015-07-30 12:45:18 +0200474 // Read the generated file from disk.
475 rtclog::EventStream parsed_stream;
476
477 ASSERT_TRUE(RtcEventLog::ParseRtcEventLog(temp_filename, &parsed_stream));
478
479 // Verify the result.
terelius2f9fd5d2015-09-04 03:39:42 -0700480 const int event_count =
Ivo Creusen301aaed2015-10-08 18:07:41 +0200481 config_count + playout_count + rtcp_count + rtp_count + 1;
Bjorn Terelius36411852015-07-30 12:45:18 +0200482 EXPECT_EQ(event_count, parsed_stream.stream_size());
483 VerifyReceiveStreamConfig(parsed_stream.stream(0), receiver_config);
484 VerifySendStreamConfig(parsed_stream.stream(1), sender_config);
Ivo Creusen301aaed2015-10-08 18:07:41 +0200485 size_t event_index = config_count, rtcp_index = 1, playout_index = 1;
terelius2f9fd5d2015-09-04 03:39:42 -0700486 for (size_t i = 1; i <= rtp_count; i++) {
487 VerifyRtpEvent(parsed_stream.stream(event_index),
Bjorn Terelius36411852015-07-30 12:45:18 +0200488 (i % 2 == 0), // Every second packet is incoming.
489 (i % 3 == 0) ? MediaType::AUDIO : MediaType::VIDEO,
terelius2f9fd5d2015-09-04 03:39:42 -0700490 rtp_packets[i - 1].data(), rtp_header_sizes[i - 1],
491 rtp_packets[i - 1].size());
492 event_index++;
493 if (i * rtcp_count >= rtcp_index * rtp_count) {
494 VerifyRtcpEvent(parsed_stream.stream(event_index),
495 rtcp_index % 2 == 0, // Every second packet is incoming.
496 rtcp_index % 3 == 0 ? MediaType::AUDIO : MediaType::VIDEO,
497 rtcp_packets[rtcp_index - 1].data(),
498 rtcp_packets[rtcp_index - 1].size());
499 event_index++;
500 rtcp_index++;
501 }
Ivo Creusen301aaed2015-10-08 18:07:41 +0200502 if (i * playout_count >= playout_index * rtp_count) {
Ivo Creusenae856f22015-09-17 16:30:16 +0200503 VerifyPlayoutEvent(parsed_stream.stream(event_index),
Ivo Creusen301aaed2015-10-08 18:07:41 +0200504 playout_ssrcs[playout_index - 1]);
terelius2f9fd5d2015-09-04 03:39:42 -0700505 event_index++;
Ivo Creusen301aaed2015-10-08 18:07:41 +0200506 playout_index++;
terelius2f9fd5d2015-09-04 03:39:42 -0700507 }
508 if (i == rtp_count / 2) {
509 VerifyLogStartEvent(parsed_stream.stream(event_index));
510 event_index++;
511 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200512 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200513
514 // Clean up temporary file - can be pretty slow.
515 remove(temp_filename.c_str());
516}
517
518TEST(RtcEventLogTest, LogSessionAndReadBack) {
terelius2f9fd5d2015-09-04 03:39:42 -0700519 // Log 5 RTP, 2 RTCP, and 0 playout events with no header extensions or CSRCS.
520 LogSessionAndReadBack(5, 2, 0, 0, 0, 321);
521
522 // Enable AbsSendTime and TransportSequenceNumbers
523 uint32_t extensions = 0;
524 for (uint32_t i = 0; i < kNumExtensions; i++) {
525 if (kExtensionTypes[i] == RTPExtensionType::kRtpExtensionAbsoluteSendTime ||
526 kExtensionTypes[i] ==
527 RTPExtensionType::kRtpExtensionTransportSequenceNumber) {
528 extensions |= 1u << i;
529 }
530 }
531 LogSessionAndReadBack(8, 2, 0, extensions, 0, 3141592653u);
532
533 extensions = (1u << kNumExtensions) - 1; // Enable all header extensions
534 LogSessionAndReadBack(9, 2, 3, extensions, 2, 2718281828u);
535
536 // Try all combinations of header extensions and up to 2 CSRCS.
537 for (extensions = 0; extensions < (1u << kNumExtensions); extensions++) {
538 for (uint32_t csrcs_count = 0; csrcs_count < 3; csrcs_count++) {
539 LogSessionAndReadBack(5 + extensions, // Number of RTP packets.
540 2 + csrcs_count, // Number of RTCP packets.
541 3 + csrcs_count, // Number of playout events
542 extensions, // Bit vector choosing extensions
543 csrcs_count, // Number of contributing sources
544 rand());
545 }
546 }
Bjorn Terelius36411852015-07-30 12:45:18 +0200547}
548
549} // namespace webrtc
550
551#endif // ENABLE_RTC_EVENT_LOG