[email protected] | f34e7963 | 2010-03-17 02:34:08 | [diff] [blame^] | 1 | // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "chrome/browser/extensions/extension_function_dispatcher.h" |
| 6 | |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 7 | #include "base/process_util.h" |
| 8 | #include "base/singleton.h" |
| 9 | #include "base/values.h" |
[email protected] | a95631cb | 2009-12-10 01:59:11 | [diff] [blame] | 10 | #include "chrome/browser/browser.h" |
| 11 | #include "chrome/browser/browser_window.h" |
[email protected] | 912256b3 | 2009-09-18 09:47:35 | [diff] [blame] | 12 | #include "chrome/browser/extensions/execute_code_in_tab_function.h" |
[email protected] | 5cbe1e2 | 2010-01-30 01:18:56 | [diff] [blame] | 13 | #include "chrome/browser/extensions/extension_accessibility_api.h" |
[email protected] | 9dd97bc | 2010-01-14 01:40:04 | [diff] [blame] | 14 | #include "chrome/browser/extensions/extension_bookmark_manager_api.h" |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 15 | #include "chrome/browser/extensions/extension_bookmarks_module.h" |
[email protected] | f9391485 | 2009-05-26 06:05:40 | [diff] [blame] | 16 | #include "chrome/browser/extensions/extension_bookmarks_module_constants.h" |
[email protected] | ec9ac0df | 2009-10-01 18:06:47 | [diff] [blame] | 17 | #include "chrome/browser/extensions/extension_browser_actions_api.h" |
[email protected] | b2725756 | 2009-11-16 23:28:26 | [diff] [blame] | 18 | #include "chrome/browser/extensions/extension_dom_ui.h" |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 19 | #include "chrome/browser/extensions/extension_function.h" |
[email protected] | de768a83 | 2009-10-30 05:25:01 | [diff] [blame] | 20 | #include "chrome/browser/extensions/extension_history_api.h" |
[email protected] | f520541 | 2010-03-16 00:19:34 | [diff] [blame] | 21 | #include "chrome/browser/extensions/extension_idle_api.h" |
[email protected] | 198bcfe | 2009-09-09 22:56:28 | [diff] [blame] | 22 | #include "chrome/browser/extensions/extension_i18n_api.h" |
[email protected] | f34e7963 | 2010-03-17 02:34:08 | [diff] [blame^] | 23 | #include "chrome/browser/extensions/extension_infobar_module.h" |
[email protected] | e916901c | 2009-05-07 00:14:31 | [diff] [blame] | 24 | #include "chrome/browser/extensions/extension_message_service.h" |
[email protected] | 438772df | 2010-02-26 18:08:43 | [diff] [blame] | 25 | #include "chrome/browser/extensions/extension_metrics_module.h" |
[email protected] | f7f3a5f | 2009-05-01 22:02:34 | [diff] [blame] | 26 | #include "chrome/browser/extensions/extension_page_actions_module.h" |
[email protected] | f9391485 | 2009-05-26 06:05:40 | [diff] [blame] | 27 | #include "chrome/browser/extensions/extension_page_actions_module_constants.h" |
[email protected] | 1c1c77a5 | 2009-11-03 00:37:31 | [diff] [blame] | 28 | #include "chrome/browser/extensions/extension_popup_api.h" |
[email protected] | 4577622 | 2009-07-15 20:21:58 | [diff] [blame] | 29 | #include "chrome/browser/extensions/extension_process_manager.h" |
[email protected] | 381162b | 2010-01-28 17:29:35 | [diff] [blame] | 30 | #include "chrome/browser/extensions/extension_processes_api.h" |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 31 | #include "chrome/browser/extensions/extension_tabs_module.h" |
[email protected] | f9391485 | 2009-05-26 06:05:40 | [diff] [blame] | 32 | #include "chrome/browser/extensions/extension_tabs_module_constants.h" |
[email protected] | 25fd1b2e | 2009-08-17 20:57:14 | [diff] [blame] | 33 | #include "chrome/browser/extensions/extension_test_api.h" |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 34 | #include "chrome/browser/extensions/extension_toolstrip_api.h" |
[email protected] | d13950e | 2009-12-04 01:43:02 | [diff] [blame] | 35 | #include "chrome/browser/extensions/extensions_quota_service.h" |
[email protected] | b1748b1d8 | 2009-11-30 20:32:56 | [diff] [blame] | 36 | #include "chrome/browser/extensions/extensions_service.h" |
[email protected] | e916901c | 2009-05-07 00:14:31 | [diff] [blame] | 37 | #include "chrome/browser/profile.h" |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 38 | #include "chrome/browser/renderer_host/render_process_host.h" |
| 39 | #include "chrome/browser/renderer_host/render_view_host.h" |
[email protected] | 3550635 | 2009-08-07 18:58:19 | [diff] [blame] | 40 | #include "chrome/common/render_messages.h" |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 41 | #include "chrome/common/result_codes.h" |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 42 | #include "chrome/common/url_constants.h" |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 43 | |
| 44 | // FactoryRegistry ------------------------------------------------------------- |
| 45 | |
| 46 | namespace { |
| 47 | |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 48 | // Template for defining ExtensionFunctionFactory. |
| 49 | template<class T> |
| 50 | ExtensionFunction* NewExtensionFunction() { |
| 51 | return new T(); |
| 52 | } |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 53 | |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 54 | // Contains a list of all known extension functions and allows clients to |
| 55 | // create instances of them. |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 56 | class FactoryRegistry { |
| 57 | public: |
| 58 | static FactoryRegistry* instance(); |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 59 | FactoryRegistry() { ResetFunctions(); } |
| 60 | |
| 61 | // Resets all functions to their default values. |
| 62 | void ResetFunctions(); |
| 63 | |
| 64 | // Adds all function names to 'names'. |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 65 | void GetAllNames(std::vector<std::string>* names); |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 66 | |
| 67 | // Allows overriding of specific functions (e.g. for testing). Functions |
| 68 | // must be previously registered. Returns true if successful. |
| 69 | bool OverrideFunction(const std::string& name, |
| 70 | ExtensionFunctionFactory factory); |
| 71 | |
| 72 | // Factory method for the ExtensionFunction registered as 'name'. |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 73 | ExtensionFunction* NewFunction(const std::string& name); |
| 74 | |
| 75 | private: |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 76 | template<class T> |
| 77 | void RegisterFunction() { |
| 78 | factories_[T::function_name()] = &NewExtensionFunction<T>; |
| 79 | } |
| 80 | |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 81 | typedef std::map<std::string, ExtensionFunctionFactory> FactoryMap; |
| 82 | FactoryMap factories_; |
| 83 | }; |
| 84 | |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 85 | FactoryRegistry* FactoryRegistry::instance() { |
| 86 | return Singleton<FactoryRegistry>::get(); |
| 87 | } |
| 88 | |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 89 | void FactoryRegistry::ResetFunctions() { |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 90 | // Register all functions here. |
| 91 | |
[email protected] | e515f5d | 2009-05-05 03:05:00 | [diff] [blame] | 92 | // Windows |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 93 | RegisterFunction<GetWindowFunction>(); |
| 94 | RegisterFunction<GetCurrentWindowFunction>(); |
| 95 | RegisterFunction<GetLastFocusedWindowFunction>(); |
| 96 | RegisterFunction<GetAllWindowsFunction>(); |
| 97 | RegisterFunction<CreateWindowFunction>(); |
| 98 | RegisterFunction<UpdateWindowFunction>(); |
| 99 | RegisterFunction<RemoveWindowFunction>(); |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 100 | |
[email protected] | e515f5d | 2009-05-05 03:05:00 | [diff] [blame] | 101 | // Tabs |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 102 | RegisterFunction<GetTabFunction>(); |
| 103 | RegisterFunction<GetSelectedTabFunction>(); |
| 104 | RegisterFunction<GetAllTabsInWindowFunction>(); |
| 105 | RegisterFunction<CreateTabFunction>(); |
| 106 | RegisterFunction<UpdateTabFunction>(); |
| 107 | RegisterFunction<MoveTabFunction>(); |
| 108 | RegisterFunction<RemoveTabFunction>(); |
| 109 | RegisterFunction<DetectTabLanguageFunction>(); |
| 110 | RegisterFunction<CaptureVisibleTabFunction>(); |
| 111 | RegisterFunction<TabsExecuteScriptFunction>(); |
| 112 | RegisterFunction<TabsInsertCSSFunction>(); |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 113 | |
[email protected] | f7f3a5f | 2009-05-01 22:02:34 | [diff] [blame] | 114 | // Page Actions. |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 115 | RegisterFunction<EnablePageActionFunction>(); |
| 116 | RegisterFunction<DisablePageActionFunction>(); |
[email protected] | 744ef17 | 2009-10-16 21:53:46 | [diff] [blame] | 117 | RegisterFunction<PageActionShowFunction>(); |
| 118 | RegisterFunction<PageActionHideFunction>(); |
| 119 | RegisterFunction<PageActionSetIconFunction>(); |
| 120 | RegisterFunction<PageActionSetTitleFunction>(); |
[email protected] | e478d670 | 2010-01-28 00:10:29 | [diff] [blame] | 121 | RegisterFunction<PageActionSetPopupFunction>(); |
[email protected] | f7f3a5f | 2009-05-01 22:02:34 | [diff] [blame] | 122 | |
[email protected] | ec9ac0df | 2009-10-01 18:06:47 | [diff] [blame] | 123 | // Browser Actions. |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 124 | RegisterFunction<BrowserActionSetIconFunction>(); |
[email protected] | 1288ba0 | 2009-10-15 00:02:24 | [diff] [blame] | 125 | RegisterFunction<BrowserActionSetTitleFunction>(); |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 126 | RegisterFunction<BrowserActionSetBadgeTextFunction>(); |
| 127 | RegisterFunction<BrowserActionSetBadgeBackgroundColorFunction>(); |
[email protected] | 85ae959 | 2010-02-03 20:58:50 | [diff] [blame] | 128 | RegisterFunction<BrowserActionSetPopupFunction>(); |
[email protected] | ec9ac0df | 2009-10-01 18:06:47 | [diff] [blame] | 129 | |
[email protected] | f7f3a5f | 2009-05-01 22:02:34 | [diff] [blame] | 130 | // Bookmarks. |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 131 | RegisterFunction<GetBookmarksFunction>(); |
| 132 | RegisterFunction<GetBookmarkChildrenFunction>(); |
[email protected] | a3c94c71 | 2009-12-18 19:23:55 | [diff] [blame] | 133 | RegisterFunction<GetBookmarkRecentFunction>(); |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 134 | RegisterFunction<GetBookmarkTreeFunction>(); |
| 135 | RegisterFunction<SearchBookmarksFunction>(); |
| 136 | RegisterFunction<RemoveBookmarkFunction>(); |
| 137 | RegisterFunction<RemoveTreeBookmarkFunction>(); |
| 138 | RegisterFunction<CreateBookmarkFunction>(); |
| 139 | RegisterFunction<MoveBookmarkFunction>(); |
| 140 | RegisterFunction<UpdateBookmarkFunction>(); |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 141 | |
[email protected] | f34e7963 | 2010-03-17 02:34:08 | [diff] [blame^] | 142 | // Infobars. |
| 143 | RegisterFunction<ShowInfoBarFunction>(); |
| 144 | |
[email protected] | 9dd97bc | 2010-01-14 01:40:04 | [diff] [blame] | 145 | // BookmarkManager |
| 146 | RegisterFunction<CopyBookmarkManagerFunction>(); |
| 147 | RegisterFunction<CutBookmarkManagerFunction>(); |
| 148 | RegisterFunction<PasteBookmarkManagerFunction>(); |
[email protected] | 03b3bbf | 2010-01-29 23:54:57 | [diff] [blame] | 149 | RegisterFunction<CanPasteBookmarkManagerFunction>(); |
[email protected] | cb6cf79 | 2010-01-28 00:04:56 | [diff] [blame] | 150 | RegisterFunction<ImportBookmarksFunction>(); |
| 151 | RegisterFunction<ExportBookmarksFunction>(); |
[email protected] | d406e2e | 2010-01-30 21:45:18 | [diff] [blame] | 152 | RegisterFunction<SortChildrenBookmarkManagerFunction>(); |
[email protected] | 9dd97bc | 2010-01-14 01:40:04 | [diff] [blame] | 153 | RegisterFunction<BookmarkManagerGetStringsFunction>(); |
[email protected] | ced90ae1 | 2010-02-20 02:06:16 | [diff] [blame] | 154 | RegisterFunction<StartDragBookmarkManagerFunction>(); |
| 155 | RegisterFunction<DropBookmarkManagerFunction>(); |
[email protected] | 9dd97bc | 2010-01-14 01:40:04 | [diff] [blame] | 156 | |
[email protected] | de768a83 | 2009-10-30 05:25:01 | [diff] [blame] | 157 | // History |
| 158 | RegisterFunction<AddUrlHistoryFunction>(); |
| 159 | RegisterFunction<DeleteAllHistoryFunction>(); |
| 160 | RegisterFunction<DeleteRangeHistoryFunction>(); |
| 161 | RegisterFunction<DeleteUrlHistoryFunction>(); |
| 162 | RegisterFunction<GetVisitsHistoryFunction>(); |
| 163 | RegisterFunction<SearchHistoryFunction>(); |
| 164 | |
[email protected] | f520541 | 2010-03-16 00:19:34 | [diff] [blame] | 165 | // Idle |
| 166 | RegisterFunction<ExtensionIdleQueryStateFunction>(); |
| 167 | |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 168 | // Toolstrips. |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 169 | RegisterFunction<ToolstripExpandFunction>(); |
| 170 | RegisterFunction<ToolstripCollapseFunction>(); |
[email protected] | 25fd1b2e | 2009-08-17 20:57:14 | [diff] [blame] | 171 | |
[email protected] | 198bcfe | 2009-09-09 22:56:28 | [diff] [blame] | 172 | // I18N. |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 173 | RegisterFunction<GetAcceptLanguagesFunction>(); |
[email protected] | 198bcfe | 2009-09-09 22:56:28 | [diff] [blame] | 174 | |
[email protected] | 1c1c77a5 | 2009-11-03 00:37:31 | [diff] [blame] | 175 | // Popup API. |
| 176 | RegisterFunction<PopupShowFunction>(); |
| 177 | |
[email protected] | 381162b | 2010-01-28 17:29:35 | [diff] [blame] | 178 | // Processes. |
| 179 | RegisterFunction<GetProcessForTabFunction>(); |
| 180 | |
[email protected] | 438772df | 2010-02-26 18:08:43 | [diff] [blame] | 181 | // Metrics. |
[email protected] | cf25e4d | 2010-03-12 21:19:34 | [diff] [blame] | 182 | RegisterFunction<MetricsRecordUserActionFunction>(); |
| 183 | RegisterFunction<MetricsRecordValueFunction>(); |
| 184 | RegisterFunction<MetricsRecordPercentageFunction>(); |
| 185 | RegisterFunction<MetricsRecordCountFunction>(); |
| 186 | RegisterFunction<MetricsRecordSmallCountFunction>(); |
| 187 | RegisterFunction<MetricsRecordMediumCountFunction>(); |
| 188 | RegisterFunction<MetricsRecordTimeFunction>(); |
| 189 | RegisterFunction<MetricsRecordMediumTimeFunction>(); |
| 190 | RegisterFunction<MetricsRecordLongTimeFunction>(); |
[email protected] | 438772df | 2010-02-26 18:08:43 | [diff] [blame] | 191 | |
[email protected] | 25fd1b2e | 2009-08-17 20:57:14 | [diff] [blame] | 192 | // Test. |
[email protected] | 61424c06 | 2009-10-14 23:14:59 | [diff] [blame] | 193 | RegisterFunction<ExtensionTestPassFunction>(); |
| 194 | RegisterFunction<ExtensionTestFailFunction>(); |
| 195 | RegisterFunction<ExtensionTestLogFunction>(); |
[email protected] | d13950e | 2009-12-04 01:43:02 | [diff] [blame] | 196 | RegisterFunction<ExtensionTestQuotaResetFunction>(); |
[email protected] | db7331a | 2010-02-25 22:10:50 | [diff] [blame] | 197 | RegisterFunction<ExtensionTestCreateIncognitoTabFunction>(); |
[email protected] | 5cbe1e2 | 2010-01-30 01:18:56 | [diff] [blame] | 198 | |
| 199 | // Accessibility. |
| 200 | RegisterFunction<GetFocusedControlFunction>(); |
| 201 | RegisterFunction<SetAccessibilityEnabledFunction>(); |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 202 | } |
| 203 | |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 204 | void FactoryRegistry::GetAllNames(std::vector<std::string>* names) { |
| 205 | for (FactoryMap::iterator iter = factories_.begin(); |
| 206 | iter != factories_.end(); ++iter) { |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 207 | names->push_back(iter->first); |
| 208 | } |
| 209 | } |
| 210 | |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 211 | bool FactoryRegistry::OverrideFunction(const std::string& name, |
| 212 | ExtensionFunctionFactory factory) { |
| 213 | FactoryMap::iterator iter = factories_.find(name); |
| 214 | if (iter == factories_.end()) { |
| 215 | return false; |
| 216 | } else { |
| 217 | iter->second = factory; |
| 218 | return true; |
| 219 | } |
| 220 | } |
| 221 | |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 222 | ExtensionFunction* FactoryRegistry::NewFunction(const std::string& name) { |
| 223 | FactoryMap::iterator iter = factories_.find(name); |
| 224 | DCHECK(iter != factories_.end()); |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 225 | ExtensionFunction* function = iter->second(); |
[email protected] | 76a3db85 | 2009-07-24 02:14:56 | [diff] [blame] | 226 | function->set_name(name); |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 227 | return function; |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 228 | } |
| 229 | |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 230 | }; // namespace |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 231 | |
[email protected] | a95631cb | 2009-12-10 01:59:11 | [diff] [blame] | 232 | // ExtensionFunctionDispatcher::Delegate --------------------------------------- |
| 233 | |
| 234 | gfx::NativeWindow ExtensionFunctionDispatcher::Delegate:: |
| 235 | GetFrameNativeWindow() { |
[email protected] | db7331a | 2010-02-25 22:10:50 | [diff] [blame] | 236 | Browser* browser = GetBrowser(true); |
[email protected] | a95631cb | 2009-12-10 01:59:11 | [diff] [blame] | 237 | // If a browser is bound to this dispatcher, then return the widget hosting |
| 238 | // the window. Extensions hosted in ExternalTabContainer objects may not |
| 239 | // have a running browser instance. |
| 240 | if (browser) |
| 241 | return browser->window()->GetNativeHandle(); |
| 242 | |
| 243 | return NULL; |
| 244 | } |
| 245 | |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 246 | // ExtensionFunctionDispatcher ------------------------------------------------- |
| 247 | |
| 248 | void ExtensionFunctionDispatcher::GetAllFunctionNames( |
| 249 | std::vector<std::string>* names) { |
| 250 | FactoryRegistry::instance()->GetAllNames(names); |
| 251 | } |
| 252 | |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 253 | bool ExtensionFunctionDispatcher::OverrideFunction( |
| 254 | const std::string& name, ExtensionFunctionFactory factory) { |
| 255 | return FactoryRegistry::instance()->OverrideFunction(name, factory); |
| 256 | } |
| 257 | |
| 258 | void ExtensionFunctionDispatcher::ResetFunctions() { |
| 259 | FactoryRegistry::instance()->ResetFunctions(); |
| 260 | } |
| 261 | |
[email protected] | 811bfe3 | 2009-07-01 08:46:25 | [diff] [blame] | 262 | std::set<ExtensionFunctionDispatcher*>* |
| 263 | ExtensionFunctionDispatcher::all_instances() { |
| 264 | static std::set<ExtensionFunctionDispatcher*> instances; |
| 265 | return &instances; |
| 266 | } |
| 267 | |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 268 | ExtensionFunctionDispatcher::ExtensionFunctionDispatcher( |
| 269 | RenderViewHost* render_view_host, |
[email protected] | 7eecaed5 | 2009-05-07 21:44:12 | [diff] [blame] | 270 | Delegate* delegate, |
[email protected] | 811bfe3 | 2009-07-01 08:46:25 | [diff] [blame] | 271 | const GURL& url) |
[email protected] | ebc1b68 | 2010-03-06 00:22:30 | [diff] [blame] | 272 | : profile_(render_view_host->process()->profile()), |
[email protected] | 68f0791 | 2010-03-05 18:33:58 | [diff] [blame] | 273 | render_view_host_(render_view_host), |
[email protected] | 7eecaed5 | 2009-05-07 21:44:12 | [diff] [blame] | 274 | delegate_(delegate), |
[email protected] | 811bfe3 | 2009-07-01 08:46:25 | [diff] [blame] | 275 | url_(url), |
[email protected] | 32dda36 | 2009-06-05 19:07:01 | [diff] [blame] | 276 | ALLOW_THIS_IN_INITIALIZER_LIST(peer_(new Peer(this))) { |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 277 | // TODO(erikkay) should we do something for these errors in Release? |
| 278 | DCHECK(url.SchemeIs(chrome::kExtensionScheme)); |
[email protected] | 3550635 | 2009-08-07 18:58:19 | [diff] [blame] | 279 | |
| 280 | Extension* extension = |
| 281 | profile()->GetExtensionsService()->GetExtensionByURL(url); |
| 282 | DCHECK(extension); |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 283 | |
[email protected] | 811bfe3 | 2009-07-01 08:46:25 | [diff] [blame] | 284 | all_instances()->insert(this); |
[email protected] | 0f605396 | 2009-07-09 19:26:35 | [diff] [blame] | 285 | |
[email protected] | 4577622 | 2009-07-15 20:21:58 | [diff] [blame] | 286 | // Notify the ExtensionProcessManager that the view was created. |
| 287 | ExtensionProcessManager* epm = profile()->GetExtensionProcessManager(); |
| 288 | epm->RegisterExtensionProcess(extension_id(), |
[email protected] | 76543b9 | 2009-08-31 17:27:45 | [diff] [blame] | 289 | render_view_host->process()->id()); |
[email protected] | 3550635 | 2009-08-07 18:58:19 | [diff] [blame] | 290 | |
[email protected] | db7331a | 2010-02-25 22:10:50 | [diff] [blame] | 291 | bool incognito_enabled = |
[email protected] | cb0ce1e02 | 2010-03-10 19:54:41 | [diff] [blame] | 292 | profile()->GetExtensionsService()->IsIncognitoEnabled(extension); |
[email protected] | db7331a | 2010-02-25 22:10:50 | [diff] [blame] | 293 | |
[email protected] | 3550635 | 2009-08-07 18:58:19 | [diff] [blame] | 294 | // Update the extension permissions. Doing this each time we create an EFD |
| 295 | // ensures that new processes are informed of permissions for newly installed |
| 296 | // extensions. |
[email protected] | cccf9093 | 2009-08-23 17:56:25 | [diff] [blame] | 297 | render_view_host->Send(new ViewMsg_Extension_SetAPIPermissions( |
[email protected] | 3550635 | 2009-08-07 18:58:19 | [diff] [blame] | 298 | extension->id(), extension->api_permissions())); |
[email protected] | cccf9093 | 2009-08-23 17:56:25 | [diff] [blame] | 299 | render_view_host->Send(new ViewMsg_Extension_SetHostPermissions( |
| 300 | extension->url(), extension->host_permissions())); |
[email protected] | db7331a | 2010-02-25 22:10:50 | [diff] [blame] | 301 | render_view_host->Send(new ViewMsg_Extension_ExtensionSetIncognitoEnabled( |
| 302 | extension->id(), incognito_enabled)); |
[email protected] | 68f0791 | 2010-03-05 18:33:58 | [diff] [blame] | 303 | |
| 304 | NotificationService::current()->Notify( |
| 305 | NotificationType::EXTENSION_FUNCTION_DISPATCHER_CREATED, |
| 306 | Source<Profile>(profile_), |
| 307 | Details<ExtensionFunctionDispatcher>(this)); |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 308 | } |
| 309 | |
[email protected] | 32dda36 | 2009-06-05 19:07:01 | [diff] [blame] | 310 | ExtensionFunctionDispatcher::~ExtensionFunctionDispatcher() { |
[email protected] | 811bfe3 | 2009-07-01 08:46:25 | [diff] [blame] | 311 | all_instances()->erase(this); |
[email protected] | 32dda36 | 2009-06-05 19:07:01 | [diff] [blame] | 312 | peer_->dispatcher_ = NULL; |
[email protected] | 68f0791 | 2010-03-05 18:33:58 | [diff] [blame] | 313 | |
| 314 | NotificationService::current()->Notify( |
| 315 | NotificationType::EXTENSION_FUNCTION_DISPATCHER_DESTROYED, |
| 316 | Source<Profile>(profile_), |
| 317 | Details<ExtensionFunctionDispatcher>(this)); |
[email protected] | 32dda36 | 2009-06-05 19:07:01 | [diff] [blame] | 318 | } |
| 319 | |
[email protected] | db7331a | 2010-02-25 22:10:50 | [diff] [blame] | 320 | Browser* ExtensionFunctionDispatcher::GetBrowser(bool include_incognito) { |
| 321 | return delegate_->GetBrowser(include_incognito); |
[email protected] | 7eecaed5 | 2009-05-07 21:44:12 | [diff] [blame] | 322 | } |
| 323 | |
[email protected] | b2725756 | 2009-11-16 23:28:26 | [diff] [blame] | 324 | ExtensionPopupHost* ExtensionFunctionDispatcher::GetPopupHost() { |
| 325 | ExtensionHost* extension_host = GetExtensionHost(); |
| 326 | if (extension_host) { |
| 327 | DCHECK(!GetExtensionDOMUI()) << |
| 328 | "Function dispatcher registered in too many environments."; |
| 329 | return extension_host->popup_host(); |
| 330 | } else { |
| 331 | ExtensionDOMUI* dom_ui = GetExtensionDOMUI(); |
| 332 | return dom_ui->popup_host(); |
| 333 | } |
| 334 | } |
| 335 | |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 336 | ExtensionHost* ExtensionFunctionDispatcher::GetExtensionHost() { |
[email protected] | 9c45b718 | 2009-08-04 16:44:43 | [diff] [blame] | 337 | return delegate_->GetExtensionHost(); |
| 338 | } |
| 339 | |
[email protected] | b2725756 | 2009-11-16 23:28:26 | [diff] [blame] | 340 | ExtensionDOMUI* ExtensionFunctionDispatcher::GetExtensionDOMUI() { |
| 341 | return delegate_->GetExtensionDOMUI(); |
| 342 | } |
| 343 | |
[email protected] | c7ad50f | 2009-09-11 06:28:15 | [diff] [blame] | 344 | Extension* ExtensionFunctionDispatcher::GetExtension() { |
| 345 | ExtensionsService* service = profile()->GetExtensionsService(); |
| 346 | DCHECK(service); |
| 347 | |
[email protected] | 61b41161 | 2009-11-10 23:17:41 | [diff] [blame] | 348 | Extension* extension = service->GetExtensionById(extension_id(), false); |
[email protected] | c7ad50f | 2009-09-11 06:28:15 | [diff] [blame] | 349 | DCHECK(extension); |
| 350 | |
| 351 | return extension; |
| 352 | } |
| 353 | |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 354 | void ExtensionFunctionDispatcher::HandleRequest(const std::string& name, |
[email protected] | e4dad9fb | 2009-10-06 18:15:58 | [diff] [blame] | 355 | const Value* args, |
[email protected] | c661918 | 2009-05-12 14:59:32 | [diff] [blame] | 356 | int request_id, |
| 357 | bool has_callback) { |
[email protected] | 32dda36 | 2009-06-05 19:07:01 | [diff] [blame] | 358 | scoped_refptr<ExtensionFunction> function( |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 359 | FactoryRegistry::instance()->NewFunction(name)); |
[email protected] | 32dda36 | 2009-06-05 19:07:01 | [diff] [blame] | 360 | function->set_dispatcher_peer(peer_); |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 361 | function->SetArgs(args); |
[email protected] | c661918 | 2009-05-12 14:59:32 | [diff] [blame] | 362 | function->set_request_id(request_id); |
| 363 | function->set_has_callback(has_callback); |
[email protected] | d13950e | 2009-12-04 01:43:02 | [diff] [blame] | 364 | ExtensionsService* service = profile()->GetExtensionsService(); |
| 365 | DCHECK(service); |
[email protected] | cb0ce1e02 | 2010-03-10 19:54:41 | [diff] [blame] | 366 | Extension* extension = service->GetExtensionById(extension_id(), false); |
| 367 | DCHECK(extension); |
| 368 | function->set_include_incognito(service->IsIncognitoEnabled(extension)); |
| 369 | |
[email protected] | d13950e | 2009-12-04 01:43:02 | [diff] [blame] | 370 | ExtensionsQuotaService* quota = service->quota_service(); |
| 371 | if (quota->Assess(extension_id(), function, args, base::TimeTicks::Now())) { |
| 372 | function->Run(); |
| 373 | } else { |
| 374 | render_view_host_->SendExtensionResponse(function->request_id(), false, |
| 375 | std::string(), QuotaLimitHeuristic::kGenericOverQuotaError); |
| 376 | } |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 377 | } |
| 378 | |
[email protected] | c661918 | 2009-05-12 14:59:32 | [diff] [blame] | 379 | void ExtensionFunctionDispatcher::SendResponse(ExtensionFunction* function, |
| 380 | bool success) { |
[email protected] | c661918 | 2009-05-12 14:59:32 | [diff] [blame] | 381 | render_view_host_->SendExtensionResponse(function->request_id(), success, |
[email protected] | b83e460 | 2009-05-15 22:58:33 | [diff] [blame] | 382 | function->GetResult(), function->GetError()); |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 383 | } |
| 384 | |
| 385 | void ExtensionFunctionDispatcher::HandleBadMessage(ExtensionFunction* api) { |
[email protected] | 25fd1b2e | 2009-08-17 20:57:14 | [diff] [blame] | 386 | LOG(ERROR) << "bad extension message " << |
[email protected] | 76543b9 | 2009-08-31 17:27:45 | [diff] [blame] | 387 | api->name() << |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 388 | " : terminating renderer."; |
| 389 | if (RenderProcessHost::run_renderer_in_process()) { |
| 390 | // In single process mode it is better if we don't suicide but just crash. |
| 391 | CHECK(false); |
| 392 | } else { |
| 393 | NOTREACHED(); |
[email protected] | 201b273 | 2009-11-13 18:57:46 | [diff] [blame] | 394 | base::KillProcess(render_view_host_->process()->GetHandle(), |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 395 | ResultCodes::KILLED_BAD_MESSAGE, false); |
| 396 | } |
| 397 | } |
| 398 | |
| 399 | Profile* ExtensionFunctionDispatcher::profile() { |
[email protected] | 68f0791 | 2010-03-05 18:33:58 | [diff] [blame] | 400 | return profile_; |
[email protected] | bfdffe2b | 2009-04-24 22:05:35 | [diff] [blame] | 401 | } |
[email protected] | a95631cb | 2009-12-10 01:59:11 | [diff] [blame] | 402 | |
| 403 | gfx::NativeWindow ExtensionFunctionDispatcher::GetFrameNativeWindow() { |
| 404 | return delegate_ ? delegate_->GetFrameNativeWindow() : NULL; |
| 405 | } |