MFC中动态替换消息响应函数记得上次CSDN上有人发贴说怎么动态替换消息响应函数,我当时是在WndProc里自己进行消息分派并把代码贴给了他,但是我感觉不是很满意,要知道,在Delphi里,消息响应函数就是一个普通函数指针,可以随时切换,简直是任何问题都没有。MFC其实也完全可以做到。
我们知道,MFC中的消息响应是通过DECLARE_MESSAGE_MAP()、BEGIN_MESSAGE_MAP()、END_MESSAGE_MAP()等几个宏构造起来的,其消息响应函数放在一个_messageEntries的数组中,并且在编译的时候已经构造好,并且是private且const的。
实现方法很简单,先搜索到需要替换的位置,然后把消息响应结构所在的地址设置为可读写,并写上新的函数地址,然后把内存属性在设置回来,就可以了。
我写了一个模板函数,如下:
//动态切换消息响应函数
#ifndef _REPLACE_FUNCTION_
#define _REPLACE_FUNCTION_
#include "stdafx.h"
template<class WndClass>
class CReplaceFunction
{
public:
typedef void (WndClass::*FuncPtr)(void);
bool operator()(const AFX_MSGMAP_ENTRY *pMesaageEntry, FuncPtr OldPtr, FuncPtr NewPtr)
{
for (int i=0; i<sizeof(pMesaageEntry); i++)
{
if (pMesaageEntry[i].pfn == OldPtr)
{
HANDLE hProc;
DWORD dwOldProtect;
//获取当前进程的句柄
hProc=OpenProcess(PROCESS_ALL_ACCESS, 0, GetCurrentProcessId());
//把要修改的那几个字节的内存改为可读写的
VirtualProtectEx(hProc, (LPVOID)&pMesaageEntry[i].pfn, sizeof(pMesaageEntry[i].pfn), PAGE_READWRITE, &dwOldProtect);
const_cast<AFX_MSGMAP_ENTRY *>(pMesaageEntry)[i].pfn = (AFX_PMSG)NewPtr;
//恢复修改前的内存属性
VirtualProtectEx(hProc, (LPVOID)&pMesaageEntry[i].pfn, sizeof(pMesaageEntry[i].pfn), dwOldProtect, &dwOldProtect);
return true;
}
}
return false;
}
};
#endif //_REPLACE_FUNCTION_
用的时候很简单,看我下面的代码:
void CReplaceFuncDlg::OnBnClickedButton1()
{
AfxMessageBox("你好");
}
void CReplaceFuncDlg::ReplaceFunction()
{
AfxMessageBox("!@#$%^谁把我换拉");
}
void CReplaceFuncDlg::OnBnClickedButton2()
{
CReplaceFunction<CReplaceFuncDlg> replacer;
replacer(_messageEntries, OnBnClickedButton1, ReplaceFunction);
}
OnBnClickedButton1和OnBnClickedButton2是按钮的消息响应函数,按下按钮1,会弹出“你好”,然后按下按钮2,就会把原来按钮1的消息响应函数替换成ReplaceFunction,所以再去点按钮1的话,显示出来的就不是“你好”,而是“!@#$%^谁把我换拉”,哈哈。好玩吧
J 其实特简单。有兴趣的话,你可以多重载几个模板函数玩玩。
温柔的毒药 2005-06-24