// 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. #ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_PROCESS_MANAGER_H_ #define CHROME_BROWSER_EXTENSIONS_EXTENSION_PROCESS_MANAGER_H_ #pragma once #include <map> #include <set> #include <string> #include "base/memory/ref_counted.h" #include "chrome/common/view_types.h" #include "content/common/notification_observer.h" #include "content/common/notification_registrar.h" class Browser; class BrowsingInstance; class Extension; class ExtensionHost; class GURL; class Profile; class RenderProcessHost; class SiteInstance; // Manages dynamic state of running Chromium extensions. There is one instance // of this class per Profile. OTR Profiles have a separate instance that keeps // track of split-mode extensions only. class ExtensionProcessManager : public NotificationObserver { public: static ExtensionProcessManager* Create(Profile* profile); virtual ~ExtensionProcessManager(); // Creates a new ExtensionHost with its associated view, grouping it in the // appropriate SiteInstance (and therefore process) based on the URL and // profile. virtual ExtensionHost* CreateView(const Extension* extension, const GURL& url, Browser* browser, ViewType::Type view_type); ExtensionHost* CreateView(const GURL& url, Browser* browser, ViewType::Type view_type); ExtensionHost* CreatePopup(const Extension* extension, const GURL& url, Browser* browser); ExtensionHost* CreatePopup(const GURL& url, Browser* browser); ExtensionHost* CreateInfobar(const Extension* extension, const GURL& url, Browser* browser); ExtensionHost* CreateInfobar(const GURL& url, Browser* browser); // Open the extension's options page. void OpenOptionsPage(const Extension* extension, Browser* browser); // Creates a new UI-less extension instance. Like CreateView, but not // displayed anywhere. virtual void CreateBackgroundHost(const Extension* extension, const GURL& url); // Gets the ExtensionHost for the background page for an extension, or NULL if // the extension isn't running or doesn't have a background page. ExtensionHost* GetBackgroundHostForExtension(const Extension* extension); // Returns the SiteInstance that the given URL belongs to. virtual SiteInstance* GetSiteInstanceForURL(const GURL& url); // Registers an extension process by |extension_id| and specifying which // |process_id| it belongs to. void RegisterExtensionProcess(const std::string& extension_id, int process_id); // Unregisters an extension process with specified |process_id|. void UnregisterExtensionProcess(int process_id); // Returns the extension process that |url| is associated with if it exists. virtual RenderProcessHost* GetExtensionProcess(const GURL& url); // Returns the process that the extension with the given ID is running in. RenderProcessHost* GetExtensionProcess(const std::string& extension_id); // Returns true if |host| is managed by this process manager. bool HasExtensionHost(ExtensionHost* host) const; typedef std::set<ExtensionHost*> ExtensionHostSet; typedef ExtensionHostSet::const_iterator const_iterator; const_iterator begin() const { return all_hosts_.begin(); } const_iterator end() const { return all_hosts_.end(); } protected: explicit ExtensionProcessManager(Profile* profile); // Called just after |host| is created so it can be registered in our lists. void OnExtensionHostCreated(ExtensionHost* host, bool is_background); // Called on browser shutdown to close our extension hosts. void CloseBackgroundHosts(); // NotificationObserver: virtual void Observe(NotificationType type, const NotificationSource& source, const NotificationDetails& details); NotificationRegistrar registrar_; // The set of all ExtensionHosts managed by this process manager. ExtensionHostSet all_hosts_; // The set of running viewless background extensions. ExtensionHostSet background_hosts_; // The BrowsingInstance shared by all extensions in this profile. This // controls process grouping. scoped_refptr<BrowsingInstance> browsing_instance_; // A map of extension ID to the render_process_id that the extension lives in. typedef std::map<std::string, int> ProcessIDMap; ProcessIDMap process_ids_; DISALLOW_COPY_AND_ASSIGN(ExtensionProcessManager); }; #endif // CHROME_BROWSER_EXTENSIONS_EXTENSION_PROCESS_MANAGER_H_