PlzNavigate data pipe

content/child/resource_dispatcher.cc

 // Copyright (c) 2012 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.
 
 // See http://dev.chromium.org/developers/design-documents/multi-process-resource-loading
 
 #include "content/child/resource_dispatcher.h"
 
 #include <utility>
 
@@ skipping 602 matching lines @@
   response->encoded_body_length = result.encoded_body_length;
 }
 
 int ResourceDispatcher::StartAsync(
     std::unique_ptr<ResourceRequest> request,
     int routing_id,
     scoped_refptr<base::SingleThreadTaskRunner> loading_task_runner,
     const url::Origin& frame_origin,
     std::unique_ptr<RequestPeer> peer,
     blink::WebURLRequest::LoadingIPCType ipc_type,
-    mojom::URLLoaderFactory* url_loader_factory) {
+    mojom::URLLoaderFactory* url_loader_factory,
+    mojo::ScopedDataPipeConsumerHandle consumer_handle) {
   CheckSchemeForReferrerPolicy(*request);
 
   // Compute a unique request_id for this renderer process.
   int request_id = MakeRequestID();
   pending_requests_[request_id] = base::MakeUnique<PendingRequestInfo>(
       std::move(peer), request->resource_type, request->origin_pid,
       frame_origin, request->url, request->download_to_file);
 
   if (resource_scheduling_filter_.get() && loading_task_runner) {
     resource_scheduling_filter_->SetRequestIdTaskRunner(request_id,
                                                         loading_task_runner);
   }
 
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner =
+      loading_task_runner ? loading_task_runner : main_thread_task_runner_;
+
+  if (consumer_handle.is_valid()) {
+    std::unique_ptr<URLLoaderClientImpl> client(
+        new URLLoaderClientImpl(request_id, this, task_runner));
+    pending_requests_[request_id]->url_loader_client = std::move(client);
+
+    task_runner->PostTask(FROM_HERE,
+                          base::Bind(&ResourceDispatcher::ContinueForNavigation,
+                                     weak_factory_.GetWeakPtr(), request_id,
+                                     base::Passed(std::move(consumer_handle))));
+
+    return request_id;
+  }
+
   if (ipc_type == blink::WebURLRequest::LoadingIPCType::Mojo) {
-    scoped_refptr<base::SingleThreadTaskRunner> task_runner =
-        loading_task_runner ? loading_task_runner : main_thread_task_runner_;
     std::unique_ptr<URLLoaderClientImpl> client(
         new URLLoaderClientImpl(request_id, this, std::move(task_runner)));
     mojom::URLLoaderAssociatedPtr url_loader;
     mojom::URLLoaderClientPtr client_ptr;
     client->Bind(&client_ptr);
     url_loader_factory->CreateLoaderAndStart(MakeRequest(&url_loader),
                                              routing_id, request_id, *request,
                                              std::move(client_ptr));
     pending_requests_[request_id]->url_loader = std::move(url_loader);
     pending_requests_[request_id]->url_loader_client = std::move(client);
@@ skipping 78 matching lines @@
 }
 
 base::TimeTicks ResourceDispatcher::ConsumeIOTimestamp() {
   if (io_timestamp_ == base::TimeTicks())
     return base::TimeTicks::Now();
   base::TimeTicks result = io_timestamp_;
   io_timestamp_ = base::TimeTicks();
   return result;
 }
 
+void ResourceDispatcher::ContinueForNavigation(
+    int request_id,
+    mojo::ScopedDataPipeConsumerHandle consumer_handle) {
+  PendingRequestInfo* request_info = GetPendingRequestInfo(request_id);
+  if (!request_info)
+    return;
+
+  URLLoaderClientImpl* client_ptr = request_info->url_loader_client.get();
+
+  // Short circuiting call to OnReceivedResponse to immediately start
+  // the request. ResourceResponseHead can be empty here because we
+  // pull the StreamOverride's one in
+  // WebURLLoaderImpl::Context::OnReceivedResponse.
+  client_ptr->OnReceiveResponse(ResourceResponseHead(),
+                                mojom::DownloadedTempFilePtr());
+  // Start streaming now.
+  client_ptr->OnStartLoadingResponseBody(std::move(consumer_handle));
+
+  // Call OnComplete now too, as it won't get called on the client.
+  // TODO(kinuko): Fill this properly.
+  ResourceRequestCompletionStatus completion_status;
+  completion_status.error_code = net::OK;
+  completion_status.was_ignored_by_handler = false;
+  completion_status.exists_in_cache = false;
+  completion_status.completion_time = base::TimeTicks::Now();
+  completion_status.encoded_data_length = -1;
+  completion_status.encoded_body_length = -1;
+  client_ptr->OnComplete(completion_status);
+}
+
 // static
 bool ResourceDispatcher::IsResourceDispatcherMessage(
     const IPC::Message& message) {
   switch (message.type()) {
     case ResourceMsg_UploadProgress::ID:
     case ResourceMsg_ReceivedResponse::ID:
     case ResourceMsg_ReceivedCachedMetadata::ID:
     case ResourceMsg_ReceivedRedirect::ID:
     case ResourceMsg_SetDataBuffer::ID:
     case ResourceMsg_DataReceived::ID:
@@ skipping 40 matching lines @@
     delete message;
   }
 }
 
 void ResourceDispatcher::SetResourceSchedulingFilter(
     scoped_refptr<ResourceSchedulingFilter> resource_scheduling_filter) {
   resource_scheduling_filter_ = resource_scheduling_filter;
 }
 
 }  // namespace content