Certificate Transparency: Start tracking logs' state

chrome/browser/profiles/profile_io_data.cc

Index: chrome/browser/profiles/profile_io_data.cc
diff --git a/chrome/browser/profiles/profile_io_data.cc b/chrome/browser/profiles/profile_io_data.cc
index c4e0cd99ea99340dd0a1c09db223a65b0e501053..4bb0b33866fb7153407a214aa97ea989a05745ed 100644
--- a/chrome/browser/profiles/profile_io_data.cc
+++ b/chrome/browser/profiles/profile_io_data.cc
@@ -4,7 +4,6 @@
 
 #include "chrome/browser/profiles/profile_io_data.h"
 
-#include <stddef.h>
 #include <string>
 #include <utility>
 
@@ -56,6 +55,7 @@
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "components/about_handler/about_protocol_handler.h"
+#include "components/certificate_transparency/tree_state_tracker.h"
 #include "components/content_settings/core/browser/content_settings_provider.h"
 #include "components/content_settings/core/browser/cookie_settings.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
@@ -1133,7 +1133,24 @@ void ProfileIOData::Init(
       new net::MultiLogCTVerifier());
   ct_verifier->AddLogs(io_thread_globals->ct_logs);
   main_request_context_->set_cert_transparency_verifier(ct_verifier.get());
+
+  VLOG(1) << "Creating TreeStateTracker observer on ProfileIOData.";
+
+  scoped_ptr<certificate_transparency::TreeStateTracker> scts_observer(
+      new certificate_transparency::TreeStateTracker(
+          io_thread_globals->ct_logs));
+  ct_verifier->SetObserver(scts_observer.get());
   cert_transparency_verifier_ = std::move(ct_verifier);
+  cert_transparency_observer_ = std::move(scts_observer);
+  VLOG(1) << "XXX Registering observer to the STHReporter.";
+
+  io_thread->RegisterSTHObserver(cert_transparency_observer_.get());
+
+  cert_transparency_observer_unregistration_ =
+      base::Bind(&IOThread::UnregisterSTHObserver, base::Unretained(io_thread),
+                 cert_transparency_observer_.get());
+
+  VLOG(1) << "TreeStateTracker observer on ProfileIOData created.";
 
   InitializeInternal(std::move(network_delegate), profile_params_.get(),
                      protocol_handlers, std::move(request_interceptors));
@@ -1280,6 +1297,12 @@ void ProfileIOData::set_channel_id_service(
 
 void ProfileIOData::DestroyResourceContext() {
   resource_context_.reset();
+  // Stop any new URLRequests from being made by the cert_transparency_observer_
+  // by stopping notifications to it from the CTVerifier.
+  cert_transparency_verifier_->SetObserver(nullptr);
+  cert_transparency_observer_unregistration_.Run();
+
+  VLOG(1) << "XXX: ProfileIOData::DestroyResourceContext";
 }
 
 scoped_ptr<net::HttpNetworkSession> ProfileIOData::CreateHttpNetworkSession(