Refactor main URLRequestContext creation to use URLRequestContextBuilder

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 <string>
 
 #include "base/basictypes.h"
 #include "base/bind.h"
@@ skipping 60 matching lines @@
 #include "net/proxy/proxy_config_service_fixed.h"
 #include "net/proxy/proxy_script_fetcher_impl.h"
 #include "net/proxy/proxy_service.h"
 #include "net/ssl/channel_id_service.h"
 #include "net/ssl/client_cert_store.h"
 #include "net/url_request/data_protocol_handler.h"
 #include "net/url_request/file_protocol_handler.h"
 #include "net/url_request/ftp_protocol_handler.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_context_builder.h"
+#include "net/url_request/url_request_context_storage.h"
 #include "net/url_request/url_request_file_job.h"
 #include "net/url_request/url_request_intercepting_job_factory.h"
 #include "net/url_request/url_request_interceptor.h"
 #include "net/url_request/url_request_job_factory_impl.h"
 
 #if defined(ENABLE_CONFIGURATION_POLICY)
 #include "chrome/browser/policy/cloud/policy_header_service_factory.h"
 #include "chrome/browser/policy/policy_helpers.h"
 #include "components/policy/core/browser/url_blacklist_manager.h"
 #include "components/policy/core/common/cloud/policy_header_io_helper.h"
@@ skipping 786 matching lines @@
   return data_reduction_proxy_io_data() &&
       data_reduction_proxy_io_data()->IsEnabled();
 }
 
 void ProfileIOData::set_data_reduction_proxy_io_data(
     scoped_ptr<data_reduction_proxy::DataReductionProxyIOData>
         data_reduction_proxy_io_data) const {
   data_reduction_proxy_io_data_ = data_reduction_proxy_io_data.Pass();
 }
 
+net::FraudulentCertificateReporter*
+ProfileIOData::fraudulent_certificate_reporter() const {
+  return main_request_context_.get()
+             ? main_request_context_->storage()
+                   ->fraudulent_certificate_reporter()
+             : nullptr;
+}
+
+net::TransportSecurityState* ProfileIOData::transport_security_state() const {
+  return main_request_context_.get()
+             ? main_request_context_->storage()->transport_security_state()
+             : nullptr;
+}
+
 base::WeakPtr<net::HttpServerProperties>
 ProfileIOData::http_server_properties() const {
   return http_server_properties_->GetWeakPtr();
 }
 
 void ProfileIOData::set_http_server_properties(
     scoped_ptr<net::HttpServerProperties> http_server_properties) const {
   http_server_properties_ = http_server_properties.Pass();
 }
 
@@ skipping 81 matching lines @@
 std::string ProfileIOData::GetSSLSessionCacheShard() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   // The SSL session cache is partitioned by setting a string. This returns a
   // unique string to partition the SSL session cache. Each time we create a
   // new profile, we'll get a fresh SSL session cache which is separate from
   // the other profiles.
   static unsigned ssl_session_cache_instance = 0;
   return base::StringPrintf("profile/%u", ssl_session_cache_instance++);
 }
 
