如何在c ++中从dll与调用应用程序进行通信

时间:2013-12-14 12:39:25

标签: c++ windows dll hook

我创建了一个系统范围的挂钩dll,在我的dll中,我试图在每次创建或销毁新进程时收到通知。
当检测到新进程时,我希望能够向调用程序发送消息,无论是布尔值还是自定义对象。 我怎么能这样做?
目前我正在使用文件记录所有名称!这太可怕了 这是到目前为止的代码:

def文件

;LIBRARY
; Def file
EXPORTS
    InstallHook
    UninstallHook

dllapi.h

void InstallHook(void);
void UninstallHook(void);

dllmain.cpp

#include "stdafx.h" 

using namespace std;
HINSTANCE currentProcessHandle;
HHOOK hookID;
string str = "1";
BOOL APIENTRY DllMain(HMODULE hModule, DWORD  ul_reason_for_call,LPVOID lpReserved)
{
    if (ul_reason_for_call == DLL_PROCESS_ATTACH)
        currentProcessHandle = hModule;
    return TRUE;
}

LRESULT CALLBACK HookProcedure(int nCode, WPARAM wparam, LPARAM lparam)
{

    if (nCode < 0) return CallNextHookEx(hookID, nCode, wparam, lparam);

    std::ofstream outfile;
    CBT_CREATEWND   *CBTHOOKCREATE;
    RECT            *CBTRECTPTR;
    RECT            CBTRECT;
    wstring         Message;

    CBTHOOKCREATE = (CBT_CREATEWND*) lparam;
    LPWSTR str = L"                     ";
    outfile.open(("d:\\test.txt"), std::ios_base::app);

    if (nCode >= 0) {
        switch (nCode)
        {
        case HCBT_CREATEWND:
            outfile << *(CBTHOOKCREATE->lpcs->lpszName) << " " << CBTHOOKCREATE->lpcs->lpszName << " Created!~ " << endl;
            //cout << "Created!~" << endl;
            break;
        case HCBT_DESTROYWND:
            outfile << "Destroied!~" << endl;
            //cout << "Destroied!~" << endl;
            break;
        default:
            //cout << "sth else" << endl;
            break;
        }
    }
    outfile.close();
    return 0;
}

void InstallHook(void)
{
    hookID = SetWindowsHookEx(WH_CBT, HookProcedure, currentProcessHandle, 0);
}

void UninstallHook(void)
{ 
    UnhookWindowsHookEx(hookID);
}

Hook Consumer控制台应用程序

// Hook Executer.cpp : Defines the entry point for the console application.
//
#include "stdafx.h"
#include "..\Dll\dllapi.h"
#include <iostream>
using namespace std;

int _tmain(int argc, _TCHAR* argv[])
{
    int num = -1;
    cout << "1.Install Hook"<<endl
        << "2.Unistall Hook"<<endl
        << "0.Exit";
    do{
        cin >> num;
        if (num ==1)
        {
            InstallHook();

        }
        else
        {
            UninstallHook();
        }
        getchar();
        system("cls");
        cout << "1.Install Hook" << endl
            << "2.Unistall Hook" << endl
            << "0.Exit";
    } while (num != 0 && num < 3);


    return 0;
}

1 个答案:

答案 0 :(得分:1)

Hook DLL由Windows加载到挂钩进程的地址空间中。您将不得不使用IPC。有关入门者,请参阅Interprocess Communications

在这种情况下可以使用的一个简单的IPC可能是Data Copy

请注意,数据复制需要接收线程中的活动(泵送)消息队列。

实现该目标的一种可能方式(许多其他可能的方法):

使用抽取的消息队列向EXE添加辅助线程。在新线程中,您将创建一个具有特定类名的虚拟不可见窗口。

这段代码非常经典:RegisterClassCreateWindowwhile GetMessage DispatchMessage

在Hook DLL中,您将拥有一个全局HWND变量。当想要使用WM_COPYDATA时,如果全局变量为null,请使用FindWindow检索HWND,并将其存储以供下次使用。该代码将在每个挂钩过程中执行至少一次。

您可能希望使用SendMessageTimeout API发送WM_COPYDATA消息。

请注意,如果快速停止/重新启动exe,某些进程可能会在全局变量中存储无效的HWND。检查Send API的返回值,如果它是“无效的hwnd”,则重做FindWindow。 (不确定这种行为是否真的可行,但无论如何......)

另请注意,在安装挂钩之前,主线程应等待在辅助线程中正确创建虚拟窗口。

如果您的挂钩EXE是一个Windowed,您不需要辅助线程,但您必须构建一个GUI。也许你也可以“只使用一个线程”并设法使用一些API,同时拥有一个活动的消息队列和你的getchar代码,但我不建议这样做。