HOOK函数(一)——进程内HOOK

　　什么是HOOK呢？其实很简单，HOOK就是对Windows消息进行拦截检查处理的一个函数。在Windows的消息机制中，当用户产生消息时，应用程序通过调用GetMessage函数取出消息，然后把消息放入到消息队列，再使用消息调度函数DispatchMessage函数讲消息调度给系统，Windows系统会调用创建窗口类时制定的窗口过程中进行次消息的处理。而HOOK函数的话，就可以对此消息进行拦截。经过此HOOK函数的处理后，再决定是屏蔽掉此消息，还是继续往下传递。至于为什么叫“钩子”，可能就是因为它可以像钩子一样把消息给钩住。话说四川话里面的“钩子”却是屁股的意思...哈哈，四川的同学读钩子肯定多了几份趣味

　　钩子函数的实现分为以下步骤：

　　1 安装钩子函数

　　安装钩子函数是通过函数SetWindowHookEx来实现的：

　　

HHOOK WINAPI SetWindowsHookExW(__in int idHook,  __in HOOKPROC lpfn,__in_opt HINSTANCE hmod,  __in DWORD dwThreadId);

　　此函数的第一个参数idHook指定将要安装的钩子过程的类型，如果我们安装键盘钩子，则设为WH_KEYBOARD，鼠标钩子则设为WH_MOUSE。 lpfn指向了相应的钩子函数，如果dwThreadId为0，或者指定了一个其他进程创建的线程之标识符，则lpfn必须指向一个位于某动态链接库中的钩子函数。如果为进程外钩子，hMod指向的是钩子函数所在的DLL的句柄,如果为进程内钩子，则设为NULL。dwThreadId指定与钩子过程相关的线程表示，为0表示与所有线程相关。函数成功返回所安装的钩子函数的句柄，否则返回NULL。PS：最后安装的钩子函数总是排在钩子链的最前面。

　　

　　2 添加全局变量

HHOOK g_hMouse = NULL;

HHOOK g_hKeyboard = NULL;

HWND g_hWnd = NULL;

     全局句柄g_hMouse、g_hKeyboard分别代表我们所要安装的鼠标钩子和键盘钩子过程的句柄，g_hWnd用来保存当前的窗口句柄。

　　3 声明钩子函数

　　如果想要监视鼠标消息，需要定义相应的鼠标钩子过程。同样，要监视键盘消息，也需要定义相应的键盘钩子过程。该钩子函数的形式必须如下：

LRESULT CALLBACK  HookProc(int nCode, WPARAM wParam, LPARAM lParam);

　　在当前钩子函数中处理完信息后，如果想把信息继续传递给下一个钩子函数，可以调用CallNextHookEx函数来实现：

LRESULT

WINAPI CallNextHookEx( __in_opt HHOOK hhk,   __in int nCode, __in WPARAM wParam，  __in LPARAM lParam

　　CallNextHookEx函数的第一个参数是钩子的句柄，就是我们调用SetWindowHookEx函数返回的句柄，其他几个参数和HookProc中是一样的。

　　4 卸载钩子函数

　　当我们不需要再使用钩子函数的时候，可以将其卸载，调用函数：

　　

BOOL WINAPI UnhookWindowsHookEx(  __in HHOOK hhk);

　　函数的唯一参数便为钩子函数的句柄。

　　说一千道一万，不如一个程序来的实在。下面便是一个实现进程内钩子函数的MFC例子。步骤如下：

　　1 新建一个名字CInnerHookDlg的MFC工程。

　　2 在CInnerHookDlgDlg.cpp中添加钩子的全局变量和当前句柄

　　

HHOOK g_hMouse = NULL;

HHOOK g_hKeyboard = NULL;

HWND g_hWnd = NULL;

　　3 然后再分别实现键盘和鼠标钩子函数,我们在键盘钩子函数中显示出当前按下去的键盘。而在鼠标钩子函数中，当我们双击右键时候，会显示出对话框。

//鼠标钩子函数

LRESULT CALLBACK MouseProc( int nCode, WPARAM wParam, LPARAM lParam )

{

...

}

    





//键盘钩子函数

LRESULT CALLBACK KeyBoardProc( int nCode, WPARAM wParam, LPARAM lParam )

{

...

}

　　4 在CCInnerHookDlgDlg的OnInitDialog()中设置钩子，添加如下代码

　　

//设置鼠标钩子

    g_hMouse = SetWindowsHookEx(WH_MOUSE, MouseProc, NULL,GetCurrentThreadId());



    //设置键盘钩子

    g_hKeyboard = SetWindowsHookEx(WH_KEYBOARD, KeyBoardProc, NULL, GetCurrentThreadId());

    

    //获取当前句柄

    g_hWnd = m_hWnd;

　　5 最后在CCInnerHookDlgDlg的OnSysCommand()中卸载钩子，添加如下代码：

UnhookWindowsHookEx(g_hMouse);

    UnhookWindowsHookEx(g_hKeyboard);

　　下面是所有的代码啦！

// CInnerHookDlgDlg.cpp : 实现文件

//



#include "stdafx.h"

#include "CInnerHookDlg.h"

#include "CInnerHookDlgDlg.h"



#ifdef _DEBUG

#define new DEBUG_NEW

#endif



HHOOK g_hMouse = NULL;

HHOOK g_hKeyboard = NULL;

HWND g_hWnd = NULL;





// 用于应用程序“关于”菜单项的 CAboutDlg 对话框



class CAboutDlg : public CDialog

{

public:

    CAboutDlg();



// 对话框数据

    enum { IDD = IDD_ABOUTBOX };



    protected:

    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持



// 实现

protected:

    DECLARE_MESSAGE_MAP()

};



CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)

