#include <iostream>
#include <Windows.h>
#include "detours.h"
#pragma comment(lib, "detours.lib")
// 定義指向原始 MessageBoxW 函數(shù)的指針
static int (WINAPI *OriginalMessageBoxW)(
	_In_opt_ HWND hWnd,
	_In_opt_ LPCWSTR lpText,
	_In_opt_ LPCWSTR lpCaption,
	_In_ UINT uType) = MessageBoxW;
// 自定義的 MessageBoxW 函數(shù)，用于替換原始函數(shù)
static int WINAPI CustomMessageBoxW(
	_In_opt_ HWND hWnd,
	_In_opt_ LPCWSTR lpText,
	_In_opt_ LPCWSTR lpCaption,
	_In_ UINT uType)
{
	// 調(diào)用原始的 MessageBoxW 函數(shù)，但修改了顯示的文本
	return OriginalMessageBoxW(hWnd, L"hello lyshark", L"MsgBox", MB_OK);
}

// 安裝鉤子
void InstallHook()
{
	// 開始一個(gè) Detours 事務(wù)
	if (DetourTransactionBegin() == NO_ERROR)
	{
		// 更新當(dāng)前線程以準(zhǔn)備進(jìn)行鉤子操作
		if (DetourUpdateThread(GetCurrentThread()) == NO_ERROR)
		{
			// 將原始函數(shù)指針替換為自定義的函數(shù)指針
			if (DetourAttach(&(PVOID&)OriginalMessageBoxW, CustomMessageBoxW) == NO_ERROR)
			{
				// 提交事務(wù)，完成鉤子安裝
				if (DetourTransactionCommit() == NO_ERROR)
				{
					printf("鉤子已成功安裝。\n");
					return;
				}
			}
		}
		// 如果任何步驟失敗，則中止事務(wù)
		DetourTransactionAbort();
	}
	printf("安裝鉤子失敗。\n");
}
// 移除鉤子
void RemoveHook()
{
	// 開始一個(gè) Detours 事務(wù)
	if (DetourTransactionBegin() == NO_ERROR)
	{
		// 更新當(dāng)前線程以準(zhǔn)備進(jìn)行鉤子操作
		if (DetourUpdateThread(GetCurrentThread()) == NO_ERROR)
		{
			// 將自定義的函數(shù)指針替換回原始函數(shù)指針
			if (DetourDetach(&(PVOID&)OriginalMessageBoxW, CustomMessageBoxW) == NO_ERROR)
			{
				// 提交事務(wù)，完成鉤子移除
				if (DetourTransactionCommit() == NO_ERROR)
				{
					printf("鉤子已成功移除。\n");
					return;
				}
			}
		}
		// 如果任何步驟失敗，則中止事務(wù)
		DetourTransactionAbort();
	}
	printf("移除鉤子失敗。\n");
}

int main(int argc, char *argv[])
{
	// 顯示原始的消息框
	MessageBoxW(NULL, L"hello world", L"MsgBox", MB_OK);
	// 安裝鉤子并顯示修改后的消息框
	InstallHook();
	MessageBoxW(NULL, L"hello world", L"MsgBox", MB_OK);
	// 移除鉤子并恢復(fù)為原始的消息框
	RemoveHook();
	MessageBoxW(NULL, L"hello world", L"MsgBox", MB_OK);
	system("pause");
	return 0;
}

#include <windows.h>
#include "detours.h"
#include "detver.h"
#pragma comment(lib, "detours.lib")
// 定義 MessageBoxW 函數(shù)指針類型
static int (WINAPI *OriginalMessageBoxW)(HWND hWnd, LPCWSTR lpText, LPCWSTR lpCaption, UINT uType) = NULL;
// 自定義的 MessageBoxW 函數(shù)
int WINAPI MyMessageBoxW(HWND hWnd, LPCWSTR lpText, LPCWSTR lpCaption, UINT uType)
{
	// 可以在這里添加自定義邏輯
	return OriginalMessageBoxW(hWnd, L"hello lyshark", lpCaption, uType);
}
// 安裝鉤子
void InstallHooks()
{
	DetourRestoreAfterWith();
	// 開始事務(wù)
	DetourTransactionBegin();
	// 更新當(dāng)前線程
	DetourUpdateThread(GetCurrentThread());
	// 查找并攔截 MessageBoxW 函數(shù)
	OriginalMessageBoxW = (int (WINAPI *)(HWND, LPCWSTR, LPCWSTR, UINT))DetourFindFunction("User32.dll", "MessageBoxW");
	if (OriginalMessageBoxW != NULL)
	{
		// 開始掛鉤
		DetourAttach(&(PVOID&)OriginalMessageBoxW, MyMessageBoxW);
	}
	// 提交事務(wù)
	DetourTransactionCommit();
}
// 卸載鉤子
void UninstallHooks()
{
	// 開始事務(wù)
	DetourTransactionBegin();
	// 更新當(dāng)前線程
	DetourUpdateThread(GetCurrentThread());
	// 撤銷攔截 MessageBoxW 函數(shù)
	if (OriginalMessageBoxW != NULL)
	{
		// 摘除鉤子
		DetourDetach(&(PVOID&)OriginalMessageBoxW, MyMessageBoxW);
	}
	// 提交事務(wù)
	DetourTransactionCommit();
}
// DLL 入口點(diǎn)
BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
{
	switch (ul_reason_for_call)
	{
	case DLL_PROCESS_ATTACH:
		// 禁用線程庫調(diào)用
		DisableThreadLibraryCalls(hModule);
		// 安裝鉤子
		InstallHooks();
		break;
	case DLL_THREAD_ATTACH:
		break;
	case DLL_THREAD_DETACH:
		break;
	case DLL_PROCESS_DETACH:
		// 卸載鉤子
		UninstallHooks();
		break;
	}
	return TRUE;
}