src/content/browser/browser_plugin/browser_plugin_message_filter.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 "content/browser/browser_plugin/browser_plugin_message_filter.h"
   6
   7 #include "base/supports_user_data.h"
   8 #include "content/browser/browser_plugin/browser_plugin_guest.h"
   9 #include "content/browser/browser_plugin/browser_plugin_guest_manager.h"
  10 #include "content/browser/web_contents/web_contents_impl.h"
  11 #include "content/common/browser_plugin/browser_plugin_constants.h"
  12 #include "content/common/browser_plugin/browser_plugin_messages.h"
  13 #include "content/common/view_messages.h"
  14 #include "content/public/browser/browser_context.h"
  15 #include "content/public/browser/browser_thread.h"
  16 #include "content/public/browser/render_view_host.h"
  17
  18 namespace content {
  19
  20 BrowserPluginMessageFilter::BrowserPluginMessageFilter(int render_process_id,
  21                                                        bool is_guest)
  22     : render_process_id_(render_process_id),
  23       is_guest_(is_guest) {
  24 }
  25
  26 BrowserPluginMessageFilter::~BrowserPluginMessageFilter() {
  27   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  28 }
  29
  30 bool BrowserPluginMessageFilter::OnMessageReceived(
  31     const IPC::Message& message,
  32     bool* message_was_ok) {
  33   // Any message requested by a BrowserPluginGuest should be routed through
  34   // a BrowserPluginGuestManager.
  35   if (BrowserPluginGuest::ShouldForwardToBrowserPluginGuest(message)) {
  36     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  37     BrowserPluginGuestManager* guest_manager = GetBrowserPluginGuestManager();
  38     if (guest_manager)
  39       guest_manager->OnMessageReceived(message, render_process_id_);
  40     // We always swallow messages destined for BrowserPluginGuestManager because
  41     // we're on the UI thread and fallback code is expected to be run on the IO
  42     // thread.
  43     return true;
  44   }
  45   return false;
  46 }
  47
  48 void BrowserPluginMessageFilter::OnDestruct() const {
  49   BrowserThread::DeleteOnIOThread::Destruct(this);
  50 }
  51
  52 void BrowserPluginMessageFilter::OverrideThreadForMessage(
  53     const IPC::Message& message, BrowserThread::ID* thread) {
  54   if (BrowserPluginGuest::ShouldForwardToBrowserPluginGuest(message))
  55     *thread = BrowserThread::UI;
  56 }
  57
  58 BrowserPluginGuestManager*
  59     BrowserPluginMessageFilter::GetBrowserPluginGuestManager() {
  60   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  61   RenderProcessHostImpl* host = static_cast<RenderProcessHostImpl*>(
  62       RenderProcessHost::FromID(render_process_id_));
  63   if (!host)
  64     return NULL;
  65
  66   BrowserContext* browser_context = host->GetBrowserContext();
  67   return static_cast<BrowserPluginGuestManager*>(
  68       browser_context->GetUserData(
  69           browser_plugin::kBrowserPluginGuestManagerKeyName));
  70 }
  71
  72 } // namespace content