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.
97 lines
2.7 KiB
C++
97 lines
2.7 KiB
C++
/*
|
|
* Copyright 2014, Synthuse.org
|
|
* Released under the Apache Version 2.0 License.
|
|
*
|
|
* last modified by ejakubowski7@gmail.com
|
|
*/
|
|
// SetMsgHook.cpp : Defines the entry point for the console application.
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "MsgHook.h"
|
|
|
|
|
|
int _tmain(int argc, _TCHAR* argv[])
|
|
{
|
|
if (argc == 1) //no args passed, show MsgHook Viewer gui
|
|
{
|
|
MsgHook_CreateMsgHookWindow(NULL);
|
|
return 0;
|
|
}
|
|
|
|
HWND hookHwnd = NULL;
|
|
long procId = 0;
|
|
TCHAR *stopStr;
|
|
|
|
for (int i = 1 ; i < argc ; i++)
|
|
{
|
|
if (_tcscmp(argv[i], _T("?")) == 0 || _tcscmp(argv[i], _T("-?")) == 0 || argc == 3 || argc > 4)
|
|
{
|
|
printf("SetMsgHook version 1.0 by Edward Jakubowski \n\n");
|
|
printf("Usage: SetMsgHook.exe [(MSG_HOOK_DLL) (MSG_HOOK_HWND_OR_ZERO) (TARGET_PID)] [MSG_HOOK_DLL] [?] \n\n");
|
|
printf(" Additional Notes:\n");
|
|
printf(" Message Hook Viewer Gui - To open the gui you must provide the path to the msg hook dll as the ");
|
|
printf("only argument when running SetMsgHook.exe. Also setting the (MSG_HOOK_HWND) argument to 0 (zero) ");
|
|
printf("will start the gui and message hook on the given Process Id.\n\n");
|
|
HWND currentHwnd = FindWindow(_T("MSGHOOKVIEW"), NULL);
|
|
printf(" Current MSG_HOOK_HWND: %ld\n", (long)currentHwnd);
|
|
return 0;
|
|
}
|
|
if (i == 1)
|
|
_tcsncpy_s(MSGHOOK_DLL_NAME, MAX_NAME_SIZE, argv[i], _TRUNCATE);
|
|
if (i == 2)
|
|
hookHwnd = (HWND)_tcstol(argv[i], &stopStr, 10);
|
|
if (i == 3)
|
|
procId = (long)_tcstol(argv[i], &stopStr, 10);
|
|
}
|
|
|
|
if (argc == 2) //one arg passed (dll), show MsgHook Viewer gui
|
|
{
|
|
printf("Starting msg hook viewer...");
|
|
MsgHook_CreateMsgHookWindow(NULL);
|
|
return 0;
|
|
}
|
|
if (argc == 4 && hookHwnd == 0)
|
|
{
|
|
printf("Starting msg hook viewer on pid %ld...", (long)procId);
|
|
TCHAR tmp[100];
|
|
_stprintf_s(tmp, _T("%ld"), (long)procId);
|
|
MsgHook_CreateMsgHookWindow(tmp);
|
|
return 0;
|
|
}
|
|
|
|
char tmp[MAX_NAME_SIZE];
|
|
size_t convertedCnt = 0;
|
|
wcstombs_s(&convertedCnt, tmp, MAX_NAME_SIZE, MSGHOOK_DLL_NAME, _TRUNCATE);
|
|
printf("MsgHook DLL: %s, HWND: %ld, PID: %ld", tmp, (long)hookHwnd, procId);
|
|
DWORD threadId = MsgHook_GetProcessMainThreadId(procId);
|
|
printf(", ThreadId: %ld\n", (long)threadId);
|
|
|
|
if (MsgHook_SetMsgHook(hookHwnd, threadId))
|
|
printf("Hook successfully initialized\n");
|
|
else
|
|
{
|
|
printf("Hook failed to initialize\n");
|
|
return -1;
|
|
}
|
|
|
|
//don't exit SetMsgHook until hooked process exits
|
|
HANDLE process = OpenProcess(SYNCHRONIZE, FALSE, procId);
|
|
while(WaitForSingleObject(process, 0) == WAIT_TIMEOUT)
|
|
{
|
|
Sleep(1000); //check once per second
|
|
if (GetCurrentHookHandle() == NULL)
|
|
{
|
|
printf("unhooked.");
|
|
break;
|
|
}
|
|
}
|
|
CloseHandle(process);
|
|
MsgHook_RemoveHook();
|
|
|
|
//_getch();
|
|
printf("done.");
|
|
return 0;
|
|
}
|
|
|