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/api/messaging/extension_message_port.h" | 5 #include "chrome/browser/extensions/api/messaging/extension_message_port.h" |
6 | 6 |
| 7 #include "base/memory/ptr_util.h" |
7 #include "base/scoped_observer.h" | 8 #include "base/scoped_observer.h" |
8 #include "chrome/browser/profiles/profile.h" | 9 #include "chrome/browser/profiles/profile.h" |
9 #include "content/public/browser/interstitial_page.h" | 10 #include "content/public/browser/interstitial_page.h" |
10 #include "content/public/browser/navigation_details.h" | 11 #include "content/public/browser/navigation_details.h" |
11 #include "content/public/browser/render_frame_host.h" | 12 #include "content/public/browser/render_frame_host.h" |
12 #include "content/public/browser/render_process_host.h" | 13 #include "content/public/browser/render_process_host.h" |
13 #include "content/public/browser/web_contents.h" | 14 #include "content/public/browser/web_contents.h" |
14 #include "content/public/browser/web_contents_observer.h" | 15 #include "content/public/browser/web_contents_observer.h" |
15 #include "extensions/browser/extension_host.h" | 16 #include "extensions/browser/extension_host.h" |
16 #include "extensions/browser/process_manager.h" | 17 #include "extensions/browser/process_manager.h" |
(...skipping 156 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
173 bool ExtensionMessagePort::HasFrame(content::RenderFrameHost* rfh) const { | 174 bool ExtensionMessagePort::HasFrame(content::RenderFrameHost* rfh) const { |
174 return frames_.find(rfh) != frames_.end(); | 175 return frames_.find(rfh) != frames_.end(); |
175 } | 176 } |
176 | 177 |
177 bool ExtensionMessagePort::IsValidPort() { | 178 bool ExtensionMessagePort::IsValidPort() { |
178 return !frames_.empty(); | 179 return !frames_.empty(); |
179 } | 180 } |
180 | 181 |
181 void ExtensionMessagePort::DispatchOnConnect( | 182 void ExtensionMessagePort::DispatchOnConnect( |
182 const std::string& channel_name, | 183 const std::string& channel_name, |
183 scoped_ptr<base::DictionaryValue> source_tab, | 184 std::unique_ptr<base::DictionaryValue> source_tab, |
184 int source_frame_id, | 185 int source_frame_id, |
185 int guest_process_id, | 186 int guest_process_id, |
186 int guest_render_frame_routing_id, | 187 int guest_render_frame_routing_id, |
187 const std::string& source_extension_id, | 188 const std::string& source_extension_id, |
188 const std::string& target_extension_id, | 189 const std::string& target_extension_id, |
189 const GURL& source_url, | 190 const GURL& source_url, |
190 const std::string& tls_channel_id) { | 191 const std::string& tls_channel_id) { |
191 ExtensionMsg_TabConnectionInfo source; | 192 ExtensionMsg_TabConnectionInfo source; |
192 if (source_tab) | 193 if (source_tab) |
193 source.tab.Swap(source_tab.get()); | 194 source.tab.Swap(source_tab.get()); |
194 source.frame_id = source_frame_id; | 195 source.frame_id = source_frame_id; |
195 | 196 |
196 ExtensionMsg_ExternalConnectionInfo info; | 197 ExtensionMsg_ExternalConnectionInfo info; |
197 info.target_id = target_extension_id; | 198 info.target_id = target_extension_id; |
198 info.source_id = source_extension_id; | 199 info.source_id = source_extension_id; |
199 info.source_url = source_url; | 200 info.source_url = source_url; |
200 info.guest_process_id = guest_process_id; | 201 info.guest_process_id = guest_process_id; |
201 info.guest_render_frame_routing_id = guest_render_frame_routing_id; | 202 info.guest_render_frame_routing_id = guest_render_frame_routing_id; |
202 | 203 |
203 SendToPort(make_scoped_ptr(new ExtensionMsg_DispatchOnConnect( | 204 SendToPort(base::WrapUnique(new ExtensionMsg_DispatchOnConnect( |
204 MSG_ROUTING_NONE, port_id_, channel_name, source, info, tls_channel_id))); | 205 MSG_ROUTING_NONE, port_id_, channel_name, source, info, tls_channel_id))); |
205 } | 206 } |
206 | 207 |
207 void ExtensionMessagePort::DispatchOnDisconnect( | 208 void ExtensionMessagePort::DispatchOnDisconnect( |
208 const std::string& error_message) { | 209 const std::string& error_message) { |
209 SendToPort(make_scoped_ptr(new ExtensionMsg_DispatchOnDisconnect( | 210 SendToPort(base::WrapUnique(new ExtensionMsg_DispatchOnDisconnect( |
210 MSG_ROUTING_NONE, port_id_, error_message))); | 211 MSG_ROUTING_NONE, port_id_, error_message))); |
211 } | 212 } |
212 | 213 |
213 void ExtensionMessagePort::DispatchOnMessage(const Message& message) { | 214 void ExtensionMessagePort::DispatchOnMessage(const Message& message) { |
214 SendToPort(make_scoped_ptr(new ExtensionMsg_DeliverMessage( | 215 SendToPort(base::WrapUnique( |
215 MSG_ROUTING_NONE, port_id_, message))); | 216 new ExtensionMsg_DeliverMessage(MSG_ROUTING_NONE, port_id_, message))); |
216 } | 217 } |
217 | 218 |
218 void ExtensionMessagePort::IncrementLazyKeepaliveCount() { | 219 void ExtensionMessagePort::IncrementLazyKeepaliveCount() { |
219 ProcessManager* pm = ProcessManager::Get(browser_context_); | 220 ProcessManager* pm = ProcessManager::Get(browser_context_); |
220 ExtensionHost* host = pm->GetBackgroundHostForExtension(extension_id_); | 221 ExtensionHost* host = pm->GetBackgroundHostForExtension(extension_id_); |
221 if (host && BackgroundInfo::HasLazyBackgroundPage(host->extension())) | 222 if (host && BackgroundInfo::HasLazyBackgroundPage(host->extension())) |
222 pm->IncrementLazyKeepaliveCount(host->extension()); | 223 pm->IncrementLazyKeepaliveCount(host->extension()); |
223 | 224 |
224 // Keep track of the background host, so when we decrement, we only do so if | 225 // Keep track of the background host, so when we decrement, we only do so if |
225 // the host hasn't reloaded. | 226 // the host hasn't reloaded. |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
269 // is not triggered for |rfh|. | 270 // is not triggered for |rfh|. |
270 if (rfh->IsRenderFrameLive()) | 271 if (rfh->IsRenderFrameLive()) |
271 frames_.insert(rfh); | 272 frames_.insert(rfh); |
272 } | 273 } |
273 | 274 |
274 void ExtensionMessagePort::UnregisterFrame(content::RenderFrameHost* rfh) { | 275 void ExtensionMessagePort::UnregisterFrame(content::RenderFrameHost* rfh) { |
275 if (frames_.erase(rfh) != 0 && frames_.empty()) | 276 if (frames_.erase(rfh) != 0 && frames_.empty()) |
276 CloseChannel(); | 277 CloseChannel(); |
277 } | 278 } |
278 | 279 |
279 void ExtensionMessagePort::SendToPort(scoped_ptr<IPC::Message> msg) { | 280 void ExtensionMessagePort::SendToPort(std::unique_ptr<IPC::Message> msg) { |
280 DCHECK_GT(frames_.size(), 0UL); | 281 DCHECK_GT(frames_.size(), 0UL); |
281 if (extension_process_) { | 282 if (extension_process_) { |
282 // All extension frames reside in the same process, so we can just send a | 283 // All extension frames reside in the same process, so we can just send a |
283 // single IPC message to the extension process as an optimization. | 284 // single IPC message to the extension process as an optimization. |
284 // The frame tracking is then only used to make sure that the port gets | 285 // The frame tracking is then only used to make sure that the port gets |
285 // closed when all frames have closed / reloaded. | 286 // closed when all frames have closed / reloaded. |
286 msg->set_routing_id(MSG_ROUTING_CONTROL); | 287 msg->set_routing_id(MSG_ROUTING_CONTROL); |
287 extension_process_->Send(msg.release()); | 288 extension_process_->Send(msg.release()); |
288 return; | 289 return; |
289 } | 290 } |
290 for (content::RenderFrameHost* rfh : frames_) { | 291 for (content::RenderFrameHost* rfh : frames_) { |
291 IPC::Message* msg_copy = new IPC::Message(*msg.get()); | 292 IPC::Message* msg_copy = new IPC::Message(*msg.get()); |
292 msg_copy->set_routing_id(rfh->GetRoutingID()); | 293 msg_copy->set_routing_id(rfh->GetRoutingID()); |
293 rfh->Send(msg_copy); | 294 rfh->Send(msg_copy); |
294 } | 295 } |
295 } | 296 } |
296 | 297 |
297 } // namespace extensions | 298 } // namespace extensions |
OLD | NEW |