/**

  * vim: set ts=4 :

  * =============================================================================

  * SourceMod BinTools Extension

  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.

  * =============================================================================

  *

  * This program is free software; you can redistribute it and/or modify it under

  * the terms of the GNU General Public License, version 3.0, as published by the

  * Free Software Foundation.

  * 

  * This program is distributed in the hope that it will be useful, but WITHOUT

  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

  * details.

  *

  * You should have received a copy of the GNU General Public License along with

  * this program.  If not, see <http://www.gnu.org/licenses/>.

  *

  * As a special exception, AlliedModders LLC gives you permission to link the

  * code of this program (as well as its derivative works) to "Half-Life 2," the

  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

  * by the Valve Corporation.  You must obey the GNU General Public License in

  * all respects for all other code used.  Additionally, AlliedModders LLC grants

  * this exception to all derivative works.  AlliedModders LLC defines further

  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

  * or <http://www.sourcemod.net/license.php>.

  *

  * Version: $Id$

  */

 #include <stdio.h>

 #include "CallMaker.h"

 #include "jit_compile.h"

 #include "extension.h"

 /* SourceMod <==> SourceHook type conversion functions */

 SourceHook::ProtoInfo::CallConvention GetSHCallConvention(SourceMod::CallConvention cv)

 {

 	if (cv < SourceMod::CallConv_ThisCall || cv > SourceMod::CallConv_Cdecl)

 	{

 		return SourceHook::ProtoInfo::CallConv_Unknown;

 	}

 	return (SourceHook::ProtoInfo::CallConvention)(cv + 1);

 }

 SourceMod::CallConvention GetSMCallConvention(SourceHook::ProtoInfo::CallConvention cv)

 {

 	assert(cv >= SourceHook::ProtoInfo::CallConv_ThisCall || cv <= SourceHook::ProtoInfo::CallConv_Cdecl);

 	return (SourceMod::CallConvention)(cv - 1);

 }

 SourceHook::PassInfo::PassType GetSHPassType(SourceMod::PassType type)

 {

 	if (type < SourceMod::PassType_Basic || type > SourceMod::PassType_Object)

 	{

 		return SourceHook::PassInfo::PassType_Unknown;

 	}

 	return (SourceHook::PassInfo::PassType)(type + 1);

 }

 SourceMod::PassType GetSMPassType(int type)

 {

 	/* SourceMod doesn't provide an Unknown type so we it must be an error */

 	assert(type >= SourceHook::PassInfo::PassType_Basic && type <= SourceHook::PassInfo::PassType_Object);

 	return (SourceMod::PassType)(type - 1);

 }

 void GetSMPassInfo(SourceMod::PassInfo *out, const SourceHook::PassInfo *in)

 {

 	out->size = in->size;

 	out->flags = in->flags;

 	out->type = GetSMPassType(in->type);

 }

 ICallWrapper *CallMaker::CreateCall(void *address,

 						 CallConvention cv,

 						 const PassInfo *retInfo,

 						 const PassInfo paramInfo[],

 						 unsigned int numParams)

 {

 	SourceHook::CProtoInfoBuilder protoInfo(GetSHCallConvention(cv));

 	for (unsigned int i=0; i<numParams; i++)

 	{

 		protoInfo.AddParam(paramInfo[i].size, GetSHPassType(paramInfo[i].type), paramInfo[i].flags,

 			NULL, NULL, NULL, NULL);

 	}

 	if (retInfo)

 	{

 		protoInfo.SetReturnType(retInfo->size, GetSHPassType(retInfo->type), retInfo->flags,

 			NULL, NULL, NULL, NULL);

 	}

 	else

 	{

 		protoInfo.SetReturnType(0, SourceHook::PassInfo::PassType_Unknown, 0,

 			NULL, NULL, NULL, NULL);

 	}

 	return g_CallMaker2.CreateCall(address, &(*protoInfo));

 }

 ICallWrapper *CallMaker::CreateVCall(unsigned int vtblIdx, 

 									 unsigned int vtblOffs, 

 									 unsigned int thisOffs, 

 									 const PassInfo *retInfo, 

 									 const PassInfo paramInfo[], 

 									 unsigned int numParams)

 {

 	SourceHook::MemFuncInfo info;

 	info.isVirtual = true;

 	info.vtblindex = vtblIdx;

 	info.vtbloffs = vtblOffs;

 	info.thisptroffs = thisOffs;

 	SourceHook::CProtoInfoBuilder protoInfo(SourceHook::ProtoInfo::CallConv_ThisCall);

 	for (unsigned int i=0; i<numParams; i++)

 	{

 		protoInfo.AddParam(paramInfo[i].size, GetSHPassType(paramInfo[i].type), paramInfo[i].flags,

 			NULL, NULL, NULL, NULL);

 	}

 	if (retInfo)

 	{

 		protoInfo.SetReturnType(retInfo->size, GetSHPassType(retInfo->type), retInfo->flags,

 			NULL, NULL, NULL, NULL);

 	}

 	else

 	{

 		protoInfo.SetReturnType(0, SourceHook::PassInfo::PassType_Unknown, 0,

 			NULL, NULL, NULL, NULL);

 	}

 	return g_CallMaker2.CreateVirtualCall(&(*protoInfo), &info);

 }

 ICallWrapper *CallMaker2::CreateCall(void *address, const SourceHook::ProtoInfo *protoInfo)

 {

 	CallWrapper *pWrapper = new CallWrapper(protoInfo);

 	pWrapper->SetCalleeAddr(address);

 	void *addr = JIT_CallCompile(pWrapper, FuncAddr_Direct);

 	pWrapper->SetCodeBaseAddr(addr);

 	return pWrapper;

 }

 ICallWrapper *CallMaker2::CreateVirtualCall(const SourceHook::ProtoInfo *protoInfo, 

 											const SourceHook::MemFuncInfo *info)

 {

 	CallWrapper *pWrapper = new CallWrapper(protoInfo);

 	pWrapper->SetMemFuncInfo(info);

 	void *addr = JIT_CallCompile(pWrapper, FuncAddr_VTable);

 	pWrapper->SetCodeBaseAddr(addr);

 	return pWrapper;

 }

 #if defined HOOKING_ENABLED

 IHookWrapper *CallMaker2::CreateVirtualHook(SourceHook::ISourceHook *pSH, 

 											const SourceHook::ProtoInfo *protoInfo,

 											const SourceHook::MemFuncInfo *info, 

 											VIRTUAL_HOOK_PROTO f)

 {

 	HookWrapper *pWrapper = new HookWrapper(pSH, protoInfo, const_cast<SourceHook::MemFuncInfo *>(info), (void *)f);

 	void *addr = JIT_HookCompile(pWrapper);

 	pWrapper->SetHookWrpAddr(addr);

 	ICallWrapper *callWrap = CreateVirtualCall(protoInfo, info);

 	pWrapper->SetCallWrapperAddr(callWrap);

 	return pWrapper;

 }

 #endif