Add embedder-specific headers to HTTP CONNECT tunnel request

components/data_reduction_proxy/browser/data_reduction_proxy_auth_request_handler.h

 // 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.
 
 #ifndef COMPONENTS_DATA_REDUCTION_PROXY_BROWSER_DATA_REDUCTION_PROXY_AUTH_REQUEST_HANDLER_H_
 #define COMPONENTS_DATA_REDUCTION_PROXY_BROWSER_DATA_REDUCTION_PROXY_AUTH_REQUEST_HANDLER_H_
 
 #include "base/gtest_prod_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/strings/string16.h"
 #include "base/time/time.h"
 #include "url/gurl.h"
 
 namespace base {
 class SingleThreadTaskRunner;
 }
 
 namespace net {
+class HostPortPair;
 class HttpRequestHeaders;
 class HttpResponseHeaders;
 class ProxyServer;
 class URLRequest;
 }
 
 namespace data_reduction_proxy {
 
 #if defined(OS_ANDROID)
 extern const char kAndroidWebViewProtocolVersion[];
@@ skipping 12 matching lines @@
   DataReductionProxyAuthRequestHandler(
       const std::string& client,
       const std::string& version,
       DataReductionProxyParams* params,
       scoped_refptr<base::SingleThreadTaskRunner> network_task_runner);
 
   virtual ~DataReductionProxyAuthRequestHandler();
 
   // Adds a 'Chrome-Proxy' header to |request_headers| with the data reduction
   // proxy authentication credentials. Only adds this header if the provided
-  // |proxy_server| is a data reduction proxy. Must be called on the IO thread.
+  // |proxy_server| is a data reduction proxy and not the data reduction proxy's
+  // CONNECT server. Must be called on the IO thread.
   void MaybeAddRequestHeader(net::URLRequest* request,
                              const net::ProxyServer& proxy_server,
                              net::HttpRequestHeaders* request_headers);
 
+  // Adds a 'Chrome-Proxy' header to |request_headers| with the data reduction
+  // proxy authentication credentials. Only adds this header if the provided
+  // |proxy_server| is the data reduction proxy's CONNECT server. Must be called
+  // on the IO thread.
+  void MaybeAddProxyTunnelRequestHandler(
+      const net::HostPortPair& proxy_server,
+      net::HttpRequestHeaders* request_headers);
+
   // Sets a new authentication key. This must be called for platforms that do
   // not have a default key defined. See the constructor implementation for
   // those platforms. Must be called on the UI thread.
   void SetKeyOnUI(const std::string& key);
 
  protected:
   void Init();
   void InitAuthenticationOnUI(const std::string& key);
 
   void AddAuthorizationHeader(net::HttpRequestHeaders* headers);
@@ skipping 19 matching lines @@
   // Stores the supplied key and sets up credentials suitable for authenticating
   // with the data reduction proxy.
   void InitAuthentication(const std::string& key);
 
   // Generates a session ID and credentials suitable for authenticating with
   // the data reduction proxy.
   void ComputeCredentials(const base::Time& now,
                           std::string* session,
                           std::string* credentials);
 
+  // Adds authentication headers only if |expects_ssl| is true and
+  // |proxy_server| is a data reduction proxy used for ssl tunneling via
+  // HTTP CONNECT, or |expect_ssl| is false and |proxy_server| is a data
+  // reduction proxy for HTTP traffic.
+  void MaybeAddRequestHeaderImpl(const net::HostPortPair& proxy_server,
+                                 bool expect_ssl,
+                                 net::HttpRequestHeaders* request_headers);
+
   // Authentication state.
   std::string key_;
 
   // Lives on the IO thread.
   std::string session_;
   std::string credentials_;
 
   // Name of the client and version of the data reduction proxy protocol to use.
   // Both live on the IO thread.
   std::string client_;
   std::string version_;
 
   // The last time the session was updated. Used to ensure that a session is
   // never used for more than twenty-four hours.
   base::Time last_update_time_;
 
   DataReductionProxyParams* data_reduction_proxy_params_;
 
   scoped_refptr<base::SingleThreadTaskRunner> network_task_runner_;
 
   DISALLOW_COPY_AND_ASSIGN(DataReductionProxyAuthRequestHandler);
 };
 
 }  // namespace data_reduction_proxy
 #endif  // COMPONENTS_DATA_REDUCTION_PROXY_BROWSER_DATA_REDUCTION_PROXY_AUTH_REQUEST_HANDLER_H_