src/net/http/http_stream_factory_impl.cc

   1 // Copyright (c) 2012 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 "net/http/http_stream_factory_impl.h"
   6
   7 #include <string>
   8
   9 #include "base/logging.h"
  10 #include "base/stl_util.h"
  11 #include "base/strings/string_number_conversions.h"
  12 #include "net/base/net_log.h"
  13 #include "net/base/net_util.h"
  14 #include "net/http/http_network_session.h"
  15 #include "net/http/http_server_properties.h"
  16 #include "net/http/http_stream_factory_impl_job.h"
  17 #include "net/http/http_stream_factory_impl_request.h"
  18 #include "net/spdy/spdy_http_stream.h"
  19 #include "url/gurl.h"
  20
  21 namespace net {
  22
  23 namespace {
  24
  25 GURL UpgradeUrlToHttps(const GURL& original_url, int port) {
  26   GURL::Replacements replacements;
  27   // new_sheme and new_port need to be in scope here because GURL::Replacements
  28   // references the memory contained by them directly.
  29   const std::string new_scheme = "https";
  30   const std::string new_port = base::IntToString(port);
  31   replacements.SetSchemeStr(new_scheme);
  32   replacements.SetPortStr(new_port);
  33   return original_url.ReplaceComponents(replacements);
  34 }
  35
  36 }  // namespace
  37
  38 HttpStreamFactoryImpl::HttpStreamFactoryImpl(HttpNetworkSession* session,
  39                                              bool for_websockets)
  40     : session_(session),
  41       for_websockets_(for_websockets) {}
  42
  43 HttpStreamFactoryImpl::~HttpStreamFactoryImpl() {
  44   DCHECK(request_map_.empty());
  45   DCHECK(spdy_session_request_map_.empty());
  46
  47   std::set<const Job*> tmp_job_set;
  48   tmp_job_set.swap(orphaned_job_set_);
  49   STLDeleteContainerPointers(tmp_job_set.begin(), tmp_job_set.end());
  50   DCHECK(orphaned_job_set_.empty());
  51
  52   tmp_job_set.clear();
  53   tmp_job_set.swap(preconnect_job_set_);
  54   STLDeleteContainerPointers(tmp_job_set.begin(), tmp_job_set.end());
  55   DCHECK(preconnect_job_set_.empty());
  56 }
  57
  58 HttpStreamRequest* HttpStreamFactoryImpl::RequestStream(
  59     const HttpRequestInfo& request_info,
  60     RequestPriority priority,
  61     const SSLConfig& server_ssl_config,
  62     const SSLConfig& proxy_ssl_config,
  63     HttpStreamRequest::Delegate* delegate,
  64     const BoundNetLog& net_log) {
  65   DCHECK(!for_websockets_);
  66   return RequestStreamInternal(request_info,
  67                                priority,
  68                                server_ssl_config,
  69                                proxy_ssl_config,
  70                                delegate,
  71                                NULL,
  72                                net_log);
  73 }
  74
  75 HttpStreamRequest* HttpStreamFactoryImpl::RequestWebSocketHandshakeStream(
  76     const HttpRequestInfo& request_info,
  77     RequestPriority priority,
  78     const SSLConfig& server_ssl_config,
  79     const SSLConfig& proxy_ssl_config,
  80     HttpStreamRequest::Delegate* delegate,
  81     WebSocketHandshakeStreamBase::CreateHelper* create_helper,
  82     const BoundNetLog& net_log) {
  83   DCHECK(for_websockets_);
  84   DCHECK(create_helper);
  85   return RequestStreamInternal(request_info,
  86                                priority,
  87                                server_ssl_config,
  88                                proxy_ssl_config,
  89                                delegate,
  90                                create_helper,
  91                                net_log);
  92 }
  93
  94 HttpStreamRequest* HttpStreamFactoryImpl::RequestStreamInternal(
  95     const HttpRequestInfo& request_info,
  96     RequestPriority priority,
  97     const SSLConfig& server_ssl_config,
  98     const SSLConfig& proxy_ssl_config,
  99     HttpStreamRequest::Delegate* delegate,
 100     WebSocketHandshakeStreamBase::CreateHelper*
 101         websocket_handshake_stream_create_helper,
 102     const BoundNetLog& net_log) {
 103   Request* request = new Request(request_info.url,
 104                                  this,
 105                                  delegate,
 106                                  websocket_handshake_stream_create_helper,
 107                                  net_log);
 108
 109   GURL alternate_url;
 110   AlternateProtocolInfo alternate =
 111       GetAlternateProtocolRequestFor(request_info.url, &alternate_url);
 112   Job* alternate_job = NULL;
 113   if (alternate.protocol != UNINITIALIZED_ALTERNATE_PROTOCOL) {
 114     // Never share connection with other jobs for FTP requests.
 115     DCHECK(!request_info.url.SchemeIs("ftp"));
 116
 117     HttpRequestInfo alternate_request_info = request_info;
 118     alternate_request_info.url = alternate_url;
 119     alternate_job =
 120         new Job(this, session_, alternate_request_info, priority,
 121                 server_ssl_config, proxy_ssl_config, net_log.net_log());
 122     request->AttachJob(alternate_job);
 123     alternate_job->MarkAsAlternate(request_info.url, alternate);
 124   }
 125
 126   Job* job = new Job(this, session_, request_info, priority,
 127                      server_ssl_config, proxy_ssl_config, net_log.net_log());
 128   request->AttachJob(job);
 129   if (alternate_job) {
 130     // Never share connection with other jobs for FTP requests.
 131     DCHECK(!request_info.url.SchemeIs("ftp"));
 132
 133     job->WaitFor(alternate_job);
 134     // Make sure to wait until we call WaitFor(), before starting
 135     // |alternate_job|, otherwise |alternate_job| will not notify |job|
 136     // appropriately.
 137     alternate_job->Start(request);
 138   }
 139   // Even if |alternate_job| has already finished, it won't have notified the
 140   // request yet, since we defer that to the next iteration of the MessageLoop,
 141   // so starting |job| is always safe.
 142   job->Start(request);
 143   return request;
 144 }
 145
 146 void HttpStreamFactoryImpl::PreconnectStreams(
 147     int num_streams,
 148     const HttpRequestInfo& request_info,
 149     RequestPriority priority,
 150     const SSLConfig& server_ssl_config,
 151     const SSLConfig& proxy_ssl_config) {
 152   DCHECK(!for_websockets_);
 153   GURL alternate_url;
 154   AlternateProtocolInfo alternate =
 155       GetAlternateProtocolRequestFor(request_info.url, &alternate_url);
 156   Job* job = NULL;
 157   if (alternate.protocol != UNINITIALIZED_ALTERNATE_PROTOCOL) {
 158     HttpRequestInfo alternate_request_info = request_info;
 159     alternate_request_info.url = alternate_url;
 160     job = new Job(this, session_, alternate_request_info, priority,
 161                   server_ssl_config, proxy_ssl_config, session_->net_log());
 162     job->MarkAsAlternate(request_info.url, alternate);
 163   } else {
 164     job = new Job(this, session_, request_info, priority,
 165                   server_ssl_config, proxy_ssl_config, session_->net_log());
 166   }
 167   preconnect_job_set_.insert(job);
 168   job->Preconnect(num_streams);
 169 }
 170
 171 const HostMappingRules* HttpStreamFactoryImpl::GetHostMappingRules() const {
 172   return session_->params().host_mapping_rules;
 173 }
 174
 175 AlternateProtocolInfo HttpStreamFactoryImpl::GetAlternateProtocolRequestFor(
 176     const GURL& original_url,
 177     GURL* alternate_url) {
 178   const AlternateProtocolInfo kNoAlternateProtocol =
 179       AlternateProtocolInfo(0,  UNINITIALIZED_ALTERNATE_PROTOCOL, 0);
 180
 181   if (!session_->params().use_alternate_protocols)
 182     return kNoAlternateProtocol;
 183
 184   if (original_url.SchemeIs("ftp"))
 185     return kNoAlternateProtocol;
 186
 187   HostPortPair origin = HostPortPair(original_url.HostNoBrackets(),
 188                                      original_url.EffectiveIntPort());
 189
 190   HttpServerProperties& http_server_properties =
 191       *session_->http_server_properties();
 192   if (!http_server_properties.HasAlternateProtocol(origin))
 193     return kNoAlternateProtocol;
 194
 195   AlternateProtocolInfo alternate =
 196       http_server_properties.GetAlternateProtocol(origin);
 197   if (alternate.is_broken) {
 198     HistogramAlternateProtocolUsage(ALTERNATE_PROTOCOL_USAGE_BROKEN);
 199     return kNoAlternateProtocol;
 200   }
 201
 202   if (!IsAlternateProtocolValid(alternate.protocol)) {
 203     NOTREACHED();
 204     return kNoAlternateProtocol;
 205   }
 206
 207   // Some shared unix systems may have user home directories (like
 208   // http://foo.com/~mike) which allow users to emit headers.  This is a bad
 209   // idea already, but with Alternate-Protocol, it provides the ability for a
 210   // single user on a multi-user system to hijack the alternate protocol.
 211   // These systems also enforce ports <1024 as restricted ports.  So don't
 212   // allow protocol upgrades to user-controllable ports.
 213   const int kUnrestrictedPort = 1024;
 214   if (!session_->params().enable_user_alternate_protocol_ports &&
 215       (alternate.port >= kUnrestrictedPort &&
 216        origin.port() < kUnrestrictedPort))
 217     return kNoAlternateProtocol;
 218
 219   origin.set_port(alternate.port);
 220   if (alternate.protocol >= NPN_SPDY_MINIMUM_VERSION &&
 221       alternate.protocol <= NPN_SPDY_MAXIMUM_VERSION) {
 222     if (!HttpStreamFactory::spdy_enabled())
 223       return kNoAlternateProtocol;
 224
 225     if (session_->HasSpdyExclusion(origin))
 226       return kNoAlternateProtocol;
 227
 228     *alternate_url = UpgradeUrlToHttps(original_url, alternate.port);
 229   } else {
 230     DCHECK_EQ(QUIC, alternate.protocol);
 231     if (!session_->params().enable_quic)
 232         return kNoAlternateProtocol;
 233
 234     // TODO(rch):  Figure out how to make QUIC iteract with PAC
 235     // scripts.  By not re-writing the URL, we will query the PAC script
 236     // for the proxy to use to reach the original URL via TCP.  But
 237     // the alternate request will be going via UDP to a different port.
 238     *alternate_url = original_url;
 239   }
 240   return alternate;
 241 }
 242
 243 void HttpStreamFactoryImpl::OrphanJob(Job* job, const Request* request) {
 244   DCHECK(ContainsKey(request_map_, job));
 245   DCHECK_EQ(request_map_[job], request);
 246   DCHECK(!ContainsKey(orphaned_job_set_, job));
 247
 248   request_map_.erase(job);
 249
 250   orphaned_job_set_.insert(job);
 251   job->Orphan(request);
 252 }
 253
 254 void HttpStreamFactoryImpl::OnNewSpdySessionReady(
 255     const base::WeakPtr<SpdySession>& spdy_session,
 256     bool direct,
 257     const SSLConfig& used_ssl_config,
 258     const ProxyInfo& used_proxy_info,
 259     bool was_npn_negotiated,
 260     NextProto protocol_negotiated,
 261     bool using_spdy,
 262     const BoundNetLog& net_log) {
 263   while (true) {
 264     if (!spdy_session)
 265       break;
 266     const SpdySessionKey& spdy_session_key = spdy_session->spdy_session_key();
 267     // Each iteration may empty out the RequestSet for |spdy_session_key| in
 268     // |spdy_session_request_map_|. So each time, check for RequestSet and use
 269     // the first one.
 270     //
 271     // TODO(willchan): If it's important, switch RequestSet out for a FIFO
 272     // queue (Order by priority first, then FIFO within same priority). Unclear
 273     // that it matters here.
 274     if (!ContainsKey(spdy_session_request_map_, spdy_session_key))
 275       break;
 276     Request* request = *spdy_session_request_map_[spdy_session_key].begin();
 277     request->Complete(was_npn_negotiated,
 278                       protocol_negotiated,
 279                       using_spdy,
 280                       net_log);
 281     if (for_websockets_) {
 282       // TODO(ricea): Restore this code path when WebSocket over SPDY
 283       // implementation is ready.
 284       NOTREACHED();
 285     } else {
 286       bool use_relative_url = direct || request->url().SchemeIs("https");
 287       request->OnStreamReady(
 288           NULL,
 289           used_ssl_config,
 290           used_proxy_info,
 291           new SpdyHttpStream(spdy_session, use_relative_url));
 292     }
 293   }
 294   // TODO(mbelshe): Alert other valid requests.
 295 }
 296
 297 void HttpStreamFactoryImpl::OnOrphanedJobComplete(const Job* job) {
 298   orphaned_job_set_.erase(job);
 299   delete job;
 300 }
 301
 302 void HttpStreamFactoryImpl::OnPreconnectsComplete(const Job* job) {
 303   preconnect_job_set_.erase(job);
 304   delete job;
 305   OnPreconnectsCompleteInternal();
 306 }
 307
 308 }  // namespace net