extensions/bintools/CallWrapper.h
author Nicholas Hastings <skamonkey@gmail.com>
Tue Jul 03 07:51:12 2012 -0400 (2012-07-03)
changeset 3573 440c51204096
parent 2504 74fea7db93e4
permissions -rw-r--r--
Fixed clientprefs prefab menu double free crash (bug 5374, r=asherkin).
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod BinTools Extension
     5  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
     6  * =============================================================================
     7  *
     8  * This program is free software; you can redistribute it and/or modify it under
     9  * the terms of the GNU General Public License, version 3.0, as published by the
    10  * Free Software Foundation.
    11  * 
    12  * This program is distributed in the hope that it will be useful, but WITHOUT
    13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
    15  * details.
    16  *
    17  * You should have received a copy of the GNU General Public License along with
    18  * this program.  If not, see <http://www.gnu.org/licenses/>.
    19  *
    20  * As a special exception, AlliedModders LLC gives you permission to link the
    21  * code of this program (as well as its derivative works) to "Half-Life 2," the
    22  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
    23  * by the Valve Corporation.  You must obey the GNU General Public License in
    24  * all respects for all other code used.  Additionally, AlliedModders LLC grants
    25  * this exception to all derivative works.  AlliedModders LLC defines further
    26  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
    27  * or <http://www.sourcemod.net/license.php>.
    28  *
    29  * Version: $Id$
    30  */
    31 
    32 #ifndef _INCLUDE_SOURCEMOD_CALLWRAPPER_H_
    33 #define _INCLUDE_SOURCEMOD_CALLWRAPPER_H_
    34 
    35 #include <IBinTools.h>
    36 #include <sourcehook_pibuilder.h>
    37 
    38 using namespace SourceMod;
    39 
    40 enum FuncAddrMethod
    41 {
    42 	FuncAddr_Direct,
    43 	FuncAddr_VTable
    44 };
    45 
    46 class CallWrapper : public ICallWrapper
    47 {
    48 public:
    49 	CallWrapper(const SourceHook::ProtoInfo *protoInfo);
    50 	~CallWrapper();
    51 public: //ICallWrapper
    52 	CallConvention GetCallConvention();
    53 	const PassEncode *GetParamInfo(unsigned int num);
    54 	const PassInfo *GetReturnInfo();
    55 	unsigned int GetParamCount();
    56 	void Execute(void *vParamStack, void *retBuffer);
    57 	void Destroy();
    58 	const SourceHook::PassInfo *GetSHReturnInfo();
    59 	SourceHook::ProtoInfo::CallConvention GetSHCallConvention();
    60 	const SourceHook::PassInfo *GetSHParamInfo(unsigned int num);
    61 	unsigned int GetParamOffset(unsigned int num);
    62 public:
    63 	void SetCalleeAddr(void *addr);
    64 	void SetCodeBaseAddr(void *addr);
    65 	void *GetCalleeAddr();
    66 	void *GetCodeBaseAddr();
    67 
    68 	void SetMemFuncInfo(const SourceHook::MemFuncInfo *funcInfo);
    69 	SourceHook::MemFuncInfo *GetMemFuncInfo();
    70 private:
    71 	PassEncode *m_Params;
    72 	SourceHook::ProtoInfo m_Info;
    73 	PassInfo *m_RetParam;
    74 	void *m_AddrCallee;
    75 	void *m_AddrCodeBase;
    76 	SourceHook::MemFuncInfo m_FuncInfo;
    77 };
    78 
    79 #endif //_INCLUDE_SOURCEMOD_CALLWRAPPER_H_