//HookDemo.cpp文件
#include <windows.h>
#include <stdio.h>
// 掛鉤指定模塊hMod對MessageBoxA的調用
BOOL SetHookApi(HMODULE hMod, LPCSTR lpstrDLLName, PROC pfnOldFun, PROC pfnNewFun);

HANDLE
WINAPI
MY_CreateIoCompletionPort(
??? HANDLE FileHandle,
??? HANDLE ExistingCompletionPort,
??? DWORD CompletionKey,
??? DWORD NumberOfConcurrentThreads
??? )
{
?return (HANDLE)3;
}

void main()
{
?::SetHookApi(::GetModuleHandle(NULL), "Kernel32.dll", (PROC)CreateIoCompletionPort, (PROC)MY_CreateIoCompletionPort);
?HANDLE h = ::CreateIoCompletionPort(NULL, NULL, 0, 0);
}

BOOL SetHookApi(HMODULE hMod, LPCSTR lpstrDLLName, PROC pfnOldFun, PROC pfnNewFun)
{
?IMAGE_DOS_HEADER* pDosHeader = (IMAGE_DOS_HEADER*)hMod;
?IMAGE_OPTIONAL_HEADER * pOptHeader = (IMAGE_OPTIONAL_HEADER *)((BYTE*)hMod + pDosHeader->e_lfanew + 24);
?IMAGE_IMPORT_DESCRIPTOR* pImportDesc = (IMAGE_IMPORT_DESCRIPTOR*)
???????????????????????????????????? ((BYTE*)hMod +
?????????????? pOptHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress);

?// 在導入表中查找user32.dll模塊。因為MessageBoxA函數從user32.dll模塊導出
?while(pImportDesc->FirstThunk)
?{
??char* pszDllName = (char*)((BYTE*)hMod + pImportDesc->Name);
??if(lstrcmpiA(pszDllName, lpstrDLLName) == 0)
??{
???break;
??}
??pImportDesc++;
?}

?if(pImportDesc->FirstThunk)
?{
??// 一個IMAGE_THUNK_DATA就是一個雙字，它指定了一個導入函數
??// 調入地址表其實是IMAGE_THUNK_DATA結構的數組，也就是DWORD數組
??IMAGE_THUNK_DATA* pThunk = (IMAGE_THUNK_DATA*)
???((BYTE*)hMod + pImportDesc->FirstThunk);

??while(pThunk->u1.Function)
??{
???// lpAddr指向的內存保存了函數的地址
???DWORD* lpAddr = (DWORD*)&(pThunk->u1.Function);
???if(*lpAddr == (DWORD)pfnOldFun)
???{
????DWORD dwOldProtect;
????MEMORY_BASIC_INFORMATION mb;
????VirtualQuery(lpAddr, &mb, sizeof(mb));
????VirtualProtect(lpAddr, sizeof(DWORD), PAGE_READWRITE, &dwOldProtect);

????// 修改IAT表項，使其指向我們自定義的函數，相當于“*lpAddr = (DWORD)MyMessageBoxA;”
????DWORD* lpNewProc = (DWORD*)pfnNewFun;

????::WriteProcessMemory(::GetCurrentProcess(),
?????lpAddr, &lpNewProc, sizeof(DWORD), NULL);
????VirtualProtect(lpAddr, sizeof(DWORD), dwOldProtect, 0);
????return TRUE;
???}

???pThunk++;
??}
?}

?return FALSE;
}

////////////////////////////////////////////////////////////////////////////////////////////////////////////
APIHIJACK.H

/*--------------------------------------------------------------------------------------------------------
? APIHIJACK.H - Based on DelayLoadProfileDLL.CPP, by Matt Pietrek for MSJ February 2000.
? http://msdn.microsoft.com/library/periodic/period00/hood0200.htm
? Adapted by Wade Brainerd, wadeb@wadeb.com
--------------------------------------------------------------------------------------------------------*/
#ifndef APIHIJACK_H
#define APIHIJACK_H

#pragma warning(disable:4200)

// Macro for convenient pointer addition.
// Essentially treats the last two parameters as DWORDs.? The first
// parameter is used to typecast the result to the appropriate pointer type.
#define MakePtr(cast, ptr, addValue ) (cast)( (DWORD)(ptr)+(DWORD)(addValue))

// Default Hook Stub Structure: Contains data about the original function, Name/Ordinal, Address
// and a Count field.? This is actually a block of assembly code.
#pragma pack( push, 1 )
struct DLPD_IAT_STUB
{
??? BYTE??? instr_CALL;
??? DWORD?? data_call;
??? BYTE??? instr_JMP;
??? DWORD?? data_JMP;
??? DWORD?? count;
??? DWORD?? pszNameOrOrdinal;

??? DLPD_IAT_STUB() : instr_CALL( 0xE8 ), instr_JMP( 0xE9 ), count( 0 ) {}
};
#pragma pack( pop )

// Example DefaultHook procedure, called from the DLPD_IAT_STUB stubs.?
// Increments "count" field of the stub.
// See the implementation for more information.
void __cdecl DefaultHook( PVOID dummy );

struct SFunctionHook
{
??? char* Name;???????? // Function name, e.g. "DirectDrawCreateEx".
??? void* HookFn;?????? // Address of your function.
??? void* OrigFn;?????? // Stored by HookAPICalls, the address of the original function.
};

struct SDLLHook
{
??? // Name of the DLL, e.g. "DDRAW.DLL"
??? char* Name;

??? // Set true to call the default for all non-hooked functions before they are executed.
??? bool UseDefault;
??? void* DefaultFn;

??? // Function hook array.? Terminated with a NULL Name field.
??? SFunctionHook Functions[];
};

// Hook functions one or more DLLs.
bool HookAPICalls( SDLLHook* Hook );

#endif

//////////////////////////////////////////////////////////////////
APIHIJACK.CPP

/*--------------------------------------------------------------------------------------------------------
??? APIHIJACK.CPP - Based on DelayLoadProfileDLL.CPP, by Matt Pietrek for MSJ February 2000.
??? http://msdn.microsoft.com/library/periodic/period00/hood0200.htm
??? Adapted by Wade Brainerd, wadeb@wadeb.com
--------------------------------------------------------------------------------------------------------*/
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <stdio.h>
#include "apihijack.h"

//===========================================================================
// Called from the DLPD_IAT_STUB stubs.? Increments "count" field of the stub

void __cdecl DefaultHook( PVOID dummy )
{
??? __asm?? pushad? // Save all general purpose registers

??? // Get return address, then subtract 5 (size of a CALL X instruction)
??? // The result points at a DLPD_IAT_STUB

??? // pointer math!? &dummy-1 really subtracts sizeof(PVOID)
??? PDWORD pRetAddr = (PDWORD)(&dummy - 1);

??? DLPD_IAT_STUB * pDLPDStub = (DLPD_IAT_STUB *)(*pRetAddr - 5);

??? pDLPDStub->count++;

??? #if 0
??? // Remove the above conditional to get a cheezy API trace from
??? // the loader process.? It's slow!
??? if ( !IMAGE_SNAP_BY_ORDINAL( pDLPDStub->pszNameOrOrdinal) )
??? {
??????? OutputDebugString( "Called hooked function: " );
??????? OutputDebugString( (PSTR)pDLPDStub->pszNameOrOrdinal );
??????? OutputDebugString( "\n" );
??? }
??? #endif

??? __asm?? popad?? // Restore all general purpose registers
}

// This function must be __cdecl!!!
void __cdecl DelayLoadProfileDLL_UpdateCount( PVOID dummy );

PIMAGE_IMPORT_DESCRIPTOR g_pFirstImportDesc;

//===========================================================================
// Given an HMODULE, returns a pointer to the PE header

PIMAGE_NT_HEADERS PEHeaderFromHModule(HMODULE hModule)
{
??? PIMAGE_NT_HEADERS pNTHeader = 0;
???
??? __try
??? {
??????? if ( PIMAGE_DOS_HEADER(hModule)->e_magic != IMAGE_DOS_SIGNATURE )
??????????? __leave;

??????? pNTHeader = PIMAGE_NT_HEADERS(PBYTE(hModule)
??????????????????? + PIMAGE_DOS_HEADER(hModule)->e_lfanew);
???????
??????? if ( pNTHeader->Signature != IMAGE_NT_SIGNATURE )
??????????? pNTHeader = 0;
??? }
??? __except( EXCEPTION_EXECUTE_HANDLER )
??? {??????
??? }

??? return pNTHeader;
}

//===========================================================================
// Builds stubs for and redirects the IAT for one DLL (pImportDesc)

