OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_event_router.h" | 5 #include "chrome/browser/extensions/extension_event_router.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/command_line.h" | 8 #include "base/command_line.h" |
9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
10 #include "base/values.h" | 10 #include "base/values.h" |
(...skipping 47 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
58 | 58 |
59 bool operator<(const ListenerProcess& that) const { | 59 bool operator<(const ListenerProcess& that) const { |
60 if (process < that.process) | 60 if (process < that.process) |
61 return true; | 61 return true; |
62 if (process == that.process && extension_id < that.extension_id) | 62 if (process == that.process && extension_id < that.extension_id) |
63 return true; | 63 return true; |
64 return false; | 64 return false; |
65 } | 65 } |
66 }; | 66 }; |
67 | 67 |
68 struct ExtensionEventRouter::ExtensionEvent { | |
69 std::string event_name; | |
70 std::string event_args; | |
71 GURL event_url; | |
72 Profile* restrict_to_profile; | |
73 std::string cross_incognito_args; | |
74 UserGestureState user_gesture; | |
75 | |
76 ExtensionEvent(const std::string& event_name, | |
77 const std::string& event_args, | |
78 const GURL& event_url, | |
79 Profile* restrict_to_profile, | |
80 const std::string& cross_incognito_args, | |
81 UserGestureState user_gesture) | |
82 : event_name(event_name), | |
83 event_args(event_args), | |
84 event_url(event_url), | |
85 restrict_to_profile(restrict_to_profile), | |
86 cross_incognito_args(cross_incognito_args), | |
87 user_gesture(user_gesture) {} | |
88 }; | |
89 | |
90 // static | 68 // static |
91 void ExtensionEventRouter::DispatchEvent(IPC::Message::Sender* ipc_sender, | 69 void ExtensionEventRouter::DispatchEvent( |
92 const std::string& extension_id, | 70 IPC::Message::Sender* ipc_sender, |
93 const std::string& event_name, | 71 const std::string& extension_id, |
94 const std::string& event_args, | 72 const std::string& event_name, |
95 const GURL& event_url, | 73 const std::string& event_args, |
96 UserGestureState user_gesture) { | 74 const GURL& event_url, |
75 UserGestureState user_gesture, | |
76 const extensions::EventFilteringInfo& info) { | |
97 ListValue args; | 77 ListValue args; |
98 args.Set(0, Value::CreateStringValue(event_name)); | 78 args.Set(0, Value::CreateStringValue(event_name)); |
99 args.Set(1, Value::CreateStringValue(event_args)); | 79 args.Set(1, Value::CreateStringValue(event_args)); |
80 args.Set(2, Value::CreateStringValue(info.AsJSONString())); | |
Aaron Boodman
2012/06/15 07:34:51
Can you juts pass this through as a Value? Seriali
koz (OOO until 15th September)
2012/06/18 23:27:27
Done.
Matt Perry
2012/06/19 00:02:55
Doesn't look like it :)
koz (OOO until 15th September)
2012/06/19 00:11:36
Oops, I think I forgot to upload this patch. I'm r
| |
100 ipc_sender->Send(new ExtensionMsg_MessageInvoke(MSG_ROUTING_CONTROL, | 81 ipc_sender->Send(new ExtensionMsg_MessageInvoke(MSG_ROUTING_CONTROL, |
101 extension_id, kDispatchEvent, args, event_url, | 82 extension_id, kDispatchEvent, args, event_url, |
102 user_gesture == USER_GESTURE_ENABLED)); | 83 user_gesture == USER_GESTURE_ENABLED)); |
103 } | 84 } |
104 | 85 |
105 ExtensionEventRouter::ExtensionEventRouter(Profile* profile) | 86 ExtensionEventRouter::ExtensionEventRouter(Profile* profile) |
106 : profile_(profile), | 87 : profile_(profile), |
107 extension_devtools_manager_( | 88 extension_devtools_manager_( |
108 ExtensionSystem::Get(profile)->devtools_manager()) { | 89 ExtensionSystem::Get(profile)->devtools_manager()), |
90 listeners_(ALLOW_THIS_IN_INITIALIZER_LIST(this)) { | |
109 registrar_.Add(this, content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, | 91 registrar_.Add(this, content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, |
110 content::NotificationService::AllSources()); | 92 content::NotificationService::AllSources()); |
111 registrar_.Add(this, content::NOTIFICATION_RENDERER_PROCESS_CLOSED, | 93 registrar_.Add(this, content::NOTIFICATION_RENDERER_PROCESS_CLOSED, |
112 content::NotificationService::AllSources()); | 94 content::NotificationService::AllSources()); |
113 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED, | 95 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED, |
114 content::Source<Profile>(profile_)); | 96 content::Source<Profile>(profile_)); |
115 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED, | 97 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED, |
116 content::Source<Profile>(profile_)); | 98 content::Source<Profile>(profile_)); |
117 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_INSTALLED, | 99 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_INSTALLED, |
118 content::Source<Profile>(profile_)); | 100 content::Source<Profile>(profile_)); |
119 } | 101 } |
120 | 102 |
121 ExtensionEventRouter::~ExtensionEventRouter() {} | 103 ExtensionEventRouter::~ExtensionEventRouter() {} |
122 | 104 |
123 void ExtensionEventRouter::AddEventListener( | 105 void ExtensionEventRouter::AddEventListener( |
124 const std::string& event_name, | 106 const std::string& event_name, |
125 content::RenderProcessHost* process, | 107 content::RenderProcessHost* process, |
126 const std::string& extension_id) { | 108 const std::string& extension_id) { |
127 ListenerProcess listener(process, extension_id); | 109 listeners_.AddListener(scoped_ptr<EventListener>(new EventListener( |
128 DCHECK_EQ(listeners_[event_name].count(listener), 0u) << event_name; | 110 event_name, extension_id, process, scoped_ptr<DictionaryValue>()))); |
129 listeners_[event_name].insert(listener); | 111 } |
112 | |
113 void ExtensionEventRouter::RemoveEventListener( | |
114 const std::string& event_name, | |
115 content::RenderProcessHost* process, | |
116 const std::string& extension_id) { | |
117 scoped_ptr<EventListener> listener(new EventListener( | |
118 event_name, extension_id, process, scoped_ptr<DictionaryValue>())); | |
119 listeners_.RemoveListener(listener.get()); | |
120 } | |
121 | |
122 void ExtensionEventRouter::OnListenerAdded(const EventListener* listener) { | |
Aaron Boodman
2012/06/15 07:34:51
Instead of the delegate interface can you just loo
koz (OOO until 15th September)
2012/06/18 23:27:27
The delegate interface was requested by Matt, and
| |
123 // We don't care about lazy events being added. | |
124 if (!listener->process) | |
125 return; | |
130 | 126 |
131 if (extension_devtools_manager_.get()) | 127 if (extension_devtools_manager_.get()) |
132 extension_devtools_manager_->AddEventListener(event_name, | 128 extension_devtools_manager_->AddEventListener(listener->event_name, |
133 process->GetID()); | 129 listener->process->GetID()); |
134 | 130 |
135 // We lazily tell the TaskManager to start updating when listeners to the | 131 // We lazily tell the TaskManager to start updating when listeners to the |
136 // processes.onUpdated or processes.onUpdatedWithMemory events arrive. | 132 // processes.onUpdated or processes.onUpdatedWithMemory events arrive. |
133 const std::string& event_name = listener->event_name; | |
137 if (event_name.compare(extension_processes_api_constants::kOnUpdated) == 0 || | 134 if (event_name.compare(extension_processes_api_constants::kOnUpdated) == 0 || |
138 event_name.compare( | 135 event_name.compare( |
139 extension_processes_api_constants::kOnUpdatedWithMemory) == 0) | 136 extension_processes_api_constants::kOnUpdatedWithMemory) == 0) |
140 ExtensionProcessesEventRouter::GetInstance()->ListenerAdded(); | 137 ExtensionProcessesEventRouter::GetInstance()->ListenerAdded(); |
141 } | 138 } |
142 | 139 |
143 void ExtensionEventRouter::RemoveEventListener( | 140 void ExtensionEventRouter::OnListenerRemoved(const EventListener* listener) { |
144 const std::string& event_name, | 141 // We don't care about lazy events being removed. |
145 content::RenderProcessHost* process, | 142 if (!listener->process) |
146 const std::string& extension_id) { | 143 return; |
147 ListenerProcess listener(process, extension_id); | |
148 DCHECK_EQ(listeners_[event_name].count(listener), 1u) << | |
149 " PID=" << process->GetID() << " extension=" << extension_id << | |
150 " event=" << event_name; | |
151 listeners_[event_name].erase(listener); | |
152 // Note: extension_id may point to data in the now-deleted listeners_ object. | |
153 // Do not use. | |
154 | 144 |
145 const std::string& event_name = listener->event_name; | |
155 if (extension_devtools_manager_.get()) | 146 if (extension_devtools_manager_.get()) |
156 extension_devtools_manager_->RemoveEventListener(event_name, | 147 extension_devtools_manager_->RemoveEventListener( |
157 process->GetID()); | 148 event_name, listener->process->GetID()); |
158 | 149 |
159 // If a processes.onUpdated or processes.onUpdatedWithMemory event listener | 150 // If a processes.onUpdated or processes.onUpdatedWithMemory event listener |
160 // is removed (or a process with one exits), then we let the extension API | 151 // is removed (or a process with one exits), then we let the extension API |
161 // know that it has one fewer listener. | 152 // know that it has one fewer listener. |
162 if (event_name.compare(extension_processes_api_constants::kOnUpdated) == 0 || | 153 if (event_name.compare(extension_processes_api_constants::kOnUpdated) == 0 || |
163 event_name.compare( | 154 event_name.compare( |
164 extension_processes_api_constants::kOnUpdatedWithMemory) == 0) | 155 extension_processes_api_constants::kOnUpdatedWithMemory) == 0) |
165 ExtensionProcessesEventRouter::GetInstance()->ListenerRemoved(); | 156 ExtensionProcessesEventRouter::GetInstance()->ListenerRemoved(); |
166 | 157 |
167 BrowserThread::PostTask( | 158 BrowserThread::PostTask( |
168 BrowserThread::IO, FROM_HERE, | 159 BrowserThread::IO, FROM_HERE, |
169 base::Bind( | 160 base::Bind( |
170 &NotifyEventListenerRemovedOnIOThread, | 161 &NotifyEventListenerRemovedOnIOThread, |
171 profile_, listener.extension_id, event_name)); | 162 profile_, listener->extension_id, listener->event_name)); |
172 } | 163 } |
173 | 164 |
174 void ExtensionEventRouter::AddLazyEventListener( | 165 void ExtensionEventRouter::AddLazyEventListener( |
175 const std::string& event_name, | 166 const std::string& event_name, |
176 const std::string& extension_id) { | 167 const std::string& extension_id) { |
177 ListenerProcess lazy_listener(NULL, extension_id); | 168 scoped_ptr<EventListener> listener(new EventListener( |
178 bool is_new = lazy_listeners_[event_name].insert(lazy_listener).second; | 169 event_name, extension_id, NULL, scoped_ptr<DictionaryValue>())); |
170 bool is_new = listeners_.AddListener(listener.Pass()); | |
171 | |
179 if (is_new) { | 172 if (is_new) { |
180 ExtensionPrefs* prefs = profile_->GetExtensionService()->extension_prefs(); | 173 ExtensionPrefs* prefs = profile_->GetExtensionService()->extension_prefs(); |
181 std::set<std::string> events = prefs->GetRegisteredEvents(extension_id); | 174 std::set<std::string> events = prefs->GetRegisteredEvents(extension_id); |
182 bool prefs_is_new = events.insert(event_name).second; | 175 bool prefs_is_new = events.insert(event_name).second; |
183 if (prefs_is_new) | 176 if (prefs_is_new) |
184 prefs->SetRegisteredEvents(extension_id, events); | 177 prefs->SetRegisteredEvents(extension_id, events); |
185 } | 178 } |
186 } | 179 } |
187 | 180 |
188 void ExtensionEventRouter::RemoveLazyEventListener( | 181 void ExtensionEventRouter::RemoveLazyEventListener( |
189 const std::string& event_name, | 182 const std::string& event_name, |
190 const std::string& extension_id) { | 183 const std::string& extension_id) { |
191 ListenerProcess lazy_listener(NULL, extension_id); | 184 scoped_ptr<EventListener> listener(new EventListener( |
192 bool did_exist = lazy_listeners_[event_name].erase(lazy_listener) > 0; | 185 event_name, extension_id, NULL, scoped_ptr<DictionaryValue>())); |
186 bool did_exist = listeners_.RemoveListener(listener.get()); | |
187 | |
193 if (did_exist) { | 188 if (did_exist) { |
194 ExtensionPrefs* prefs = profile_->GetExtensionService()->extension_prefs(); | 189 ExtensionPrefs* prefs = profile_->GetExtensionService()->extension_prefs(); |
195 std::set<std::string> events = prefs->GetRegisteredEvents(extension_id); | 190 std::set<std::string> events = prefs->GetRegisteredEvents(extension_id); |
196 bool prefs_did_exist = events.erase(event_name) > 0; | 191 bool prefs_did_exist = events.erase(event_name) > 0; |
197 DCHECK(prefs_did_exist); | 192 DCHECK(prefs_did_exist); |
198 prefs->SetRegisteredEvents(extension_id, events); | 193 prefs->SetRegisteredEvents(extension_id, events); |
199 } | 194 } |
200 } | 195 } |
201 | 196 |
197 void ExtensionEventRouter::AddFilteredEventListener( | |
198 const std::string& event_name, | |
199 content::RenderProcessHost* process, | |
200 const std::string& extension_id, | |
201 const base::DictionaryValue& filter, | |
202 bool add_lazy_listener) { | |
203 listeners_.AddListener(scoped_ptr<EventListener>(new EventListener( | |
204 event_name, extension_id, process, | |
205 scoped_ptr<DictionaryValue>(filter.DeepCopy())))); | |
206 | |
207 if (add_lazy_listener) { | |
208 bool added = listeners_.AddListener(scoped_ptr<EventListener>( | |
209 new EventListener(event_name, extension_id, NULL, | |
210 scoped_ptr<DictionaryValue>(filter.DeepCopy())))); | |
211 | |
212 if (added) { | |
213 ExtensionPrefs* prefs = | |
214 profile_->GetExtensionService()->extension_prefs(); | |
215 prefs->AddFilterToEvent(event_name, extension_id, &filter); | |
216 } | |
217 } | |
218 } | |
219 | |
220 void ExtensionEventRouter::RemoveFilteredEventListener( | |
221 const std::string& event_name, | |
222 content::RenderProcessHost* process, | |
223 const std::string& extension_id, | |
224 const base::DictionaryValue& filter, | |
225 bool remove_lazy_listener) { | |
226 scoped_ptr<EventListener> listener(new EventListener( | |
227 event_name, extension_id, process, | |
228 scoped_ptr<DictionaryValue>(filter.DeepCopy()))); | |
229 | |
230 listeners_.RemoveListener(listener.get()); | |
231 | |
232 if (remove_lazy_listener) { | |
233 listener->process = NULL; | |
234 bool removed = listeners_.RemoveListener(listener.get()); | |
235 | |
236 if (removed) { | |
237 ExtensionPrefs* prefs = | |
238 profile_->GetExtensionService()->extension_prefs(); | |
239 prefs->RemoveFilterFromEvent(event_name, extension_id, &filter); | |
240 } | |
241 } | |
242 } | |
243 | |
202 bool ExtensionEventRouter::HasEventListener(const std::string& event_name) { | 244 bool ExtensionEventRouter::HasEventListener(const std::string& event_name) { |
203 return (HasEventListenerImpl(listeners_, "", event_name) || | 245 return listeners_.HasListenerForEvent(event_name); |
204 HasEventListenerImpl(lazy_listeners_, "", event_name)); | |
205 } | 246 } |
206 | 247 |
207 bool ExtensionEventRouter::ExtensionHasEventListener( | 248 bool ExtensionEventRouter::ExtensionHasEventListener( |
208 const std::string& extension_id, const std::string& event_name) { | 249 const std::string& extension_id, const std::string& event_name) { |
209 return (HasEventListenerImpl(listeners_, extension_id, event_name) || | 250 return listeners_.HasListenerForExtension(extension_id, event_name); |
210 HasEventListenerImpl(lazy_listeners_, extension_id, event_name)); | |
211 } | 251 } |
212 | 252 |
213 bool ExtensionEventRouter::HasEventListenerImpl( | 253 bool ExtensionEventRouter::HasEventListenerImpl( |
214 const ListenerMap& listener_map, | 254 const ListenerMap& listener_map, |
215 const std::string& extension_id, | 255 const std::string& extension_id, |
216 const std::string& event_name) { | 256 const std::string& event_name) { |
217 ListenerMap::const_iterator it = listener_map.find(event_name); | 257 ListenerMap::const_iterator it = listener_map.find(event_name); |
218 if (it == listener_map.end()) | 258 if (it == listener_map.end()) |
219 return false; | 259 return false; |
220 | 260 |
221 const std::set<ListenerProcess>& listeners = it->second; | 261 const std::set<ListenerProcess>& listeners = it->second; |
222 if (extension_id.empty()) | 262 if (extension_id.empty()) |
223 return !listeners.empty(); | 263 return !listeners.empty(); |
224 | 264 |
225 for (std::set<ListenerProcess>::const_iterator listener = listeners.begin(); | 265 for (std::set<ListenerProcess>::const_iterator listener = listeners.begin(); |
226 listener != listeners.end(); ++listener) { | 266 listener != listeners.end(); ++listener) { |
227 if (listener->extension_id == extension_id) | 267 if (listener->extension_id == extension_id) |
228 return true; | 268 return true; |
229 } | 269 } |
230 return false; | 270 return false; |
231 } | 271 } |
232 | 272 |
233 void ExtensionEventRouter::DispatchEventToRenderers( | 273 void ExtensionEventRouter::DispatchEventToRenderers( |
234 const std::string& event_name, | 274 const std::string& event_name, |
235 const std::string& event_args, | 275 const std::string& event_args, |
236 Profile* restrict_to_profile, | 276 Profile* restrict_to_profile, |
237 const GURL& event_url) { | 277 const GURL& event_url, |
278 extensions::EventFilteringInfo info) { | |
238 linked_ptr<ExtensionEvent> event( | 279 linked_ptr<ExtensionEvent> event( |
239 new ExtensionEvent(event_name, event_args, event_url, | 280 new ExtensionEvent(event_name, event_args, event_url, |
240 restrict_to_profile, "", USER_GESTURE_UNKNOWN)); | 281 restrict_to_profile, "", USER_GESTURE_UNKNOWN, info)); |
241 DispatchEventImpl("", event); | 282 DispatchEventImpl("", event); |
242 } | 283 } |
243 | 284 |
244 void ExtensionEventRouter::DispatchEventToExtension( | 285 void ExtensionEventRouter::DispatchEventToExtension( |
245 const std::string& extension_id, | 286 const std::string& extension_id, |
246 const std::string& event_name, | 287 const std::string& event_name, |
247 const std::string& event_args, | 288 const std::string& event_args, |
248 Profile* restrict_to_profile, | 289 Profile* restrict_to_profile, |
249 const GURL& event_url) { | 290 const GURL& event_url) { |
250 DCHECK(!extension_id.empty()); | 291 DCHECK(!extension_id.empty()); |
251 linked_ptr<ExtensionEvent> event( | 292 linked_ptr<ExtensionEvent> event( |
252 new ExtensionEvent(event_name, event_args, event_url, | 293 new ExtensionEvent(event_name, event_args, event_url, |
253 restrict_to_profile, "", USER_GESTURE_UNKNOWN)); | 294 restrict_to_profile, "", USER_GESTURE_UNKNOWN, |
295 EventFilteringInfo())); | |
254 DispatchEventImpl(extension_id, event); | 296 DispatchEventImpl(extension_id, event); |
255 } | 297 } |
256 | 298 |
257 void ExtensionEventRouter::DispatchEventToExtension( | 299 void ExtensionEventRouter::DispatchEventToExtension( |
258 const std::string& extension_id, | 300 const std::string& extension_id, |
259 const std::string& event_name, | 301 const std::string& event_name, |
260 const std::string& event_args, | 302 const std::string& event_args, |
261 Profile* restrict_to_profile, | 303 Profile* restrict_to_profile, |
262 const GURL& event_url, | 304 const GURL& event_url, |
263 UserGestureState user_gesture) { | 305 UserGestureState user_gesture) { |
264 DCHECK(!extension_id.empty()); | 306 DCHECK(!extension_id.empty()); |
265 linked_ptr<ExtensionEvent> event( | 307 linked_ptr<ExtensionEvent> event( |
266 new ExtensionEvent(event_name, event_args, event_url, | 308 new ExtensionEvent(event_name, event_args, event_url, |
267 restrict_to_profile, "", user_gesture)); | 309 restrict_to_profile, "", user_gesture, |
310 EventFilteringInfo())); | |
268 DispatchEventImpl(extension_id, event); | 311 DispatchEventImpl(extension_id, event); |
269 } | 312 } |
270 | 313 |
271 void ExtensionEventRouter::DispatchEventsToRenderersAcrossIncognito( | 314 void ExtensionEventRouter::DispatchEventsToRenderersAcrossIncognito( |
272 const std::string& event_name, | 315 const std::string& event_name, |
273 const std::string& event_args, | 316 const std::string& event_args, |
274 Profile* restrict_to_profile, | 317 Profile* restrict_to_profile, |
275 const std::string& cross_incognito_args, | 318 const std::string& cross_incognito_args, |
276 const GURL& event_url) { | 319 const GURL& event_url) { |
277 linked_ptr<ExtensionEvent> event( | 320 linked_ptr<ExtensionEvent> event( |
278 new ExtensionEvent(event_name, event_args, event_url, | 321 new ExtensionEvent(event_name, event_args, event_url, |
279 restrict_to_profile, cross_incognito_args, | 322 restrict_to_profile, cross_incognito_args, |
280 USER_GESTURE_UNKNOWN)); | 323 USER_GESTURE_UNKNOWN, EventFilteringInfo())); |
281 DispatchEventImpl("", event); | 324 DispatchEventImpl("", event); |
282 } | 325 } |
283 | 326 |
284 void ExtensionEventRouter::DispatchEventImpl( | 327 void ExtensionEventRouter::DispatchEventImpl( |
285 const std::string& extension_id, | 328 const std::string& restrict_to_extension_id, |
286 const linked_ptr<ExtensionEvent>& event) { | 329 const linked_ptr<ExtensionEvent>& event) { |
287 // We don't expect to get events from a completely different profile. | 330 // We don't expect to get events from a completely different profile. |
288 DCHECK(!event->restrict_to_profile || | 331 DCHECK(!event->restrict_to_profile || |
289 profile_->IsSameProfile(event->restrict_to_profile)); | 332 profile_->IsSameProfile(event->restrict_to_profile)); |
290 | 333 |
291 LoadLazyBackgroundPagesForEvent(extension_id, event); | 334 std::set<const EventListener*> listeners(listeners_.GetEventTargets(*event)); |
292 | 335 for (std::set<const EventListener*>::iterator it = listeners.begin(); |
293 ListenerMap::iterator it = listeners_.find(event->event_name); | 336 it != listeners.end(); it++) { |
294 if (it == listeners_.end()) | 337 const EventListener* listener = *it; |
295 return; | 338 if (listener->process) { |
296 | 339 if (restrict_to_extension_id.empty() || |
297 std::set<ListenerProcess>& listeners = it->second; | 340 restrict_to_extension_id == listener->extension_id) |
298 for (std::set<ListenerProcess>::iterator listener = listeners.begin(); | 341 DispatchEventToProcess(listener->extension_id, listener->process, |
299 listener != listeners.end(); ++listener) { | 342 event); |
300 if (!extension_id.empty() && extension_id != listener->extension_id) | 343 } else { |
301 continue; | 344 DispatchLazyEvent(listener->extension_id, event); |
302 | 345 } |
303 DispatchEventToListener(*listener, event); | |
304 } | 346 } |
305 } | 347 } |
306 | 348 |
307 void ExtensionEventRouter::DispatchEventToListener( | 349 void ExtensionEventRouter::DispatchLazyEvent( |
308 const ListenerProcess& listener, | 350 const std::string& extension_id, |
351 const linked_ptr<ExtensionEvent>& event) { | |
352 ExtensionService* service = profile_->GetExtensionService(); | |
353 // Check both the original and the incognito profile to see if we | |
354 // should load a lazy bg page to handle the event. The latter case | |
355 // occurs in the case of split-mode extensions. | |
356 const Extension* extension = service->extensions()->GetByID(extension_id); | |
357 if (extension) { | |
358 MaybeLoadLazyBackgroundPage(profile_, extension, event, extension_id); | |
359 if (profile_->HasOffTheRecordProfile() && | |
360 extension->incognito_split_mode()) { | |
361 MaybeLoadLazyBackgroundPage( | |
362 profile_->GetOffTheRecordProfile(), extension, event, extension_id); | |
363 } | |
364 } | |
365 } | |
366 | |
367 void ExtensionEventRouter::DispatchEventToProcess( | |
368 const std::string& extension_id, | |
369 content::RenderProcessHost* process, | |
309 const linked_ptr<ExtensionEvent>& event) { | 370 const linked_ptr<ExtensionEvent>& event) { |
310 ExtensionService* service = profile_->GetExtensionService(); | 371 ExtensionService* service = profile_->GetExtensionService(); |
311 const Extension* extension = service->extensions()->GetByID( | 372 const Extension* extension = service->extensions()->GetByID( |
312 listener.extension_id); | 373 extension_id); |
313 | 374 |
314 // The extension could have been removed, but we do not unregister it until | 375 // The extension could have been removed, but we do not unregister it until |
315 // the extension process is unloaded. | 376 // the extension process is unloaded. |
316 if (!extension) | 377 if (!extension) |
317 return; | 378 return; |
318 | 379 |
319 Profile* listener_profile = Profile::FromBrowserContext( | 380 Profile* listener_profile = Profile::FromBrowserContext( |
320 listener.process->GetBrowserContext()); | 381 process->GetBrowserContext()); |
321 extensions::ProcessMap* process_map = | 382 extensions::ProcessMap* process_map = |
322 listener_profile->GetExtensionService()->process_map(); | 383 listener_profile->GetExtensionService()->process_map(); |
323 // If the event is privileged, only send to extension processes. Otherwise, | 384 // If the event is privileged, only send to extension processes. Otherwise, |
324 // it's OK to send to normal renderers (e.g., for content scripts). | 385 // it's OK to send to normal renderers (e.g., for content scripts). |
325 if (ExtensionAPI::GetSharedInstance()->IsPrivileged(event->event_name) && | 386 if (ExtensionAPI::GetSharedInstance()->IsPrivileged(event->event_name) && |
326 !process_map->Contains(extension->id(), listener.process->GetID())) { | 387 !process_map->Contains(extension->id(), process->GetID())) { |
327 return; | 388 return; |
328 } | 389 } |
329 | 390 |
330 const std::string* event_args; | 391 const std::string* event_args; |
331 if (!CanDispatchEventToProfile(listener_profile, extension, | 392 if (!CanDispatchEventToProfile(listener_profile, extension, |
332 event, &event_args)) | 393 event, &event_args)) |
333 return; | 394 return; |
334 | 395 |
335 DispatchEvent(listener.process, listener.extension_id, | 396 DispatchEvent(process, extension_id, |
336 event->event_name, *event_args, | 397 event->event_name, *event_args, |
337 event->event_url, event->user_gesture); | 398 event->event_url, event->user_gesture, |
399 event->info); | |
338 IncrementInFlightEvents(listener_profile, extension); | 400 IncrementInFlightEvents(listener_profile, extension); |
339 } | 401 } |
340 | 402 |
341 bool ExtensionEventRouter::CanDispatchEventToProfile( | 403 bool ExtensionEventRouter::CanDispatchEventToProfile( |
342 Profile* profile, | 404 Profile* profile, |
343 const Extension* extension, | 405 const Extension* extension, |
344 const linked_ptr<ExtensionEvent>& event, | 406 const linked_ptr<ExtensionEvent>& event, |
345 const std::string** event_args) { | 407 const std::string** event_args) { |
346 *event_args = &event->event_args; | 408 *event_args = &event->event_args; |
347 | 409 |
348 // Is this event from a different profile than the renderer (ie, an | 410 // Is this event from a different profile than the renderer (ie, an |
349 // incognito tab event sent to a normal process, or vice versa). | 411 // incognito tab event sent to a normal process, or vice versa). |
350 bool cross_incognito = event->restrict_to_profile && | 412 bool cross_incognito = event->restrict_to_profile && |
351 profile != event->restrict_to_profile; | 413 profile != event->restrict_to_profile; |
352 if (cross_incognito && | 414 if (cross_incognito && |
353 !profile->GetExtensionService()->CanCrossIncognito(extension)) { | 415 !profile->GetExtensionService()->CanCrossIncognito(extension)) { |
354 if (event->cross_incognito_args.empty()) | 416 if (event->cross_incognito_args.empty()) |
355 return false; | 417 return false; |
356 // Send the event with different arguments to extensions that can't | 418 // Send the event with different arguments to extensions that can't |
357 // cross incognito. | 419 // cross incognito. |
358 *event_args = &event->cross_incognito_args; | 420 *event_args = &event->cross_incognito_args; |
359 } | 421 } |
360 | 422 |
361 return true; | 423 return true; |
362 } | 424 } |
363 | 425 |
364 void ExtensionEventRouter::LoadLazyBackgroundPagesForEvent( | |
365 const std::string& extension_id, | |
366 const linked_ptr<ExtensionEvent>& event) { | |
367 ExtensionService* service = profile_->GetExtensionService(); | |
368 | |
369 ListenerMap::iterator it = lazy_listeners_.find(event->event_name); | |
370 if (it == lazy_listeners_.end()) | |
371 return; | |
372 | |
373 std::set<ListenerProcess>& listeners = it->second; | |
374 for (std::set<ListenerProcess>::iterator listener = listeners.begin(); | |
375 listener != listeners.end(); ++listener) { | |
376 if (!extension_id.empty() && extension_id != listener->extension_id) | |
377 continue; | |
378 | |
379 // Check both the original and the incognito profile to see if we | |
380 // should load a lazy bg page to handle the event. The latter case | |
381 // occurs in the case of split-mode extensions. | |
382 const Extension* extension = service->extensions()->GetByID( | |
383 listener->extension_id); | |
384 if (extension) { | |
385 MaybeLoadLazyBackgroundPage(profile_, extension, event); | |
386 if (profile_->HasOffTheRecordProfile() && | |
387 extension->incognito_split_mode()) { | |
388 MaybeLoadLazyBackgroundPage( | |
389 profile_->GetOffTheRecordProfile(), extension, event); | |
390 } | |
391 } | |
392 } | |
393 } | |
394 | |
395 void ExtensionEventRouter::MaybeLoadLazyBackgroundPage( | 426 void ExtensionEventRouter::MaybeLoadLazyBackgroundPage( |
396 Profile* profile, | 427 Profile* profile, |
397 const Extension* extension, | 428 const Extension* extension, |
398 const linked_ptr<ExtensionEvent>& event) { | 429 const linked_ptr<ExtensionEvent>& event, |
430 const std::string& extension_id) { | |
399 const std::string* event_args; | 431 const std::string* event_args; |
400 if (!CanDispatchEventToProfile(profile, extension, event, &event_args)) | 432 if (!CanDispatchEventToProfile(profile, extension, event, &event_args)) |
401 return; | 433 return; |
402 | 434 |
403 extensions::LazyBackgroundTaskQueue* queue = | 435 extensions::LazyBackgroundTaskQueue* queue = |
404 ExtensionSystem::Get(profile)->lazy_background_task_queue(); | 436 ExtensionSystem::Get(profile)->lazy_background_task_queue(); |
405 if (queue->ShouldEnqueueTask(profile, extension)) { | 437 if (queue->ShouldEnqueueTask(profile, extension)) { |
406 queue->AddPendingTask( | 438 queue->AddPendingTask( |
407 profile, extension->id(), | 439 profile, extension->id(), |
408 base::Bind(&ExtensionEventRouter::DispatchPendingEvent, | 440 base::Bind(&ExtensionEventRouter::DispatchPendingEvent, |
(...skipping 20 matching lines...) Expand all Loading... | |
429 ExtensionSystem::Get(profile)->process_manager(); | 461 ExtensionSystem::Get(profile)->process_manager(); |
430 ExtensionHost* host = pm->GetBackgroundHostForExtension(extension_id); | 462 ExtensionHost* host = pm->GetBackgroundHostForExtension(extension_id); |
431 // The event ACK is routed to the background host, so this should never be | 463 // The event ACK is routed to the background host, so this should never be |
432 // NULL. | 464 // NULL. |
433 CHECK(host); | 465 CHECK(host); |
434 CHECK(host->extension()->has_lazy_background_page()); | 466 CHECK(host->extension()->has_lazy_background_page()); |
435 pm->DecrementLazyKeepaliveCount(host->extension()); | 467 pm->DecrementLazyKeepaliveCount(host->extension()); |
436 } | 468 } |
437 | 469 |
438 void ExtensionEventRouter::DispatchPendingEvent( | 470 void ExtensionEventRouter::DispatchPendingEvent( |
439 const linked_ptr<ExtensionEvent>& event, ExtensionHost* host) { | 471 const linked_ptr<ExtensionEvent>& event, |
472 ExtensionHost* host) { | |
440 if (!host) | 473 if (!host) |
441 return; | 474 return; |
442 | 475 |
443 ListenerProcess listener(host->render_process_host(), | 476 if (listeners_.HasProcessListener(host->render_process_host(), |
444 host->extension()->id()); | 477 host->extension()->id())) |
445 if (listeners_[event->event_name].count(listener) > 0u) | 478 DispatchEventToProcess(host->extension()->id(), |
446 DispatchEventToListener(listener, event); | 479 host->render_process_host(), event); |
447 } | 480 } |
448 | 481 |
449 void ExtensionEventRouter::Observe( | 482 void ExtensionEventRouter::Observe( |
450 int type, | 483 int type, |
451 const content::NotificationSource& source, | 484 const content::NotificationSource& source, |
452 const content::NotificationDetails& details) { | 485 const content::NotificationDetails& details) { |
453 switch (type) { | 486 switch (type) { |
454 case content::NOTIFICATION_RENDERER_PROCESS_TERMINATED: | 487 case content::NOTIFICATION_RENDERER_PROCESS_TERMINATED: |
455 case content::NOTIFICATION_RENDERER_PROCESS_CLOSED: { | 488 case content::NOTIFICATION_RENDERER_PROCESS_CLOSED: { |
456 content::RenderProcessHost* renderer = | 489 content::RenderProcessHost* renderer = |
457 content::Source<content::RenderProcessHost>(source).ptr(); | 490 content::Source<content::RenderProcessHost>(source).ptr(); |
458 // Remove all event listeners associated with this renderer. | 491 // Remove all event listeners associated with this renderer. |
459 for (ListenerMap::iterator it = listeners_.begin(); | 492 listeners_.RemoveListenersForProcess(renderer); |
460 it != listeners_.end(); ) { | |
461 ListenerMap::iterator current_it = it++; | |
462 for (std::set<ListenerProcess>::iterator jt = | |
463 current_it->second.begin(); | |
464 jt != current_it->second.end(); ) { | |
465 std::set<ListenerProcess>::iterator current_jt = jt++; | |
466 if (current_jt->process == renderer) { | |
467 RemoveEventListener(current_it->first, | |
468 current_jt->process, | |
469 current_jt->extension_id); | |
470 } | |
471 } | |
472 } | |
473 break; | 493 break; |
474 } | 494 } |
475 case chrome::NOTIFICATION_EXTENSION_LOADED: { | 495 case chrome::NOTIFICATION_EXTENSION_LOADED: { |
476 // Add all registered lazy listeners to our cache. | 496 // Add all registered lazy listeners to our cache. |
477 const Extension* extension = | 497 const Extension* extension = |
478 content::Details<const Extension>(details).ptr(); | 498 content::Details<const Extension>(details).ptr(); |
499 ExtensionPrefs* prefs = | |
500 profile_->GetExtensionService()->extension_prefs(); | |
479 std::set<std::string> registered_events = | 501 std::set<std::string> registered_events = |
480 profile_->GetExtensionService()->extension_prefs()-> | 502 prefs->GetRegisteredEvents(extension->id()); |
481 GetRegisteredEvents(extension->id()); | 503 const DictionaryValue* filtered_events = |
482 ListenerProcess lazy_listener(NULL, extension->id()); | 504 prefs->GetFilteredEvents(extension->id()); |
483 for (std::set<std::string>::iterator it = registered_events.begin(); | 505 if (filtered_events) |
484 it != registered_events.end(); ++it) { | 506 listeners_.AddLazyListenersFromPreferences(extension->id(), |
485 lazy_listeners_[*it].insert(lazy_listener); | 507 registered_events, |
486 } | 508 *filtered_events); |
487 break; | 509 break; |
488 } | 510 } |
489 case chrome::NOTIFICATION_EXTENSION_UNLOADED: { | 511 case chrome::NOTIFICATION_EXTENSION_UNLOADED: { |
490 // Remove all registered lazy listeners from our cache. | 512 // Remove all registered lazy listeners from our cache. |
491 extensions::UnloadedExtensionInfo* unloaded = | 513 extensions::UnloadedExtensionInfo* unloaded = |
492 content::Details<extensions::UnloadedExtensionInfo>(details).ptr(); | 514 content::Details<extensions::UnloadedExtensionInfo>(details).ptr(); |
493 ListenerProcess lazy_listener(NULL, unloaded->extension->id()); | 515 listeners_.RemoveLazyListenersFor(unloaded->extension->id()); |
494 for (ListenerMap::iterator it = lazy_listeners_.begin(); | |
495 it != lazy_listeners_.end(); ++it) { | |
496 it->second.erase(lazy_listener); | |
497 } | |
498 break; | 516 break; |
499 } | 517 } |
500 case chrome::NOTIFICATION_EXTENSION_INSTALLED: { | 518 case chrome::NOTIFICATION_EXTENSION_INSTALLED: { |
501 // Dispatch the onInstalled event. | 519 // Dispatch the onInstalled event. |
502 const Extension* extension = | 520 const Extension* extension = |
503 content::Details<const Extension>(details).ptr(); | 521 content::Details<const Extension>(details).ptr(); |
504 MessageLoop::current()->PostTask(FROM_HERE, | 522 MessageLoop::current()->PostTask(FROM_HERE, |
505 base::Bind(&extensions::RuntimeEventRouter::DispatchOnInstalledEvent, | 523 base::Bind(&extensions::RuntimeEventRouter::DispatchOnInstalledEvent, |
506 profile_, extension->id())); | 524 profile_, extension->id())); |
507 break; | 525 break; |
508 } | 526 } |
509 default: | 527 default: |
510 NOTREACHED(); | 528 NOTREACHED(); |
511 return; | 529 return; |
512 } | 530 } |
513 } | 531 } |
532 | |
533 ExtensionEvent::ExtensionEvent( | |
534 const std::string& event_name, | |
535 const std::string& event_args, | |
536 const GURL& event_url, | |
537 Profile* restrict_to_profile, | |
538 const std::string& cross_incognito_args, | |
539 ExtensionEventRouter::UserGestureState user_gesture, | |
540 const extensions::EventFilteringInfo& info) | |
541 : event_name(event_name), | |
542 event_args(event_args), | |
543 event_url(event_url), | |
544 restrict_to_profile(restrict_to_profile), | |
545 cross_incognito_args(cross_incognito_args), | |
546 user_gesture(user_gesture), | |
547 info(info) { | |
548 } | |
549 | |
550 ExtensionEvent::~ExtensionEvent() { | |
551 } | |
OLD | NEW |