Track all extension processes in ExtensionInfoMap

chrome/browser/extensions/extension_info_map.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 "chrome/browser/extensions/extension_info_map.h"
 
 #include "chrome/common/extensions/extension.h"
 #include "content/browser/browser_thread.h"
 
 namespace {
 
-static void CheckOnValidThread() {
+void CheckOnValidThread() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
 }
 
 }  // namespace
 
 
 struct ExtensionInfoMap::ExtraData {
   // When the extension was installed.
   base::Time install_time;
 
@@ skipping 66 matching lines @@
     return iter->second.incognito_enabled;
   return false;
 }
 
 bool ExtensionInfoMap::CanCrossIncognito(const Extension* extension) {
   // This is duplicated from ExtensionService :(.
   return IsIncognitoEnabled(extension->id()) &&
       !extension->incognito_split_mode();
 }
 
-// These are duplicated from ExtensionProcessManager so that we can have the
-// information on the IO thread :(.
-void ExtensionInfoMap::BindingsEnabledForProcess(int render_process_id) {
-  extension_bindings_process_ids_.insert(render_process_id);
+void ExtensionInfoMap::RegisterExtensionProcess(const std::string& extension_id,
+                                                int process_id) {
+  DCHECK(!IsExtensionInProcess(extension_id, process_id));
+  extension_process_ids_.insert(
+      ExtensionProcessIDMap::value_type(extension_id, process_id));
 }
 
-void ExtensionInfoMap::BindingsDisabledForProcess(int render_process_id) {
-  extension_bindings_process_ids_.erase(render_process_id);
+void ExtensionInfoMap::UnregisterExtensionProcess(
+    const std::string& extension_id,
+    int process_id) {
+  ExtensionProcessIDMap::iterator iter =
+      std::find(extension_process_ids_.begin(),
+                extension_process_ids_.end(),
+                ExtensionProcessIDMap::value_type(extension_id, process_id));
+  if (iter != extension_process_ids_.end())
+    extension_process_ids_.erase(iter);
 }
 
-bool ExtensionInfoMap::AreBindingsEnabledForProcess(
-    int render_process_id) const {
-  // Must behave logically the same as AreBindingsEnabledForProcess() in
-  // extension_process_manager.cc.
-  return extension_bindings_process_ids_.find(render_process_id) !=
-      extension_bindings_process_ids_.end();
+bool ExtensionInfoMap::IsExtensionInProcess(
+    const std::string& extension_id, int process_id) const {
+  return std::find(
+      extension_process_ids_.begin(),
+      extension_process_ids_.end(),
+      ExtensionProcessIDMap::value_type(extension_id, process_id)) !=
+          extension_process_ids_.end();
 }