Index: webrtc/api/rtcstatscollector.cc |
diff --git a/webrtc/api/rtcstatscollector.cc b/webrtc/api/rtcstatscollector.cc |
index 5f60ea13cecadb7defc5169701e06a2d7308dc26..27e6fcab943c8fb622487fabae2fab4302c3686b 100644 |
--- a/webrtc/api/rtcstatscollector.cc |
+++ b/webrtc/api/rtcstatscollector.cc |
@@ -23,6 +23,15 @@ |
namespace webrtc { |
+namespace { |
+ |
+std::string RTCIceCandidatePairStatsID(const cricket::ConnectionInfo& info) { |
+ return "RTCIceCandidatePair_" + info.local_candidate.id() + "_" + |
+ info.remote_candidate.id(); |
+} |
+ |
+} // namespace |
+ |
const char* CandidateTypeToRTCIceCandidateType(const std::string& type) { |
if (type == cricket::LOCAL_PORT_TYPE) |
return RTCIceCandidateType::kHost; |
@@ -107,9 +116,12 @@ void RTCStatsCollector::ProducePartialResultsOnSignalingThread( |
SessionStats session_stats; |
if (pc_->session()->GetTransportStats(&session_stats)) { |
- ProduceCertificateStats_s(timestamp_us, session_stats, report.get()); |
- ProduceIceCandidateAndPairStats_s(timestamp_us, session_stats, |
- report.get()); |
+ std::map<std::string, CertificateStatsPair> transport_certificates = |
+ ProduceCertificateStats_s(timestamp_us, session_stats, report.get()); |
+ ProduceIceCandidateAndPairStats_s( |
+ timestamp_us, session_stats, report.get()); |
+ ProduceTransportStats_s( |
+ timestamp_us, session_stats, transport_certificates, report.get()); |
} |
ProducePeerConnectionStats_s(timestamp_us, report.get()); |
@@ -176,46 +188,63 @@ void RTCStatsCollector::DeliverCachedReport() { |
callbacks_.clear(); |
} |
-void RTCStatsCollector::ProduceCertificateStats_s( |
+std::map<std::string, RTCStatsCollector::CertificateStatsPair> |
+RTCStatsCollector::ProduceCertificateStats_s( |
hta-webrtc
2016/10/12 22:52:35
I don't like this function. It's doing two things,
hbos
2016/10/17 19:33:13
Yes, that's much nicer. Done.
|
int64_t timestamp_us, const SessionStats& session_stats, |
RTCStatsReport* report) const { |
RTC_DCHECK(signaling_thread_->IsCurrent()); |
+ std::map<std::string, CertificateStatsPair> transport_certificates; |
for (const auto& transport_stats : session_stats.transport_stats) { |
+ CertificateStatsPair certificate_stats_pair; |
rtc::scoped_refptr<rtc::RTCCertificate> local_certificate; |
if (pc_->session()->GetLocalCertificate( |
transport_stats.second.transport_name, &local_certificate)) { |
- ProduceCertificateStatsFromSSLCertificateAndChain_s( |
- timestamp_us, local_certificate->ssl_certificate(), report); |
+ certificate_stats_pair.local = |
+ ProduceCertificateStatsFromSSLCertificateAndChain_s( |
+ timestamp_us, local_certificate->ssl_certificate(), report); |
} |
std::unique_ptr<rtc::SSLCertificate> remote_certificate = |
pc_->session()->GetRemoteSSLCertificate( |
transport_stats.second.transport_name); |
if (remote_certificate) { |
- ProduceCertificateStatsFromSSLCertificateAndChain_s( |
- timestamp_us, *remote_certificate.get(), report); |
+ certificate_stats_pair.remote = |
+ ProduceCertificateStatsFromSSLCertificateAndChain_s( |
+ timestamp_us, *remote_certificate.get(), report); |
+ } |
+ if (certificate_stats_pair.local && certificate_stats_pair.remote) { |
+ transport_certificates.insert( |
+ std::make_pair(transport_stats.second.transport_name, |
+ std::move(certificate_stats_pair))); |
} |
} |
+ return transport_certificates; |
} |
-void RTCStatsCollector::ProduceCertificateStatsFromSSLCertificateAndChain_s( |
+RTCCertificateStats* |
+RTCStatsCollector::ProduceCertificateStatsFromSSLCertificateAndChain_s( |
hta-webrtc
2016/10/12 22:52:35
This one too is now dual-function.
hbos
2016/10/17 19:33:13
Done.
|
int64_t timestamp_us, const rtc::SSLCertificate& certificate, |
RTCStatsReport* report) const { |
RTC_DCHECK(signaling_thread_->IsCurrent()); |
std::unique_ptr<rtc::SSLCertificateStats> ssl_stats = |
certificate.GetStats(); |
- RTCCertificateStats* prev_stats = nullptr; |
+ RTCCertificateStats* first_certificate_stats = nullptr; |
+ RTCCertificateStats* prev_certificate_stats = nullptr; |
for (rtc::SSLCertificateStats* s = ssl_stats.get(); s; |
s = s->issuer.get()) { |
- RTCCertificateStats* stats = new RTCCertificateStats( |
+ RTCCertificateStats* certificate_stats = new RTCCertificateStats( |
"RTCCertificate_" + s->fingerprint, timestamp_us); |
- stats->fingerprint = s->fingerprint; |
- stats->fingerprint_algorithm = s->fingerprint_algorithm; |
- stats->base64_certificate = s->base64_certificate; |
- if (prev_stats) |
- prev_stats->issuer_certificate_id = stats->id(); |
- report->AddStats(std::unique_ptr<RTCCertificateStats>(stats)); |
- prev_stats = stats; |
+ certificate_stats->fingerprint = s->fingerprint; |
+ certificate_stats->fingerprint_algorithm = s->fingerprint_algorithm; |
+ certificate_stats->base64_certificate = s->base64_certificate; |
+ if (prev_certificate_stats) |
+ prev_certificate_stats->issuer_certificate_id = certificate_stats->id(); |
+ report->AddStats(std::unique_ptr<RTCCertificateStats>(certificate_stats)); |
+ if (!first_certificate_stats) |
+ first_certificate_stats = certificate_stats; |
+ prev_certificate_stats = certificate_stats; |
} |
+ RTC_DCHECK(first_certificate_stats); |
+ return first_certificate_stats; |
} |
void RTCStatsCollector::ProduceIceCandidateAndPairStats_s( |
@@ -226,10 +255,9 @@ void RTCStatsCollector::ProduceIceCandidateAndPairStats_s( |
for (const auto& channel_stats : transport_stats.second.channel_stats) { |
for (const cricket::ConnectionInfo& info : |
channel_stats.connection_infos) { |
- const std::string& id = "RTCIceCandidatePair_" + |
- info.local_candidate.id() + "_" + info.remote_candidate.id(); |
std::unique_ptr<RTCIceCandidatePairStats> candidate_pair_stats( |
- new RTCIceCandidatePairStats(id, timestamp_us)); |
+ new RTCIceCandidatePairStats(RTCIceCandidatePairStatsID(info), |
+ timestamp_us)); |
// TODO(hbos): Set all of the |RTCIceCandidatePairStats|'s members, |
// crbug.com/633550. |
@@ -337,4 +365,40 @@ void RTCStatsCollector::ProducePeerConnectionStats_s( |
report->AddStats(std::move(stats)); |
} |
+void RTCStatsCollector::ProduceTransportStats_s( |
+ int64_t timestamp_us, const SessionStats& session_stats, |
+ const std::map<std::string, CertificateStatsPair>& transport_certificates, |
+ RTCStatsReport* report) const { |
+ RTC_DCHECK(signaling_thread_->IsCurrent()); |
+ for (const auto& transport : session_stats.transport_stats) { |
+ std::unique_ptr<RTCTransportStats> transport_stats( |
+ new RTCTransportStats("RTCTransport_" + transport.second.transport_name, |
+ timestamp_us)); |
+ transport_stats->bytes_sent = 0; |
+ transport_stats->bytes_received = 0; |
+ transport_stats->active_connection = false; |
+ for (const auto& channel_stats : transport.second.channel_stats) { |
+ for (const cricket::ConnectionInfo& info : |
+ channel_stats.connection_infos) { |
+ *transport_stats->bytes_sent += info.sent_total_bytes; |
+ *transport_stats->bytes_received += info.recv_total_bytes; |
+ if (info.best_connection) { |
+ transport_stats->active_connection = true; |
+ transport_stats->selected_candidate_pair_id = |
+ RTCIceCandidatePairStatsID(info); |
+ } |
+ } |
+ } |
+ // TODO(hbos): Set transport_stats->rtcp_transport_stats_id. |
+ // crbug.com/653873 |
+ const auto& it = transport_certificates.find( |
+ transport.second.transport_name); |
+ if (it != transport_certificates.end()) { |
+ transport_stats->local_certificate_id = it->second.local->id(); |
+ transport_stats->remote_certificate_id = it->second.remote->id();; |
+ } |
+ report->AddStats(std::move(transport_stats)); |
+ } |
+} |
+ |
} // namespace webrtc |