network service: Create URLLoader for service worker navigation case

content/browser/service_worker/service_worker_request_handler.h

 // Copyright 2014 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.
 
 #ifndef CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_REQUEST_HANDLER_H_
 #define CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_REQUEST_HANDLER_H_
 
 #include <memory>
 
 #include "base/callback_forward.h"
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "base/supports_user_data.h"
 #include "base/time/time.h"
+#include "content/browser/service_worker/service_worker_url_loader_factory.h"
 #include "content/common/content_export.h"
 #include "content/common/service_worker/service_worker_status_code.h"
 #include "content/common/service_worker/service_worker_types.h"
+#include "content/common/url_loader_factory.mojom.h"
 #include "content/public/common/request_context_frame_type.h"
 #include "content/public/common/request_context_type.h"
 #include "content/public/common/resource_type.h"
 #include "net/url_request/url_request_job_factory.h"
 
 namespace net {
 class NetworkDelegate;
 class URLRequest;
 class URLRequestInterceptor;
 }
 
 namespace storage {
 class BlobStorageContext;
 }
 
 namespace content {
 
 class ResourceContext;
+struct ResourceRequest;
 class ResourceRequestBodyImpl;
 class ServiceWorkerContextCore;
 class ServiceWorkerContextWrapper;
 class ServiceWorkerNavigationHandleCore;
 class ServiceWorkerProviderHost;
 class WebContents;
 
 // Abstract base class for routing network requests to ServiceWorkers.
 // Created one per URLRequest and attached to each request.
 class CONTENT_EXPORT ServiceWorkerRequestHandler
     : public base::SupportsUserData::Data {
  public:
   // PlzNavigate
   // Attaches a newly created handler if the given |request| needs to be handled
   // by ServiceWorker.
   static void InitializeForNavigation(
-      net::URLRequest* request,
+      net::URLRequest* url_request,
       ServiceWorkerNavigationHandleCore* navigation_handle_core,
       storage::BlobStorageContext* blob_storage_context,
       bool skip_service_worker,
+      ResourceType resource_type,
+      RequestContextType request_context_type,
+      RequestContextFrameType frame_type,
+      bool is_parent_frame_secure,
+      scoped_refptr<ResourceRequestBodyImpl> body,
+      const base::Callback<WebContents*(void)>& web_contents_getter);
+
+  // PlzNavigate and --enable-network-service
+  // Same as InitializeForNavigation(), but instead of attaching to a request,
+  // returns a URLLoaderFactoryPtr if the request needs to be handled.
+  static mojom::URLLoaderFactoryPtrInfo InitializeForNavigationNetworkService(
+      const ResourceRequest& resource_request,
+      ResourceContext* resource_context,
+      ServiceWorkerNavigationHandleCore* navigation_handle_core,
+      storage::BlobStorageContext* blob_storage_context,
+      bool skip_service_worker,
       ResourceType resource_type,
       RequestContextType request_context_type,
       RequestContextFrameType frame_type,
       bool is_parent_frame_secure,
       scoped_refptr<ResourceRequestBodyImpl> body,
       const base::Callback<WebContents*(void)>& web_contents_getter);
 
   // Attaches a newly created handler if the given |request| needs to
   // be handled by ServiceWorker.
   // TODO(kinuko): While utilizing UserData to attach data to URLRequest
@@ skipping 36 matching lines @@
       const net::URLRequest* request);
 
   ~ServiceWorkerRequestHandler() override;
 
   // Called via custom URLRequestJobFactory.
   virtual net::URLRequestJob* MaybeCreateJob(
       net::URLRequest* request,
       net::NetworkDelegate* network_delegate,
       ResourceContext* context) = 0;
 
+  virtual mojom::URLLoaderFactoryPtrInfo MaybeCreateURLLoader(
+      const ResourceRequest& resource_request,
+      ResourceContext* resource_context,
+      std::unique_ptr<ServiceWorkerRequestHandler> request_handler) {
+    return mojom::URLLoaderFactoryPtrInfo();
+  }
+
   // Methods to support cross site navigations.
   void PrepareForCrossSiteTransfer(int old_process_id);
   void CompleteCrossSiteTransfer(int new_process_id,
                                  int new_provider_id);
   void MaybeCompleteCrossSiteTransferInOldProcess(
       int old_process_id);
 
   // Useful for detecting storage partition mismatches in the context of cross
   // site transfer navigations.
   bool SanityCheckIsSameContext(ServiceWorkerContextWrapper* wrapper);
 
+  using HandlerCreated =
+      base::Callback<void(std::unique_ptr<ServiceWorkerRequestHandler>)>;
+
  protected:
   ServiceWorkerRequestHandler(
       base::WeakPtr<ServiceWorkerContextCore> context,
       base::WeakPtr<ServiceWorkerProviderHost> provider_host,
       base::WeakPtr<storage::BlobStorageContext> blob_storage_context,
       ResourceType resource_type);
 
   base::WeakPtr<ServiceWorkerContextCore> context_;
   base::WeakPtr<ServiceWorkerProviderHost> provider_host_;
   base::WeakPtr<storage::BlobStorageContext> blob_storage_context_;
   ResourceType resource_type_;
 
  private:
+  static void InitializeForNavigationImpl(
+      const GURL& url,
+      ServiceWorkerNavigationHandleCore* navigation_handle_core,
+      storage::BlobStorageContext* blob_storage_context,
+      bool skip_service_worker,
+      ResourceType resource_type,
+      RequestContextType request_context_type,
+      RequestContextFrameType frame_type,
+      bool is_parent_frame_secure,
+      scoped_refptr<ResourceRequestBodyImpl> body,
+      const base::Callback<WebContents*(void)>& web_contents_getter,
+      const HandlerCreated& on_handler_created);
+
   std::unique_ptr<ServiceWorkerProviderHost> host_for_cross_site_transfer_;
   int old_process_id_;
   int old_provider_id_;
 
   DISALLOW_COPY_AND_ASSIGN(ServiceWorkerRequestHandler);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_REQUEST_HANDLER_H_