1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/extensions/bintools/CallWrapper.cpp Sun Mar 30 07:00:22 2008 +0000
1.3 @@ -0,0 +1,111 @@
1.4 +/**
1.5 + * vim: set ts=4 :
1.6 + * =============================================================================
1.7 + * SourceMod BinTools Extension
1.8 + * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
1.9 + * =============================================================================
1.10 + *
1.11 + * This program is free software; you can redistribute it and/or modify it under
1.12 + * the terms of the GNU General Public License, version 3.0, as published by the
1.13 + * Free Software Foundation.
1.14 + *
1.15 + * This program is distributed in the hope that it will be useful, but WITHOUT
1.16 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.17 + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.18 + * details.
1.19 + *
1.20 + * You should have received a copy of the GNU General Public License along with
1.21 + * this program. If not, see <http://www.gnu.org/licenses/>.
1.22 + *
1.23 + * As a special exception, AlliedModders LLC gives you permission to link the
1.24 + * code of this program (as well as its derivative works) to "Half-Life 2," the
1.25 + * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
1.26 + * by the Valve Corporation. You must obey the GNU General Public License in
1.27 + * all respects for all other code used. Additionally, AlliedModders LLC grants
1.28 + * this exception to all derivative works. AlliedModders LLC defines further
1.29 + * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
1.30 + * or <http://www.sourcemod.net/license.php>.
1.31 + *
1.32 + * Version: $Id$
1.33 + */
1.34 +
1.35 +#include "extension.h"
1.36 +#include "CallWrapper.h"
1.37 +
1.38 +CallWrapper::CallWrapper(CallConvention cv, const PassInfo *paramInfo, const PassInfo *retInfo, unsigned int numParams)
1.39 +{
1.40 + m_Cv = cv;
1.41 +
1.42 + if (numParams)
1.43 + {
1.44 + m_Params = new PassEncode[numParams];
1.45 + for (size_t i=0; i<numParams; i++)
1.46 + {
1.47 + m_Params[i].info = paramInfo[i];
1.48 + }
1.49 + } else {
1.50 + m_Params = NULL;
1.51 + }
1.52 +
1.53 + if (retInfo)
1.54 + {
1.55 + m_RetParam = new PassInfo;
1.56 + *m_RetParam = *retInfo;
1.57 + } else {
1.58 + m_RetParam = NULL;
1.59 + }
1.60 +
1.61 + m_NumParams = numParams;
1.62 +
1.63 + /* Calculate virtual stack offsets for each parameter */
1.64 + size_t offs = 0;
1.65 +
1.66 + if (cv == CallConv_ThisCall)
1.67 + {
1.68 + offs += sizeof(void *);
1.69 + }
1.70 + for (size_t i=0; i<numParams; i++)
1.71 + {
1.72 + m_Params[i].offset = offs;
1.73 + offs += m_Params[i].info.size;
1.74 + }
1.75 +}
1.76 +
1.77 +CallWrapper::~CallWrapper()
1.78 +{
1.79 + delete [] m_Params;
1.80 + delete m_RetParam;
1.81 + g_SPEngine->FreePageMemory(m_Addrs[ADDR_CODEBASE]);
1.82 +}
1.83 +
1.84 +void CallWrapper::Destroy()
1.85 +{
1.86 + delete this;
1.87 +}
1.88 +
1.89 +CallConvention CallWrapper::GetCallConvention()
1.90 +{
1.91 + return m_Cv;
1.92 +}
1.93 +
1.94 +const PassEncode *CallWrapper::GetParamInfo(unsigned int num)
1.95 +{
1.96 + return (num+1 > m_NumParams) ? NULL : &m_Params[num];
1.97 +}
1.98 +
1.99 +const PassInfo *CallWrapper::GetReturnInfo()
1.100 +{
1.101 + return m_RetParam;
1.102 +}
1.103 +
1.104 +unsigned int CallWrapper::GetParamCount()
1.105 +{
1.106 + return m_NumParams;
1.107 +}
1.108 +
1.109 +void CallWrapper::Execute(void *vParamStack, void *retBuffer)
1.110 +{
1.111 + typedef void (*CALL_EXECUTE)(void *, void *);
1.112 + CALL_EXECUTE fn = (CALL_EXECUTE)m_Addrs[ADDR_CODEBASE];
1.113 + fn(vParamStack, retBuffer);
1.114 +}