[Chromoting] The desktop process now creates a pre-connected pipe and passes (with some help of the…

remoting/host/desktop_session_win.cc

Index: remoting/host/desktop_session_win.cc
diff --git a/remoting/host/desktop_session_win.cc b/remoting/host/desktop_session_win.cc
index 897a87d483656b502b05cd9ecd0ee305621785da..8472e127160aad5e375fc2f7570311ac210a5b40 100644
--- a/remoting/host/desktop_session_win.cc
+++ b/remoting/host/desktop_session_win.cc
@@ -5,12 +5,16 @@
 #include "remoting/host/desktop_session_win.h"
 
 #include "base/path_service.h"
-#include "base/single_thread_task_runner.h"
+#include "ipc/ipc_message_macros.h"
+#include "remoting/base/auto_thread_task_runner.h"
+#include "remoting/host/chromoting_messages.h"
 #include "remoting/host/daemon_process.h"
 #include "remoting/host/win/worker_process_launcher.h"
 #include "remoting/host/win/wts_console_monitor.h"
 #include "remoting/host/win/wts_session_process_delegate.h"
 
+using base::win::ScopedHandle;
+
 namespace {
 
 const FilePath::CharType kDesktopBinaryName[] =
@@ -25,8 +29,8 @@ const char kDaemonIpcSecurityDescriptor[] = "O:SYG:SYD:(A;;GA;;;SY)";
 namespace remoting {
 
 DesktopSessionWin::DesktopSessionWin(
-    scoped_refptr<base::SingleThreadTaskRunner> main_task_runner,
-    scoped_refptr<base::SingleThreadTaskRunner> io_task_runner,
+    scoped_refptr<AutoThreadTaskRunner> main_task_runner,
+    scoped_refptr<AutoThreadTaskRunner> io_task_runner,
     DaemonProcess* daemon_process,
     int id,
     WtsConsoleMonitor* monitor)
@@ -48,14 +52,29 @@ DesktopSessionWin::~DesktopSessionWin() {
 
 void DesktopSessionWin::OnChannelConnected(int32 peer_pid) {
   DCHECK(main_task_runner_->BelongsToCurrentThread());
+
+  // Obtain the handle of the desktop process. It will be passed to the network
+  // process so it would be able to duplicate handles of shared memory objects
+  // from the desktop process.
+  desktop_process_.Set(OpenProcess(PROCESS_DUP_HANDLE, false, peer_pid));
+  if (!desktop_process_.IsValid()) {
+    RestartDesktopProcess(FROM_HERE);
+    return;
+  }
+
+  VLOG(1) << "IPC: daemon <- desktop (" << peer_pid << ")";
 }
 
 bool DesktopSessionWin::OnMessageReceived(const IPC::Message& message) {
   DCHECK(main_task_runner_->BelongsToCurrentThread());
 
-  // TODO(alexeypa): Process ChromotingDesktopHostMsg_Initialized messages here.
-  // See http://crbug.com/134694.
-  return false;
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(DesktopSessionWin, message)
+    IPC_MESSAGE_HANDLER(ChromotingDesktopDaemonMsg_DesktopAttached,
+                        OnDesktopSessionAgentAttached)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
 }
 
 void DesktopSessionWin::OnPermanentError() {
@@ -100,4 +119,21 @@ void DesktopSessionWin::OnSessionDetached() {
   launcher_.reset();
 }
 
+void DesktopSessionWin::OnDesktopSessionAgentAttached(
+      IPC::PlatformFileForTransit desktop_pipe) {
+  if (!daemon_process()->OnDesktopSessionAgentAttached(id(),
+                                                       desktop_process_,
+                                                       desktop_pipe)) {
+    RestartDesktopProcess(FROM_HERE);
+  }
+}
+
+void DesktopSessionWin::RestartDesktopProcess(
+    const tracked_objects::Location& location) {
+  DCHECK(main_task_runner_->BelongsToCurrentThread());
+
+  launcher_->Send(new ChromotingDaemonDesktopMsg_Crash(
+      location.function_name(), location.file_name(), location.line_number()));
+}
+
 }  // namespace remoting