|
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. |
|
4 |
|
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 } |