Delegate decision what site instances can be rendered in what process to chrome

chrome/browser/extensions/extension_process_manager.cc

 // Copyright (c) 2011 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 "base/command_line.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 
 #include "chrome/browser/ui/browser_window.h"
 #include "content/browser/browsing_instance.h"
 #if defined(OS_MACOSX)
@@ skipping 86 matching lines @@
   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED,
                  Source<Profile>(original_profile));
   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED,
                  Source<Profile>(profile));
   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_HOST_VIEW_SHOULD_CLOSE,
                  Source<Profile>(profile));
   // We can listen to everything for SITE_INSTANCE_DELETED because we check the
   // |site_instance_id| in UnregisterExtensionSiteInstance.
   registrar_.Add(this, content::NOTIFICATION_SITE_INSTANCE_DELETED,
                  NotificationService::AllBrowserContextsAndSources());
+  // Same for NOTIFICATION_RENDERER_PROCESS_CLOSED.
+  registrar_.Add(this, content::NOTIFICATION_RENDERER_PROCESS_CLOSED,
+                 NotificationService::AllBrowserContextsAndSources());
   registrar_.Add(this, content::NOTIFICATION_APP_TERMINATING,
                  NotificationService::AllSources());
 }
 
 ExtensionProcessManager::~ExtensionProcessManager() {
   CloseBackgroundHosts();
   DCHECK(background_hosts_.empty());
 }
 
 ExtensionHost* ExtensionProcessManager::CreateViewHost(
@@ skipping 122 matching lines @@
   extension_ids_[site_instance_id] = extension_id;
 }
 
 void ExtensionProcessManager::UnregisterExtensionSiteInstance(
     int site_instance_id) {
   SiteInstanceIDMap::iterator it = extension_ids_.find(site_instance_id);
   if (it != extension_ids_.end())
     extension_ids_.erase(it++);
 }
 
+void ExtensionProcessManager::RegisterProcessHostWithBindings(int host_id) {
+  process_ids_.insert(host_id);
+}
+
+void ExtensionProcessManager::UnregisterProcessHostWithBindings(int host_id) {
+  std::set<int>::iterator it = process_ids_.find(host_id);

[jam, 2011/09/26 15:46:16]

nit: you can also just write process_ids_.erase(host_id);

+  if (it != process_ids_.end())
+    process_ids_.erase(it);
+}
+
+bool ExtensionProcessManager::IsProcessHostWithBindings(int host_id) const {
+  return process_ids_.find(host_id) != process_ids_.end();
+}
+
 RenderProcessHost* ExtensionProcessManager::GetExtensionProcess(
     const GURL& url) {
   if (!browsing_instance_->HasSiteInstance(url))
     return NULL;
   scoped_refptr<SiteInstance> site(
       browsing_instance_->GetSiteInstanceForURL(url));
   if (site->HasProcess())
     return site->GetProcess();
   return NULL;
 }
@@ skipping 69 matching lines @@
       background_hosts_.erase(host);
       break;
     }
 
     case content::NOTIFICATION_SITE_INSTANCE_DELETED: {
       SiteInstance* site_instance = Source<SiteInstance>(source).ptr();
       UnregisterExtensionSiteInstance(site_instance->id());
       break;
     }
 
+    case content::NOTIFICATION_RENDERER_PROCESS_CLOSED: {
+      RenderProcessHost* process_host =
+          Source<RenderProcessHost>(source).ptr();
+      UnregisterProcessHostWithBindings(process_host->id());
+      break;
+    }
+
     case chrome::NOTIFICATION_EXTENSION_HOST_VIEW_SHOULD_CLOSE: {
       ExtensionHost* host = Details<ExtensionHost>(details).ptr();
       if (host->extension_host_type() == ViewType::EXTENSION_BACKGROUND_PAGE) {
         delete host;
         // |host| should deregister itself from our structures.
         CHECK(background_hosts_.find(host) == background_hosts_.end());
       }
       break;
     }
 
@@ skipping 151 matching lines @@
         if (service && service->is_ready())
           CreateBackgroundHostsForProfileStartup(this, service->extensions());
       }
       break;
     }
     default:
       ExtensionProcessManager::Observe(type, source, details);
       break;
   }
 }