Add a function to add extra headers from NavigationThrottle

content/browser/frame_host/navigation_handle_impl.cc

 // Copyright 2015 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 "content/browser/frame_host/navigation_handle_impl.h"
 
 #include "base/command_line.h"
 #include "content/browser/frame_host/frame_tree_node.h"
 #include "content/browser/frame_host/navigator.h"
 #include "content/browser/frame_host/navigator_delegate.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
 #include "content/browser/service_worker/service_worker_navigation_handle.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/storage_partition.h"
 #include "content/public/common/content_client.h"
 #include "content/public/common/content_switches.h"
 #include "net/url_request/redirect_info.h"
 #include "third_party/WebKit/public/web/WebSandboxFlags.h"
 
 namespace content {
 
 namespace {
 
 void UpdateThrottleCheckResult(
     NavigationThrottle::ThrottleCheckResult* to_update,
-    NavigationThrottle::ThrottleCheckResult result) {
+    NavigationThrottle::ThrottleCheckResult result,
+    const NavigationHandleImpl::ExtraHeadersList& headers) {
   *to_update = result;
 }
 
 }  // namespace
 
 // static
 scoped_ptr<NavigationHandleImpl> NavigationHandleImpl::Create(
     const GURL& url,
     FrameTreeNode* frame_tree_node) {
   return scoped_ptr<NavigationHandleImpl>(
@@ skipping 173 matching lines @@
   NavigationThrottle::ThrottleCheckResult result = NavigationThrottle::DEFER;
   WillRedirectRequest(new_url, new_method_is_post, new_referrer_url,
                       new_is_external_protocol,
                       base::Bind(&UpdateThrottleCheckResult, &result));
 
   // Reset the callback to ensure it will not be called later.
   complete_callback_.Reset();
   return result;
 }
 
+void NavigationHandleImpl::AddExtraHeader(const std::string& name,
+                                          const std::string& value) {
+  // Only set headers during throttle checks, when the handler of the request
+  // will be able to add them to the network request.
+  DCHECK(!complete_callback_.is_null());
+  extra_headers_to_add_.push_back(
+      std::pair<std::string, std::string>(name, value));
+}
+
 void NavigationHandleImpl::WillStartRequest(
     bool is_post,
     const Referrer& sanitized_referrer,
     bool has_user_gesture,
     ui::PageTransition transition,
     bool is_external_protocol,
     const ThrottleChecksFinishedCallback& callback) {
   // Update the navigation parameters.
   is_post_ = is_post;
   sanitized_referrer_ = sanitized_referrer;
@@ skipping 129 matching lines @@
   }
   next_index_ = 0;
   state_ = WILL_REDIRECT_REQUEST;
   return NavigationThrottle::PROCEED;
 }
 
 void NavigationHandleImpl::RunCompleteCallback(
     NavigationThrottle::ThrottleCheckResult result) {
   DCHECK(result != NavigationThrottle::DEFER);
   if (!complete_callback_.is_null())
-    complete_callback_.Run(result);
+    complete_callback_.Run(result, extra_headers_to_add_);
 
   complete_callback_.Reset();
+  extra_headers_to_add_.clear();
 }
 
 }  // namespace content