TnbWindowsHooker.h

[詳解]

#pragma once
#include "TnbDef.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
#ifndef _TnbDOXYGEN //Document作成用シンボル
 
#if ( _WIN32_WINNT < 0x0400 )
 
    #define WH_KEYBOARD_LL      13
    #define WH_MOUSE_LL         14
 
    /*
     * Low level hook flags
     */
    #define LLKHF_EXTENDED          (KF_EXTENDED >> 8)
    #define LLKHF_INJECTED          0x00000010
    #define LLKHF_ALTDOWN           (KF_ALTDOWN >> 8)
    #define LLKHF_UP                (KF_UP >> 8)
    #define LLMHF_INJECTED          0x00000001
 
    /*
     * Structure used by WH_KEYBOARD_LL
     */
    typedef struct tagKBDLLHOOKSTRUCT {
        DWORD   vkCode;
        DWORD   scanCode;
        DWORD   flags;
        DWORD   time;
        ULONG   dwExtraInfo;
    } KBDLLHOOKSTRUCT, FAR *LPKBDLLHOOKSTRUCT, *PKBDLLHOOKSTRUCT;
 
    /*
     * Structure used by WH_MOUSE_LL
     */
    typedef struct tagMSLLHOOKSTRUCT {
        POINT   pt;
        DWORD   mouseData;
        DWORD   flags;
        DWORD   time;
        ULONG   dwExtraInfo;
    } MSLLHOOKSTRUCT, FAR *LPMSLLHOOKSTRUCT, *PMSLLHOOKSTRUCT;
 
#endif // (_WIN32_WINNT < 0x0400)
 
#ifndef WH_HARDWARE
    #define WH_HARDWARE         8
#endif
 
// ウィンドウズフック関係
namespace WinHook
{
    #ifdef TnbGLOBALHOOK
        #pragma comment(linker, "/section:.tnbwinhook,RWS")
        #pragma data_seg(".tnbwinhook")
    #endif
 
    // ウィンドウフック対象のインスタンスハンドル
    _SELECTANY HINSTANCE    g_hInstance     = NULL;
    // ウィンドウフック対象のスレッドID
    _SELECTANY DWORD        g_dwThreadId    = 0;
 
    #define _WINHOOKMAC(NAME) \
        _SELECTANY HHOOK    s_hhk##NAME = NULL;                                             \
        static HHOOK Hook##NAME(bool b, HOOKPROC PF) {                                      \
            if ( ! b && s_hhk##NAME != NULL ){                                              \
              ::UnhookWindowsHookEx(s_hhk##NAME); s_hhk##NAME = NULL;                       \
            }                                                                               \
            if ( b && s_hhk##NAME == NULL ){                                                \
              s_hhk##NAME = SetWindowsHookEx(WH_##NAME, PF, g_hInstance, g_dwThreadId);     \
            }                                                                               \
            return s_hhk##NAME;                                                             \
        }
 
    _WINHOOKMAC( MSGFILTER      )
    _WINHOOKMAC( JOURNALRECORD  )
    _WINHOOKMAC( JOURNALPLAYBACK)
    _WINHOOKMAC( KEYBOARD       )
    _WINHOOKMAC( GETMESSAGE     )
    _WINHOOKMAC( CALLWNDPROC    )
    _WINHOOKMAC( CBT            )
    _WINHOOKMAC( SYSMSGFILTER   )
    _WINHOOKMAC( MOUSE          )
    _WINHOOKMAC( HARDWARE       )
    _WINHOOKMAC( DEBUG          )
    _WINHOOKMAC( SHELL          )
    _WINHOOKMAC( FOREGROUNDIDLE )
    _WINHOOKMAC( CALLWNDPROCRET )
    _WINHOOKMAC( KEYBOARD_LL    )
    _WINHOOKMAC( MOUSE_LL       )
    #ifdef TnbGLOBALHOOK
        #pragma data_seg()
    #endif
    #undef _WINHOOKMAC
};
 
#endif //_TnbDOXYGEN
 
 
 
class CWindowsHooker
{
    typedef HHOOK (*LPHOOKFUNC)(bool, HOOKPROC);
 
    //Unhook処理用サブクラス
    class CReject
    {
    public: ~CReject(void);
    };
 
    // 指定のHookタイプの関数アドレス取得
    static LPHOOKFUNC m_GetFunc(int iHookType)
    {
        switch ( iHookType )
        {
        case WH_MSGFILTER:      return WinHook::HookMSGFILTER;
        case WH_JOURNALRECORD:  return WinHook::HookJOURNALRECORD;
        case WH_JOURNALPLAYBACK:return WinHook::HookJOURNALPLAYBACK;
        case WH_KEYBOARD:       return WinHook::HookKEYBOARD;
        case WH_GETMESSAGE:     return WinHook::HookGETMESSAGE;
        case WH_CALLWNDPROC :   return WinHook::HookCALLWNDPROC;
        case WH_CBT:            return WinHook::HookCBT;
        case WH_SYSMSGFILTER:   return WinHook::HookSYSMSGFILTER;
        case WH_MOUSE:          return WinHook::HookMOUSE;
        case WH_HARDWARE:       return WinHook::HookHARDWARE;
        case WH_DEBUG:          return WinHook::HookDEBUG;
        case WH_SHELL:          return WinHook::HookSHELL;
        case WH_FOREGROUNDIDLE: return WinHook::HookFOREGROUNDIDLE;
        case WH_CALLWNDPROCRET: return WinHook::HookCALLWNDPROCRET;
        case WH_KEYBOARD_LL:    return WinHook::HookKEYBOARD_LL;
        case WH_MOUSE_LL:       return WinHook::HookMOUSE_LL;
        default:    break;
        }
        return NULL;
    }
 
    CWindowsHooker(void);
 
public:
 
    static void OnDllMain(HANDLE hModule, DWORD reason)
    {
        switch ( reason )
        {
        case DLL_PROCESS_ATTACH:
            WinHook::g_hInstance = static_cast<HINSTANCE>(hModule);
            WinHook::g_dwThreadId = 0;
            break;
        case DLL_PROCESS_DETACH:
            UnhookAll();
            break;
        case DLL_THREAD_ATTACH:
        case DLL_THREAD_DETACH:
        default:
            break;
        }
    }
 
    static void SetEnvironment(HINSTANCE hInstance, DWORD dwThreadId = 0)
    {
        WinHook::g_hInstance = hInstance;
        WinHook::g_dwThreadId = dwThreadId;
    }
 
    static HHOOK Hook(int iHookType, HOOKPROC PF)
    {
        static CReject s_cReje;
        LPHOOKFUNC P = m_GetFunc(iHookType);
        if ( P != NULL ){ return P(true, PF); }
        return NULL;
    }
    
    static bool Unhook(int iHookType)
    {
        LPHOOKFUNC P = m_GetFunc(iHookType);
        if ( P != NULL ){ P(false, NULL); }
        return P != NULL;
    }
 
    static void UnhookAll(void)
    {
        for ( int i = WH_MIN; i < 30; i++ )
        {
            Unhook(i);
        }
    }
};
 
 
 
inline CWindowsHooker::CReject::~CReject(void)
{
    CWindowsHooker::UnhookAll();
}
 
 
 
};//TNB