| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ | 5 #ifndef CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ |
| 6 #define CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ | 6 #define CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 #include <set> | 9 #include <set> |
| 10 #include <string> | 10 #include <string> |
| 11 #include <vector> | 11 #include <vector> |
| 12 | 12 |
| 13 #include "base/callback.h" | 13 #include "base/callback.h" |
| 14 #include "base/compiler_specific.h" | 14 #include "base/compiler_specific.h" |
| 15 #include "base/memory/linked_ptr.h" | 15 #include "base/memory/linked_ptr.h" |
| 16 #include "chrome/browser/extensions/location_bar_controller.h" | 16 #include "chrome/browser/extensions/location_bar_controller.h" |
| 17 #include "content/public/browser/web_contents_observer.h" | 17 #include "content/public/browser/web_contents_observer.h" |
| 18 #include "extensions/browser/extension_registry_observer.h" |
| 18 #include "extensions/common/permissions/permissions_data.h" | 19 #include "extensions/common/permissions/permissions_data.h" |
| 19 #include "extensions/common/user_script.h" | 20 #include "extensions/common/user_script.h" |
| 20 | 21 |
| 21 namespace content { | 22 namespace content { |
| 22 class WebContents; | 23 class WebContents; |
| 23 } | 24 } |
| 24 | 25 |
| 25 namespace IPC { | 26 namespace IPC { |
| 26 class Message; | 27 class Message; |
| 27 } | 28 } |
| 28 | 29 |
| 29 class ExtensionAction; | 30 class ExtensionAction; |
| 30 | 31 |
| 31 namespace extensions { | 32 namespace extensions { |
| 32 class Extension; | 33 class Extension; |
| 34 class ExtensionRegistry; |
| 33 | 35 |
| 34 // The provider for ExtensionActions corresponding to scripts which are actively | 36 // The provider for ExtensionActions corresponding to scripts which are actively |
| 35 // running or need permission. | 37 // running or need permission. |
| 36 // TODO(rdevlin.cronin): This isn't really a controller, but it has good parity | 38 // TODO(rdevlin.cronin): This isn't really a controller, but it has good parity |
| 37 // with LocationBar"Controller". | 39 // with LocationBar"Controller". |
| 38 class ActiveScriptController : public content::WebContentsObserver { | 40 class ActiveScriptController : public content::WebContentsObserver, |
| 41 public ExtensionRegistryObserver { |
| 39 public: | 42 public: |
| 40 explicit ActiveScriptController(content::WebContents* web_contents); | 43 explicit ActiveScriptController(content::WebContents* web_contents); |
| 41 virtual ~ActiveScriptController(); | 44 virtual ~ActiveScriptController(); |
| 42 | 45 |
| 43 // Returns the ActiveScriptController for the given |web_contents|, or NULL | 46 // Returns the ActiveScriptController for the given |web_contents|, or NULL |
| 44 // if one does not exist. | 47 // if one does not exist. |
| 45 static ActiveScriptController* GetForWebContents( | 48 static ActiveScriptController* GetForWebContents( |
| 46 content::WebContents* web_contents); | 49 content::WebContents* web_contents); |
| 47 | 50 |
| 48 // Notifies the ActiveScriptController that an extension has been granted | 51 // Notifies the ActiveScriptController that an extension has been granted |
| (...skipping 12 matching lines...) Expand all Loading... |
| 61 // been clicked, running any pending tasks that were previously shelved. | 64 // been clicked, running any pending tasks that were previously shelved. |
| 62 void OnClicked(const Extension* extension); | 65 void OnClicked(const Extension* extension); |
| 63 | 66 |
| 64 // Returns true if there is an active script injection action for |extension|. | 67 // Returns true if there is an active script injection action for |extension|. |
| 65 bool HasActiveScriptAction(const Extension* extension); | 68 bool HasActiveScriptAction(const Extension* extension); |
| 66 | 69 |
| 67 // Returns the action to display for the given |extension|, or NULL if no | 70 // Returns the action to display for the given |extension|, or NULL if no |
| 68 // action should be displayed. | 71 // action should be displayed. |
| 69 ExtensionAction* GetActionForExtension(const Extension* extension); | 72 ExtensionAction* GetActionForExtension(const Extension* extension); |
| 70 | 73 |
| 71 // Notifies that the given |extension| has been unloaded; forwarded from the | |
| 72 // ExtensionRegistryObserver method. | |
| 73 void OnExtensionUnloaded(const Extension* extension); | |
| 74 | |
| 75 #if defined(UNIT_TEST) | 74 #if defined(UNIT_TEST) |
| 76 // Only used in tests. | 75 // Only used in tests. |
| 77 PermissionsData::AccessType RequiresUserConsentForScriptInjectionForTesting( | 76 PermissionsData::AccessType RequiresUserConsentForScriptInjectionForTesting( |
| 78 const Extension* extension, | 77 const Extension* extension, |
| 79 UserScript::InjectionType type) { | 78 UserScript::InjectionType type) { |
| 80 return RequiresUserConsentForScriptInjection(extension, type); | 79 return RequiresUserConsentForScriptInjection(extension, type); |
| 81 } | 80 } |
| 82 void RequestScriptInjectionForTesting(const Extension* extension, | 81 void RequestScriptInjectionForTesting(const Extension* extension, |
| 83 const base::Closure& callback) { | 82 const base::Closure& callback) { |
| 84 return RequestScriptInjection(extension, callback); | 83 return RequestScriptInjection(extension, callback); |
| (...skipping 21 matching lines...) Expand all Loading... |
| 106 | 105 |
| 107 // Handle the RequestScriptInjectionPermission message. | 106 // Handle the RequestScriptInjectionPermission message. |
| 108 void OnRequestScriptInjectionPermission( | 107 void OnRequestScriptInjectionPermission( |
| 109 const std::string& extension_id, | 108 const std::string& extension_id, |
| 110 UserScript::InjectionType script_type, | 109 UserScript::InjectionType script_type, |
| 111 int64 request_id); | 110 int64 request_id); |
| 112 | 111 |
| 113 // Grants permission for the given request to run. | 112 // Grants permission for the given request to run. |
| 114 void PermitScriptInjection(int64 request_id); | 113 void PermitScriptInjection(int64 request_id); |
| 115 | 114 |
| 115 // Log metrics. |
| 116 void LogUMA() const; |
| 117 |
| 116 // content::WebContentsObserver implementation. | 118 // content::WebContentsObserver implementation. |
| 117 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; | 119 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; |
| 118 virtual void DidNavigateMainFrame( | 120 virtual void DidNavigateMainFrame( |
| 119 const content::LoadCommittedDetails& details, | 121 const content::LoadCommittedDetails& details, |
| 120 const content::FrameNavigateParams& params) OVERRIDE; | 122 const content::FrameNavigateParams& params) OVERRIDE; |
| 121 | 123 |
| 122 // Log metrics. | 124 // ExtensionRegistryObserver: |
| 123 void LogUMA() const; | 125 virtual void OnExtensionUnloaded( |
| 126 content::BrowserContext* browser_context, |
| 127 const Extension* extension, |
| 128 UnloadedExtensionInfo::Reason reason) OVERRIDE; |
| 124 | 129 |
| 125 // Whether or not the ActiveScriptController is enabled (corresponding to the | 130 // Whether or not the ActiveScriptController is enabled (corresponding to the |
| 126 // kActiveScriptEnforcement switch). If it is not, it acts as an empty shell, | 131 // kActiveScriptEnforcement switch). If it is not, it acts as an empty shell, |
| 127 // always allowing scripts to run and never displaying actions. | 132 // always allowing scripts to run and never displaying actions. |
| 128 bool enabled_; | 133 bool enabled_; |
| 129 | 134 |
| 130 // The map of extension_id:pending_request of all pending requests. | 135 // The map of extension_id:pending_request of all pending requests. |
| 131 PendingRequestMap pending_requests_; | 136 PendingRequestMap pending_requests_; |
| 132 | 137 |
| 133 // The extensions which have been granted permission to run on the given page. | 138 // The extensions which have been granted permission to run on the given page. |
| 134 // TODO(rdevlin.cronin): Right now, this just keeps track of extensions that | 139 // TODO(rdevlin.cronin): Right now, this just keeps track of extensions that |
| 135 // have been permitted to run on the page via this interface. Instead, it | 140 // have been permitted to run on the page via this interface. Instead, it |
| 136 // should incorporate more fully with ActiveTab. | 141 // should incorporate more fully with ActiveTab. |
| 137 std::set<std::string> permitted_extensions_; | 142 std::set<std::string> permitted_extensions_; |
| 138 | 143 |
| 139 // Script badges that have been generated for extensions. This is both those | 144 // Script badges that have been generated for extensions. This is both those |
| 140 // with actions already declared that are copied and normalised, and actions | 145 // with actions already declared that are copied and normalised, and actions |
| 141 // that get generated for extensions that haven't declared anything. | 146 // that get generated for extensions that haven't declared anything. |
| 142 typedef std::map<std::string, linked_ptr<ExtensionAction> > ActiveScriptMap; | 147 typedef std::map<std::string, linked_ptr<ExtensionAction> > ActiveScriptMap; |
| 143 ActiveScriptMap active_script_actions_; | 148 ActiveScriptMap active_script_actions_; |
| 144 | 149 |
| 150 ScopedObserver<ExtensionRegistry, ExtensionRegistryObserver> |
| 151 extension_registry_observer_; |
| 152 |
| 145 DISALLOW_COPY_AND_ASSIGN(ActiveScriptController); | 153 DISALLOW_COPY_AND_ASSIGN(ActiveScriptController); |
| 146 }; | 154 }; |
| 147 | 155 |
| 148 } // namespace extensions | 156 } // namespace extensions |
| 149 | 157 |
| 150 #endif // CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ | 158 #endif // CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ |
| OLD | NEW |