97 lines
3.4 KiB
C++
97 lines
3.4 KiB
C++
// Copyright (c) 2017 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "shell/browser/io_thread.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "components/net_log/chrome_net_log.h"
|
|
#include "content/public/browser/browser_thread.h"
|
|
#include "content/public/browser/network_service_instance.h"
|
|
#include "net/cert/caching_cert_verifier.h"
|
|
#include "net/cert/cert_verifier.h"
|
|
#include "net/cert/cert_verify_proc.h"
|
|
#include "net/cert/multi_threaded_cert_verifier.h"
|
|
#include "net/proxy_resolution/proxy_resolution_service.h"
|
|
#include "net/url_request/url_request_context.h"
|
|
#include "services/network/network_service.h"
|
|
#include "services/network/public/cpp/features.h"
|
|
#include "services/network/url_request_context_builder_mojo.h"
|
|
#include "shell/browser/net/url_request_context_getter.h"
|
|
|
|
using content::BrowserThread;
|
|
|
|
IOThread::IOThread(net_log::ChromeNetLog* net_log,
|
|
SystemNetworkContextManager* system_network_context_manager)
|
|
: net_log_(net_log) {
|
|
BrowserThread::SetIOThreadDelegate(this);
|
|
|
|
system_network_context_manager->SetUp(
|
|
&network_context_request_, &network_context_params_,
|
|
&http_auth_static_params_, &http_auth_dynamic_params_);
|
|
}
|
|
|
|
IOThread::~IOThread() {
|
|
BrowserThread::SetIOThreadDelegate(nullptr);
|
|
}
|
|
|
|
void IOThread::RegisterURLRequestContextGetter(
|
|
electron::URLRequestContextGetter* getter) {
|
|
base::AutoLock lock(lock_);
|
|
|
|
DCHECK(!base::FeatureList::IsEnabled(network::features::kNetworkService));
|
|
DCHECK_EQ(0u, request_context_getters_.count(getter));
|
|
request_context_getters_.insert(getter);
|
|
}
|
|
|
|
void IOThread::DeregisterURLRequestContextGetter(
|
|
electron::URLRequestContextGetter* getter) {
|
|
base::AutoLock lock(lock_);
|
|
|
|
DCHECK(!base::FeatureList::IsEnabled(network::features::kNetworkService));
|
|
DCHECK_EQ(1u, request_context_getters_.count(getter));
|
|
request_context_getters_.erase(getter);
|
|
}
|
|
|
|
void IOThread::Init() {
|
|
if (!base::FeatureList::IsEnabled(network::features::kNetworkService)) {
|
|
std::unique_ptr<network::URLRequestContextBuilderMojo> builder =
|
|
std::make_unique<network::URLRequestContextBuilderMojo>();
|
|
|
|
// Enable file:// support.
|
|
builder->set_file_enabled(true);
|
|
|
|
auto cert_verifier = std::make_unique<net::CachingCertVerifier>(
|
|
std::make_unique<net::MultiThreadedCertVerifier>(
|
|
net::CertVerifyProc::CreateDefault(nullptr)));
|
|
builder->SetCertVerifier(std::move(cert_verifier));
|
|
|
|
// Create the network service, so that shared host resolver
|
|
// gets created which is required to set the auth preferences below.
|
|
network::NetworkService* network_service = content::GetNetworkServiceImpl();
|
|
network_service->SetUpHttpAuth(std::move(http_auth_static_params_));
|
|
network_service->ConfigureHttpAuthPrefs(
|
|
std::move(http_auth_dynamic_params_));
|
|
|
|
system_network_context_ = network_service->CreateNetworkContextWithBuilder(
|
|
std::move(network_context_request_), std::move(network_context_params_),
|
|
std::move(builder), &system_request_context_);
|
|
}
|
|
}
|
|
|
|
void IOThread::CleanUp() {
|
|
if (!base::FeatureList::IsEnabled(network::features::kNetworkService)) {
|
|
system_request_context_->proxy_resolution_service()->OnShutdown();
|
|
|
|
base::AutoLock lock(lock_);
|
|
for (auto* getter : request_context_getters_) {
|
|
getter->NotifyContextShuttingDown();
|
|
}
|
|
|
|
system_network_context_.reset();
|
|
}
|
|
|
|
if (net_log_)
|
|
net_log_->ShutDownBeforeThreadPool();
|
|
}
|