13f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// Use of this source code is governed by a BSD-style license that can be 3c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// found in the LICENSE file. 4c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 53f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen#include "base/debug/debug_on_start_win.h" 6c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 73f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen#include <windows.h> 8c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 9c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "base/base_switches.h" 10c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "base/basictypes.h" 11513209b27ff55e2841eac0e4120199c23acce758Ben Murdoch#include "base/debug/debugger.h" 12c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 133f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsennamespace base { 143f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsennamespace debug { 153f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen 16c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// Minimalist implementation to try to find a command line argument. We can use 17c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// kernel32 exported functions but not the CRT functions because we're too early 18c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// in the process startup. 19c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// The code is not that bright and will find things like ---argument or 20c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// /-/argument. 21c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// Note: command_line is non-destructively modified. 22c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottbool DebugOnStart::FindArgument(wchar_t* command_line, const char* argument_c) { 23dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen wchar_t argument[50] = {}; 24c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (int i = 0; argument_c[i]; ++i) 25c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott argument[i] = argument_c[i]; 26c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 27c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int argument_len = lstrlen(argument); 28c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int command_line_len = lstrlen(command_line); 29c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (command_line_len > argument_len) { 30c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott wchar_t first_char = command_line[0]; 31c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott wchar_t last_char = command_line[argument_len+1]; 32c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Try to find an argument. 33c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ((first_char == L'-' || first_char == L'/') && 34c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (last_char == L' ' || last_char == 0 || last_char == L'=')) { 35c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott command_line[argument_len+1] = 0; 36c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Skip the - or / 37c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (lstrcmpi(command_line+1, argument) == 0) { 38c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Found it. 39c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott command_line[argument_len+1] = last_char; 40c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return true; 41c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 42c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Fix back. 43c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott command_line[argument_len+1] = last_char; 44c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 45c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Continue searching. 46c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ++command_line; 47c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott --command_line_len; 48c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 49c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return false; 50c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 51c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 52c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// static 53c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint __cdecl DebugOnStart::Init() { 54c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Try to find the argument. 55c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (FindArgument(GetCommandLine(), switches::kDebugOnStart)) { 56c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // We can do 2 things here: 57c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // - Ask for a debugger to attach to us. This involve reading the registry 58c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // key and creating the process. 59c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // - Do a int3. 60c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 61c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // It will fails if we run in a sandbox. That is expected. 62513209b27ff55e2841eac0e4120199c23acce758Ben Murdoch base::debug::SpawnDebuggerOnProcess(GetCurrentProcessId()); 63c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 64c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Wait for a debugger to come take us. 65513209b27ff55e2841eac0e4120199c23acce758Ben Murdoch base::debug::WaitForDebugger(60, false); 66c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (FindArgument(GetCommandLine(), switches::kWaitForDebugger)) { 67c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Wait for a debugger to come take us. 68513209b27ff55e2841eac0e4120199c23acce758Ben Murdoch base::debug::WaitForDebugger(60, true); 69c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 70c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 71c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 723f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen 733f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen} // namespace debug 743f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen} // namespace base 75