dvander@950: #include "extension.h"
dvander@950: #include "vcallbuilder.h"
dvander@950: 
dvander@950: enum SDKLibrary
dvander@950: {
dvander@950: 	SDKLibrary_Server,	/**< server.dll/server_i486.so */
dvander@950: 	SDKLibrary_Engine,	/**< engine.dll/engine_*.so */
dvander@950: };
dvander@950: 
dvander@950: enum SDKPassMethod
dvander@950: {
dvander@950: 	SDKPass_Pointer,		/**< Pass as a pointer */
dvander@950: 	SDKPass_Plain,			/**< Pass as plain data */
dvander@950: 	SDKPass_ByValue,		/**< Pass an object by value */
dvander@950: 	SDKPass_ByRef,			/**< Pass an object by reference */
dvander@950: };
dvander@950: 
dvander@950: int s_vtbl_index = 0;
dvander@950: void *s_call_addr = NULL;
dvander@950: ValveCallType s_vcalltype = ValveCall_Static;
dvander@950: bool s_has_return = false;
dvander@950: ValvePassInfo s_return;
dvander@950: unsigned int s_numparams = 0;
dvander@950: ValvePassInfo s_params[SP_MAX_EXEC_PARAMS];
dvander@950: 
dvander@950: inline void DecodePassMethod(ValveType vtype, SDKPassMethod method, PassType &type, unsigned int &flags)
dvander@950: {
dvander@950: 	if (method == SDKPass_Pointer)
dvander@950: 	{
dvander@950: 		type = PassType_Basic;
dvander@950: 		if (vtype == Valve_POD
dvander@950: 			|| vtype == Valve_Float)
dvander@950: 		{
dvander@950: 			flags = PASSFLAG_BYREF;
dvander@950: 		} else {
dvander@950: 			flags = PASSFLAG_BYVAL;
dvander@950: 		}
dvander@950: 	} else if (method == SDKPass_Plain) {
dvander@950: 		type = PassType_Basic;
dvander@950: 		flags = PASSFLAG_BYVAL;
dvander@950: 	} else if (method == SDKPass_ByValue) {
dvander@950: 		if (vtype == Valve_Vector
dvander@950: 			|| vtype == Valve_QAngle)
dvander@950: 		{
dvander@950: 			type = PassType_Object;
dvander@950: 		} else {
dvander@950: 			type = PassType_Basic;
dvander@950: 		}
dvander@950: 		flags = PASSFLAG_BYVAL;
dvander@950: 	} else if (method == SDKPass_ByRef) {
dvander@950: 		if (vtype == Valve_Vector
dvander@950: 			|| vtype == Valve_QAngle)
dvander@950: 		{
dvander@950: 			type = PassType_Object;
dvander@950: 		} else {
dvander@950: 			type = PassType_Basic;
dvander@950: 		}
dvander@950: 		flags = PASSFLAG_BYREF;
dvander@950: 	}
dvander@950: }
dvander@950: 
dvander@950: static cell_t StartPrepSDKCall(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	s_numparams = 0;
dvander@950: 	s_vtbl_index = 0;
dvander@950: 	s_call_addr = NULL;
dvander@950: 	s_has_return = false;
dvander@950: 	s_vcalltype = (ValveCallType)params[1];
dvander@950: 
dvander@950: 	return 1;
dvander@950: }
dvander@950: 
dvander@950: static cell_t PrepSDKCall_SetVirtual(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	s_vtbl_index = params[1];
dvander@950: 
dvander@950: 	return 1;
dvander@950: }
dvander@950: 
dvander@950: static cell_t PrepSDKCall_SetSignature(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	void *addrInBase = NULL;
dvander@950: 	if (params[1] == SDKLibrary_Server)
dvander@950: 	{
dvander@950: 		addrInBase = g_SMAPI->serverFactory(false);
dvander@950: 	} else if (params[1] == SDKLibrary_Engine) {
dvander@950: 		addrInBase = g_SMAPI->engineFactory(false);
dvander@950: 	}
dvander@950: 	if (addrInBase == NULL)
dvander@950: 	{
dvander@950: 		return 0;
dvander@950: 	}
dvander@950: 
dvander@950: 	char *sig;
dvander@950: 	pContext->LocalToString(params[2], &sig);
dvander@950: 
dvander@950: #if defined PLATFORM_LINUX
dvander@950: 	if (sig[0] == '@')
dvander@950: 	{
dvander@950: 		Dl_info info;
dvander@950: 		if (dladdr(addrInBase, &info) == 0)
dvander@950: 		{
dvander@950: 			return 0;
dvander@950: 		}
dvander@950: 		void *handle = dlopen(info.dli_fname, RTLD_NOW);
dvander@950: 		if (!handle)
dvander@950: 		{
dvander@950: 			return 0;
dvander@950: 		}
dvander@950: 		s_call_addr = dlsym(handle, &sig[1]);
dvander@950: 		dlclose(handle);
dvander@950: 
dvander@950: 		return (s_call_addr != NULL) ? 1 : 0;
dvander@950: 	}
dvander@950: #endif
dvander@950: 
dvander@950: 	s_call_addr = memutils->FindPattern(addrInBase, sig, params[3]);
dvander@950: 
dvander@950: 	return (s_call_addr != NULL) ? 1 : 0;
dvander@950: }
dvander@950: 
dvander@950: static cell_t PrepSDKCall_SetFromConf(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	IGameConfig *conf;
dvander@950: 
dvander@950: 	if (params[1] == BAD_HANDLE)
dvander@950: 	{
dvander@950: 		conf = g_pGameConf;
dvander@950: 	} else {
dvander@950: 		HandleError err;
dvander@950: 		if ((conf = gameconfs->ReadHandle(params[1], pContext->GetIdentity(), &err)) == NULL)
dvander@950: 		{
dvander@950: 			return pContext->ThrowNativeError("Invalid Handle %x (error %d)", params[1], err);
dvander@950: 		}
dvander@950: 	}
dvander@950: 	
dvander@950: 	char *key;
dvander@950: 	pContext->LocalToString(params[3], &key);
dvander@950: 
dvander@950: 	if (params[2] == 0)
dvander@950: 	{
dvander@950: 		return conf->GetOffset(key, &s_vtbl_index) ? 1 : 0;
dvander@950: 	} else if (params[2] == 1) {
dvander@950: 		bool result = conf->GetMemSig(key, &s_call_addr) ? 1 : 0;
dvander@950: 		return (result && s_call_addr != NULL) ? 1 : 0;
dvander@950: 	}
dvander@950: 
dvander@950: 	return 0;
dvander@950: }
dvander@950: 
dvander@950: static cell_t PrepSDKCall_SetReturnInfo(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	s_has_return = true;
dvander@950: 	s_return.vtype = (ValveType)params[1];
dvander@950: 	DecodePassMethod(s_return.vtype, (SDKPassMethod)params[2], s_return.type, s_return.flags);
dvander@950: 	s_return.decflags = params[3];
dvander@950: 	s_return.encflags = params[4];
dvander@950: 
dvander@950: 	return 1;
dvander@950: }
dvander@950: 
dvander@950: static cell_t PrepSDKCall_AddParameter(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	if (s_numparams >= SP_MAX_EXEC_PARAMS)
dvander@950: 	{
dvander@950: 		return pContext->ThrowNativeError("Parameter limit for SDK calls reached");
dvander@950: 	}
dvander@950: 
dvander@950: 	ValvePassInfo *info = &s_params[s_numparams++];
dvander@950: 	info->vtype = (ValveType)params[1];
dvander@950: 	DecodePassMethod(info->vtype, (SDKPassMethod)params[2], info->type, info->flags);
dvander@950: 	info->decflags = params[3];
dvander@950: 	info->encflags = params[4];
dvander@950: 
dvander@950: 	return 1;
dvander@950: }
dvander@950: 
dvander@950: static cell_t EndPrepSDKCall(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	ValveCall *vc = NULL;
dvander@950: 	if (s_vtbl_index)
dvander@950: 	{
dvander@950: 		vc = CreateValveVCall(s_vtbl_index, s_vcalltype, s_has_return ? &s_return : NULL, s_params, s_numparams);
dvander@953: 	} else if (s_call_addr) {
dvander@953: 		vc = CreateValveCall(s_call_addr, s_vcalltype, s_has_return ? &s_return : NULL, s_params, s_numparams);
dvander@950: 	}
dvander@950: 
dvander@950: 	if (!vc)
dvander@950: 	{
dvander@950: 		return BAD_HANDLE;
dvander@950: 	}
dvander@950: 
dvander@950: 	Handle_t hndl = handlesys->CreateHandle(g_CallHandle, vc, pContext->GetIdentity(), myself->GetIdentity(), NULL);
dvander@950: 	if (!hndl)
dvander@950: 	{
dvander@950: 		delete vc;
dvander@950: 	}
dvander@950: 
dvander@950: 	return hndl;
dvander@950: }
dvander@950: 
dvander@950: static cell_t SDKCall(IPluginContext *pContext, const cell_t *params)
dvander@950: {
dvander@950: 	ValveCall *vc;
dvander@950: 	HandleError err;
dvander@950: 	HandleSecurity sec(pContext->GetIdentity(), myself->GetIdentity());
dvander@950: 
dvander@950: 	if ((err = handlesys->ReadHandle(params[1], g_CallHandle, &sec, (void **)&vc)) != HandleError_None)
dvander@950: 	{
dvander@950: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", params[1], err);
dvander@950: 	}
dvander@950: 
dvander@950: 	unsigned char *ptr = vc->stk_get();
dvander@950: 
dvander@950: 	unsigned int numparams = (unsigned)params[0];
dvander@950: 	unsigned int startparam = 2;
dvander@950: 	/* Do we need to write a thispointer?  */
dvander@950: 
dvander@950: 	if (vc->thisinfo && 
dvander@950: 		(vc->thisinfo->vtype == Valve_CBaseEntity
dvander@950: 		 || vc->thisinfo->vtype == Valve_CBasePlayer))
dvander@950: 	{
dvander@950: 		if (startparam > numparams)
dvander@950: 		{
dvander@950: 			vc->stk_put(ptr);
dvander@950: 			return pContext->ThrowNativeError("Expected 1 parameter for entity pointer; found none");
dvander@950: 		}
dvander@950: 		if (DecodeValveParam(pContext, 
dvander@950: 			params[startparam], 
dvander@950: 			vc->thisinfo->vtype, 
dvander@950: 			vc->thisinfo->decflags | VDECODE_FLAG_BYREF,
dvander@950: 			vc->thisinfo->type,
dvander@950: 			ptr) == Data_Fail)
dvander@950: 		{
dvander@950: 			vc->stk_put(ptr);
dvander@950: 			return 0;
dvander@950: 		}
dvander@950: 		startparam++;
dvander@950: 	}
dvander@950: 
dvander@950: 	/* See if we need to skip any more parameters */
dvander@950: 	unsigned int retparam = startparam;
dvander@950: 	if (vc->retinfo)
dvander@950: 	{
dvander@950: 		if (vc->retinfo->vtype == Valve_String)
dvander@950: 		{
dvander@950: 			startparam += 2;
dvander@950: 		} else if (vc->retinfo->vtype == Valve_Vector
dvander@950: 					|| vc->retinfo->vtype == Valve_QAngle)
dvander@950: 		{
dvander@950: 			startparam += 1;
dvander@950: 		}
dvander@950: 	}
dvander@950: 
dvander@950: 	unsigned int callparams = vc->call->GetParamCount();
dvander@950: 	bool will_copyback = false;
dvander@950: 	for (unsigned int i=0; i<callparams; i++)
dvander@950: 	{
dvander@950: 		unsigned int p = startparam + i;
dvander@950: 		if (p > numparams)
dvander@950: 		{
dvander@950: 			vc->stk_put(ptr);
dvander@950: 			return pContext->ThrowNativeError("Expected %dth parameter, found none", p);
dvander@950: 		}
dvander@950: 		if (DecodeValveParam(pContext,
dvander@950: 			params[p],
dvander@950: 			vc->vparams[i].vtype,
dvander@950: 			vc->vparams[i].decflags | VDECODE_FLAG_BYREF,
dvander@950: 			vc->vparams[i].type,
dvander@950: 			ptr + vc->vparams[i].offset) == Data_Fail)
dvander@950: 		{
dvander@950: 			vc->stk_put(ptr);
dvander@950: 			return 0;
dvander@950: 		}
dvander@950: 		if (vc->vparams[i].encflags & VENCODE_FLAG_COPYBACK)
dvander@950: 		{
dvander@950: 			will_copyback = true;
dvander@950: 		}
dvander@950: 	}
dvander@950: 
dvander@950: 	/* Make the actual call */
dvander@950: 	vc->call->Execute(ptr, vc->retbuf);
dvander@950: 
dvander@950: 	/* Do we need to copy anything back? */
dvander@950: 	if (will_copyback)
dvander@950: 	{
dvander@950: 		for (unsigned int i=0; i<callparams; i++)
dvander@950: 		{
dvander@950: 			if (vc->vparams[i].encflags & VENCODE_FLAG_COPYBACK)
dvander@950: 			{
dvander@950: 				if (EncodeValveParam(pContext, 
dvander@950: 					startparam + i, 
dvander@950: 					vc->vparams[i].vtype, 
dvander@950: 					vc->vparams[i].type, 
dvander@950: 					ptr + vc->vparams[i].offset) == Data_Fail)
dvander@950: 				{
dvander@950: 					vc->stk_put(ptr);
dvander@950: 					return 0;
dvander@950: 				}
dvander@950: 			}
dvander@950: 		}
dvander@950: 	}
dvander@950: 
dvander@950: 	/* Save stack once and for all */
dvander@950: 	vc->stk_put(ptr);
dvander@950: 
dvander@950: 	/* Figure out how to decode the return information */
dvander@950: 	if (vc->retinfo)
dvander@950: 	{
dvander@950: 		if (vc->retinfo->vtype == Valve_String)
dvander@950: 		{
dvander@950: 			if (numparams < 3)
dvander@950: 			{
dvander@950: 				return pContext->ThrowNativeError("Expected arguments (2,3) for string storage");
dvander@950: 			}
dvander@950: 			cell_t *addr;
dvander@950: 			size_t written;
dvander@950: 			pContext->LocalToPhysAddr(params[retparam+1], &addr);
dvander@950: 			pContext->StringToLocalUTF8(params[retparam], *addr, *(char **)vc->retbuf, &written);
dvander@950: 			return (cell_t)written;
dvander@950: 		} else if (vc->retinfo->vtype == Valve_Vector
dvander@950: 					|| vc->retinfo->vtype == Valve_QAngle)
dvander@950: 		{
dvander@950: 			if (numparams < 2)
dvander@950: 			{
dvander@950: 				return pContext->ThrowNativeError("Expected argument (2) for Float[3] storage");
dvander@950: 			}
dvander@950: 			if (EncodeValveParam(pContext, params[retparam], vc->retinfo->vtype, vc->retinfo->type, vc->retbuf)
dvander@950: 				== Data_Fail)
dvander@950: 			{
dvander@950: 				return 0;
dvander@950: 			}
dvander@950: 		} else if (vc->retinfo->vtype == Valve_CBaseEntity
dvander@950: 					|| vc->retinfo->vtype == Valve_CBasePlayer)
dvander@950: 		{
dvander@950: 			CBaseEntity *pEntity = *(CBaseEntity **)(vc->retbuf);
dvander@950: 			if (!pEntity)
dvander@950: 			{
dvander@950: 				return -1;
dvander@950: 			}
dvander@950: 			edict_t *pEdict = gameents->BaseEntityToEdict(pEntity);
dvander@950: 			if (!pEdict || pEdict->IsFree())
dvander@950: 			{
dvander@950: 				return -1;
dvander@950: 			}
dvander@950: 			return engine->IndexOfEdict(pEdict);
dvander@950: 		} else if (vc->retinfo->vtype == Valve_Edict) {
dvander@950: 			edict_t *pEdict = *(edict_t **)(vc->retbuf);
dvander@950: 			if (!pEdict || pEdict->IsFree())
dvander@950: 			{
dvander@950: 				return -1;
dvander@950: 			}
dvander@950: 			return engine->IndexOfEdict(pEdict);
dvander@950: 		} else {
dvander@950: 			cell_t *addr = (cell_t *)vc->retbuf;
dvander@950: 			if (vc->retinfo->flags & PASSFLAG_BYREF)
dvander@950: 			{
dvander@950: 				addr = *(cell_t **)addr;
dvander@950: 			}
dvander@950: 			return *addr;
dvander@950: 		}
dvander@950: 	}
dvander@950: 
dvander@950: 	return 0;
dvander@950: }
dvander@950: 
dvander@950: sp_nativeinfo_t g_CallNatives[] = 
dvander@950: {
dvander@950: 	{"StartPrepSDKCall",			StartPrepSDKCall},
dvander@950: 	{"PrepSDKCall_SetVirtual",		PrepSDKCall_SetVirtual},
dvander@950: 	{"PrepSDKCall_SetSignature",	PrepSDKCall_SetSignature},
dvander@950: 	{"PrepSDKCall_SetFromConf",		PrepSDKCall_SetFromConf},
dvander@950: 	{"PrepSDKCall_SetReturnInfo",	PrepSDKCall_SetReturnInfo},
dvander@950: 	{"PrepSDKCall_AddParameter",	PrepSDKCall_AddParameter},
dvander@950: 	{"EndPrepSDKCall",				EndPrepSDKCall},
dvander@950: 	{"SDKCall",						SDKCall},
dvander@950: 	{NULL,							NULL},
dvander@950: };