/**

  * vim: set ts=4 :

  * =============================================================================

  * SourceMod

  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.

  * =============================================================================

  *

  * This program is free software; you can redistribute it and/or modify it under

  * the terms of the GNU General Public License, version 3.0, as published by the

  * Free Software Foundation.

  * 

  * This program 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 General Public License for more

  * details.

  *

  * You should have received a copy of the GNU General Public License along with

  * this program.  If not, see <http://www.gnu.org/licenses/>.

  *

  * As a special exception, AlliedModders LLC gives you permission to link the

  * code of this program (as well as its derivative works) to "Half-Life 2," the

  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

  * by the Valve Corporation.  You must obey the GNU General Public License in

  * all respects for all other code used.  Additionally, AlliedModders LLC grants

  * this exception to all derivative works.  AlliedModders LLC defines further

  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

  * 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