/**

 * vim: set ts=4 :

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

 * SourceMod SDKTools 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 "vcallbuilder.h"

#include "extension.h"

ValveCall::ValveCall()

{

	call = NULL;

	vparams = NULL;

	retinfo = NULL;

	thisinfo = NULL;

	retbuf = NULL;

}

ValveCall::~ValveCall()

{

	while (!stk.empty())

	{

		unsigned char *ptr = stk.front();

		delete [] ptr;

		stk.pop();

	}

	if (call)

	{

		call->Destroy();

	}

	delete [] retbuf;

	delete [] vparams;

}

unsigned char *ValveCall::stk_get()

{

	unsigned char *ptr;

	if (stk.empty())

	{

		ptr = new unsigned char[stackSize];

	} else {

		ptr = stk.front();

		stk.pop();

	}

	return ptr;

}

void ValveCall::stk_put(unsigned char *ptr)

{

	stk.push(ptr);

}

ValveCall *CreateValveCall(void *addr,

						   ValveCallType vcalltype,

						   const ValvePassInfo *retInfo,

						   const ValvePassInfo *params,

						   unsigned int numParams)

{

	if (numParams > 32)

	{

		return NULL;

	}

	ValveCall *vc = new ValveCall;

	vc->type = vcalltype;

	size_t size = 0;

	vc->stackSize = 0;

	/* Get return information - encode only */

	PassInfo retBuf;

	size_t retBufSize = 0;

	bool retbuf_needs_extra;

	if (retInfo)

	{

		if ((size = ValveParamToBinParam(retInfo->vtype, retInfo->type, retInfo->flags, &retBuf, retbuf_needs_extra)) == 0)

		{

			delete vc;

			return NULL;

		}

		retBufSize = retBuf.size;

	}

	/* Get parameter info */

	PassInfo paramBuf[32];

	size_t sizes[32];

	size_t normSize = 0;

	size_t extraSize = 0;

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

	{

		bool needs_extra;

		if ((size = ValveParamToBinParam(params[i].vtype, 

			params[i].type,

			params[i].flags,

			&paramBuf[i],

			needs_extra)) == 0)

		{

			delete vc;

			return NULL;

		}

		if (needs_extra)

		{

			sizes[i] = size;

		} else {

			sizes[i] = 0;

		}

		normSize += paramBuf[i].size;

		extraSize += sizes[i];

	}

	/* Get thisinfo if needed */

	ValvePassInfo thisbuf;

	ValvePassInfo *thisinfo = NULL;

	CallConvention cv = CallConv_Cdecl;

	if (vcalltype != ValveCall_Static)

	{

		thisinfo = &thisbuf;

		thisinfo->type = PassType_Basic;

		switch (vcalltype)

		{

		case ValveCall_Entity:

			thisinfo->vtype = Valve_CBaseEntity;

			thisinfo->flags = PASSFLAG_BYVAL;

			thisinfo->decflags |= VDECODE_FLAG_ALLOWWORLD;

			break;

		case ValveCall_Player:

			thisinfo->vtype = Valve_CBasePlayer;

			thisinfo->flags = PASSFLAG_BYVAL;

			thisinfo->decflags = 0;

			break;

		default:

			thisinfo->vtype = Valve_POD;

			thisinfo->flags = PASSFLAG_ASPOINTER;

			thisinfo->decflags = 0;

			break;

		}

		thisinfo->encflags = 0;

		thisinfo->offset = 0;

		normSize += sizeof(void *);

		cv = CallConv_ThisCall;

	}

	/* Now we can try creating the call */

	if ((vc->call = g_pBinTools->CreateCall(addr,

		cv,

		(retInfo ? &retBuf : NULL),

		paramBuf,

		numParams))

		== NULL)

	{

		if (!vc->call)

		{

			delete vc;

			return NULL;

		}

	}

	/* Allocate extra space for thisptr AND ret buffer, even if we don't use it */

	vc->vparams = new ValvePassInfo[numParams + 2];

	/* We've got the call and everything is encoded.

	 * It's time to save the valve specific information and helper variables.

	 */

	if (retInfo)

	{

		/* Allocate and copy */

		vc->retinfo = &(vc->vparams[numParams]);

		*vc->retinfo = *retInfo;

		vc->retinfo->offset = 0;

		vc->retinfo->obj_offset = retbuf_needs_extra ? sizeof(void *) : 0;

		/* Allocate stack space */

		vc->retbuf = new unsigned char[retBufSize];

	} else {

		vc->retinfo = NULL;

		vc->retbuf = NULL;

	}

	if (thisinfo)

	{

		/* Allocate and copy */

		vc->thisinfo = &(vc->vparams[numParams + 1]);

		*vc->thisinfo = *thisinfo;

		vc->thisinfo->offset = 0;

		vc->thisinfo->obj_offset = 0;

	} else {

		vc->thisinfo = NULL;

	}

	/* Now, save info about each parameter. */

	size_t last_extra_offset = 0;

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

	{

		/* Copy */

		vc->vparams[i] = params[i];

		vc->vparams[i].offset = vc->call->GetParamInfo(i)->offset;

		vc->vparams[i].obj_offset = last_extra_offset;

		last_extra_offset += sizes[i];

	}

	vc->stackSize = normSize + extraSize;

	vc->stackEnd = normSize;

	return vc;

}

