chrome/browser/extensions/api/messaging/extension_message_port.cc

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