diff --git a/core/CrazyDebugger.cpp b/core/CrazyDebugger.cpp new file mode 100644 index 00000000..7966efad --- /dev/null +++ b/core/CrazyDebugger.cpp @@ -0,0 +1,127 @@ +/** + * vim: set ts=4 : + * =============================================================== + * SourceMod (C)2004-2007 AlliedModders LLC. All rights reserved. + * =============================================================== + * + * This file is not open source and may not be copied without explicit + * written permission of AlliedModders LLC. This file may not be redistributed + * in whole or significant part. + * For information, see LICENSE.txt or http://www.sourcemod.net/license.php + * + * Version: $Id$ + */ +#if defined CRAZY_DEBUG +#include "sm_globals.h" +#include "sourcemm_api.h" +#include "Tlhelp32.h" +#include "LibrarySys.h" +#include "minidump.h" + +BOOL +WINAPI +MiniDumpWriteDump2( + IN HANDLE hProcess, + IN DWORD ProcessId, + IN HANDLE hFile, + IN MINIDUMP_TYPE DumpType, + IN CONST PMINIDUMP_EXCEPTION_INFORMATION ExceptionParam, OPTIONAL + IN CONST PMINIDUMP_USER_STREAM_INFORMATION UserStreamParam, OPTIONAL + IN CONST PMINIDUMP_CALLBACK_INFORMATION CallbackParam OPTIONAL + ) +{ + DumpType = (MINIDUMP_TYPE)((int)DumpType|MiniDumpWithFullMemory|MiniDumpWithHandleData); + return MiniDumpWriteDump(hProcess, ProcessId, hFile, DumpType, ExceptionParam, UserStreamParam, CallbackParam); +} + +FARPROC WINAPI GetProcAddress2(HMODULE hModule, LPCSTR lpProcName) +{ + if (strcmp(lpProcName, "MiniDumpWriteDump") == 0) + { + return (FARPROC)MiniDumpWriteDump2; + } + + return GetProcAddress(hModule, lpProcName); +} + +class CrazyWindowsDebugger : public SMGlobalClass +{ +public: + void OnSourceModAllInitialized() + { + HANDLE hModuleList = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, GetCurrentProcessId()); + MODULEENTRY32 me32; + + me32.dwSize = sizeof(MODULEENTRY32); + + if (!Module32First(hModuleList, &me32)) + { + Error("Could not initialize crazy debugger!"); + } + + bool found = false; + + do + { + if (strcasecmp(me32.szModule, "steam.dll") == 0) + { + IMAGE_DOS_HEADER *dos = (IMAGE_DOS_HEADER *)me32.modBaseAddr; + if (dos->e_magic != IMAGE_DOS_SIGNATURE) + { + Error("[SM] Could not detect steam.dll with valid DOS signature"); + } + char *base = (char *)dos; + IMAGE_NT_HEADERS *nt = (IMAGE_NT_HEADERS *)(base + dos->e_lfanew); + if (nt->Signature != IMAGE_NT_SIGNATURE) + { + Error("[SM] Could not detect steam.dll with valid NT signature"); + } + IMAGE_IMPORT_DESCRIPTOR *desc = + (IMAGE_IMPORT_DESCRIPTOR *) + (base + nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress); + if (base == (char *)desc) + { + Error("[SM] Could not find the steam.dll IAT"); + } + while (desc->Name) + { + if (desc->FirstThunk != 0) + { + IMAGE_THUNK_DATA *data = (IMAGE_THUNK_DATA *)(base + desc->OriginalFirstThunk); + DWORD *iat = (DWORD *)(base + desc->FirstThunk); + while (data->u1.Function) + { + if ((data->u1.Ordinal & IMAGE_ORDINAL_FLAG32) != IMAGE_ORDINAL_FLAG32) + { + IMAGE_IMPORT_BY_NAME *import = (IMAGE_IMPORT_BY_NAME *)(base + data->u1.AddressOfData); + if (strcmp((char *)import->Name, "GetProcAddress") == 0) + { + DWORD oldprot, oldprot2; + VirtualProtect(iat, 4, PAGE_READWRITE, &oldprot); + *iat = (DWORD)GetProcAddress2; + VirtualProtect(iat, 4, oldprot, &oldprot2); + found = true; + goto _end; + } + } + data++; + iat++; + } + } + desc++; + } + break; + } + } while (Module32Next(hModuleList, &me32)); + +_end: + + if (!found) + { + Error("Could not find steam.dll's GetProcAddress IAT entry"); + } + + CloseHandle(hModuleList); + } +} s_CrazyDebugger; +#endif diff --git a/core/msvc8/sourcemod_mm.sln b/core/msvc8/sourcemod_mm.sln index 6a65d2ef..6d3ef0e6 100644 --- a/core/msvc8/sourcemod_mm.sln +++ b/core/msvc8/sourcemod_mm.sln @@ -5,10 +5,13 @@ Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "sourcemod_mm", "sourcemod_m EndProject Global GlobalSection(SolutionConfigurationPlatforms) = preSolution + CrazyDebug|Win32 = CrazyDebug|Win32 Debug|Win32 = Debug|Win32 Release|Win32 = Release|Win32 EndGlobalSection GlobalSection(ProjectConfigurationPlatforms) = postSolution + {E39527CD-7CAB-4420-97CC-DA1B93B260BC}.CrazyDebug|Win32.ActiveCfg = CrazyDebug|Win32 + {E39527CD-7CAB-4420-97CC-DA1B93B260BC}.CrazyDebug|Win32.Build.0 = CrazyDebug|Win32 {E39527CD-7CAB-4420-97CC-DA1B93B260BC}.Debug|Win32.ActiveCfg = Debug|Win32 {E39527CD-7CAB-4420-97CC-DA1B93B260BC}.Debug|Win32.Build.0 = Debug|Win32 {E39527CD-7CAB-4420-97CC-DA1B93B260BC}.Release|Win32.ActiveCfg = Release|Win32 diff --git a/core/msvc8/sourcemod_mm.vcproj b/core/msvc8/sourcemod_mm.vcproj index 7568de8d..d13cd56c 100644 --- a/core/msvc8/sourcemod_mm.vcproj +++ b/core/msvc8/sourcemod_mm.vcproj @@ -172,6 +172,85 @@ Name="VCPostBuildEventTool" /> + + + + + + + + + + + + + + + + + + + + @@ -209,6 +288,10 @@ RelativePath="..\CoreConfig.cpp" > + +