LunaHook-mirror/LunaHook/veh_hook.h

/**
veh_hook Vectored Exception Handler hooking library
Version: 24-March-2008
**/

#ifndef LIST_T_H_INCLUDED
#define LIST_T_H_INCLUDED
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include<functional>
// VEH Hooking types
#define VEH_HK_INT3 0
#define VEH_HK_MEM 1
#define VEH_HK_HW 2
// -

#define OPCODE_INT3 "\xCC"


//typedef void (*pfvoid)();
//typedef void (*newFuncType)(PCONTEXT);
using newFuncType = std::function<bool(PCONTEXT)>;

typedef struct veh_node
{
	struct veh_node* last;
	struct veh_node* next;
	void* origFunc;
	newFuncType newFunc;
	void* handle;
	DWORD hooktype;
	void* baseAddr; // Address of the page in which origFunc resides.
	BYTE origBaseByte;
	DWORD OldProtect;
} veh_node_t;

typedef struct
{
	veh_node_t* head;
	veh_node_t* tail;
} veh_list_t;

// VEH hook interface functions for creating and removing hooks.
bool add_veh_hook(void* origFunc, newFuncType newFunc, DWORD hook_type=VEH_HK_INT3);
bool remove_veh_hook(void* origFunc);

// The VEH dispathing function is called by Windows every time an exception is encountered.
// the function dispatches calls to the correct inctercept function.
LONG CALLBACK veh_dispatch(PEXCEPTION_POINTERS ExceptionInfo);

// Functions used internally by the library.
veh_list_t* new_veh_list();
veh_node_t* create_veh_node(void* origFunc, newFuncType newFunc, void* handle, DWORD hook_type);
void insert_veh_node(veh_list_t* list, veh_node_t*);
void remove_veh_node(veh_list_t* list, void* origFunc);
veh_node_t* get_veh_node(veh_list_t* list, void* origFunc, int range=0);

#endif // LIST_T_H_INCLUDED