{

}



void CAboutDlg::DoDataExchange(CDataExchange* pDX)

{

    CDialog::DoDataExchange(pDX);

}



BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)

END_MESSAGE_MAP()





// CCInnerHookDlgDlg 对话框

CCInnerHookDlgDlg::CCInnerHookDlgDlg(CWnd* pParent /*=NULL*/)

    : CDialog(CCInnerHookDlgDlg::IDD, pParent)

{

    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

}



void CCInnerHookDlgDlg::DoDataExchange(CDataExchange* pDX)

{

    CDialog::DoDataExchange(pDX);

}



BEGIN_MESSAGE_MAP(CCInnerHookDlgDlg, CDialog)

    ON_WM_SYSCOMMAND()

    ON_WM_PAINT()

    ON_WM_QUERYDRAGICON()

    //}}AFX_MSG_MAP

END_MESSAGE_MAP()



//鼠标钩子函数

LRESULT CALLBACK MouseProc( int nCode, WPARAM wParam, LPARAM lParam )

{

    LPMOUSEHOOKSTRUCT pMouseHook = (MOUSEHOOKSTRUCT*)(lParam);



    if (nCode >= 0)

    {

        //双击右键弹出当前窗口的窗口名

        if (wParam == WM_RBUTTONDBLCLK)

        {

            HWND hWnd = pMouseHook->hwnd;  //鼠标所在的窗口

            HWND hParent;



            //获取顶层窗口

            while(hWnd != NULL)

            {

                hParent = GetParent(hWnd);

                if (hParent == NULL)

                    break;

                hWnd = hParent;

            }



            if (hWnd != NULL)

            {

                //得到窗口名

                TCHAR strDlgName[MAX_PATH] = {0};

                GetWindowText(hWnd,strDlgName, MAX_PATH);

                AfxMessageBox( strDlgName );

            }

        }

    }



    return CallNextHookEx(g_hMouse,nCode, wParam, lParam);

}



//键盘钩子函数

LRESULT CALLBACK KeyBoardProc( int nCode, WPARAM wParam, LPARAM lParam )

{

    LRESULT lReturnValue;

    lReturnValue = CallNextHookEx(g_hKeyboard, nCode, wParam, lParam);



    //键盘按键是否已经松动

    if ( (lParam & 0x80000000) && (HC_ACTION == nCode))

    {

        char c = (char) lParam;

        CString strText(_T("按下了键"));

        strText += c;

        AfxMessageBox(strText);  //显示当前的按键

    }



    //如果按下F4，则退出程序

    if (wParam == VK_F4)

    {

        SendMessage(g_hWnd, WM_CLOSE,0,0);  

    }



    return lReturnValue;

}





// CCInnerHookDlgDlg 消息处理程序



BOOL CCInnerHookDlgDlg::OnInitDialog()

{

    CDialog::OnInitDialog();



    // 将“关于...”菜单项添加到系统菜单中。



    // IDM_ABOUTBOX 必须在系统命令范围内。

    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);

    ASSERT(IDM_ABOUTBOX < 0xF000);



    CMenu* pSysMenu = GetSystemMenu(FALSE);

    if (pSysMenu != NULL)

    {

        CString strAboutMenu;

        strAboutMenu.LoadString(IDS_ABOUTBOX);

        if (!strAboutMenu.IsEmpty())

        {

            pSysMenu->AppendMenu(MF_SEPARATOR);

            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);

        }

    }



    // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动

    //  执行此操作

    SetIcon(m_hIcon, TRUE);            // 设置大图标

    SetIcon(m_hIcon, FALSE);        // 设置小图标



    // TODO: 在此添加额外的初始化代码



    //设置鼠标钩子

    g_hMouse = SetWindowsHookEx(WH_MOUSE, MouseProc, NULL,GetCurrentThreadId());



    //设置键盘钩子

    g_hKeyboard = SetWindowsHookEx(WH_KEYBOARD, KeyBoardProc, NULL, GetCurrentThreadId());

    

    //获取当前句柄

    g_hWnd = m_hWnd;



    return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE

}

void CCInnerHookDlgDlg::OnSysCommand(UINT nID, LPARAM lParam)

{

    if ((nID & 0xFFF0) == IDM_ABOUTBOX)

    {

        CAboutDlg dlgAbout;

        dlgAbout.DoModal();

    }

    else

    {

        CDialog::OnSysCommand(nID, lParam);

    }



    UnhookWindowsHookEx(g_hMouse); UnhookWindowsHookEx(g_hKeyboard); }



// 如果向对话框添加最小化按钮，则需要下面的代码

//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，

//  这将由框架自动完成。



void CCInnerHookDlgDlg::OnPaint()

{

    if (IsIconic())

    {

        CPaintDC dc(this); // 用于绘制的设备上下文



        SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);



        // 使图标在工作区矩形中居中

        int cxIcon = GetSystemMetrics(SM_CXICON);

        int cyIcon = GetSystemMetrics(SM_CYICON);

        CRect rect;

        GetClientRect(&rect);

        int x = (rect.Width() - cxIcon + 1) / 2;

        int y = (rect.Height() - cyIcon + 1) / 2;



        // 绘制图标

        dc.DrawIcon(x, y, m_hIcon);

    }

    else

    {

        CDialog::OnPaint();

    }

}



//当用户拖动最小化窗口时系统调用此函数取得光标

//显示。

HCURSOR CCInnerHookDlgDlg::OnQueryDragIcon()

{

    return static_cast<HCURSOR>(m_hIcon);

}