content/utility/utility_thread_impl.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/utility/utility_thread_impl.h"
   6
   7 #include <stddef.h>
   8
   9 #include "base/command_line.h"
  10 #include "base/memory/scoped_vector.h"
  11 #include "content/child/blink_platform_impl.h"
  12 #include "content/child/child_process.h"
  13 #include "content/common/child_process_messages.h"
  14 #include "content/common/utility_messages.h"
  15 #include "content/public/common/content_switches.h"
  16 #include "content/public/utility/content_utility_client.h"
  17 #include "content/utility/utility_blink_platform_impl.h"
  18 #include "content/utility/utility_process_control_impl.h"
  19 #include "ipc/ipc_sync_channel.h"
  20 #include "third_party/WebKit/public/web/WebKit.h"
  21
  22 #if defined(OS_POSIX) && defined(ENABLE_PLUGINS)
  23 #include "base/files/file_path.h"
  24 #include "content/common/plugin_list.h"
  25 #endif
  26
  27 namespace content {
  28
  29 namespace {
  30
  31 template<typename SRC, typename DEST>
  32 void ConvertVector(const SRC& src, DEST* dest) {
  33   dest->reserve(src.size());
  34   for (typename SRC::const_iterator i = src.begin(); i != src.end(); ++i)
  35     dest->push_back(typename DEST::value_type(*i));
  36 }
  37
  38 }  // namespace
  39
  40 UtilityThreadImpl::UtilityThreadImpl()
  41     : ChildThreadImpl(ChildThreadImpl::Options::Builder().Build()) {
  42   Init();
  43 }
  44
  45 UtilityThreadImpl::UtilityThreadImpl(const InProcessChildThreadParams& params)
  46     : ChildThreadImpl(ChildThreadImpl::Options::Builder()
  47                           .InBrowserProcess(params)
  48                           .Build()) {
  49   Init();
  50 }
  51
  52 UtilityThreadImpl::~UtilityThreadImpl() {
  53 }
  54
  55 void UtilityThreadImpl::Shutdown() {
  56   ChildThreadImpl::Shutdown();
  57
  58   if (!IsInBrowserProcess())
  59     blink::shutdown();
  60 }
  61
  62 void UtilityThreadImpl::ReleaseProcessIfNeeded() {
  63   if (batch_mode_)
  64     return;
  65
  66   if (IsInBrowserProcess()) {
  67     // Close the channel to cause UtilityProcessHostImpl to be deleted. We need
  68     // to take a different code path than the multi-process case because that
  69     // depends on the child process going away to close the channel, but that
  70     // can't happen when we're in single process mode.
  71     channel()->Close();
  72   } else {
  73     ChildProcess::current()->ReleaseProcess();
  74   }
  75 }
  76
  77 void UtilityThreadImpl::Init() {
  78   batch_mode_ = false;
  79   ChildProcess::current()->AddRefProcess();
  80   if (!IsInBrowserProcess()) {
  81     // We can only initialize WebKit on one thread, and in single process mode
  82     // we run the utility thread on separate thread. This means that if any code
  83     // needs WebKit initialized in the utility process, they need to have
  84     // another path to support single process mode.
  85     blink_platform_impl_.reset(new UtilityBlinkPlatformImpl);
  86     blink::initialize(blink_platform_impl_.get());
  87   }
  88   GetContentClient()->utility()->UtilityThreadStarted();
  89
  90   process_control_.reset(new UtilityProcessControlImpl);
  91   service_registry()->AddService(base::Bind(
  92       &UtilityThreadImpl::BindProcessControlRequest, base::Unretained(this)));
  93
  94   GetContentClient()->utility()->RegisterMojoServices(service_registry());
  95 }
  96
  97 bool UtilityThreadImpl::OnControlMessageReceived(const IPC::Message& msg) {
  98   if (GetContentClient()->utility()->OnMessageReceived(msg))
  99     return true;
 100
 101   bool handled = true;
 102   IPC_BEGIN_MESSAGE_MAP(UtilityThreadImpl, msg)
 103     IPC_MESSAGE_HANDLER(UtilityMsg_BatchMode_Started, OnBatchModeStarted)
 104     IPC_MESSAGE_HANDLER(UtilityMsg_BatchMode_Finished, OnBatchModeFinished)
 105 #if defined(OS_POSIX) && defined(ENABLE_PLUGINS)
 106     IPC_MESSAGE_HANDLER(UtilityMsg_LoadPlugins, OnLoadPlugins)
 107 #endif
 108     IPC_MESSAGE_UNHANDLED(handled = false)
 109   IPC_END_MESSAGE_MAP()
 110   return handled;
 111 }
 112
 113 void UtilityThreadImpl::OnBatchModeStarted() {
 114   batch_mode_ = true;
 115 }
 116
 117 void UtilityThreadImpl::OnBatchModeFinished() {
 118   batch_mode_ = false;
 119   ReleaseProcessIfNeeded();
 120 }
 121
 122 #if defined(OS_POSIX) && defined(ENABLE_PLUGINS)
 123 void UtilityThreadImpl::OnLoadPlugins(
 124     const std::vector<base::FilePath>& plugin_paths) {
 125   PluginList* plugin_list = PluginList::Singleton();
 126
 127   std::vector<WebPluginInfo> plugins;
 128   // TODO(bauerb): If we restart loading plugins, we might mess up the logic in
 129   // PluginList::ShouldLoadPlugin due to missing the previously loaded plugins
 130   // in |plugin_groups|.
 131   for (size_t i = 0; i < plugin_paths.size(); ++i) {
 132     WebPluginInfo plugin;
 133     if (!plugin_list->LoadPluginIntoPluginList(
 134         plugin_paths[i], &plugins, &plugin))
 135       Send(new UtilityHostMsg_LoadPluginFailed(i, plugin_paths[i]));
 136     else
 137       Send(new UtilityHostMsg_LoadedPlugin(i, plugin));
 138   }
 139
 140   ReleaseProcessIfNeeded();
 141 }
 142 #endif
 143
 144 void UtilityThreadImpl::BindProcessControlRequest(
 145     mojo::InterfaceRequest<ProcessControl> request) {
 146   DCHECK(process_control_);
 147   process_control_bindings_.AddBinding(process_control_.get(), request.Pass());
 148 }
 149
 150 }  // namespace content