1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/extensions/tf2/natives.cpp Sun Mar 30 07:00:22 2008 +0000
1.3 @@ -0,0 +1,159 @@
1.4 +/**
1.5 + * vim: set ts=4 :
1.6 + * =============================================================================
1.7 + * SourceMod Counter-Strike:Source 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 "RegNatives.h"
1.37 +
1.38 +#define REGISTER_NATIVE_ADDR(name, code) \
1.39 + void *addr; \
1.40 + if (!g_pGameConf->GetMemSig(name, &addr)) \
1.41 + { \
1.42 + return pContext->ThrowNativeError("Failed to locate function"); \
1.43 + } \
1.44 + code; \
1.45 + g_RegNatives.Register(pWrapper);
1.46 +
1.47 +inline CBaseEntity *GetCBaseEntity(int num, bool onlyPlayers)
1.48 +{
1.49 + edict_t *pEdict = engine->PEntityOfEntIndex(num);
1.50 + if (!pEdict || pEdict->IsFree())
1.51 + {
1.52 + return NULL;
1.53 + }
1.54 +
1.55 + if (num > 0 && num < playerhelpers->GetMaxClients())
1.56 + {
1.57 + IGamePlayer *pPlayer = playerhelpers->GetGamePlayer(pEdict);
1.58 + if (!pPlayer || !pPlayer->IsConnected())
1.59 + {
1.60 + return NULL;
1.61 + }
1.62 + }
1.63 + else if (onlyPlayers)
1.64 + {
1.65 + return NULL;
1.66 + }
1.67 +
1.68 + IServerUnknown *pUnk;
1.69 + if ((pUnk=pEdict->GetUnknown()) == NULL)
1.70 + {
1.71 + return NULL;
1.72 + }
1.73 +
1.74 + return pUnk->GetBaseEntity();
1.75 +}
1.76 +
1.77 +
1.78 +// native TF2_Burn(client)
1.79 +cell_t TF2_Burn(IPluginContext *pContext, const cell_t *params)
1.80 +{
1.81 + static ICallWrapper *pWrapper = NULL;
1.82 +
1.83 + // CTFPlayerShared::Burn(CTFPlayer*)
1.84 + if (!pWrapper)
1.85 + {
1.86 + REGISTER_NATIVE_ADDR("Burn",
1.87 + PassInfo pass[1]; \
1.88 + pass[0].flags = PASSFLAG_BYVAL; \
1.89 + pass[0].size = sizeof(CBaseEntity *); \
1.90 + pass[0].type = PassType_Basic; \
1.91 + pWrapper = g_pBinTools->CreateCall(addr, CallConv_ThisCall, NULL, pass, 1))
1.92 + }
1.93 +
1.94 + CBaseEntity *pEntity;
1.95 + if (!(pEntity=GetCBaseEntity(params[1], true)))
1.96 + {
1.97 + return pContext->ThrowNativeError("Client index %d is not valid", params[1]);
1.98 + }
1.99 +
1.100 + void *obj = (void *)((uint8_t *)pEntity + playerSharedOffset->GetOffset());
1.101 +
1.102 + unsigned char vstk[sizeof(void *) + sizeof(CBaseEntity *)];
1.103 + unsigned char *vptr = vstk;
1.104 +
1.105 + *(void **)vptr = obj;
1.106 + vptr += sizeof(void *);
1.107 + *(CBaseEntity **)vptr = pEntity;
1.108 +
1.109 + pWrapper->Execute(vstk, NULL);
1.110 +
1.111 + return 1;
1.112 +}
1.113 +
1.114 +// native TF2_Invuln(client, bool:something, bool:anothersomething)
1.115 +cell_t TF2_Invuln(IPluginContext *pContext, const cell_t *params)
1.116 +{
1.117 + static ICallWrapper *pWrapper = NULL;
1.118 +
1.119 + //CTFPlayerShared::SetInvulnerable(bool, bool)
1.120 + if (!pWrapper)
1.121 + {
1.122 + REGISTER_NATIVE_ADDR("Invuln",
1.123 + PassInfo pass[2]; \
1.124 + pass[0].flags = PASSFLAG_BYVAL; \
1.125 + pass[0].size = sizeof(bool); \
1.126 + pass[0].type = PassType_Basic; \
1.127 + pass[1].flags = PASSFLAG_BYVAL; \
1.128 + pass[1].size = sizeof(bool); \
1.129 + pass[1].type = PassType_Basic; \
1.130 + pWrapper = g_pBinTools->CreateCall(addr, CallConv_ThisCall, NULL, pass, 2))
1.131 + }
1.132 +
1.133 + CBaseEntity *pEntity;
1.134 + if (!(pEntity=GetCBaseEntity(params[1], true)))
1.135 + {
1.136 + return pContext->ThrowNativeError("Client index %d is not valid", params[1]);
1.137 + }
1.138 +
1.139 + void *obj = (void *)((uint8_t *)pEntity + playerSharedOffset->GetOffset());
1.140 +
1.141 + unsigned char vstk[sizeof(void *) + 2*sizeof(bool)];
1.142 + unsigned char *vptr = vstk;
1.143 +
1.144 +
1.145 + *(void **)vptr = obj;
1.146 + vptr += sizeof(bool);
1.147 + *(bool *)vptr = !!params[2];
1.148 + vptr += sizeof(bool);
1.149 + *(bool *)vptr = !!params[3];
1.150 +
1.151 + pWrapper->Execute(vstk, NULL);
1.152 +
1.153 + return 1;
1.154 +}
1.155 +
1.156 +
1.157 +sp_nativeinfo_t g_TFNatives[] =
1.158 +{
1.159 + {"TF2_Burn", TF2_Burn},
1.160 + {"TF2_Invuln", TF2_Invuln},
1.161 + {NULL, NULL}
1.162 +};
