remoting/host/desktop_process_main.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 // This file implements the Windows service controlling Me2Me host processes
   6 // running within user sessions.
   7
   8 #include "base/at_exit.h"
   9 #include "base/bind.h"
  10 #include "base/bind_helpers.h"
  11 #include "base/command_line.h"
  12 #include "base/file_path.h"
  13 #include "base/message_loop.h"
  14 #include "base/run_loop.h"
  15 #include "base/scoped_native_library.h"
  16 #include "base/stringprintf.h"
  17 #include "base/utf_string_conversions.h"
  18 #include "base/win/windows_version.h"
  19 #include "remoting/base/auto_thread_task_runner.h"
  20 #include "remoting/base/breakpad.h"
  21 #include "remoting/host/desktop_process.h"
  22 #include "remoting/host/host_exit_codes.h"
  23 #include "remoting/host/logging.h"
  24 #include "remoting/host/usage_stats_consent.h"
  25
  26 #if defined(OS_MACOSX)
  27 #include "base/mac/scoped_nsautorelease_pool.h"
  28 #endif  // defined(OS_MACOSX)
  29
  30 #if defined(OS_WIN)
  31 #include <commctrl.h>
  32 #endif  // defined(OS_WIN)
  33
  34 namespace {
  35
  36 // The command line switch specifying the name of the daemon IPC endpoint.
  37 const char kDaemonIpcSwitchName[] = "daemon-pipe";
  38
  39 // "--help" or "--?" prints the usage message.
  40 const char kHelpSwitchName[] = "help";
  41 const char kQuestionSwitchName[] = "?";
  42
  43 const char kUsageMessage[] =
  44   "\n"
  45   "Usage: %s [options]\n"
  46   "\n"
  47   "Options:\n"
  48   "  --help, --?     - Print this message.\n";
  49
  50 void Usage(const FilePath& program_name) {
  51   std::string display_name = UTF16ToUTF8(program_name.LossyDisplayName());
  52   LOG(INFO) << StringPrintf(kUsageMessage, display_name.c_str());
  53 }
  54
  55 }  // namespace
  56
  57 int main(int argc, char** argv) {
  58 #if defined(OS_MACOSX)
  59   // Needed so we don't leak objects when threads are created.
  60   base::mac::ScopedNSAutoreleasePool pool;
  61 #endif
  62
  63   CommandLine::Init(argc, argv);
  64
  65   // This object instance is required by Chrome code (for example,
  66   // LazyInstance, MessageLoop).
  67   base::AtExitManager exit_manager;
  68
  69   remoting::InitHostLogging();
  70
  71   const CommandLine* command_line = CommandLine::ForCurrentProcess();
  72   if (command_line->HasSwitch(kHelpSwitchName) ||
  73       command_line->HasSwitch(kQuestionSwitchName)) {
  74     Usage(command_line->GetProgram());
  75     return remoting::kSuccessExitCode;
  76   }
  77
  78   std::string channel_name =
  79       command_line->GetSwitchValueASCII(kDaemonIpcSwitchName);
  80
  81   if (channel_name.empty()) {
  82     Usage(command_line->GetProgram());
  83     return remoting::kUsageExitCode;
  84   }
  85
  86   MessageLoop message_loop(MessageLoop::TYPE_UI);
  87   base::RunLoop run_loop;
  88   base::Closure quit_ui_task_runner = base::Bind(
  89       base::IgnoreResult(&base::SingleThreadTaskRunner::PostTask),
  90       message_loop.message_loop_proxy(),
  91       FROM_HERE, run_loop.QuitClosure());
  92   scoped_refptr<remoting::AutoThreadTaskRunner> ui_task_runner =
  93       new remoting::AutoThreadTaskRunner(message_loop.message_loop_proxy(),
  94                                          quit_ui_task_runner);
  95
  96   remoting::DesktopProcess desktop_process(ui_task_runner, channel_name);
  97   if (!desktop_process.Start())
  98     return remoting::kInitializationFailed;
  99
 100   // Run the UI message loop.
 101   ui_task_runner = NULL;
 102   run_loop.Run();
 103
 104   return remoting::kSuccessExitCode;
 105 }
 106
 107 #if defined(OS_WIN)
 108
 109 int CALLBACK WinMain(HINSTANCE instance,
 110                      HINSTANCE previous_instance,
 111                      LPSTR raw_command_line,
 112                      int show_command) {
 113 #ifdef OFFICIAL_BUILD
 114   if (remoting::IsUsageStatsAllowed()) {
 115     remoting::InitializeCrashReporting();
 116   }
 117 #endif  // OFFICIAL_BUILD
 118
 119   // Register and initialize common controls.
 120   INITCOMMONCONTROLSEX info;
 121   info.dwSize = sizeof(info);
 122   info.dwICC = ICC_STANDARD_CLASSES;
 123   InitCommonControlsEx(&info);
 124
 125   // Mark the process as DPI-aware, so Windows won't scale coordinates in APIs.
 126   // N.B. This API exists on Vista and above.
 127   if (base::win::GetVersion() >= base::win::VERSION_VISTA) {
 128     FilePath path(base::GetNativeLibraryName(UTF8ToUTF16("user32")));
 129     base::ScopedNativeLibrary user32(path);
 130     CHECK(user32.is_valid());
 131
 132     typedef BOOL (WINAPI * SetProcessDPIAwareFn)();
 133     SetProcessDPIAwareFn set_process_dpi_aware =
 134         static_cast<SetProcessDPIAwareFn>(
 135             user32.GetFunctionPointer("SetProcessDPIAware"));
 136     set_process_dpi_aware();
 137   }
 138
 139   // CommandLine::Init() ignores the passed |argc| and |argv| on Windows getting
 140   // the command line from GetCommandLineW(), so we can safely pass NULL here.
 141   return main(0, NULL);
 142 }
 143
 144 #endif  // defined(OS_WIN)