dvander@2545: /**
dvander@2545:  * vim: set ts=4 :
dvander@2545:  * =============================================================================
dvander@2545:  * SourcePawn
dvander@2545:  * Copyright (C) 2004-2009 AlliedModders LLC.  All rights reserved.
dvander@2545:  * =============================================================================
dvander@2545:  *
dvander@2545:  * This program is free software; you can redistribute it and/or modify it under
dvander@2545:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@2545:  * Free Software Foundation.
dvander@2545:  * 
dvander@2545:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@2545:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@2545:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@2545:  * details.
dvander@2545:  *
dvander@2545:  * You should have received a copy of the GNU General Public License along with
dvander@2545:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@2545:  *
dvander@2545:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@2545:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@2545:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@2545:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@2545:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@2545:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@2545:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@2545:  * or <http://www.sourcemod.net/license.php>.
dvander@2545:  *
dvander@2545:  * Version: $Id$
dvander@2545:  */
dvander@2545: 
dvander@2545: #include "NativeInvoker.h"
dvander@2545: #include "ShareSys.h"
dvander@2545: 
dvander@2545: NativeInterface g_NInvoke;
dvander@2545: 
dvander@2545: NativeInvoker::NativeInvoker()
dvander@2545: {
dvander@2545: }
dvander@2545: 
dvander@2545: NativeInvoker::~NativeInvoker()
dvander@2545: {
dvander@2545: }
dvander@2545: 
dvander@2545: const char *NativeInterface::GetInterfaceName()
dvander@2545: {
dvander@2545: 	return SMINTERFACE_NINVOKE_NAME;
dvander@2545: }
dvander@2545: 
dvander@2545: unsigned int NativeInterface::GetInterfaceVersion()
dvander@2545: {
dvander@2545: 	return SMINTERFACE_NINVOKE_VERSION;
dvander@2545: }
dvander@2545: 
dvander@2545: void NativeInterface::OnSourceModAllInitialized()
dvander@2545: {
dvander@3872: 	sharesys->AddInterface(NULL, &g_NInvoke);
dvander@2545: }
dvander@2545: 
dvander@2545: IPluginRuntime *NativeInterface::CreateRuntime(const char *name, size_t bytes)
dvander@2545: {
dvander@2545: 	return g_pSourcePawn2->CreateEmptyRuntime(name, bytes);
dvander@2545: }
dvander@2545: 
dvander@2545: INativeInvoker *NativeInterface::CreateInvoker()
dvander@2545: {
dvander@2545: 	return new NativeInvoker();
dvander@2545: }
dvander@2545: 
dvander@2545: bool NativeInvoker::Start(IPluginContext *pContext, const char *name)
dvander@2545: {
dvander@2545: 	NativeEntry *entry;
dvander@2545: 	
dvander@2545: 	entry = g_ShareSys.FindNative(name);
dvander@2545: 	if (entry == NULL)
dvander@2545: 	{
dvander@2545: 		return false;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	native = NULL;
dvander@2545: 	if (entry->replacement.owner != NULL)
dvander@2545: 	{
dvander@2545: 		native = entry->replacement.func;
dvander@2545: 	}
dvander@2545: 	else if (entry->owner != NULL)
dvander@2545: 	{
dvander@2545: 		native = entry->func;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	if (native == NULL)
dvander@2545: 	{
dvander@2545: 		return false;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	this->pContext = pContext;
dvander@2545: 
dvander@2545: 	m_curparam = 0;
dvander@2545: 	m_errorstate = SP_ERROR_NONE;
dvander@2545: 
dvander@2545: 	return true;
dvander@2545: }
dvander@2545: 
dvander@2545: cell_t NativeInvoker::PushCell(cell_t cell)
dvander@2545: {
dvander@2545: 	if (m_curparam >= SP_MAX_EXEC_PARAMS)
dvander@2545: 	{
dvander@2545: 		return SetError(SP_ERROR_PARAMS_MAX);
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	m_info[m_curparam].marked = false;
dvander@2545: 	m_params[m_curparam] = cell;
dvander@2545: 	m_curparam++;
dvander@2545: 
dvander@2545: 	return SP_ERROR_NONE;
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::PushCellByRef(cell_t *cell, int flags)
dvander@2545: {
dvander@2545: 	return PushArray(cell, 1, flags);
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::PushFloat(float number)
dvander@2545: {
dvander@2545: 	cell_t val = *(cell_t *)&number;
dvander@2545: 
dvander@2545: 	return PushCell(val);
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::PushFloatByRef(float *number, int flags)
dvander@2545: {
dvander@2545: 	return PushCellByRef((cell_t *)number, flags);
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::PushArray(cell_t *inarray, unsigned int cells, int copyback)
dvander@2545: {
dvander@2545: 	if (m_curparam >= SP_MAX_EXEC_PARAMS)
dvander@2545: 	{
dvander@2545: 		return SetError(SP_ERROR_PARAMS_MAX);
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	ParamInfo *info = &m_info[m_curparam];
dvander@2545: 
dvander@2545: 	info->flags = inarray ? copyback : 0;
dvander@2545: 	info->marked = true;
dvander@2545: 	info->size = cells;
dvander@2545: 	info->str.is_sz = false;
dvander@2545: 	info->orig_addr = inarray;
dvander@2545: 
dvander@2545: 	m_curparam++;
dvander@2545: 
dvander@2545: 	return SP_ERROR_NONE;
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::PushString(const char *string)
dvander@2545: {
dvander@2545: 	return _PushString(string, SM_PARAM_STRING_COPY, 0, strlen(string)+1);
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::PushStringEx(char *buffer, size_t length, int sz_flags, int cp_flags)
dvander@2545: {
dvander@2545: 	return _PushString(buffer, sz_flags, cp_flags, length);
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::_PushString(const char *string, int sz_flags, int cp_flags, size_t len)
dvander@2545: {
dvander@2545: 	if (m_curparam >= SP_MAX_EXEC_PARAMS)
dvander@2545: 	{
dvander@2545: 		return SetError(SP_ERROR_PARAMS_MAX);
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	ParamInfo *info = &m_info[m_curparam];
dvander@2545: 
dvander@2545: 	info->marked = true;
dvander@2545: 	info->orig_addr = (cell_t *)string;
dvander@2545: 	info->flags = cp_flags;
dvander@2545: 	info->size = len;
dvander@2545: 	info->str.sz_flags = sz_flags;
dvander@2545: 	info->str.is_sz = true;
dvander@2545: 
dvander@2545: 	m_curparam++;
dvander@2545: 
dvander@2545: 	return SP_ERROR_NONE;
dvander@2545: }
dvander@2545: 
dvander@2545: void NativeInvoker::Cancel()
dvander@2545: {
dvander@2545: 	if (pContext == NULL)
dvander@2545: 	{
dvander@2545: 		return;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	m_errorstate = SP_ERROR_NONE;
dvander@2545: 	m_curparam = 0;
dvander@2545: 	pContext = NULL;
dvander@2545: 	native = NULL;
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::SetError(int err)
dvander@2545: {
dvander@2545: 	m_errorstate = err;
dvander@2545: 	return err;
dvander@2545: }
dvander@2545: 
dvander@2545: int NativeInvoker::Invoke(cell_t *result)
dvander@2545: {
dvander@2545: 	int err = SP_ERROR_NONE;
dvander@2545: 
dvander@2545: 	if (pContext == NULL)
dvander@2545: 	{
dvander@2545: 		return SP_ERROR_INVALID_NATIVE;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	if (m_errorstate != SP_ERROR_NONE)
dvander@2545: 	{
dvander@2545: 		err = m_errorstate;
dvander@2545: 		Cancel();
dvander@2545: 		return err;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	cell_t tresult;
dvander@2545: 
dvander@2545: 	if (result == NULL)
dvander@2545: 	{
dvander@2545: 		result = &tresult;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	//This is for re-entrancy!
dvander@2545: 	IPluginContext *ctx = pContext;
dvander@2545: 	cell_t _temp_params[SP_MAX_EXEC_PARAMS + 1];
dvander@2545: 	cell_t *temp_params = &_temp_params[1];
dvander@2545: 	ParamInfo temp_info[SP_MAX_EXEC_PARAMS];
dvander@2545: 	unsigned int numparams = m_curparam;
dvander@2545: 	unsigned int i;
dvander@2545: 	bool docopies = true;
dvander@2545: 
dvander@2545: 	if (numparams)
dvander@2545: 	{
dvander@2545: 		//Save the info locally, then reset it for re-entrant calls.
dvander@2545: 		memcpy(temp_info, m_info, numparams * sizeof(ParamInfo));
dvander@2545: 	}
dvander@2545: 	m_curparam = 0;
dvander@2545: 	pContext = NULL;
dvander@2545: 
dvander@2545: 	/* Initialize 0th parameter */
dvander@2545: 	_temp_params[0] = numparams;
dvander@2545: 
dvander@2545: 	/* Browse the parameters and build arrays */
dvander@2545: 	for (i = 0; i < numparams; i++)
dvander@2545: 	{
dvander@2545: 		/* Is this marked as an array? */
dvander@2545: 		if (temp_info[i].marked)
dvander@2545: 		{
dvander@2545: 			if (!temp_info[i].str.is_sz)
dvander@2545: 			{
dvander@2545: 				/* Allocate a normal/generic array */
dvander@2545: 				if ((err = ctx->HeapAlloc(temp_info[i].size, 
dvander@2545: 										  &temp_info[i].local_addr,
dvander@2545: 										  &temp_info[i].phys_addr))
dvander@2545: 					!= SP_ERROR_NONE)
dvander@2545: 				{
dvander@2545: 					break;
dvander@2545: 				}
dvander@2545: 				if (temp_info[i].orig_addr)
dvander@2545: 				{
dvander@2545: 					memcpy(temp_info[i].phys_addr, temp_info[i].orig_addr, sizeof(cell_t) * temp_info[i].size);
dvander@2545: 				}
dvander@2545: 			}
dvander@2545: 			else
dvander@2545: 			{
dvander@2545: 				/* Calculate cells required for the string */
dvander@2545: 				size_t cells = (temp_info[i].size + sizeof(cell_t) - 1) / sizeof(cell_t);
dvander@2545: 
dvander@2545: 				/* Allocate the buffer */
dvander@2545: 				if ((err = ctx->HeapAlloc(cells,
dvander@2545: 										  &temp_info[i].local_addr,
dvander@2545: 										  &temp_info[i].phys_addr))
dvander@2545: 					!= SP_ERROR_NONE)
dvander@2545: 				{
dvander@2545: 					break;
dvander@2545: 				}
dvander@2545: 				/* Copy original string if necessary */
dvander@2545: 				if ((temp_info[i].str.sz_flags & SM_PARAM_STRING_COPY) && (temp_info[i].orig_addr != NULL))
dvander@2545: 				{
dvander@2545: 					/* Cut off UTF-8 properly */
dvander@2545: 					if (temp_info[i].str.sz_flags & SM_PARAM_STRING_UTF8)
dvander@2545: 					{
dvander@2545: 						if ((err = ctx->StringToLocalUTF8(temp_info[i].local_addr, 
dvander@2545: 														  temp_info[i].size, 
dvander@2545: 														  (const char *)temp_info[i].orig_addr,
dvander@2545: 														  NULL))
dvander@2545: 							!= SP_ERROR_NONE)
dvander@2545: 						{
dvander@2545: 							break;
dvander@2545: 						}
dvander@2545: 					}
dvander@2545: 					/* Copy a binary blob */
dvander@2545: 					else if (temp_info[i].str.sz_flags & SM_PARAM_STRING_BINARY)
dvander@2545: 					{
dvander@2545: 						memmove(temp_info[i].phys_addr, temp_info[i].orig_addr, temp_info[i].size);
dvander@2545: 					}
dvander@2545: 					/* Copy ASCII characters */
dvander@2545: 					else
dvander@2545: 					{
dvander@2545: 						if ((err = ctx->StringToLocal(temp_info[i].local_addr,
dvander@2545: 													  temp_info[i].size,
dvander@2545: 													  (const char *)temp_info[i].orig_addr))
dvander@2545: 							!= SP_ERROR_NONE)
dvander@2545: 						{
dvander@2545: 							break;
dvander@2545: 						}
dvander@2545: 					}
dvander@2545: 				}
dvander@2545: 			} /* End array/string calculation */
dvander@2545: 			/* Update the pushed parameter with the byref local address */
dvander@2545: 			temp_params[i] = temp_info[i].local_addr;
dvander@2545: 		}
dvander@2545: 		else
dvander@2545: 		{
dvander@2545: 			/* Just copy the value normally */
dvander@2545: 			temp_params[i] = m_params[i];
dvander@2545: 		}
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	/* Make the call if we can */
dvander@2545: 	if (err == SP_ERROR_NONE)
dvander@2545: 	{
dvander@2545: 		*result = native(ctx, _temp_params);
dvander@2545: 		if (ctx->GetLastNativeError() != SP_ERROR_NONE)
dvander@2545: 		{
dvander@2545: 			docopies = false;
dvander@2545: 			ctx->ClearLastNativeError();
dvander@2545: 		}
dvander@2545: 	}
dvander@2545: 	else
dvander@2545: 	{
dvander@2545: 		docopies = false;
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	/* i should be equal to the last valid parameter + 1 */
dvander@2545: 	while (i--)
dvander@2545: 	{
dvander@2545: 		if (!temp_info[i].marked)
dvander@2545: 		{
dvander@2545: 			continue;
dvander@2545: 		}
dvander@2545: 
dvander@2545: 		if (docopies && (temp_info[i].flags & SM_PARAM_COPYBACK))
dvander@2545: 		{
dvander@2545: 			if (temp_info[i].orig_addr)
dvander@2545: 			{
dvander@2545: 				if (temp_info[i].str.is_sz)
dvander@2545: 				{
dvander@2545: 					memcpy(temp_info[i].orig_addr, temp_info[i].phys_addr, temp_info[i].size);
dvander@2545: 
dvander@2545: 				}
dvander@2545: 				else
dvander@2545: 				{
dvander@2545: 					if (temp_info[i].size == 1)
dvander@2545: 					{
dvander@2545: 						*temp_info[i].orig_addr = *(temp_info[i].phys_addr);
dvander@2545: 					}
dvander@2545: 					else
dvander@2545: 					{
dvander@2545: 						memcpy(temp_info[i].orig_addr, 
dvander@2545: 							temp_info[i].phys_addr, 
dvander@2545: 							temp_info[i].size * sizeof(cell_t));
dvander@2545: 					}
dvander@2545: 				}
dvander@2545: 			}
dvander@2545: 		}
dvander@2545: 
dvander@2545: 		if ((err = ctx->HeapPop(temp_info[i].local_addr)) != SP_ERROR_NONE)
dvander@2545: 		{
dvander@2545: 			return err;
dvander@2545: 		}
dvander@2545: 	}
dvander@2545: 
dvander@2545: 	return err;
dvander@2545: }