| /* |
| * Copyright 2018 The WebRTC project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #ifndef PC_TEST_FAKE_PEER_CONNECTION_FOR_STATS_H_ |
| #define PC_TEST_FAKE_PEER_CONNECTION_FOR_STATS_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <set> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "media/base/fake_media_engine.h" |
| #include "media/base/media_channel.h" |
| #include "pc/channel.h" |
| #include "pc/stream_collection.h" |
| #include "pc/test/fake_data_channel_controller.h" |
| #include "pc/test/fake_peer_connection_base.h" |
| |
| namespace webrtc { |
| |
| // Fake VoiceMediaChannel where the result of GetStats can be configured. |
| class FakeVoiceMediaSendChannelForStats |
| : public cricket::FakeVoiceMediaSendChannel { |
| public: |
| explicit FakeVoiceMediaSendChannelForStats(TaskQueueBase* network_thread) |
| : cricket::FakeVoiceMediaSendChannel(cricket::AudioOptions(), |
| network_thread) {} |
| |
| void SetStats(const cricket::VoiceMediaInfo& voice_info) { |
| send_stats_ = cricket::VoiceMediaSendInfo(); |
| send_stats_->senders = voice_info.senders; |
| send_stats_->send_codecs = voice_info.send_codecs; |
| } |
| |
| // VoiceMediaChannel overrides. |
| bool GetStats(cricket::VoiceMediaSendInfo* info) override { |
| if (send_stats_) { |
| *info = *send_stats_; |
| return true; |
| } |
| return false; |
| } |
| |
| private: |
| absl::optional<cricket::VoiceMediaSendInfo> send_stats_; |
| }; |
| |
| class FakeVoiceMediaReceiveChannelForStats |
| : public cricket::FakeVoiceMediaReceiveChannel { |
| public: |
| explicit FakeVoiceMediaReceiveChannelForStats(TaskQueueBase* network_thread) |
| : cricket::FakeVoiceMediaReceiveChannel(cricket::AudioOptions(), |
| network_thread) {} |
| |
| void SetStats(const cricket::VoiceMediaInfo& voice_info) { |
| receive_stats_ = cricket::VoiceMediaReceiveInfo(); |
| receive_stats_->receivers = voice_info.receivers; |
| receive_stats_->receive_codecs = voice_info.receive_codecs; |
| receive_stats_->device_underrun_count = voice_info.device_underrun_count; |
| } |
| |
| // VoiceMediaChannel overrides. |
| bool GetStats(cricket::VoiceMediaReceiveInfo* info, |
| bool get_and_clear_legacy_stats) override { |
| if (receive_stats_) { |
| *info = *receive_stats_; |
| return true; |
| } |
| return false; |
| } |
| |
| private: |
| absl::optional<cricket::VoiceMediaReceiveInfo> receive_stats_; |
| }; |
| |
| // Fake VideoMediaChannel where the result of GetStats can be configured. |
| class FakeVideoMediaSendChannelForStats |
| : public cricket::FakeVideoMediaSendChannel { |
| public: |
| explicit FakeVideoMediaSendChannelForStats(TaskQueueBase* network_thread) |
| : cricket::FakeVideoMediaSendChannel(cricket::VideoOptions(), |
| network_thread) {} |
| |
| void SetStats(const cricket::VideoMediaInfo& video_info) { |
| send_stats_ = cricket::VideoMediaSendInfo(); |
| send_stats_->senders = video_info.senders; |
| send_stats_->aggregated_senders = video_info.aggregated_senders; |
| send_stats_->send_codecs = video_info.send_codecs; |
| } |
| |
| // VideoMediaChannel overrides. |
| bool GetStats(cricket::VideoMediaSendInfo* info) override { |
| if (send_stats_) { |
| *info = *send_stats_; |
| return true; |
| } |
| return false; |
| } |
| |
| private: |
| absl::optional<cricket::VideoMediaSendInfo> send_stats_; |
| }; |
| |
| class FakeVideoMediaReceiveChannelForStats |
| : public cricket::FakeVideoMediaReceiveChannel { |
| public: |
| explicit FakeVideoMediaReceiveChannelForStats(TaskQueueBase* network_thread) |
| : cricket::FakeVideoMediaReceiveChannel(cricket::VideoOptions(), |
| network_thread) {} |
| |
| void SetStats(const cricket::VideoMediaInfo& video_info) { |
| receive_stats_ = cricket::VideoMediaReceiveInfo(); |
| receive_stats_->receivers = video_info.receivers; |
| receive_stats_->receive_codecs = video_info.receive_codecs; |
| } |
| |
| // VideoMediaChannel overrides. |
| bool GetStats(cricket::VideoMediaReceiveInfo* info) override { |
| if (receive_stats_) { |
| *info = *receive_stats_; |
| return true; |
| } |
| return false; |
| } |
| |
| private: |
| absl::optional<cricket::VideoMediaReceiveInfo> receive_stats_; |
| }; |
| |
| constexpr bool kDefaultRtcpMuxRequired = true; |
| constexpr bool kDefaultSrtpRequired = true; |
| |
| class VoiceChannelForTesting : public cricket::VoiceChannel { |
| public: |
| VoiceChannelForTesting( |
| rtc::Thread* worker_thread, |
| rtc::Thread* network_thread, |
| rtc::Thread* signaling_thread, |
| std::unique_ptr<cricket::VoiceMediaSendChannelInterface> send_channel, |
| std::unique_ptr<cricket::VoiceMediaReceiveChannelInterface> |
| receive_channel, |
| const std::string& content_name, |
| bool srtp_required, |
| webrtc::CryptoOptions crypto_options, |
| rtc::UniqueRandomIdGenerator* ssrc_generator, |
| std::string transport_name) |
| : VoiceChannel(worker_thread, |
| network_thread, |
| signaling_thread, |
| std::move(send_channel), |
| std::move(receive_channel), |
| content_name, |
| srtp_required, |
| std::move(crypto_options), |
| ssrc_generator), |
| test_transport_name_(std::move(transport_name)) {} |
| |
| private: |
| absl::string_view transport_name() const override { |
| return test_transport_name_; |
| } |
| |
| const std::string test_transport_name_; |
| }; |
| |
| class VideoChannelForTesting : public cricket::VideoChannel { |
| public: |
| VideoChannelForTesting( |
| rtc::Thread* worker_thread, |
| rtc::Thread* network_thread, |
| rtc::Thread* signaling_thread, |
| std::unique_ptr<cricket::VideoMediaSendChannelInterface> send_channel, |
| std::unique_ptr<cricket::VideoMediaReceiveChannelInterface> |
| receive_channel, |
| const std::string& content_name, |
| bool srtp_required, |
| webrtc::CryptoOptions crypto_options, |
| rtc::UniqueRandomIdGenerator* ssrc_generator, |
| std::string transport_name) |
| : VideoChannel(worker_thread, |
| network_thread, |
| signaling_thread, |
| std::move(send_channel), |
| std::move(receive_channel), |
| content_name, |
| srtp_required, |
| std::move(crypto_options), |
| ssrc_generator), |
| test_transport_name_(std::move(transport_name)) {} |
| |
| private: |
| absl::string_view transport_name() const override { |
| return test_transport_name_; |
| } |
| |
| const std::string test_transport_name_; |
| }; |
| |
| // This class is intended to be fed into the StatsCollector and |
| // RTCStatsCollector so that the stats functionality can be unit tested. |
| // Individual tests can configure this fake as needed to simulate scenarios |
| // under which to test the stats collectors. |
| class FakePeerConnectionForStats : public FakePeerConnectionBase { |
| public: |
| // TODO(steveanton): Add support for specifying separate threads to test |
| // multi-threading correctness. |
| FakePeerConnectionForStats() |
| : network_thread_(rtc::Thread::Current()), |
| worker_thread_(rtc::Thread::Current()), |
| signaling_thread_(rtc::Thread::Current()), |
| // TODO(hta): remove separate thread variables and use context. |
| dependencies_(MakeDependencies()), |
| context_(ConnectionContext::Create(&dependencies_)), |
| local_streams_(StreamCollection::Create()), |
| remote_streams_(StreamCollection::Create()), |
| data_channel_controller_(network_thread_) {} |
| |
| ~FakePeerConnectionForStats() { |
| for (auto transceiver : transceivers_) { |
| transceiver->internal()->ClearChannel(); |
| } |
| } |
| |
| static PeerConnectionFactoryDependencies MakeDependencies() { |
| PeerConnectionFactoryDependencies dependencies; |
| dependencies.network_thread = rtc::Thread::Current(); |
| dependencies.worker_thread = rtc::Thread::Current(); |
| dependencies.signaling_thread = rtc::Thread::Current(); |
| dependencies.media_engine = std::make_unique<cricket::FakeMediaEngine>(); |
| return dependencies; |
| } |
| |
| rtc::scoped_refptr<StreamCollection> mutable_local_streams() { |
| return local_streams_; |
| } |
| |
| rtc::scoped_refptr<StreamCollection> mutable_remote_streams() { |
| return remote_streams_; |
| } |
| |
| rtc::scoped_refptr<RtpSenderInterface> AddSender( |
| rtc::scoped_refptr<RtpSenderInternal> sender) { |
| // TODO(steveanton): Switch tests to use RtpTransceivers directly. |
| auto sender_proxy = RtpSenderProxyWithInternal<RtpSenderInternal>::Create( |
| signaling_thread_, sender); |
| GetOrCreateFirstTransceiverOfType(sender->media_type()) |
| ->internal() |
| ->AddSender(sender_proxy); |
| return sender_proxy; |
| } |
| |
| void RemoveSender(rtc::scoped_refptr<RtpSenderInterface> sender) { |
| GetOrCreateFirstTransceiverOfType(sender->media_type()) |
| ->internal() |
| ->RemoveSender(sender.get()); |
| } |
| |
| rtc::scoped_refptr<RtpReceiverInterface> AddReceiver( |
| rtc::scoped_refptr<RtpReceiverInternal> receiver) { |
| // TODO(steveanton): Switch tests to use RtpTransceivers directly. |
| auto receiver_proxy = |
| RtpReceiverProxyWithInternal<RtpReceiverInternal>::Create( |
| signaling_thread_, worker_thread_, receiver); |
| GetOrCreateFirstTransceiverOfType(receiver->media_type()) |
| ->internal() |
| ->AddReceiver(receiver_proxy); |
| return receiver_proxy; |
| } |
| |
| void RemoveReceiver(rtc::scoped_refptr<RtpReceiverInterface> receiver) { |
| GetOrCreateFirstTransceiverOfType(receiver->media_type()) |
| ->internal() |
| ->RemoveReceiver(receiver.get()); |
| } |
| |
| std::pair<FakeVoiceMediaSendChannelForStats*, |
| FakeVoiceMediaReceiveChannelForStats*> |
| AddVoiceChannel( |
| const std::string& mid, |
| const std::string& transport_name, |
| cricket::VoiceMediaInfo initial_stats = cricket::VoiceMediaInfo()) { |
| auto voice_media_send_channel = |
| std::make_unique<FakeVoiceMediaSendChannelForStats>(network_thread_); |
| auto voice_media_receive_channel = |
| std::make_unique<FakeVoiceMediaReceiveChannelForStats>(network_thread_); |
| auto* voice_media_send_channel_ptr = voice_media_send_channel.get(); |
| auto* voice_media_receive_channel_ptr = voice_media_receive_channel.get(); |
| auto voice_channel = std::make_unique<VoiceChannelForTesting>( |
| worker_thread_, network_thread_, signaling_thread_, |
| std::move(voice_media_send_channel), |
| std::move(voice_media_receive_channel), mid, kDefaultSrtpRequired, |
| webrtc::CryptoOptions(), context_->ssrc_generator(), transport_name); |
| auto transceiver = |
| GetOrCreateFirstTransceiverOfType(cricket::MEDIA_TYPE_AUDIO) |
| ->internal(); |
| if (transceiver->channel()) { |
| // This transceiver already has a channel, create a new one. |
| transceiver = |
| CreateTransceiverOfType(cricket::MEDIA_TYPE_AUDIO)->internal(); |
| } |
| RTC_DCHECK(!transceiver->channel()); |
| transceiver->SetChannel(std::move(voice_channel), |
| [](const std::string&) { return nullptr; }); |
| voice_media_send_channel_ptr->SetStats(initial_stats); |
| voice_media_receive_channel_ptr->SetStats(initial_stats); |
| return std::make_pair(voice_media_send_channel_ptr, |
| voice_media_receive_channel_ptr); |
| } |
| |
| std::pair<FakeVideoMediaSendChannelForStats*, |
| FakeVideoMediaReceiveChannelForStats*> |
| AddVideoChannel( |
| const std::string& mid, |
| const std::string& transport_name, |
| cricket::VideoMediaInfo initial_stats = cricket::VideoMediaInfo()) { |
| auto video_media_send_channel = |
| std::make_unique<FakeVideoMediaSendChannelForStats>(network_thread_); |
| auto video_media_receive_channel = |
| std::make_unique<FakeVideoMediaReceiveChannelForStats>(network_thread_); |
| auto video_media_send_channel_ptr = video_media_send_channel.get(); |
| auto video_media_receive_channel_ptr = video_media_receive_channel.get(); |
| auto video_channel = std::make_unique<VideoChannelForTesting>( |
| worker_thread_, network_thread_, signaling_thread_, |
| std::move(video_media_send_channel), |
| std::move(video_media_receive_channel), mid, kDefaultSrtpRequired, |
| webrtc::CryptoOptions(), context_->ssrc_generator(), transport_name); |
| auto transceiver = |
| GetOrCreateFirstTransceiverOfType(cricket::MEDIA_TYPE_VIDEO) |
| ->internal(); |
| if (transceiver->channel()) { |
| // This transceiver already has a channel, create a new one. |
| transceiver = |
| CreateTransceiverOfType(cricket::MEDIA_TYPE_VIDEO)->internal(); |
| } |
| RTC_DCHECK(!transceiver->channel()); |
| transceiver->SetChannel(std::move(video_channel), |
| [](const std::string&) { return nullptr; }); |
| video_media_send_channel_ptr->SetStats(initial_stats); |
| video_media_receive_channel_ptr->SetStats(initial_stats); |
| return std::make_pair(video_media_send_channel_ptr, |
| video_media_receive_channel_ptr); |
| } |
| |
| void AddSctpDataChannel(const std::string& label) { |
| AddSctpDataChannel(label, InternalDataChannelInit()); |
| } |
| |
| void AddSctpDataChannel(const std::string& label, |
| const InternalDataChannelInit& init) { |
| // TODO(bugs.webrtc.org/11547): Supply a separate network thread. |
| AddSctpDataChannel(SctpDataChannel::Create( |
| data_channel_controller_.weak_ptr(), label, false, init, |
| rtc::Thread::Current(), rtc::Thread::Current())); |
| } |
| |
| void AddSctpDataChannel(rtc::scoped_refptr<SctpDataChannel> data_channel) { |
| sctp_data_channels_.push_back(data_channel); |
| } |
| |
| void SetTransportStats(const std::string& transport_name, |
| const cricket::TransportChannelStats& channel_stats) { |
| SetTransportStats( |
| transport_name, |
| std::vector<cricket::TransportChannelStats>{channel_stats}); |
| } |
| |
| void SetTransportStats( |
| const std::string& transport_name, |
| const std::vector<cricket::TransportChannelStats>& channel_stats_list) { |
| cricket::TransportStats transport_stats; |
| transport_stats.transport_name = transport_name; |
| transport_stats.channel_stats = channel_stats_list; |
| transport_stats_by_name_[transport_name] = transport_stats; |
| } |
| |
| void SetCallStats(const Call::Stats& call_stats) { call_stats_ = call_stats; } |
| |
| void SetAudioDeviceStats( |
| absl::optional<AudioDeviceModule::Stats> audio_device_stats) { |
| audio_device_stats_ = audio_device_stats; |
| } |
| |
| void SetLocalCertificate( |
| const std::string& transport_name, |
| rtc::scoped_refptr<rtc::RTCCertificate> certificate) { |
| local_certificates_by_transport_[transport_name] = certificate; |
| } |
| |
| void SetRemoteCertChain(const std::string& transport_name, |
| std::unique_ptr<rtc::SSLCertChain> chain) { |
| remote_cert_chains_by_transport_[transport_name] = std::move(chain); |
| } |
| |
| // PeerConnectionInterface overrides. |
| |
| rtc::scoped_refptr<StreamCollectionInterface> local_streams() override { |
| return local_streams_; |
| } |
| |
| rtc::scoped_refptr<StreamCollectionInterface> remote_streams() override { |
| return remote_streams_; |
| } |
| |
| std::vector<rtc::scoped_refptr<RtpSenderInterface>> GetSenders() |
| const override { |
| std::vector<rtc::scoped_refptr<RtpSenderInterface>> senders; |
| for (auto transceiver : transceivers_) { |
| for (auto sender : transceiver->internal()->senders()) { |
| senders.push_back(sender); |
| } |
| } |
| return senders; |
| } |
| |
| std::vector<rtc::scoped_refptr<RtpReceiverInterface>> GetReceivers() |
| const override { |
| std::vector<rtc::scoped_refptr<RtpReceiverInterface>> receivers; |
| for (auto transceiver : transceivers_) { |
| for (auto receiver : transceiver->internal()->receivers()) { |
| receivers.push_back(receiver); |
| } |
| } |
| return receivers; |
| } |
| |
| // PeerConnectionInternal overrides. |
| |
| rtc::Thread* network_thread() const override { return network_thread_; } |
| |
| rtc::Thread* worker_thread() const override { return worker_thread_; } |
| |
| rtc::Thread* signaling_thread() const override { return signaling_thread_; } |
| |
| std::vector< |
| rtc::scoped_refptr<RtpTransceiverProxyWithInternal<RtpTransceiver>>> |
| GetTransceiversInternal() const override { |
| return transceivers_; |
| } |
| |
| std::vector<DataChannelStats> GetDataChannelStats() const override { |
| RTC_DCHECK_RUN_ON(signaling_thread()); |
| std::vector<DataChannelStats> stats; |
| for (const auto& channel : sctp_data_channels_) |
| stats.push_back(channel->GetStats()); |
| return stats; |
| } |
| |
| cricket::CandidateStatsList GetPooledCandidateStats() const override { |
| return {}; |
| } |
| |
| std::map<std::string, cricket::TransportStats> GetTransportStatsByNames( |
| const std::set<std::string>& transport_names) override { |
| RTC_DCHECK_RUN_ON(network_thread_); |
| std::map<std::string, cricket::TransportStats> transport_stats_by_name; |
| for (const std::string& transport_name : transport_names) { |
| transport_stats_by_name[transport_name] = |
| GetTransportStatsByName(transport_name); |
| } |
| return transport_stats_by_name; |
| } |
| |
| Call::Stats GetCallStats() override { return call_stats_; } |
| |
| absl::optional<AudioDeviceModule::Stats> GetAudioDeviceStats() override { |
| return audio_device_stats_; |
| } |
| |
| bool GetLocalCertificate( |
| const std::string& transport_name, |
| rtc::scoped_refptr<rtc::RTCCertificate>* certificate) override { |
| auto it = local_certificates_by_transport_.find(transport_name); |
| if (it != local_certificates_by_transport_.end()) { |
| *certificate = it->second; |
| return true; |
| } else { |
| return false; |
| } |
| } |
| |
| std::unique_ptr<rtc::SSLCertChain> GetRemoteSSLCertChain( |
| const std::string& transport_name) override { |
| auto it = remote_cert_chains_by_transport_.find(transport_name); |
| if (it != remote_cert_chains_by_transport_.end()) { |
| return it->second->Clone(); |
| } else { |
| return nullptr; |
| } |
| } |
| |
| private: |
| cricket::TransportStats GetTransportStatsByName( |
| const std::string& transport_name) { |
| auto it = transport_stats_by_name_.find(transport_name); |
| if (it != transport_stats_by_name_.end()) { |
| // If specific transport stats have been specified, return those. |
| return it->second; |
| } |
| // Otherwise, generate some dummy stats. |
| cricket::TransportChannelStats channel_stats; |
| channel_stats.component = cricket::ICE_CANDIDATE_COMPONENT_RTP; |
| cricket::TransportStats transport_stats; |
| transport_stats.transport_name = transport_name; |
| transport_stats.channel_stats.push_back(channel_stats); |
| return transport_stats; |
| } |
| |
| rtc::scoped_refptr<RtpTransceiverProxyWithInternal<RtpTransceiver>> |
| GetOrCreateFirstTransceiverOfType(cricket::MediaType media_type) { |
| for (auto transceiver : transceivers_) { |
| if (transceiver->internal()->media_type() == media_type) { |
| return transceiver; |
| } |
| } |
| return CreateTransceiverOfType(media_type); |
| } |
| |
| rtc::scoped_refptr<RtpTransceiverProxyWithInternal<RtpTransceiver>> |
| CreateTransceiverOfType(cricket::MediaType media_type) { |
| auto transceiver = RtpTransceiverProxyWithInternal<RtpTransceiver>::Create( |
| signaling_thread_, |
| rtc::make_ref_counted<RtpTransceiver>(media_type, context_.get())); |
| transceivers_.push_back(transceiver); |
| return transceiver; |
| } |
| |
| rtc::Thread* const network_thread_; |
| rtc::Thread* const worker_thread_; |
| rtc::Thread* const signaling_thread_; |
| |
| PeerConnectionFactoryDependencies dependencies_; |
| rtc::scoped_refptr<ConnectionContext> context_; |
| |
| rtc::scoped_refptr<StreamCollection> local_streams_; |
| rtc::scoped_refptr<StreamCollection> remote_streams_; |
| |
| std::vector< |
| rtc::scoped_refptr<RtpTransceiverProxyWithInternal<RtpTransceiver>>> |
| transceivers_; |
| |
| FakeDataChannelController data_channel_controller_; |
| |
| std::vector<rtc::scoped_refptr<SctpDataChannel>> sctp_data_channels_; |
| |
| std::map<std::string, cricket::TransportStats> transport_stats_by_name_; |
| |
| Call::Stats call_stats_; |
| |
| absl::optional<AudioDeviceModule::Stats> audio_device_stats_; |
| |
| std::map<std::string, rtc::scoped_refptr<rtc::RTCCertificate>> |
| local_certificates_by_transport_; |
| std::map<std::string, std::unique_ptr<rtc::SSLCertChain>> |
| remote_cert_chains_by_transport_; |
| }; |
| |
| } // namespace webrtc |
| |
| #endif // PC_TEST_FAKE_PEER_CONNECTION_FOR_STATS_H_ |