菜单也是一个窗口, 假如我们能得到它的窗口的句柄, 要实现像添加阴影这样的效果, 就不会很难了。
可惜我们根本找不到这个窗口是在哪里被创建的,也没办法很容易地取得它的窗口句柄,甚至几乎难以相信它是一个窗口,因为我实在找不到它的窗口句柄啊。经过对许多别人已经做好的类的源代码的"研究", 我终于找到了一个方法。那就是万能的钩子,如果说在Windows里面抓"人",连钩子也办不到的话,那我就不知道该用什么方法实现了,呵呵。
下面我就一起来看看如何抓到这些"可恶"的家伙吧。为了便于移植,我们就写一个专用的类吧,就取名为CMenuWndHook。添加两个静态成员先:
static CMap m_WndMenuMap;
static HHOOK m_hMenuHook;
被我们抓到的这些家伙肯定不止一个,我们需要一个映射模板类来保存它们的句柄和对应的CMenuWndHook 类对象的指针。m_hMenuHook则为我们将要创建的钩子的钩子句柄。再在CPP文件中初始化它们:
CMap CMenuWndHook::m_WndMenuMap;
HHOOK CMenuWndHook::m_hMenuHook = NULL;
下面再添加两个函数来做安装与卸载hook之用, 它们都是静态函数:
void CMenuWndHook::InstallHook()
{
if (m_hMenuHook == NULL)
{
m_hMenuHook = ::SetWindowsHookEx(WH_CALLWNDPROC,WindowHook,AfxGetApp()->m_hInstance,::GetCurrentThreadId());
}
}
Windows之下一般用上面的SetWindowsHookEx API函数来安装HOOK,它的函数原型如下:
HHOOK SetWindowsHookEx(int idHook, //钩子的类型,即它处理的消息类型
HOOKPROC lpfn, //子函数的入口地址,当钩子钩到任何消息后先调用这个函数。
// (如果dwThreadId参数为0,或是一个由别的进程创建的线程的标识,
//lpfn必须指向DLL中的钩子子程。除此以外,lpfn可以指向当前进
//程的一段钩子子程代码)
HINSTANCE hMod, //应用程序实例的句柄。标识包含lpfn所指的子程的DLL。
// 如果dwThreadId标识当前进程创建的一个线程,
//而且子程代码位于当前进程,hMod必须为NULL。
//可以很简单的设定其为本应用程序的实例句柄。
DWORD dwThreadId //与安装的钩子子程相关联的线程的标识符。
//如果为0,钩子子程与所有的线程关联,即为全局钩子。
//但这时,你钩子只能是放在DLL中。
);
函数成功则返回钩子子程的句柄,失败返回NULL。 我们用到的是WH_CALLWNDPROC类型的钩子,它使你可以监视发送到窗口过程的消息, 系统在消息发送到 接收窗口过程之前会调用你指定的WH_CALLWNDPROC Hook 子程,这样你就可以等它们自投罗网,然后就可以 对它们为所欲为了。 卸载钩子就简单多了,只需要调用UnhookWindowsHookEx即可,当然,我们还需要额外做一点清理工作:
void CMenuWndHook::UnInstallHook()
{
POSITION pos = m_WndMenuMap.GetStartPosition();
while (pos != NULL)
{
HWND hwnd;
CMenuWndHook *pMenuWndHook;
m_WndMenuMap.GetNextAssoc(pos, hwnd, pMenuWndHook);
delete pMenuWndHook;
pMenuWndHook= NULL;
}
m_WndMenuMap.RemoveAll();
if (m_hMenuHook != NULL)
{
::UnhookWindowsHookEx(m_hMenuHook);
}
}
在介绍如何安装钩子时,提到要一个钩子子程,这个子程必须按下面的格式声明,否则不能使用:
LRESULT CALLBACK WindowHook(int code, WPARAM wParam, LPARAM lParam); 函数名随意,同样把它声明为静态函数,下面各位注意了,我们的逮捕行动就是在这个函数中展开的:
LRESULT CALLBACK CMenuWndHook::WindowHook(int code, WPARAM wParam, LPARAM lParam)
{
//如果你安装的是WH_CALLWNDPROC类型的钩子的话,系统就会传递一个这个家伙的指针:
CWPSTRUCT* pStruct = (CWPSTRUCT*)lParam;
while (code == HC_ACTION)
{
HWND hWnd = pStruct->hwnd;
// 截获 WM_CREATE 消息, 为了保证不抓错"人",我们必须严格确定这是否是我们要抓的家伙,
// 这样我们就可以在它们刚出头就把它们逮住:
if(pStruct->message != WM_CREATE &&pStruct->message != 0x01E2)
{
break;
}
// 是否为菜单类 ----------------------------------------
TCHAR strClassName[10];
int Count = ::GetClassName(hWnd,strClassName,sizeof(strClassName) / sizeof(strClassName[0]));
// 再次确认它的身份(菜单窗口类的类名为"#32768",且为6个字符长):
if (Count != 6 || _tcscmp(strClassName, _T("#32768")) != 0 )
{
// 对不起,认错人了,pass :-)
break;
}
//是否已经被子类化------------------------------------
// 我们抓到一个之后,会给它用SetProp挂个牌(后面会介绍)
if(::GetProp(pStruct->hwnd, CoolMenu_oldProc) ! = NULL )
{
// 已经在编? pass.
break;
}
// 抓到一个,给它登记注册(这个函数我会在后面介绍), 而且不能登记失败, :)
VERIFY(AddWndHook(pStruct->hwnd) != NULL);
//下面该叫它去洗心革面了-----------------
//取得原来的窗口过程 ----------------------------------
WNDPROC oldWndProc = (WNDPROC)(long)::GetWindowLong(pStruct->hwnd, GWL_WNDPROC);
if (oldWndProc == NULL)
{
break;
}
ASSERT(oldWndProc != CoolMenuProc); //这个过程一样不能出错
// 保存到窗口的属性中 ----------------------------------
// 哈哈,给它打个记号吧 (SetProp API函数是用来给一个窗口加上一个属性的,
// RemoveProp 则是删除一个属性,GetProp 是取得一个属性的值)
// CoolMenu_oldProc 为一字符数组, 我在CPP文件的开头声明了它,表示你要
// 添加的属性名: const TCHAR CoolMenu_oldProc[]=_T("CoolMenu_oldProc");
// 这里保存的是它的原来的窗口过程,这种该随身带的东西还是让它自己拿着比较好
if (!SetProp(pStruct->hwnd,CoolMenu_oldProc, oldWndProc))
{
break;
}
// 子类化----------------------------------------------
// 这个不用我说了吧,这里我们用了偷梁换柱的方法,呵呵,这可是子类化的惯技了:
if (!SetWindowLong(pStruct->hwnd, GWL_WNDPROC,(DWORD)(ULONG)CoolMenuProc) )
{
//没有成功!!唉,就放过他吧,虽然忙了半天了,不过这种情况我想是不可能发生的!
::RemoveProp(pStruct->hwnd, CoolMenu_oldProc);
break;
}
}
// 这句可是绝对不能少的,叫那些闲杂人等该干什么就干什么去,不要?
// 嘿嘿,看你的程序怎么死吧!
return CallNextHookEx(m_hMenuHook, code, wParam, lParam);
}