dvander@1111: /**
dvander@1111: * vim: set ts=4 :
dvander@1231: * ================================================================
dvander@1231: * SourceMod
dvander@1231: * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved.
dvander@1231: * ================================================================
dvander@1111: *
dvander@1231: * This program is free software; you can redistribute it and/or
dvander@1231: * modify it under the terms of the GNU General Public License,
dvander@1231: * version 3.0, as published by the Free Software Foundation.
dvander@1231: *
dvander@1231: * This program is distributed in the hope that it will be useful,
dvander@1231: * but WITHOUT ANY WARRANTY; without even the implied warranty of
dvander@1231: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
dvander@1231: * GNU General Public License for more details.
dvander@1231: *
dvander@1231: * You should have received a copy of the GNU General Public License
dvander@1231: * along with this program. If not, see .
dvander@1231: *
dvander@1231: * As a special exception, AlliedModders LLC gives you permission to
dvander@1231: * link the code of this program (as well as its derivative works) to
dvander@1231: * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any
dvander@1231: * Game MODs that run on software by the Valve Corporation. You must
dvander@1231: * obey the GNU General Public License in all respects for all other
dvander@1231: * code used. Additionally, AlliedModders LLC grants this exception
dvander@1231: * to all derivative works. AlliedModders LLC defines further
dvander@1231: * exceptions, found in LICENSE.txt (as of this writing, version
dvander@1231: * JULY-31-2007), or .
dvander@1111: *
dvander@1111: * Version: $Id$
dvander@1111: */
dvander@1231:
dvander@1111: #if defined CRAZY_DEBUG
dvander@1111: #include "sm_globals.h"
dvander@1111: #include "sourcemm_api.h"
dvander@1111: #include "Tlhelp32.h"
dvander@1111: #include "LibrarySys.h"
dvander@1111: #include "minidump.h"
dvander@1111:
dvander@1111: BOOL
dvander@1111: WINAPI
dvander@1111: MiniDumpWriteDump2(
dvander@1111: IN HANDLE hProcess,
dvander@1111: IN DWORD ProcessId,
dvander@1111: IN HANDLE hFile,
dvander@1111: IN MINIDUMP_TYPE DumpType,
dvander@1111: IN CONST PMINIDUMP_EXCEPTION_INFORMATION ExceptionParam, OPTIONAL
dvander@1111: IN CONST PMINIDUMP_USER_STREAM_INFORMATION UserStreamParam, OPTIONAL
dvander@1111: IN CONST PMINIDUMP_CALLBACK_INFORMATION CallbackParam OPTIONAL
dvander@1111: )
dvander@1111: {
dvander@1111: DumpType = (MINIDUMP_TYPE)((int)DumpType|MiniDumpWithFullMemory|MiniDumpWithHandleData);
dvander@1111: return MiniDumpWriteDump(hProcess, ProcessId, hFile, DumpType, ExceptionParam, UserStreamParam, CallbackParam);
dvander@1111: }
dvander@1111:
dvander@1111: FARPROC WINAPI GetProcAddress2(HMODULE hModule, LPCSTR lpProcName)
dvander@1111: {
dvander@1111: if (strcmp(lpProcName, "MiniDumpWriteDump") == 0)
dvander@1111: {
dvander@1111: return (FARPROC)MiniDumpWriteDump2;
dvander@1111: }
dvander@1111:
dvander@1111: return GetProcAddress(hModule, lpProcName);
dvander@1111: }
dvander@1111:
dvander@1111: class CrazyWindowsDebugger : public SMGlobalClass
dvander@1111: {
dvander@1111: public:
dvander@1111: void OnSourceModAllInitialized()
dvander@1111: {
dvander@1111: HANDLE hModuleList = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, GetCurrentProcessId());
dvander@1111: MODULEENTRY32 me32;
dvander@1111:
dvander@1111: me32.dwSize = sizeof(MODULEENTRY32);
dvander@1111:
dvander@1111: if (!Module32First(hModuleList, &me32))
dvander@1111: {
dvander@1111: Error("Could not initialize crazy debugger!");
dvander@1111: }
dvander@1111:
dvander@1111: bool found = false;
dvander@1111:
dvander@1111: do
dvander@1111: {
dvander@1111: if (strcasecmp(me32.szModule, "steam.dll") == 0)
dvander@1111: {
dvander@1111: IMAGE_DOS_HEADER *dos = (IMAGE_DOS_HEADER *)me32.modBaseAddr;
dvander@1111: if (dos->e_magic != IMAGE_DOS_SIGNATURE)
dvander@1111: {
dvander@1111: Error("[SM] Could not detect steam.dll with valid DOS signature");
dvander@1111: }
dvander@1111: char *base = (char *)dos;
dvander@1111: IMAGE_NT_HEADERS *nt = (IMAGE_NT_HEADERS *)(base + dos->e_lfanew);
dvander@1111: if (nt->Signature != IMAGE_NT_SIGNATURE)
dvander@1111: {
dvander@1111: Error("[SM] Could not detect steam.dll with valid NT signature");
dvander@1111: }
dvander@1111: IMAGE_IMPORT_DESCRIPTOR *desc =
dvander@1111: (IMAGE_IMPORT_DESCRIPTOR *)
dvander@1111: (base + nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress);
dvander@1111: if (base == (char *)desc)
dvander@1111: {
dvander@1111: Error("[SM] Could not find the steam.dll IAT");
dvander@1111: }
dvander@1111: while (desc->Name)
dvander@1111: {
dvander@1111: if (desc->FirstThunk != 0)
dvander@1111: {
dvander@1111: IMAGE_THUNK_DATA *data = (IMAGE_THUNK_DATA *)(base + desc->OriginalFirstThunk);
dvander@1111: DWORD *iat = (DWORD *)(base + desc->FirstThunk);
dvander@1111: while (data->u1.Function)
dvander@1111: {
dvander@1111: if ((data->u1.Ordinal & IMAGE_ORDINAL_FLAG32) != IMAGE_ORDINAL_FLAG32)
dvander@1111: {
dvander@1111: IMAGE_IMPORT_BY_NAME *import = (IMAGE_IMPORT_BY_NAME *)(base + data->u1.AddressOfData);
dvander@1111: if (strcmp((char *)import->Name, "GetProcAddress") == 0)
dvander@1111: {
dvander@1111: DWORD oldprot, oldprot2;
dvander@1111: VirtualProtect(iat, 4, PAGE_READWRITE, &oldprot);
dvander@1111: *iat = (DWORD)GetProcAddress2;
dvander@1111: VirtualProtect(iat, 4, oldprot, &oldprot2);
dvander@1111: found = true;
dvander@1111: goto _end;
dvander@1111: }
dvander@1111: }
dvander@1111: data++;
dvander@1111: iat++;
dvander@1111: }
dvander@1111: }
dvander@1111: desc++;
dvander@1111: }
dvander@1111: break;
dvander@1111: }
dvander@1111: } while (Module32Next(hModuleList, &me32));
dvander@1111:
dvander@1111: _end:
dvander@1111:
dvander@1111: if (!found)
dvander@1111: {
dvander@1111: Error("Could not find steam.dll's GetProcAddress IAT entry");
dvander@1111: }
dvander@1111:
dvander@1111: CloseHandle(hModuleList);
dvander@1111: }
dvander@1111: } s_CrazyDebugger;
dvander@1111: #endif