本文通过一个DEMO来演示如何HOOK桌面消息。
代码详见:
需求:截获桌面窗口鼠标单击事件,解析所选中的桌面 Item,并将解析后的 item 信息发送给主调程序,并将信息显示在一个窗口上面。如下图:
思路:
1. 确定HOOK的类型。很明显,这一个进程外的HOOK,我们的应用程序DesktopCaptor2.exe 需要捕获 Explorer.exe 这个进程的桌面窗口所在的线程的消息。因此,需要将HOOK过程放在一个独立的DLL 中去,然后使用 SetWindowsHookEx 将HOOK过程安装到HOOK链中去。
2. 如何解析点击的桌面 Item 信息?这个其实也比较好做,由于桌面窗口本身是一个 listview 控件(不解释),因此,我们可以通过 listview 拿到桌面窗口的简单信息。
3. 如何将解析后的桌面 Item 信息发送给主调程序,让它弹出一个窗口,并显示桌面 Item 信息?我们这里采用WM_COPYDATA 将从桌面进程获取到信息发送到我们的应用程序。
工程目录如下:
在主调程序(DesktopCaptor2.exe)是一个简单Win32 Dialog的程序。在这个程序中,我们干了两件事情:
1. 调用 DekstopHook 工程中的 DesktopHook.h 中的两个导出函数,通过这两个函数,对 HOOK 过程安装和卸载。
2. 接收来自于Explorer.exe 进程发送的 WM_COPYDATA 消息,还原桌面 Item 数据,并将在点击桌面 Item 的位置弹出一个窗口出来,显示 Item 信息。至于这个窗口如何制作,我就不再描述了。
以下为部分代码:
- #include "CommonDef.h"
- #include "DesktopHook.h"
- #include "FloatWin.h"
-
- const UINT WM_DESKTOP_CLICKED_ITEM = RegisterWindowMessage(L"WM_DESKTOP_CLICKED_ITEM");
-
- BOOL g_isCaptured = FALSE;
- CFloatWin* g_floatWin = NULL;
-
- INT_PTR WINAPI DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) ;
-
- int WINAPI _tWinMain(HINSTANCE hinstExe, HINSTANCE, PTSTR pszCmdLine, int)
- {
- HWND hwnd = FindWindow(TEXT("#32770"), TEXT("DesktopCaptor2"));
- if (IsWindow(hwnd))
- {
-
- MessageBox(GetForegroundWindow(), L"An instance is already running", L"Error", MB_ICONERROR);
- }
- else
- {
- DialogBox(hinstExe, MAKEINTRESOURCE(IDD_DESKTOP_CAPTOR), NULL, DlgProc);
- }
- return(0);
- }
-
- INT_PTR CALLBACK DlgProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
- {
- UNREFERENCED_PARAMETER(lParam);
- switch (message)
- {
- case WM_INITDIALOG:
- {
-
- SendMessage(hDlg, WM_SETICON, ICON_BIG, (LPARAM)
- LoadIcon((HINSTANCE) GetWindowLongPtr(hDlg, GWLP_HINSTANCE),
- MAKEINTRESOURCE(IDI_DESKTOPCAPTOR2)));
-
-
- int nScreenWidth = ::GetSystemMetrics(SM_CXSCREEN);
- int nScreenHeight = ::GetSystemMetrics(SM_CYSCREEN);
- RECT rect = { 0 };
- GetWindowRect(hDlg, &rect);
- SetWindowPos(
- hDlg,
- HWND_TOP,
- nScreenWidth - (rect.right - rect.left),
- 0,
- 0, 0,
- SWP_NOSIZE);
-
- g_floatWin = CFloatWin::getInstance();
- }
- return (INT_PTR)TRUE;
-
- case WM_COMMAND:
- {
- UINT wmId = LOWORD(wParam);
- UINT wmEvent = HIWORD(wParam);
-
- switch (wmId)
- {
- case IDOK:
- case IDCANCEL:
- EndDialog(hDlg, LOWORD(wParam));
- return (INT_PTR)TRUE;
- case IDC_START_CAPTOR:
- if (FALSE == g_isCaptured)
- {
- g_isCaptured = CreateDesktopEventCaptor(hDlg);
- }
- break;
- case IDC_STOP_CAPTOR:
- if (TRUE == g_isCaptured)
- {
- CloseDesktopEventCaptor();
- g_isCaptured = FALSE;
- }
- break;
- default:
- return DefWindowProc(hDlg, message, wParam, lParam);
- }
- }
- break;
- case WM_COPYDATA:
- {
- COPYDATASTRUCT* pCopyData = (COPYDATASTRUCT*)lParam;
- if (pCopyData->dwData == WM_DESKTOP_CLICKED_ITEM)
- {
- DesktopItemData itemData(*(DesktopItemData*)pCopyData->lpData);
- g_floatWin->ShowWindow(TRUE, &itemData);
- }
- }
- break;
- }
- return (INT_PTR)FALSE;
- }
我们重点说明一下DekstopHookDLL中的内容。
DesktopHook.h 中定义的是一些导出接口,代码如下:
- #ifndef _DESKTOPHOOK_H_
- #define _DESKTOPHOOK_H_
-
-
-
-
-
- #ifdef DESKTOPHOOK_EXPORTS
- #define DESKTOPHOOK_API __declspec(dllexport)
- #else
- #define DESKTOPHOOK_API __declspec(dllimport)
- #endif
-
-
- typedef struct DESKTOPHOOK_API _DesktopItemData
- {
- POINT point;
- WCHAR szName[MAX_PATH];
- int nIndex;
-
- _DesktopItemData()
- {
- ZeroMemory(szName, MAX_PATH);
- point.x = point.y = 0;
- nIndex = -1;
- }
-
- _DesktopItemData(POINT pt, WCHAR* pszName, int nIx)
- {
- if (NULL != pszName)
- {
- point.x = pt.x;
- point.y = pt.y;
-
- ZeroMemory(szName, MAX_PATH);
- _tcscpy_s(szName, MAX_PATH, pszName);
-
- nIndex = nIx;
- }
- }
-
- _DesktopItemData(const _DesktopItemData& itemDataRef)
- {
- point.x = itemDataRef.point.x;
- point.y = itemDataRef.point.y;
-
- ZeroMemory(szName, MAX_PATH);
- _tcscpy_s(szName, MAX_PATH, itemDataRef.szName);
- nIndex = itemDataRef.nIndex;
- }
-
- _DesktopItemData& operator = (const _DesktopItemData& itemDataRef)
- {
- point.x = itemDataRef.point.x;
- point.y = itemDataRef.point.y;
-
- ZeroMemory(szName, MAX_PATH);
- _tcscpy_s(szName, MAX_PATH, itemDataRef.szName);
- nIndex = itemDataRef.nIndex;
-
- return *this;
- }
-
- } DesktopItemData, *LPDesktopItemData;
-
- EXTERN_C DESKTOPHOOK_API BOOL CreateDesktopEventCaptor(HWND hNotifierhWnd);
- EXTERN_C DESKTOPHOOK_API void CloseDesktopEventCaptor();
-
-
-
-
-
- #endif // _DESKTOPHOOK_H_
其中:
1. DesktopItemData 结构体是用来存放解析桌面 Item 的数据。
2. CreateDesktopEventCaptor 函数是用来安装 HOOK;
3. CloseDesktopEventCaptor 函数是用来卸载 HOOK;
以下是 DesktopHook.cpp 中的实现代码:
-
-
-
- #include "stdafx.h"
- #include "DesktopHook.h"
- #include "DesktopItem.h"
-
- #pragma data_seg("SHARED_DATA")
- HWND g_hNotifierWnd = NULL;
- HHOOK g_hPostMsgHook = NULL;
- WCHAR g_szBuf[MAX_PATH] = {0};
- #pragma data_seg()
- #pragma comment(linker, "/SECTION:SHARED_DATA,RWS")
-
-
- const UINT WM_DESKTOP_CLICKED_ITEM = RegisterWindowMessage(L"WM_DESKTOP_CLICKED_ITEM");
-
- HMODULE g_hModule;
- HWND g_hDesktopWnd = NULL;
-
-
-
- BOOL g_bDoubleClick = FALSE;
- UINT_PTR g_timerID = 0;
- POINT g_clickPt;
- CDesktopItem g_singleClickDesktopItem = CDesktopItem::Empty;
- DesktopItemData g_desktopItemData;
-
-
- static HWND FindShellWindow();
- static LRESULT CALLBACK GetMsgProc(int code,WPARAM wParam,LPARAM lParam);
- static VOID CALLBACK TimerProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime);
-
- EXTERN_C DESKTOPHOOK_API BOOL CreateDesktopEventCaptor(HWND hNotifierhWnd)
- {
- g_hDesktopWnd = FindShellWindow();
- if (NULL == g_hDesktopWnd)
- {
- OutputDebugString(L"Can not find desktop window handle");
- return FALSE;
- }
-
- g_hNotifierWnd = hNotifierhWnd;
-
-
- DWORD targetThreadid = ::GetWindowThreadProcessId(g_hDesktopWnd, NULL);
-
- g_hPostMsgHook = ::SetWindowsHookEx(WH_GETMESSAGE, &GetMsgProc, g_hModule, targetThreadid);
-
- if (g_hPostMsgHook != NULL)
- {
- return TRUE;
- }
-
- return FALSE;
- }
-
- EXTERN_C DESKTOPHOOK_API void CloseDesktopEventCaptor()
- {
- if (g_hPostMsgHook != NULL)
- {
- ::UnhookWindowsHookEx(g_hPostMsgHook);
- }
- }
-
- HWND FindShellWindow()
- {
-
-
- UINT uFindCount = 0;
- HWND hSysListView32Wnd = NULL;
- while (NULL == hSysListView32Wnd && uFindCount < 10)
- {
- HWND hParentWnd = ::GetShellWindow();
- HWND hSHELLDLL_DefViewWnd = ::FindWindowEx(hParentWnd, NULL, L"SHELLDLL_DefView", NULL);
- hSysListView32Wnd = ::FindWindowEx(hSHELLDLL_DefViewWnd, NULL, L"SysListView32", L"FolderView");
-
- if (NULL == hSysListView32Wnd)
- {
- hParentWnd = ::FindWindowEx(NULL, NULL, L"WorkerW", L"");
- while((!hSHELLDLL_DefViewWnd) && hParentWnd)
- {
- hSHELLDLL_DefViewWnd = ::FindWindowEx(hParentWnd, NULL, L"SHELLDLL_DefView", NULL);
- hParentWnd = FindWindowEx(NULL, hParentWnd, L"WorkerW", L"");
- }
- hSysListView32Wnd = ::FindWindowEx(hSHELLDLL_DefViewWnd, 0, L"SysListView32", L"FolderView");
- }
-
- if (NULL == hSysListView32Wnd)
- {
- Sleep(1000);
- uFindCount++;
- }
- else
- {
- break;
- }
- }
-
- return hSysListView32Wnd;
- }
-
-
- LRESULT CALLBACK GetMsgProc(int code, WPARAM wParam, LPARAM lParam)
- {
- MSG* pMsg = (MSG*)lParam;
- if( NULL != pMsg && pMsg->hwnd != NULL )
- {
- switch(pMsg->message)
- {
- case WM_MOUSEMOVE:
- {
-
- }
- break;
- case WM_LBUTTONUP:
- {
-
- if (g_bDoubleClick)
- {
- OutputDebugString(L"g_bDoubleClick == TRUE");
- g_singleClickDesktopItem = CDesktopItem::Empty;
- g_bDoubleClick = FALSE;
- ::KillTimer(NULL, g_timerID);
- }
- else
- {
- OutputDebugString(L"g_bDoubleClick == FALSE");
- ::KillTimer(NULL, g_timerID);
- g_clickPt.x = pMsg->pt.x;
- g_clickPt.y = pMsg->pt.y;
- g_singleClickDesktopItem = pMsg;
- g_timerID = ::SetTimer(NULL, 1, ::GetDoubleClickTime(), TimerProc);
- }
- }
- break;
- case WM_LBUTTONDBLCLK:
- g_bDoubleClick = TRUE;
- break;
- }
- }
-
- return ::CallNextHookEx(g_hPostMsgHook, code, wParam, lParam);
- }
-
- VOID CALLBACK TimerProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime)
- {
- UNREFERENCED_PARAMETER(hwnd);
- UNREFERENCED_PARAMETER(uMsg);
- UNREFERENCED_PARAMETER(idEvent);
- UNREFERENCED_PARAMETER(dwTime);
- if (g_timerID == idEvent &&
- g_bDoubleClick == FALSE &&
- g_singleClickDesktopItem != CDesktopItem::Empty)
- {
- OutputDebugString(L"SendMessageTimeout Process begin");
-
- wstring strName = g_singleClickDesktopItem.GetItemName();
-
-
-
- ZeroMemory(&g_desktopItemData, sizeof(DesktopItemData));
- g_desktopItemData.point.x = g_clickPt.x;
- g_desktopItemData.point.y = g_clickPt.y;
- _tcscpy_s(g_desktopItemData.szName, MAX_PATH, strName.c_str());
- g_desktopItemData.nIndex = g_singleClickDesktopItem.GetItemIndex();
-
- COPYDATASTRUCT copyData;
- copyData.dwData = (UINT_PTR)WM_DESKTOP_CLICKED_ITEM;
- copyData.cbData = sizeof(DesktopItemData);
- copyData.lpData = &g_desktopItemData;
- DWORD_PTR rt = -1;
- SetActiveWindow(g_hNotifierWnd);
- SetForegroundWindow(g_hNotifierWnd);
-
- SendMessage(g_hNotifierWnd, WM_COPYDATA, (WPARAM)g_hDesktopWnd, (LPARAM)©Data);
-
- OutputDebugString(L"SendMessageTimeout Process end");
- }
- ::KillTimer(NULL, g_timerID);
- }
说明:在该实现中,CreateDesktopEventCaptor 函数与 CloseDesktopEventCaptor 函数被 DesktopCaptor2.exe 进程调用,此时,DesktopHook.dll 会被加载到 DesktopCaptor2.exe 所在进程,当使用 CreateDesktopEventCaptor 函数后,HOOK 过程函数 GetMsgProc 会被安装到 Explorer.exe 进程中去,此时,DesktopHook.dll 会被加载到 Explorer.exe 中去。当用鼠标单击桌面 Item 时,消息将被传递到 GetMsgProc 中去,然后,在这个函数中发送一个 WM_COPYDATA 消息给DesktopCaptor2 的窗口,实际上,这相当于是 Explorer.exe 进程发送的 WM_COPYDATA 消息到 DesktopCaptor2.exe 进程 。需要注意一点的是,在 GetMsgProc 中发送消息的时候,g_hNotifierWnd 必须要设置成为共享数据,因为它是DesktopCaptor2.exe 进程在调用 DesktopHook.dll 的 CreateDesktopEventCaptor 函数的时候被设置的,要想在Explorer.exe进程中继续有效,需要将之设置为 DLL共享数据。
另外,这里还有另外两个问题:
- 如何找到 Desktop 窗口句柄? 通过Spy++,我们得到:
其中,蓝色部分就是我们的桌面窗口句柄。我们的 FindShellWindow 函数就是为了干这个事情,但是有时候,这个函数会失败,因此,我们最多循环10次去查找桌面窗口句柄。
- WH_GETMESSAGE 类型的消息 HOOK 只能钩住使用 PostMessage 方式发送的消息,这点很重要,否则,如果是以SendMessage发送的方式,则需要使用 WH_CALLWNDPROC 或者 WH_CALLWNDPROCRET 的 HOOK 方式。
- 如何解析点击桌面的 Item 信息?
实际上,由于桌面窗口本身是一个 ListView 控件,通过ListView 的控件的API,我们便能够拿到相关的信息(当然这里只是一个简单的信息,深层次的信息还需要深掘)。
首先,通过下面的函数,能够拿到选中的 Item 的 ID,其中 hwnd 就是桌面窗口句柄。
UINT ListView_GetSelectedCount(
HWND hwnd
);
其次,通过下面的函数,传入选中的 Item 的 ID,我们就能拿到 Item 对应的文本。
void ListView_GetItemText(
HWND hwnd,
int iItem,
int iSubItem,
LPTSTR pszText,
int cchTextMax
);
在这里,我使用了一个 CDesktopItem 类来专门干这个事情。
具体请详见附录代码。
至此,简单的代码讲解就结束了。
下面说一下如何调试。
因为本例子是进程外的HOOK,因此调试起来有很多不方便的地方。调试的难处在于如何调试HOOK过程函数。先说一个简单的例子,可能大家经常会遇到这种情况:假如一个Solution下有ProjectA和 Project B,它们都是exe,但涉及到相互发消息,有人就会打开2个Visual Studio,同时进行调试,这样的确可以做到,但总感觉不太方便。实际上,在同一个Visual Studio中,是可以同时调试多个程序的。对于刚刚这种情况,只需要在每个Project A 和 ProjectB 上面分别右击->Debug->Start new instance 即可。
然而,对于本例,这样做是不行的,因为 DesktopHook 是一个DLL工程,本身是无法进行独立调试的。因此,要想同时调试 DesktopCaptor2 工程和 DesktopHook 工程,需要按如下操作:
- 将DesktopHook工程设置为默认启动的工程。如图:
- 将 DesktopCaptor2 工程通过 右击->Debug->Start new instance 启动起来,点击 Start Desktop Captor 按钮启动HOOK,将DesktopHook .dll 注入到进程Explorer.exe中去。
- 将 DesktopHook 工程到附加(Attach)到Explorer.exe进程中去。如图:
点击 Tools->Attachto Process...
然后找到Explorer.exe,点击Attach按钮即可。
通过上面的这种方式,我们就能够很简单的在一个工程中,调试两个不同的进程的程序。这时,我们将断点打在DesktopHook 工程的函数 GetMsgProc中,并在DesktopCaptor2 工程 WM_COPYDATA 内部打上断点,发现点击桌面图标时,断点会走到GetMsgProc内部,当发送完消息后,就能走到DesktopCaptor2 工程 WM_COPYDATA 内部。。
另外,你调试的时候,要注意一下当前用户的权限以及Visual Studio的权限。如果当前用户是管理员组用户,而你的VisualStudio是以管理员启动进来的,那么DesktopCaptor2.exe将也是管理员权限,但此时,Explorer.exe却是普通权限,此时,在GetMsgProc内部发送WM_COPYDATA是会出现问题的,因为不能向高权限进程发送消息。
阅读(1439) | 评论(0) | 转发(0) |