[Proof-of-concept] PlzNavigate and Service Worker

content/browser/service_worker/service_worker_request_handler.cc

 // 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.
 
 #include "content/browser/service_worker/service_worker_request_handler.h"
 
 #include <string>
 
+#include "base/command_line.h"
+#include "content/browser/loader/navigation_url_loader_impl_core.h"
 #include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
 #include "content/browser/service_worker/service_worker_provider_host.h"
 #include "content/browser/service_worker/service_worker_registration.h"
 #include "content/browser/service_worker/service_worker_url_request_job.h"
 #include "content/browser/service_worker/service_worker_utils.h"
 #include "content/common/resource_request_body.h"
 #include "content/common/service_worker/service_worker_types.h"
 #include "content/public/browser/resource_context.h"
+#include "content/public/common/content_switches.h"
 #include "net/base/net_util.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_interceptor.h"
 #include "storage/browser/blob/blob_storage_context.h"
-
 namespace content {
 
 namespace {
 
 int kUserDataKey;  // Key value is not important.
 
 class ServiceWorkerRequestInterceptor
     : public net::URLRequestInterceptor {
  public:
   explicit ServiceWorkerRequestInterceptor(ResourceContext* resource_context)
@@ skipping 23 matching lines @@
     storage::BlobStorageContext* blob_storage_context,
     int process_id,
     int provider_id,
     bool skip_service_worker,
     FetchRequestMode request_mode,
     FetchCredentialsMode credentials_mode,
     ResourceType resource_type,
     RequestContextType request_context_type,
     RequestContextFrameType frame_type,
     scoped_refptr<ResourceRequestBody> body) {
+  // PlzNavigate:
+  if (request->url().SchemeIsBlob()) {

[michaeln, 2015/07/24 22:22:27]

Is there a more direct way of identifying plznavigate requests? Using the
request url is a little overloaded.

(aside: Wow, so as far as the loader logic in the renderer knows, its loading a
blob?)

[horo, 2015/07/27 09:11:17]

Removed these code from ServiceWorkerRequestHandler::InitializeHandler().

+    scoped_ptr<ServiceWorkerProviderHost> navigation_provider_host(
+        context_wrapper->context()->TakeNavigationProviderHost(request->url()));
+    if (navigation_provider_host) {
+      context_wrapper->context()->TransferProviderHostIn(
+          process_id, provider_id, navigation_provider_host.Pass());
+    }
+  }
   if (!request->url().SchemeIsHTTPOrHTTPS())
     return;
 
   if (!context_wrapper || !context_wrapper->context() ||
       provider_id == kInvalidServiceWorkerProviderId) {
     return;
   }
 
   ServiceWorkerProviderHost* provider_host =
       context_wrapper->context()->GetProviderHost(process_id, provider_id);
   if (!provider_host || !provider_host->IsContextAlive())
     return;
 
   if (skip_service_worker) {
     if (ServiceWorkerUtils::IsMainResourceType(resource_type)) {
       provider_host->SetDocumentUrl(net::SimplifyUrlForRequest(request->url()));
       provider_host->SetTopmostFrameUrl(request->first_party_for_cookies());
       // A page load with skip_service_worker should be triggered by
       // shift-reload, so retain all live matching registrations.
       provider_host->AddAllMatchingRegistrations();
     }
     return;
   }
-
   scoped_ptr<ServiceWorkerRequestHandler> handler(
       provider_host->CreateRequestHandler(request_mode,
                                           credentials_mode,
                                           resource_type,
                                           request_context_type,
                                           frame_type,
                                           blob_storage_context->AsWeakPtr(),
                                           body));
   if (!handler)
     return;
 
   request->SetUserData(&kUserDataKey, handler.release());
 }
 
+void ServiceWorkerRequestHandler::InitializeNavigationHandler(
+    net::URLRequest* request,
+    ServiceWorkerContextWrapper* context_wrapper,
+    storage::BlobStorageContext* blob_storage_context,
+    bool skip_service_worker,
+    bool is_main_frame,
+    scoped_refptr<ResourceRequestBody> body,
+    NavigationURLLoaderImplCore* loader) {
+  CHECK(base::CommandLine::ForCurrentProcess()->HasSwitch(
+      switches::kEnableBrowserSideNavigation));
+  if (!request->url().SchemeIsHTTPOrHTTPS())
+    return;
+  if (!context_wrapper || !context_wrapper->context())
+    return;
+  scoped_ptr<ServiceWorkerProviderHost> provider_host(
+      context_wrapper->context()->CreateNavigationProviderHost());
+  loader->SetServiceWorkerProviderHost(provider_host.Pass());
+  if (skip_service_worker) {
+    // TODO(horo): Does it work correctly?.
+    provider_host->SetDocumentUrl(net::SimplifyUrlForRequest(request->url()));
+    provider_host->SetTopmostFrameUrl(request->first_party_for_cookies());
+    provider_host->AddAllMatchingRegistrations();
+    return;
+  }
+  scoped_ptr<ServiceWorkerRequestHandler> handler(
+      loader->service_worker_provider_host()->CreateRequestHandler(
+          FETCH_REQUEST_MODE_NO_CORS, FETCH_CREDENTIALS_MODE_SAME_ORIGIN,
+          is_main_frame ? RESOURCE_TYPE_MAIN_FRAME : RESOURCE_TYPE_SUB_FRAME,
+          REQUEST_CONTEXT_TYPE_LOCATION,         // TODO(horo): pass in
+                                                 // FrameHostMsg_BeginNavigation
+          REQUEST_CONTEXT_FRAME_TYPE_TOP_LEVEL,  // TODO(horo): pass in
+                                                 // FrameHostMsg_BeginNavigation
+          blob_storage_context->AsWeakPtr(), body));
+  if (!handler)
+    return;
+  request->SetUserData(&kUserDataKey, handler.release());
+}
+
 ServiceWorkerRequestHandler* ServiceWorkerRequestHandler::GetHandler(
     net::URLRequest* request) {
   return static_cast<ServiceWorkerRequestHandler*>(
       request->GetUserData(&kUserDataKey));
 }
 
 scoped_ptr<net::URLRequestInterceptor>
 ServiceWorkerRequestHandler::CreateInterceptor(
     ResourceContext* resource_context) {
   return scoped_ptr<net::URLRequestInterceptor>(
@@ skipping 52 matching lines @@
     ResourceType resource_type)
     : context_(context),
       provider_host_(provider_host),
       blob_storage_context_(blob_storage_context),
       resource_type_(resource_type),
       old_process_id_(0),
       old_provider_id_(kInvalidServiceWorkerProviderId) {
 }
 
 }  // namespace content