OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 #include "chrome/browser/extensions/extension_function_dispatcher.h" | 5 #include "chrome/browser/extensions/extension_function_dispatcher.h" |
6 | 6 |
7 #include "base/process_util.h" | 7 #include "base/process_util.h" |
8 #include "base/singleton.h" | 8 #include "base/singleton.h" |
9 #include "base/values.h" | 9 #include "base/values.h" |
10 #include "chrome/browser/extensions/extension_bookmarks_module.h" | 10 #include "chrome/browser/extensions/extension_bookmarks_module.h" |
11 #include "chrome/browser/extensions/extension_bookmarks_module_constants.h" | 11 #include "chrome/browser/extensions/extension_bookmarks_module_constants.h" |
12 #include "chrome/browser/extensions/extension_function.h" | 12 #include "chrome/browser/extensions/extension_function.h" |
13 #include "chrome/browser/extensions/extension_message_service.h" | 13 #include "chrome/browser/extensions/extension_message_service.h" |
14 #include "chrome/browser/extensions/extension_page_actions_module.h" | 14 #include "chrome/browser/extensions/extension_page_actions_module.h" |
15 #include "chrome/browser/extensions/extension_page_actions_module_constants.h" | 15 #include "chrome/browser/extensions/extension_page_actions_module_constants.h" |
16 #include "chrome/browser/extensions/extension_process_manager.h" | 16 #include "chrome/browser/extensions/extension_process_manager.h" |
17 #include "chrome/browser/extensions/extension_tabs_module.h" | 17 #include "chrome/browser/extensions/extension_tabs_module.h" |
18 #include "chrome/browser/extensions/extension_tabs_module_constants.h" | 18 #include "chrome/browser/extensions/extension_tabs_module_constants.h" |
19 #include "chrome/browser/extensions/extension_toolstrip_api.h" | |
20 #include "chrome/browser/profile.h" | 19 #include "chrome/browser/profile.h" |
21 #include "chrome/browser/renderer_host/render_process_host.h" | 20 #include "chrome/browser/renderer_host/render_process_host.h" |
22 #include "chrome/browser/renderer_host/render_view_host.h" | 21 #include "chrome/browser/renderer_host/render_view_host.h" |
23 #include "chrome/common/result_codes.h" | 22 #include "chrome/common/result_codes.h" |
24 #include "chrome/common/url_constants.h" | |
25 | 23 |
26 // FactoryRegistry ------------------------------------------------------------- | 24 // FactoryRegistry ------------------------------------------------------------- |
27 | 25 |
28 namespace { | 26 namespace { |
29 | 27 |
30 // Template for defining ExtensionFunctionFactory. | 28 // Template for defining ExtensionFunctionFactory. |
31 template<class T> | 29 template<class T> |
32 ExtensionFunction* NewExtensionFunction() { | 30 ExtensionFunction* NewExtensionFunction() { |
33 return new T(); | 31 return new T(); |
34 } | 32 } |
(...skipping 27 matching lines...) Expand all Loading... |
62 FactoryRegistry* FactoryRegistry::instance() { | 60 FactoryRegistry* FactoryRegistry::instance() { |
63 return Singleton<FactoryRegistry>::get(); | 61 return Singleton<FactoryRegistry>::get(); |
64 } | 62 } |
65 | 63 |
66 void FactoryRegistry::ResetFunctions() { | 64 void FactoryRegistry::ResetFunctions() { |
67 // Register all functions here. | 65 // Register all functions here. |
68 | 66 |
69 namespace tabs = extension_tabs_module_constants; | 67 namespace tabs = extension_tabs_module_constants; |
70 namespace page_actions = extension_page_actions_module_constants; | 68 namespace page_actions = extension_page_actions_module_constants; |
71 namespace bookmarks = extension_bookmarks_module_constants; | 69 namespace bookmarks = extension_bookmarks_module_constants; |
72 namespace toolstrip = extension_toolstrip_api_functions; | |
73 | 70 |
74 // Windows | 71 // Windows |
75 factories_[tabs::kGetWindowFunction] = | 72 factories_[tabs::kGetWindowFunction] = |
76 &NewExtensionFunction<GetWindowFunction>; | 73 &NewExtensionFunction<GetWindowFunction>; |
77 factories_[tabs::kGetCurrentWindowFunction] = | 74 factories_[tabs::kGetCurrentWindowFunction] = |
78 &NewExtensionFunction<GetCurrentWindowFunction>; | 75 &NewExtensionFunction<GetCurrentWindowFunction>; |
79 factories_[tabs::kGetLastFocusedWindowFunction] = | 76 factories_[tabs::kGetLastFocusedWindowFunction] = |
80 &NewExtensionFunction<GetLastFocusedWindowFunction>; | 77 &NewExtensionFunction<GetLastFocusedWindowFunction>; |
81 factories_[tabs::kGetAllWindowsFunction] = | 78 factories_[tabs::kGetAllWindowsFunction] = |
82 &NewExtensionFunction<GetAllWindowsFunction>; | 79 &NewExtensionFunction<GetAllWindowsFunction>; |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
123 factories_[bookmarks::kRemoveBookmarkFunction] = | 120 factories_[bookmarks::kRemoveBookmarkFunction] = |
124 &NewExtensionFunction<RemoveBookmarkFunction>; | 121 &NewExtensionFunction<RemoveBookmarkFunction>; |
125 factories_[bookmarks::kRemoveBookmarkTreeFunction] = | 122 factories_[bookmarks::kRemoveBookmarkTreeFunction] = |
126 &NewExtensionFunction<RemoveBookmarkFunction>; | 123 &NewExtensionFunction<RemoveBookmarkFunction>; |
127 factories_[bookmarks::kCreateBookmarkFunction] = | 124 factories_[bookmarks::kCreateBookmarkFunction] = |
128 &NewExtensionFunction<CreateBookmarkFunction>; | 125 &NewExtensionFunction<CreateBookmarkFunction>; |
129 factories_[bookmarks::kMoveBookmarkFunction] = | 126 factories_[bookmarks::kMoveBookmarkFunction] = |
130 &NewExtensionFunction<MoveBookmarkFunction>; | 127 &NewExtensionFunction<MoveBookmarkFunction>; |
131 factories_[bookmarks::kSetBookmarkTitleFunction] = | 128 factories_[bookmarks::kSetBookmarkTitleFunction] = |
132 &NewExtensionFunction<SetBookmarkTitleFunction>; | 129 &NewExtensionFunction<SetBookmarkTitleFunction>; |
133 | |
134 // Toolstrips. | |
135 factories_[toolstrip::kExpandFunction] = | |
136 &NewExtensionFunction<ToolstripExpandFunction>; | |
137 factories_[toolstrip::kCollapseFunction] = | |
138 &NewExtensionFunction<ToolstripCollapseFunction>; | |
139 } | 130 } |
140 | 131 |
141 void FactoryRegistry::GetAllNames(std::vector<std::string>* names) { | 132 void FactoryRegistry::GetAllNames(std::vector<std::string>* names) { |
142 for (FactoryMap::iterator iter = factories_.begin(); | 133 for (FactoryMap::iterator iter = factories_.begin(); |
143 iter != factories_.end(); ++iter) { | 134 iter != factories_.end(); ++iter) { |
144 names->push_back(iter->first); | 135 names->push_back(iter->first); |
145 } | 136 } |
146 } | 137 } |
147 | 138 |
148 bool FactoryRegistry::OverrideFunction(const std::string& name, | 139 bool FactoryRegistry::OverrideFunction(const std::string& name, |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
189 } | 180 } |
190 | 181 |
191 ExtensionFunctionDispatcher::ExtensionFunctionDispatcher( | 182 ExtensionFunctionDispatcher::ExtensionFunctionDispatcher( |
192 RenderViewHost* render_view_host, | 183 RenderViewHost* render_view_host, |
193 Delegate* delegate, | 184 Delegate* delegate, |
194 const GURL& url) | 185 const GURL& url) |
195 : render_view_host_(render_view_host), | 186 : render_view_host_(render_view_host), |
196 delegate_(delegate), | 187 delegate_(delegate), |
197 url_(url), | 188 url_(url), |
198 ALLOW_THIS_IN_INITIALIZER_LIST(peer_(new Peer(this))) { | 189 ALLOW_THIS_IN_INITIALIZER_LIST(peer_(new Peer(this))) { |
199 // TODO(erikkay) should we do something for these errors in Release? | |
200 DCHECK(url.SchemeIs(chrome::kExtensionScheme)); | |
201 DCHECK(profile()->GetExtensionsService()->GetExtensionByURL(url)); | |
202 | |
203 all_instances()->insert(this); | 190 all_instances()->insert(this); |
204 | 191 |
205 // Notify the ExtensionProcessManager that the view was created. | 192 // Notify the ExtensionProcessManager that the view was created. |
206 ExtensionProcessManager* epm = profile()->GetExtensionProcessManager(); | 193 ExtensionProcessManager* epm = profile()->GetExtensionProcessManager(); |
207 epm->RegisterExtensionProcess(extension_id(), | 194 epm->RegisterExtensionProcess(extension_id(), |
208 render_view_host->process()->pid()); | 195 render_view_host->process()->pid()); |
209 } | 196 } |
210 | 197 |
211 ExtensionFunctionDispatcher::~ExtensionFunctionDispatcher() { | 198 ExtensionFunctionDispatcher::~ExtensionFunctionDispatcher() { |
212 all_instances()->erase(this); | 199 all_instances()->erase(this); |
213 peer_->dispatcher_ = NULL; | 200 peer_->dispatcher_ = NULL; |
214 } | 201 } |
215 | 202 |
216 Browser* ExtensionFunctionDispatcher::GetBrowser() { | 203 Browser* ExtensionFunctionDispatcher::GetBrowser() { |
217 DCHECK(delegate_); | 204 DCHECK(delegate_); |
218 | 205 |
219 Browser* retval = delegate_->GetBrowser(); | 206 Browser* retval = delegate_->GetBrowser(); |
220 return retval; | 207 return retval; |
221 } | 208 } |
222 | 209 |
223 ExtensionHost* ExtensionFunctionDispatcher::GetExtensionHost() { | |
224 DCHECK(delegate_); | |
225 return delegate_->GetExtensionHost(); | |
226 } | |
227 | |
228 void ExtensionFunctionDispatcher::HandleRequest(const std::string& name, | 210 void ExtensionFunctionDispatcher::HandleRequest(const std::string& name, |
229 const std::string& args, | 211 const std::string& args, |
230 int request_id, | 212 int request_id, |
231 bool has_callback) { | 213 bool has_callback) { |
232 scoped_refptr<ExtensionFunction> function( | 214 scoped_refptr<ExtensionFunction> function( |
233 FactoryRegistry::instance()->NewFunction(name)); | 215 FactoryRegistry::instance()->NewFunction(name)); |
234 function->set_dispatcher_peer(peer_); | 216 function->set_dispatcher_peer(peer_); |
235 function->SetArgs(args); | 217 function->SetArgs(args); |
236 function->set_request_id(request_id); | 218 function->set_request_id(request_id); |
237 function->set_has_callback(has_callback); | 219 function->set_has_callback(has_callback); |
(...skipping 15 matching lines...) Expand all Loading... |
253 } else { | 235 } else { |
254 NOTREACHED(); | 236 NOTREACHED(); |
255 base::KillProcess(render_view_host_->process()->process().handle(), | 237 base::KillProcess(render_view_host_->process()->process().handle(), |
256 ResultCodes::KILLED_BAD_MESSAGE, false); | 238 ResultCodes::KILLED_BAD_MESSAGE, false); |
257 } | 239 } |
258 } | 240 } |
259 | 241 |
260 Profile* ExtensionFunctionDispatcher::profile() { | 242 Profile* ExtensionFunctionDispatcher::profile() { |
261 return render_view_host_->process()->profile(); | 243 return render_view_host_->process()->profile(); |
262 } | 244 } |
OLD | NEW |