Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // | 4 // |
| 5 // This file implements the Windows service controlling Me2Me host processes | 5 // This file implements the Windows service controlling Me2Me host processes |
| 6 // running within user sessions. | 6 // running within user sessions. |
| 7 | 7 |
| 8 #include "remoting/host/wts_session_process_launcher_win.h" | 8 #include "remoting/host/wts_session_process_launcher_win.h" |
| 9 | 9 |
| 10 #include <windows.h> | 10 #include <windows.h> |
| 11 #include <sddl.h> | 11 #include <sddl.h> |
| 12 #include <limits> | 12 #include <limits> |
| 13 | 13 |
| 14 #include "base/base_switches.h" | 14 #include "base/base_switches.h" |
| 15 #include "base/bind.h" | 15 #include "base/bind.h" |
| 16 #include "base/bind_helpers.h" | 16 #include "base/bind_helpers.h" |
| 17 #include "base/command_line.h" | 17 #include "base/command_line.h" |
| 18 #include "base/file_path.h" | |
| 19 #include "base/file_util.h" | |
| 18 #include "base/logging.h" | 20 #include "base/logging.h" |
| 19 #include "base/single_thread_task_runner.h" | 21 #include "base/single_thread_task_runner.h" |
| 22 #include "base/path_service.h" | |
| 20 #include "base/process_util.h" | 23 #include "base/process_util.h" |
| 21 #include "base/rand_util.h" | 24 #include "base/rand_util.h" |
| 22 #include "base/stringprintf.h" | 25 #include "base/stringprintf.h" |
| 23 #include "base/win/scoped_handle.h" | 26 #include "base/win/scoped_handle.h" |
| 24 #include "ipc/ipc_channel_proxy.h" | 27 #include "ipc/ipc_channel_proxy.h" |
| 25 #include "ipc/ipc_message.h" | 28 #include "ipc/ipc_message.h" |
| 26 #include "ipc/ipc_message_macros.h" | 29 #include "ipc/ipc_message_macros.h" |
| 27 #include "remoting/host/constants.h" | 30 #include "remoting/host/constants.h" |
| 28 #include "remoting/host/chromoting_messages.h" | 31 #include "remoting/host/chromoting_messages.h" |
| 29 #include "remoting/host/launch_process_in_session_win.h" | 32 #include "remoting/host/launch_process_in_session_win.h" |
| 30 #include "remoting/host/sas_injector.h" | 33 #include "remoting/host/sas_injector.h" |
| 31 #include "remoting/host/wts_console_monitor_win.h" | 34 #include "remoting/host/wts_console_monitor_win.h" |
| 32 | 35 |
| 33 using base::win::ScopedHandle; | 36 using base::win::ScopedHandle; |
| 34 using base::TimeDelta; | 37 using base::TimeDelta; |
| 35 | 38 |
| 36 namespace { | 39 namespace { |
| 37 | 40 |
| 38 // The minimum and maximum delays between attempts to inject host process into | 41 // The minimum and maximum delays between attempts to inject host process into |
| 39 // a session. | 42 // a session. |
| 40 const int kMaxLaunchDelaySeconds = 60; | 43 const int kMaxLaunchDelaySeconds = 60; |
| 41 const int kMinLaunchDelaySeconds = 1; | 44 const int kMinLaunchDelaySeconds = 1; |
| 42 | 45 |
| 46 const FilePath::CharType kMe2meHostBinaryName[] = | |
| 47 FILE_PATH_LITERAL("remoting_me2me_host.exe"); | |
| 48 | |
| 43 // Match the pipe name prefix used by Chrome IPC channels. | 49 // Match the pipe name prefix used by Chrome IPC channels. |
| 44 const wchar_t kChromePipeNamePrefix[] = L"\\\\.\\pipe\\chrome."; | 50 const wchar_t kChromePipeNamePrefix[] = L"\\\\.\\pipe\\chrome."; |
| 45 | 51 |
| 46 // The IPC channel name is passed to the host in the command line. | 52 // The IPC channel name is passed to the host in the command line. |
| 47 const char kChromotingIpcSwitchName[] = "chromoting-ipc"; | 53 const char kChromotingIpcSwitchName[] = "chromoting-ipc"; |
| 48 | 54 |
| 49 // The command line parameters that should be copied from the service's command | 55 // The command line parameters that should be copied from the service's command |
| 50 // line to the host process. | 56 // line to the host process. |
| 51 const char* kCopiedSwitchNames[] = { | 57 const char* kCopiedSwitchNames[] = { |
| 52 "auth-config", "host-config", switches::kV, switches::kVModule }; | 58 "auth-config", "host-config", switches::kV, switches::kVModule }; |
| (...skipping 154 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 207 } // namespace | 213 } // namespace |
| 208 | 214 |
| 209 namespace remoting { | 215 namespace remoting { |
| 210 | 216 |
| 211 // Session id that does not represent any session. | 217 // Session id that does not represent any session. |
| 212 const uint32 kInvalidSessionId = 0xffffffff; | 218 const uint32 kInvalidSessionId = 0xffffffff; |
| 213 | 219 |
| 214 WtsSessionProcessLauncher::WtsSessionProcessLauncher( | 220 WtsSessionProcessLauncher::WtsSessionProcessLauncher( |
| 215 const Shutdownable::Callback& done, | 221 const Shutdownable::Callback& done, |
| 216 WtsConsoleMonitor* monitor, | 222 WtsConsoleMonitor* monitor, |
| 217 const FilePath& host_binary, | |
| 218 scoped_refptr<base::SingleThreadTaskRunner> main_message_loop, | 223 scoped_refptr<base::SingleThreadTaskRunner> main_message_loop, |
| 219 scoped_refptr<base::SingleThreadTaskRunner> ipc_message_loop) | 224 scoped_refptr<base::SingleThreadTaskRunner> ipc_message_loop) |
| 220 : Shutdownable(main_message_loop, done), | 225 : Shutdownable(main_message_loop, done), |
| 221 host_binary_(host_binary), | |
| 222 main_message_loop_(main_message_loop), | 226 main_message_loop_(main_message_loop), |
| 223 ipc_message_loop_(ipc_message_loop), | 227 ipc_message_loop_(ipc_message_loop), |
| 224 monitor_(monitor), | 228 monitor_(monitor), |
| 225 state_(StateDetached) { | 229 state_(StateDetached) { |
| 226 monitor_->AddWtsConsoleObserver(this); | 230 monitor_->AddWtsConsoleObserver(this); |
| 227 } | 231 } |
| 228 | 232 |
| 229 WtsSessionProcessLauncher::~WtsSessionProcessLauncher() { | 233 WtsSessionProcessLauncher::~WtsSessionProcessLauncher() { |
| 230 monitor_->RemoveWtsConsoleObserver(this); | 234 monitor_->RemoveWtsConsoleObserver(this); |
| 231 if (state_ != StateDetached) { | 235 if (state_ != StateDetached) { |
| (...skipping 10 matching lines...) Expand all Loading... | |
| 242 void WtsSessionProcessLauncher::LaunchProcess() { | 246 void WtsSessionProcessLauncher::LaunchProcess() { |
| 243 DCHECK(main_message_loop_->BelongsToCurrentThread()); | 247 DCHECK(main_message_loop_->BelongsToCurrentThread()); |
| 244 DCHECK(state_ == StateStarting); | 248 DCHECK(state_ == StateStarting); |
| 245 DCHECK(!timer_.IsRunning()); | 249 DCHECK(!timer_.IsRunning()); |
| 246 DCHECK(process_.handle() == NULL); | 250 DCHECK(process_.handle() == NULL); |
| 247 DCHECK(process_watcher_.GetWatchedObject() == NULL); | 251 DCHECK(process_watcher_.GetWatchedObject() == NULL); |
| 248 DCHECK(chromoting_channel_.get() == NULL); | 252 DCHECK(chromoting_channel_.get() == NULL); |
| 249 | 253 |
| 250 launch_time_ = base::Time::Now(); | 254 launch_time_ = base::Time::Now(); |
| 251 | 255 |
| 256 // Construct the host binary name. | |
| 257 FilePath exe_path; | |
| 258 if (!PathService::Get(base::FILE_EXE, &exe_path)) { | |
|
simonmorris
2012/07/27 22:08:56
You could use DIR_EXE and avoid the call to DirNam
alexeypa (please no reviews)
2012/07/27 22:35:29
Done.
| |
| 259 LOG(ERROR) << "Failed to get the executable file name."; | |
| 260 Shutdown(); | |
| 261 return; | |
| 262 } | |
| 263 FilePath host_binary = exe_path.DirName().Append(kMe2meHostBinaryName); | |
| 264 | |
| 252 std::wstring channel_name; | 265 std::wstring channel_name; |
| 253 ScopedHandle pipe; | 266 ScopedHandle pipe; |
| 254 if (CreatePipeForIpcChannel(this, &channel_name, &pipe)) { | 267 if (CreatePipeForIpcChannel(this, &channel_name, &pipe)) { |
| 255 // Wrap the pipe into an IPC channel. | 268 // Wrap the pipe into an IPC channel. |
| 256 chromoting_channel_.reset(new IPC::ChannelProxy( | 269 chromoting_channel_.reset(new IPC::ChannelProxy( |
| 257 IPC::ChannelHandle(pipe.Get()), | 270 IPC::ChannelHandle(pipe.Get()), |
| 258 IPC::Channel::MODE_SERVER, | 271 IPC::Channel::MODE_SERVER, |
| 259 this, | 272 this, |
| 260 ipc_message_loop_)); | 273 ipc_message_loop_)); |
| 261 | 274 |
| 262 // Create the host process command line passing the name of the IPC channel | 275 // Create the host process command line passing the name of the IPC channel |
| 263 // to use and copying known switches from the service's command line. | 276 // to use and copying known switches from the service's command line. |
| 264 CommandLine command_line(host_binary_); | 277 CommandLine command_line(host_binary); |
| 265 command_line.AppendSwitchNative(kChromotingIpcSwitchName, channel_name); | 278 command_line.AppendSwitchNative(kChromotingIpcSwitchName, channel_name); |
| 266 command_line.CopySwitchesFrom(*CommandLine::ForCurrentProcess(), | 279 command_line.CopySwitchesFrom(*CommandLine::ForCurrentProcess(), |
| 267 kCopiedSwitchNames, | 280 kCopiedSwitchNames, |
| 268 _countof(kCopiedSwitchNames)); | 281 _countof(kCopiedSwitchNames)); |
| 269 | 282 |
| 270 // Try to launch the process and attach an object watcher to the returned | 283 // Try to launch the process and attach an object watcher to the returned |
| 271 // handle so that we get notified when the process terminates. | 284 // handle so that we get notified when the process terminates. |
| 272 if (LaunchProcessInSession(host_binary_, | 285 if (LaunchProcessInSession(host_binary, |
| 273 command_line.GetCommandLineString(), | 286 command_line.GetCommandLineString(), |
| 274 session_token_, | 287 session_token_, |
| 275 &process_)) { | 288 &process_)) { |
| 276 if (process_watcher_.StartWatching(process_.handle(), this)) { | 289 if (process_watcher_.StartWatching(process_.handle(), this)) { |
| 277 state_ = StateAttached; | 290 state_ = StateAttached; |
| 278 return; | 291 return; |
| 279 } else { | 292 } else { |
| 280 LOG(ERROR) << "Failed to arm the process watcher."; | 293 LOG(ERROR) << "Failed to arm the process watcher."; |
| 281 process_.Terminate(0); | 294 process_.Terminate(0); |
| 282 process_.Close(); | 295 process_.Close(); |
| (...skipping 185 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 468 | 481 |
| 469 void WtsSessionProcessLauncher::DoShutdown() { | 482 void WtsSessionProcessLauncher::DoShutdown() { |
| 470 if (state_ != StateDetached) { | 483 if (state_ != StateDetached) { |
| 471 OnSessionDetached(); | 484 OnSessionDetached(); |
| 472 } | 485 } |
| 473 | 486 |
| 474 CompleteShutdown(); | 487 CompleteShutdown(); |
| 475 } | 488 } |
| 476 | 489 |
| 477 } // namespace remoting | 490 } // namespace remoting |
| OLD | NEW |