WH_CBT フックを使用して、アプリの起動を検知する [C++Builder]

//---------------------------------------------------------------------------

#ifndef Unit1H
#define Unit1H
//---------------------------------------------------------------------------
#include <System.Classes.hpp>
#include <Vcl.Controls.hpp>
#include <Vcl.StdCtrls.hpp>
#include <Vcl.Forms.hpp>

#include "cbt_hook.h" // CbtHook.dll 使うためのインクルードファイル

class TForm1 : public TForm
{
private:	// ユーザー宣言
  // CbtHook.dll のモジュールのハンドル。LodLibrary() したときの返値。
  HINSTANCE m_hHookInst = NULL;

  // CbtHook.dll から PostMessage される ウィンドウメッセージの ID。
  Cardinal m_hookMessageId = NULL;

  // CbtHook.dll の関数
  GetHookMessage_t m_GetHookMessage;
  StartHook_t m_StartHookCBT;
  StopHook_t m_StopHookCBT;

  // CbtHook.dll の操作関数。
  void _StartCbtHook(); // ロードして WH_CBT フックを開始する
  void _StopCbtHook();  // 開放して WH_CBT フックを終了する

  // ウィンドウメッセージを受信する関数
  void __fastcall WndProc(Messages::TMessage &msg); // override TForm::WndProc
  void _WndProcHook(Messages::TMessage &msg); // call by WndProc()

public:		// ユーザー宣言
  __fastcall TForm1(TComponent* Owner);
};
//---------------------------------------------------------------------------
extern PACKAGE TForm1 *Form1;
//---------------------------------------------------------------------------
#endif

void __fastcall TForm1::FormShow(TObject *Sender)
{
    // WH_CBT フックを開始する
    _StartCbtHook();
}
//---------------------------------------------------------------------------

void __fastcall TForm1::FormClose(TObject *Sender, TCloseAction &Action)
{
    // WH_CBT フックを終了する
    _StopCbtHook();
}
//---------------------------------------------------------------------------

void TForm1::_StartCbtHook()
{
    // CbtHook.dll のフルパスを作る。
    UnicodeString exePath = ExtractFilePath(Application->ExeName);
    UnicodeString hookDllPath = exePath + L"CbtHook.dll";

    // CbtHook.dll をロードする。
    m_hHookInst = LoadLibrary(hookDllPath.c_str());
    if (m_hHookInst == NULL) {
        // 失敗した (fileを削除しているなど)
        return;
    }

    // CbtHook.dll に実装している関数をロードする。
    m_GetHookMessage = (GetHookMessage_t)GetProcAddress(m_hHookInst, "GetHookMessage");
    m_StartHookCBT = (StartHook_t)GetProcAddress(m_hHookInst, "StartHookCBT");
    m_StopHookCBT = (StopHook_t)GetProcAddress(m_hHookInst, "StopHookCBT");
    if (m_GetHookMessage == NULL || m_StartHookCBT == NULL || m_StopHookCBT == NULL) {
        // load出来たが関数を取得出来ない
        FreeLibrary(m_hHookInst);
        m_hHookInst = NULL;
        return;
    }

    // CbtHook.dll から PostMessage される ウィンドウメッセージの ID を取得する。
    // 登録するメッセージ識別子は CbtHook.dll から GetHookMessage() で得た値を使う。
    UnicodeString HookMessage = m_GetHookMessage();
    m_hookMessageId = RegisterWindowMessage(HookMessage.c_str());
    if (m_hookMessageId == 0) {
        FreeLibrary(m_hHookInst);
        m_hHookInst = NULL;
        return;
    }

    // WH_CBT フックを開始する。 (CbtHook.dll に指示する)
    m_StartHookCBT(Handle);
}
//---------------------------------------------------------------------------

void TForm1::_StopCbtHook()
{
    if (m_hHookInst == NULL) return;

    // WH_CBT フックを終了する。 (CbtHook.dll に指示する)
    m_StopHookCBT();

    // CbtHook.dll を開放する。
    FreeLibrary(m_hHookInst);
	m_hHookInst = NULL;
    m_hookMessageId = NULL;
}
//---------------------------------------------------------------------------

