AppCache and StoragePartition'ing

content/browser/worker_host/worker_service_impl.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.
 
 #include "content/browser/worker_host/worker_service_impl.h"
 
 #include <string>
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/sys_info.h"
 #include "base/threading/thread.h"
 #include "content/browser/debugger/worker_devtools_manager.h"
+#include "content/browser/storage_partition_impl.h"
 #include "content/browser/worker_host/worker_message_filter.h"
 #include "content/browser/worker_host/worker_process_host.h"
 #include "content/common/view_messages.h"
 #include "content/common/worker_messages.h"
 #include "content/public/browser/child_process_data.h"
 #include "content/public/browser/resource_context.h"
 #include "content/public/browser/worker_service_observer.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/process_type.h"
 #include "net/base/registry_controlled_domains/registry_controlled_domain.h"
@@ skipping 65 matching lines @@
   // the queued workers, or a renderer has shut down, in which case it doesn't
   // affect anything.  We call this function in both scenarios because then we
   // don't have to keep track which filters are from worker processes.
   TryStartingQueuedWorker();
 }
 
 void WorkerServiceImpl::CreateWorker(
     const ViewHostMsg_CreateWorker_Params& params,
     int route_id,
     WorkerMessageFilter* filter,
-    ResourceContext* resource_context) {
+    ResourceContext* resource_context,
+    StoragePartitionImpl* storage_partition) {
   // Generate a unique route id for the browser-worker communication that's
   // unique among all worker processes.  That way when the worker process sends
   // a wrapped IPC message through us, we know which WorkerProcessHost to give
   // it to.
   WorkerProcessHost::WorkerInstance instance(
       params.url,
       params.name,
       next_worker_route_id(),
       0,
       params.script_resource_appcache_id,
-      resource_context);
+      resource_context,
+      storage_partition);
   instance.AddFilter(filter, route_id);
   instance.worker_document_set()->Add(
       filter, params.document_id, filter->render_process_id(),
       params.render_view_route_id);
 
   CreateWorkerFromInstance(instance);
 }
 
 void WorkerServiceImpl::LookupSharedWorker(
     const ViewHostMsg_CreateWorker_Params& params,
     int route_id,
     WorkerMessageFilter* filter,
     ResourceContext* resource_context,
+    StoragePartitionImpl* storage_partition,
     bool* exists,
     bool* url_mismatch) {
   *exists = true;
   WorkerProcessHost::WorkerInstance* instance = FindSharedWorkerInstance(
-      params.url, params.name, resource_context);
+      params.url, params.name, resource_context, storage_partition);
 
   if (!instance) {
     // If no worker instance currently exists, we need to create a pending
     // instance - this is to make sure that any subsequent lookups passing a
     // mismatched URL get the appropriate url_mismatch error at lookup time.
     // Having named shared workers was a Really Bad Idea due to details like
     // this.
-    instance = CreatePendingInstance(params.url, params.name, resource_context);
+    instance = CreatePendingInstance(params.url, params.name, resource_context,
+                                     storage_partition);
     *exists = false;
   }
 
   // Make sure the passed-in instance matches the URL - if not, return an
   // error.
   if (params.url != instance->url()) {
     *url_mismatch = true;
     *exists = false;
   } else {
     *url_mismatch = false;
@@ skipping 77 matching lines @@
       queued_workers_.push_back(instance);
       return true;
     }
   }
 
   // Check to see if this shared worker is already running (two pages may have
   // tried to start up the worker simultaneously).
   // See if a worker with this name already exists.
   WorkerProcessHost::WorkerInstance* existing_instance =
       FindSharedWorkerInstance(
-          instance.url(), instance.name(), instance.resource_context());
+          instance.url(), instance.name(), instance.resource_context(),
+          instance.storage_partition());
   WorkerProcessHost::WorkerInstance::FilterInfo filter_info =
       instance.GetFilter();
   // If this worker is already running, no need to create a new copy. Just
   // inform the caller that the worker has been created.
   if (existing_instance) {
     // Walk the worker's filter list to see if this client is listed. If not,
     // then it means that the worker started by the client already exited so
     // we should not attach to this new one (http://crbug.com/29243).
     if (!existing_instance->HasFilter(filter_info.first, filter_info.second))
       return false;
     filter_info.first->Send(new ViewMsg_WorkerCreated(filter_info.second));
     return true;
   }
 
   // Look to see if there's a pending instance.
   WorkerProcessHost::WorkerInstance* pending = FindPendingInstance(
-      instance.url(), instance.name(), instance.resource_context());
+      instance.url(), instance.name(), instance.resource_context(),
+      instance.storage_partition());
   // If there's no instance *and* no pending instance (or there is a pending
   // instance but it does not contain our filter info), then it means the
   // worker started up and exited already. Log a warning because this should
   // be a very rare occurrence and is probably a bug, but it *can* happen so
   // handle it gracefully.
   if (!pending ||
       !pending->HasFilter(filter_info.first, filter_info.second)) {
     DLOG(WARNING) << "Pending worker already exited";
     return false;
   }
 
   // Assign the accumulated document set and filter list for this pending
   // worker to the new instance.
   DCHECK(!pending->worker_document_set()->IsEmpty());
   instance.ShareDocumentSet(*pending);
   for (WorkerProcessHost::WorkerInstance::FilterList::const_iterator i =
            pending->filters().begin();
        i != pending->filters().end(); ++i) {
     instance.AddFilter(i->first, i->second);
   }
   RemovePendingInstances(
-      instance.url(), instance.name(), instance.resource_context());
+      instance.url(), instance.name(), instance.resource_context(),
+      instance.storage_partition());
 
   // Remove any queued instances of this worker and copy over the filter to
   // this instance.
   for (WorkerProcessHost::Instances::iterator iter = queued_workers_.begin();
        iter != queued_workers_.end();) {
     if (iter->Matches(instance.url(), instance.name(),
-                      instance.resource_context())) {
+                      instance.resource_context(),
+                      instance.storage_partition())) {
       DCHECK(iter->NumFilters() == 1);
       WorkerProcessHost::WorkerInstance::FilterInfo filter_info =
           iter->GetFilter();
       instance.AddFilter(filter_info.first, filter_info.second);
       iter = queued_workers_.erase(iter);
     } else {
       ++iter;
     }
   }
 
   if (!worker) {
     WorkerMessageFilter* first_filter = instance.filters().begin()->first;
-    worker = new WorkerProcessHost(instance.resource_context());
+    worker = new WorkerProcessHost(instance.resource_context(),
+                                   instance.storage_partition());
     // TODO(atwilson): This won't work if the message is from a worker process.
     // We don't support that yet though (this message is only sent from
     // renderers) but when we do, we'll need to add code to pass in the current
     // worker's document set for nested workers.
     if (!worker->Init(first_filter->render_process_id())) {
       delete worker;
       return false;
     }
   }
 
