servificied service worker interception

content/browser/service_worker/service_worker_controllee_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_CONTROLLEE_REQUEST_HANDLER_H_
 #define CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_CONTROLLEE_REQUEST_HANDLER_H_
 
 #include <stdint.h>
 
 #include "base/gtest_prod_util.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/time/time.h"
+#include "content/browser/service_worker/service_worker_controllee_url_loader_factory.h"
 #include "content/browser/service_worker/service_worker_request_handler.h"
 #include "content/browser/service_worker/service_worker_url_request_job.h"
 #include "content/common/service_worker/service_worker_types.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 "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorkerResponseType.h"
 #include "url/gurl.h"
 
 namespace net {
 class NetworkDelegate;
 class URLRequest;
 }
 
 namespace content {
 
 class ResourceRequestBodyImpl;
 class ServiceWorkerRegistration;
 class ServiceWorkerVersion;
 
 // A request handler derivative used to handle requests from
 // controlled documents.
 class CONTENT_EXPORT ServiceWorkerControlleeRequestHandler
     : public ServiceWorkerRequestHandler,
-      public ServiceWorkerURLRequestJob::Delegate {
+      public ServiceWorkerURLRequestJob::Delegate,
+      public ServiceWorkerControlleeURLLoaderFactory::Delegate {
  public:
   ServiceWorkerControlleeRequestHandler(
       base::WeakPtr<ServiceWorkerContextCore> context,
       base::WeakPtr<ServiceWorkerProviderHost> provider_host,
       base::WeakPtr<storage::BlobStorageContext> blob_storage_context,
       FetchRequestMode request_mode,
       FetchCredentialsMode credentials_mode,
       FetchRedirectMode redirect_mode,
       ResourceType resource_type,
       RequestContextType request_context_type,
       RequestContextFrameType frame_type,
+#if 0
+      scoped_refptr<ResourceRequestBodyImpl> body,
+      NetworkFallbackCallback network_fallback_callback);
+#endif
       scoped_refptr<ResourceRequestBodyImpl> body);
   ~ServiceWorkerControlleeRequestHandler() override;
 
   // Called via custom URLRequestJobFactory.
   net::URLRequestJob* MaybeCreateJob(
       net::URLRequest* request,
       net::NetworkDelegate* network_delegate,
       ResourceContext* resource_context) override;
 
+  mojom::URLLoaderFactoryPtr MaybeGetURLLoaderFactory(
+      const ResourceRequest& resource_request,
+      ResourceContext* resource_context,
+      std::unique_ptr<ServiceWorkerRequestHandler> request_handler) override;
+
+  class JobWrapper;
+
  private:
   FRIEND_TEST_ALL_PREFIXES(ServiceWorkerControlleeRequestHandlerTest,
                            ActivateWaitingVersion);
   typedef ServiceWorkerControlleeRequestHandler self;
 
   // For main resource case.
-  void PrepareForMainResource(const net::URLRequest* request);
+  void PrepareForMainResource(const GURL& url,
+                              const GURL& first_party_for_cookies);
   void DidLookupRegistrationForMainResource(
       ServiceWorkerStatusCode status,
       scoped_refptr<ServiceWorkerRegistration> registration);
   void OnVersionStatusChanged(
       ServiceWorkerRegistration* registration,
       ServiceWorkerVersion* version);
 
   void DidUpdateRegistration(
       const scoped_refptr<ServiceWorkerRegistration>& original_registration,
       ServiceWorkerStatusCode status,
@@ skipping 17 matching lines @@
   bool RequestStillValid(
       ServiceWorkerMetrics::URLRequestJobResult* result) override;
   void MainResourceLoadFailed() override;
 
   // Sets |job_| to nullptr, and clears all extra response info associated with
   // that job, except for timing information.
   void ClearJob();
 
   const bool is_main_resource_load_;
   const bool is_main_frame_load_;
-  base::WeakPtr<ServiceWorkerURLRequestJob> job_;
+  std::unique_ptr<JobWrapper> job_;
   FetchRequestMode request_mode_;
   FetchCredentialsMode credentials_mode_;
   FetchRedirectMode redirect_mode_;
   RequestContextType request_context_type_;
   RequestContextFrameType frame_type_;
   scoped_refptr<ResourceRequestBodyImpl> body_;
   ResourceContext* resource_context_;
   GURL stripped_url_;
   bool force_update_started_;
 
   // True if the next time this request is started, the response should be
   // delivered from the network, bypassing the ServiceWorker. Cleared after the
   // next intercept opportunity, for main frame requests.
   bool use_network_;
 
+  // When --enable-network-service, used to fallback to the network if the SW
+  // decides a request needs to be restarted after it's already returned the
+  // URLLoaderFactory in the main creation flow.
+  // NetworkFallbackCallback network_fallback_callback_;
+
   base::WeakPtrFactory<ServiceWorkerControlleeRequestHandler> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ServiceWorkerControlleeRequestHandler);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_CONTROLLEE_REQUEST_HANDLER_H_