
[1] https://w3c.github.io/webrtc-stats/#certificatestats-dict* BUG=chromium:627816, chromium:629436 NOTRY=True Review-Url: https://codereview.webrtc.org/2243123002 Cr-Commit-Position: refs/heads/master@{#14484}
603 lines
23 KiB
C++
603 lines
23 KiB
C++
/*
|
|
* Copyright 2016 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.
|
|
*/
|
|
|
|
#include "webrtc/api/rtcstatscollector.h"
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "webrtc/api/jsepsessiondescription.h"
|
|
#include "webrtc/api/stats/rtcstats_objects.h"
|
|
#include "webrtc/api/stats/rtcstatsreport.h"
|
|
#include "webrtc/api/test/mock_datachannel.h"
|
|
#include "webrtc/api/test/mock_peerconnection.h"
|
|
#include "webrtc/api/test/mock_webrtcsession.h"
|
|
#include "webrtc/base/checks.h"
|
|
#include "webrtc/base/fakeclock.h"
|
|
#include "webrtc/base/fakesslidentity.h"
|
|
#include "webrtc/base/gunit.h"
|
|
#include "webrtc/base/logging.h"
|
|
#include "webrtc/base/thread_checker.h"
|
|
#include "webrtc/base/timedelta.h"
|
|
#include "webrtc/base/timeutils.h"
|
|
#include "webrtc/media/base/fakemediaengine.h"
|
|
|
|
using testing::_;
|
|
using testing::Invoke;
|
|
using testing::Return;
|
|
using testing::ReturnRef;
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
|
|
const int64_t kGetStatsReportTimeoutMs = 1000;
|
|
|
|
struct CertificateInfo {
|
|
rtc::scoped_refptr<rtc::RTCCertificate> certificate;
|
|
std::vector<std::string> ders;
|
|
std::vector<std::string> pems;
|
|
std::vector<std::string> fingerprints;
|
|
};
|
|
|
|
std::unique_ptr<CertificateInfo> CreateFakeCertificateAndInfoFromDers(
|
|
const std::vector<std::string>& ders) {
|
|
RTC_CHECK(!ders.empty());
|
|
std::unique_ptr<CertificateInfo> info(new CertificateInfo());
|
|
info->ders = ders;
|
|
for (const std::string& der : ders) {
|
|
info->pems.push_back(rtc::SSLIdentity::DerToPem(
|
|
"CERTIFICATE",
|
|
reinterpret_cast<const unsigned char*>(der.c_str()),
|
|
der.length()));
|
|
}
|
|
info->certificate =
|
|
rtc::RTCCertificate::Create(std::unique_ptr<rtc::FakeSSLIdentity>(
|
|
new rtc::FakeSSLIdentity(rtc::FakeSSLCertificate(info->pems))));
|
|
// Strip header/footer and newline characters of PEM strings.
|
|
for (size_t i = 0; i < info->pems.size(); ++i) {
|
|
rtc::replace_substrs("-----BEGIN CERTIFICATE-----", 27,
|
|
"", 0, &info->pems[i]);
|
|
rtc::replace_substrs("-----END CERTIFICATE-----", 25,
|
|
"", 0, &info->pems[i]);
|
|
rtc::replace_substrs("\n", 1,
|
|
"", 0, &info->pems[i]);
|
|
}
|
|
// Fingerprint of leaf certificate.
|
|
std::unique_ptr<rtc::SSLFingerprint> fp(
|
|
rtc::SSLFingerprint::Create("sha-1",
|
|
&info->certificate->ssl_certificate()));
|
|
EXPECT_TRUE(fp);
|
|
info->fingerprints.push_back(fp->GetRfc4572Fingerprint());
|
|
// Fingerprints of the rest of the chain.
|
|
std::unique_ptr<rtc::SSLCertChain> chain =
|
|
info->certificate->ssl_certificate().GetChain();
|
|
if (chain) {
|
|
for (size_t i = 0; i < chain->GetSize(); i++) {
|
|
fp.reset(rtc::SSLFingerprint::Create("sha-1", &chain->Get(i)));
|
|
EXPECT_TRUE(fp);
|
|
info->fingerprints.push_back(fp->GetRfc4572Fingerprint());
|
|
}
|
|
}
|
|
EXPECT_EQ(info->ders.size(), info->fingerprints.size());
|
|
return info;
|
|
}
|
|
|
|
class RTCStatsCollectorTestHelper : public SetSessionDescriptionObserver {
|
|
public:
|
|
RTCStatsCollectorTestHelper()
|
|
: worker_thread_(rtc::Thread::Current()),
|
|
network_thread_(rtc::Thread::Current()),
|
|
channel_manager_(new cricket::ChannelManager(
|
|
new cricket::FakeMediaEngine(),
|
|
worker_thread_,
|
|
network_thread_)),
|
|
media_controller_(
|
|
MediaControllerInterface::Create(cricket::MediaConfig(),
|
|
worker_thread_,
|
|
channel_manager_.get())),
|
|
session_(media_controller_.get()),
|
|
pc_() {
|
|
// Default return values for mocks.
|
|
EXPECT_CALL(pc_, session()).WillRepeatedly(Return(&session_));
|
|
EXPECT_CALL(pc_, sctp_data_channels()).WillRepeatedly(
|
|
ReturnRef(data_channels_));
|
|
EXPECT_CALL(session_, GetTransportStats(_)).WillRepeatedly(Return(false));
|
|
}
|
|
|
|
rtc::ScopedFakeClock& fake_clock() { return fake_clock_; }
|
|
MockWebRtcSession& session() { return session_; }
|
|
MockPeerConnection& pc() { return pc_; }
|
|
std::vector<rtc::scoped_refptr<DataChannel>>& data_channels() {
|
|
return data_channels_;
|
|
}
|
|
|
|
// SetSessionDescriptionObserver overrides.
|
|
void OnSuccess() override {}
|
|
void OnFailure(const std::string& error) override {
|
|
RTC_NOTREACHED() << error;
|
|
}
|
|
|
|
private:
|
|
rtc::ScopedFakeClock fake_clock_;
|
|
rtc::Thread* const worker_thread_;
|
|
rtc::Thread* const network_thread_;
|
|
std::unique_ptr<cricket::ChannelManager> channel_manager_;
|
|
std::unique_ptr<webrtc::MediaControllerInterface> media_controller_;
|
|
MockWebRtcSession session_;
|
|
MockPeerConnection pc_;
|
|
|
|
std::vector<rtc::scoped_refptr<DataChannel>> data_channels_;
|
|
};
|
|
|
|
class RTCTestStats : public RTCStats {
|
|
public:
|
|
RTCTestStats(const std::string& id, int64_t timestamp_us)
|
|
: RTCStats(id, timestamp_us),
|
|
dummy_stat("dummyStat") {}
|
|
|
|
WEBRTC_RTCSTATS_IMPL(RTCStats, RTCTestStats,
|
|
&dummy_stat);
|
|
|
|
RTCStatsMember<int32_t> dummy_stat;
|
|
};
|
|
|
|
const char RTCTestStats::kType[] = "test-stats";
|
|
|
|
// Overrides the stats collection to verify thread usage and that the resulting
|
|
// partial reports are merged.
|
|
class FakeRTCStatsCollector : public RTCStatsCollector,
|
|
public RTCStatsCollectorCallback {
|
|
public:
|
|
static rtc::scoped_refptr<FakeRTCStatsCollector> Create(
|
|
PeerConnection* pc,
|
|
int64_t cache_lifetime_us) {
|
|
return rtc::scoped_refptr<FakeRTCStatsCollector>(
|
|
new rtc::RefCountedObject<FakeRTCStatsCollector>(
|
|
pc, cache_lifetime_us));
|
|
}
|
|
|
|
// RTCStatsCollectorCallback implementation.
|
|
void OnStatsDelivered(
|
|
const rtc::scoped_refptr<const RTCStatsReport>& report) override {
|
|
EXPECT_TRUE(signaling_thread_->IsCurrent());
|
|
rtc::CritScope cs(&lock_);
|
|
delivered_report_ = report;
|
|
}
|
|
|
|
void VerifyThreadUsageAndResultsMerging() {
|
|
GetStatsReport(rtc::scoped_refptr<RTCStatsCollectorCallback>(this));
|
|
EXPECT_TRUE_WAIT(HasVerifiedResults(), kGetStatsReportTimeoutMs);
|
|
}
|
|
|
|
bool HasVerifiedResults() {
|
|
EXPECT_TRUE(signaling_thread_->IsCurrent());
|
|
rtc::CritScope cs(&lock_);
|
|
if (!delivered_report_)
|
|
return false;
|
|
EXPECT_EQ(produced_on_signaling_thread_, 1);
|
|
EXPECT_EQ(produced_on_worker_thread_, 1);
|
|
EXPECT_EQ(produced_on_network_thread_, 1);
|
|
|
|
EXPECT_TRUE(delivered_report_->Get("SignalingThreadStats"));
|
|
EXPECT_TRUE(delivered_report_->Get("WorkerThreadStats"));
|
|
EXPECT_TRUE(delivered_report_->Get("NetworkThreadStats"));
|
|
|
|
produced_on_signaling_thread_ = 0;
|
|
produced_on_worker_thread_ = 0;
|
|
produced_on_network_thread_ = 0;
|
|
delivered_report_ = nullptr;
|
|
return true;
|
|
}
|
|
|
|
protected:
|
|
FakeRTCStatsCollector(
|
|
PeerConnection* pc,
|
|
int64_t cache_lifetime)
|
|
: RTCStatsCollector(pc, cache_lifetime),
|
|
signaling_thread_(pc->session()->signaling_thread()),
|
|
worker_thread_(pc->session()->worker_thread()),
|
|
network_thread_(pc->session()->network_thread()) {
|
|
}
|
|
|
|
void ProducePartialResultsOnSignalingThread(int64_t timestamp_us) override {
|
|
EXPECT_TRUE(signaling_thread_->IsCurrent());
|
|
{
|
|
rtc::CritScope cs(&lock_);
|
|
EXPECT_FALSE(delivered_report_);
|
|
++produced_on_signaling_thread_;
|
|
}
|
|
|
|
rtc::scoped_refptr<RTCStatsReport> signaling_report =
|
|
RTCStatsReport::Create();
|
|
signaling_report->AddStats(std::unique_ptr<const RTCStats>(
|
|
new RTCTestStats("SignalingThreadStats", timestamp_us)));
|
|
AddPartialResults(signaling_report);
|
|
}
|
|
void ProducePartialResultsOnWorkerThread(int64_t timestamp_us) override {
|
|
EXPECT_TRUE(worker_thread_->IsCurrent());
|
|
{
|
|
rtc::CritScope cs(&lock_);
|
|
EXPECT_FALSE(delivered_report_);
|
|
++produced_on_worker_thread_;
|
|
}
|
|
|
|
rtc::scoped_refptr<RTCStatsReport> worker_report = RTCStatsReport::Create();
|
|
worker_report->AddStats(std::unique_ptr<const RTCStats>(
|
|
new RTCTestStats("WorkerThreadStats", timestamp_us)));
|
|
AddPartialResults(worker_report);
|
|
}
|
|
void ProducePartialResultsOnNetworkThread(int64_t timestamp_us) override {
|
|
EXPECT_TRUE(network_thread_->IsCurrent());
|
|
{
|
|
rtc::CritScope cs(&lock_);
|
|
EXPECT_FALSE(delivered_report_);
|
|
++produced_on_network_thread_;
|
|
}
|
|
|
|
rtc::scoped_refptr<RTCStatsReport> network_report =
|
|
RTCStatsReport::Create();
|
|
network_report->AddStats(std::unique_ptr<const RTCStats>(
|
|
new RTCTestStats("NetworkThreadStats", timestamp_us)));
|
|
AddPartialResults(network_report);
|
|
}
|
|
|
|
private:
|
|
rtc::Thread* const signaling_thread_;
|
|
rtc::Thread* const worker_thread_;
|
|
rtc::Thread* const network_thread_;
|
|
|
|
rtc::CriticalSection lock_;
|
|
rtc::scoped_refptr<const RTCStatsReport> delivered_report_;
|
|
int produced_on_signaling_thread_ = 0;
|
|
int produced_on_worker_thread_ = 0;
|
|
int produced_on_network_thread_ = 0;
|
|
};
|
|
|
|
class StatsCallback : public RTCStatsCollectorCallback {
|
|
public:
|
|
static rtc::scoped_refptr<StatsCallback> Create(
|
|
rtc::scoped_refptr<const RTCStatsReport>* report_ptr = nullptr) {
|
|
return rtc::scoped_refptr<StatsCallback>(
|
|
new rtc::RefCountedObject<StatsCallback>(report_ptr));
|
|
}
|
|
|
|
void OnStatsDelivered(
|
|
const rtc::scoped_refptr<const RTCStatsReport>& report) override {
|
|
EXPECT_TRUE(thread_checker_.CalledOnValidThread());
|
|
report_ = report;
|
|
if (report_ptr_)
|
|
*report_ptr_ = report_;
|
|
}
|
|
|
|
rtc::scoped_refptr<const RTCStatsReport> report() const {
|
|
EXPECT_TRUE(thread_checker_.CalledOnValidThread());
|
|
return report_;
|
|
}
|
|
|
|
protected:
|
|
explicit StatsCallback(rtc::scoped_refptr<const RTCStatsReport>* report_ptr)
|
|
: report_ptr_(report_ptr) {}
|
|
|
|
private:
|
|
rtc::ThreadChecker thread_checker_;
|
|
rtc::scoped_refptr<const RTCStatsReport> report_;
|
|
rtc::scoped_refptr<const RTCStatsReport>* report_ptr_;
|
|
};
|
|
|
|
class RTCStatsCollectorTest : public testing::Test {
|
|
public:
|
|
RTCStatsCollectorTest()
|
|
: test_(new rtc::RefCountedObject<RTCStatsCollectorTestHelper>()),
|
|
collector_(RTCStatsCollector::Create(
|
|
&test_->pc(), 50 * rtc::kNumMicrosecsPerMillisec)) {
|
|
}
|
|
|
|
rtc::scoped_refptr<const RTCStatsReport> GetStatsReport() {
|
|
rtc::scoped_refptr<StatsCallback> callback = StatsCallback::Create();
|
|
collector_->GetStatsReport(callback);
|
|
EXPECT_TRUE_WAIT(callback->report(), kGetStatsReportTimeoutMs);
|
|
return callback->report();
|
|
}
|
|
|
|
void ExpectReportContainsCertificateInfo(
|
|
const rtc::scoped_refptr<const RTCStatsReport>& report,
|
|
const CertificateInfo& cert_info) {
|
|
for (size_t i = 0; i < cert_info.fingerprints.size(); ++i) {
|
|
const RTCStats* stats = report->Get(
|
|
"RTCCertificate_" + cert_info.fingerprints[i]);
|
|
EXPECT_TRUE(stats);
|
|
const RTCCertificateStats& cert_stats =
|
|
stats->cast_to<const RTCCertificateStats>();
|
|
EXPECT_EQ(*cert_stats.fingerprint, cert_info.fingerprints[i]);
|
|
EXPECT_EQ(*cert_stats.fingerprint_algorithm, "sha-1");
|
|
EXPECT_EQ(*cert_stats.base64_certificate, cert_info.pems[i]);
|
|
if (i + 1 < cert_info.fingerprints.size()) {
|
|
EXPECT_EQ(*cert_stats.issuer_certificate_id,
|
|
"RTCCertificate_" + cert_info.fingerprints[i + 1]);
|
|
} else {
|
|
EXPECT_FALSE(cert_stats.issuer_certificate_id.is_defined());
|
|
}
|
|
}
|
|
}
|
|
|
|
protected:
|
|
rtc::scoped_refptr<RTCStatsCollectorTestHelper> test_;
|
|
rtc::scoped_refptr<RTCStatsCollector> collector_;
|
|
};
|
|
|
|
TEST_F(RTCStatsCollectorTest, SingleCallback) {
|
|
rtc::scoped_refptr<const RTCStatsReport> result;
|
|
collector_->GetStatsReport(StatsCallback::Create(&result));
|
|
EXPECT_TRUE_WAIT(result, kGetStatsReportTimeoutMs);
|
|
}
|
|
|
|
TEST_F(RTCStatsCollectorTest, MultipleCallbacks) {
|
|
rtc::scoped_refptr<const RTCStatsReport> a;
|
|
rtc::scoped_refptr<const RTCStatsReport> b;
|
|
rtc::scoped_refptr<const RTCStatsReport> c;
|
|
collector_->GetStatsReport(StatsCallback::Create(&a));
|
|
collector_->GetStatsReport(StatsCallback::Create(&b));
|
|
collector_->GetStatsReport(StatsCallback::Create(&c));
|
|
EXPECT_TRUE_WAIT(a, kGetStatsReportTimeoutMs);
|
|
EXPECT_TRUE_WAIT(b, kGetStatsReportTimeoutMs);
|
|
EXPECT_TRUE_WAIT(c, kGetStatsReportTimeoutMs);
|
|
EXPECT_EQ(a.get(), b.get());
|
|
EXPECT_EQ(b.get(), c.get());
|
|
}
|
|
|
|
TEST_F(RTCStatsCollectorTest, CachedStatsReports) {
|
|
// Caching should ensure |a| and |b| are the same report.
|
|
rtc::scoped_refptr<const RTCStatsReport> a = GetStatsReport();
|
|
rtc::scoped_refptr<const RTCStatsReport> b = GetStatsReport();
|
|
EXPECT_EQ(a.get(), b.get());
|
|
// Invalidate cache by clearing it.
|
|
collector_->ClearCachedStatsReport();
|
|
rtc::scoped_refptr<const RTCStatsReport> c = GetStatsReport();
|
|
EXPECT_NE(b.get(), c.get());
|
|
// Invalidate cache by advancing time.
|
|
test_->fake_clock().AdvanceTime(rtc::TimeDelta::FromMilliseconds(51));
|
|
rtc::scoped_refptr<const RTCStatsReport> d = GetStatsReport();
|
|
EXPECT_TRUE(d);
|
|
EXPECT_NE(c.get(), d.get());
|
|
}
|
|
|
|
TEST_F(RTCStatsCollectorTest, MultipleCallbacksWithInvalidatedCacheInBetween) {
|
|
rtc::scoped_refptr<const RTCStatsReport> a;
|
|
rtc::scoped_refptr<const RTCStatsReport> b;
|
|
rtc::scoped_refptr<const RTCStatsReport> c;
|
|
collector_->GetStatsReport(StatsCallback::Create(&a));
|
|
collector_->GetStatsReport(StatsCallback::Create(&b));
|
|
// Cache is invalidated after 50 ms.
|
|
test_->fake_clock().AdvanceTime(rtc::TimeDelta::FromMilliseconds(51));
|
|
collector_->GetStatsReport(StatsCallback::Create(&c));
|
|
EXPECT_TRUE_WAIT(a, kGetStatsReportTimeoutMs);
|
|
EXPECT_TRUE_WAIT(b, kGetStatsReportTimeoutMs);
|
|
EXPECT_TRUE_WAIT(c, kGetStatsReportTimeoutMs);
|
|
EXPECT_EQ(a.get(), b.get());
|
|
// The act of doing |AdvanceTime| processes all messages. If this was not the
|
|
// case we might not require |c| to be fresher than |b|.
|
|
EXPECT_NE(c.get(), b.get());
|
|
}
|
|
|
|
TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsSingle) {
|
|
std::unique_ptr<CertificateInfo> local_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(
|
|
std::vector<std::string>({ "(local) single certificate" }));
|
|
std::unique_ptr<CertificateInfo> remote_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(
|
|
std::vector<std::string>({ "(remote) single certificate" }));
|
|
|
|
// Mock the session to return the local and remote certificates.
|
|
EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
|
[this](SessionStats* stats) {
|
|
stats->transport_stats["transport"].transport_name = "transport";
|
|
return true;
|
|
}));
|
|
EXPECT_CALL(test_->session(), GetLocalCertificate(_, _)).WillRepeatedly(
|
|
Invoke([this, &local_certinfo](const std::string& transport_name,
|
|
rtc::scoped_refptr<rtc::RTCCertificate>* certificate) {
|
|
if (transport_name == "transport") {
|
|
*certificate = local_certinfo->certificate;
|
|
return true;
|
|
}
|
|
return false;
|
|
}));
|
|
EXPECT_CALL(test_->session(),
|
|
GetRemoteSSLCertificate_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
|
[this, &remote_certinfo](const std::string& transport_name) {
|
|
if (transport_name == "transport")
|
|
return remote_certinfo->certificate->ssl_certificate().GetReference();
|
|
return static_cast<rtc::SSLCertificate*>(nullptr);
|
|
}));
|
|
|
|
rtc::scoped_refptr<const RTCStatsReport> report = GetStatsReport();
|
|
ExpectReportContainsCertificateInfo(report, *local_certinfo.get());
|
|
ExpectReportContainsCertificateInfo(report, *remote_certinfo.get());
|
|
}
|
|
|
|
TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsMultiple) {
|
|
std::unique_ptr<CertificateInfo> audio_local_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(
|
|
std::vector<std::string>({ "(local) audio" }));
|
|
audio_local_certinfo = CreateFakeCertificateAndInfoFromDers(
|
|
audio_local_certinfo->ders);
|
|
std::unique_ptr<CertificateInfo> audio_remote_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(
|
|
std::vector<std::string>({ "(remote) audio" }));
|
|
audio_remote_certinfo = CreateFakeCertificateAndInfoFromDers(
|
|
audio_remote_certinfo->ders);
|
|
|
|
std::unique_ptr<CertificateInfo> video_local_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(
|
|
std::vector<std::string>({ "(local) video" }));
|
|
video_local_certinfo = CreateFakeCertificateAndInfoFromDers(
|
|
video_local_certinfo->ders);
|
|
std::unique_ptr<CertificateInfo> video_remote_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(
|
|
std::vector<std::string>({ "(remote) video" }));
|
|
video_remote_certinfo = CreateFakeCertificateAndInfoFromDers(
|
|
video_remote_certinfo->ders);
|
|
|
|
// Mock the session to return the local and remote certificates.
|
|
EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
|
[this](SessionStats* stats) {
|
|
stats->transport_stats["audio"].transport_name = "audio";
|
|
stats->transport_stats["video"].transport_name = "video";
|
|
return true;
|
|
}));
|
|
EXPECT_CALL(test_->session(), GetLocalCertificate(_, _)).WillRepeatedly(
|
|
Invoke([this, &audio_local_certinfo, &video_local_certinfo](
|
|
const std::string& transport_name,
|
|
rtc::scoped_refptr<rtc::RTCCertificate>* certificate) {
|
|
if (transport_name == "audio") {
|
|
*certificate = audio_local_certinfo->certificate;
|
|
return true;
|
|
}
|
|
if (transport_name == "video") {
|
|
*certificate = video_local_certinfo->certificate;
|
|
return true;
|
|
}
|
|
return false;
|
|
}));
|
|
EXPECT_CALL(test_->session(),
|
|
GetRemoteSSLCertificate_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
|
[this, &audio_remote_certinfo, &video_remote_certinfo](
|
|
const std::string& transport_name) {
|
|
if (transport_name == "audio") {
|
|
return audio_remote_certinfo->certificate->ssl_certificate()
|
|
.GetReference();
|
|
}
|
|
if (transport_name == "video") {
|
|
return video_remote_certinfo->certificate->ssl_certificate()
|
|
.GetReference();
|
|
}
|
|
return static_cast<rtc::SSLCertificate*>(nullptr);
|
|
}));
|
|
|
|
rtc::scoped_refptr<const RTCStatsReport> report = GetStatsReport();
|
|
ExpectReportContainsCertificateInfo(report, *audio_local_certinfo.get());
|
|
ExpectReportContainsCertificateInfo(report, *audio_remote_certinfo.get());
|
|
ExpectReportContainsCertificateInfo(report, *video_local_certinfo.get());
|
|
ExpectReportContainsCertificateInfo(report, *video_remote_certinfo.get());
|
|
}
|
|
|
|
TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsChain) {
|
|
std::vector<std::string> local_ders;
|
|
local_ders.push_back("(local) this");
|
|
local_ders.push_back("(local) is");
|
|
local_ders.push_back("(local) a");
|
|
local_ders.push_back("(local) chain");
|
|
std::unique_ptr<CertificateInfo> local_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(local_ders);
|
|
std::vector<std::string> remote_ders;
|
|
remote_ders.push_back("(remote) this");
|
|
remote_ders.push_back("(remote) is");
|
|
remote_ders.push_back("(remote) another");
|
|
remote_ders.push_back("(remote) chain");
|
|
std::unique_ptr<CertificateInfo> remote_certinfo =
|
|
CreateFakeCertificateAndInfoFromDers(remote_ders);
|
|
|
|
// Mock the session to return the local and remote certificates.
|
|
EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
|
[this](SessionStats* stats) {
|
|
stats->transport_stats["transport"].transport_name = "transport";
|
|
return true;
|
|
}));
|
|
EXPECT_CALL(test_->session(), GetLocalCertificate(_, _)).WillRepeatedly(
|
|
Invoke([this, &local_certinfo](const std::string& transport_name,
|
|
rtc::scoped_refptr<rtc::RTCCertificate>* certificate) {
|
|
if (transport_name == "transport") {
|
|
*certificate = local_certinfo->certificate;
|
|
return true;
|
|
}
|
|
return false;
|
|
}));
|
|
EXPECT_CALL(test_->session(),
|
|
GetRemoteSSLCertificate_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
|
[this, &remote_certinfo](const std::string& transport_name) {
|
|
if (transport_name == "transport")
|
|
return remote_certinfo->certificate->ssl_certificate().GetReference();
|
|
return static_cast<rtc::SSLCertificate*>(nullptr);
|
|
}));
|
|
|
|
rtc::scoped_refptr<const RTCStatsReport> report = GetStatsReport();
|
|
ExpectReportContainsCertificateInfo(report, *local_certinfo.get());
|
|
ExpectReportContainsCertificateInfo(report, *remote_certinfo.get());
|
|
}
|
|
|
|
TEST_F(RTCStatsCollectorTest, CollectRTCPeerConnectionStats) {
|
|
int64_t before = rtc::TimeUTCMicros();
|
|
rtc::scoped_refptr<const RTCStatsReport> report = GetStatsReport();
|
|
int64_t after = rtc::TimeUTCMicros();
|
|
EXPECT_EQ(report->GetStatsOfType<RTCPeerConnectionStats>().size(),
|
|
static_cast<size_t>(1)) << "Expecting 1 RTCPeerConnectionStats.";
|
|
const RTCStats* stats = report->Get("RTCPeerConnection");
|
|
EXPECT_TRUE(stats);
|
|
EXPECT_LE(before, stats->timestamp_us());
|
|
EXPECT_LE(stats->timestamp_us(), after);
|
|
{
|
|
// Expected stats with no data channels
|
|
const RTCPeerConnectionStats& pcstats =
|
|
stats->cast_to<RTCPeerConnectionStats>();
|
|
EXPECT_EQ(*pcstats.data_channels_opened, static_cast<uint32_t>(0));
|
|
EXPECT_EQ(*pcstats.data_channels_closed, static_cast<uint32_t>(0));
|
|
}
|
|
|
|
test_->data_channels().push_back(
|
|
new MockDataChannel(DataChannelInterface::kConnecting));
|
|
test_->data_channels().push_back(
|
|
new MockDataChannel(DataChannelInterface::kOpen));
|
|
test_->data_channels().push_back(
|
|
new MockDataChannel(DataChannelInterface::kClosing));
|
|
test_->data_channels().push_back(
|
|
new MockDataChannel(DataChannelInterface::kClosed));
|
|
|
|
collector_->ClearCachedStatsReport();
|
|
report = GetStatsReport();
|
|
EXPECT_EQ(report->GetStatsOfType<RTCPeerConnectionStats>().size(),
|
|
static_cast<size_t>(1)) << "Expecting 1 RTCPeerConnectionStats.";
|
|
stats = report->Get("RTCPeerConnection");
|
|
EXPECT_TRUE(stats);
|
|
{
|
|
// Expected stats with the above four data channels
|
|
// TODO(hbos): When the |RTCPeerConnectionStats| is the number of data
|
|
// channels that have been opened and closed, not the numbers currently
|
|
// open/closed, we would expect opened >= closed and (opened - closed) to be
|
|
// the number currently open. crbug.com/636818.
|
|
const RTCPeerConnectionStats& pcstats =
|
|
stats->cast_to<RTCPeerConnectionStats>();
|
|
EXPECT_EQ(*pcstats.data_channels_opened, static_cast<uint32_t>(1));
|
|
EXPECT_EQ(*pcstats.data_channels_closed, static_cast<uint32_t>(3));
|
|
}
|
|
}
|
|
|
|
class RTCStatsCollectorTestWithFakeCollector : public testing::Test {
|
|
public:
|
|
RTCStatsCollectorTestWithFakeCollector()
|
|
: test_(new rtc::RefCountedObject<RTCStatsCollectorTestHelper>()),
|
|
collector_(FakeRTCStatsCollector::Create(
|
|
&test_->pc(), 50 * rtc::kNumMicrosecsPerMillisec)) {
|
|
}
|
|
|
|
protected:
|
|
rtc::scoped_refptr<RTCStatsCollectorTestHelper> test_;
|
|
rtc::scoped_refptr<FakeRTCStatsCollector> collector_;
|
|
};
|
|
|
|
TEST_F(RTCStatsCollectorTestWithFakeCollector, ThreadUsageAndResultsMerging) {
|
|
collector_->VerifyThreadUsageAndResultsMerging();
|
|
}
|
|
|
|
} // namespace
|
|
|
|
} // namespace webrtc
|