ValveCall *CreateValveVCall(unsigned int vtableIdx,

							ValveCallType vcalltype,

							const ValvePassInfo *retInfo,

							const ValvePassInfo *params,

							unsigned int numParams)

{

	if (numParams > 32)

	{

		return NULL;

	}

	ValveCall *vc = new ValveCall;

	vc->type = vcalltype;

	size_t size = 0;

	vc->stackSize = 0;

	/* Get return information - encode only */

	PassInfo retBuf;

	size_t retBufSize = 0;

	bool retbuf_needs_extra;

	if (retInfo)

	{

		if ((size = ValveParamToBinParam(retInfo->vtype, retInfo->type, retInfo->flags, &retBuf, retbuf_needs_extra)) == 0)

		{

			delete vc;

			return NULL;

		}

		retBufSize = retBuf.size;

	}

	/* Get parameter info */

	PassInfo paramBuf[32];

	size_t sizes[32];

	size_t normSize = 0;

	size_t extraSize = 0;

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

	{

		bool needs_extra;

		if ((size = ValveParamToBinParam(params[i].vtype, 

										params[i].type,

										params[i].flags,

										&paramBuf[i],

										needs_extra)) == 0)

		{

			delete vc;

			return NULL;

		}

		if (needs_extra)

		{

			sizes[i] = size;

		} else {

			sizes[i] = 0;

		}

		normSize += paramBuf[i].size;

		extraSize += sizes[i];

	}

	/* Now we can try creating the call */

	if ((vc->call = g_pBinTools->CreateVCall(vtableIdx, 

		0, 

		0, 

		(retInfo ? &retBuf : NULL),

		paramBuf,

		numParams))

		== NULL)

	{

		if (!vc->call)

		{

			delete vc;

			return NULL;

		}

	}

	/* Allocate extra space for thisptr AND ret buffer, even if we don't use it */

	vc->vparams = new ValvePassInfo[numParams + 2];

	/* We've got the call and everything is encoded.

	 * It's time to save the valve specific information and helper variables.

	 */

	if (retInfo)

	{

		/* Allocate and copy */

		vc->retinfo = &(vc->vparams[numParams]);

		*vc->retinfo = *retInfo;

		vc->retinfo->offset = 0;

		vc->retinfo->obj_offset = retbuf_needs_extra ? sizeof(void *) : 0;

		/* Allocate stack space */

		vc->retbuf = new unsigned char[retBufSize];

	} else {

		vc->retinfo = NULL;

		vc->retbuf = NULL;

	}

	/* Save the this info for the dynamic decoder */

	vc->thisinfo = &(vc->vparams[numParams + 1]);

	vc->thisinfo->type = PassType_Basic;

	switch (vcalltype)

	{

	case ValveCall_Entity:

		vc->thisinfo->vtype = Valve_CBaseEntity;

		vc->thisinfo->flags = PASSFLAG_BYVAL;

		vc->thisinfo->decflags = VDECODE_FLAG_ALLOWWORLD;

		break;

	case ValveCall_Player:

		vc->thisinfo->vtype = Valve_CBasePlayer;

		vc->thisinfo->flags = PASSFLAG_BYVAL;

		vc->thisinfo->decflags = 0;

		break;

	default:

		vc->thisinfo->vtype = Valve_POD;

		vc->thisinfo->flags = PASSFLAG_ASPOINTER;

		vc->thisinfo->decflags = 0;

		break;

	}

	vc->thisinfo->encflags = 0;

	vc->thisinfo->offset = 0;

	vc->thisinfo->obj_offset = 0;

	normSize += sizeof(void *);

	/* Now, save info about each parameter. */

	size_t last_extra_offset = 0;

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

	{

		/* Copy */

		vc->vparams[i] = params[i];

		vc->vparams[i].offset = vc->call->GetParamInfo(i)->offset;

		vc->vparams[i].obj_offset = last_extra_offset;

		last_extra_offset += sizes[i];

	}

	vc->stackSize = normSize + extraSize;

	vc->stackEnd = normSize;

	return vc;

}