sourcemod-central: extensions/bintools/HookWrapper.h@74fea7db93e4

     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: CallMaker.h 1964 2008-03-27 04:54:56Z damagedsoul $

    30  */

    32 #ifndef _INCLUDE_SOURCEMOD_HOOKWRAPPER_H_

    33 #define _INCLUDE_SOURCEMOD_HOOKWRAPPER_H_

    35 #if defined HOOKING_ENABLED

    37 #include "smsdk_ext.h"

    38 #include "sourcehook.h"

    39 #include "IBinTools.h"

    41 #define VTABLE_PATCH_OFFS		2

    43 using namespace SourceMod;

    45 class HookWrapper : public IHookWrapper

    46 {

    47 public:

    48 	HookWrapper(SourceHook::ISourceHook *pSH,

    49 				const SourceHook::ProtoInfo *proto,

    50 				SourceHook::MemFuncInfo *memfunc,

    51 				void *addr

    52 				);

    53 	~HookWrapper();

    54 public: //IHookWrapper

    55 	ISMDelegate *CreateDelegate(void *data);

    56 	unsigned int GetParamCount();

    57 	unsigned int GetParamOffset(unsigned int argnum, unsigned int *size);

    58 	void PerformRecall(void *params, void *retval);

    59 	void Destroy();

    60 public:

    61 	unsigned int GetParamSize();

    62 	unsigned int GetRetSize();

    63 	SourceHook::ProtoInfo *GetProtoInfo();

    64 	void *GetHandlerAddr();

    65 	void SetHookWrpAddr(void *addr);

    66 	void SetCallWrapperAddr(ICallWrapper *wrap);

    67 private:

    68 	void *m_FuncAddr;

    69 	void *m_HookWrapper;

    70 	SourceHook::ISourceHook *m_pSH;

    71 	unsigned int *m_ParamOffs;

    72 	unsigned int m_ParamSize;

    73 	unsigned int m_RetSize;

    74 	SourceHook::MemFuncInfo m_MemFuncInfo;

    75 	SourceHook::ProtoInfo m_ProtoInfo;

    76 	ICallWrapper *m_CallWrapper;

    77 };

    79 class SMDelegate : public ISMDelegate

    80 {

    81 public:

    82 	SMDelegate(void *data);

    83 public: //SourceHook::ISHDelegate

    84 	bool IsEqual(ISHDelegate *pOtherDeleg);

    85 	void DeleteThis();

    86 	void Call();

    87 public: //ISMDelegate

    88 	void *GetUserData();

    89 public:

    90 	void PatchVtable(void *addr);

    91 private:

    92 	void *m_Data;

    93 };

    95 #endif

    97 #endif //_INCLUDE_SOURCEMOD_HOOKWRAPPER_H_

author	Matt Woodrow <pred@alliedmods.net>
	Thu Jan 08 18:17:17 2009 +1300 (2009-01-08)
changeset 2504	74fea7db93e4
permissions	-rw-r--r--