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
3 // found in the LICENSE file.
4
5 #include "chrome/browser/extensions/api/messaging/extension_message_port.h"
6
7 #include "chrome/browser/profiles/profile.h"
8 #include "content/public/browser/render_process_host.h"
9 #include "extensions/browser/extension_host.h"
10 #include "extensions/browser/extension_system.h"
11 #include "extensions/browser/process_manager.h"
12 #include "extensions/common/extension_messages.h"
13 #include "extensions/common/manifest_handlers/background_info.h"
14
15 namespace extensions {
16
ExtensionMessagePort(content::RenderProcessHost * process,int routing_id,const std::string & extension_id)17 ExtensionMessagePort::ExtensionMessagePort(content::RenderProcessHost* process,
18 int routing_id,
19 const std::string& extension_id)
20 : process_(process),
21 routing_id_(routing_id),
22 extension_id_(extension_id),
23 background_host_ptr_(NULL) {
24 }
25
DispatchOnConnect(int dest_port_id,const std::string & channel_name,const base::DictionaryValue & source_tab,const std::string & source_extension_id,const std::string & target_extension_id,const GURL & source_url,const std::string & tls_channel_id)26 void ExtensionMessagePort::DispatchOnConnect(
27 int dest_port_id,
28 const std::string& channel_name,
29 const base::DictionaryValue& source_tab,
30 const std::string& source_extension_id,
31 const std::string& target_extension_id,
32 const GURL& source_url,
33 const std::string& tls_channel_id) {
34 ExtensionMsg_ExternalConnectionInfo info;
35 info.target_id = target_extension_id;
36 info.source_id = source_extension_id;
37 info.source_url = source_url;
38 process_->Send(new ExtensionMsg_DispatchOnConnect(
39 routing_id_, dest_port_id, channel_name, source_tab, info,
40 tls_channel_id));
41 }
42
DispatchOnDisconnect(int source_port_id,const std::string & error_message)43 void ExtensionMessagePort::DispatchOnDisconnect(
44 int source_port_id,
45 const std::string& error_message) {
46 process_->Send(new ExtensionMsg_DispatchOnDisconnect(
47 routing_id_, source_port_id, error_message));
48 }
49
DispatchOnMessage(const Message & message,int target_port_id)50 void ExtensionMessagePort::DispatchOnMessage(const Message& message,
51 int target_port_id) {
52 process_->Send(new ExtensionMsg_DeliverMessage(
53 routing_id_, target_port_id, message));
54 }
55
IncrementLazyKeepaliveCount()56 void ExtensionMessagePort::IncrementLazyKeepaliveCount() {
57 Profile* profile =
58 Profile::FromBrowserContext(process_->GetBrowserContext());
59 extensions::ProcessManager* pm =
60 ExtensionSystem::Get(profile)->process_manager();
61 ExtensionHost* host = pm->GetBackgroundHostForExtension(extension_id_);
62 if (host && BackgroundInfo::HasLazyBackgroundPage(host->extension()))
63 pm->IncrementLazyKeepaliveCount(host->extension());
64
65 // Keep track of the background host, so when we decrement, we only do so if
66 // the host hasn't reloaded.
67 background_host_ptr_ = host;
68 }
69
DecrementLazyKeepaliveCount()70 void ExtensionMessagePort::DecrementLazyKeepaliveCount() {
71 Profile* profile =
72 Profile::FromBrowserContext(process_->GetBrowserContext());
73 extensions::ProcessManager* pm =
74 ExtensionSystem::Get(profile)->process_manager();
75 ExtensionHost* host = pm->GetBackgroundHostForExtension(extension_id_);
76 if (host && host == background_host_ptr_)
77 pm->DecrementLazyKeepaliveCount(host->extension());
78 }
79
GetRenderProcessHost()80 content::RenderProcessHost* ExtensionMessagePort::GetRenderProcessHost() {
81 return process_;
82 }
83
84 } // namespace extensions
85