//=============================================================================
//  Application側でメッセージ通知を受取る
//  ここではフォームのWndProcメソッドを使用
//=============================================================================
void __fastcall TForm1::WndProc(Messages::TMessage &msg)
{
    if (msg.Msg == m_hookMessageId) {
		// CbtHook.dll からの通知を受信した。
        _WndProcHook(msg);
    }

    // デフォルトの WndProc() を実行する。
    TForm::WndProc(msg);
}
//---------------------------------------------------------------------------

void TForm1::_WndProcHook(Messages::TMessage &msg)
{
    if (m_hookMessageId == NULL) return;

    UINT msgId = m_hookMessageId;
    WPARAM wParam = msg.WParam;
    int nCode = (int)msg.LParam;

    switch (nCode) {
        // wParam = hWnd イベント
        case HCBT_CREATEWND:
        case HCBT_DESTROYWND:
            break;
        case HCBT_ACTIVATE:
        case HCBT_SETFOCUS:
            // ここにイベントを受信したときの処理を書く
            break;
        case HCBT_MOVESIZE:
        case HCBT_MINMAX:
        case HCBT_SYSCOMMAND:
        case HCBT_CLICKSKIPPED:
        case HCBT_KEYSKIPPED:
        case HCBT_QS:
        default:
            break;
    }  // switch (cd->dwData)
    return;
}

#ifndef CBT_HOOK_H
#define CBT_HOOK_H

// CbtHook.dll の型宣言
typedef char * __stdcall (*GetHookMessage_t)(void);
typedef bool __stdcall (*StartHook_t)(HWND);
typedef void __stdcall (*StopHook_t)(void);

#endif // CBT_HOOK_H

#ifndef HOOKUNIT_H
#define HOOKUNIT_H

#include <windows.h>

#include "cbt_hook.h" // CbtHook.dll を使うためのインクルードファイル
#ifdef _HOOK_CPP
#define DECLSPEC __declspec(dllexport)
#else
#define DECLSPEC __declspec(dllimport)
#endif

extern "C" DECLSPEC const char *__stdcall GetHookMessage();
extern "C" DECLSPEC bool __stdcall StartHookCBT(HWND);
extern "C" DECLSPEC void __stdcall StopHookCBT();

// 共有メモリの内容の構造体
struct THookInfo {
    HWND hWndHookOwner;  // 通知を受け取るアプリのウィンドウハンドル
    UINT msgId;                 // 通知時のメッセージID
};

#endif /* HOOKUNIT_H */

//---------------------------------------------------------------------------
#include <windows.h>
#include "cbt_hook.h" // CbtHook.dll を使うためのインクルードファイル
#define _HOOK_CPP
#include "CbtHookUnit.h"

// グローバル変数 ............................................................
// 識別子 - システム全体で一意となる文字列にすること。
const char *g_HookMessage = "MyApp_CbtHookMsg";  // 自アプリ宛メッセージ識別子
const char *g_MapFileName = "MyApp_CbtHookMapFile";  // 共有マップファイル名

// フックの開始／終了を制御する自exeにアタッチした CbtHook.dll のみ知っている情報。
HHOOK g_HookHandleCBT = NULL;  // SetWindowsHookEx() で得たハンドル。

// DLL_PROCESS_ATTACH した各々のexeで異なる値が入る変数。
// g_hookDataには、フックが挿入される全アプリで共有したいデータを入れる。
// フックで取得した情報の通知先の自アプリのウィンドウハンドルなど。
HINSTANCE g_hinstDLL = NULL;
HANDLE g_hMapFile = NULL;
THookInfo *g_tHookInfo = NULL;

// 内部関数プロトタイプ宣言 ..................................................
bool _DllProcessAttach(HINSTANCE, LPVOID);
void _DllProcessDetach(LPVOID);
bool _MapHookData();
void _UnmapHookData();
LRESULT CALLBACK _CbtProc(int, WPARAM, LPARAM);
void _HandleHookEvent(int, WPARAM, LPARAM);
bool _IsEventOccuredOwnWnd(HWND, HWND);

