PlzNavigate: ensure we don't create extra-processes for ServiceWorkers

content/browser/browsing_instance.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/browsing_instance.h"
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "content/browser/site_instance_impl.h"
 #include "content/common/site_isolation_policy.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/url_constants.h"
 
 namespace content {
 
+namespace {
+
+using BrowsingInstanceList = std::set<BrowsingInstance*>;
+base::LazyInstance<BrowsingInstanceList>::DestructorAtExit
+    g_browsing_instance_list = LAZY_INSTANCE_INITIALIZER;
+
+}  // namespace
+
+// static
+scoped_refptr<SiteInstanceImpl> BrowsingInstance::FindSiteInstance(
+    BrowserContext* browser_context,
+    const GURL& url) {
+  for (BrowsingInstance* browsing_instance : g_browsing_instance_list.Get()) {
+    if (browsing_instance->browser_context() == browser_context &&
+        browsing_instance->HasSiteInstance(url)) {
+      return browsing_instance->GetSiteInstanceForURL(url);
+    }
+  }
+  std::string site = SiteInstanceImpl::GetSiteForURL(browser_context, url)
+                         .possibly_invalid_spec();
+  for (BrowsingInstance* browsing_instance : g_browsing_instance_list.Get()) {
+    if (browsing_instance->browser_context() != browser_context)
+      continue;
+    if (browsing_instance->site_to_site_instance_map_.find(site) !=
+        browsing_instance->site_to_site_instance_map_.end()) {
+      for (SiteInstanceImpl* instance :
+           browsing_instance->site_to_site_instance_map_[site]) {
+        return instance;
+      }
+    }
+  }
+  return nullptr;
+}
+
 BrowsingInstance::BrowsingInstance(BrowserContext* browser_context)
-    : browser_context_(browser_context),
-      active_contents_count_(0u) {
+    : browser_context_(browser_context), active_contents_count_(0u) {
+  g_browsing_instance_list.Get().insert(this);
 }
 
 bool BrowsingInstance::HasSiteInstance(const GURL& url) {
   std::string site =
       SiteInstanceImpl::GetSiteForURL(browser_context_, url)
           .possibly_invalid_spec();
-
   return site_instance_map_.find(site) != site_instance_map_.end();
 }
 
 scoped_refptr<SiteInstanceImpl> BrowsingInstance::GetSiteInstanceForURL(
     const GURL& url) {
   std::string site = SiteInstanceImpl::GetSiteForURL(browser_context_, url)
                          .possibly_invalid_spec();
 
   SiteInstanceMap::iterator i = site_instance_map_.find(site);
   if (i != site_instance_map_.end())
@@ skipping 57 matching lines @@
   // comments in RegisterSiteInstance.)
   SiteInstanceMap::iterator i = site_instance_map_.find(site);
   if (i != site_instance_map_.end() && i->second == site_instance) {
     // Matches, so erase it.
     site_instance_map_.erase(i);
   }
   if (default_subframe_site_instance_ == site_instance)
     default_subframe_site_instance_ = nullptr;
 }
 
+void BrowsingInstance::RegisterUsedFor(SiteInstanceImpl* site_instance,
+                                       std::string site) {
+  auto instances = site_to_site_instance_map_.find(site);
+  if (instances == site_to_site_instance_map_.end())
+    site_to_site_instance_map_[site] = std::set<SiteInstanceImpl*>();
+  site_to_site_instance_map_[site].insert(site_instance);
+}
+
+void BrowsingInstance::UnregisterUsedFor(SiteInstanceImpl* site_instance,
+                                         const std::set<std::string>& sites) {
+  for (const auto& site : sites) {
+    if (site_to_site_instance_map_.find(site) !=
+        site_to_site_instance_map_.end()) {
+      site_to_site_instance_map_[site].erase(site_instance);
+      if (site_to_site_instance_map_[site].empty())
+        site_to_site_instance_map_.erase(site);
+    }
+  }
+}
+
 BrowsingInstance::~BrowsingInstance() {
+  g_browsing_instance_list.Get().erase(this);
   // We should only be deleted when all of the SiteInstances that refer to
   // us are gone.
   DCHECK(site_instance_map_.empty());
   DCHECK_EQ(0u, active_contents_count_);
 }
 
 }  // namespace content