win8/viewer/metro_viewer_process_host.cc

   1 // Copyright (c) 2013 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 "win8/viewer/metro_viewer_process_host.h"
   6
   7 #include <shlobj.h>
   8
   9 #include "base/command_line.h"
  10 #include "base/file_util.h"
  11 #include "base/files/file_path.h"
  12 #include "base/memory/ref_counted.h"
  13 #include "base/process/process.h"
  14 #include "base/process/process_handle.h"
  15 #include "base/strings/string16.h"
  16 #include "base/synchronization/waitable_event.h"
  17 #include "base/time/time.h"
  18 #include "base/win/scoped_comptr.h"
  19 #include "ipc/ipc_message.h"
  20 #include "ipc/ipc_message_macros.h"
  21 #include "ui/aura/remote_window_tree_host_win.h"
  22 #include "ui/metro_viewer/metro_viewer_messages.h"
  23 #include "win8/viewer/metro_viewer_constants.h"
  24
  25 namespace {
  26
  27 const int kViewerProcessConnectionTimeoutSecs = 60;
  28
  29 }  // namespace
  30
  31 namespace win8 {
  32
  33 MetroViewerProcessHost::InternalMessageFilter::InternalMessageFilter(
  34     MetroViewerProcessHost* owner) : owner_(owner) {
  35 }
  36
  37 void MetroViewerProcessHost::InternalMessageFilter::OnChannelConnected(
  38     int32 /* peer_pid */) {
  39   owner_->NotifyChannelConnected();
  40 }
  41
  42 MetroViewerProcessHost::MetroViewerProcessHost(
  43     base::SingleThreadTaskRunner* ipc_task_runner) {
  44
  45   channel_.reset(new IPC::ChannelProxy(
  46       kMetroViewerIPCChannelName,
  47       IPC::Channel::MODE_NAMED_SERVER,
  48       this,
  49       ipc_task_runner));
  50 }
  51
  52 MetroViewerProcessHost::~MetroViewerProcessHost() {
  53   if (!channel_)
  54     return;
  55
  56   base::ProcessId viewer_process_id = GetViewerProcessId();
  57   channel_->Close();
  58   if (message_filter_) {
  59     // Wait for the viewer process to go away.
  60     if (viewer_process_id != base::kNullProcessId) {
  61       base::ProcessHandle viewer_process = NULL;
  62       base::OpenProcessHandleWithAccess(
  63           viewer_process_id,
  64           PROCESS_QUERY_INFORMATION | SYNCHRONIZE,
  65           &viewer_process);
  66       if (viewer_process) {
  67         ::WaitForSingleObject(viewer_process, INFINITE);
  68         ::CloseHandle(viewer_process);
  69       }
  70     }
  71     channel_->RemoveFilter(message_filter_);
  72   }
  73 }
  74
  75 base::ProcessId MetroViewerProcessHost::GetViewerProcessId() {
  76   if (channel_)
  77     return channel_->peer_pid();
  78   return base::kNullProcessId;
  79 }
  80
  81 bool MetroViewerProcessHost::LaunchViewerAndWaitForConnection(
  82     const base::string16& app_user_model_id) {
  83   DCHECK_EQ(base::kNullProcessId, channel_->peer_pid());
  84
  85   channel_connected_event_.reset(new base::WaitableEvent(false, false));
  86
  87   message_filter_ = new InternalMessageFilter(this);
  88   channel_->AddFilter(message_filter_);
  89
  90   base::win::ScopedComPtr<IApplicationActivationManager> activator;
  91   HRESULT hr = activator.CreateInstance(CLSID_ApplicationActivationManager);
  92   if (SUCCEEDED(hr)) {
  93     DWORD pid = 0;
  94     // Use the "connect" verb to
  95     hr = activator->ActivateApplication(
  96         app_user_model_id.c_str(), kMetroViewerConnectVerb, AO_NONE, &pid);
  97   }
  98
  99   LOG_IF(ERROR, FAILED(hr)) << "Tried and failed to launch Metro Chrome. "
 100                             << "hr=" << std::hex << hr;
 101
 102   // Having launched the viewer process, now we wait for it to connect.
 103   bool success =
 104       channel_connected_event_->TimedWait(base::TimeDelta::FromSeconds(
 105           kViewerProcessConnectionTimeoutSecs));
 106   channel_connected_event_.reset();
 107   return success;
 108 }
 109
 110 bool MetroViewerProcessHost::Send(IPC::Message* msg) {
 111   return channel_->Send(msg);
 112 }
 113
 114 bool MetroViewerProcessHost::OnMessageReceived(
 115     const IPC::Message& message) {
 116   DCHECK(CalledOnValidThread());
 117   bool handled = true;
 118   IPC_BEGIN_MESSAGE_MAP(MetroViewerProcessHost, message)
 119     IPC_MESSAGE_HANDLER(MetroViewerHostMsg_SetTargetSurface, OnSetTargetSurface)
 120     IPC_MESSAGE_HANDLER(MetroViewerHostMsg_OpenURL, OnOpenURL)
 121     IPC_MESSAGE_HANDLER(MetroViewerHostMsg_SearchRequest, OnHandleSearchRequest)
 122     IPC_MESSAGE_HANDLER(MetroViewerHostMsg_WindowSizeChanged,
 123                         OnWindowSizeChanged)
 124     IPC_MESSAGE_UNHANDLED(handled = false)
 125   IPC_END_MESSAGE_MAP()
 126   return handled ? true :
 127       aura::RemoteWindowTreeHostWin::Instance()->OnMessageReceived(message);
 128 }
 129
 130 void MetroViewerProcessHost::NotifyChannelConnected() {
 131   if (channel_connected_event_)
 132     channel_connected_event_->Signal();
 133 }
 134
 135 }  // namespace win8