浅谈C++ Hook 钩子的使用介绍

1. 什么是Hook钩子？

Hook钩子是一种可以监控和修改系统、进程或应用程序行为的技术。在Windows操作系统下，可以通过Hook技术对API函数进行钩取，实现拦截API调用并进行自定义的处理。

2. Hook钩子的类型

在Windows操作系统中，可以使用以下两种类型的Hook钩子：

2.1 系统级钩子

系统级钩子可以拦截系统级事件，包括键盘、鼠标、消息、调试等，这意味着它们可以在操作系统和应用程序之间运行。由于系统级钩子是全局的，因此需要管理员权限才能安装和使用。

2.2 进程级钩子

进程级钩子仅能拦截进程级事件，例如进程通信、线程和窗口消息等。由于进程级钩子只能用于当前进程，因此不需要管理员权限。

3. Hook钩子的基本步骤

使用Hook钩子的基本步骤如下：

1. 找到要拦截的API函数的地址
2. 注册钩子函数
3. 安装钩子
4. 卸载钩子

下面我们来通过示例进行说明。

4. 示例1：拦截MessageBox函数

下面代码演示了如何使用Hook钩子拦截MessageBox函数，并将其强制转换为一个自定义的对话框。该示例代码使用了一个系统级钩子。

#include <windows.h>

HHOOK g_hHook;  // 钩子句柄

// 自定义对话框的回调函数
INT_PTR CALLBACK MyDlgProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
    switch(message)
    {
        case WM_INITDIALOG:
            // 在这里添加代码来初始化对话框
            return (INT_PTR)TRUE;

        case WM_COMMAND:
            if(LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
            {
                EndDialog(hDlg, LOWORD(wParam));
                return (INT_PTR)TRUE;
            }
            break;            
    }
    return (INT_PTR)FALSE;
}

// Hook钩子函数，拦截MessageBox函数
LRESULT CALLBACK MyHookProc(int nCode, WPARAM wParam, LPARAM lParam)
{
    if(nCode == HC_ACTION)
    {
        CWPSTRUCT* pMsg = (CWPSTRUCT*)lParam;
        if(pMsg != NULL && pMsg->message == WM_INITDIALOG)
        {
            // 将MessageBox对话框转换为自定义对话框
            HWND hMsgBox = pMsg->hwnd;
            HWND hParent = GetParent(hMsgBox);
            INT_PTR nResult = DialogBox(GetModuleHandle(NULL), MAKEINTRESOURCE(IDD_MYDIALOG), hParent, MyDlgProc);

            // 关闭MessageBox对话框
            SendMessage(hMsgBox, WM_CLOSE, 0, 0);
            return nResult ? nResult : TRUE;
        }
    }
    return CallNextHookEx(g_hHook, nCode, wParam, lParam);
}

int main()
{
    // 注册钩子函数
    g_hHook = SetWindowsHookEx(WH_CALLWNDPROCRET, MyHookProc, NULL, 0);

    // 调用MessageBox
    MessageBox(NULL, TEXT("Hello, world!"), TEXT("Message"), MB_OK);

    // 卸载钩子
    UnhookWindowsHookEx(g_hHook);
    return 0;
}

5. 示例2：拦截CreateFile函数

下面的示例代码演示了如何使用Hook钩子拦截CreateFile函数，并将文件名修改为指定的文件名。

#include <windows.h>

HHOOK g_hHook;  // 钩子句柄

// Hook钩子函数，拦截CreateFile函数
HANDLE WINAPI MyHookProc(LPCTSTR lpFileName, DWORD dwDesiredAccess, DWORD dwShareMode,
                         LPSECURITY_ATTRIBUTES lpSecurityAttributes, DWORD dwCreationDisposition,
                         DWORD dwFlagsAndAttributes, HANDLE hTemplateFile)
{
    // 将文件名修改为指定的文件名
    TCHAR szNewFileName[] = TEXT("C:\\Demo.txt");
    return CreateFile(szNewFileName, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, hTemplateFile);
}

int main()
{
    // 注册钩子函数
    g_hHook = SetWindowsHookEx(WH_CBT, (HOOKPROC)MyHookProc, NULL, GetCurrentThreadId());

    // 调用CreateFile
    HANDLE hFile = CreateFile(TEXT("C:\\Test.txt"), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if(hFile != INVALID_HANDLE_VALUE)
    {
        // 在这里添加代码来处理文件

        // 关闭文件句柄
        CloseHandle(hFile);
    }

    // 卸载钩子
    UnhookWindowsHookEx(g_hHook);
    return 0;
}