components/cronet/android/url_request_context_adapter.cc

   1 // Copyright 2014 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 "components/cronet/android/url_request_context_adapter.h"
   6
   7 #include <limits>
   8
   9 #include "base/bind.h"
  10 #include "base/files/file_util.h"
  11 #include "base/files/scoped_file.h"
  12 #include "base/message_loop/message_loop.h"
  13 #include "base/single_thread_task_runner.h"
  14 #include "components/cronet/url_request_context_config.h"
  15 #include "net/android/network_change_notifier_factory_android.h"
  16 #include "net/base/net_errors.h"
  17 #include "net/base/net_log_logger.h"
  18 #include "net/base/net_util.h"
  19 #include "net/base/network_change_notifier.h"
  20 #include "net/base/network_delegate_impl.h"
  21 #include "net/cert/cert_verifier.h"
  22 #include "net/http/http_auth_handler_factory.h"
  23 #include "net/http/http_network_layer.h"
  24 #include "net/http/http_server_properties.h"
  25 #include "net/proxy/proxy_service.h"
  26 #include "net/ssl/ssl_config_service_defaults.h"
  27 #include "net/url_request/static_http_user_agent_settings.h"
  28 #include "net/url_request/url_request_context_builder.h"
  29 #include "net/url_request/url_request_context_storage.h"
  30 #include "net/url_request/url_request_job_factory_impl.h"
  31
  32 namespace {
  33
  34 class BasicNetworkDelegate : public net::NetworkDelegateImpl {
  35  public:
  36   BasicNetworkDelegate() {}
  37   ~BasicNetworkDelegate() override {}
  38
  39  private:
  40   // net::NetworkDelegate implementation.
  41   int OnBeforeURLRequest(net::URLRequest* request,
  42                          const net::CompletionCallback& callback,
  43                          GURL* new_url) override {
  44     return net::OK;
  45   }
  46
  47   int OnBeforeSendHeaders(net::URLRequest* request,
  48                           const net::CompletionCallback& callback,
  49                           net::HttpRequestHeaders* headers) override {
  50     return net::OK;
  51   }
  52
  53   void OnSendHeaders(net::URLRequest* request,
  54                      const net::HttpRequestHeaders& headers) override {}
  55
  56   int OnHeadersReceived(
  57       net::URLRequest* request,
  58       const net::CompletionCallback& callback,
  59       const net::HttpResponseHeaders* original_response_headers,
  60       scoped_refptr<net::HttpResponseHeaders>* _response_headers,
  61       GURL* allowed_unsafe_redirect_url) override {
  62     return net::OK;
  63   }
  64
  65   void OnBeforeRedirect(net::URLRequest* request,
  66                         const GURL& new_location) override {}
  67
  68   void OnResponseStarted(net::URLRequest* request) override {}
  69
  70   void OnRawBytesRead(const net::URLRequest& request,
  71                       int bytes_read) override {}
  72
  73   void OnCompleted(net::URLRequest* request, bool started) override {}
  74
  75   void OnURLRequestDestroyed(net::URLRequest* request) override {}
  76
  77   void OnPACScriptError(int line_number,
  78                         const base::string16& error) override {}
  79
  80   NetworkDelegate::AuthRequiredResponse OnAuthRequired(
  81       net::URLRequest* request,
  82       const net::AuthChallengeInfo& auth_info,
  83       const AuthCallback& callback,
  84       net::AuthCredentials* credentials) override {
  85     return net::NetworkDelegate::AUTH_REQUIRED_RESPONSE_NO_ACTION;
  86   }
  87
  88   bool OnCanGetCookies(const net::URLRequest& request,
  89                        const net::CookieList& cookie_list) override {
  90     return false;
  91   }
  92
  93   bool OnCanSetCookie(const net::URLRequest& request,
  94                       const std::string& cookie_line,
  95                       net::CookieOptions* options) override {
  96     return false;
  97   }
  98
  99   bool OnCanAccessFile(const net::URLRequest& request,
 100                        const base::FilePath& path) const override {
 101     return false;
 102   }
 103
 104   bool OnCanThrottleRequest(
 105       const net::URLRequest& request) const override {
 106     return false;
 107   }
 108
 109   DISALLOW_COPY_AND_ASSIGN(BasicNetworkDelegate);
 110 };
 111
 112 }  // namespace
 113
 114 namespace cronet {
 115
 116 URLRequestContextAdapter::URLRequestContextAdapter(
 117     URLRequestContextAdapterDelegate* delegate,
 118     std::string user_agent)
 119     : load_disable_cache_(true),
 120       is_context_initialized_(false) {
 121   delegate_ = delegate;
 122   user_agent_ = user_agent;
 123 }
 124
 125 void URLRequestContextAdapter::Initialize(
 126     scoped_ptr<URLRequestContextConfig> config) {
 127   network_thread_ = new base::Thread("network");
 128   base::Thread::Options options;
 129   options.message_loop_type = base::MessageLoop::TYPE_IO;
 130   network_thread_->StartWithOptions(options);
 131   config_ = config.Pass();
 132 }
 133
 134 void URLRequestContextAdapter::InitRequestContextOnMainThread() {
 135   proxy_config_service_.reset(net::ProxyService::CreateSystemProxyConfigService(
 136       GetNetworkTaskRunner(), NULL));
 137   GetNetworkTaskRunner()->PostTask(
 138       FROM_HERE,
 139       base::Bind(&URLRequestContextAdapter::InitRequestContextOnNetworkThread,
 140                  this));
 141 }
 142
 143 void URLRequestContextAdapter::InitRequestContextOnNetworkThread() {
 144   DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
 145   DCHECK(config_);
 146   // TODO(mmenke):  Add method to have the builder enable SPDY.
 147   net::URLRequestContextBuilder context_builder;
 148   context_builder.set_network_delegate(new BasicNetworkDelegate());
 149   context_builder.set_proxy_config_service(proxy_config_service_.get());
 150   config_->ConfigureURLRequestContextBuilder(&context_builder);
 151
 152   context_.reset(context_builder.Build());
 153
 154   // Currently (circa M39) enabling QUIC requires setting probability threshold.
 155   if (config_->enable_quic) {
 156     context_->http_server_properties()
 157         ->SetAlternateProtocolProbabilityThreshold(0.0f);
 158     for (size_t hint = 0; hint < config_->quic_hints.size(); ++hint) {
 159       const URLRequestContextConfig::QuicHint& quic_hint =
 160           *config_->quic_hints[hint];
 161       if (quic_hint.host.empty()) {
 162         LOG(ERROR) << "Empty QUIC hint host: " << quic_hint.host;
 163         continue;
 164       }
 165
 166       url::CanonHostInfo host_info;
 167       std::string canon_host(net::CanonicalizeHost(quic_hint.host, &host_info));
 168       if (!host_info.IsIPAddress() &&
 169           !net::IsCanonicalizedHostCompliant(canon_host)) {
 170         LOG(ERROR) << "Invalid QUIC hint host: " << quic_hint.host;
 171         continue;
 172       }
 173
 174       if (quic_hint.port <= std::numeric_limits<uint16>::min() ||
 175           quic_hint.port > std::numeric_limits<uint16>::max()) {
 176         LOG(ERROR) << "Invalid QUIC hint port: "
 177                    << quic_hint.port;
 178         continue;
 179       }
 180
 181       if (quic_hint.alternate_port <= std::numeric_limits<uint16>::min() ||
 182           quic_hint.alternate_port > std::numeric_limits<uint16>::max()) {
 183         LOG(ERROR) << "Invalid QUIC hint alternate port: "
 184                    << quic_hint.alternate_port;
 185         continue;
 186       }
 187
 188       net::HostPortPair quic_hint_host_port_pair(canon_host,
 189                                                  quic_hint.port);
 190       net::AlternativeService alternative_service(
 191           net::AlternateProtocol::QUIC, "",
 192           static_cast<uint16>(quic_hint.alternate_port));
 193       context_->http_server_properties()->SetAlternativeService(
 194           quic_hint_host_port_pair, alternative_service, 1.0f);
 195     }
 196   }
 197   load_disable_cache_ = config_->load_disable_cache;
 198   config_.reset(NULL);
 199
 200   if (VLOG_IS_ON(2)) {
 201     net_log_observer_.reset(new NetLogObserver());
 202     context_->net_log()->AddThreadSafeObserver(net_log_observer_.get(),
 203                                                net::NetLog::LOG_ALL_BUT_BYTES);
 204   }
 205
 206   is_context_initialized_ = true;
 207   while (!tasks_waiting_for_context_.empty()) {
 208     tasks_waiting_for_context_.front().Run();
 209     tasks_waiting_for_context_.pop();
 210   }
 211
 212   delegate_->OnContextInitialized(this);
 213 }
 214
 215 void URLRequestContextAdapter::PostTaskToNetworkThread(
 216     const tracked_objects::Location& posted_from,
 217     const RunAfterContextInitTask& callback) {
 218   GetNetworkTaskRunner()->PostTask(
 219       posted_from,
 220       base::Bind(
 221           &URLRequestContextAdapter::RunTaskAfterContextInitOnNetworkThread,
 222           this,
 223           callback));
 224 }
 225
 226 void URLRequestContextAdapter::RunTaskAfterContextInitOnNetworkThread(
 227     const RunAfterContextInitTask& callback) {
 228   DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
 229   if (is_context_initialized_) {
 230     callback.Run();
 231     return;
 232   }
 233   tasks_waiting_for_context_.push(callback);
 234 }
 235
 236 URLRequestContextAdapter::~URLRequestContextAdapter() {
 237   DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
 238   if (net_log_observer_) {
 239     context_->net_log()->RemoveThreadSafeObserver(net_log_observer_.get());
 240     net_log_observer_.reset();
 241   }
 242   StopNetLogHelper();
 243   // TODO(mef): Ensure that |network_thread_| is destroyed properly.
 244 }
 245
 246 const std::string& URLRequestContextAdapter::GetUserAgent(
 247     const GURL& url) const {
 248   return user_agent_;
 249 }
 250
 251 net::URLRequestContext* URLRequestContextAdapter::GetURLRequestContext() {
 252   DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
 253   if (!context_) {
 254     LOG(ERROR) << "URLRequestContext is not set up";
 255   }
 256   return context_.get();
 257 }
 258
 259 scoped_refptr<base::SingleThreadTaskRunner>
 260 URLRequestContextAdapter::GetNetworkTaskRunner() const {
 261   return network_thread_->message_loop_proxy();
 262 }
 263
 264 void URLRequestContextAdapter::StartNetLogToFile(const std::string& file_name) {
 265   PostTaskToNetworkThread(
 266       FROM_HERE,
 267       base::Bind(
 268           &URLRequestContextAdapter::StartNetLogToFileHelper, this, file_name));
 269 }
 270
 271 void URLRequestContextAdapter::StopNetLog() {
 272   PostTaskToNetworkThread(
 273       FROM_HERE, base::Bind(&URLRequestContextAdapter::StopNetLogHelper, this));
 274 }
 275
 276 void URLRequestContextAdapter::StartNetLogToFileHelper(
 277     const std::string& file_name) {
 278   DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
 279   // Do nothing if already logging to a file.
 280   if (net_log_logger_)
 281     return;
 282
 283   base::FilePath file_path(file_name);
 284   base::ScopedFILE file(base::OpenFile(file_path, "w"));
 285   if (!file)
 286     return;
 287
 288   net_log_logger_.reset(new net::NetLogLogger());
 289   net_log_logger_->StartObserving(context_->net_log(), file.Pass(), nullptr,
 290                                   context_.get());
 291 }
 292
 293 void URLRequestContextAdapter::StopNetLogHelper() {
 294   DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
 295   if (net_log_logger_) {
 296     net_log_logger_->StopObserving(context_.get());
 297     net_log_logger_.reset();
 298   }
 299 }
 300
 301 void NetLogObserver::OnAddEntry(const net::NetLog::Entry& entry) {
 302   VLOG(2) << "Net log entry: type=" << entry.type()
 303           << ", source=" << entry.source().type << ", phase=" << entry.phase();
 304 }
 305
 306 }  // namespace cronet