Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 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 | #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" |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 18 | #include "webrtc/base/buffer.h" |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 19 | #include "webrtc/base/checks.h" |
| 20 | #include "webrtc/base/scoped_ptr.h" |
| 21 | #include "webrtc/call.h" |
Peter Boström | 5c389d3 | 2015-09-25 13:58:30 +0200 | [diff] [blame] | 22 | #include "webrtc/call/rtc_event_log.h" |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 23 | #include "webrtc/modules/rtp_rtcp/source/rtp_sender.h" |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 24 | #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 Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 28 | |
| 29 | // Files generated at build-time by the protobuf compiler. |
| 30 | #ifdef WEBRTC_ANDROID_PLATFORM_BUILD |
Peter Boström | 5c389d3 | 2015-09-25 13:58:30 +0200 | [diff] [blame] | 31 | #include "external/webrtc/webrtc/call/rtc_event_log.pb.h" |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 32 | #else |
Peter Boström | 5c389d3 | 2015-09-25 13:58:30 +0200 | [diff] [blame] | 33 | #include "webrtc/call/rtc_event_log.pb.h" |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 34 | #endif |
| 35 | |
| 36 | namespace webrtc { |
| 37 | |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 38 | namespace { |
| 39 | |
| 40 | const RTPExtensionType kExtensionTypes[] = { |
| 41 | RTPExtensionType::kRtpExtensionTransmissionTimeOffset, |
| 42 | RTPExtensionType::kRtpExtensionAudioLevel, |
| 43 | RTPExtensionType::kRtpExtensionAbsoluteSendTime, |
| 44 | RTPExtensionType::kRtpExtensionVideoRotation, |
| 45 | RTPExtensionType::kRtpExtensionTransportSequenceNumber}; |
| 46 | const char* kExtensionNames[] = {RtpExtension::kTOffset, |
| 47 | RtpExtension::kAudioLevel, |
| 48 | RtpExtension::kAbsSendTime, |
| 49 | RtpExtension::kVideoRotation, |
| 50 | RtpExtension::kTransportSequenceNumber}; |
| 51 | const size_t kNumExtensions = 5; |
| 52 | |
Peter Boström | 5c389d3 | 2015-09-25 13:58:30 +0200 | [diff] [blame] | 53 | } // namespace |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 54 | |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 55 | // TODO(terelius): Place this definition with other parsing functions? |
| 56 | MediaType 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 Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 87 | if ((type == rtclog::Event::AUDIO_PLAYOUT_EVENT) != |
| 88 | event.has_audio_playout_event()) |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 89 | return ::testing::AssertionFailure() |
| 90 | << "Event of type " << type << " has " |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 91 | << (event.has_audio_playout_event() ? "" : "no ") |
| 92 | << "audio_playout event"; |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 93 | 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 | |
| 121 | void 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()); |
pbos | da903ea | 2015-10-02 02:36:56 -0700 | [diff] [blame] | 134 | if (config.rtp.rtcp_mode == RtcpMode::kCompound) |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 135 | 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 | |
| 184 | void 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 | |
| 229 | void 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 | |
| 251 | void 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 Creusen | ae856f2 | 2015-09-17 16:30:16 +0200 | [diff] [blame] | 270 | void VerifyPlayoutEvent(const rtclog::Event& event, uint32_t ssrc) { |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 271 | ASSERT_TRUE(IsValidBasicEvent(event)); |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 272 | 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()); |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 276 | } |
| 277 | |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 278 | void VerifyLogStartEvent(const rtclog::Event& event) { |
| 279 | ASSERT_TRUE(IsValidBasicEvent(event)); |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 280 | EXPECT_EQ(rtclog::Event::LOG_START, event.type()); |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 281 | } |
| 282 | |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 283 | /* |
| 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 | */ |
| 288 | size_t GenerateRtpPacket(uint32_t extensions_bitvector, |
| 289 | uint32_t csrcs_count, |
| 290 | uint8_t* packet, |
| 291 | size_t packet_size) { |
henrikg | 91d6ede | 2015-09-17 00:24:34 -0700 | [diff] [blame] | 292 | RTC_CHECK_GE(packet_size, 16 + 4 * csrcs_count + 4 * kNumExtensions); |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 293 | Clock* clock = Clock::GetRealTimeClock(); |
| 294 | |
Peter Boström | ac547a6 | 2015-09-17 23:03:57 +0200 | [diff] [blame] | 295 | RTPSender rtp_sender(false, // bool audio |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 296 | 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 | |
| 339 | void 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 | |
| 345 | void GenerateVideoReceiveConfig(uint32_t extensions_bitvector, |
| 346 | VideoReceiveStream::Config* config) { |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 347 | // 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. |
pbos | da903ea | 2015-10-02 02:36:56 -0700 | [diff] [blame] | 356 | config->rtp.rtcp_mode = |
| 357 | rand() % 2 ? RtcpMode::kCompound : RtcpMode::kReducedSize; |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 358 | config->rtp.rtcp_xr.receiver_reference_time_report = (rand() % 2 == 1); |
| 359 | config->rtp.remb = (rand() % 2 == 1); |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 360 | // 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)); |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 365 | // 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 Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 372 | } |
| 373 | |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 374 | void GenerateVideoSendConfig(uint32_t extensions_bitvector, |
| 375 | VideoSendStream::Config* config) { |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 376 | // 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"; |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 386 | // 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 Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 393 | } |
| 394 | |
| 395 | // Test for the RtcEventLog class. Dumps some RTP packets to disk, then reads |
| 396 | // them back to see if they match. |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 397 | void LogSessionAndReadBack(size_t rtp_count, |
| 398 | size_t rtcp_count, |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 399 | size_t playout_count, |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 400 | uint32_t extensions_bitvector, |
| 401 | uint32_t csrcs_count, |
| 402 | unsigned random_seed) { |
| 403 | ASSERT_LE(rtcp_count, rtp_count); |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 404 | ASSERT_LE(playout_count, rtp_count); |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 405 | std::vector<rtc::Buffer> rtp_packets; |
| 406 | std::vector<rtc::Buffer> rtcp_packets; |
| 407 | std::vector<size_t> rtp_header_sizes; |
Ivo Creusen | ae856f2 | 2015-09-17 16:30:16 +0200 | [diff] [blame] | 408 | std::vector<uint32_t> playout_ssrcs; |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 409 | |
solenberg | 4fbae2b | 2015-08-28 04:07:10 -0700 | [diff] [blame] | 410 | VideoReceiveStream::Config receiver_config(nullptr); |
| 411 | VideoSendStream::Config sender_config(nullptr); |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 412 | |
| 413 | srand(random_seed); |
| 414 | |
| 415 | // Create rtp_count RTP packets containing random data. |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 416 | for (size_t i = 0; i < rtp_count; i++) { |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 417 | 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 Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 422 | } |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 423 | // 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 Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 428 | } |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 429 | // Create playout_count random SSRCs to use when logging AudioPlayout events. |
| 430 | for (size_t i = 0; i < playout_count; i++) { |
Ivo Creusen | ae856f2 | 2015-09-17 16:30:16 +0200 | [diff] [blame] | 431 | playout_ssrcs.push_back(static_cast<uint32_t>(rand())); |
| 432 | } |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 433 | // Create configurations for the video streams. |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 434 | GenerateVideoReceiveConfig(extensions_bitvector, &receiver_config); |
| 435 | GenerateVideoSendConfig(extensions_bitvector, &sender_config); |
| 436 | const int config_count = 2; |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 437 | |
| 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 Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 450 | size_t rtcp_index = 1, playout_index = 1; |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 451 | for (size_t i = 1; i <= rtp_count; i++) { |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 452 | log_dumper->LogRtpHeader( |
| 453 | (i % 2 == 0), // Every second packet is incoming. |
| 454 | (i % 3 == 0) ? MediaType::AUDIO : MediaType::VIDEO, |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 455 | 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 Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 464 | if (i * playout_count >= playout_index * rtp_count) { |
| 465 | log_dumper->LogAudioPlayout(playout_ssrcs[playout_index - 1]); |
| 466 | playout_index++; |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 467 | } |
| 468 | if (i == rtp_count / 2) { |
| 469 | log_dumper->StartLogging(temp_filename, 10000000); |
| 470 | } |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 471 | } |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 472 | } |
| 473 | |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 474 | // 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. |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 480 | const int event_count = |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 481 | config_count + playout_count + rtcp_count + rtp_count + 1; |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 482 | 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 Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 485 | size_t event_index = config_count, rtcp_index = 1, playout_index = 1; |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 486 | for (size_t i = 1; i <= rtp_count; i++) { |
| 487 | VerifyRtpEvent(parsed_stream.stream(event_index), |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 488 | (i % 2 == 0), // Every second packet is incoming. |
| 489 | (i % 3 == 0) ? MediaType::AUDIO : MediaType::VIDEO, |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 490 | 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 Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 502 | if (i * playout_count >= playout_index * rtp_count) { |
Ivo Creusen | ae856f2 | 2015-09-17 16:30:16 +0200 | [diff] [blame] | 503 | VerifyPlayoutEvent(parsed_stream.stream(event_index), |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 504 | playout_ssrcs[playout_index - 1]); |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 505 | event_index++; |
Ivo Creusen | 301aaed | 2015-10-08 18:07:41 +0200 | [diff] [blame^] | 506 | playout_index++; |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 507 | } |
| 508 | if (i == rtp_count / 2) { |
| 509 | VerifyLogStartEvent(parsed_stream.stream(event_index)); |
| 510 | event_index++; |
| 511 | } |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 512 | } |
Bjorn Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 513 | |
| 514 | // Clean up temporary file - can be pretty slow. |
| 515 | remove(temp_filename.c_str()); |
| 516 | } |
| 517 | |
| 518 | TEST(RtcEventLogTest, LogSessionAndReadBack) { |
terelius | 2f9fd5d | 2015-09-04 03:39:42 -0700 | [diff] [blame] | 519 | // 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 Terelius | 3641185 | 2015-07-30 12:45:18 +0200 | [diff] [blame] | 547 | } |
| 548 | |
| 549 | } // namespace webrtc |
| 550 | |
| 551 | #endif // ENABLE_RTC_EVENT_LOG |