[Mojo-Loading] Dispatch body data after response is received

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 71 matching lines @@
       : binding_(this),
         request_id_(request_id),
         resource_dispatcher_(resource_dispatcher),
         task_runner_(std::move(task_runner)) {}
   ~URLLoaderClientImpl() override {
     if (body_consumer_)
       body_consumer_->Cancel();
   }
 
   void OnReceiveResponse(const ResourceResponseHead& response_head) override {
+    has_received_response_ = true;
+    if (body_consumer_)
+      body_consumer_->Start(task_runner_.get());
     resource_dispatcher_->OnMessageReceived(
         ResourceMsg_ReceivedResponse(request_id_, response_head));
   }
 
   void OnStartLoadingResponseBody(
       mojo::ScopedDataPipeConsumerHandle body) override {
     DCHECK(!body_consumer_);
     body_consumer_ = new URLResponseBodyConsumer(
         request_id_, resource_dispatcher_, std::move(body), task_runner_);
+    if (has_received_response_)
+      body_consumer_->Start(task_runner_.get());
   }
 
   void OnComplete(const ResourceRequestCompletionStatus& status) override {
     if (!body_consumer_) {
       resource_dispatcher_->OnMessageReceived(
           ResourceMsg_RequestComplete(request_id_, status));
       return;
     }
     body_consumer_->OnComplete(status);
   }
 
   mojom::URLLoaderClientPtr CreateInterfacePtrAndBind() {
     return binding_.CreateInterfacePtrAndBind();
   }
 
  private:
   mojo::Binding<mojom::URLLoaderClient> binding_;
   scoped_refptr<URLResponseBodyConsumer> body_consumer_;
   const int request_id_;
+  bool has_received_response_ = false;
   ResourceDispatcher* const resource_dispatcher_;
   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
 };
 
 void CheckSchemeForReferrerPolicy(const ResourceRequest& request) {
   if ((request.referrer_policy == blink::WebReferrerPolicyDefault ||
        request.referrer_policy ==
            blink::WebReferrerPolicyNoReferrerWhenDowngrade) &&
       request.referrer.SchemeIsCryptographic() &&
       !request.url.SchemeIsCryptographic()) {
@@ skipping 693 matching lines @@
     delete message;
   }
 }
 
 void ResourceDispatcher::SetResourceSchedulingFilter(
     scoped_refptr<ResourceSchedulingFilter> resource_scheduling_filter) {
   resource_scheduling_filter_ = resource_scheduling_filter;
 }
 
 }  // namespace content