1a156deca5
NOTE: the JIT is now embeddable out-of-box and usable by other projects which is pretty cool. I will commit a shell app demonstrating this soon --HG-- extra : convert_revision : svn%3A39bc706e-5318-0410-9160-8a85361fbb7c/trunk%402400
455 lines
12 KiB
C++
455 lines
12 KiB
C++
/**
|
|
* vim: set ts=4 :
|
|
* =============================================================================
|
|
* SourceMod
|
|
* 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 <sh_list.h>
|
|
#include <sh_string.h>
|
|
#include "sm_trie.h"
|
|
#include "sm_globals.h"
|
|
#include "PluginSys.h"
|
|
#include "sourcemod.h"
|
|
#include "sm_stringutil.h"
|
|
|
|
using namespace SourceHook;
|
|
|
|
static cell_t s_curparams[SP_MAX_EXEC_PARAMS+1];
|
|
static FakeNative *s_curnative = NULL;
|
|
static IPluginContext *s_curcaller = NULL;
|
|
|
|
cell_t FakeNativeRouter(IPluginContext *pContext, const cell_t *params, void *pData)
|
|
{
|
|
FakeNative *native = (FakeNative *)pData;
|
|
|
|
/* Check if too many parameters were passed */
|
|
if (params[0] > SP_MAX_EXEC_PARAMS)
|
|
{
|
|
return pContext->ThrowNativeError("Called native with too many parameters (%d>%d)", params[9], SP_MAX_EXEC_PARAMS);
|
|
}
|
|
|
|
/* Check if the native is paused */
|
|
if (native->ctx->GetRuntime()->IsPaused())
|
|
{
|
|
return pContext->ThrowNativeError("Plugin owning this native is currently paused.");
|
|
}
|
|
|
|
CPlugin *pCaller = g_PluginSys.GetPluginByCtx(pContext->GetContext());
|
|
|
|
/* Save any old data on the stack */
|
|
FakeNative *pSaveNative = s_curnative;
|
|
IPluginContext *pSaveCaller = s_curcaller;
|
|
cell_t save_params[SP_MAX_EXEC_PARAMS+1];
|
|
if (pSaveNative != NULL)
|
|
{
|
|
/* Copy all old parameters */
|
|
for (cell_t i=0; i<=s_curparams[0]; i++)
|
|
{
|
|
save_params[i] = s_curparams[i];
|
|
}
|
|
}
|
|
|
|
/* Save the current parameters */
|
|
s_curnative = native;
|
|
s_curcaller = pContext;
|
|
for (cell_t i=0; i<=params[0]; i++)
|
|
{
|
|
s_curparams[i] = params[i];
|
|
}
|
|
|
|
/* Push info and execute. */
|
|
cell_t result = 0;
|
|
native->call->PushCell(pCaller->GetMyHandle());
|
|
native->call->PushCell(params[0]);
|
|
int error;
|
|
if ((error=native->call->Execute(&result)) != SP_ERROR_NONE)
|
|
{
|
|
if (pContext->GetLastNativeError() == SP_ERROR_NONE)
|
|
{
|
|
pContext->ThrowNativeErrorEx(error, "Error encountered while processing a dynamic native");
|
|
}
|
|
}
|
|
|
|
/* Restore everything from the stack if necessary */
|
|
s_curnative = pSaveNative;
|
|
s_curcaller = pSaveCaller;
|
|
if (pSaveNative != NULL)
|
|
{
|
|
for (cell_t i=0; i<=save_params[0]; i++)
|
|
{
|
|
s_curparams[i] = save_params[i];
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
static cell_t CreateNative(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
char *name;
|
|
CPlugin *pPlugin;
|
|
|
|
pContext->LocalToString(params[1], &name);
|
|
|
|
IPluginFunction *pFunction = pContext->GetFunctionById(params[2]);
|
|
if (!pFunction)
|
|
{
|
|
return pContext->ThrowNativeError("Function %x is not a valid function", params[2]);
|
|
}
|
|
|
|
pPlugin = g_PluginSys.GetPluginByCtx(pContext->GetContext());
|
|
|
|
if (!pPlugin->AddFakeNative(pFunction, name, FakeNativeRouter))
|
|
{
|
|
return pContext->ThrowNativeError("Fatal error creating dynamic native!");
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t ThrowNativeError(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
g_SourceMod.SetGlobalTarget(SOURCEMOD_SERVER_LANGUAGE);
|
|
|
|
char buffer[512];
|
|
|
|
g_SourceMod.FormatString(buffer, sizeof(buffer), pContext, params, 2);
|
|
|
|
if (pContext->GetLastNativeError() != SP_ERROR_NONE)
|
|
{
|
|
s_curcaller->ThrowNativeError("Error encountered while processing a dynamic native");
|
|
} else {
|
|
s_curcaller->ThrowNativeErrorEx(params[1], "%s", buffer);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static cell_t GetNativeStringLength(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
int err;
|
|
char *str;
|
|
if ((err=s_curcaller->LocalToString(s_curparams[param], &str)) != SP_ERROR_NONE)
|
|
{
|
|
return err;
|
|
}
|
|
|
|
cell_t *addr;
|
|
pContext->LocalToPhysAddr(params[2], &addr);
|
|
*addr = (cell_t)strlen(str);
|
|
|
|
return SP_ERROR_NONE;
|
|
}
|
|
|
|
static cell_t GetNativeString(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
int err;
|
|
char *str;
|
|
if ((err=s_curcaller->LocalToString(s_curparams[param], &str)) != SP_ERROR_NONE)
|
|
{
|
|
return err;
|
|
}
|
|
|
|
size_t bytes = 0;
|
|
pContext->StringToLocalUTF8(params[2], params[3], str, &bytes);
|
|
|
|
cell_t *addr;
|
|
pContext->LocalToPhysAddr(params[4], &addr);
|
|
*addr = (cell_t)bytes;
|
|
|
|
return SP_ERROR_NONE;
|
|
}
|
|
|
|
static cell_t SetNativeString(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
char *str;
|
|
pContext->LocalToString(params[2], &str);
|
|
|
|
int err;
|
|
size_t bytes = 0;
|
|
if (params[4])
|
|
{
|
|
err = s_curcaller->StringToLocalUTF8(s_curparams[param], params[3], str, &bytes);
|
|
} else {
|
|
err = s_curcaller->StringToLocal(s_curparams[param], params[3], str);
|
|
/* Eww */
|
|
bytes = strlen(str);
|
|
if (bytes >= (size_t)params[3])
|
|
{
|
|
bytes = params[3] - 1;
|
|
}
|
|
}
|
|
|
|
if (err != SP_ERROR_NONE)
|
|
{
|
|
return err;
|
|
}
|
|
|
|
cell_t *addr;
|
|
pContext->LocalToPhysAddr(params[5], &addr);
|
|
*addr = (cell_t)bytes;
|
|
|
|
return SP_ERROR_NONE;
|
|
}
|
|
|
|
static cell_t GetNativeCell(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
return s_curparams[param];
|
|
}
|
|
|
|
static cell_t GetNativeCellRef(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
cell_t *addr;
|
|
if (s_curcaller->LocalToPhysAddr(s_curparams[param], &addr) != SP_ERROR_NONE)
|
|
{
|
|
return s_curcaller->ThrowNativeErrorEx(SP_ERROR_INVALID_ADDRESS, "Invalid address value");
|
|
}
|
|
|
|
return *addr;
|
|
}
|
|
|
|
static cell_t SetNativeCellRef(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
cell_t *addr;
|
|
if (s_curcaller->LocalToPhysAddr(s_curparams[param], &addr) != SP_ERROR_NONE)
|
|
{
|
|
return s_curcaller->ThrowNativeErrorEx(SP_ERROR_INVALID_ADDRESS, "Invalid address value");
|
|
}
|
|
|
|
*addr = params[2];
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t GetNativeArray(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
int err;
|
|
cell_t *addr;
|
|
if ((err=s_curcaller->LocalToPhysAddr(s_curparams[param], &addr)) != SP_ERROR_NONE)
|
|
{
|
|
return err;
|
|
}
|
|
|
|
cell_t *src;
|
|
pContext->LocalToPhysAddr(params[2], &src);
|
|
|
|
memcpy(src, addr, sizeof(cell_t) * params[3]);
|
|
|
|
return SP_ERROR_NONE;
|
|
}
|
|
|
|
static cell_t SetNativeArray(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
cell_t param = params[1];
|
|
if (param < 1 || param > s_curparams[0])
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", param);
|
|
}
|
|
|
|
int err;
|
|
cell_t *addr;
|
|
if ((err=s_curcaller->LocalToPhysAddr(s_curparams[param], &addr)) != SP_ERROR_NONE)
|
|
{
|
|
return err;
|
|
}
|
|
|
|
cell_t *src;
|
|
pContext->LocalToPhysAddr(params[2], &src);
|
|
|
|
memcpy(addr, src, sizeof(cell_t) * params[3]);
|
|
|
|
return SP_ERROR_NONE;
|
|
}
|
|
|
|
static cell_t FormatNativeString(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!s_curnative || (s_curnative->ctx != pContext))
|
|
{
|
|
return pContext->ThrowNativeError("Not called from inside a native function");
|
|
}
|
|
|
|
int out_param = params[1];
|
|
int fmt_param = params[2];
|
|
int var_param = params[3];
|
|
|
|
/* Validate input */
|
|
if (out_param && (out_param < 1 || out_param > s_curparams[0]))
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", out_param);
|
|
}
|
|
if (fmt_param && (fmt_param < 1 || fmt_param > s_curparams[0]))
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", fmt_param);
|
|
}
|
|
if (var_param && (var_param < 1 || var_param > s_curparams[0] + 1))
|
|
{
|
|
return pContext->ThrowNativeErrorEx(SP_ERROR_PARAM, "Invalid parameter number: %d", fmt_param);
|
|
}
|
|
|
|
/* Get buffer information */
|
|
int err;
|
|
char *output_buffer;
|
|
char *format_buffer;
|
|
|
|
if (out_param)
|
|
{
|
|
if ((err=s_curcaller->LocalToString(s_curparams[out_param], &output_buffer)) != SP_ERROR_NONE)
|
|
{
|
|
return err;
|
|
}
|
|
} else {
|
|
pContext->LocalToString(params[6], &output_buffer);
|
|
}
|
|
|
|
if (fmt_param)
|
|
{
|
|
if ((err=s_curcaller->LocalToString(s_curparams[fmt_param], &format_buffer)) != SP_ERROR_NONE)
|
|
{
|
|
return err;
|
|
}
|
|
} else {
|
|
pContext->LocalToString(params[7], &format_buffer);
|
|
}
|
|
|
|
/* Get maximum length */
|
|
size_t maxlen = (size_t)params[4];
|
|
|
|
/* Do the format */
|
|
size_t written = atcprintf(output_buffer, maxlen, format_buffer, s_curcaller, s_curparams, &var_param);
|
|
|
|
cell_t *addr;
|
|
pContext->LocalToPhysAddr(params[5], &addr);
|
|
*addr = (cell_t)written;
|
|
|
|
return s_curcaller->GetLastNativeError();
|
|
}
|
|
|
|
//tee hee
|
|
REGISTER_NATIVES(nativeNatives)
|
|
{
|
|
{"CreateNative", CreateNative},
|
|
{"GetNativeArray", GetNativeArray},
|
|
{"GetNativeCell", GetNativeCell},
|
|
{"GetNativeCellRef", GetNativeCellRef},
|
|
{"GetNativeString", GetNativeString},
|
|
{"GetNativeStringLength", GetNativeStringLength},
|
|
{"FormatNativeString", FormatNativeString},
|
|
{"ThrowNativeError", ThrowNativeError},
|
|
{"SetNativeArray", SetNativeArray},
|
|
{"SetNativeCellRef", SetNativeCellRef},
|
|
{"SetNativeString", SetNativeString},
|
|
{NULL, NULL},
|
|
};
|