mirror of
https://github.com/Artikash/Textractor.git
synced 2024-12-24 01:14:12 +08:00
63 lines
1.9 KiB
C++
63 lines
1.9 KiB
C++
#pragma once
|
|
|
|
// host.h
|
|
// 8/23/2013 jichi
|
|
// Branch: ITH/IHF.h, rev 105
|
|
|
|
#define DLLEXPORT __declspec(dllexport)
|
|
|
|
#include <Windows.h>
|
|
#include "textthread.h"
|
|
#include <string>
|
|
#include <functional>
|
|
#include "../vnrhook/include/types.h"
|
|
|
|
struct ProcessRecord
|
|
{
|
|
HANDLE process_handle;
|
|
HANDLE hookman_mutex;
|
|
HANDLE hookman_section;
|
|
LPVOID hookman_map;
|
|
HANDLE hostPipe;
|
|
};
|
|
|
|
typedef std::function<void(DWORD)> ProcessEventCallback;
|
|
typedef std::function<void(TextThread*)> ThreadEventCallback;
|
|
|
|
struct ThreadParameterHasher
|
|
{
|
|
size_t operator()(const ThreadParameter& tp) const
|
|
{
|
|
return std::hash<__int64>()(tp.pid << 6) + std::hash<__int64>()(tp.hook) + std::hash<__int64>()(tp.retn) + std::hash<__int64>()(tp.spl);
|
|
}
|
|
};
|
|
|
|
namespace Host
|
|
{
|
|
DLLEXPORT void Start(ProcessEventCallback onAttach, ProcessEventCallback onDetach, ThreadEventCallback onCreate, ThreadEventCallback onRemove);
|
|
DLLEXPORT void Close();
|
|
DLLEXPORT bool InjectProcess(DWORD pid, DWORD timeout = 5000);
|
|
DLLEXPORT bool DetachProcess(DWORD pid);
|
|
|
|
DLLEXPORT bool InsertHook(DWORD pid, HookParam hp, std::string name = "");
|
|
DLLEXPORT bool RemoveHook(DWORD pid, DWORD addr);
|
|
DLLEXPORT HookParam GetHookParam(DWORD pid, DWORD addr);
|
|
DLLEXPORT HookParam GetHookParam(ThreadParameter tp);
|
|
DLLEXPORT std::wstring GetHookName(DWORD pid, DWORD addr);
|
|
|
|
DLLEXPORT TextThread* GetThread(ThreadParameter tp);
|
|
DLLEXPORT void AddConsoleOutput(std::wstring text);
|
|
|
|
DLLEXPORT void RegisterThreadCreateCallback(ThreadEventCallback cf);
|
|
DLLEXPORT void RegisterThreadRemoveCallback(ThreadEventCallback cf);
|
|
DLLEXPORT void RegisterProcessAttachCallback(ProcessEventCallback cf);
|
|
DLLEXPORT void RegisterProcessDetachCallback(ProcessEventCallback cf);
|
|
}
|
|
|
|
void DispatchText(ThreadParameter tp, const BYTE *text, int len);
|
|
void RemoveThreads(bool(*RemoveIf)(ThreadParameter, ThreadParameter), ThreadParameter cmp);
|
|
void RegisterProcess(DWORD pid, HANDLE hostPipe);
|
|
void UnregisterProcess(DWORD pid);
|
|
|
|
// EOF
|