// dllmain.cpp : 定义 DLL 应用程序的入口点。
#include "stdafx.h"

DWORD oldProtect;
BYTE  JmpBtye[5];
BYTE  OldByte[5];
void * H1_Messagebox;
void *  F1_Messagebox();
bool H1_MessageBoxA();
void UnHook();
//定义函数原型
int WINAPI My_MessageBoxA(
	HWND hwnd,
	LPCTSTR lpTExt,
	LPCTSTR lpCaption,
	UINT uType);
BOOL APIENTRY DllMain( HMODULE hModule,
                       DWORD  ul_reason_for_call,
                       LPVOID lpReserved
                     )
{
    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
		H1_MessageBoxA();
		break;
    case DLL_PROCESS_DETACH:

        break;
    }
    return TRUE;
}

//自定义函数初始化
int WINAPI My_MessageBoxA(
	HWND hwnd,
	LPCTSTR lpTExt,
	LPCTSTR lpCaption,
	UINT uType)
{
	char newText[1024] = { 0 };
	lstrcpy(newText, lpTExt);
	lstrcat(newText, TEXT("\n\tMessageBox Hacked by pediy.com!"));
	uType |= MB_ICONERROR;
	return 0;
};
//取模块地址
void *  F1_Messagebox()
{
	void * addr = 0;
	HMODULE hModule = LoadLibraryA("User32.dll");
	addr = (void *)GetProcAddress(hModule, "MessageBoxA");
	return addr;
}

//开始hook
bool H1_MessageBoxA()
{

	void *addr = F1_Messagebox();
	H1_Messagebox = addr;
	if (addr == 0)
	{
		MessageBoxA(NULL, "error", NULL, 0);
		return false;
	}

	VirtualProtect((void*)addr, 5, PAGE_EXECUTE_READWRITE, &oldProtect);
	JmpBtye[0] = 0xE9;
	*(DWORD*)JmpBtye[1]= (DWORD)((long long)My_MessageBoxA - (long long)addr - 5);
	memcpy(OldByte, (void*)addr, 5);
	memcpy((void*)addr, JmpBtye, 5);
	

}
//恢复
void UnHook()
{
	memcpy((void*)H1_Messagebox, OldByte, 5);
	DWORD p;
	VirtualProtect((void*)H1_Messagebox, 5, oldProtect, &p);

}