15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdlib.h> 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/at_exit.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_handle.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/logging.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// "--help" or "--?" prints the usage message. 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kHelpSwitchName[] = "help"; 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kQuestionSwitchName[] = "?"; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kUsageMessage[] = 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Usage: %s <pid>\n" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) " pid - PID of the process to be crashed.\n"; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Exit codes: 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kSuccessExitCode = 0; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kUsageExitCode = 1; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kErrorExitCode = 2; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void usage(const char* program_name) { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fprintf(stderr, kUsageMessage, program_name); 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int main(int argc, char** argv) { 37cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) base::CommandLine::Init(argc, argv); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::AtExitManager exit_manager; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) remoting::InitHostLogging(); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 43cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) const base::CommandLine* command_line = 44cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) base::CommandLine::ForCurrentProcess(); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (command_line->HasSwitch(kHelpSwitchName) || 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) command_line->HasSwitch(kQuestionSwitchName)) { 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) usage(argv[0]); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return kSuccessExitCode; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 51cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) base::CommandLine::StringVector args = command_line->GetArgs(); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (args.size() != 1) { 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) usage(argv[0]); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return kUsageExitCode; 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pid = _wtoi(args[0].c_str()); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pid == 0) { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Invalid process PID: " << args[0]; 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return kErrorExitCode; 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD desired_access = PROCESS_CREATE_THREAD | PROCESS_QUERY_INFORMATION | 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PROCESS_VM_OPERATION | PROCESS_VM_WRITE | PROCESS_VM_READ; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedHandle process; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) process.Set(OpenProcess(desired_access, FALSE, pid)); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!process.IsValid()) { 68cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to open the process " << pid; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return kErrorExitCode; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD thread_id; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedHandle thread; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) thread.Set(CreateRemoteThread(process.Get(), NULL, 0, NULL, NULL, 0, 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &thread_id)); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!thread.IsValid()) { 77cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to create a remote thread in " << pid; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return kErrorExitCode; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return kSuccessExitCode; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 83