#pragma argsused
//=============================================================================
// DllMain()
//=============================================================================
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fwdreason, LPVOID lpvReserved)
{
    switch (fwdreason) {
        case DLL_PROCESS_ATTACH:  // ロードされるときの処理
            if (_DllProcessAttach(hinstDLL, lpvReserved) == false) return FALSE;
            break;
        case DLL_THREAD_ATTACH:
            break;
        case DLL_THREAD_DETACH:
            break;
        case DLL_PROCESS_DETACH:  // アンロードされるときの処理
            _DllProcessDetach(lpvReserved);
            break;
    }
    return TRUE;
}

//---------------------------------------------------------------------------
// _DllProcessAttach()
//---------------------------------------------------------------------------
bool _DllProcessAttach(HINSTANCE hinstDLL, LPVOID lpvReserved)
{
    if (g_hinstDLL) return true;

    // Dll instanceの保存
    g_hinstDLL = hinstDLL;

    // メモリマップドファイルの作成
    if (_MapHookData() == false) {
        _UnmapHookData();
        return false;
    }

    return true;
}

//---------------------------------------------------------------------------
// _DllProcessDetach()
//---------------------------------------------------------------------------
void _DllProcessDetach(LPVOID lpvReserved)
{
    if (g_hinstDLL == NULL) return;

    // フックの停止
    StopHookCBT();

    //  メモリマップドファイルのクローズ
    _UnmapHookData();

    // 保存した Dll instance のクリア
    g_hinstDLL = NULL;
}

//=============================================================================
// GetHookMessage() - Application側でメッセージ通知を受取るために
// RegisterWindowMessage する関数
//=============================================================================
extern "C" __declspec(dllexport) const char *__stdcall GetHookMessage()
{
    return g_HookMessage;
}

//=============================================================================
// StartHookCBT() - WH_CBT フック関数の登録/解除
// 登録するフック関数は _CbtProc()
// PARAMETER
//   HWND hWnd :
//       フックイベント発生時に通知を受け取るウィンドウのハンドル
//=============================================================================
extern "C" __declspec(dllexport) bool __stdcall StartHookCBT(
    HWND hWnd)
{
    if (g_tHookInfo == NULL) return false;

    if (g_HookHandleCBT != NULL)
        // フックが解除されていない.
        return true;

    // フック情報構造体初期化とフック関数の登録
    g_tHookInfo->hWndHookOwner = hWnd;

    // アプリケーションへ通知するメッセージを登録
    g_tHookInfo->msgId = RegisterWindowMessage(g_HookMessage);
    if (g_tHookInfo->msgId == 0) {
        return false;
    }

    // WH_CBT フック開始 - _CbtProc() をフックチェーンにインストールする
    g_HookHandleCBT = SetWindowsHookEx(
        /* idHook    =*/WH_CBT,
        /* lpFn      =*/(HOOKPROC)_CbtProc,
        /* hmod      =*/g_hinstDLL,
        /* dwThreadId=*/0);
    if (g_HookHandleCBT == NULL) {
        // 失敗してるので失敗(false)を返す.
        return false;
    }

    // フック成功
    return true;
}

//=============================================================================
// StopHookCBT() - フックの解除
//=============================================================================
extern "C" __declspec(dllexport) void __stdcall StopHookCBT()
{
    if (g_HookHandleCBT == NULL) return;

    // フック解除 - _CbtProc() をフックチェーンから削除する
    BOOL ret = UnhookWindowsHookEx(g_HookHandleCBT);
    if (ret == 0) {
        return;
    }

    g_HookHandleCBT = NULL;
    return;
}