bool RedirectIAT( SDLLHook* DLLHook, PIMAGE_IMPORT_DESCRIPTOR pImportDesc, PVOID pBaseLoadAddr )
{
??? PIMAGE_THUNK_DATA pIAT;???? // Ptr to import address table
??? PIMAGE_THUNK_DATA pINT;???? // Ptr to import names table
??? PIMAGE_THUNK_DATA pIteratingIAT;

??? // Figure out which OS platform we're on
??? OSVERSIONINFO osvi;
??? osvi.dwOSVersionInfoSize = sizeof(osvi);
??? GetVersionEx( &osvi );

??? // If no import names table, we can't redirect this, so bail
??? if ( pImportDesc->OriginalFirstThunk == 0 )
??????? return false;

??? pIAT = MakePtr( PIMAGE_THUNK_DATA, pBaseLoadAddr, pImportDesc->FirstThunk );
??? pINT = MakePtr( PIMAGE_THUNK_DATA, pBaseLoadAddr, pImportDesc->OriginalFirstThunk );

??? // Count how many entries there are in this IAT.? Array is 0 terminated
??? pIteratingIAT = pIAT;
??? unsigned cFuncs = 0;
??? while ( pIteratingIAT->u1.Function )
??? {
??????? cFuncs++;
??????? pIteratingIAT++;
??? }

??? if ( cFuncs == 0 )? // If no imported functions, we're done!
??????? return false;

??? // These next few lines ensure that we'll be able to modify the IAT,
??? // which is often in a read-only section in the EXE.
??? DWORD flOldProtect, flNewProtect, flDontCare;
??? MEMORY_BASIC_INFORMATION mbi;
???
??? // Get the current protection attributes???????????????????????????
??? VirtualQuery( pIAT, &mbi, sizeof(mbi) );
???
??? // remove ReadOnly and ExecuteRead attributes, add on ReadWrite flag
??? flNewProtect = mbi.Protect;
??? flNewProtect &= ~(PAGE_READONLY | PAGE_EXECUTE_READ);
??? flNewProtect |= (PAGE_READWRITE);
???
??? if ( !VirtualProtect(?? pIAT, sizeof(PVOID) * cFuncs,
??????????????????????????? flNewProtect, &flOldProtect) )
??? {
??????? return false;
??? }

??? // If the Default hook is enabled, build an array of redirection stubs in the processes memory.
??? DLPD_IAT_STUB * pStubs = 0;
??? if ( DLLHook->UseDefault )
??? {
??????? // Allocate memory for the redirection stubs.? Make one extra stub at the
??????? // end to be a sentinel
??????? pStubs = new DLPD_IAT_STUB[ cFuncs + 1];
??????? if ( !pStubs )
??????????? return false;
??? }

??? // Scan through the IAT, completing the stubs and redirecting the IAT
??? // entries to point to the stubs
??? pIteratingIAT = pIAT;

??? while ( pIteratingIAT->u1.Function )
??? {
??????? void* HookFn = 0;? // Set to either the SFunctionHook or pStubs.

??????? if ( !IMAGE_SNAP_BY_ORDINAL( pINT->u1.Ordinal ) )? // import by name
??????? {
??????????? PIMAGE_IMPORT_BY_NAME pImportName = MakePtr( PIMAGE_IMPORT_BY_NAME, pBaseLoadAddr, pINT->u1.AddressOfData );

??????????? // Iterate through the hook functions, searching for this import.
??????????? SFunctionHook* FHook = DLLHook->Functions;
??????????? while ( FHook->Name )
??????????? {
??????????????? if ( lstrcmpi( FHook->Name, (char*)pImportName->Name ) == 0 )
??????????????? {
??????????????????? OutputDebugString( "Hooked function: " );
??????????????????? OutputDebugString( (char*)pImportName->Name );
??????????????????? OutputDebugString( "\n" );

??????????????????? // Save the old function in the SFunctionHook structure and get the new one.
??????????????????? FHook->OrigFn = pIteratingIAT->u1.Function;
??????????????????? HookFn = FHook->HookFn;
??????????????????? break;
??????????????? }

??????????????? FHook++;
??????????? }

??????????? // If the default function is enabled, store the name for the user.
??????????? if ( DLLHook->UseDefault )
??????????????? pStubs->pszNameOrOrdinal = (DWORD)&pImportName->Name;
??????? }
??????? else
??????? {
??????????? // If the default function is enabled, store the ordinal for the user.
??????????? if ( DLLHook->UseDefault )
??????????????? pStubs->pszNameOrOrdinal = pINT->u1.Ordinal;
??????? }

??????? // If the default function is enabled, fill in the fields to the stub code.
??????? if ( DLLHook->UseDefault )
??????? {
??????????? pStubs->data_call = (DWORD)(PDWORD)DLLHook->DefaultFn
??????????????????????????????? - (DWORD)(PDWORD)&pStubs->instr_JMP;
??????????? pStubs->data_JMP = *(PDWORD)pIteratingIAT - (DWORD)(PDWORD)&pStubs->count;

??????????? // If it wasn't manually hooked, use the Stub function.
??????????? if ( !HookFn )
??????????????? HookFn = (void*)pStubs;
??????? }

??????? // Replace the IAT function pointer if we have a hook.
??????? if ( HookFn )
??????? {
??????????? // Cheez-o hack to see if what we're importing is code or data.
??????????? // If it's code, we shouldn't be able to write to it
??????????? if ( IsBadWritePtr( (PVOID)pIteratingIAT->u1.Function, 1 ) )
??????????? {
??????????????? pIteratingIAT->u1.Function = (PDWORD)HookFn;
??????????? }
??????????? else if ( osvi.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS )
??????????? {
??????????????? // Special hack for Win9X, which builds stubs for imported
??????????????? // functions in system DLLs (Loaded above 2GB).? These stubs are
??????????????? // writeable, so we have to explicitly check for this case
??????????????? if ( pIteratingIAT->u1.Function > (PDWORD)0x80000000 )
??????????????????? pIteratingIAT->u1.Function = (PDWORD)HookFn;
??????????? }
??????? }

??????? if ( DLLHook->UseDefault )
??????????? pStubs++;?????????? // Advance to next stub

??????? pIteratingIAT++;??? // Advance to next IAT entry
??????? pINT++;???????????? // Advance to next INT entry
??? }

??? if ( DLLHook->UseDefault )
??????? pStubs->pszNameOrOrdinal = 0;?? // Final stub is a sentinel

??? // Put the page attributes back the way they were.
??? VirtualProtect( pIAT, sizeof(PVOID) * cFuncs, flOldProtect, &flDontCare);
???
??? return true;
}

