#include "stdafx.h"
#include <Windows.h>
#include "CAnyHook.h"
CAnyHook::CAnyHook()
{
	memset(oldData, 0, 7);
	memset(newData, 0, 7);
	isHooked = false;
}

CAnyHook::CAnyHook(void* hookAddr, void* hookTarget)
{
	CAnyHook();
	hook(hookAddr, hookTarget);
}

void CAnyHook::hook(void* hookAddr, void* hookTarget)
{
	if (!isHooked)
	{
		mHookAddr = hookAddr;
		mTargetaddr = hookTarget;
		writeData(oldData, hookAddr, 7);
		newData[0] = 0x8b;
		newData[1] = 0xff;
		newData[2] = 0xe9;
		*(DWORD*)& newData[3] = (DWORD)hookTarget - (DWORD)hookAddr - 5 - 2;
		writeData(hookAddr, newData, 7);
		isHooked = true;
	}
}



void CAnyHook::writeData(void* patchAddr, void* patchData, DWORD patchSize)
{
	DWORD oldProtect = 0;
	VirtualProtect(patchAddr, patchSize, PAGE_EXECUTE_READWRITE, &oldProtect);
	memcpy(patchAddr, patchData, patchSize);
}

void CAnyHook::unHook()
{
	if(isHooked)
	{
		writeData(mHookAddr, oldData, 7);
		isHooked = false;
	}
}

void CAnyHook::reHook()
{
	if(!isHooked)
	{
		writeData(mHookAddr, newData, 7);
		isHooked = true;
	}
}