Certificate Transparency: Start tracking logs' state

chrome/browser/profiles/profile_io_data.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/profiles/profile_io_data.h"
 
-#include <stddef.h>
 #include <string>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/debug/alias.h"
 #include "base/logging.h"
@@ skipping 31 matching lines @@
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ssl/chrome_expect_ct_reporter.h"
 #include "chrome/browser/ui/search/new_tab_page_interceptor_service.h"
 #include "chrome/browser/ui/search/new_tab_page_interceptor_service_factory.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/features.h"
 #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"
 #include "components/cookie_config/cookie_store_util.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_io_data.h"
 #include "components/dom_distiller/core/url_constants.h"
 #include "components/metrics/metrics_pref_names.h"
 #include "components/net_log/chrome_net_log.h"
 #include "components/policy/core/browser/url_blacklist_manager.h"
 #include "components/policy/core/common/cloud/policy_header_io_helper.h"
@@ skipping 1057 matching lines @@
   // Install the New Tab Page Interceptor.
   if (profile_params_->new_tab_page_interceptor.get()) {
     request_interceptors.push_back(
         profile_params_->new_tab_page_interceptor.release());
   }
 
   scoped_ptr<net::MultiLogCTVerifier> ct_verifier(
       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.";
+
+  /*
+  if (!IsOffTheRecord()) {
+    VLOG(1) << "New Disk-based TreeStateStorage in "
+            << profile_params_->path.AsUTF8Unsafe();
+    tree_storage.reset(new certificate_transparency::DiskTreeStateStorage(
+        profile_params_->path.Append("CertificateTransparency")));
+  } else {
+    VLOG(1) << "New Memory-based TreeStateStorage.";
+    tree_storage.reset(new certificate_transparency::MemoryTreeStateStorage());
+  }
+  */
+
+  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_globals->sth_observers_registrar.SignUpForRegistration(
+      cert_transparency_observer_.get());
+  cert_transparency_observer_unregistration_ = base::Bind(
+      &net::ct::STHReporterRegistrar::UnregisterObserver,
+      base::Unretained(&(io_thread_globals->sth_observers_registrar)),
+      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));
 
   profile_params_.reset();
   initialized_ = true;
 }
 
 void ProfileIOData::ApplyProfileParamsToContext(
     net::URLRequestContext* context) const {
@@ skipping 126 matching lines @@
     delete this;
 }
 
 void ProfileIOData::set_channel_id_service(
     net::ChannelIDService* channel_id_service) const {
   channel_id_service_.reset(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(
     const ProfileParams& profile_params) const {
   net::HttpNetworkSession::Params params;
   net::URLRequestContext* context = main_request_context();
 
   IOThread* const io_thread = profile_params.io_thread;
 
   io_thread->InitializeNetworkSessionParams(&params);
@@ skipping 26 matching lines @@
       make_scoped_ptr(new DevToolsNetworkTransactionFactory(
           network_controller_handle_.GetController(), shared_session)),
       std::move(backend), true /* set_up_quic_server_info */));
 }
 
 void ProfileIOData::SetCookieSettingsForTesting(
     content_settings::CookieSettings* cookie_settings) {
   DCHECK(!cookie_settings_.get());
   cookie_settings_ = cookie_settings;
 }