+namespace {
+
+class ChromeFraudulentCertificateReporterFactory
+    : public net::FraudulentCertificateReporterFactory {
+ public:
+  net::FraudulentCertificateReporter* Create(
+      net::URLRequestContext* context) const override {
+    return new ChromeFraudulentCertificateReporter(context);
+  }
+};
+
+}
+
+net::CertVerifier* ProfileIOData::CertVerifierDelegateMainContext::Get(
+    net::URLRequestContext* context) {
+  return owner_->GetCertVerifierForMainContext(context);
+}
+
+net::HttpTransactionFactory*
+ProfileIOData::HttpTransactionFactoryDelegate::Create(
+    net::URLRequestContext* context) {
+  return owner_->CreateMainHttpFactory(
+      profile_params_, main_backend_, context).release();
+}
+
 void ProfileIOData::Init(
     content::ProtocolHandlerMap* protocol_handlers,
     content::URLRequestInterceptorScopedVector request_interceptors) const {
   // The basic logic is implemented here. The specific initialization
   // is done in InitializeInternal(), implemented by subtypes. Static helper
   // functions have been provided to assist in common operations.
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   DCHECK(!initialized_);
 
   // TODO(jhawkins): Remove once crbug.com/102004 is fixed.
   CHECK(initialized_on_UI_thread_);
 
   // TODO(jhawkins): Return to DCHECK once crbug.com/102004 is fixed.
   CHECK(profile_params_.get());
 
   IOThread* const io_thread = profile_params_->io_thread;
   IOThread::Globals* const io_thread_globals = io_thread->globals();
   const base::CommandLine& command_line =
       *base::CommandLine::ForCurrentProcess();
 
   // Create the common request contexts.
-  main_request_context_.reset(new net::URLRequestContext());
+  net::URLRequestContextBuilder context_builder;
   extensions_request_context_.reset(new net::URLRequestContext());
 
   scoped_ptr<ChromeNetworkDelegate> network_delegate(
       new ChromeNetworkDelegate(
 #if defined(ENABLE_EXTENSIONS)
           io_thread_globals->extension_event_router_forwarder.get(),
 #else
           NULL,
 #endif
           &enable_referrers_));
 #if defined(ENABLE_EXTENSIONS)
   network_delegate->set_extension_info_map(
       profile_params_->extension_info_map.get());
   if (!command_line.HasSwitch(switches::kDisableExtensionsHttpThrottling)) {
     extension_throttle_manager_.reset(
         new extensions::ExtensionThrottleManager());
   }
 #endif
 
 #if defined(ENABLE_CONFIGURATION_POLICY)
   network_delegate->set_url_blacklist_manager(url_blacklist_manager_.get());
 #endif
   network_delegate->set_profile(profile_params_->profile);
   network_delegate->set_profile_path(profile_params_->path);
   network_delegate->set_cookie_settings(profile_params_->cookie_settings.get());
   network_delegate->set_enable_do_not_track(&enable_do_not_track_);
   network_delegate->set_force_google_safe_search(&force_google_safesearch_);
   network_delegate->set_force_youtube_safety_mode(&force_youtube_safety_mode_);
-  fraudulent_certificate_reporter_.reset(
-      new ChromeFraudulentCertificateReporter(main_request_context_.get()));
+
+  context_builder.set_fraudulent_certificate_reporter_factory(
+      new ChromeFraudulentCertificateReporterFactory);
 
   // NOTE: Proxy service uses the default io thread network delegate, not the
   // delegate just created.
   proxy_service_.reset(
       ProxyServiceFactory::CreateProxyService(
           io_thread->net_log(),
           io_thread_globals->proxy_script_fetcher_context.get(),
           io_thread_globals->system_network_delegate.get(),
           profile_params_->proxy_config_service.release(),
           command_line,
           quick_check_enabled_.GetValue()));
-  transport_security_state_.reset(new net::TransportSecurityState());
-  base::SequencedWorkerPool* pool = BrowserThread::GetBlockingPool();
-  transport_security_persister_.reset(
-      new net::TransportSecurityPersister(
-          transport_security_state_.get(),
-          profile_params_->path,
-          pool->GetSequencedTaskRunnerWithShutdownBehavior(
-              pool->GetSequenceToken(),
-              base::SequencedWorkerPool::BLOCK_SHUTDOWN),
-          IsOffTheRecord()));
+  context_builder.set_transport_security_state(
+      new net::TransportSecurityState());
 
   // Take ownership over these parameters.
   cookie_settings_ = profile_params_->cookie_settings;
   host_content_settings_map_ = profile_params_->host_content_settings_map;
 #if defined(ENABLE_EXTENSIONS)
   extension_info_map_ = profile_params_->extension_info_map;
 #endif
 
-  resource_context_->host_resolver_ = io_thread_globals->host_resolver.get();
-  resource_context_->request_context_ = main_request_context_.get();
-
   if (profile_params_->resource_prefetch_predictor_observer_) {
     resource_prefetch_predictor_observer_.reset(
         profile_params_->resource_prefetch_predictor_observer_.release());
   }
 
 #if defined(ENABLE_SUPERVISED_USERS)
   supervised_user_url_filter_ = profile_params_->supervised_user_url_filter;
 #endif
 
+  context_builder.set_cert_verifier_delegate(
+      new CertVerifierDelegateMainContext(this));
+
+  // 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());
+  }
+
+  // We save a copy of the network delegate pointer as it may be wrapped before
+  // it is stored in the URLResourceContext, and we may need it during the
+  // post setup init.
+  ChromeNetworkDelegate* network_delegate_raw_ptr = network_delegate.get();
+  InitializeInternal(
+      network_delegate.Pass(), profile_params_.get(),
+      protocol_handlers, request_interceptors.Pass(),
+      &context_builder);
+
+  // Now that the builder is initialized, proceed with building the main
+  // context.
+  main_request_context_.reset(context_builder.BuildMainContext());
+
+  base::SequencedWorkerPool* pool = BrowserThread::GetBlockingPool();
+  transport_security_persister_.reset(new net::TransportSecurityPersister(
+      main_request_context_->storage()->transport_security_state(),
+      profile_params_->path,
+      pool->GetSequencedTaskRunnerWithShutdownBehavior(
+          pool->GetSequenceToken(), base::SequencedWorkerPool::BLOCK_SHUTDOWN),
+      IsOffTheRecord()));
+
+  InitPostContextSetup(network_delegate_raw_ptr);
+
+  profile_params_.reset();
+  initialized_ = true;
+}
+
+net::CertVerifier* ProfileIOData::GetCertVerifierForMainContext(
+    net::URLRequestContext* context) const {
+  IOThread* const io_thread = profile_params_->io_thread;
+  IOThread::Globals* const io_thread_globals = io_thread->globals();
+
+  resource_context_->host_resolver_ = io_thread_globals->host_resolver.get();
+  resource_context_->request_context_ = context;
 #if defined(OS_CHROMEOS)
   username_hash_ = profile_params_->username_hash;
   use_system_key_slot_ = profile_params_->use_system_key_slot;
   if (use_system_key_slot_)
     EnableNSSSystemKeySlotForResourceContext(resource_context_.get());
 
   crypto::ScopedPK11Slot public_slot =
       crypto::GetPublicSlotForChromeOSUser(username_hash_);
   // The private slot won't be ready by this point. It shouldn't be necessary
   // for cert trust purposes anyway.
   scoped_refptr<net::CertVerifyProc> verify_proc(
       new chromeos::CertVerifyProcChromeOS(public_slot.Pass()));
   if (policy_cert_verifier_) {
     DCHECK_EQ(policy_cert_verifier_, cert_verifier_.get());
     policy_cert_verifier_->InitializeOnIOThread(verify_proc);
   } else {
     cert_verifier_.reset(new net::MultiThreadedCertVerifier(verify_proc.get()));
   }
-  main_request_context_->set_cert_verifier(cert_verifier_.get());
+  return cert_verifier_.get();
 #else
-  main_request_context_->set_cert_verifier(
-      io_thread_globals->cert_verifier.get());
+  return io_thread_globals->cert_verifier.get();
 #endif
-
-  // 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());
-  }
-
-  InitializeInternal(
-      network_delegate.Pass(), profile_params_.get(),
-      protocol_handlers, request_interceptors.Pass());
-
-  profile_params_.reset();
-  initialized_ = true;
 }