//===========================================================================
// Top level routine to find the EXE's imports, and redirect them
bool HookAPICalls( SDLLHook* Hook )
{
??? if ( !Hook )
??????? return false;

??? HMODULE hModEXE = GetModuleHandle( 0 );

??? PIMAGE_NT_HEADERS pExeNTHdr = PEHeaderFromHModule( hModEXE );
???
??? if ( !pExeNTHdr )
??????? return false;

??? DWORD importRVA = pExeNTHdr->OptionalHeader.DataDirectory
??????????????????????? [IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress;
??? if ( !importRVA )
??????? return false;

??? // Convert imports RVA to a usable pointer
??? PIMAGE_IMPORT_DESCRIPTOR pImportDesc = MakePtr( PIMAGE_IMPORT_DESCRIPTOR,
??????????????????????????????????????????????????? hModEXE, importRVA );

??? // Save off imports address in a global for later use
??? g_pFirstImportDesc = pImportDesc;??

??? // Iterate through each import descriptor, and redirect if appropriate
??? while ( pImportDesc->FirstThunk )
??? {
??????? PSTR pszImportModuleName = MakePtr( PSTR, hModEXE, pImportDesc->Name);

??????? if ( lstrcmpi( pszImportModuleName, Hook->Name ) == 0 )
??????? {
??????????? OutputDebugString( "Found " );
??????????? OutputDebugString( Hook->Name );
??????????? OutputDebugString( "...\n" );

??????????? RedirectIAT( Hook, pImportDesc, (PVOID)hModEXE );
??????? }
???????
??????? pImportDesc++;? // Advance to next import descriptor
??? }

??? return true;
}

SDLLHook D3DHook =
{
??? "DDRAW.DLL",
??? false, NULL,??// Default hook disabled, NULL function pointer.
??? {
??????? { "DirectDrawCreateEx", MyDirectDrawCreateEx },
??????? { NULL, NULL }
??? }
};

// Hook function.
HRESULT WINAPI MyDirectDrawCreateEx( GUID FAR * lpGuid, LPVOID? *lplpDD, REFIID? iid,IUnknown FAR *pUnkOuter )
{
??? // Let the world know we're working.
??? MessageBeep( MB_ICONINFORMATION );

??? OutputDebugString( "TESTDLL: MyDirectDrawCreateEx called.\n" );

??? DirectDrawCreateEx_Type OldFn =
??????? (DirectDrawCreateEx_Type)D3DHook.Functions[D3DFN_DirectDrawCreateEx].OrigFn;
??? return OldFn( lpGuid, lplpDD, iid, pUnkOuter );
}

HookAPICalls( &D3DHook );