Extensions: Keep track of loaded extensions in RendererStartupHelper.

extensions/browser/renderer_startup_helper.h

 // 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.
 
 #ifndef EXTENSIONS_BROWSER_RENDERER_STARTUP_HELPER_H_
 #define EXTENSIONS_BROWSER_RENDERER_STARTUP_HELPER_H_
 
 #include <map>
 #include <set>
 
@@ skipping 29 matching lines @@
   // This class sends messages to all renderers started for |browser_context|.
   explicit RendererStartupHelper(content::BrowserContext* browser_context);
   ~RendererStartupHelper() override;
 
   // content::NotificationObserver overrides:
   void Observe(int type,
                const content::NotificationSource& source,
                const content::NotificationDetails& details) override;
 
   // Sends a message to the specified |process| activating the given extension
-  // once the process is initialized.
+  // once the process is initialized. OnExtensionLoaded should have already been
+  // called for the extension.
   void ActivateExtensionInProcess(const Extension& extension,
                                   content::RenderProcessHost* process);
 
   // Sends a message to all initialized processes to [un]load the given
   // extension. We have explicit calls for these (rather than using an
   // ExtensionRegistryObserver) because this needs to happen before other
   // initialization which might rely on the renderers being notified.
   void OnExtensionUnloaded(const Extension& extension);
   void OnExtensionLoaded(const Extension& extension);
 
  private:
+  friend class RendererStartupHelperTest;
+
   // Initializes the specified process, informing it of system state and loaded
   // extensions.
   void InitializeProcess(content::RenderProcessHost* process);
 
   // Untracks the given process.
   void UntrackProcess(content::RenderProcessHost* process);
 
   content::BrowserContext* browser_context_;  // Not owned.
 
+  // Tracks the set of loaded extensions and the processes they are loaded in.
+  std::map<ExtensionId, std::set<content::RenderProcessHost*>>
+      extension_process_map_;
+
   // The set of render processes that have had the initial batch of IPC messages
   // sent, including the set of loaded extensions. Further messages that
   // activate, load, or unload extensions should not be sent until after this
   // happens.
   std::set<content::RenderProcessHost*> initialized_processes_;
 
   // The set of ids for extensions that are active in a process that has not
   // been initialized. The activation message will be sent the process is
   // initialized.
   std::map<content::RenderProcessHost*, std::set<ExtensionId>>
@@ skipping 25 matching lines @@
   content::BrowserContext* GetBrowserContextToUse(
       content::BrowserContext* context) const override;
   bool ServiceIsCreatedWithBrowserContext() const override;
 
   DISALLOW_COPY_AND_ASSIGN(RendererStartupHelperFactory);
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_RENDERER_STARTUP_HELPER_H_