+//#endif
 
 void ProfileIOData::ApplyProfileParamsToContext(
     net::URLRequestContext* context) const {
   context->set_http_user_agent_settings(
       chrome_http_user_agent_settings_.get());
   context->set_ssl_config_service(profile_params_->ssl_config_service.get());
 }
 
+void ProfileIOData::ApplyProfileParamsToContext(
+    net::URLRequestContextBuilder* context_builder) const {
+  context_builder->set_http_user_agent_settings_unowned(
+      chrome_http_user_agent_settings_.get());
+  context_builder->set_ssl_config_service(
+      profile_params_->ssl_config_service.get());
+}
+
 scoped_ptr<net::URLRequestJobFactory> ProfileIOData::SetUpJobFactoryDefaults(
     scoped_ptr<net::URLRequestJobFactoryImpl> job_factory,
     content::URLRequestInterceptorScopedVector request_interceptors,
     scoped_ptr<ProtocolHandlerRegistry::JobInterceptorFactory>
         protocol_handler_interceptor,
     net::NetworkDelegate* network_delegate,
     net::FtpTransactionFactory* ftp_transaction_factory) const {
   // NOTE(willchan): Keep these protocol handlers in sync with
   // ProfileIOData::IsHandledProtocol().
   bool set_protocol = job_factory->SetProtocolHandler(
@@ skipping 105 matching lines @@
           base::Bind(&NotifyContextGettersOfShutdownOnIO,
               base::Passed(&context_getters)));
     }
   }
 
   bool posted = BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE, this);
   if (!posted)
     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();
 }
 
 scoped_ptr<net::HttpCache> ProfileIOData::CreateMainHttpFactory(
     const ProfileParams* profile_params,
-    net::HttpCache::BackendFactory* main_backend) const {
+    net::HttpCache::BackendFactory* main_backend,
+    net::URLRequestContext* context) const {
   net::HttpNetworkSession::Params params;
-  net::URLRequestContext* context = main_request_context();
 
   IOThread* const io_thread = profile_params->io_thread;
 
   io_thread->InitializeNetworkSessionParams(&params);
 
   params.host_resolver = context->host_resolver();
   params.cert_verifier = context->cert_verifier();
   params.channel_id_service = context->channel_id_service();
   params.transport_security_state = context->transport_security_state();
   params.cert_transparency_verifier = context->cert_transparency_verifier();
@@ skipping 22 matching lines @@
       new DevToolsNetworkTransactionFactory(
           network_controller_.get(), shared_session),
       shared_session->net_log(), backend));
 }
 
 void ProfileIOData::SetCookieSettingsForTesting(
     content_settings::CookieSettings* cookie_settings) {
   DCHECK(!cookie_settings_.get());
   cookie_settings_ = cookie_settings;
 }