chenx221/Textractor
1
0
Fork 0
forked from Public-Mirror/Textractor
Code Pull Requests Activity

using raii in vnrhook

Browse Source
This commit is contained in:
Akash Mozumdar 2018-12-01 15:52:52 -05:00
parent bc644bbe4e
commit 2f544afaa5
1 changed files with 23 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
vnrhook/main.cc
View File

@ -17,14 +17,14 @@
#include "texthook.h" #include "texthook.h"
#include "util/growl.h" #include "util/growl.h"
std::unique_ptr<WinMutex> sectionMutex; std::unique_ptr<WinMutex> viewMutex;
namespace namespace
{ {
HANDLE hSection, hookPipe; AutoHandle<> hookPipe = INVALID_HANDLE_VALUE, mappedFile = INVALID_HANDLE_VALUE;
TextHook* hooks; TextHook* hooks;
bool running; bool running;
int currentHook = 0, userhookCount = 0; int currentHook = 0;
DWORD DUMMY; DWORD DUMMY;
} }
@ -34,7 +34,8 @@ DWORD WINAPI Pipe(LPVOID)
{ {
DWORD count = 0; DWORD count = 0;
BYTE buffer[PIPE_BUFFER_SIZE] = {}; BYTE buffer[PIPE_BUFFER_SIZE] = {};
HANDLE hostPipe = hookPipe = INVALID_HANDLE_VALUE; AutoHandle<> hostPipe = INVALID_HANDLE_VALUE;
hookPipe = INVALID_HANDLE_VALUE;
while (hookPipe == INVALID_HANDLE_VALUE || hostPipe == INVALID_HANDLE_VALUE) while (hookPipe == INVALID_HANDLE_VALUE || hostPipe == INVALID_HANDLE_VALUE)
{ {
@ -68,7 +69,7 @@ DWORD WINAPI Pipe(LPVOID)
case HOST_COMMAND_NEW_HOOK: case HOST_COMMAND_NEW_HOOK:
{ {
auto info = *(InsertHookCmd*)buffer; auto info = *(InsertHookCmd*)buffer;
NewHook(info.hp, info.name, 0); NewHook(info.hp, "UserHook", 0);
} }
break; break;
case HOST_COMMAND_DETACH: case HOST_COMMAND_DETACH:
@ -77,10 +78,9 @@ DWORD WINAPI Pipe(LPVOID)
} }
break; break;
} }
CloseHandle(hostPipe);
CloseHandle(hookPipe);
} }
hookPipe = INVALID_HANDLE_VALUE;
for (int i = 0; i < MAX_HOOK; ++i) if (hooks[i].hp.insertion_address) hooks[i].Clear();
FreeLibraryAndExitThread(GetModuleHandleW(ITH_DLL), 0); FreeLibraryAndExitThread(GetModuleHandleW(ITH_DLL), 0);
return 0; return 0;
} }
@ -95,17 +95,15 @@ void TextOutput(ThreadParam tp, BYTE* text, int len)
WriteFile(hookPipe, buffer, sizeof(ThreadParam) + len, &DUMMY, nullptr); WriteFile(hookPipe, buffer, sizeof(ThreadParam) + len, &DUMMY, nullptr);
} }
void ConsoleOutput(LPCSTR text) void ConsoleOutput(LPCSTR text, ...)
{ {
ConsoleOutputNotif buffer(text); ConsoleOutputNotif buffer;
va_list args;
va_start(args, text);
vsprintf_s<MESSAGE_SIZE>(buffer.message, text, args);
WriteFile(hookPipe, &buffer, sizeof(buffer), &DUMMY, nullptr); WriteFile(hookPipe, &buffer, sizeof(buffer), &DUMMY, nullptr);
} }
void ConsoleOutput(std::string text)
{
ConsoleOutput(text.c_str());
}
void NotifyHookRemove(uint64_t addr) void NotifyHookRemove(uint64_t addr)
{ {
HookRemovedNotif buffer(addr); HookRemovedNotif buffer(addr);
@ -118,13 +116,13 @@ BOOL WINAPI DllMain(HINSTANCE hModule, DWORD fdwReason, LPVOID)
{ {
case DLL_PROCESS_ATTACH: case DLL_PROCESS_ATTACH:
{ {
sectionMutex = std::make_unique<WinMutex>(ITH_HOOKMAN_MUTEX_ + std::to_wstring(GetCurrentProcessId())); viewMutex = std::make_unique<WinMutex>(ITH_HOOKMAN_MUTEX_ + std::to_wstring(GetCurrentProcessId()));
if (GetLastError() == ERROR_ALREADY_EXISTS) return FALSE; if (GetLastError() == ERROR_ALREADY_EXISTS) return FALSE;
DisableThreadLibraryCalls(hModule); DisableThreadLibraryCalls(hModule);
// jichi 9/25/2013: Interprocedural communication with vnrsrv. // jichi 9/25/2013: Interprocedural communication with vnrsrv.
hSection = CreateFileMappingW(INVALID_HANDLE_VALUE, nullptr, PAGE_EXECUTE_READWRITE, 0, HOOK_SECTION_SIZE, (ITH_SECTION_ + std::to_wstring(GetCurrentProcessId())).c_str()); mappedFile = CreateFileMappingW(INVALID_HANDLE_VALUE, nullptr, PAGE_EXECUTE_READWRITE, 0, HOOK_SECTION_SIZE, (ITH_SECTION_ + std::to_wstring(GetCurrentProcessId())).c_str());
hooks = (TextHook*)MapViewOfFile(hSection, FILE_MAP_ALL_ACCESS | FILE_MAP_EXECUTE, 0, 0, HOOK_BUFFER_SIZE); hooks = (TextHook*)MapViewOfFile(mappedFile, FILE_MAP_ALL_ACCESS | FILE_MAP_EXECUTE, 0, 0, HOOK_BUFFER_SIZE);
memset(hooks, 0, HOOK_BUFFER_SIZE); memset(hooks, 0, HOOK_BUFFER_SIZE);
MH_Initialize(); MH_Initialize();
@ -136,36 +134,27 @@ BOOL WINAPI DllMain(HINSTANCE hModule, DWORD fdwReason, LPVOID)
case DLL_PROCESS_DETACH: case DLL_PROCESS_DETACH:
{ {
running = false; running = false;
for (int i = 0; i < MAX_HOOK; ++i) if (hooks[i].hp.insertion_address) hooks[i].ClearHook();
UnmapViewOfFile(hooks); UnmapViewOfFile(hooks);
MH_Uninitialize(); MH_Uninitialize();
CloseHandle(hSection);
} }
break; break;
} }
return TRUE; return TRUE;
} }
//extern "C" {
void NewHook(HookParam hp, LPCSTR lpname, DWORD flag) void NewHook(HookParam hp, LPCSTR lpname, DWORD flag)
{ {
std::string name = lpname; if (++currentHook >= MAX_HOOK) return ConsoleOutput(TOO_MANY_HOOKS);
if (++currentHook < MAX_HOOK) if (lpname && *lpname) strcpy_s<HOOK_NAME_SIZE>(hp.name, lpname);
{ ConsoleOutput(INSERTING_HOOK, hp.name);
if (name.empty()) name = "UserHook " + std::to_string(userhookCount++); RemoveHook(hp.address, 0);
ConsoleOutput(INSERTING_HOOK + name); if (!hooks[currentHook].Insert(hp, flag)) ConsoleOutput(HOOK_FAILED);
// jichi 7/13/2014: This function would raise when too many hooks added
hooks[currentHook].InitHook(hp, name.c_str(), flag);
if (!hooks[currentHook].InsertHook()) ConsoleOutput(HOOK_FAILED);
}
else ConsoleOutput(TOO_MANY_HOOKS);
} }
void RemoveHook(uint64_t addr) void RemoveHook(uint64_t addr, int maxOffset)
{ {
for (int i = 0; i < MAX_HOOK; i++) for (int i = 0; i < MAX_HOOK; i++)
if (abs((long long)(hooks[i].hp.insertion_address - addr)) < 9) return hooks[i].ClearHook(); if (abs((long long)(hooks[i].hp.insertion_address - addr)) <= maxOffset) return hooks[i].Clear();
} }
// EOF // EOF