added craaaaazy debugger so we can get full minidumps

--HG-- extra : convert_revision : svn%3A39bc706e-5318-0410-9160-8a85361fbb7c/trunk%401117
2007-07-14 17:36:11 +00:00
parent 1f919c1f09
commit 68901b71d2
3 changed files with 213 additions and 0 deletions
@@ -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