dvander@160: #include <stdio.h>
dvander@160: #include "sourcemod.h"
dvander@160: #include "sourcemm_api.h"
dvander@160: #include "systems/LibrarySys.h"
dvander@160: #include "vm/sp_vm_engine.h"
dvander@160: #include <sh_string.h>
dvander@176: #include "PluginSys.h"
dvander@180: #include "ForwardSys.h"
dvander@160: 
dvander@160: SourcePawnEngine g_SourcePawn;
dvander@160: SourceModBase g_SourceMod;
dvander@160: 
dvander@160: ILibrary *g_pJIT = NULL;
dvander@160: SourceHook::String g_BaseDir;
dvander@160: ISourcePawnEngine *g_pSourcePawn = &g_SourcePawn;
dvander@160: IVirtualMachine *g_pVM;
dvander@160: 
dvander@160: typedef int (*GIVEENGINEPOINTER)(ISourcePawnEngine *);
dvander@160: typedef unsigned int (*GETEXPORTCOUNT)();
dvander@160: typedef IVirtualMachine *(*GETEXPORT)(unsigned int);
dvander@160: typedef void (*NOTIFYSHUTDOWN)();
dvander@160: 
dvander@160: void ShutdownJIT()
dvander@160: {
dvander@160: 	NOTIFYSHUTDOWN notify = (NOTIFYSHUTDOWN)g_pJIT->GetSymbolAddress("NotifyShutdown");
dvander@160: 	if (notify)
dvander@160: 	{
dvander@160: 		notify();
dvander@160: 	}
dvander@160: 
dvander@160: 	g_pJIT->CloseLibrary();
dvander@160: }
dvander@160: 
dvander@160: bool SourceModBase::InitializeSourceMod(char *error, size_t err_max, bool late)
dvander@160: {
dvander@160: 	//:TODO: we need a localinfo system!
dvander@160: 	g_BaseDir.assign(g_SMAPI->GetBaseDir());
dvander@160: 
dvander@160: 	/* Attempt to load the JIT! */
dvander@160: 	char file[PLATFORM_MAX_PATH];
dvander@160: 	char myerror[255];
dvander@160: 	g_SMAPI->PathFormat(file, sizeof(file), "%s/addons/sourcemod/bin/sourcepawn.jit.x86.%s",
dvander@160: 		g_BaseDir.c_str(),
dvander@160: 		PLATFORM_LIB_EXT
dvander@160: 		);
dvander@160: 
dvander@160: 	g_pJIT = g_LibSys.OpenLibrary(file, myerror, sizeof(myerror));
dvander@160: 	if (!g_pJIT)
dvander@160: 	{
dvander@160: 		if (error && err_max)
dvander@160: 		{
dvander@160: 			snprintf(error, err_max, "%s (failed to load bin/sourcepawn.jit.x86.%s)", 
dvander@160: 				myerror,
dvander@160: 				PLATFORM_LIB_EXT);
dvander@160: 		}
dvander@160: 		return false;
dvander@160: 	}
dvander@160: 
dvander@160: 	int err;
dvander@160: 	GIVEENGINEPOINTER jit_init = (GIVEENGINEPOINTER)g_pJIT->GetSymbolAddress("GiveEnginePointer");
dvander@160: 	if (!jit_init)
dvander@160: 	{
dvander@160: 		ShutdownJIT();
dvander@160: 		if (error && err_max)
dvander@160: 		{
dvander@160: 			snprintf(error, err_max, "Failed to find GiveEnginePointer in JIT!");
dvander@160: 		}
dvander@160: 		return false;
dvander@160: 	}
dvander@160: 
dvander@160: 	if ((err=jit_init(g_pSourcePawn)) != 0)
dvander@160: 	{
dvander@160: 		ShutdownJIT();
dvander@160: 		if (error && err_max)
dvander@160: 		{
dvander@160: 			snprintf(error, err_max, "GiveEnginePointer returned %d in the JIT", err);
dvander@160: 		}
dvander@160: 		return false;
dvander@160: 	}
dvander@160: 
dvander@160: 	GETEXPORTCOUNT jit_getnum = (GETEXPORTCOUNT)g_pJIT->GetSymbolAddress("GetExportCount");
dvander@160: 	GETEXPORT jit_get = (GETEXPORT)g_pJIT->GetSymbolAddress("GetExport");
dvander@160: 	if (!jit_get)
dvander@160: 	{
dvander@160: 		ShutdownJIT();
dvander@160: 		if (error && err_max)
dvander@160: 		{
dvander@160: 			snprintf(error, err_max, "JIT is missing a necessary export!");
dvander@160: 		}
dvander@160: 		return false;
dvander@160: 	}
dvander@160: 
dvander@160: 	unsigned int num = jit_getnum();
dvander@160: 	if (!num || ((g_pVM=jit_get(0)) == NULL))
dvander@160: 	{
dvander@160: 		ShutdownJIT();
dvander@160: 		if (error && err_max)
dvander@160: 		{
dvander@160: 			snprintf(error, err_max, "JIT did not export any virtual machines!");
dvander@160: 		}
dvander@160: 		return false;
dvander@160: 	}
dvander@160: 
dvander@176: 	unsigned int api = g_pVM->GetAPIVersion();
dvander@176: 	if (api != SOURCEPAWN_VM_API_VERSION)
dvander@176: 	{
dvander@176: 		ShutdownJIT();
dvander@176: 		if (error && err_max)
dvander@176: 		{
dvander@176: 			snprintf(error, err_max, "JIT is not a compatible version");
dvander@176: 		}
dvander@176: 		return false;
dvander@176: 	}
dvander@176: 
dvander@176: 	g_SMAPI->PathFormat(file, sizeof(file), "%s/addons/sourcemod/plugins/test.smx", g_BaseDir.c_str());
dvander@176: 	IPlugin *pPlugin = g_PluginMngr.LoadPlugin(file, false, PluginType_Global, error, err_max);
dvander@177: 	IPluginFunction *func = pPlugin->GetFunctionByName("Test");
dvander@180: 	IPluginFunction *func2 = pPlugin->GetFunctionByName("Test2");
dvander@177: 	cell_t result = 2;
dvander@177: 	cell_t val = 6;
dvander@180: 	ParamType types[] = {Param_Cell, Param_CellByRef};
dvander@180: 	va_list ap = va_start(ap, late);
dvander@180: 	CForward *fwd = CForward::CreateForward(NULL, ET_Custom, 2, types, ap);
dvander@180: 	fwd->AddFunction(func);
dvander@180: 	fwd->AddFunction(func2);
dvander@180: 	fwd->PushCell(1);
dvander@180: 	fwd->PushCellByRef(&val, SMFUNC_COPYBACK_ALWAYS);
dvander@180: 	fwd->Execute(&result, NULL);
dvander@176: 	g_PluginMngr.UnloadPlugin(pPlugin);
dvander@176: 
dvander@160: 	return true;
dvander@160: }