dvander@601: /**
ds@653:  * vim: set ts=4 :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
ds@1947:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
dvander@601:  *
ds@1330:  * This program is free software; you can redistribute it and/or modify it under
ds@1330:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330:  * Free Software Foundation.
dvander@1231:  * 
ds@1330:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330:  * details.
dvander@1231:  *
ds@1330:  * You should have received a copy of the GNU General Public License along with
ds@1330:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
ds@1330:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330:  * or <http://www.sourcemod.net/license.php>.
dvander@601:  *
dvander@601:  * Version: $Id$
dvander@601:  */
dvander@601: 
dvander@601: #include <sh_list.h>
dvander@601: #include <sh_string.h>
dvander@601: #include "sm_trie.h"
dvander@601: #include "sm_globals.h"
dvander@601: #include "PluginSys.h"
dvander@601: #include "sourcemod.h"
dvander@601: #include "sm_stringutil.h"
dvander@601: 
dvander@601: using namespace SourceHook;
dvander@601: 
dvander@601: static cell_t s_curparams[SP_MAX_EXEC_PARAMS+1];
dvander@601: static FakeNative *s_curnative = NULL;
dvander@601: static IPluginContext *s_curcaller = NULL;
dvander@601: 
dvander@601: cell_t FakeNativeRouter(IPluginContext *pContext, const cell_t *params, void *pData)
dvander@601: {
dvander@601: 	FakeNative *native = (FakeNative *)pData;
dvander@601: 
dvander@601: 	/* Check if too many parameters were passed */
dvander@601: 	if (params[0] > SP_MAX_EXEC_PARAMS)
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Called native with too many parameters (%d>%d)", params[9], SP_MAX_EXEC_PARAMS);
dvander@601: 	}
dvander@601: 
dvander@601: 	/* Check if the native is paused */
dvander@2193: 	if (native->ctx->GetRuntime()->IsPaused())
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Plugin owning this native is currently paused.");
dvander@601: 	}
dvander@601: 
dvander@601: 	CPlugin *pCaller = g_PluginSys.GetPluginByCtx(pContext->GetContext());
dvander@601: 
dvander@601: 	/* Save any old data on the stack */
dvander@601: 	FakeNative *pSaveNative = s_curnative;
dvander@601: 	IPluginContext *pSaveCaller = s_curcaller;
dvander@601: 	cell_t save_params[SP_MAX_EXEC_PARAMS+1];
dvander@601: 	if (pSaveNative != NULL)
dvander@601: 	{
dvander@601: 		/* Copy all old parameters */
dvander@601: 		for (cell_t i=0; i<=s_curparams[0]; i++)
dvander@601: 		{
dvander@601: 			save_params[i] = s_curparams[i];
dvander@601: 		}
dvander@601: 	}
dvander@601: 
dvander@601: 	/* Save the current parameters */
dvander@601: 	s_curnative = native;
dvander@601: 	s_curcaller = pContext;
dvander@601: 	for (cell_t i=0; i<=params[0]; i++)
dvander@601: 	{
dvander@601: 		s_curparams[i] = params[i];
dvander@601: 	}
dvander@601: 
dvander@601: 	/* Push info and execute. */
dvander@601: 	cell_t result = 0;
dvander@601: 	native->call->PushCell(pCaller->GetMyHandle());
dvander@601: 	native->call->PushCell(params[0]);
dvander@805: 	int error;
dvander@805: 	if ((error=native->call->Execute(&result)) != SP_ERROR_NONE)
dvander@601: 	{
dvander@2193: 		if (pContext->GetLastNativeError() == SP_ERROR_NONE)
dvander@601: 		{
dvander@805: 			pContext->ThrowNativeErrorEx(error, "Error encountered while processing a dynamic native");
dvander@601: 		}
dvander@601: 	}
dvander@601: 
dvander@601: 	/* Restore everything from the stack if necessary */
dvander@699: 	s_curnative = pSaveNative;
dvander@699: 	s_curcaller = pSaveCaller;
dvander@601: 	if (pSaveNative != NULL)
dvander@601: 	{
dvander@601: 		for (cell_t i=0; i<=save_params[0]; i++)
dvander@601: 		{
dvander@601: 			s_curparams[i] = save_params[i];
dvander@601: 		}
dvander@601: 	}
dvander@601: 
dvander@601: 	return result;
dvander@601: }
dvander@601: 
dvander@601: static cell_t CreateNative(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	char *name;
dvander@2086: 	CPlugin *pPlugin;
dvander@2086: 
dvander@601: 	pContext->LocalToString(params[1], &name);
dvander@601: 
dvander@601: 	IPluginFunction *pFunction = pContext->GetFunctionById(params[2]);
dvander@601: 	if (!pFunction)
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Function %x is not a valid function", params[2]);
dvander@601: 	}
dvander@601: 
dvander@2086: 	pPlugin = g_PluginSys.GetPluginByCtx(pContext->GetContext());
dvander@2086: 
dvander@2086: 	if (!pPlugin->AddFakeNative(pFunction, name, FakeNativeRouter))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Fatal error creating dynamic native!");
dvander@601: 	}
dvander@601: 
dvander@601: 	return 1;
dvander@601: }
dvander@601: 
dvander@601: static cell_t ThrowNativeError(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@2042: 	g_SourceMod.SetGlobalTarget(SOURCEMOD_SERVER_LANGUAGE);
dvander@1011: 
dvander@601: 	char buffer[512];
dvander@601: 
dvander@601: 	g_SourceMod.FormatString(buffer, sizeof(buffer), pContext, params, 2);
dvander@601: 
dvander@2193: 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		s_curcaller->ThrowNativeError("Error encountered while processing a dynamic native");
dvander@601: 	} else {
dvander@601: 		s_curcaller->ThrowNativeErrorEx(params[1], "%s", buffer);
dvander@601: 	}
dvander@601: 
dvander@601: 	return 0;
dvander@601: }
dvander@601: 
dvander@601: static cell_t GetNativeStringLength(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	int err;
dvander@601: 	char *str;
dvander@601: 	if ((err=s_curcaller->LocalToString(s_curparams[param], &str)) != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		return err;
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t *addr;
dvander@601: 	pContext->LocalToPhysAddr(params[2], &addr);
dvander@601: 	*addr = (cell_t)strlen(str);
dvander@601: 
dvander@601: 	return SP_ERROR_NONE;
dvander@601: }
dvander@601: 
dvander@601: static cell_t GetNativeString(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	int err;
dvander@601: 	char *str;
dvander@601: 	if ((err=s_curcaller->LocalToString(s_curparams[param], &str)) != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		return err;
dvander@601: 	}
dvander@601: 
dvander@601: 	size_t bytes = 0;
dvander@601: 	pContext->StringToLocalUTF8(params[2], params[3], str, &bytes);
dvander@601: 
dvander@601: 	cell_t *addr;
dvander@601: 	pContext->LocalToPhysAddr(params[4], &addr);
dvander@601: 	*addr = (cell_t)bytes;
dvander@601: 
dvander@601: 	return SP_ERROR_NONE;
dvander@601: }
dvander@601: 
dvander@601: static cell_t SetNativeString(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	char *str;
dvander@601: 	pContext->LocalToString(params[2], &str);
dvander@601: 
dvander@601: 	int err;
dvander@601: 	size_t bytes = 0;
dvander@601: 	if (params[4])
dvander@601: 	{
dvander@601: 		err = s_curcaller->StringToLocalUTF8(s_curparams[param], params[3], str, &bytes);
dvander@601: 	} else {
dvander@601: 		err = s_curcaller->StringToLocal(s_curparams[param], params[3], str);
dvander@601: 		/* Eww */
dvander@601: 		bytes = strlen(str);
dvander@601: 		if (bytes >= (size_t)params[3])
dvander@601: 		{
dvander@601: 			bytes = params[3] - 1;
dvander@601: 		}
dvander@601: 	}
dvander@601: 
dvander@601: 	if (err != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		return err;
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t *addr;
dvander@601: 	pContext->LocalToPhysAddr(params[5], &addr);
dvander@601: 	*addr = (cell_t)bytes;
dvander@601: 
dvander@601: 	return SP_ERROR_NONE;
dvander@601: }
dvander@601: 
dvander@601: static cell_t GetNativeCell(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	return s_curparams[param];
dvander@601: }
dvander@601: 
dvander@601: static cell_t GetNativeCellRef(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t *addr;
dvander@601: 	if (s_curcaller->LocalToPhysAddr(s_curparams[param], &addr) != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		return s_curcaller->ThrowNativeErrorEx(SP_ERROR_INVALID_ADDRESS, "Invalid address value");
dvander@601: 	}
dvander@601: 
dvander@601: 	return *addr;
dvander@601: }
dvander@601: 
dvander@601: static cell_t SetNativeCellRef(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t *addr;
dvander@601: 	if (s_curcaller->LocalToPhysAddr(s_curparams[param], &addr) != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		return s_curcaller->ThrowNativeErrorEx(SP_ERROR_INVALID_ADDRESS, "Invalid address value");
dvander@601: 	}
dvander@601: 
dvander@601: 	*addr = params[2];
dvander@601: 
dvander@601: 	return 1;
dvander@601: }
dvander@601: 
dvander@601: static cell_t GetNativeArray(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	int err;
dvander@601: 	cell_t *addr;
dvander@601: 	if ((err=s_curcaller->LocalToPhysAddr(s_curparams[param], &addr)) != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		return err;
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t *src;
dvander@601: 	pContext->LocalToPhysAddr(params[2], &src);
dvander@601: 
dvander@601: 	memcpy(src, addr, sizeof(cell_t) * params[3]);
dvander@601: 
dvander@601: 	return SP_ERROR_NONE;
dvander@601: }
dvander@601: 
dvander@601: static cell_t SetNativeArray(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t param = params[1];
dvander@601: 	if (param < 1 || param > s_curparams[0])
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
dvander@601: 	}
dvander@601: 
dvander@601: 	int err;
dvander@601: 	cell_t *addr;
dvander@601: 	if ((err=s_curcaller->LocalToPhysAddr(s_curparams[param], &addr)) != SP_ERROR_NONE)
dvander@601: 	{
dvander@601: 		return err;
dvander@601: 	}
dvander@601: 
dvander@601: 	cell_t *src;
dvander@601: 	pContext->LocalToPhysAddr(params[2], &src);
dvander@601: 
dvander@601: 	memcpy(addr, src, sizeof(cell_t) * params[3]);
dvander@601: 
dvander@601: 	return SP_ERROR_NONE;
dvander@601: }
dvander@601: 
dvander@601: static cell_t FormatNativeString(IPluginContext *pContext, const cell_t *params)
dvander@601: {
dvander@601: 	if (!s_curnative || (s_curnative->ctx != pContext))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeError("Not called from inside a native function");
dvander@601: 	}
dvander@601: 
dvander@601: 	int out_param = params[1];
dvander@601: 	int fmt_param = params[2];
dvander@601: 	int var_param = params[3];
dvander@601: 
dvander@601: 	/* Validate input */
dvander@601: 	if (out_param && (out_param < 1 || out_param > s_curparams[0]))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", out_param);
dvander@601: 	}
dvander@601: 	if (fmt_param && (fmt_param < 1 || fmt_param > s_curparams[0]))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", fmt_param);
dvander@601: 	}
dvander@601: 	if (var_param && (var_param < 1 || var_param > s_curparams[0] + 1))
dvander@601: 	{
dvander@601: 		return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", fmt_param);
dvander@601: 	}
dvander@601: 
dvander@601: 	/* Get buffer information */
dvander@601: 	int err;
dvander@601: 	char *output_buffer;
dvander@601: 	char *format_buffer;
dvander@601: 
dvander@601: 	if (out_param)
dvander@601: 	{
dvander@601: 		if ((err=s_curcaller->LocalToString(s_curparams[out_param], &output_buffer)) != SP_ERROR_NONE)
dvander@601: 		{
dvander@601: 			return err;
dvander@601: 		}
dvander@601: 	} else {
dvander@601: 		pContext->LocalToString(params[6], &output_buffer);
dvander@601: 	}
dvander@601: 
dvander@601: 	if (fmt_param)
dvander@601: 	{
dvander@601: 		if ((err=s_curcaller->LocalToString(s_curparams[fmt_param], &format_buffer)) != SP_ERROR_NONE)
dvander@601: 		{
dvander@601: 			return err;
dvander@601: 		}
dvander@601: 	} else {
dvander@601: 		pContext->LocalToString(params[7], &format_buffer);
dvander@601: 	}
dvander@601: 
dvander@601: 	/* Get maximum length */
dvander@601: 	size_t maxlen = (size_t)params[4];
dvander@601: 
dvander@601: 	/* Do the format */
dvander@601: 	size_t written = atcprintf(output_buffer, maxlen, format_buffer, s_curcaller, s_curparams, &var_param);
dvander@601: 
dvander@601: 	cell_t *addr;
dvander@601: 	pContext->LocalToPhysAddr(params[5], &addr);
dvander@601: 	*addr = (cell_t)written;
dvander@601: 
dvander@2193: 	return s_curcaller->GetLastNativeError();
dvander@601: }
dvander@601: 
dvander@601: //tee hee
dvander@601: REGISTER_NATIVES(nativeNatives)
dvander@601: {
dvander@601: 	{"CreateNative",			CreateNative},
dvander@601: 	{"GetNativeArray",			GetNativeArray},
dvander@601: 	{"GetNativeCell",			GetNativeCell},
dvander@601: 	{"GetNativeCellRef",		GetNativeCellRef},
dvander@601: 	{"GetNativeString",			GetNativeString},
dvander@601: 	{"GetNativeStringLength",	GetNativeStringLength},
dvander@601: 	{"FormatNativeString",		FormatNativeString},
dvander@601: 	{"ThrowNativeError",		ThrowNativeError},
dvander@601: 	{"SetNativeArray",			SetNativeArray},
dvander@601: 	{"SetNativeCellRef",		SetNativeCellRef},
dvander@601: 	{"SetNativeString",			SetNativeString},
dvander@601: 	{NULL,						NULL},
dvander@601: };