| 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/win/host_service.h" | 8 #include "remoting/host/win/host_service.h" |
| 9 | 9 |
| 10 #include <windows.h> | 10 #include <windows.h> |
| 11 #include <wtsapi32.h> | 11 #include <wtsapi32.h> |
| 12 #include <stdio.h> | 12 #include <stdio.h> |
| 13 | 13 |
| 14 #include "base/at_exit.h" | 14 #include "base/at_exit.h" |
| 15 #include "base/base_paths.h" | 15 #include "base/base_paths.h" |
| 16 #include "base/bind.h" | 16 #include "base/bind.h" |
| 17 #include "base/command_line.h" | 17 #include "base/command_line.h" |
| 18 #include "base/file_util.h" | 18 #include "base/file_path.h" |
| 19 #include "base/logging.h" | 19 #include "base/logging.h" |
| 20 #include "base/message_loop.h" | 20 #include "base/message_loop.h" |
| 21 #include "base/path_service.h" | |
| 22 #include "base/single_thread_task_runner.h" | 21 #include "base/single_thread_task_runner.h" |
| 23 #include "base/stringprintf.h" | 22 #include "base/stringprintf.h" |
| 24 #include "base/threading/thread.h" | 23 #include "base/threading/thread.h" |
| 25 #include "base/utf_string_conversions.h" | 24 #include "base/utf_string_conversions.h" |
| 26 #include "base/win/wrapped_window_proc.h" | 25 #include "base/win/wrapped_window_proc.h" |
| 27 #include "remoting/base/breakpad.h" | 26 #include "remoting/base/breakpad.h" |
| 28 #include "remoting/base/scoped_sc_handle_win.h" | 27 #include "remoting/base/scoped_sc_handle_win.h" |
| 29 #include "remoting/base/stoppable.h" | 28 #include "remoting/base/stoppable.h" |
| 30 #include "remoting/host/branding.h" | 29 #include "remoting/host/branding.h" |
| 31 #include "remoting/host/usage_stats_consent.h" | 30 #include "remoting/host/usage_stats_consent.h" |
| (...skipping 11 matching lines...) Expand all Loading... |
| 43 const wchar_t kSessionNotificationWindowClass[] = | 42 const wchar_t kSessionNotificationWindowClass[] = |
| 44 L"Chromoting_SessionNotificationWindow"; | 43 L"Chromoting_SessionNotificationWindow"; |
| 45 | 44 |
| 46 // Command line actions and switches: | 45 // Command line actions and switches: |
| 47 // "run" sumply runs the service as usual. | 46 // "run" sumply runs the service as usual. |
| 48 const wchar_t kRunActionName[] = L"run"; | 47 const wchar_t kRunActionName[] = L"run"; |
| 49 | 48 |
| 50 // "--console" runs the service interactively for debugging purposes. | 49 // "--console" runs the service interactively for debugging purposes. |
| 51 const char kConsoleSwitchName[] = "console"; | 50 const char kConsoleSwitchName[] = "console"; |
| 52 | 51 |
| 53 // "--host-binary" specifies the host binary to run in console session. | |
| 54 const char kHostBinarySwitchName[] = "host-binary"; | |
| 55 | |
| 56 // "--help" or "--?" prints the usage message. | 52 // "--help" or "--?" prints the usage message. |
| 57 const char kHelpSwitchName[] = "help"; | 53 const char kHelpSwitchName[] = "help"; |
| 58 const char kQuestionSwitchName[] = "?"; | 54 const char kQuestionSwitchName[] = "?"; |
| 59 | 55 |
| 60 const char kUsageMessage[] = | 56 const char kUsageMessage[] = |
| 61 "\n" | 57 "\n" |
| 62 "Usage: %s [action] [options]\n" | 58 "Usage: %s [action] [options]\n" |
| 63 "\n" | 59 "\n" |
| 64 "Actions:\n" | 60 "Actions:\n" |
| 65 " run - Run the service (default if no action was specified).\n" | 61 " run - Run the service (default if no action was specified).\n" |
| 66 "\n" | 62 "\n" |
| 67 "Options:\n" | 63 "Options:\n" |
| 68 " --console - Run the service interactively for debugging purposes.\n" | 64 " --console - Run the service interactively for debugging purposes.\n" |
| 69 " --host-binary - Specifies the host binary to run.\n" | |
| 70 " --help, --? - Print this message.\n"; | 65 " --help, --? - Print this message.\n"; |
| 71 | 66 |
| 72 // Exit codes: | 67 // Exit codes: |
| 73 const int kSuccessExitCode = 0; | 68 const int kSuccessExitCode = 0; |
| 74 const int kUsageExitCode = 1; | 69 const int kUsageExitCode = 1; |
| 75 const int kErrorExitCode = 2; | 70 const int kErrorExitCode = 2; |
| 76 | 71 |
| 77 void usage(const char* program_name) { | 72 void usage(const char* program_name) { |
| 78 fprintf(stderr, kUsageMessage, program_name); | 73 fprintf(stderr, kUsageMessage, program_name); |
| 79 } | 74 } |
| (...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 166 LOG(ERROR) << "Invalid command line: more than one action requested."; | 161 LOG(ERROR) << "Invalid command line: more than one action requested."; |
| 167 return false; | 162 return false; |
| 168 } | 163 } |
| 169 if (args[0] != kRunActionName) { | 164 if (args[0] != kRunActionName) { |
| 170 LOG(ERROR) << "Invalid command line: invalid action specified: " | 165 LOG(ERROR) << "Invalid command line: invalid action specified: " |
| 171 << args[0]; | 166 << args[0]; |
| 172 return false; | 167 return false; |
| 173 } | 168 } |
| 174 } | 169 } |
| 175 | 170 |
| 176 if (command_line->HasSwitch(kHostBinarySwitchName)) { | |
| 177 host_binary_ = command_line->GetSwitchValuePath(kHostBinarySwitchName); | |
| 178 } else { | |
| 179 LOG(ERROR) << "Invalid command line: --" << kHostBinarySwitchName | |
| 180 << " is required."; | |
| 181 return false; | |
| 182 } | |
| 183 | |
| 184 // Run interactively if needed. | 171 // Run interactively if needed. |
| 185 if (run_routine_ == &HostService::RunAsService && | 172 if (run_routine_ == &HostService::RunAsService && |
| 186 command_line->HasSwitch(kConsoleSwitchName)) { | 173 command_line->HasSwitch(kConsoleSwitchName)) { |
| 187 run_routine_ = &HostService::RunInConsole; | 174 run_routine_ = &HostService::RunInConsole; |
| 188 } | 175 } |
| 189 | 176 |
| 190 return true; | 177 return true; |
| 191 } | 178 } |
| 192 | 179 |
| 193 int HostService::Run() { | 180 int HostService::Run() { |
| 194 return (this->*run_routine_)(); | 181 return (this->*run_routine_)(); |
| 195 } | 182 } |
| 196 | 183 |
| 197 void HostService::RunMessageLoop(MessageLoop* message_loop) { | 184 void HostService::RunMessageLoop(MessageLoop* message_loop) { |
| 198 // Launch the I/O thread. | 185 // Launch the I/O thread. |
| 199 base::Thread io_thread(kIoThreadName); | 186 base::Thread io_thread(kIoThreadName); |
| 200 base::Thread::Options io_thread_options(MessageLoop::TYPE_IO, 0); | 187 base::Thread::Options io_thread_options(MessageLoop::TYPE_IO, 0); |
| 201 if (!io_thread.StartWithOptions(io_thread_options)) { | 188 if (!io_thread.StartWithOptions(io_thread_options)) { |
| 202 LOG(FATAL) << "Failed to start the I/O thread"; | 189 LOG(FATAL) << "Failed to start the I/O thread"; |
| 203 stopped_event_.Signal(); | 190 stopped_event_.Signal(); |
| 204 return; | 191 return; |
| 205 } | 192 } |
| 206 | 193 |
| 207 // Create the session process launcher. | 194 // Create the session process launcher. |
| 208 launcher_.reset(new WtsSessionProcessLauncher( | 195 launcher_.reset(new WtsSessionProcessLauncher( |
| 209 base::Bind(&HostService::OnLauncherShutdown, base::Unretained(this)), | 196 base::Bind(&HostService::OnLauncherShutdown, base::Unretained(this)), |
| 210 this, | 197 this, |
| 211 host_binary_, | |
| 212 main_task_runner_, | 198 main_task_runner_, |
| 213 io_thread.message_loop_proxy())); | 199 io_thread.message_loop_proxy())); |
| 214 | 200 |
| 215 // Run the service. | 201 // Run the service. |
| 216 message_loop->Run(); | 202 message_loop->Run(); |
| 217 | 203 |
| 218 // Release the control handler. | 204 // Release the control handler. |
| 219 stopped_event_.Signal(); | 205 stopped_event_.Signal(); |
| 220 } | 206 } |
| 221 | 207 |
| (...skipping 213 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 435 } | 421 } |
| 436 | 422 |
| 437 remoting::HostService* service = remoting::HostService::GetInstance(); | 423 remoting::HostService* service = remoting::HostService::GetInstance(); |
| 438 if (!service->InitWithCommandLine(command_line)) { | 424 if (!service->InitWithCommandLine(command_line)) { |
| 439 usage(argv[0]); | 425 usage(argv[0]); |
| 440 return kUsageExitCode; | 426 return kUsageExitCode; |
| 441 } | 427 } |
| 442 | 428 |
| 443 return service->Run(); | 429 return service->Run(); |
| 444 } | 430 } |
| OLD | NEW |