@@ skipping 206 matching lines @@
     int worker_route_id) {
   WorkerDevToolsManager::GetInstance()->WorkerDestroyed(
       process, worker_route_id);
   FOR_EACH_OBSERVER(WorkerServiceObserver, observers_,
                     WorkerDestroyed(process->GetData().id, worker_route_id));
 }
 
 WorkerProcessHost::WorkerInstance* WorkerServiceImpl::FindSharedWorkerInstance(
     const GURL& url,
     const string16& name,
-    ResourceContext* resource_context) {
+    ResourceContext* resource_context,
+    StoragePartitionImpl* storage_partition) {
   for (WorkerProcessHostIterator iter; !iter.Done(); ++iter) {
     for (WorkerProcessHost::Instances::iterator instance_iter =
              iter->mutable_instances().begin();
          instance_iter != iter->mutable_instances().end();
          ++instance_iter) {
-      if (instance_iter->Matches(url, name, resource_context))
+      if (instance_iter->Matches(url, name, resource_context,
+                                 storage_partition))
         return &(*instance_iter);
     }
   }
   return NULL;
 }
 
 WorkerProcessHost::WorkerInstance* WorkerServiceImpl::FindPendingInstance(
     const GURL& url,
     const string16& name,
-    ResourceContext* resource_context) {
+    ResourceContext* resource_context,
+    StoragePartitionImpl* storage_partition) {
   // Walk the pending instances looking for a matching pending worker.
   for (WorkerProcessHost::Instances::iterator iter =
            pending_shared_workers_.begin();
        iter != pending_shared_workers_.end();
        ++iter) {
-    if (iter->Matches(url, name, resource_context)) {
+    if (iter->Matches(url, name, resource_context, storage_partition)) {
       return &(*iter);
     }
   }
   return NULL;
 }
 
 
 void WorkerServiceImpl::RemovePendingInstances(
     const GURL& url,
     const string16& name,
-    ResourceContext* resource_context) {
+    ResourceContext* resource_context,
+    StoragePartitionImpl* storage_partition) {
   // Walk the pending instances looking for a matching pending worker.
   for (WorkerProcessHost::Instances::iterator iter =
            pending_shared_workers_.begin();
        iter != pending_shared_workers_.end(); ) {
-    if (iter->Matches(url, name, resource_context)) {
+    if (iter->Matches(url, name, resource_context, storage_partition)) {
       iter = pending_shared_workers_.erase(iter);
     } else {
       ++iter;
     }
   }
 }
 
 WorkerProcessHost::WorkerInstance* WorkerServiceImpl::CreatePendingInstance(
     const GURL& url,
     const string16& name,
-    ResourceContext* resource_context) {
+    ResourceContext* resource_context,
+    StoragePartitionImpl* storage_partition) {
   // Look for an existing pending shared worker.
   WorkerProcessHost::WorkerInstance* instance =
-      FindPendingInstance(url, name, resource_context);
+      FindPendingInstance(url, name, resource_context, storage_partition);
   if (instance)
     return instance;
 
   // No existing pending worker - create a new one.
-  WorkerProcessHost::WorkerInstance pending(url, true, name, resource_context);
+  WorkerProcessHost::WorkerInstance pending(
+      url, true, name, resource_context, storage_partition);
   pending_shared_workers_.push_back(pending);
   return &pending_shared_workers_.back();
 }
 
 }  // namespace content