【第十节】windows sdk编程：截获控件消息

目录

一、控件消息的截获概述

二、相关函数

三、示例代码

四、高级技巧与注意事项  

五、常见问题解答  

六、总结 

---

一、控件消息的截获概述

        控件的消息处理函数通常由系统定义，开发者无需干预。但在需要实现特殊交互逻辑（如自定义点击效果、消息过滤或增强功能）时，可通过消息处理函数替换技术截获控件消息。此技术通过替换控件的默认消息处理函数（Window Procedure），在自定义处理完成后，调用原函数确保消息链完整。该技术广泛应用于界面美化、输入验证和功能扩展等场景。

二、相关函数

        以下是两个关键的函数，用于获取和设置窗口或控件的属性：

LONG GetWindowLong(
    HWND hWnd,  // 窗口句柄
    int nIndex  // 索引
);

LONG SetWindowLong(
    HWND hWnd,      // 窗口句柄
    int nIndex,     // 索引
    LONG dwNewLong  // 新值
);

        `nIndex` 参数可以是以下宏之一：

        通过以下代码，可以将控件的消息处理函数替换为自定义的函数 `MyMsgProc`，并将原处理函数记录在 `OldMsgProc` 中：

OldMsgProc = (WNDPROC)SetWindowLong(hControlWnd, GWL_WNDPROC, (LONG)MyMsgProc);

        这种方法不仅可以用于截获控件的消息，还可以用于获取和修改大多数窗口的信息。

实现原理

        （1）替换消息处理函数：通过 SetWindowLongPtr 将控件的 WNDPROC 替换为自定义函数。

        （2）保存原函数指针：将原函数指针保存至安全位置（如全局变量或窗口的 GWLP_USERDATA）。

        （3）消息处理流程：

                自定义函数处理特定消息（如 WM_LBUTTONDOWN）。

                未处理的消息通过 CallWindowProc 传递给原函数，确保控件正常行为。

三、示例代码

        以下是一个完整的示例代码，展示了如何截获按钮控件的消息并自定义其行为：

cpp文件

#include 
#include 
#include "resource.h" // 包含资源头文件
#pragma comment(lib, "comctl32.lib")

// 函数声明
INT_PTR CALLBACK DialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
LRESULT CALLBACK NewBtnProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);

// WinMain 入口函数
int WINAPI WinMain(
    _In_ HINSTANCE hInstance,
    _In_opt_ HINSTANCE hPrevInstance,
    _In_ LPSTR lpCmdLine,
    _In_ int nShowCmd
) {
    // 初始化通用控件
    INITCOMMONCONTROLSEX icc = { sizeof(INITCOMMONCONTROLSEX), ICC_STANDARD_CLASSES };
    InitCommonControlsEx(&icc);

    // 创建并显示对话框
    DialogBox(hInstance, MAKEINTRESOURCE(IDD_DIALOG1), NULL, DialogProc);
    return 0;
}

// 自定义按钮消息处理函数
LRESULT CALLBACK NewBtnProc(
    _In_ HWND hwnd,
    _In_ UINT uMsg,
    _In_ WPARAM wParam,
    _In_ LPARAM lParam
) {
    // 从窗口属性中获取原处理函数
    WNDPROC oldProc = (WNDPROC)GetWindowLongPtr(hwnd, GWLP_USERDATA);

    switch (uMsg) {
    case WM_LBUTTONDOWN:
        // 自定义处理左键点击消息
        MessageBox(hwnd, L"按钮已被劫持！\n点击后将恢复正常状态", L" 窗口劫持通知", MB_OK | MB_ICONWARNING);

        // 恢复原消息处理函数
        SetWindowLongPtr(hwnd, GWLP_WNDPROC, (LONG_PTR)oldProc);
        SetWindowLongPtr(hwnd, GWLP_USERDATA, 0); // 清理窗口属性
        return 0; // 拦截消息

    case WM_DESTROY:
        // 清理窗口属性
        SetWindowLongPtr(hwnd, GWLP_USERDATA, 0);
        break;

    default:
        break;
    }

    // 调用原消息处理函数
    return CallWindowProc(oldProc, hwnd, uMsg, wParam, lParam);
}

