您的位置:首页 > 其它

vc 利用钩子实现菜单阴影效果

2012-06-28 10:14 204 查看
  菜单也是一个窗口, 假如我们能得到它的窗口的句柄, 要实现像添加阴影这样的效果, 就不会很难了。

可惜我们根本找不到这个窗口是在哪里被创建的,也没办法很容易地取得它的窗口句柄,甚至几乎难以相信它是一个窗口,因为我实在找不到它的窗口句柄啊。经过对许多别人已经做好的类的源代码的"研究", 我终于找到了一个方法。那就是万能的钩子,如果说在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);

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: