jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 1 | // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "content/network/network_context.h" |
| 6 | |
| 7 | #include "base/command_line.h" |
| 8 | #include "base/logging.h" |
yzshen | 6da6865f | 2017-04-18 17:49:13 | [diff] [blame] | 9 | #include "base/strings/string_number_conversions.h" |
mmenke | 68a5021 | 2017-06-12 20:20:31 | [diff] [blame] | 10 | #include "components/network_session_configurator/common/network_switches.h" |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 11 | #include "content/network/cache_url_loader.h" |
mmenke | f73a512 | 2017-07-13 23:27:39 | [diff] [blame^] | 12 | #include "content/network/network_service_impl.h" |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 13 | #include "content/network/network_service_url_loader_factory_impl.h" |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 14 | #include "content/network/url_loader_impl.h" |
| 15 | #include "content/public/common/content_client.h" |
yzshen | 6da6865f | 2017-04-18 17:49:13 | [diff] [blame] | 16 | #include "content/public/common/content_switches.h" |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 17 | #include "net/dns/host_resolver.h" |
| 18 | #include "net/dns/mapped_host_resolver.h" |
mmenke | 7c7f4faf5 | 2017-06-01 00:25:22 | [diff] [blame] | 19 | #include "net/http/http_network_session.h" |
yzshen | a90291c | 2017-04-26 16:22:52 | [diff] [blame] | 20 | #include "net/proxy/proxy_config.h" |
| 21 | #include "net/proxy/proxy_config_service_fixed.h" |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 22 | #include "net/url_request/url_request_context.h" |
| 23 | #include "net/url_request/url_request_context_builder.h" |
| 24 | |
| 25 | namespace content { |
| 26 | |
| 27 | namespace { |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 28 | |
| 29 | std::unique_ptr<net::URLRequestContext> MakeURLRequestContext() { |
| 30 | net::URLRequestContextBuilder builder; |
mmenke | 7c7f4faf5 | 2017-06-01 00:25:22 | [diff] [blame] | 31 | net::HttpNetworkSession::Params params; |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 32 | const base::CommandLine* command_line = |
| 33 | base::CommandLine::ForCurrentProcess(); |
yzshen | 6da6865f | 2017-04-18 17:49:13 | [diff] [blame] | 34 | if (command_line->HasSwitch(switches::kIgnoreCertificateErrors)) |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 35 | params.ignore_certificate_errors = true; |
yzshen | 6da6865f | 2017-04-18 17:49:13 | [diff] [blame] | 36 | |
| 37 | if (command_line->HasSwitch(switches::kTestingFixedHttpPort)) { |
| 38 | int value; |
| 39 | base::StringToInt( |
| 40 | command_line->GetSwitchValueASCII(switches::kTestingFixedHttpPort), |
| 41 | &value); |
| 42 | params.testing_fixed_http_port = value; |
| 43 | } |
| 44 | if (command_line->HasSwitch(switches::kTestingFixedHttpsPort)) { |
| 45 | int value; |
| 46 | base::StringToInt( |
| 47 | command_line->GetSwitchValueASCII(switches::kTestingFixedHttpsPort), |
| 48 | &value); |
| 49 | params.testing_fixed_https_port = value; |
| 50 | } |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 51 | builder.set_http_network_session_params(params); |
yzshen | 6da6865f | 2017-04-18 17:49:13 | [diff] [blame] | 52 | if (command_line->HasSwitch(switches::kHostResolverRules)) { |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 53 | std::unique_ptr<net::HostResolver> host_resolver( |
| 54 | net::HostResolver::CreateDefaultResolver(nullptr)); |
| 55 | std::unique_ptr<net::MappedHostResolver> remapped_host_resolver( |
| 56 | new net::MappedHostResolver(std::move(host_resolver))); |
| 57 | remapped_host_resolver->SetRulesFromString( |
yzshen | 6da6865f | 2017-04-18 17:49:13 | [diff] [blame] | 58 | command_line->GetSwitchValueASCII(switches::kHostResolverRules)); |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 59 | builder.set_host_resolver(std::move(remapped_host_resolver)); |
| 60 | } |
| 61 | builder.set_accept_language("en-us,en"); |
| 62 | builder.set_user_agent(GetContentClient()->GetUserAgent()); |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 63 | net::URLRequestContextBuilder::HttpCacheParams cache_params; |
| 64 | |
| 65 | // We store the cache in memory so we can run many shells in parallel when |
| 66 | // running tests, otherwise the network services in each shell will corrupt |
| 67 | // the disk cache. |
| 68 | cache_params.type = net::URLRequestContextBuilder::HttpCacheParams::IN_MEMORY; |
| 69 | |
| 70 | builder.EnableHttpCache(cache_params); |
| 71 | builder.set_file_enabled(true); |
jam | 73438654 | 2017-05-08 21:21:35 | [diff] [blame] | 72 | builder.set_data_enabled(true); |
scottmg | fd5066f | 2017-04-14 00:49:17 | [diff] [blame] | 73 | |
yzshen | a90291c | 2017-04-26 16:22:52 | [diff] [blame] | 74 | if (command_line->HasSwitch(switches::kProxyServer)) { |
| 75 | net::ProxyConfig config; |
| 76 | config.proxy_rules().ParseFromString( |
| 77 | command_line->GetSwitchValueASCII(switches::kProxyServer)); |
| 78 | std::unique_ptr<net::ProxyConfigService> fixed_config_service = |
| 79 | base::MakeUnique<net::ProxyConfigServiceFixed>(config); |
| 80 | builder.set_proxy_config_service(std::move(fixed_config_service)); |
| 81 | } else { |
| 82 | builder.set_proxy_service(net::ProxyService::CreateDirect()); |
| 83 | } |
| 84 | |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 85 | return builder.Build(); |
| 86 | } |
| 87 | |
| 88 | } // namespace |
| 89 | |
mmenke | f73a512 | 2017-07-13 23:27:39 | [diff] [blame^] | 90 | NetworkContext::NetworkContext(NetworkServiceImpl* network_service, |
mmenke | 502a613 | 2017-06-29 00:36:29 | [diff] [blame] | 91 | mojom::NetworkContextRequest request, |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 92 | mojom::NetworkContextParamsPtr params) |
mmenke | 502a613 | 2017-06-29 00:36:29 | [diff] [blame] | 93 | : network_service_(network_service), |
| 94 | url_request_context_(MakeURLRequestContext()), |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 95 | params_(std::move(params)), |
mmenke | 502a613 | 2017-06-29 00:36:29 | [diff] [blame] | 96 | binding_(this, std::move(request)) { |
| 97 | network_service_->RegisterNetworkContext(this); |
| 98 | binding_.set_connection_error_handler( |
| 99 | base::Bind(&NetworkContext::OnConnectionError, base::Unretained(this))); |
| 100 | } |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 101 | |
mmenke | f73a512 | 2017-07-13 23:27:39 | [diff] [blame^] | 102 | // TODO(mmenke): Share URLRequestContextBulder configuration between two |
| 103 | // constructors. Can only share them once consumer code is ready for its |
| 104 | // corresponding options to be overwritten. |
| 105 | NetworkContext::NetworkContext( |
| 106 | mojom::NetworkContextRequest request, |
| 107 | mojom::NetworkContextParamsPtr params, |
| 108 | std::unique_ptr<net::URLRequestContextBuilder> builder) |
| 109 | : network_service_(nullptr), |
| 110 | url_request_context_(builder->Build()), |
| 111 | params_(std::move(params)), |
| 112 | binding_(this, std::move(request)) {} |
| 113 | |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 114 | NetworkContext::~NetworkContext() { |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 115 | // Call each URLLoaderImpl and ask it to release its net::URLRequest, as the |
| 116 | // corresponding net::URLRequestContext is going away with this |
| 117 | // NetworkContext. The loaders can be deregistering themselves in Cleanup(), |
mmenke | 28bcb85f | 2017-06-23 01:14:39 | [diff] [blame] | 118 | // so have to be careful. |
| 119 | while (!url_loaders_.empty()) |
| 120 | (*url_loaders_.begin())->Cleanup(); |
mmenke | 502a613 | 2017-06-29 00:36:29 | [diff] [blame] | 121 | |
| 122 | // May be nullptr in tests. |
| 123 | if (network_service_) |
| 124 | network_service_->DeregisterNetworkContext(this); |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 125 | } |
| 126 | |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 127 | std::unique_ptr<NetworkContext> NetworkContext::CreateForTesting() { |
| 128 | return base::WrapUnique(new NetworkContext); |
| 129 | } |
| 130 | |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 131 | void NetworkContext::RegisterURLLoader(URLLoaderImpl* url_loader) { |
| 132 | DCHECK(url_loaders_.count(url_loader) == 0); |
| 133 | url_loaders_.insert(url_loader); |
| 134 | } |
| 135 | |
| 136 | void NetworkContext::DeregisterURLLoader(URLLoaderImpl* url_loader) { |
mmenke | 28bcb85f | 2017-06-23 01:14:39 | [diff] [blame] | 137 | size_t removed_count = url_loaders_.erase(url_loader); |
| 138 | DCHECK(removed_count); |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 139 | } |
| 140 | |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 141 | void NetworkContext::CreateURLLoaderFactory( |
| 142 | mojom::URLLoaderFactoryRequest request, |
| 143 | uint32_t process_id) { |
| 144 | loader_factory_bindings_.AddBinding( |
| 145 | base::MakeUnique<NetworkServiceURLLoaderFactoryImpl>(this, process_id), |
| 146 | std::move(request)); |
| 147 | } |
| 148 | |
| 149 | void NetworkContext::HandleViewCacheRequest(const GURL& url, |
| 150 | mojom::URLLoaderClientPtr client) { |
| 151 | StartCacheURLLoader(url, url_request_context_.get(), std::move(client)); |
| 152 | } |
| 153 | |
mmenke | 502a613 | 2017-06-29 00:36:29 | [diff] [blame] | 154 | void NetworkContext::Cleanup() { |
| 155 | // The NetworkService is going away, so have to destroy the |
| 156 | // net::URLRequestContext held by this NetworkContext. |
| 157 | delete this; |
| 158 | } |
| 159 | |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 160 | NetworkContext::NetworkContext() |
mmenke | 502a613 | 2017-06-29 00:36:29 | [diff] [blame] | 161 | : network_service_(nullptr), |
| 162 | url_request_context_(MakeURLRequestContext()), |
jam | c190586 | 2017-05-16 14:45:30 | [diff] [blame] | 163 | binding_(this) {} |
| 164 | |
mmenke | 502a613 | 2017-06-29 00:36:29 | [diff] [blame] | 165 | void NetworkContext::OnConnectionError() { |
| 166 | // Don't delete |this| in response to connection errors when it was created by |
| 167 | // CreateForTesting. |
| 168 | if (network_service_) |
| 169 | delete this; |
| 170 | } |
| 171 | |
jam | 6f02ddc | 2017-04-12 01:43:50 | [diff] [blame] | 172 | } // namespace content |