// 对话框消息处理函数
INT_PTR CALLBACK DialogProc(
    _In_ HWND hwndDlg,
    _In_ UINT uMsg,
    _In_ WPARAM wParam,
    _In_ LPARAM lParam
) {
    switch (uMsg) {
    case WM_INITDIALOG:
        // 初始化对话框
        break;

    case WM_CLOSE:
        // 关闭对话框
        EndDialog(hwndDlg, 0);
        break;

    case WM_COMMAND: {
        WORD wId = LOWORD(wParam);
        if (wId == IDC_BUTTON1) {
            // 获取按钮句柄
            HWND hButton = (HWND)lParam;

            // 替换按钮的消息处理函数
            WNDPROC oldProc = (WNDPROC)SetWindowLongPtr(
                hButton,
                GWLP_WNDPROC,
                (LONG_PTR)NewBtnProc
            );

            if (oldProc == NULL) {
                // 错误处理
                MessageBox(hwndDlg, L"替换消息处理函数失败！", L"❌ 错误", MB_OK | MB_ICONERROR);
                break;
            }

            // 保存原函数指针到窗口属性
            SetWindowLongPtr(hButton, GWLP_USERDATA, (LONG_PTR)oldProc);

            // 提示用户
            MessageBox(hwndDlg, L"按钮正常点击\n下次点击时按钮将被劫持", L"✅ 按钮点击", MB_OK | MB_ICONINFORMATION);
        }
        break;
    }

    default:
        break;
    }
    return 0;
}

resource.h文件

#ifndef RESOURCE_H
#define RESOURCE_H

#define IDC_BUTTON1     106
#define IDD_DIALOG1     107
#define IDC_LIST1       108
#endif // RESOURCE_H

rc文件

#include 
#include 
#include 
#include "resource.h"

// 对话框资源定义
IDD_DIALOG1 DIALOGEX 0, 0, 300, 200
STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | WS_POPUP | WS_CAPTION | WS_SYSMENU
CAPTION "窗口子类化示例"
FONT 9, "MS Shell Dlg", 0, 0, 0x1
BEGIN
    PUSHBUTTON      "点击我",IDC_BUTTON1,100,80,100,40
END

代码说明

1. WinMain 函数：程序的入口点，创建并显示对话框。
2. DialogProc 函数：处理对话框的消息，包括初始化、关闭和按钮点击事件。
3. NewBtnProc 函数：自定义的按钮消息处理函数，用于截获并处理按钮的左键点击消息。
4. 全局变量 `g_oldBtn`：用于保存原按钮的消息处理函数，以便在自定义处理完成后恢复。

主要功能

- 消息截获：通过 `SetWindowLong` 函数将按钮的消息处理函数替换为自定义的 `NewBtnProc`。
- 自定义处理：在 `NewBtnProc` 中处理 `WM_LBUTTONDOWN` 消息，弹出一个消息框。
- 恢复原处理函数：在自定义处理完成后，恢复按钮的原始消息处理函数。

四、高级技巧与注意事项  

（1）动态消息拦截  
- 持续拦截：若需长期拦截消息，避免在自定义函数中恢复原处理函数。  
- 条件拦截：根据运行时状态动态决定是否处理消息，例如：  

 if (g_isInterceptEnabled) {
      // 自定义处理
      return 0;
  } else {
      return CallWindowProc(oldProc, hwnd, uMsg, wParam, lParam);
  }

（2）多控件管理  
- 遍历子控件：通过 `EnumChildWindows` 批量替换多个控件的消息处理函数。  
- 标识控件：使用 `GetDlgCtrlID` 区分不同控件，实现差异化处理。  

（3）错误处理  
- 函数返回值检查：  

  WNDPROC oldProc = (WNDPROC)SetWindowLongPtr(hButton, GWLP_WNDPROC, ...);
  if (oldProc == NULL) {
      DWORD err = GetLastError();
      // 处理错误
  }

（4）消息处理优先级  
- 子类化（Subclassing）：消息处理函数替换是子类化的一种形式，适用于单个控件。  
- 超类化（Superclassing）：创建新窗口类时修改消息处理，适用于同类所有控件。  

五、常见问题解答  

Q1: 如何防止消息截获导致程序崩溃？
- 原因：未正确保存或传递原函数指针。  
- 解决：  
  1. 使用 `SetProp`/`GetProp` 替代全局变量。  
  2. 确保所有路径最终调用 `CallWindowProc`。  

Q2: 如何截获对话框自身的消息？
- 方法：替换对话框的 `DlgProc`：  

 // 保存原对话框函数
  g_oldDlgProc = (WNDPROC)SetWindowLongPtr(hDlg, DWLP_DLGPROC, (LONG_PTR)NewDlgProc);

Q3: 截获消息后控件无法正常显示？ 
- 检查点：  
  1. 是否遗漏了 `WM_PAINT` 或 `WM_ERASEBKGND` 消息的处理。  
  2. 是否未调用原函数的默认处理逻辑。  

六、总结 

        通过替换控件的消息处理函数，开发者可以实现高度定制化的交互逻辑。关键点包括：  
（1）使用 `SetWindowLongPtr` 安全替换函数。  
（2）通过窗口属性或类成员保存原函数指针。  
（3）所有消息必须传递至原函数（除非明确拦截）。  
（4）注意64位兼容性和资源管理。  

        此技术结合其他Windows API（如钩子、Owner Draw），可进一步实现复杂UI效果，是Windows GUI开发的核心技能之一。