a1009aed38
All plugin and include file headers also have been changed to say about GPL3 instead of GPL2. (This day shall henceforth be known as the Eighty Column Massacre of '07) --HG-- extra : convert_revision : svn%3A39bc706e-5318-0410-9160-8a85361fbb7c/trunk%401336
370 lines
8.7 KiB
C++
370 lines
8.7 KiB
C++
/**
|
|
* vim: set ts=4 :
|
|
* =============================================================================
|
|
* SourceMod SDKTools Extension
|
|
* Copyright (C) 2004-2007 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 "vcallbuilder.h"
|
|
#include "extension.h"
|
|
|
|
ValveCall::ValveCall()
|
|
{
|
|
call = NULL;
|
|
vparams = NULL;
|
|
retinfo = NULL;
|
|
thisinfo = NULL;
|
|
retbuf = NULL;
|
|
}
|
|
|
|
ValveCall::~ValveCall()
|
|
{
|
|
while (!stk.empty())
|
|
{
|
|
unsigned char *ptr = stk.front();
|
|
delete [] ptr;
|
|
stk.pop();
|
|
}
|
|
if (call)
|
|
{
|
|
call->Destroy();
|
|
}
|
|
delete [] retbuf;
|
|
delete [] vparams;
|
|
}
|
|
|
|
unsigned char *ValveCall::stk_get()
|
|
{
|
|
unsigned char *ptr;
|
|
if (stk.empty())
|
|
{
|
|
ptr = new unsigned char[stackSize];
|
|
} else {
|
|
ptr = stk.front();
|
|
stk.pop();
|
|
}
|
|
return ptr;
|
|
}
|
|
|
|
void ValveCall::stk_put(unsigned char *ptr)
|
|
{
|
|
stk.push(ptr);
|
|
}
|
|
|
|
ValveCall *CreateValveCall(void *addr,
|
|
ValveCallType vcalltype,
|
|
const ValvePassInfo *retInfo,
|
|
const ValvePassInfo *params,
|
|
unsigned int numParams)
|
|
{
|
|
if (numParams > 32)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
ValveCall *vc = new ValveCall;
|
|
|
|
vc->type = vcalltype;
|
|
|
|
size_t size = 0;
|
|
vc->stackSize = 0;
|
|
|
|
/* Get return information - encode only */
|
|
PassInfo retBuf;
|
|
size_t retBufSize = 0;
|
|
bool retbuf_needs_extra;
|
|
if (retInfo)
|
|
{
|
|
if ((size = ValveParamToBinParam(retInfo->vtype, retInfo->type, retInfo->flags, &retBuf, retbuf_needs_extra)) == 0)
|
|
{
|
|
delete vc;
|
|
return NULL;
|
|
}
|
|
retBufSize = retBuf.size;
|
|
}
|
|
|
|
/* Get parameter info */
|
|
PassInfo paramBuf[32];
|
|
size_t sizes[32];
|
|
size_t normSize = 0;
|
|
size_t extraSize = 0;
|
|
for (unsigned int i=0; i<numParams; i++)
|
|
{
|
|
bool needs_extra;
|
|
if ((size = ValveParamToBinParam(params[i].vtype,
|
|
params[i].type,
|
|
params[i].flags,
|
|
¶mBuf[i],
|
|
needs_extra)) == 0)
|
|
{
|
|
delete vc;
|
|
return NULL;
|
|
}
|
|
if (needs_extra)
|
|
{
|
|
sizes[i] = size;
|
|
} else {
|
|
sizes[i] = 0;
|
|
}
|
|
normSize += paramBuf[i].size;
|
|
extraSize += sizes[i];
|
|
}
|
|
|
|
|
|
/* Get thisinfo if needed */
|
|
ValvePassInfo thisbuf;
|
|
ValvePassInfo *thisinfo = NULL;
|
|
CallConvention cv = CallConv_Cdecl;
|
|
if (vcalltype != ValveCall_Static)
|
|
{
|
|
thisinfo = &thisbuf;
|
|
thisinfo->type = PassType_Basic;
|
|
switch (vcalltype)
|
|
{
|
|
case ValveCall_Entity:
|
|
thisinfo->vtype = Valve_CBaseEntity;
|
|
thisinfo->flags = PASSFLAG_BYVAL;
|
|
thisinfo->decflags |= VDECODE_FLAG_ALLOWWORLD;
|
|
break;
|
|
case ValveCall_Player:
|
|
thisinfo->vtype = Valve_CBasePlayer;
|
|
thisinfo->flags = PASSFLAG_BYVAL;
|
|
thisinfo->decflags = 0;
|
|
break;
|
|
default:
|
|
thisinfo->vtype = Valve_POD;
|
|
thisinfo->flags = PASSFLAG_ASPOINTER;
|
|
thisinfo->decflags = 0;
|
|
break;
|
|
}
|
|
thisinfo->encflags = 0;
|
|
thisinfo->offset = 0;
|
|
normSize += sizeof(void *);
|
|
cv = CallConv_ThisCall;
|
|
}
|
|
|
|
/* Now we can try creating the call */
|
|
if ((vc->call = g_pBinTools->CreateCall(addr,
|
|
cv,
|
|
(retInfo ? &retBuf : NULL),
|
|
paramBuf,
|
|
numParams))
|
|
== NULL)
|
|
{
|
|
if (!vc->call)
|
|
{
|
|
delete vc;
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
/* Allocate extra space for thisptr AND ret buffer, even if we don't use it */
|
|
vc->vparams = new ValvePassInfo[numParams + 2];
|
|
|
|
/* We've got the call and everything is encoded.
|
|
* It's time to save the valve specific information and helper variables.
|
|
*/
|
|
if (retInfo)
|
|
{
|
|
/* Allocate and copy */
|
|
vc->retinfo = &(vc->vparams[numParams]);
|
|
*vc->retinfo = *retInfo;
|
|
vc->retinfo->offset = 0;
|
|
vc->retinfo->obj_offset = retbuf_needs_extra ? sizeof(void *) : 0;
|
|
/* Allocate stack space */
|
|
vc->retbuf = new unsigned char[retBufSize];
|
|
} else {
|
|
vc->retinfo = NULL;
|
|
vc->retbuf = NULL;
|
|
}
|
|
|
|
if (thisinfo)
|
|
{
|
|
/* Allocate and copy */
|
|
vc->thisinfo = &(vc->vparams[numParams + 1]);
|
|
*vc->thisinfo = *thisinfo;
|
|
vc->thisinfo->offset = 0;
|
|
vc->thisinfo->obj_offset = 0;
|
|
} else {
|
|
vc->thisinfo = NULL;
|
|
}
|
|
|
|
/* Now, save info about each parameter. */
|
|
size_t last_extra_offset = 0;
|
|
for (unsigned int i=0; i<numParams; i++)
|
|
{
|
|
/* Copy */
|
|
vc->vparams[i] = params[i];
|
|
vc->vparams[i].offset = vc->call->GetParamInfo(i)->offset;
|
|
vc->vparams[i].obj_offset = last_extra_offset;
|
|
last_extra_offset += sizes[i];
|
|
}
|
|
|
|
vc->stackSize = normSize + extraSize;
|
|
vc->stackEnd = normSize;
|
|
|
|
return vc;
|
|
}
|
|
|
|
ValveCall *CreateValveVCall(unsigned int vtableIdx,
|
|
ValveCallType vcalltype,
|
|
const ValvePassInfo *retInfo,
|
|
const ValvePassInfo *params,
|
|
unsigned int numParams)
|
|
{
|
|
if (numParams > 32)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
ValveCall *vc = new ValveCall;
|
|
|
|
vc->type = vcalltype;
|
|
|
|
size_t size = 0;
|
|
vc->stackSize = 0;
|
|
|
|
/* Get return information - encode only */
|
|
PassInfo retBuf;
|
|
size_t retBufSize = 0;
|
|
bool retbuf_needs_extra;
|
|
if (retInfo)
|
|
{
|
|
if ((size = ValveParamToBinParam(retInfo->vtype, retInfo->type, retInfo->flags, &retBuf, retbuf_needs_extra)) == 0)
|
|
{
|
|
delete vc;
|
|
return NULL;
|
|
}
|
|
retBufSize = retBuf.size;
|
|
}
|
|
|
|
/* Get parameter info */
|
|
PassInfo paramBuf[32];
|
|
size_t sizes[32];
|
|
size_t normSize = 0;
|
|
size_t extraSize = 0;
|
|
for (unsigned int i=0; i<numParams; i++)
|
|
{
|
|
bool needs_extra;
|
|
if ((size = ValveParamToBinParam(params[i].vtype,
|
|
params[i].type,
|
|
params[i].flags,
|
|
¶mBuf[i],
|
|
needs_extra)) == 0)
|
|
{
|
|
delete vc;
|
|
return NULL;
|
|
}
|
|
if (needs_extra)
|
|
{
|
|
sizes[i] = size;
|
|
} else {
|
|
sizes[i] = 0;
|
|
}
|
|
normSize += paramBuf[i].size;
|
|
extraSize += sizes[i];
|
|
}
|
|
|
|
/* Now we can try creating the call */
|
|
if ((vc->call = g_pBinTools->CreateVCall(vtableIdx,
|
|
0,
|
|
0,
|
|
(retInfo ? &retBuf : NULL),
|
|
paramBuf,
|
|
numParams))
|
|
== NULL)
|
|
{
|
|
if (!vc->call)
|
|
{
|
|
delete vc;
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
/* Allocate extra space for thisptr AND ret buffer, even if we don't use it */
|
|
vc->vparams = new ValvePassInfo[numParams + 2];
|
|
|
|
/* We've got the call and everything is encoded.
|
|
* It's time to save the valve specific information and helper variables.
|
|
*/
|
|
if (retInfo)
|
|
{
|
|
/* Allocate and copy */
|
|
vc->retinfo = &(vc->vparams[numParams]);
|
|
*vc->retinfo = *retInfo;
|
|
vc->retinfo->offset = 0;
|
|
vc->retinfo->obj_offset = retbuf_needs_extra ? sizeof(void *) : 0;
|
|
/* Allocate stack space */
|
|
vc->retbuf = new unsigned char[retBufSize];
|
|
} else {
|
|
vc->retinfo = NULL;
|
|
vc->retbuf = NULL;
|
|
}
|
|
|
|
/* Save the this info for the dynamic decoder */
|
|
vc->thisinfo = &(vc->vparams[numParams + 1]);
|
|
vc->thisinfo->type = PassType_Basic;
|
|
switch (vcalltype)
|
|
{
|
|
case ValveCall_Entity:
|
|
vc->thisinfo->vtype = Valve_CBaseEntity;
|
|
vc->thisinfo->flags = PASSFLAG_BYVAL;
|
|
vc->thisinfo->decflags = VDECODE_FLAG_ALLOWWORLD;
|
|
break;
|
|
case ValveCall_Player:
|
|
vc->thisinfo->vtype = Valve_CBasePlayer;
|
|
vc->thisinfo->flags = PASSFLAG_BYVAL;
|
|
vc->thisinfo->decflags = 0;
|
|
break;
|
|
default:
|
|
vc->thisinfo->vtype = Valve_POD;
|
|
vc->thisinfo->flags = PASSFLAG_ASPOINTER;
|
|
vc->thisinfo->decflags = 0;
|
|
break;
|
|
}
|
|
vc->thisinfo->encflags = 0;
|
|
vc->thisinfo->offset = 0;
|
|
vc->thisinfo->obj_offset = 0;
|
|
normSize += sizeof(void *);
|
|
|
|
/* Now, save info about each parameter. */
|
|
size_t last_extra_offset = 0;
|
|
for (unsigned int i=0; i<numParams; i++)
|
|
{
|
|
/* Copy */
|
|
vc->vparams[i] = params[i];
|
|
vc->vparams[i].offset = vc->call->GetParamInfo(i)->offset;
|
|
vc->vparams[i].obj_offset = last_extra_offset;
|
|
last_extra_offset += sizes[i];
|
|
}
|
|
|
|
vc->stackSize = normSize + extraSize;
|
|
vc->stackEnd = normSize;
|
|
|
|
return vc;
|
|
}
|