d100d23259
Added support for message hook viewer to hook both 32 and 64 bit applications. Fixed filtering on the msg hook viewer Added custom filtering on msg hook viewer Added Process Id targetting on msg hook viewer Added SetMsgHook.exe command line app as an alternative way of starting msg hook viewer.
62 lines
2.1 KiB
C++
62 lines
2.1 KiB
C++
// dllmain.cpp : Defines the entry point for the DLL application.
|
|
#include "stdafx.h"
|
|
|
|
HANDLE hMappedFile;
|
|
GLOBALDATA* pData;
|
|
bool bStartingProcess = false;
|
|
|
|
|
|
BOOL APIENTRY DllMain( HMODULE hModule,
|
|
DWORD ul_reason_for_call,
|
|
LPVOID lpReserved
|
|
)
|
|
{
|
|
switch (ul_reason_for_call)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
{
|
|
//printf("debug DLL_PROCESS_ATTACH hModule: %ld\n", (long)hModule);
|
|
TCHAR szBaseName[_MAX_FNAME], szTmp[_MAX_FNAME];
|
|
memset((void *)&szBaseName, '\0', sizeof(TCHAR) * _MAX_FNAME);
|
|
|
|
if (GetModuleBaseName(GetCurrentProcess(), (HMODULE)hModule, szTmp, sizeof(szTmp)))// compute MMF-filename from current module base name, uses Psapi
|
|
_wsplitpath_s(szTmp, NULL, NULL, szBaseName, _MAX_FNAME, NULL, NULL, NULL, NULL);
|
|
//_wsplitpath(szTmp, NULL, NULL, szBaseName, NULL);
|
|
|
|
wcscat_s(szBaseName, TEXT("MsgHookSharedMem")); // add specifier string
|
|
if (IsCurrentProcess64Bit())
|
|
wcscat_s(szBaseName, TEXT("64")); // add bit specifier
|
|
else
|
|
wcscat_s(szBaseName, TEXT("32")); // add bit specifier
|
|
|
|
hMappedFile = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, sizeof(GLOBALDATA), szBaseName);
|
|
pData = (GLOBALDATA*)MapViewOfFile(hMappedFile, FILE_MAP_WRITE, 0, 0, 0);
|
|
bStartingProcess = (hMappedFile != NULL) && (GetLastError() != ERROR_ALREADY_EXISTS);
|
|
|
|
if(bStartingProcess) // if the MMF doesn't exist, we have the first instance
|
|
{
|
|
pData->g_hInstance = hModule; // so set the instance handle
|
|
pData->g_hWnd = NULL; // and initialize the other handles
|
|
pData->g_CwpHook = NULL;
|
|
pData->g_MsgHook = NULL;
|
|
pData->g_CwpHookProc = NULL;
|
|
}
|
|
else
|
|
{
|
|
//open
|
|
hMappedFile = OpenFileMapping(FILE_MAP_ALL_ACCESS, false, szBaseName);
|
|
pData = (GLOBALDATA*)MapViewOfFile(hMappedFile, FILE_MAP_ALL_ACCESS, 0, 0, 0);
|
|
}
|
|
DisableThreadLibraryCalls((HMODULE)hModule);
|
|
}
|
|
break;
|
|
case DLL_THREAD_ATTACH:
|
|
case DLL_THREAD_DETACH:
|
|
case DLL_PROCESS_DETACH:
|
|
CloseHandle(hMappedFile); // on detaching the DLL, close the MMF
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|