hook.cpp

#include "hook.h"
#include<QDebug>
 /// 静态变量定义
QMutex XGlobalHook::mutex;
QSharedPointer<XGlobalHook> XGlobalHook::hook_instance;
/// 全局变量定义
HINSTANCE g_hInstance = Q_NULLPTR;
HHOOK g_hMouseHook = Q_NULLPTR;  //鼠标钩子 句柄
HHOOK g_hKeyHook = Q_NULLPTR;    //键盘钩子 句柄

 /// 全局函数定义
 // DllMain
BOOL WINAPI DllMain(
     HINSTANCE hinstDLL,  // DLL模块的句柄
     DWORD fdwReason,     // 调用的情况
     LPVOID lpReserved    // reserved
     );

 // 鼠标钩子处理函数
LRESULT CALLBACK MouseProc(int nCode,       // 钩子码
                            WPARAM wParam,   // 消息ID
                            LPARAM lParam    // 鼠标坐标
                            );
 // 键盘钩子处理函数
LRESULT CALLBACK KeyboardProc(int nCode,            // 钩子码
                               WPARAM wParam,        // 虚拟键码
                               LPARAM lParam         // 键击信息
                               );


XGlobalHook::~XGlobalHook()
{
   uninstallMouseHook();
   uninstallKeyHook();
}

QSharedPointer<XGlobalHook>&XGlobalHook::instance()
{
   if (hook_instance.isNull())
   {
     QMutexLocker mutexLocker(&mutex);
     if (hook_instance.isNull())
       hook_instance = QSharedPointer<XGlobalHook>(new XGlobalHook());
   }
   return hook_instance;
}

bool XGlobalHook::installMouseHook()
{
   g_hMouseHook =SetWindowsHookEx(WH_MOUSE_LL, (HOOKPROC)MouseProc, g_hInstance, 0);
   return Q_LIKELY(g_hMouseHook);
}

bool XGlobalHook::uninstallMouseHook()
{
   if (Q_UNLIKELY(!g_hMouseHook)) {
     return true;
   }

   bool success = false;
   success = UnhookWindowsHookEx(g_hMouseHook);
   if (success) {
     g_hMouseHook = Q_NULLPTR;
   }
   return success;
}

bool XGlobalHook::isMouseHookRunning() const
{
   return Q_LIKELY(g_hMouseHook);
}

bool XGlobalHook::installKeyHook()
{
   g_hKeyHook =SetWindowsHookEx(WH_KEYBOARD_LL, (HOOKPROC)KeyboardProc, g_hInstance, 0);
   return Q_LIKELY(g_hKeyHook);
}

bool XGlobalHook::uninstallKeyHook()
{
   if (Q_UNLIKELY(!g_hKeyHook)) {
     return true;
   }

   bool success = false;
   success = UnhookWindowsHookEx(g_hKeyHook);
   if (success) {
     g_hKeyHook = Q_NULLPTR;
   }
   return success;
}

bool XGlobalHook::isKeyHookRunning() const
{
   return Q_LIKELY(g_hKeyHook);
}
 /// DLL 钩子接口
#if defined HOOK_LIBRARY
void XGlobalHook::onMouseEvent(int type, PMOUSEHOOKSTRUCT pMouseHookStruct, bool* is_shield)
{
   emit mouseEvent((button_type)type, pMouseHookStruct, is_shield);
}

void XGlobalHook::onKeyEvent(PKBDLLHOOKSTRUCT pKeyHookStruct)
{
   emit keyEvent(pKeyHookStruct);
}

#endif

LRESULT CALLBACK MouseProc(int nCode, WPARAM wParam, LPARAM lParam)
{
    int type;
    switch (wParam)
    {
    case WM_LBUTTONDOWN: type=0;break;
    case WM_RBUTTONDOWN: type=1;break;
    case WM_NCMBUTTONDOWN: type=2;break;
    case WM_MOUSEMOVE: type=3;break;
    case WM_LBUTTONUP: type=4;break;
    case WM_RBUTTONUP: type=5;break;
	case WM_MOUSEWHEEL: type=6;break;
    default: type=-1;break;
    }
    PMOUSEHOOKSTRUCT pmshs = reinterpret_cast<PMOUSEHOOKSTRUCT>(lParam);
    bool is_shield = false;
    xHook->onMouseEvent(type, pmshs, &is_shield);
    if(is_shield)//屏蔽鼠标事件
    {
        return 1;
    }
    return CallNextHookEx(g_hMouseHook,nCode,wParam,lParam);
}

LRESULT CALLBACK KeyboardProc(int nCode,WPARAM wParam,LPARAM lParam)
{
   PKBDLLHOOKSTRUCT pkbhs = reinterpret_cast<PKBDLLHOOKSTRUCT>(lParam);
   xHook->onKeyEvent(pkbhs);
   return CallNextHookEx(g_hKeyHook, nCode, wParam, lParam);
}

BOOL WINAPI DllMain(
     HINSTANCE hinstDLL,  // DLL模块的句柄
     DWORD fdwReason,     // 调用的情况
     LPVOID lpReserved )  // reserved
{
   Q_UNUSED(lpReserved);
   // 在不同的情况下都会调用DllMain函数，分别处理
   switch( fdwReason )
   {
     // 加载Dll
     case DLL_PROCESS_ATTACH:
     {
       g_hInstance = hinstDLL;
       break;
     }
       // 新建线程
     case DLL_THREAD_ATTACH:
     break;
       // 线程退出
     case DLL_THREAD_DETACH:
       break;
       // 释放Dll
     case DLL_PROCESS_DETACH:
       g_hInstance = Q_NULLPTR;
     break;
   }
   return TRUE;
}