//-----------------------------------------------------------------------------
// _MapHookData() - メモリマップドファイルの作成
//-----------------------------------------------------------------------------
bool _MapHookData()
{
    if (g_hMapFile) return false;
    if (g_tHookInfo) return false;

    g_hMapFile = CreateFileMapping(
        /* hFile = */ (HANDLE)INVALID_HANDLE_VALUE,
        /* lpFileMappingAttributes = */ NULL,
        /* flProtect         = */ PAGE_READWRITE,
        /* dwMaximumSizeHigh = */ 0,
        /* dwMaximuSizeLow   = */ sizeof(THookInfo),
        /* lpName            = */ g_MapFileName);
    if (g_hMapFile == NULL) {
        return false;
    }

    g_tHookInfo = (THookInfo *)MapViewOfFile(g_hMapFile, FILE_MAP_ALL_ACCESS,
                                               0, 0, sizeof(THookInfo));
    if (g_tHookInfo == NULL) {
        return false;
    }

    return true;
}

//-----------------------------------------------------------------------------
// _UnmapHookData() - メモリマップドファイルの破棄
//-----------------------------------------------------------------------------
void _UnmapHookData()
{
    if (g_HookHandleCBT != NULL) return;

    if (g_tHookInfo) {
        if (UnmapViewOfFile(g_tHookInfo) == 0) {
            return;
        }
        g_tHookInfo = NULL;
    }

    if (g_hMapFile) {
        CloseHandle(g_hMapFile);
        g_hMapFile = NULL;
    }
}

//=============================================================================
// フックのコールバック関数
//=============================================================================
LRESULT CALLBACK _CbtProc(int nCode, WPARAM wParam, LPARAM lParam)
{
    if (nCode >= 0) {
        // 0以上ならフックイベントを処理する
        _HandleHookEvent(nCode, wParam, lParam);
    }

    return CallNextHookEx(NULL, nCode, wParam, lParam);
}

//-----------------------------------------------------------------------------
// _HandleHookEvent
//-----------------------------------------------------------------------------
void _HandleHookEvent(int nCode, WPARAM wParam, LPARAM lParam)
{
    if (g_tHookInfo == NULL) return;  // 共有メモリが未作成なら処理しない

    // 共有データに入っている情報を変数に取り出す
    HWND hWndHookOwner = (g_tHookInfo) ? g_tHookInfo->hWndHookOwner : NULL;
    UINT msgId = (g_tHookInfo) ? g_tHookInfo->msgId : NULL;
    if (hWndHookOwner == NULL) return;
    if (msgId == NULL) return;

    // イベントを処理する
    switch (nCode) {
        case HCBT_CREATEWND:  // 3 ウィンドウが作成されようとしている
        case HCBT_DESTROYWND:  // 4 ウィンドウが破棄されようとしている
            break;
        case HCBT_ACTIVATE:    // 5 システムがウィンドウをアクティブ化しようとしている
        case HCBT_SETFOCUS:    // 9 ウィンドウがキーボードフォーカスを受け取ろうとしている
            // HCBT_CREATEWND ではなく、HCBT_ACTIVATE/SETFOCCUS をアプリ起動として
            // 使用する。連続発生する HCBT_CREATEWND ではウィンドウタイトル等の情報が
            // 取れるタイミングが分からないため。情報取得できる HCBT_ACTIVE/SETFOCUS
            // を起動の検知に使用する。
            // PostMessage()したメッセージは、TForm1::WndProc()で受信される。
            // FormMain.cpp も CbtHookUnit.cpp も g_HooMessage の同じ文字列で
            // RegisterWindowMessage() して msgId を得ているので、同じ値であれば
            // CbtHookUnit -> FormMain と投げられたメッセージであると識別できる。
            PostMessage(hWndHookOwner, msgId, (WPARAM)wParam, (LPARAM)nCode);            
            break;
        case HCBT_MOVESIZE:    // 0 ウィンドウが移動／サイズ変更されようとしている
        case HCBT_MINMAX:      // 1 最大化/最小化
        case HCBT_SYSCOMMAND:  // 8 システムコマンドが実行されようとしている
        case HCBT_CLICKSKIPPED:// 6 システムメッセージキューからマウスメッセージを削除した
        case HCBT_KEYSKIPPED:  // 7 システムメッセージ待ち行列からキーボードメッセージを除去した
        case HCBT_QS:          // 2 システムメッセージ待ち行列から WM_QUEUESYNC メッセージを取り出した
        default:
            break;
    }  // switch(nCode)
}