root/lang/cplusplus/SleipnirGoogleSuggest/HookDll.cpp

/**
 *
 */

#include "StdAfx.h"
#include "Common.h"

namespace SleipnirGoogleSuggest {

// �t�b�N�֘A�͓D�L���̂ł���namespace�lj�
namespace Hook {

#pragma section ("mysecdrfgsr",read,write,shared)
#pragma data_seg("mysecdrfgsr")
// DLL�����L�����O���[�o���ϐ�
static struct {
    HHOOK hHook;        // Hook()�̃n���h��
    LONG  isHookBegan;  // �͂��߂��Ńt�b�N�v���V�[�W�����n�܂����B XXX �����K�v
} shared_globals = {};
#pragma data_seg()

// �O���[�o���ϐ�
static struct {
    bool isHookCalled;        // Hook()�̃��C�����[�`������������s�������悤�ɂ�����
    HMODULE hModule;          // DLL�̃��W���[���n���h��
    WNDPROC DefaultEditProc;  // �T�u�N���X���ȑO�̃G�f�B�b�g�{�b�N�X�̃E�B���h�E�v���V�[�W��
} globals = {};

/**
 * GoogleSuggest�@�\�̉�
 */
void Detach() {
    HWND hWnd = GetSearchBarEdit();
    SetWindowLongPtr(hWnd,GWLP_WNDPROC,(LONG_PTR)globals.DefaultEditProc);

    // XXX �����댯���H
    Debug("FreeLibrary()");
    FreeLibrary(GetModuleHandle(NULL));
    Debug("FreeLibrary() OK");
}

/**
 * GoogleSuggest�@�\�����ꂽ�G�W�b�g�{�b�N�X�̃E�B���h�E�v���V�[�W��
 */
LRESULT CALLBACK SubclassedEditProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam) {
    if (msg == WM_KEYDOWN) {
        MessageBox(NULL,_T("keydown"),_T(""),MB_OK);
    }
    if (msg == WM_LBUTTONDOWN) {
        MessageBox(NULL,_T(""),_T(""),MB_OK);
        Detach();
    }
    return globals.DefaultEditProc(hWnd,msg,wParam,lParam);
}

/**
 * GoogleSuggest�@�\�̒lj�
 * ���v���Z�X�Ŏ�s�����v������ */
void Attach() {
    HWND hWnd = GetSearchBarEdit();
    globals.DefaultEditProc = (WNDPROC)GetWindowLongPtr(hWnd, GWLP_WNDPROC);
    SetWindowLongPtr(hWnd,GWLP_WNDPROC,(LONG_PTR)SubclassedEditProc);
}

/**
 * �T�u�N���X�����邽��leipnir�̃E�B���h�E�v���V�[�W���Ƀt�b�N������
 */
LRESULT CALLBACK Hook(int nCode,WPARAM wParam,LPARAM lParam) {
    if (nCode < 0 || nCode != HC_ACTION || wParam != 0 || globals.isHookCalled) {
        return CallNextHookEx(shared_globals.hHook, nCode, wParam, lParam);
    }
    globals.isHookCalled = true;

    // �t�b�N�֐��n�܂�����m
    InterlockedExchange(&shared_globals.isHookBegan,1);

    Debug("start hHook: " << (int)shared_globals.hHook);

    // Increase reference count via LoadLibrary
    TCHAR lib_name[MAX_PATH];
    if (!GetModuleFileName(globals.hModule, lib_name, MAX_PATH)) {
        Debug("!GetModuleFileName()");
        abort();
    }
    if (!LoadLibrary(lib_name)) {
        Debug("!LoadLibrary()");
        abort();
    }
    // �t�b�N��
    if (!UnhookWindowsHookEx(shared_globals.hHook)) {
        Debug("!UnhookWindowsHookEx()");
        abort();
    }

    Debug("end");
    Attach();
    // XXX hHook�͖�����Ă��邩�������H�H�����̂��H�H
    return CallNextHookEx(shared_globals.hHook, nCode, wParam, lParam);
}

/**
 * GoogleSuggest�@�\��ݍ��ފ֐�
 * �ʂ̃v���Z�X�����яo��
 */
__declspec(dllexport) bool CALLBACK AttachFromOtherProcess(DWORD SleipnirThreadId)
{
    if (shared_globals.hHook) {
        Debug("already hooked");
        return false;
    }
    shared_globals.hHook = SetWindowsHookEx(WH_CALLWNDPROCRET,Hook,globals.hModule,SleipnirThreadId);
    if (!shared_globals.hHook) {
        Debug("!SetWindowsHookEx()");
        return false;
    }

    // �t�b�N�֐����n�܂��ő҂�
    bool released = false;
    for (int loop=0; loop<100; loop++) {
        Sleep(1000);
        // XXX �ǂݍ��݂̓�������ƃ_���H
        if (shared_globals.isHookBegan) {
            released = true;
            break;
        }
    }
    if (!released) {
        Debug("timeout");
        return false;
    }

    Debug("OK");
    return true;
}
}
}


/**
 * DllMain
 */
BOOL APIENTRY DllMain( HMODULE hModule,
                       DWORD  ul_reason_for_call,
                       LPVOID lpReserved
                     )
{
    UNREFERENCED_PARAMETER(lpReserved);

    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
        SleipnirGoogleSuggest::Hook::globals.hModule = hModule;
    case DLL_THREAD_ATTACH:
    case DLL_THREAD_DETACH:
    case DLL_PROCESS_DETACH:
        break;
    }
    return TRUE;
}