net/http/proxy_client_socket.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/proxy_client_socket.h"
   6
   7 #include "base/metrics/histogram.h"
   8 #include "base/stringprintf.h"
   9 #include "googleurl/src/gurl.h"
  10 #include "net/base/host_port_pair.h"
  11 #include "net/base/net_errors.h"
  12 #include "net/base/net_util.h"
  13 #include "net/http/http_auth_controller.h"
  14 #include "net/http/http_request_info.h"
  15 #include "net/http/http_response_headers.h"
  16 #include "net/http/http_response_info.h"
  17
  18 namespace net {
  19
  20 // static
  21 void ProxyClientSocket::BuildTunnelRequest(
  22     const HttpRequestInfo& request_info,
  23     const HttpRequestHeaders& auth_headers,
  24     const HostPortPair& endpoint,
  25     std::string* request_line,
  26     HttpRequestHeaders* request_headers) {
  27   // RFC 2616 Section 9 says the Host request-header field MUST accompany all
  28   // HTTP/1.1 requests.  Add "Proxy-Connection: keep-alive" for compat with
  29   // HTTP/1.0 proxies such as Squid (required for NTLM authentication).
  30   *request_line = StringPrintf(
  31       "CONNECT %s HTTP/1.1\r\n", endpoint.ToString().c_str());
  32   request_headers->SetHeader(HttpRequestHeaders::kHost,
  33                              GetHostAndOptionalPort(request_info.url));
  34   request_headers->SetHeader(HttpRequestHeaders::kProxyConnection,
  35                              "keep-alive");
  36
  37   std::string user_agent;
  38   if (request_info.extra_headers.GetHeader(HttpRequestHeaders::kUserAgent,
  39                                             &user_agent))
  40     request_headers->SetHeader(HttpRequestHeaders::kUserAgent, user_agent);
  41
  42   request_headers->MergeFrom(auth_headers);
  43 }
  44
  45 // static
  46 int ProxyClientSocket::HandleProxyAuthChallenge(HttpAuthController* auth,
  47                                                 HttpResponseInfo* response,
  48                                                 const BoundNetLog& net_log) {
  49   DCHECK(response->headers);
  50   int rv = auth->HandleAuthChallenge(response->headers, false, true, net_log);
  51   response->auth_challenge = auth->auth_info();
  52   if (rv == OK)
  53     return ERR_PROXY_AUTH_REQUESTED;
  54   return rv;
  55 }
  56
  57 // static
  58 void ProxyClientSocket::LogBlockedTunnelResponse(int http_status_code,
  59                                                  const GURL& url,
  60                                                  bool is_https_proxy) {
  61   if (is_https_proxy) {
  62     UMA_HISTOGRAM_CUSTOM_ENUMERATION(
  63         "Net.BlockedTunnelResponse.HttpsProxy",
  64         HttpUtil::MapStatusCodeForHistogram(http_status_code),
  65         HttpUtil::GetStatusCodesForHistogram());
  66   } else {
  67     UMA_HISTOGRAM_CUSTOM_ENUMERATION(
  68         "Net.BlockedTunnelResponse.HttpProxy",
  69         HttpUtil::MapStatusCodeForHistogram(http_status_code),
  70         HttpUtil::GetStatusCodesForHistogram());
  71   }
  72 }
  73
  74 // static
  75 bool ProxyClientSocket::SanitizeProxyRedirect(HttpResponseInfo* response,
  76                                               const GURL& url) {
  77   DCHECK(response && response->headers);
  78
  79   std::string location;
  80   if (!response->headers->IsRedirect(&location))
  81     return false;
  82
  83   // Return minimal headers; set "Content-length: 0" to ignore response body.
  84   std::string fake_response_headers =
  85       base::StringPrintf("HTTP/1.0 302 Found\n"
  86                          "Location: %s\n"
  87                          "Content-length: 0\n"
  88                          "Connection: close\n"
  89                          "\n",
  90                          location.c_str());
  91   std::string raw_headers =
  92       HttpUtil::AssembleRawHeaders(fake_response_headers.data(),
  93                                    fake_response_headers.length());
  94   response->headers = new HttpResponseHeaders(raw_headers);
  95
  96   return true;
  97 }
  98
  99 }  // namespace net