DLL安装钩子实例

生成钩子dll

头文件

#ifndef _KBHOOK_H_
#define _KBHOOK_H_
#include <Windows.h>
#ifdef _KEYHOOKLIB
#define _KEYHOOKLIB_EXPORTS _declspec(dllexport)
#else
#define _KEYHOOKLIB_EXPORTS _declspec(dllimport)
#endif
#define HK_KEY WM_USER + 101

BOOL _KEYHOOKLIB_EXPORTS WINAPI SetKeyHook(BOOL bInstall, DWORD dwThread = 0, HWND hWndCall = NULL);

#endif

cpp

// KBHook.cpp : 定义 DLL 应用程序的导出函数。
//

#include "stdafx.h"
#define _KEYHOOKLIB
#include "KBHool.h"

#pragma data_seg("KBShared")
HWND    g_hWndCaller = NULL;
HHOOK   g_hHook = NULL;
#pragma  data_seg()


HMODULE WINAPI ModuleFromAddress(LPCVOID pv)
{
    MEMORY_BASIC_INFORMATION mbi;
    if (::VirtualQuery(pv, &mbi, sizeof(MEMORY_BASIC_INFORMATION)) !=0)
    {
        return (HMODULE)mbi.AllocationBase;
    }
    return NULL;
}

LRESULT CALLBACK KeyHookProc(int ncode , WPARAM wParam, LPARAM lParam)
{
    if (ncode < 0 || ncode == HC_NOREMOVE)
    {
        return CallNextHookEx(g_hHook, ncode, wParam, lParam);
    }
    if (lParam & 0x40000000)
    {
        return CallNextHookEx(g_hHook, ncode, wParam, lParam);
    }
    PostMessage(g_hWndCaller, HK_KEY, wParam, lParam);
    return CallNextHookEx(g_hHook, ncode, wParam, lParam);
}
BOOL WINAPI SetKeyHook(BOOL bInstall, DWORD dwThread /* = 0 */, HWND hWndCall /* = NULL */)
{
    BOOL bOK = FALSE;
    g_hWndCaller = hWndCall;
    if (bInstall)
    {
        g_hHook = ::SetWindowsHookEx(WH_KEYBOARD_LL,KeyHookProc, ModuleFromAddress(KeyHookProc), dwThread);
        int LastError = GetLastError();
        bOK = (g_hHook != NULL);
    }
    else
    {
        bOK = UnhookWindowsHookEx(g_hHook);
        g_hHook = NULL;
    }
    return bOK;
}

 

 

使用的例子

// UseKeyHookDlg.h : 头文件
//

#pragma once
#include "../KBHook/KBHool.h"

// CUseKeyHookDlg 对话框
class CUseKeyHookDlg : public CDialogEx
{
// 构造
public:
	CUseKeyHookDlg(CWnd* pParent = NULL);	// 标准构造函数

// 对话框数据
	enum { IDD = IDD_USEKEYHOOK_DIALOG };

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


// 实现
protected:
	HICON m_hIcon;

	// 生成的消息映射函数
	virtual BOOL OnInitDialog();
	afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
	afx_msg void OnPaint();
    afx_msg LRESULT OnHookKey(WPARAM wParam, LPARAM lParam);
	afx_msg HCURSOR OnQueryDragIcon();
	DECLARE_MESSAGE_MAP()
public:
    afx_msg void OnBnClickedCancel();
};

 

// UseKeyHookDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "UseKeyHook.h"
#include "UseKeyHookDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


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

class CAboutDlg : public CDialogEx
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

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

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CUseKeyHookDlg 对话框




CUseKeyHookDlg::CUseKeyHookDlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(CUseKeyHookDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CUseKeyHookDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CUseKeyHookDlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
    ON_MESSAGE(HK_KEY, OnHookKey)
    ON_BN_CLICKED(IDCANCEL, &CUseKeyHookDlg::OnBnClickedCancel)
END_MESSAGE_MAP()


// CUseKeyHookDlg 消息处理程序

BOOL CUseKeyHookDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

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

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

    if (!SetKeyHook(TRUE, 0, m_hWnd))
    {
        MessageBox(_T("安装钩子失败"));
    }
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

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

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

void CUseKeyHookDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialogEx::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CUseKeyHookDlg::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
	{
		CDialogEx::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CUseKeyHookDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

LRESULT CUseKeyHookDlg::OnHookKey(WPARAM wParam, LPARAM lParam)
{   
    KBDLLHOOKSTRUCT *kblp=(KBDLLHOOKSTRUCT*)lParam;

    DWORD dwvk = kblp->vkCode;
    DWORD dwMsg = 1;
    dwMsg += kblp->scanCode << 16;
    dwMsg += kblp->flags << 24;
    char KeyName[20] = {0};
    ::GetKeyNameText(dwMsg, KeyName, 20);
    CString NewItem;
    NewItem.Format("用户按键: %s\r\n", KeyName);
    CString strEdit;
    GetDlgItem(IDC_STATIC)->GetWindowText(strEdit);
    strEdit += NewItem;
    GetDlgItem(IDC_STATIC)->SetWindowText(NewItem);
    MessageBeep(MB_OK);
    return TRUE;
}


void CUseKeyHookDlg::OnBnClickedCancel()
{
    // TODO: 在此添加控件通知处理程序代码
    SetKeyHook(FALSE);
    CDialogEx::OnCancel();
}