Add embedder-specific headers to HTTP CONNECT tunnel request

components/data_reduction_proxy/browser/data_reduction_proxy_auth_request_handler.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/data_reduction_proxy/browser/data_reduction_proxy_auth_request_handler.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string_split.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
 #include "components/data_reduction_proxy/browser/data_reduction_proxy_params.h"
 #include "components/data_reduction_proxy/browser/data_reduction_proxy_protocol.h"
 #include "components/data_reduction_proxy/browser/data_reduction_proxy_settings.h"
 #include "components/data_reduction_proxy/common/data_reduction_proxy_headers.h"
 #include "components/data_reduction_proxy/common/data_reduction_proxy_switches.h"
 #include "crypto/random.h"
+#include "net/base/host_port_pair.h"
 #include "net/proxy/proxy_server.h"
 #include "net/url_request/url_request.h"
 #include "url/gurl.h"
 
 namespace data_reduction_proxy {
 
 // The empty version for the authentication protocol. Currently used by
 // Android webview.
 #if defined(OS_ANDROID)
 const char kAndroidWebViewProtocolVersion[] = "";
@@ skipping 66 matching lines @@
   crypto::RandBytes(output, length);
 }
 
 void DataReductionProxyAuthRequestHandler::MaybeAddRequestHeader(
     net::URLRequest* request,
     const net::ProxyServer& proxy_server,
     net::HttpRequestHeaders* request_headers) {
   DCHECK(network_task_runner_->BelongsToCurrentThread());
   if (!proxy_server.is_valid())
     return;
-  if (data_reduction_proxy_params_ &&
-      data_reduction_proxy_params_->IsDataReductionProxy(
-          proxy_server.host_port_pair(), NULL)) {
-    AddAuthorizationHeader(request_headers);
-  }
+  if (proxy_server.is_direct())
+    return;
+  MaybeAddRequestHeaderImpl(proxy_server.host_port_pair(),
+                            false,
+                            request_headers);
+}
+
+void DataReductionProxyAuthRequestHandler::MaybeAddProxyTunnelRequestHandler(
+    const net::HostPortPair& proxy_server,
+    net::HttpRequestHeaders* request_headers) {
+  DCHECK(network_task_runner_->BelongsToCurrentThread());
+  MaybeAddRequestHeaderImpl(proxy_server, true, request_headers);
 }
 
 void DataReductionProxyAuthRequestHandler::AddAuthorizationHeader(
     net::HttpRequestHeaders* headers) {
   base::Time now = Now();
   if (now - last_update_time_ > base::TimeDelta::FromHours(24)) {
     last_update_time_ = now;
     ComputeCredentials(last_update_time_, &session_, &credentials_);
   }
   const char kChromeProxyHeader[] = "Chrome-Proxy";
@@ skipping 60 matching lines @@
   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
   std::string key =
     command_line.GetSwitchValueASCII(switches::kDataReductionProxyKey);
 #if defined(SPDY_PROXY_AUTH_VALUE)
   if (key.empty())
     key = SPDY_PROXY_AUTH_VALUE;
 #endif
   return key;
 }
 
+void DataReductionProxyAuthRequestHandler::MaybeAddRequestHeaderImpl(
+    const net::HostPortPair& proxy_server,
+    bool expect_ssl,
+    net::HttpRequestHeaders* request_headers) {
+  if (proxy_server.IsEmpty())
+    return;
+  if (data_reduction_proxy_params_ &&
+      data_reduction_proxy_params_->IsDataReductionProxy(proxy_server, NULL) &&
+      net::HostPortPair::FromURL(
+          data_reduction_proxy_params_->ssl_origin()).Equals(
+              proxy_server) == expect_ssl) {
+    AddAuthorizationHeader(request_headers);
+  }
+}
+
 }  // namespace data_reduction_proxy