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> |
(...skipping 16 matching lines...) Expand all Loading... |
27 } | 27 } |
28 | 28 |
29 class ExtensionAction; | 29 class ExtensionAction; |
30 | 30 |
31 namespace extensions { | 31 namespace extensions { |
32 class Extension; | 32 class Extension; |
33 | 33 |
34 // The provider for ExtensionActions corresponding to scripts which are actively | 34 // The provider for ExtensionActions corresponding to scripts which are actively |
35 // running or need permission. | 35 // running or need permission. |
36 // TODO(rdevlin.cronin): This isn't really a controller, but it has good parity | 36 // TODO(rdevlin.cronin): This isn't really a controller, but it has good parity |
37 // with PageAction"Controller". | 37 // with LocationBar"Controller". |
38 class ActiveScriptController : public LocationBarController::ActionProvider, | 38 class ActiveScriptController : public content::WebContentsObserver { |
39 public content::WebContentsObserver { | |
40 public: | 39 public: |
41 explicit ActiveScriptController(content::WebContents* web_contents); | 40 explicit ActiveScriptController(content::WebContents* web_contents); |
42 virtual ~ActiveScriptController(); | 41 virtual ~ActiveScriptController(); |
43 | 42 |
44 // Returns the ActiveScriptController for the given |web_contents|, or NULL | 43 // Returns the ActiveScriptController for the given |web_contents|, or NULL |
45 // if one does not exist. | 44 // if one does not exist. |
46 static ActiveScriptController* GetForWebContents( | 45 static ActiveScriptController* GetForWebContents( |
47 content::WebContents* web_contents); | 46 content::WebContents* web_contents); |
48 | 47 |
49 // Notifies the ActiveScriptController that an extension has been granted | 48 // Notifies the ActiveScriptController that an extension has been granted |
50 // active tab permissions. This will run any pending injections for that | 49 // active tab permissions. This will run any pending injections for that |
51 // extension. | 50 // extension. |
52 void OnActiveTabPermissionGranted(const Extension* extension); | 51 void OnActiveTabPermissionGranted(const Extension* extension); |
53 | 52 |
54 // Notifies the ActiveScriptController of detected ad injection. | 53 // Notifies the ActiveScriptController of detected ad injection. |
55 void OnAdInjectionDetected(const std::set<std::string>& ad_injectors); | 54 void OnAdInjectionDetected(const std::set<std::string>& ad_injectors); |
56 | 55 |
57 // Adds the visible origin to |extension|'s active permissions, granting | 56 // Adds the visible origin to |extension|'s active permissions, granting |
58 // |extension| permission to always run script injections on the origin. | 57 // |extension| permission to always run script injections on the origin. |
59 void AlwaysRunOnVisibleOrigin(const Extension* extension); | 58 void AlwaysRunOnVisibleOrigin(const Extension* extension); |
60 | 59 |
61 // Notifies the ActiveScriptController that the action for |extension| has | 60 // Notifies the ActiveScriptController that the action for |extension| has |
62 // been clicked, running any pending tasks that were previously shelved. | 61 // been clicked, running any pending tasks that were previously shelved. |
63 void OnClicked(const Extension* extension); | 62 void OnClicked(const Extension* extension); |
64 | 63 |
65 // Returns true if there is an active script injection action for |extension|. | 64 // Returns true if there is an active script injection action for |extension|. |
66 bool HasActiveScriptAction(const Extension* extension); | 65 bool HasActiveScriptAction(const Extension* extension); |
67 | 66 |
68 // LocationBarControllerProvider implementation. | 67 // Returns the action to display for the given |extension|, or NULL if no |
69 virtual ExtensionAction* GetActionForExtension( | 68 // action should be displayed. |
70 const Extension* extension) OVERRIDE; | 69 ExtensionAction* GetActionForExtension(const Extension* extension); |
71 virtual void OnNavigated() OVERRIDE; | 70 |
72 virtual void OnExtensionUnloaded(const Extension* extension) OVERRIDE; | 71 // Notifies that the given |extension| has been unloaded; forwarded from the |
| 72 // ExtensionRegistryObserver method. |
| 73 void OnExtensionUnloaded(const Extension* extension); |
73 | 74 |
74 #if defined(UNIT_TEST) | 75 #if defined(UNIT_TEST) |
75 // Only used in tests. | 76 // Only used in tests. |
76 PermissionsData::AccessType RequiresUserConsentForScriptInjectionForTesting( | 77 PermissionsData::AccessType RequiresUserConsentForScriptInjectionForTesting( |
77 const Extension* extension, | 78 const Extension* extension, |
78 UserScript::InjectionType type) { | 79 UserScript::InjectionType type) { |
79 return RequiresUserConsentForScriptInjection(extension, type); | 80 return RequiresUserConsentForScriptInjection(extension, type); |
80 } | 81 } |
81 void RequestScriptInjectionForTesting(const Extension* extension, | 82 void RequestScriptInjectionForTesting(const Extension* extension, |
82 const base::Closure& callback) { | 83 const base::Closure& callback) { |
(...skipping 24 matching lines...) Expand all Loading... |
107 void OnRequestScriptInjectionPermission( | 108 void OnRequestScriptInjectionPermission( |
108 const std::string& extension_id, | 109 const std::string& extension_id, |
109 UserScript::InjectionType script_type, | 110 UserScript::InjectionType script_type, |
110 int64 request_id); | 111 int64 request_id); |
111 | 112 |
112 // Grants permission for the given request to run. | 113 // Grants permission for the given request to run. |
113 void PermitScriptInjection(int64 request_id); | 114 void PermitScriptInjection(int64 request_id); |
114 | 115 |
115 // content::WebContentsObserver implementation. | 116 // content::WebContentsObserver implementation. |
116 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; | 117 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; |
| 118 virtual void DidNavigateMainFrame( |
| 119 const content::LoadCommittedDetails& details, |
| 120 const content::FrameNavigateParams& params) OVERRIDE; |
117 | 121 |
118 // Log metrics. | 122 // Log metrics. |
119 void LogUMA() const; | 123 void LogUMA() const; |
120 | 124 |
121 // Whether or not the ActiveScriptController is enabled (corresponding to the | 125 // Whether or not the ActiveScriptController is enabled (corresponding to the |
122 // kActiveScriptEnforcement switch). If it is not, it acts as an empty shell, | 126 // kActiveScriptEnforcement switch). If it is not, it acts as an empty shell, |
123 // always allowing scripts to run and never displaying actions. | 127 // always allowing scripts to run and never displaying actions. |
124 bool enabled_; | 128 bool enabled_; |
125 | 129 |
126 // The map of extension_id:pending_request of all pending requests. | 130 // The map of extension_id:pending_request of all pending requests. |
(...skipping 10 matching lines...) Expand all Loading... |
137 // that get generated for extensions that haven't declared anything. | 141 // that get generated for extensions that haven't declared anything. |
138 typedef std::map<std::string, linked_ptr<ExtensionAction> > ActiveScriptMap; | 142 typedef std::map<std::string, linked_ptr<ExtensionAction> > ActiveScriptMap; |
139 ActiveScriptMap active_script_actions_; | 143 ActiveScriptMap active_script_actions_; |
140 | 144 |
141 DISALLOW_COPY_AND_ASSIGN(ActiveScriptController); | 145 DISALLOW_COPY_AND_ASSIGN(ActiveScriptController); |
142 }; | 146 }; |
143 | 147 |
144 } // namespace extensions | 148 } // namespace extensions |
145 | 149 |
146 #endif // CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ | 150 #endif // CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_ |
OLD | NEW |