jlmiller@webrtc.org | 5f93d0a | 2015-01-20 21:36:13 +0000 | [diff] [blame] | 1 | /* |
kjellander | 65c7f67 | 2016-02-12 00:05:01 -0800 | [diff] [blame] | 2 | * Copyright 2008 The WebRTC project authors. All Rights Reserved. |
jlmiller@webrtc.org | 5f93d0a | 2015-01-20 21:36:13 +0000 | [diff] [blame] | 3 | * |
kjellander | 65c7f67 | 2016-02-12 00:05:01 -0800 | [diff] [blame] | 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. |
jlmiller@webrtc.org | 5f93d0a | 2015-01-20 21:36:13 +0000 | [diff] [blame] | 9 | */ |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 10 | |
Anton Sukhanov | 4f08faa | 2019-05-21 11:12:57 -0700 | [diff] [blame] | 11 | #include "pc/channel_manager.h" |
| 12 | |
deadbeef | 112b2e9 | 2017-02-10 20:13:37 -0800 | [diff] [blame] | 13 | #include <memory> |
| 14 | |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 15 | #include "api/rtc_error.h" |
Niels Möller | 65f17ca | 2019-09-12 13:59:36 +0200 | [diff] [blame] | 16 | #include "api/transport/media/media_transport_config.h" |
Jonas Oreland | a3aa9bd | 2019-04-17 07:38:40 +0200 | [diff] [blame] | 17 | #include "api/video/builtin_video_bitrate_allocator_factory.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 18 | #include "media/base/fake_media_engine.h" |
| 19 | #include "media/base/test_utils.h" |
| 20 | #include "media/engine/fake_webrtc_call.h" |
| 21 | #include "p2p/base/dtls_transport_internal.h" |
| 22 | #include "p2p/base/fake_dtls_transport.h" |
| 23 | #include "p2p/base/p2p_constants.h" |
| 24 | #include "p2p/base/packet_transport_internal.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 25 | #include "pc/dtls_srtp_transport.h" |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 26 | #include "rtc_base/checks.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 27 | #include "rtc_base/thread.h" |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 28 | #include "test/gtest.h" |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 29 | |
deadbeef | 1a2183d | 2017-02-10 23:44:49 -0800 | [diff] [blame] | 30 | namespace { |
deadbeef | 7af91dd | 2016-12-13 11:29:11 -0800 | [diff] [blame] | 31 | const bool kDefaultSrtpRequired = true; |
| 32 | } |
| 33 | |
| 34 | namespace cricket { |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 35 | |
| 36 | static const AudioCodec kAudioCodecs[] = { |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 37 | AudioCodec(97, "voice", 1, 2, 3), |
| 38 | AudioCodec(111, "OPUS", 48000, 32000, 2), |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 39 | }; |
| 40 | |
| 41 | static const VideoCodec kVideoCodecs[] = { |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 42 | VideoCodec(99, "H264"), |
| 43 | VideoCodec(100, "VP8"), |
| 44 | VideoCodec(96, "rtx"), |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 45 | }; |
| 46 | |
Mirko Bonadei | 6a489f2 | 2019-04-09 15:11:12 +0200 | [diff] [blame] | 47 | class ChannelManagerTest : public ::testing::Test { |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 48 | protected: |
deadbeef | cbecd35 | 2015-09-23 11:50:27 -0700 | [diff] [blame] | 49 | ChannelManagerTest() |
tommi | e725159 | 2017-07-14 14:44:46 -0700 | [diff] [blame] | 50 | : network_(rtc::Thread::CreateWithSocketServer()), |
| 51 | worker_(rtc::Thread::Create()), |
Jonas Oreland | a3aa9bd | 2019-04-17 07:38:40 +0200 | [diff] [blame] | 52 | video_bitrate_allocator_factory_( |
| 53 | webrtc::CreateBuiltinVideoBitrateAllocatorFactory()), |
tommi | e725159 | 2017-07-14 14:44:46 -0700 | [diff] [blame] | 54 | fme_(new cricket::FakeMediaEngine()), |
stefan | c1aeaf0 | 2015-10-15 07:26:07 -0700 | [diff] [blame] | 55 | fdme_(new cricket::FakeDataEngine()), |
deadbeef | 112b2e9 | 2017-02-10 20:13:37 -0800 | [diff] [blame] | 56 | cm_(new cricket::ChannelManager( |
| 57 | std::unique_ptr<MediaEngineInterface>(fme_), |
| 58 | std::unique_ptr<DataEngineInterface>(fdme_), |
Steve Anton | c9e1560 | 2017-11-06 15:40:09 -0800 | [diff] [blame] | 59 | rtc::Thread::Current(), |
deadbeef | 112b2e9 | 2017-02-10 20:13:37 -0800 | [diff] [blame] | 60 | rtc::Thread::Current())), |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 61 | fake_call_() { |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 62 | fme_->SetAudioCodecs(MAKE_VECTOR(kAudioCodecs)); |
| 63 | fme_->SetVideoCodecs(MAKE_VECTOR(kVideoCodecs)); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 64 | } |
| 65 | |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 66 | std::unique_ptr<webrtc::RtpTransportInternal> CreateDtlsSrtpTransport() { |
Mirko Bonadei | 317a1f0 | 2019-09-17 17:06:18 +0200 | [diff] [blame] | 67 | rtp_dtls_transport_ = std::make_unique<FakeDtlsTransport>( |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 68 | "fake_dtls_transport", cricket::ICE_CANDIDATE_COMPONENT_RTP); |
Mirko Bonadei | 317a1f0 | 2019-09-17 17:06:18 +0200 | [diff] [blame] | 69 | auto dtls_srtp_transport = std::make_unique<webrtc::DtlsSrtpTransport>( |
Karl Wiberg | 918f50c | 2018-07-05 11:40:33 +0200 | [diff] [blame] | 70 | /*rtcp_mux_required=*/true); |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 71 | dtls_srtp_transport->SetDtlsTransports(rtp_dtls_transport_.get(), |
| 72 | /*rtcp_dtls_transport=*/nullptr); |
| 73 | return dtls_srtp_transport; |
| 74 | } |
| 75 | |
Anton Sukhanov | 98a462c | 2018-10-17 13:15:42 -0700 | [diff] [blame] | 76 | void TestCreateDestroyChannels( |
| 77 | webrtc::RtpTransportInternal* rtp_transport, |
Anton Sukhanov | 4f08faa | 2019-05-21 11:12:57 -0700 | [diff] [blame] | 78 | webrtc::MediaTransportConfig media_transport_config) { |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 79 | cricket::VoiceChannel* voice_channel = cm_->CreateVoiceChannel( |
Anton Sukhanov | 4f08faa | 2019-05-21 11:12:57 -0700 | [diff] [blame] | 80 | &fake_call_, cricket::MediaConfig(), rtp_transport, |
| 81 | media_transport_config, rtc::Thread::Current(), cricket::CN_AUDIO, |
| 82 | kDefaultSrtpRequired, webrtc::CryptoOptions(), &ssrc_generator_, |
| 83 | AudioOptions()); |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 84 | EXPECT_TRUE(voice_channel != nullptr); |
| 85 | cricket::VideoChannel* video_channel = cm_->CreateVideoChannel( |
Anton Sukhanov | 4f08faa | 2019-05-21 11:12:57 -0700 | [diff] [blame] | 86 | &fake_call_, cricket::MediaConfig(), rtp_transport, |
| 87 | media_transport_config, rtc::Thread::Current(), cricket::CN_VIDEO, |
| 88 | kDefaultSrtpRequired, webrtc::CryptoOptions(), &ssrc_generator_, |
| 89 | VideoOptions(), video_bitrate_allocator_factory_.get()); |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 90 | EXPECT_TRUE(video_channel != nullptr); |
| 91 | cricket::RtpDataChannel* rtp_data_channel = cm_->CreateRtpDataChannel( |
| 92 | cricket::MediaConfig(), rtp_transport, rtc::Thread::Current(), |
Amit Hilbuch | bcd39d4 | 2019-01-25 17:13:56 -0800 | [diff] [blame] | 93 | cricket::CN_DATA, kDefaultSrtpRequired, webrtc::CryptoOptions(), |
| 94 | &ssrc_generator_); |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 95 | EXPECT_TRUE(rtp_data_channel != nullptr); |
| 96 | cm_->DestroyVideoChannel(video_channel); |
| 97 | cm_->DestroyVoiceChannel(voice_channel); |
| 98 | cm_->DestroyRtpDataChannel(rtp_data_channel); |
| 99 | cm_->Terminate(); |
| 100 | } |
| 101 | |
| 102 | std::unique_ptr<DtlsTransportInternal> rtp_dtls_transport_; |
tommi | e725159 | 2017-07-14 14:44:46 -0700 | [diff] [blame] | 103 | std::unique_ptr<rtc::Thread> network_; |
| 104 | std::unique_ptr<rtc::Thread> worker_; |
Jonas Oreland | a3aa9bd | 2019-04-17 07:38:40 +0200 | [diff] [blame] | 105 | std::unique_ptr<webrtc::VideoBitrateAllocatorFactory> |
| 106 | video_bitrate_allocator_factory_; |
deadbeef | 112b2e9 | 2017-02-10 20:13:37 -0800 | [diff] [blame] | 107 | // |fme_| and |fdme_| are actually owned by |cm_|. |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 108 | cricket::FakeMediaEngine* fme_; |
| 109 | cricket::FakeDataEngine* fdme_; |
deadbeef | 112b2e9 | 2017-02-10 20:13:37 -0800 | [diff] [blame] | 110 | std::unique_ptr<cricket::ChannelManager> cm_; |
stefan | c1aeaf0 | 2015-10-15 07:26:07 -0700 | [diff] [blame] | 111 | cricket::FakeCall fake_call_; |
Amit Hilbuch | bcd39d4 | 2019-01-25 17:13:56 -0800 | [diff] [blame] | 112 | rtc::UniqueRandomIdGenerator ssrc_generator_; |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 113 | }; |
| 114 | |
| 115 | // Test that we startup/shutdown properly. |
| 116 | TEST_F(ChannelManagerTest, StartupShutdown) { |
| 117 | EXPECT_FALSE(cm_->initialized()); |
buildbot@webrtc.org | d4e598d | 2014-07-29 17:36:52 +0000 | [diff] [blame] | 118 | EXPECT_EQ(rtc::Thread::Current(), cm_->worker_thread()); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 119 | EXPECT_TRUE(cm_->Init()); |
| 120 | EXPECT_TRUE(cm_->initialized()); |
| 121 | cm_->Terminate(); |
| 122 | EXPECT_FALSE(cm_->initialized()); |
| 123 | } |
| 124 | |
| 125 | // Test that we startup/shutdown properly with a worker thread. |
| 126 | TEST_F(ChannelManagerTest, StartupShutdownOnThread) { |
tommi | e725159 | 2017-07-14 14:44:46 -0700 | [diff] [blame] | 127 | network_->Start(); |
| 128 | worker_->Start(); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 129 | EXPECT_FALSE(cm_->initialized()); |
buildbot@webrtc.org | d4e598d | 2014-07-29 17:36:52 +0000 | [diff] [blame] | 130 | EXPECT_EQ(rtc::Thread::Current(), cm_->worker_thread()); |
tommi | e725159 | 2017-07-14 14:44:46 -0700 | [diff] [blame] | 131 | EXPECT_TRUE(cm_->set_network_thread(network_.get())); |
| 132 | EXPECT_EQ(network_.get(), cm_->network_thread()); |
| 133 | EXPECT_TRUE(cm_->set_worker_thread(worker_.get())); |
| 134 | EXPECT_EQ(worker_.get(), cm_->worker_thread()); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 135 | EXPECT_TRUE(cm_->Init()); |
| 136 | EXPECT_TRUE(cm_->initialized()); |
Danil Chapovalov | 33b01f2 | 2016-05-11 19:55:27 +0200 | [diff] [blame] | 137 | // Setting the network or worker thread while initialized should fail. |
| 138 | EXPECT_FALSE(cm_->set_network_thread(rtc::Thread::Current())); |
buildbot@webrtc.org | d4e598d | 2014-07-29 17:36:52 +0000 | [diff] [blame] | 139 | EXPECT_FALSE(cm_->set_worker_thread(rtc::Thread::Current())); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 140 | cm_->Terminate(); |
| 141 | EXPECT_FALSE(cm_->initialized()); |
| 142 | } |
| 143 | |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 144 | TEST_F(ChannelManagerTest, SetVideoRtxEnabled) { |
| 145 | std::vector<VideoCodec> codecs; |
perkj | 2675274 | 2016-10-24 01:21:16 -0700 | [diff] [blame] | 146 | const VideoCodec rtx_codec(96, "rtx"); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 147 | |
| 148 | // By default RTX is disabled. |
magjed | 3cf8ece | 2016-11-10 03:36:53 -0800 | [diff] [blame] | 149 | cm_->GetSupportedVideoCodecs(&codecs); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 150 | EXPECT_FALSE(ContainsMatchingCodec(codecs, rtx_codec)); |
| 151 | |
| 152 | // Enable and check. |
| 153 | EXPECT_TRUE(cm_->SetVideoRtxEnabled(true)); |
magjed | 3cf8ece | 2016-11-10 03:36:53 -0800 | [diff] [blame] | 154 | cm_->GetSupportedVideoCodecs(&codecs); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 155 | EXPECT_TRUE(ContainsMatchingCodec(codecs, rtx_codec)); |
| 156 | |
| 157 | // Disable and check. |
| 158 | EXPECT_TRUE(cm_->SetVideoRtxEnabled(false)); |
magjed | 3cf8ece | 2016-11-10 03:36:53 -0800 | [diff] [blame] | 159 | cm_->GetSupportedVideoCodecs(&codecs); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 160 | EXPECT_FALSE(ContainsMatchingCodec(codecs, rtx_codec)); |
| 161 | |
| 162 | // Cannot toggle rtx after initialization. |
| 163 | EXPECT_TRUE(cm_->Init()); |
| 164 | EXPECT_FALSE(cm_->SetVideoRtxEnabled(true)); |
| 165 | EXPECT_FALSE(cm_->SetVideoRtxEnabled(false)); |
| 166 | |
| 167 | // Can set again after terminate. |
| 168 | cm_->Terminate(); |
| 169 | EXPECT_TRUE(cm_->SetVideoRtxEnabled(true)); |
magjed | 3cf8ece | 2016-11-10 03:36:53 -0800 | [diff] [blame] | 170 | cm_->GetSupportedVideoCodecs(&codecs); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 171 | EXPECT_TRUE(ContainsMatchingCodec(codecs, rtx_codec)); |
| 172 | } |
| 173 | |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 174 | TEST_F(ChannelManagerTest, CreateDestroyChannels) { |
Zhi Huang | 2dfc42d | 2017-12-04 13:38:48 -0800 | [diff] [blame] | 175 | EXPECT_TRUE(cm_->Init()); |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 176 | auto rtp_transport = CreateDtlsSrtpTransport(); |
Anton Sukhanov | 4f08faa | 2019-05-21 11:12:57 -0700 | [diff] [blame] | 177 | TestCreateDestroyChannels(rtp_transport.get(), |
| 178 | webrtc::MediaTransportConfig()); |
Anton Sukhanov | 98a462c | 2018-10-17 13:15:42 -0700 | [diff] [blame] | 179 | } |
| 180 | |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 181 | TEST_F(ChannelManagerTest, CreateDestroyChannelsOnThread) { |
Zhi Huang | 2dfc42d | 2017-12-04 13:38:48 -0800 | [diff] [blame] | 182 | network_->Start(); |
| 183 | worker_->Start(); |
| 184 | EXPECT_TRUE(cm_->set_worker_thread(worker_.get())); |
| 185 | EXPECT_TRUE(cm_->set_network_thread(network_.get())); |
| 186 | EXPECT_TRUE(cm_->Init()); |
Zhi Huang | e830e68 | 2018-03-30 10:48:35 -0700 | [diff] [blame] | 187 | auto rtp_transport = CreateDtlsSrtpTransport(); |
Anton Sukhanov | 4f08faa | 2019-05-21 11:12:57 -0700 | [diff] [blame] | 188 | TestCreateDestroyChannels(rtp_transport.get(), |
| 189 | webrtc::MediaTransportConfig()); |
Zhi Huang | 2dfc42d | 2017-12-04 13:38:48 -0800 | [diff] [blame] | 190 | } |
| 191 | |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 +0000 | [diff] [blame] | 192 | } // namespace cricket |