blob: ddd0e1a4d849831f2c43ee710fe276fb1f9e2703 [file] [log] [blame]
deadbeefe814a0d2017-02-25 18:15:09 -08001/*
2 * Copyright 2017 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 <memory>
12#include <utility> // For std::pair, std::move.
13
14#include "webrtc/api/ortc/ortcfactoryinterface.h"
deadbeefe814a0d2017-02-25 18:15:09 -080015#include "webrtc/ortc/testrtpparameters.h"
16#include "webrtc/p2p/base/udptransport.h"
17#include "webrtc/pc/test/fakeaudiocapturemodule.h"
18#include "webrtc/pc/test/fakeperiodicvideocapturer.h"
19#include "webrtc/pc/test/fakevideotrackrenderer.h"
Edward Lemurc20978e2017-07-06 19:44:34 +020020#include "webrtc/rtc_base/criticalsection.h"
21#include "webrtc/rtc_base/fakenetwork.h"
22#include "webrtc/rtc_base/gunit.h"
23#include "webrtc/rtc_base/virtualsocketserver.h"
deadbeefe814a0d2017-02-25 18:15:09 -080024
25namespace {
26
zhihuangd3501ad2017-03-03 14:39:06 -080027const int kDefaultTimeout = 10000; // 10 seconds.
28const int kReceivingDuration = 1000; // 1 second.
deadbeefe814a0d2017-02-25 18:15:09 -080029// Default number of audio/video frames to wait for before considering a test a
30// success.
31const int kDefaultNumFrames = 3;
32const rtc::IPAddress kIPv4LocalHostAddress =
33 rtc::IPAddress(0x7F000001); // 127.0.0.1
34
zhihuangd3501ad2017-03-03 14:39:06 -080035static const char kTestKeyParams1[] =
36 "inline:WVNfX19zZW1jdGwgKskgewkyMjA7fQp9CnVubGVz";
37static const char kTestKeyParams2[] =
38 "inline:PS1uQCVeeCFCanVmcjkpaywjNWhcYD0mXXtxaVBR";
39static const char kTestKeyParams3[] =
40 "inline:WVNfX19zZW1jdGwgKskgewkyMjA7fQp9CnVubGVa";
41static const char kTestKeyParams4[] =
42 "inline:WVNfX19zZW1jdGwgKskgewkyMjA7fQp9CnVubGVb";
43static const cricket::CryptoParams kTestCryptoParams1(1,
44 "AES_CM_128_HMAC_SHA1_80",
45 kTestKeyParams1,
46 "");
47static const cricket::CryptoParams kTestCryptoParams2(1,
48 "AES_CM_128_HMAC_SHA1_80",
49 kTestKeyParams2,
50 "");
51static const cricket::CryptoParams kTestCryptoParams3(1,
52 "AES_CM_128_HMAC_SHA1_80",
53 kTestKeyParams3,
54 "");
55static const cricket::CryptoParams kTestCryptoParams4(1,
56 "AES_CM_128_HMAC_SHA1_80",
57 kTestKeyParams4,
58 "");
deadbeefe814a0d2017-02-25 18:15:09 -080059} // namespace
60
61namespace webrtc {
62
63// Used to test that things work end-to-end when using the default
64// implementations of threads/etc. provided by OrtcFactory, with the exception
65// of using a virtual network.
66//
67// By default, the virtual network manager doesn't enumerate any networks, but
68// sockets can still be created in this state.
69class OrtcFactoryIntegrationTest : public testing::Test {
70 public:
71 OrtcFactoryIntegrationTest()
deadbeef98e186c2017-05-16 18:00:06 -070072 : network_thread_(&virtual_socket_server_),
deadbeefe814a0d2017-02-25 18:15:09 -080073 fake_audio_capture_module1_(FakeAudioCaptureModule::Create()),
74 fake_audio_capture_module2_(FakeAudioCaptureModule::Create()) {
75 // Sockets are bound to the ANY address, so this is needed to tell the
76 // virtual network which address to use in this case.
77 virtual_socket_server_.SetDefaultRoute(kIPv4LocalHostAddress);
78 network_thread_.Start();
79 // Need to create after network thread is started.
80 ortc_factory1_ = OrtcFactoryInterface::Create(
81 &network_thread_, nullptr, &fake_network_manager_,
82 nullptr, fake_audio_capture_module1_)
83 .MoveValue();
84 ortc_factory2_ = OrtcFactoryInterface::Create(
85 &network_thread_, nullptr, &fake_network_manager_,
86 nullptr, fake_audio_capture_module2_)
87 .MoveValue();
88 }
89
90 protected:
91 typedef std::pair<std::unique_ptr<UdpTransportInterface>,
92 std::unique_ptr<UdpTransportInterface>>
93 UdpTransportPair;
94 typedef std::pair<std::unique_ptr<RtpTransportInterface>,
95 std::unique_ptr<RtpTransportInterface>>
96 RtpTransportPair;
zhihuangd3501ad2017-03-03 14:39:06 -080097 typedef std::pair<std::unique_ptr<SrtpTransportInterface>,
98 std::unique_ptr<SrtpTransportInterface>>
99 SrtpTransportPair;
deadbeefe814a0d2017-02-25 18:15:09 -0800100 typedef std::pair<std::unique_ptr<RtpTransportControllerInterface>,
101 std::unique_ptr<RtpTransportControllerInterface>>
102 RtpTransportControllerPair;
103
104 // Helper function that creates one UDP transport each for |ortc_factory1_|
105 // and |ortc_factory2_|, and connects them.
106 UdpTransportPair CreateAndConnectUdpTransportPair() {
107 auto transport1 = ortc_factory1_->CreateUdpTransport(AF_INET).MoveValue();
108 auto transport2 = ortc_factory2_->CreateUdpTransport(AF_INET).MoveValue();
109 transport1->SetRemoteAddress(
110 rtc::SocketAddress(virtual_socket_server_.GetDefaultRoute(AF_INET),
111 transport2->GetLocalAddress().port()));
112 transport2->SetRemoteAddress(
113 rtc::SocketAddress(virtual_socket_server_.GetDefaultRoute(AF_INET),
114 transport1->GetLocalAddress().port()));
115 return {std::move(transport1), std::move(transport2)};
116 }
117
118 // Creates one transport controller each for |ortc_factory1_| and
119 // |ortc_factory2_|.
120 RtpTransportControllerPair CreateRtpTransportControllerPair() {
121 return {ortc_factory1_->CreateRtpTransportController().MoveValue(),
122 ortc_factory2_->CreateRtpTransportController().MoveValue()};
123 }
124
125 // Helper function that creates a pair of RtpTransports between
126 // |ortc_factory1_| and |ortc_factory2_|. Expected to be called with the
127 // result of CreateAndConnectUdpTransportPair. |rtcp_udp_transports| can be
128 // empty if RTCP muxing is used. |transport_controllers| can be empty if
129 // these transports are being created using a default transport controller.
130 RtpTransportPair CreateRtpTransportPair(
131 const RtcpParameters& rtcp_parameters,
132 const UdpTransportPair& rtp_udp_transports,
133 const UdpTransportPair& rtcp_udp_transports,
134 const RtpTransportControllerPair& transport_controllers) {
135 auto transport_result1 = ortc_factory1_->CreateRtpTransport(
136 rtcp_parameters, rtp_udp_transports.first.get(),
137 rtcp_udp_transports.first.get(), transport_controllers.first.get());
138 auto transport_result2 = ortc_factory2_->CreateRtpTransport(
139 rtcp_parameters, rtp_udp_transports.second.get(),
140 rtcp_udp_transports.second.get(), transport_controllers.second.get());
141 return {transport_result1.MoveValue(), transport_result2.MoveValue()};
142 }
143
zhihuangd3501ad2017-03-03 14:39:06 -0800144 SrtpTransportPair CreateSrtpTransportPair(
145 const RtcpParameters& rtcp_parameters,
146 const UdpTransportPair& rtp_udp_transports,
147 const UdpTransportPair& rtcp_udp_transports,
148 const RtpTransportControllerPair& transport_controllers) {
149 auto transport_result1 = ortc_factory1_->CreateSrtpTransport(
150 rtcp_parameters, rtp_udp_transports.first.get(),
151 rtcp_udp_transports.first.get(), transport_controllers.first.get());
152 auto transport_result2 = ortc_factory2_->CreateSrtpTransport(
153 rtcp_parameters, rtp_udp_transports.second.get(),
154 rtcp_udp_transports.second.get(), transport_controllers.second.get());
155 return {transport_result1.MoveValue(), transport_result2.MoveValue()};
156 }
157
deadbeefe814a0d2017-02-25 18:15:09 -0800158 // For convenience when |rtcp_udp_transports| and |transport_controllers|
159 // aren't needed.
160 RtpTransportPair CreateRtpTransportPair(
161 const RtcpParameters& rtcp_parameters,
162 const UdpTransportPair& rtp_udp_transports) {
163 return CreateRtpTransportPair(rtcp_parameters, rtp_udp_transports,
164 UdpTransportPair(),
165 RtpTransportControllerPair());
166 }
167
zhihuangd3501ad2017-03-03 14:39:06 -0800168 SrtpTransportPair CreateSrtpTransportPairAndSetKeys(
169 const RtcpParameters& rtcp_parameters,
170 const UdpTransportPair& rtp_udp_transports) {
171 SrtpTransportPair srtp_transports = CreateSrtpTransportPair(
172 rtcp_parameters, rtp_udp_transports, UdpTransportPair(),
173 RtpTransportControllerPair());
174 EXPECT_TRUE(srtp_transports.first->SetSrtpSendKey(kTestCryptoParams1).ok());
175 EXPECT_TRUE(
176 srtp_transports.first->SetSrtpReceiveKey(kTestCryptoParams2).ok());
177 EXPECT_TRUE(
178 srtp_transports.second->SetSrtpSendKey(kTestCryptoParams2).ok());
179 EXPECT_TRUE(
180 srtp_transports.second->SetSrtpReceiveKey(kTestCryptoParams1).ok());
181 return srtp_transports;
182 }
183
184 SrtpTransportPair CreateSrtpTransportPairAndSetMismatchingKeys(
185 const RtcpParameters& rtcp_parameters,
186 const UdpTransportPair& rtp_udp_transports) {
187 SrtpTransportPair srtp_transports = CreateSrtpTransportPair(
188 rtcp_parameters, rtp_udp_transports, UdpTransportPair(),
189 RtpTransportControllerPair());
190 EXPECT_TRUE(srtp_transports.first->SetSrtpSendKey(kTestCryptoParams1).ok());
191 EXPECT_TRUE(
192 srtp_transports.first->SetSrtpReceiveKey(kTestCryptoParams2).ok());
193 EXPECT_TRUE(
194 srtp_transports.second->SetSrtpSendKey(kTestCryptoParams1).ok());
195 EXPECT_TRUE(
196 srtp_transports.second->SetSrtpReceiveKey(kTestCryptoParams2).ok());
197 return srtp_transports;
198 }
199
deadbeefe814a0d2017-02-25 18:15:09 -0800200 // Ends up using fake audio capture module, which was passed into OrtcFactory
201 // on creation.
202 rtc::scoped_refptr<webrtc::AudioTrackInterface> CreateLocalAudioTrack(
203 const std::string& id,
204 OrtcFactoryInterface* ortc_factory) {
205 // Disable echo cancellation to make test more efficient.
206 cricket::AudioOptions options;
207 options.echo_cancellation.emplace(true);
208 rtc::scoped_refptr<webrtc::AudioSourceInterface> source =
209 ortc_factory->CreateAudioSource(options);
210 return ortc_factory->CreateAudioTrack(id, source);
211 }
212
213 // Stores created capturer in |fake_video_capturers_|.
214 rtc::scoped_refptr<webrtc::VideoTrackInterface>
215 CreateLocalVideoTrackAndFakeCapturer(const std::string& id,
216 OrtcFactoryInterface* ortc_factory) {
217 cricket::FakeVideoCapturer* fake_capturer =
218 new webrtc::FakePeriodicVideoCapturer();
219 fake_video_capturers_.push_back(fake_capturer);
220 rtc::scoped_refptr<webrtc::VideoTrackSourceInterface> source =
221 ortc_factory->CreateVideoSource(
222 std::unique_ptr<cricket::VideoCapturer>(fake_capturer));
223 return rtc::scoped_refptr<webrtc::VideoTrackInterface>(
224 ortc_factory->CreateVideoTrack(id, source));
225 }
226
zhihuangd3501ad2017-03-03 14:39:06 -0800227 // Helper function used to test two way RTP senders and receivers with basic
228 // configurations.
229 // If |expect_success| is true, waits for kDefaultTimeout for
230 // kDefaultNumFrames frames to be received by all RtpReceivers.
231 // If |expect_success| is false, simply waits for |kReceivingDuration|, and
232 // stores the number of received frames in |received_audio_frame1_| etc.
233 void BasicTwoWayRtpSendersAndReceiversTest(RtpTransportPair srtp_transports,
234 bool expect_success) {
235 received_audio_frames1_ = 0;
236 received_audio_frames2_ = 0;
237 rendered_video_frames1_ = 0;
238 rendered_video_frames2_ = 0;
239 // Create all the senders and receivers (four per endpoint).
240 auto audio_sender_result1 = ortc_factory1_->CreateRtpSender(
241 cricket::MEDIA_TYPE_AUDIO, srtp_transports.first.get());
242 auto video_sender_result1 = ortc_factory1_->CreateRtpSender(
243 cricket::MEDIA_TYPE_VIDEO, srtp_transports.first.get());
244 auto audio_receiver_result1 = ortc_factory1_->CreateRtpReceiver(
245 cricket::MEDIA_TYPE_AUDIO, srtp_transports.first.get());
246 auto video_receiver_result1 = ortc_factory1_->CreateRtpReceiver(
247 cricket::MEDIA_TYPE_VIDEO, srtp_transports.first.get());
248 ASSERT_TRUE(audio_sender_result1.ok());
249 ASSERT_TRUE(video_sender_result1.ok());
250 ASSERT_TRUE(audio_receiver_result1.ok());
251 ASSERT_TRUE(video_receiver_result1.ok());
252 auto audio_sender1 = audio_sender_result1.MoveValue();
253 auto video_sender1 = video_sender_result1.MoveValue();
254 auto audio_receiver1 = audio_receiver_result1.MoveValue();
255 auto video_receiver1 = video_receiver_result1.MoveValue();
256
257 auto audio_sender_result2 = ortc_factory2_->CreateRtpSender(
258 cricket::MEDIA_TYPE_AUDIO, srtp_transports.second.get());
259 auto video_sender_result2 = ortc_factory2_->CreateRtpSender(
260 cricket::MEDIA_TYPE_VIDEO, srtp_transports.second.get());
261 auto audio_receiver_result2 = ortc_factory2_->CreateRtpReceiver(
262 cricket::MEDIA_TYPE_AUDIO, srtp_transports.second.get());
263 auto video_receiver_result2 = ortc_factory2_->CreateRtpReceiver(
264 cricket::MEDIA_TYPE_VIDEO, srtp_transports.second.get());
265 ASSERT_TRUE(audio_sender_result2.ok());
266 ASSERT_TRUE(video_sender_result2.ok());
267 ASSERT_TRUE(audio_receiver_result2.ok());
268 ASSERT_TRUE(video_receiver_result2.ok());
269 auto audio_sender2 = audio_sender_result2.MoveValue();
270 auto video_sender2 = video_sender_result2.MoveValue();
271 auto audio_receiver2 = audio_receiver_result2.MoveValue();
272 auto video_receiver2 = video_receiver_result2.MoveValue();
273
274 // Add fake tracks.
275 RTCError error = audio_sender1->SetTrack(
276 CreateLocalAudioTrack("audio", ortc_factory1_.get()));
277 EXPECT_TRUE(error.ok());
278 error = video_sender1->SetTrack(
279 CreateLocalVideoTrackAndFakeCapturer("video", ortc_factory1_.get()));
280 EXPECT_TRUE(error.ok());
281 error = audio_sender2->SetTrack(
282 CreateLocalAudioTrack("audio", ortc_factory2_.get()));
283 EXPECT_TRUE(error.ok());
284 error = video_sender2->SetTrack(
285 CreateLocalVideoTrackAndFakeCapturer("video", ortc_factory2_.get()));
286 EXPECT_TRUE(error.ok());
287
288 // "sent_X_parameters1" are the parameters that endpoint 1 sends with and
289 // endpoint 2 receives with.
290 RtpParameters sent_opus_parameters1 =
291 MakeMinimalOpusParametersWithSsrc(0xdeadbeef);
292 RtpParameters sent_vp8_parameters1 =
293 MakeMinimalVp8ParametersWithSsrc(0xbaadfeed);
294 RtpParameters sent_opus_parameters2 =
295 MakeMinimalOpusParametersWithSsrc(0x13333337);
296 RtpParameters sent_vp8_parameters2 =
297 MakeMinimalVp8ParametersWithSsrc(0x12345678);
298
299 // Configure the senders' and receivers' parameters.
300 EXPECT_TRUE(audio_receiver1->Receive(sent_opus_parameters2).ok());
301 EXPECT_TRUE(video_receiver1->Receive(sent_vp8_parameters2).ok());
302 EXPECT_TRUE(audio_receiver2->Receive(sent_opus_parameters1).ok());
303 EXPECT_TRUE(video_receiver2->Receive(sent_vp8_parameters1).ok());
304 EXPECT_TRUE(audio_sender1->Send(sent_opus_parameters1).ok());
305 EXPECT_TRUE(video_sender1->Send(sent_vp8_parameters1).ok());
306 EXPECT_TRUE(audio_sender2->Send(sent_opus_parameters2).ok());
307 EXPECT_TRUE(video_sender2->Send(sent_vp8_parameters2).ok());
308
309 FakeVideoTrackRenderer fake_video_renderer1(
310 static_cast<VideoTrackInterface*>(video_receiver1->GetTrack().get()));
311 FakeVideoTrackRenderer fake_video_renderer2(
312 static_cast<VideoTrackInterface*>(video_receiver2->GetTrack().get()));
313
314 if (expect_success) {
315 EXPECT_TRUE_WAIT(
316 fake_audio_capture_module1_->frames_received() > kDefaultNumFrames &&
317 fake_video_renderer1.num_rendered_frames() > kDefaultNumFrames &&
318 fake_audio_capture_module2_->frames_received() >
319 kDefaultNumFrames &&
320 fake_video_renderer1.num_rendered_frames() > kDefaultNumFrames,
321 kDefaultTimeout);
322 } else {
323 WAIT(false, kReceivingDuration);
324 rendered_video_frames1_ = fake_video_renderer1.num_rendered_frames();
325 rendered_video_frames2_ = fake_video_renderer2.num_rendered_frames();
326 received_audio_frames1_ = fake_audio_capture_module1_->frames_received();
327 received_audio_frames2_ = fake_audio_capture_module2_->frames_received();
328 }
329 }
330
deadbeefe814a0d2017-02-25 18:15:09 -0800331 rtc::VirtualSocketServer virtual_socket_server_;
332 rtc::Thread network_thread_;
333 rtc::FakeNetworkManager fake_network_manager_;
334 rtc::scoped_refptr<FakeAudioCaptureModule> fake_audio_capture_module1_;
335 rtc::scoped_refptr<FakeAudioCaptureModule> fake_audio_capture_module2_;
336 std::unique_ptr<OrtcFactoryInterface> ortc_factory1_;
337 std::unique_ptr<OrtcFactoryInterface> ortc_factory2_;
338 // Actually owned by video tracks.
339 std::vector<cricket::FakeVideoCapturer*> fake_video_capturers_;
zhihuangd3501ad2017-03-03 14:39:06 -0800340 int received_audio_frames1_ = 0;
341 int received_audio_frames2_ = 0;
342 int rendered_video_frames1_ = 0;
343 int rendered_video_frames2_ = 0;
deadbeefe814a0d2017-02-25 18:15:09 -0800344};
345
zhihuangacfb0172017-03-28 13:45:18 -0700346// Disable for TSan v2, see
347// https://bugs.chromium.org/p/webrtc/issues/detail?id=7366 for details.
348#if !defined(THREAD_SANITIZER)
349
deadbeefe814a0d2017-02-25 18:15:09 -0800350// Very basic end-to-end test with a single pair of audio RTP sender and
351// receiver.
352//
353// Uses muxed RTCP, and minimal parameters with a hard-coded config that's
354// known to work.
355TEST_F(OrtcFactoryIntegrationTest, BasicOneWayAudioRtpSenderAndReceiver) {
356 auto udp_transports = CreateAndConnectUdpTransportPair();
357 auto rtp_transports =
358 CreateRtpTransportPair(MakeRtcpMuxParameters(), udp_transports);
359
360 auto sender_result = ortc_factory1_->CreateRtpSender(
361 cricket::MEDIA_TYPE_AUDIO, rtp_transports.first.get());
362 auto receiver_result = ortc_factory2_->CreateRtpReceiver(
363 cricket::MEDIA_TYPE_AUDIO, rtp_transports.second.get());
364 ASSERT_TRUE(sender_result.ok());
365 ASSERT_TRUE(receiver_result.ok());
366 auto sender = sender_result.MoveValue();
367 auto receiver = receiver_result.MoveValue();
368
369 RTCError error =
370 sender->SetTrack(CreateLocalAudioTrack("audio", ortc_factory1_.get()));
371 EXPECT_TRUE(error.ok());
372
373 RtpParameters opus_parameters = MakeMinimalOpusParameters();
374 EXPECT_TRUE(receiver->Receive(opus_parameters).ok());
375 EXPECT_TRUE(sender->Send(opus_parameters).ok());
376 // Sender and receiver are connected and configured; audio frames should be
377 // able to flow at this point.
378 EXPECT_TRUE_WAIT(
379 fake_audio_capture_module2_->frames_received() > kDefaultNumFrames,
380 kDefaultTimeout);
381}
382
383// Very basic end-to-end test with a single pair of video RTP sender and
384// receiver.
385//
386// Uses muxed RTCP, and minimal parameters with a hard-coded config that's
387// known to work.
388TEST_F(OrtcFactoryIntegrationTest, BasicOneWayVideoRtpSenderAndReceiver) {
389 auto udp_transports = CreateAndConnectUdpTransportPair();
390 auto rtp_transports =
391 CreateRtpTransportPair(MakeRtcpMuxParameters(), udp_transports);
392
393 auto sender_result = ortc_factory1_->CreateRtpSender(
394 cricket::MEDIA_TYPE_VIDEO, rtp_transports.first.get());
395 auto receiver_result = ortc_factory2_->CreateRtpReceiver(
396 cricket::MEDIA_TYPE_VIDEO, rtp_transports.second.get());
397 ASSERT_TRUE(sender_result.ok());
398 ASSERT_TRUE(receiver_result.ok());
399 auto sender = sender_result.MoveValue();
400 auto receiver = receiver_result.MoveValue();
401
402 RTCError error = sender->SetTrack(
403 CreateLocalVideoTrackAndFakeCapturer("video", ortc_factory1_.get()));
404 EXPECT_TRUE(error.ok());
405
406 RtpParameters vp8_parameters = MakeMinimalVp8Parameters();
407 EXPECT_TRUE(receiver->Receive(vp8_parameters).ok());
408 EXPECT_TRUE(sender->Send(vp8_parameters).ok());
409 FakeVideoTrackRenderer fake_renderer(
410 static_cast<VideoTrackInterface*>(receiver->GetTrack().get()));
411 // Sender and receiver are connected and configured; video frames should be
412 // able to flow at this point.
413 EXPECT_TRUE_WAIT(fake_renderer.num_rendered_frames() > kDefaultNumFrames,
414 kDefaultTimeout);
415}
416
417// Test that if the track is changed while sending, the sender seamlessly
418// transitions to sending it and frames are received end-to-end.
419//
420// Only doing this for video, since given that audio is sourced from a single
421// fake audio capture module, the audio track is just a dummy object.
422// TODO(deadbeef): Change this when possible.
423TEST_F(OrtcFactoryIntegrationTest, SetTrackWhileSending) {
424 auto udp_transports = CreateAndConnectUdpTransportPair();
425 auto rtp_transports =
426 CreateRtpTransportPair(MakeRtcpMuxParameters(), udp_transports);
427
428 auto sender_result = ortc_factory1_->CreateRtpSender(
429 cricket::MEDIA_TYPE_VIDEO, rtp_transports.first.get());
430 auto receiver_result = ortc_factory2_->CreateRtpReceiver(
431 cricket::MEDIA_TYPE_VIDEO, rtp_transports.second.get());
432 ASSERT_TRUE(sender_result.ok());
433 ASSERT_TRUE(receiver_result.ok());
434 auto sender = sender_result.MoveValue();
435 auto receiver = receiver_result.MoveValue();
436
437 RTCError error = sender->SetTrack(
438 CreateLocalVideoTrackAndFakeCapturer("video_1", ortc_factory1_.get()));
439 EXPECT_TRUE(error.ok());
440 RtpParameters vp8_parameters = MakeMinimalVp8Parameters();
441 EXPECT_TRUE(receiver->Receive(vp8_parameters).ok());
442 EXPECT_TRUE(sender->Send(vp8_parameters).ok());
443 FakeVideoTrackRenderer fake_renderer(
444 static_cast<VideoTrackInterface*>(receiver->GetTrack().get()));
445 // Expect for some initial number of frames to be received.
446 EXPECT_TRUE_WAIT(fake_renderer.num_rendered_frames() > kDefaultNumFrames,
447 kDefaultTimeout);
448 // Stop the old capturer, set a new track, and verify new frames are received
449 // from the new track. Stopping the old capturer ensures that we aren't
450 // actually still getting frames from it.
451 fake_video_capturers_[0]->Stop();
452 int prev_num_frames = fake_renderer.num_rendered_frames();
453 error = sender->SetTrack(
454 CreateLocalVideoTrackAndFakeCapturer("video_2", ortc_factory1_.get()));
455 EXPECT_TRUE(error.ok());
456 EXPECT_TRUE_WAIT(
457 fake_renderer.num_rendered_frames() > kDefaultNumFrames + prev_num_frames,
458 kDefaultTimeout);
459}
460
461// End-to-end test with two pairs of RTP senders and receivers, for audio and
462// video.
463//
464// Uses muxed RTCP, and minimal parameters with hard-coded configs that are
465// known to work.
466TEST_F(OrtcFactoryIntegrationTest,
467 BasicTwoWayAudioVideoRtpSendersAndReceivers) {
468 auto udp_transports = CreateAndConnectUdpTransportPair();
469 auto rtp_transports =
470 CreateRtpTransportPair(MakeRtcpMuxParameters(), udp_transports);
zhihuangd3501ad2017-03-03 14:39:06 -0800471 bool expect_success = true;
472 BasicTwoWayRtpSendersAndReceiversTest(std::move(rtp_transports),
473 expect_success);
474}
deadbeefe814a0d2017-02-25 18:15:09 -0800475
zhihuangd3501ad2017-03-03 14:39:06 -0800476TEST_F(OrtcFactoryIntegrationTest,
477 BasicTwoWayAudioVideoSrtpSendersAndReceivers) {
478 auto udp_transports = CreateAndConnectUdpTransportPair();
479 auto srtp_transports = CreateSrtpTransportPairAndSetKeys(
480 MakeRtcpMuxParameters(), udp_transports);
481 bool expect_success = true;
482 BasicTwoWayRtpSendersAndReceiversTest(std::move(srtp_transports),
483 expect_success);
484}
deadbeefe814a0d2017-02-25 18:15:09 -0800485
zhihuangd3501ad2017-03-03 14:39:06 -0800486// Tests that the packets cannot be decoded if the keys are mismatched.
487TEST_F(OrtcFactoryIntegrationTest, SrtpSendersAndReceiversWithMismatchingKeys) {
488 auto udp_transports = CreateAndConnectUdpTransportPair();
489 auto srtp_transports = CreateSrtpTransportPairAndSetMismatchingKeys(
490 MakeRtcpMuxParameters(), udp_transports);
491 bool expect_success = false;
492 BasicTwoWayRtpSendersAndReceiversTest(std::move(srtp_transports),
493 expect_success);
494 // No frames are expected to be decoded.
495 EXPECT_TRUE(received_audio_frames1_ == 0 && received_audio_frames2_ == 0 &&
496 rendered_video_frames1_ == 0 && rendered_video_frames2_ == 0);
497}
deadbeefe814a0d2017-02-25 18:15:09 -0800498
zhihuangd3501ad2017-03-03 14:39:06 -0800499// Tests that the frames cannot be decoded if only one side uses SRTP.
500TEST_F(OrtcFactoryIntegrationTest, OneSideSrtpSenderAndReceiver) {
501 auto rtcp_parameters = MakeRtcpMuxParameters();
502 auto udp_transports = CreateAndConnectUdpTransportPair();
503 auto rtcp_udp_transports = UdpTransportPair();
504 auto transport_controllers = RtpTransportControllerPair();
505 auto transport_result1 = ortc_factory1_->CreateRtpTransport(
506 rtcp_parameters, udp_transports.first.get(),
507 rtcp_udp_transports.first.get(), transport_controllers.first.get());
508 auto transport_result2 = ortc_factory2_->CreateSrtpTransport(
509 rtcp_parameters, udp_transports.second.get(),
510 rtcp_udp_transports.second.get(), transport_controllers.second.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800511
zhihuangd3501ad2017-03-03 14:39:06 -0800512 auto rtp_transport = transport_result1.MoveValue();
513 auto srtp_transport = transport_result2.MoveValue();
514 EXPECT_TRUE(srtp_transport->SetSrtpSendKey(kTestCryptoParams1).ok());
515 EXPECT_TRUE(srtp_transport->SetSrtpReceiveKey(kTestCryptoParams2).ok());
516 bool expect_success = false;
517 BasicTwoWayRtpSendersAndReceiversTest(
518 {std::move(rtp_transport), std::move(srtp_transport)}, expect_success);
deadbeefe814a0d2017-02-25 18:15:09 -0800519
zhihuangd3501ad2017-03-03 14:39:06 -0800520 // The SRTP side is not expected to decode any audio or video frames.
521 // The RTP side is not expected to decode any video frames while it is
522 // possible that the encrypted audio frames can be accidentally decoded which
523 // is why received_audio_frames1_ is not validated.
524 EXPECT_TRUE(received_audio_frames2_ == 0 && rendered_video_frames1_ == 0 &&
525 rendered_video_frames2_ == 0);
deadbeefe814a0d2017-02-25 18:15:09 -0800526}
527
528// End-to-end test with two pairs of RTP senders and receivers, for audio and
529// video. Unlike the test above, this attempts to make the parameters as
zhihuangd3501ad2017-03-03 14:39:06 -0800530// complex as possible. The senders and receivers use the SRTP transport with
531// different keys.
deadbeefe814a0d2017-02-25 18:15:09 -0800532//
533// Uses non-muxed RTCP, with separate audio/video transports, and a full set of
534// parameters, as would normally be used in a PeerConnection.
535//
536// TODO(deadbeef): Update this test as more audio/video features become
537// supported.
zhihuangd3501ad2017-03-03 14:39:06 -0800538TEST_F(OrtcFactoryIntegrationTest,
539 FullTwoWayAudioVideoSrtpSendersAndReceivers) {
deadbeefe814a0d2017-02-25 18:15:09 -0800540 // We want four pairs of UDP transports for this test, for audio/video and
541 // RTP/RTCP.
542 auto audio_rtp_udp_transports = CreateAndConnectUdpTransportPair();
543 auto audio_rtcp_udp_transports = CreateAndConnectUdpTransportPair();
544 auto video_rtp_udp_transports = CreateAndConnectUdpTransportPair();
545 auto video_rtcp_udp_transports = CreateAndConnectUdpTransportPair();
546
547 // Since we have multiple RTP transports on each side, we need an RTP
548 // transport controller.
549 auto transport_controllers = CreateRtpTransportControllerPair();
550
551 RtcpParameters audio_rtcp_parameters;
552 audio_rtcp_parameters.mux = false;
zhihuangd3501ad2017-03-03 14:39:06 -0800553 auto audio_srtp_transports =
554 CreateSrtpTransportPair(audio_rtcp_parameters, audio_rtp_udp_transports,
555 audio_rtcp_udp_transports, transport_controllers);
deadbeefe814a0d2017-02-25 18:15:09 -0800556
557 RtcpParameters video_rtcp_parameters;
558 video_rtcp_parameters.mux = false;
559 video_rtcp_parameters.reduced_size = true;
zhihuangd3501ad2017-03-03 14:39:06 -0800560 auto video_srtp_transports =
561 CreateSrtpTransportPair(video_rtcp_parameters, video_rtp_udp_transports,
562 video_rtcp_udp_transports, transport_controllers);
563
564 // Set keys for SRTP transports.
565 audio_srtp_transports.first->SetSrtpSendKey(kTestCryptoParams1);
566 audio_srtp_transports.first->SetSrtpReceiveKey(kTestCryptoParams2);
567 video_srtp_transports.first->SetSrtpSendKey(kTestCryptoParams3);
568 video_srtp_transports.first->SetSrtpReceiveKey(kTestCryptoParams4);
569
570 audio_srtp_transports.second->SetSrtpSendKey(kTestCryptoParams2);
571 audio_srtp_transports.second->SetSrtpReceiveKey(kTestCryptoParams1);
572 video_srtp_transports.second->SetSrtpSendKey(kTestCryptoParams4);
573 video_srtp_transports.second->SetSrtpReceiveKey(kTestCryptoParams3);
deadbeefe814a0d2017-02-25 18:15:09 -0800574
575 // Create all the senders and receivers (four per endpoint).
576 auto audio_sender_result1 = ortc_factory1_->CreateRtpSender(
zhihuangd3501ad2017-03-03 14:39:06 -0800577 cricket::MEDIA_TYPE_AUDIO, audio_srtp_transports.first.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800578 auto video_sender_result1 = ortc_factory1_->CreateRtpSender(
zhihuangd3501ad2017-03-03 14:39:06 -0800579 cricket::MEDIA_TYPE_VIDEO, video_srtp_transports.first.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800580 auto audio_receiver_result1 = ortc_factory1_->CreateRtpReceiver(
zhihuangd3501ad2017-03-03 14:39:06 -0800581 cricket::MEDIA_TYPE_AUDIO, audio_srtp_transports.first.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800582 auto video_receiver_result1 = ortc_factory1_->CreateRtpReceiver(
zhihuangd3501ad2017-03-03 14:39:06 -0800583 cricket::MEDIA_TYPE_VIDEO, video_srtp_transports.first.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800584 ASSERT_TRUE(audio_sender_result1.ok());
585 ASSERT_TRUE(video_sender_result1.ok());
586 ASSERT_TRUE(audio_receiver_result1.ok());
587 ASSERT_TRUE(video_receiver_result1.ok());
588 auto audio_sender1 = audio_sender_result1.MoveValue();
589 auto video_sender1 = video_sender_result1.MoveValue();
590 auto audio_receiver1 = audio_receiver_result1.MoveValue();
591 auto video_receiver1 = video_receiver_result1.MoveValue();
592
593 auto audio_sender_result2 = ortc_factory2_->CreateRtpSender(
zhihuangd3501ad2017-03-03 14:39:06 -0800594 cricket::MEDIA_TYPE_AUDIO, audio_srtp_transports.second.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800595 auto video_sender_result2 = ortc_factory2_->CreateRtpSender(
zhihuangd3501ad2017-03-03 14:39:06 -0800596 cricket::MEDIA_TYPE_VIDEO, video_srtp_transports.second.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800597 auto audio_receiver_result2 = ortc_factory2_->CreateRtpReceiver(
zhihuangd3501ad2017-03-03 14:39:06 -0800598 cricket::MEDIA_TYPE_AUDIO, audio_srtp_transports.second.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800599 auto video_receiver_result2 = ortc_factory2_->CreateRtpReceiver(
zhihuangd3501ad2017-03-03 14:39:06 -0800600 cricket::MEDIA_TYPE_VIDEO, video_srtp_transports.second.get());
deadbeefe814a0d2017-02-25 18:15:09 -0800601 ASSERT_TRUE(audio_sender_result2.ok());
602 ASSERT_TRUE(video_sender_result2.ok());
603 ASSERT_TRUE(audio_receiver_result2.ok());
604 ASSERT_TRUE(video_receiver_result2.ok());
605 auto audio_sender2 = audio_sender_result2.MoveValue();
606 auto video_sender2 = video_sender_result2.MoveValue();
607 auto audio_receiver2 = audio_receiver_result2.MoveValue();
608 auto video_receiver2 = video_receiver_result2.MoveValue();
609
610 RTCError error = audio_sender1->SetTrack(
611 CreateLocalAudioTrack("audio", ortc_factory1_.get()));
612 EXPECT_TRUE(error.ok());
613 error = video_sender1->SetTrack(
614 CreateLocalVideoTrackAndFakeCapturer("video", ortc_factory1_.get()));
615 EXPECT_TRUE(error.ok());
616 error = audio_sender2->SetTrack(
617 CreateLocalAudioTrack("audio", ortc_factory2_.get()));
618 EXPECT_TRUE(error.ok());
619 error = video_sender2->SetTrack(
620 CreateLocalVideoTrackAndFakeCapturer("video", ortc_factory2_.get()));
621 EXPECT_TRUE(error.ok());
622
623 // Use different codecs in different directions for extra challenge.
624 RtpParameters opus_send_parameters = MakeFullOpusParameters();
625 RtpParameters isac_send_parameters = MakeFullIsacParameters();
626 RtpParameters vp8_send_parameters = MakeFullVp8Parameters();
627 RtpParameters vp9_send_parameters = MakeFullVp9Parameters();
628
629 // Remove "payload_type" from receive parameters. Receiver will need to
630 // discern the payload type from packets received.
631 RtpParameters opus_receive_parameters = opus_send_parameters;
632 RtpParameters isac_receive_parameters = isac_send_parameters;
633 RtpParameters vp8_receive_parameters = vp8_send_parameters;
634 RtpParameters vp9_receive_parameters = vp9_send_parameters;
635 opus_receive_parameters.encodings[0].codec_payload_type.reset();
636 isac_receive_parameters.encodings[0].codec_payload_type.reset();
637 vp8_receive_parameters.encodings[0].codec_payload_type.reset();
638 vp9_receive_parameters.encodings[0].codec_payload_type.reset();
639
640 // Configure the senders' and receivers' parameters.
641 //
642 // Note: Intentionally, the top codec in the receive parameters does not
643 // match the codec sent by the other side. If "Receive" is called with a list
644 // of codecs, the receiver should be prepared to receive any of them, not
645 // just the one on top.
646 EXPECT_TRUE(audio_receiver1->Receive(opus_receive_parameters).ok());
647 EXPECT_TRUE(video_receiver1->Receive(vp8_receive_parameters).ok());
648 EXPECT_TRUE(audio_receiver2->Receive(isac_receive_parameters).ok());
649 EXPECT_TRUE(video_receiver2->Receive(vp9_receive_parameters).ok());
650 EXPECT_TRUE(audio_sender1->Send(opus_send_parameters).ok());
651 EXPECT_TRUE(video_sender1->Send(vp8_send_parameters).ok());
652 EXPECT_TRUE(audio_sender2->Send(isac_send_parameters).ok());
653 EXPECT_TRUE(video_sender2->Send(vp9_send_parameters).ok());
654
655 FakeVideoTrackRenderer fake_video_renderer1(
656 static_cast<VideoTrackInterface*>(video_receiver1->GetTrack().get()));
657 FakeVideoTrackRenderer fake_video_renderer2(
658 static_cast<VideoTrackInterface*>(video_receiver2->GetTrack().get()));
659
660 // Senders and receivers are connected and configured; audio and video frames
661 // should be able to flow at this point.
662 EXPECT_TRUE_WAIT(
663 fake_audio_capture_module1_->frames_received() > kDefaultNumFrames &&
664 fake_video_renderer1.num_rendered_frames() > kDefaultNumFrames &&
665 fake_audio_capture_module2_->frames_received() > kDefaultNumFrames &&
666 fake_video_renderer2.num_rendered_frames() > kDefaultNumFrames,
667 kDefaultTimeout);
668}
669
670// TODO(deadbeef): End-to-end test for multiple senders/receivers of the same
671// media type, once that's supported. Currently, it is not because the
672// BaseChannel model relies on there being a single VoiceChannel and
673// VideoChannel, and these only support a single set of codecs/etc. per
674// send/receive direction.
675
676// TODO(deadbeef): End-to-end test for simulcast, once that's supported by this
677// API.
678
zhihuangacfb0172017-03-28 13:45:18 -0700679#endif // if !defined(THREAD_SANITIZER)
680
deadbeefe814a0d2017-02-25 18:15:09 -0800681} // namespace webrtc