license.bot | f003cfe | 2008-08-24 09:55:55 +0900 | [diff] [blame] | 1 | // Copyright (c) 2006-2008 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. |
brettw@google.com | cc1eb1b | 2008-08-09 07:07:48 +0900 | [diff] [blame] | 4 | |
initial.commit | 3f4a732 | 2008-07-27 06:49:38 +0900 | [diff] [blame] | 5 | #include <windows.h> |
| 6 | |
| 7 | #include "base/debug_on_start.h" |
| 8 | |
| 9 | #include "base/base_switches.h" |
| 10 | #include "base/basictypes.h" |
| 11 | #include "base/debug_util.h" |
| 12 | |
| 13 | // Minimalist implementation to try to find a command line argument. We can use |
| 14 | // kernel32 exported functions but not the CRT functions because we're too early |
| 15 | // in the process startup. |
| 16 | // The code is not that bright and will find things like ---argument or |
| 17 | // /-/argument. |
| 18 | // Note: command_line is non-destructively modified. |
| 19 | bool DebugOnStart::FindArgument(wchar_t* command_line, const wchar_t* argument) |
| 20 | { |
| 21 | int argument_len = lstrlen(argument); |
| 22 | int command_line_len = lstrlen(command_line); |
| 23 | while (command_line_len > argument_len) { |
| 24 | wchar_t first_char = command_line[0]; |
| 25 | wchar_t last_char = command_line[argument_len+1]; |
| 26 | // Try to find an argument. |
| 27 | if ((first_char == L'-' || first_char == L'/') && |
| 28 | (last_char == L' ' || last_char == 0 || last_char == L'=')) { |
| 29 | command_line[argument_len+1] = 0; |
| 30 | // Skip the - or / |
| 31 | if (lstrcmpi(command_line+1, argument) == 0) { |
| 32 | // Found it. |
| 33 | command_line[argument_len+1] = last_char; |
| 34 | return true; |
| 35 | } |
| 36 | // Fix back. |
| 37 | command_line[argument_len+1] = last_char; |
| 38 | } |
| 39 | // Continue searching. |
| 40 | ++command_line; |
| 41 | --command_line_len; |
| 42 | } |
| 43 | return false; |
| 44 | } |
| 45 | |
| 46 | // static |
| 47 | int __cdecl DebugOnStart::Init() { |
| 48 | // Try to find the argument. |
| 49 | if (FindArgument(GetCommandLine(), switches::kDebugOnStart)) { |
| 50 | // We can do 2 things here: |
| 51 | // - Ask for a debugger to attach to us. This involve reading the registry |
| 52 | // key and creating the process. |
| 53 | // - Do a int3. |
| 54 | |
| 55 | // It will fails if we run in a sandbox. That is expected. |
| 56 | DebugUtil::SpawnDebuggerOnProcess(GetCurrentProcessId()); |
| 57 | |
| 58 | // Wait for a debugger to come take us. |
| 59 | DebugUtil::WaitForDebugger(60, false); |
| 60 | } else if (FindArgument(GetCommandLine(), switches::kWaitForDebugger)) { |
| 61 | // Wait for a debugger to come take us. |
| 62 | DebugUtil::WaitForDebugger(60, true); |
| 63 | } |
| 64 | return 0; |
| 65 | } |