remoting/tools/breakpad_tester_win.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 #include <windows.h>
   6 #include <stdlib.h>
   7
   8 #include "base/at_exit.h"
   9 #include "base/command_line.h"
  10 #include "base/win/scoped_handle.h"
  11 #include "remoting/host/logging.h"
  12
  13 namespace {
  14
  15 // "--help" or "--?" prints the usage message.
  16 const char kHelpSwitchName[] = "help";
  17 const char kQuestionSwitchName[] = "?";
  18
  19 const char kUsageMessage[] =
  20   "\n"
  21   "Usage: %s <pid>\n"
  22   "\n"
  23   "  pid  - PID of the process to be crashed.\n";
  24
  25 // Exit codes:
  26 const int kSuccessExitCode = 0;
  27 const int kUsageExitCode = 1;
  28 const int kErrorExitCode = 2;
  29
  30 void usage(const char* program_name) {
  31   fprintf(stderr, kUsageMessage, program_name);
  32 }
  33
  34 }  // namespace
  35
  36 int main(int argc, char** argv) {
  37   CommandLine::Init(argc, argv);
  38
  39   base::AtExitManager exit_manager;
  40
  41   remoting::InitHostLogging();
  42
  43   const CommandLine* command_line = CommandLine::ForCurrentProcess();
  44   if (command_line->HasSwitch(kHelpSwitchName) ||
  45       command_line->HasSwitch(kQuestionSwitchName)) {
  46     usage(argv[0]);
  47     return kSuccessExitCode;
  48   }
  49
  50   CommandLine::StringVector args = command_line->GetArgs();
  51   if (args.size() != 1) {
  52     usage(argv[0]);
  53     return kUsageExitCode;
  54   }
  55
  56   int pid = _wtoi(args[0].c_str());
  57   if (pid == 0) {
  58     LOG(ERROR) << "Invalid process PID: " << args[0];
  59     return kErrorExitCode;
  60   }
  61
  62   DWORD desired_access = PROCESS_CREATE_THREAD | PROCESS_QUERY_INFORMATION |
  63       PROCESS_VM_OPERATION | PROCESS_VM_WRITE | PROCESS_VM_READ;
  64   base::win::ScopedHandle process;
  65   process.Set(OpenProcess(desired_access, FALSE, pid));
  66   if (!process.IsValid()) {
  67     LOG_GETLASTERROR(ERROR) << "Failed to open the process " << pid;
  68     return kErrorExitCode;
  69   }
  70
  71   DWORD thread_id;
  72   base::win::ScopedHandle thread;
  73   thread.Set(CreateRemoteThread(process.Get(), NULL, 0, NULL, NULL, 0,
  74                                 &thread_id));
  75   if (!thread.IsValid()) {
  76     LOG_GETLASTERROR(ERROR) << "Failed to create a remote thread in " << pid;
  77     return kErrorExitCode;
  78   }
  79
  80   return kSuccessExitCode;
  81 }