mousehook.cpp

VisualC高级编程技术精粹.rar

// Mousehook.cpp : Defines the initialization routines for the DLL.
//

#include "stdafx.h"
#include <afxdllx.h>
#include "Mousehook.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#pragma data_seg("mydata")
     HWND glhPrevTarWnd = NULL; //上次鼠标所指的窗口句柄 
     HWND glhDisplayWnd = NULL; //显示目标窗口标题编辑框的句柄
	 HHOOK glhHook = NULL; //安装的鼠标勾子句柄 
	 HINSTANCE glhInstance = NULL; //DLL实例句柄 
#pragma data_seg()

HWND XYZWindowFromPoint(HWND hwndParent,      // handle to parent window
                        POINT point,          // structure with point coordinates
                        UINT uFlags  = CWP_SKIPINVISIBLE       // skip options
					   );

static AFX_EXTENSION_MODULE MousehookDLL = { NULL, NULL };

extern "C" int APIENTRY
DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
{
	// Remove this if you use lpReserved
	UNREFERENCED_PARAMETER(lpReserved);

	if (dwReason == DLL_PROCESS_ATTACH)
	{
		TRACE0("MOUSEHOOK.DLL Initializing!\n");
		
		// Extension DLL one-time initialization
		if (!AfxInitExtensionModule(MousehookDLL, hInstance))
			return 0;

		// Insert this DLL into the resource chain
		// NOTE: If this Extension DLL is being implicitly linked to by
		//  an MFC Regular DLL (such as an ActiveX Control)
		//  instead of an MFC application, then you will want to
		//  remove this line from DllMain and put it in a separate
		//  function exported from this Extension DLL.  The Regular DLL
		//  that uses this Extension DLL should then explicitly call that
		//  function to initialize this Extension DLL.  Otherwise,
		//  the CDynLinkLibrary object will not be attached to the
		//  Regular DLL's resource chain, and serious problems will
		//  result.

		new CDynLinkLibrary(MousehookDLL);
		glhInstance=hInstance; //插入保存DLL实例句柄
	}
	else if (dwReason == DLL_PROCESS_DETACH)
	{
		TRACE0("MOUSEHOOK.DLL Terminating!\n");
		// Terminate the library before destructors are called
		AfxTermExtensionModule(MousehookDLL);
	}
	return 1;   // ok   
}

// 钩子函数的实现： 
LRESULT WINAPI MouseProc(int nCode,WPARAM wparam,LPARAM lparam)
{
	LPMOUSEHOOKSTRUCT pMouseHook = (MOUSEHOOKSTRUCT FAR *)lparam;
	if (nCode >= 0)
	{
		HWND glhTargetWnd = XYZWindowFromPoint(NULL,pMouseHook->pt);    
		if (glhTargetWnd != glhPrevTarWnd)
		{
			char szCaption[256];

			// 取目标窗口标题
			GetWindowText(glhTargetWnd,szCaption,100);
			
			// 保存目标窗口
			if (IsWindow(glhDisplayWnd))
			{
				SendMessage(glhDisplayWnd,
				            WM_SETTEXT,
							0,
							(LPARAM)(LPCTSTR)szCaption);
			}
			glhPrevTarWnd = glhTargetWnd;			
		}
	}
	
	// 继续传递消息
	return CallNextHookEx(glhHook,nCode,wparam,lparam);
}

CMouseHook::CMouseHook()
{

}

CMouseHook::~CMouseHook()
{
}

// 安装钩子并设定接收显示窗口句柄
BOOL CMouseHook::StartHook(HWND hWnd)
{
	BOOL bResult = FALSE;

	glhHook = SetWindowsHookEx(WH_MOUSE,
		                       MouseProc,
		                       glhInstance,
		                       0);
	
	if (glhHook != NULL)
	{
		bResult = TRUE;
	}
	
	// 设置显示目标窗口标题编辑框的句柄
	glhDisplayWnd = hWnd;
	
	return bResult; 
}

// 卸载钩子
BOOL CMouseHook::StopHook() 
{
	BOOL bResult = FALSE;
	if (glhHook)
	{
		bResult = UnhookWindowsHookEx(glhHook);
		if (bResult)
		{
			glhPrevTarWnd = NULL;
			glhDisplayWnd = NULL;
			glhHook = NULL;
		}
	}

	return bResult;
}

// 返回光标所在的子窗口句柄
HWND XYZWindowFromPoint(HWND hwndParent,   // handle to parent window
                        POINT point,       // structure with point coordinates
                        UINT uFlags        // skip options
)
{
	// 返回光标所在的子窗口句柄
	if (hwndParent != NULL)
	{
		return ::ChildWindowFromPointEx(hwndParent, point, uFlags);
	}
	
	// Find the smallest "window" still containing the point
	// Doing this prevents us from stopping at the first window containing the point
	RECT rect, rectSearch;
	HWND pWnd, hWnd, hSearchWnd;

	hWnd = ::WindowFromPoint(point);// 得到光标(point)所在点的窗口句柄
	if (hWnd != NULL)
	{
		// Get the size and parent for compare later
		::GetWindowRect(hWnd, &rect);	// 得到整个窗口在屏幕上的矩形框位置
		pWnd = ::GetParent(hWnd);       // 得到父窗口句柄

		// We only search further if the window has a parent
		if (pWnd != NULL)
		{
			// Search from the window down in the Z-Order
			hSearchWnd = hWnd;
			do
			{
				//如果找不到窗口，该函数就会返回NULL
				hSearchWnd = ::GetWindow(hSearchWnd, GW_HWNDNEXT);
				
				// Does the search window also contain the point, have the same parent, and is visible?
				::GetWindowRect(hSearchWnd, &rectSearch);
				if (::PtInRect(&rectSearch, point) && ::GetParent(hSearchWnd) 
					== pWnd && ::IsWindowVisible(hSearchWnd))
				{
					// 比较面积
					if (((rectSearch.right - rectSearch.left) * (rectSearch.bottom - rectSearch.top)) 
						< ((rect.right - rect.left) * (rect.bottom - rect.top)))
					{
						// Found new smaller window, update compare window
						hWnd = hSearchWnd;
						::GetWindowRect(hWnd, &rect);
					}
				}
			}while (hSearchWnd != NULL);			
		}
	}

	return hWnd;
}