Use mojo for Chrome Loading, Part 1

content/child/resource_dispatcher.h

 // 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
 
 #ifndef CONTENT_CHILD_RESOURCE_DISPATCHER_H_
 #define CONTENT_CHILD_RESOURCE_DISPATCHER_H_
 
 #include <stdint.h>
 
 #include <deque>
 #include <map>
 #include <memory>
 #include <string>
 
 #include "base/containers/hash_tables.h"
 #include "base/macros.h"
 #include "base/memory/linked_ptr.h"
 #include "base/memory/shared_memory.h"
 #include "base/memory/weak_ptr.h"
 #include "base/single_thread_task_runner.h"
 #include "base/time/time.h"
 #include "content/common/content_export.h"
+#include "content/common/url_loader.mojom.h"
 #include "content/public/common/resource_type.h"
 #include "ipc/ipc_listener.h"
 #include "ipc/ipc_sender.h"
 #include "net/base/request_priority.h"
+#include "third_party/WebKit/public/platform/WebURLRequest.h"
 #include "url/gurl.h"
 
 namespace net {
 struct RedirectInfo;
 }
 
 namespace content {
 class RequestPeer;
 class ResourceDispatcherDelegate;
 class ResourceRequestBodyImpl;
 class ResourceSchedulingFilter;
 struct ResourceResponseInfo;
 struct RequestInfo;
 struct ResourceRequest;
 struct ResourceRequestCompletionStatus;
 struct ResourceResponseHead;
 class SharedMemoryReceivedDataFactory;
 struct SiteIsolationResponseMetaData;
 struct SyncLoadResponse;
 
+namespace mojom {
+class URLLoaderFactory;
+}  // namespace mojom
+
 // This class serves as a communication interface to the ResourceDispatcherHost
 // in the browser process. It can be used from any child process.
 // Virtual methods are for tests.
 class CONTENT_EXPORT ResourceDispatcher : public IPC::Listener {
  public:
   ResourceDispatcher(
       IPC::Sender* sender,
       scoped_refptr<base::SingleThreadTaskRunner> main_thread_task_runner);
   ~ResourceDispatcher() override;
 
   // IPC::Listener implementation.
   bool OnMessageReceived(const IPC::Message& message) override;
 
   // Call this method to load the resource synchronously (i.e., in one shot).
   // This is an alternative to the StartAsync method. Be warned that this method
   // will block the calling thread until the resource is fully downloaded or an
   // error occurs. It could block the calling thread for a long time, so only
   // use this if you really need it!  There is also no way for the caller to
   // interrupt this method. Errors are reported via the status field of the
   // response parameter.
   virtual void StartSync(const RequestInfo& request_info,
                          ResourceRequestBodyImpl* request_body,
-                         SyncLoadResponse* response);
+                         SyncLoadResponse* response,
+                         blink::WebURLRequest::LoadingIPCType ipc_type,
+                         mojom::URLLoaderFactory* url_loader_factory);
 
   // Call this method to initiate the request. If this method succeeds, then
   // the peer's methods will be called asynchronously to report various events.
-  // Returns the request id.
+  // Returns the request id. |url_loader_factory| must be non-null if and only
+  // if |ipc_type| is LoadingIPCType::Mojo.
   virtual int StartAsync(const RequestInfo& request_info,
                          ResourceRequestBodyImpl* request_body,
-                         std::unique_ptr<RequestPeer> peer);
+                         std::unique_ptr<RequestPeer> peer,
+                         blink::WebURLRequest::LoadingIPCType ipc_type,
+                         mojom::URLLoaderFactory* url_loader_factory);
 
   // Removes a request from the |pending_requests_| list, returning true if the
   // request was found and removed.
   bool RemovePendingRequest(int request_id);
 
   // Cancels a request in the |pending_requests_| list.  The request will be
   // removed from the dispatcher as well.
   virtual void Cancel(int request_id);
 
   // Toggles the is_deferred attribute for the specified request.
@@ skipping 23 matching lines @@
 
   void SetMainThreadTaskRunner(
       scoped_refptr<base::SingleThreadTaskRunner> main_thread_task_runner) {
     main_thread_task_runner_ = main_thread_task_runner;
   }
 
   void SetResourceSchedulingFilter(
       scoped_refptr<ResourceSchedulingFilter> resource_scheduling_filter);
 
  private:
+  friend class URLResponseBodyConsumer;
   friend class ResourceDispatcherTest;
 
   typedef std::deque<IPC::Message*> MessageQueue;
   struct PendingRequestInfo {
     PendingRequestInfo(std::unique_ptr<RequestPeer> peer,
                        ResourceType resource_type,
                        int origin_pid,
                        const GURL& frame_origin,
                        const GURL& request_url,
                        bool download_to_file);
@@ skipping 16 matching lines @@
     GURL response_url;
     bool download_to_file;
     std::unique_ptr<IPC::Message> pending_redirect_message;
     base::TimeTicks request_start;
     base::TimeTicks response_start;
     base::TimeTicks completion_time;
     linked_ptr<base::SharedMemory> buffer;
     scoped_refptr<SharedMemoryReceivedDataFactory> received_data_factory;
     std::unique_ptr<SiteIsolationResponseMetaData> site_isolation_metadata;
     int buffer_size;
+
+    // For mojo loading.
+    mojom::URLLoaderPtr url_loader;
+    std::unique_ptr<mojom::URLLoaderClient> url_loader_client;
   };
   using PendingRequestMap = std::map<int, std::unique_ptr<PendingRequestInfo>>;
 
   // Helper to lookup the info based on the request_id.
   // May return NULL if the request as been canceled from the client side.
   PendingRequestInfo* GetPendingRequestInfo(int request_id);
 
   // Follows redirect, if any, for the given request.
   void FollowPendingRedirect(int request_id, PendingRequestInfo* request_info);
 
@@ skipping 75 matching lines @@
   scoped_refptr<ResourceSchedulingFilter> resource_scheduling_filter_;
 
   base::WeakPtrFactory<ResourceDispatcher> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ResourceDispatcher);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_CHILD_RESOURCE_DISPATCHER_H_