Implement transfer navigation with mojo

content/browser/loader/navigation_resource_throttle.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/loader/navigation_resource_throttle.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "base/callback.h"
+#include "base/callback_helpers.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted.h"
 #include "content/browser/frame_host/navigation_handle_impl.h"
 #include "content/browser/frame_host/navigator.h"
 #include "content/browser/frame_host/render_frame_host_impl.h"
 #include "content/browser/loader/navigation_resource_handler.h"
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
 #include "content/browser/loader/resource_loader.h"
 #include "content/browser/loader/resource_request_info_impl.h"
@@ skipping 151 matching lines @@
       render_frame_host, headers, connection_info, ssl_status, request_id,
       should_replace_current_entry, is_download, is_stream, transfer_callback,
       base::Bind(&SendCheckResultToIOThread, callback));
 }
 
 }  // namespace
 
 NavigationResourceThrottle::NavigationResourceThrottle(
     net::URLRequest* request,
     ResourceDispatcherHostDelegate* resource_dispatcher_host_delegate,
-    RequestContextType request_context_type)
+    RequestContextType request_context_type,
+    const TransferCallback& on_transfer)
     : request_(request),
       resource_dispatcher_host_delegate_(resource_dispatcher_host_delegate),
       request_context_type_(request_context_type),
       in_cross_site_transition_(false),
       on_transfer_done_result_(NavigationThrottle::DEFER),
+      on_transfer_(on_transfer),
       weak_ptr_factory_(this) {}
 
 NavigationResourceThrottle::~NavigationResourceThrottle() {}
 
 void NavigationResourceThrottle::WillStartRequest(bool* defer) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   const ResourceRequestInfoImpl* info =
       ResourceRequestInfoImpl::ForRequest(request_);
   if (!info)
     return;
@@ skipping 157 matching lines @@
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   in_cross_site_transition_ = true;
   ResourceRequestInfoImpl* info =
       ResourceRequestInfoImpl::ForRequest(request_);
   ResourceDispatcherHostImpl::Get()->MarkAsTransferredNavigation(
       info->GetGlobalRequestID(),
       base::Bind(&NavigationResourceThrottle::OnTransferComplete,
                  weak_ptr_factory_.GetWeakPtr()));
 }
 
-void NavigationResourceThrottle::OnTransferComplete() {
+void NavigationResourceThrottle::OnTransferComplete(
+    mojom::URLLoaderAssociatedRequest mojo_request,
+    mojom::URLLoaderClientAssociatedPtr url_loader_client) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   DCHECK(in_cross_site_transition_);
   in_cross_site_transition_ = false;
 
+  if (on_transfer_) {

[clamy, 2016/11/21 17:55:54]

Could you dcheck this only happens when loading with mojo is activated?

+    base::ResetAndReturn(&on_transfer_)
+        .Run(std::move(mojo_request), std::move(url_loader_client));
+  }
+
   // If the results of the checks on the UI thread are known, unblock the
   // navigation. Otherwise, wait until the callback has executed.
   if (on_transfer_done_result_ != NavigationThrottle::DEFER) {
     OnUIChecksPerformed(on_transfer_done_result_);
     on_transfer_done_result_ = NavigationThrottle::DEFER;
   }
 }
 
 }  // namespace content