| /* | |
| * Unit tests for the debugger facility | |
| * | |
| * Copyright (c) 2007 Francois Gouget for CodeWeavers | |
| * | |
| * This library is free software; you can redistribute it and/or | |
| * modify it under the terms of the GNU Lesser General Public | |
| * License as published by the Free Software Foundation; either | |
| * version 2.1 of the License, or (at your option) any later version. | |
| * | |
| * This library is distributed in the hope that it will be useful, | |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| * Lesser General Public License for more details. | |
| * | |
| * You should have received a copy of the GNU Lesser General Public | |
| * License along with this library; if not, write to the Free Software | |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA | |
| */ | |
| #include <stdio.h> | |
| #include <assert.h> | |
| #include <windows.h> | |
| #include <winreg.h> | |
| #include "wine/test.h" | |
| #ifndef STATUS_DEBUGGER_INACTIVE | |
| #define STATUS_DEBUGGER_INACTIVE ((NTSTATUS) 0xC0000354) | |
| #endif | |
| static int myARGC; | |
| static char** myARGV; | |
| static BOOL (WINAPI *pDebugActiveProcessStop)(DWORD); | |
| static BOOL (WINAPI *pDebugSetProcessKillOnExit)(BOOL); | |
| /* Copied from the process test */ | |
| static void get_file_name(char* buf) | |
| { | |
| char path[MAX_PATH]; | |
| buf[0] = '\0'; | |
| GetTempPathA(sizeof(path), path); | |
| GetTempFileNameA(path, "wt", 0, buf); | |
| } | |
| static void get_events(const char* name, HANDLE *start_event, HANDLE *done_event) | |
| { | |
| const char* basename; | |
| char* event_name; | |
| basename=strrchr(name, '\\'); | |
| basename=(basename ? basename+1 : name); | |
| event_name=HeapAlloc(GetProcessHeap(), 0, 6+strlen(basename)+1); | |
| sprintf(event_name, "start_%s", basename); | |
| *start_event=CreateEvent(NULL, 0,0, event_name); | |
| sprintf(event_name, "done_%s", basename); | |
| *done_event=CreateEvent(NULL, 0,0, event_name); | |
| HeapFree(GetProcessHeap(), 0, event_name); | |
| } | |
| static void save_blackbox(const char* logfile, void* blackbox, int size) | |
| { | |
| HANDLE hFile; | |
| DWORD written; | |
| hFile=CreateFileA(logfile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0); | |
| if (hFile == INVALID_HANDLE_VALUE) | |
| return; | |
| WriteFile(hFile, blackbox, size, &written, NULL); | |
| CloseHandle(hFile); | |
| } | |
| static int load_blackbox(const char* logfile, void* blackbox, int size) | |
| { | |
| HANDLE hFile; | |
| DWORD read; | |
| BOOL ret; | |
| hFile=CreateFileA(logfile, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0); | |
| if (hFile == INVALID_HANDLE_VALUE) | |
| { | |
| ok(0, "unable to open '%s'\n", logfile); | |
| return 0; | |
| } | |
| ret=ReadFile(hFile, blackbox, size, &read, NULL); | |
| ok(read == size, "wrong size for '%s': read=%d\n", logfile, read); | |
| CloseHandle(hFile); | |
| return 1; | |
| } | |
| typedef struct | |
| { | |
| DWORD pid; | |
| } crash_blackbox_t; | |
| static void doCrash(int argc, char** argv) | |
| { | |
| char* p; | |
| if (argc >= 4) | |
| { | |
| crash_blackbox_t blackbox; | |
| blackbox.pid=GetCurrentProcessId(); | |
| save_blackbox(argv[3], &blackbox, sizeof(blackbox)); | |
| } | |
| /* Just crash */ | |
| trace("child: crashing...\n"); | |
| p=NULL; | |
| *p=0; | |
| } | |
| typedef struct | |
| { | |
| int argc; | |
| DWORD pid; | |
| BOOL debug_rc; | |
| DWORD debug_err; | |
| BOOL attach_rc; | |
| DWORD attach_err; | |
| BOOL nokill_rc; | |
| DWORD nokill_err; | |
| BOOL detach_rc; | |
| DWORD detach_err; | |
| } debugger_blackbox_t; | |
| static void doDebugger(int argc, char** argv) | |
| { | |
| const char* logfile; | |
| debugger_blackbox_t blackbox; | |
| HANDLE start_event, done_event, debug_event; | |
| blackbox.argc=argc; | |
| logfile=(argc >= 4 ? argv[3] : NULL); | |
| blackbox.pid=(argc >= 5 ? atol(argv[4]) : 0); | |
| if (strstr(myARGV[2], "attach")) | |
| { | |
| blackbox.attach_rc=DebugActiveProcess(blackbox.pid); | |
| if (!blackbox.attach_rc) | |
| blackbox.attach_err=GetLastError(); | |
| } | |
| else | |
| blackbox.attach_rc=TRUE; | |
| debug_event=(argc >= 6 ? (HANDLE)atol(argv[5]) : NULL); | |
| if (debug_event && strstr(myARGV[2], "event")) | |
| { | |
| blackbox.debug_rc=SetEvent(debug_event); | |
| if (!blackbox.debug_rc) | |
| blackbox.debug_err=GetLastError(); | |
| } | |
| else | |
| blackbox.debug_rc=TRUE; | |
| get_events(logfile, &start_event, &done_event); | |
| if (strstr(myARGV[2], "order")) | |
| { | |
| trace("debugger: waiting for the start signal...\n"); | |
| WaitForSingleObject(start_event, INFINITE); | |
| } | |
| if (strstr(myARGV[2], "nokill")) | |
| { | |
| blackbox.nokill_rc=pDebugSetProcessKillOnExit(FALSE); | |
| if (!blackbox.nokill_rc) | |
| blackbox.nokill_err=GetLastError(); | |
| } | |
| else | |
| blackbox.nokill_rc=TRUE; | |
| if (strstr(myARGV[2], "detach")) | |
| { | |
| blackbox.detach_rc=pDebugActiveProcessStop(blackbox.pid); | |
| if (!blackbox.detach_rc) | |
| blackbox.detach_err=GetLastError(); | |
| } | |
| else | |
| blackbox.detach_rc=TRUE; | |
| save_blackbox(logfile, &blackbox, sizeof(blackbox)); | |
| trace("debugger: done debugging...\n"); | |
| SetEvent(done_event); | |
| /* Just exit with a known value */ | |
| ExitProcess(0xdeadbeef); | |
| } | |
| static void crash_and_debug(HKEY hkey, const char* argv0, const char* dbgtasks) | |
| { | |
| DWORD ret; | |
| HANDLE start_event, done_event; | |
| char* cmd; | |
| char dbglog[MAX_PATH]; | |
| char childlog[MAX_PATH]; | |
| PROCESS_INFORMATION info; | |
| STARTUPINFOA startup; | |
| DWORD exit_code; | |
| crash_blackbox_t crash_blackbox; | |
| debugger_blackbox_t dbg_blackbox; | |
| ret=RegSetValueExA(hkey, "auto", 0, REG_SZ, (BYTE*)"1", 2); | |
| ok(ret == ERROR_SUCCESS, "unable to set AeDebug/auto: ret=%d\n", ret); | |
| get_file_name(dbglog); | |
| get_events(dbglog, &start_event, &done_event); | |
| cmd=HeapAlloc(GetProcessHeap(), 0, strlen(argv0)+10+strlen(dbgtasks)+1+strlen(dbglog)+34+1); | |
| sprintf(cmd, "%s debugger %s %s %%ld %%ld", argv0, dbgtasks, dbglog); | |
| ret=RegSetValueExA(hkey, "debugger", 0, REG_SZ, (BYTE*)cmd, strlen(cmd)+1); | |
| ok(ret == ERROR_SUCCESS, "unable to set AeDebug/debugger: ret=%d\n", ret); | |
| HeapFree(GetProcessHeap(), 0, cmd); | |
| get_file_name(childlog); | |
| cmd=HeapAlloc(GetProcessHeap(), 0, strlen(argv0)+16+strlen(dbglog)+1); | |
| sprintf(cmd, "%s debugger crash %s", argv0, childlog); | |
| memset(&startup, 0, sizeof(startup)); | |
| startup.cb = sizeof(startup); | |
| startup.dwFlags = STARTF_USESHOWWINDOW; | |
| startup.wShowWindow = SW_SHOWNORMAL; | |
| ret=CreateProcessA(NULL, cmd, NULL, NULL, FALSE, 0, NULL, NULL, &startup, &info); | |
| ok(ret, "CreateProcess: err=%d\n", GetLastError()); | |
| HeapFree(GetProcessHeap(), 0, cmd); | |
| CloseHandle(info.hThread); | |
| /* The process exits... */ | |
| trace("waiting for child exit...\n"); | |
| ok(WaitForSingleObject(info.hProcess, 60000) == WAIT_OBJECT_0, "Timed out waiting for the child to crash\n"); | |
| ok(GetExitCodeProcess(info.hProcess, &exit_code), "GetExitCodeProcess failed: err=%d\n", GetLastError()); | |
| if (strstr(dbgtasks, "code2")) | |
| { | |
| /* If, after attaching to the debuggee, the debugger exits without | |
| * detaching, then the debuggee gets a special exit code. | |
| */ | |
| ok(exit_code == 0xffffffff || /* Win 9x */ | |
| exit_code == 0x80 || /* NT4 */ | |
| exit_code == STATUS_DEBUGGER_INACTIVE, /* Win >= XP */ | |
| "wrong exit code : %08x\n", exit_code); | |
| } | |
| else | |
| ok(exit_code == STATUS_ACCESS_VIOLATION, "exit code = %08x instead of STATUS_ACCESS_VIOLATION\n", exit_code); | |
| CloseHandle(info.hProcess); | |
| /* ...before the debugger */ | |
| if (strstr(dbgtasks, "order")) | |
| ok(SetEvent(start_event), "SetEvent(start_event) failed\n"); | |
| trace("waiting for the debugger...\n"); | |
| ok(WaitForSingleObject(done_event, 60000) == WAIT_OBJECT_0, "Timed out waiting for the debugger\n"); | |
| assert(load_blackbox(childlog, &crash_blackbox, sizeof(crash_blackbox))); | |
| assert(load_blackbox(dbglog, &dbg_blackbox, sizeof(dbg_blackbox))); | |
| ok(dbg_blackbox.argc == 6, "wrong debugger argument count: %d\n", dbg_blackbox.argc); | |
| ok(dbg_blackbox.pid == crash_blackbox.pid, "the child and debugged pids don't match: %d != %d\n", crash_blackbox.pid, dbg_blackbox.pid); | |
| ok(dbg_blackbox.debug_rc, "debugger: SetEvent(debug_event) failed err=%d\n", dbg_blackbox.debug_err); | |
| ok(dbg_blackbox.attach_rc, "DebugActiveProcess(%d) failed err=%d\n", dbg_blackbox.pid, dbg_blackbox.attach_err); | |
| ok(dbg_blackbox.nokill_rc, "DebugSetProcessKillOnExit(FALSE) failed err=%d\n", dbg_blackbox.nokill_err); | |
| ok(dbg_blackbox.detach_rc, "DebugActiveProcessStop(%d) failed err=%d\n", dbg_blackbox.pid, dbg_blackbox.detach_err); | |
| assert(DeleteFileA(dbglog) != 0); | |
| assert(DeleteFileA(childlog) != 0); | |
| } | |
| static void crash_and_winedbg(HKEY hkey, const char* argv0) | |
| { | |
| DWORD ret; | |
| char* cmd; | |
| PROCESS_INFORMATION info; | |
| STARTUPINFOA startup; | |
| DWORD exit_code; | |
| ret=RegSetValueExA(hkey, "auto", 0, REG_SZ, (BYTE*)"1", 2); | |
| ok(ret == ERROR_SUCCESS, "unable to set AeDebug/auto: ret=%d\n", ret); | |
| cmd=HeapAlloc(GetProcessHeap(), 0, strlen(argv0)+15+1); | |
| sprintf(cmd, "%s debugger crash", argv0); | |
| memset(&startup, 0, sizeof(startup)); | |
| startup.cb = sizeof(startup); | |
| startup.dwFlags = STARTF_USESHOWWINDOW; | |
| startup.wShowWindow = SW_SHOWNORMAL; | |
| ret=CreateProcessA(NULL, cmd, NULL, NULL, FALSE, 0, NULL, NULL, &startup, &info); | |
| ok(ret, "CreateProcess: err=%d\n", GetLastError()); | |
| HeapFree(GetProcessHeap(), 0, cmd); | |
| CloseHandle(info.hThread); | |
| trace("waiting for child exit...\n"); | |
| ok(WaitForSingleObject(info.hProcess, 60000) == WAIT_OBJECT_0, "Timed out waiting for the child to crash\n"); | |
| ok(GetExitCodeProcess(info.hProcess, &exit_code), "GetExitCodeProcess failed: err=%d\n", GetLastError()); | |
| ok(exit_code == STATUS_ACCESS_VIOLATION, "exit code = %08x\n", exit_code); | |
| CloseHandle(info.hProcess); | |
| } | |
| static void test_ExitCode(void) | |
| { | |
| static const char* AeDebug="Software\\Microsoft\\Windows NT\\CurrentVersion\\AeDebug"; | |
| char test_exe[MAX_PATH]; | |
| DWORD ret; | |
| HKEY hkey; | |
| DWORD disposition; | |
| LPBYTE auto_val=NULL; | |
| DWORD auto_size, auto_type; | |
| LPBYTE debugger_val=NULL; | |
| DWORD debugger_size, debugger_type; | |
| GetModuleFileNameA(GetModuleHandle(NULL), test_exe, sizeof(test_exe)); | |
| if (GetFileAttributes(test_exe) == INVALID_FILE_ATTRIBUTES) | |
| strcat(test_exe, ".so"); | |
| if (GetFileAttributesA(test_exe) == INVALID_FILE_ATTRIBUTES) | |
| { | |
| ok(0, "could not find the test executable '%s'\n", test_exe); | |
| return; | |
| } | |
| ret=RegCreateKeyExA(HKEY_LOCAL_MACHINE, AeDebug, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, &disposition); | |
| if (ret == ERROR_SUCCESS) | |
| { | |
| auto_size=0; | |
| ret=RegQueryValueExA(hkey, "auto", NULL, &auto_type, NULL, &auto_size); | |
| if (ret == ERROR_SUCCESS) | |
| { | |
| auto_val=HeapAlloc(GetProcessHeap(), 0, auto_size); | |
| RegQueryValueExA(hkey, "auto", NULL, &auto_type, auto_val, &auto_size); | |
| } | |
| debugger_size=0; | |
| ret=RegQueryValueExA(hkey, "debugger", NULL, &debugger_type, NULL, &debugger_size); | |
| if (ret == ERROR_SUCCESS) | |
| { | |
| debugger_val=HeapAlloc(GetProcessHeap(), 0, debugger_size); | |
| RegQueryValueExA(hkey, "debugger", NULL, &debugger_type, debugger_val, &debugger_size); | |
| } | |
| } | |
| else if (ret == ERROR_ACCESS_DENIED) | |
| { | |
| skip("not enough privileges to change the debugger\n"); | |
| return; | |
| } | |
| else if (ret != ERROR_FILE_NOT_FOUND) | |
| { | |
| ok(0, "could not open the AeDebug key: %d\n", ret); | |
| return; | |
| } | |
| if (debugger_val && debugger_type == REG_SZ && | |
| strstr((char*)debugger_val, "winedbg --auto")) | |
| crash_and_winedbg(hkey, test_exe); | |
| crash_and_debug(hkey, test_exe, "dbg,none"); | |
| crash_and_debug(hkey, test_exe, "dbg,event,order"); | |
| crash_and_debug(hkey, test_exe, "dbg,attach,event,code2"); | |
| if (pDebugSetProcessKillOnExit) | |
| crash_and_debug(hkey, test_exe, "dbg,attach,event,nokill"); | |
| if (pDebugActiveProcessStop) | |
| crash_and_debug(hkey, test_exe, "dbg,attach,event,detach"); | |
| if (disposition == REG_CREATED_NEW_KEY) | |
| { | |
| RegCloseKey(hkey); | |
| RegDeleteKeyA(HKEY_LOCAL_MACHINE, AeDebug); | |
| } | |
| else | |
| { | |
| if (auto_val) | |
| { | |
| RegSetValueExA(hkey, "auto", 0, auto_type, auto_val, auto_size); | |
| HeapFree(GetProcessHeap(), 0, auto_val); | |
| } | |
| else | |
| RegDeleteValueA(hkey, "auto"); | |
| if (debugger_val) | |
| { | |
| RegSetValueExA(hkey, "debugger", 0, debugger_type, debugger_val, debugger_size); | |
| HeapFree(GetProcessHeap(), 0, debugger_val); | |
| } | |
| else | |
| RegDeleteValueA(hkey, "debugger"); | |
| RegCloseKey(hkey); | |
| } | |
| } | |
| START_TEST(debugger) | |
| { | |
| HMODULE hdll; | |
| hdll=GetModuleHandle("kernel32.dll"); | |
| pDebugActiveProcessStop=(void*)GetProcAddress(hdll, "DebugActiveProcessStop"); | |
| pDebugSetProcessKillOnExit=(void*)GetProcAddress(hdll, "DebugSetProcessKillOnExit"); | |
| myARGC=winetest_get_mainargs(&myARGV); | |
| if (myARGC >= 3 && strcmp(myARGV[2], "crash") == 0) | |
| { | |
| doCrash(myARGC, myARGV); | |
| } | |
| else if (myARGC >= 3 && strncmp(myARGV[2], "dbg,", 4) == 0) | |
| { | |
| doDebugger(myARGC, myARGV); | |
| } | |
| else | |
| { | |
| test_ExitCode(); | |
| } | |
| } |