Add pushback immunity / extensive burn native

Author: nosoop

gamedata/tf2.utils.nosoop.txt

@@ -147,6 +147,20 @@
 		}
 		"Signatures"
 		{
+			"CTFPlayerShared::IsImmuneToPushback()"
+			{
+				// contains unique ref to string "spunup_push_force_immunity"
+				"library"	"server"
+				"linux"		"@_ZNK15CTFPlayerShared18IsImmuneToPushbackEv"
+				"windows"	"\x55\x8B\xEC\x51\x56\x8B\xF1\x57\x6A\x1C"
+			}
+			"CTFPlayerShared::Burn()"
+			{
+				// contains unique ref to string "afterburn_immunity"
+				"library"	"server"
+				"linux"		"@_ZN15CTFPlayerShared4BurnEP9CTFPlayerP13CTFWeaponBasef"
+				"windows"	"\x55\x8B\xEC\x83\xEC\x08\x56\x8B\xF1\x8B\x8E\x90\x01\x00\x00\x8B\x01"
+			}
 			"CTFPlayer::GetEntityForLoadoutSlot()"
 			{
 				// called a few blocks after function with unique x-ref string "enable_misc2_noisemaker"

scripting/include/tf2utils.inc

@@ -131,6 +131,30 @@ native float TF2Util_GetPlayerBurnDuration(int client);
  */
 native void TF2Util_SetPlayerBurnDuration(int client, float duration);
 
+/**
+ * Sets a client on fire.  This is effectively the same as `TF2_IgnitePlayer` but with the
+ * additional ability to set the weapon (allowing attribute modifiers to take effect).
+ * 
+ * @param client      Target client index.
+ * @param attacker    Attacker client index, or INVALID_ENT_REFERENCE for none.
+ * @param duration    Duration of fire, in seconds.
+ * @param weapon      Weapon entity, or INVALID_ENT_REFERENCE for none.
+ * 
+ * @note              As of Jungle Inferno, fire duration is capped to 10 seconds.
+ * @error             Invalid clients, or non-weapon entity.
+ */
+native void TF2Util_IgnitePlayer(int client, int attacker, float duration,
+		int weapon = INVALID_ENT_REFERENCE);
+
+/**
+ * Returns true if the given player is immune to pushback.
+ * 
+ * @param client    Client index.
+ * 
+ * @error           Invalid client or not in game.
+ */
+native bool TF2Util_IsPlayerImmuneToPushback(int client);
+
 /**
  * Returns the player's respawn time override, if any.  This is the number of seconds that a
  * player will spend waiting to be respawned after their last death (CTFPlayer::m_flDeathTime).

scripting/tf2utils.sp

@@ -11,7 +11,7 @@
 
 #include <stocksoup/memory>
 
-#define PLUGIN_VERSION "0.17.0"
+#define PLUGIN_VERSION "0.18.0"
 public Plugin myinfo = {
 	name = "TF2 Utils",
 	author = "nosoop",
@@ -40,6 +40,8 @@ Handle g_SDKCallIsEntityWearable;
 Handle g_SDKCallPlayerEquipWearable;
 
 Handle g_SDKCallPointInRespawnRoom;
+Handle g_SDKCallPlayerSharedImmuneToPushback;
+Handle g_SDKCallPlayerSharedBurn;
 
 Address offs_ConditionNames;
 Address offs_CTFPlayer_aObjects;
@@ -77,6 +79,9 @@ public APLRes AskPluginLoad2(Handle self, bool late, char[] error, int maxlen) {
 	CreateNative("TF2Util_SetPlayerConditionProvider", Native_SetPlayerConditionProvider);
 	CreateNative("TF2Util_GetPlayerBurnDuration", Native_GetPlayerBurnDuration);
 	CreateNative("TF2Util_SetPlayerBurnDuration", Native_SetPlayerBurnDuration);
+	CreateNative("TF2Util_IgnitePlayer", Native_IgnitePlayer);
+	CreateNative("TF2Util_IsPlayerImmuneToPushback", Native_IsPlayerImmuneToPushback);
+	
 	CreateNative("TF2Util_GetPlayerRespawnTimeOverride", Native_GetPlayerRespawnTimeOverride);
 	CreateNative("TF2Util_SetPlayerRespawnTimeOverride", Native_SetPlayerRespawnTimeOverride);
 
@@ -145,6 +150,19 @@ public void OnPluginStart() {
 	PrepSDKCall_AddParameter(SDKType_Bool, SDKPass_Plain);
 	g_SDKCallPlayerSharedGetMaxHealth = EndPrepSDKCall();
 
+	StartPrepSDKCall(SDKCall_Raw);
+	PrepSDKCall_SetFromConf(hGameConf, SDKConf_Signature, "CTFPlayerShared::Burn()");
+	PrepSDKCall_AddParameter(SDKType_CBasePlayer, SDKPass_Pointer, VDECODE_FLAG_ALLOWNULL);
+	PrepSDKCall_AddParameter(SDKType_CBaseEntity, SDKPass_Pointer, VDECODE_FLAG_ALLOWNULL);
+	PrepSDKCall_AddParameter(SDKType_Float, SDKPass_Plain);
+	g_SDKCallPlayerSharedBurn = EndPrepSDKCall();
+	
+	StartPrepSDKCall(SDKCall_Raw);
+	PrepSDKCall_SetFromConf(hGameConf, SDKConf_Signature,
+			"CTFPlayerShared::IsImmuneToPushback()");
+	PrepSDKCall_SetReturnInfo(SDKType_Bool, SDKPass_Plain);
+	g_SDKCallPlayerSharedImmuneToPushback = EndPrepSDKCall();
+	
 	StartPrepSDKCall(SDKCall_Entity);
 	PrepSDKCall_SetReturnInfo(SDKType_PlainOldData, SDKPass_Plain);
 	PrepSDKCall_SetFromConf(hGameConf, SDKConf_Virtual, "CBaseEntity::GetMaxHealth()");
@@ -364,9 +382,7 @@ int Native_GetMaxHealthBoost(Handle plugin, int nParams) {
 		ThrowNativeError(SP_ERROR_NATIVE, "Client index %d is invalid", client);
 	}
 
-	Address offs_Shared = view_as<Address>(GetEntSendPropOffs(client, "m_Shared", true));
-	
-	return SDKCall(g_SDKCallPlayerSharedGetMaxHealth, GetEntityAddress(client) + offs_Shared,
+	return SDKCall(g_SDKCallPlayerSharedGetMaxHealth, GetPlayerSharedAddress(client),
 			bIgnoreAttributes, bIgnoreOverheal);
 }
 
@@ -530,6 +546,19 @@ int Native_GetWeaponMaxClip(Handle plugin, int nParams) {
 	return SDKCall(g_SDKCallWeaponGetMaxClip, entity);
 }
 
+// bool(int client);
+int Native_IsPlayerImmuneToPushback(Handle plugin, int nParams) {
+	int client = GetNativeCell(1);
+	
+	if (!(0 < client <= MaxClients)) {
+		return ThrowNativeError(SP_ERROR_NATIVE, "Client %d index is not valid", client);
+	} else if (!IsClientInGame(client)) {
+		return ThrowNativeError(SP_ERROR_NATIVE, "Client %d is not in game", client);
+	}
+	
+	return SDKCall(g_SDKCallPlayerSharedImmuneToPushback, GetPlayerSharedAddress(client));
+}
+
 // bool(const float[3] position, int entity, bool bRestrictToSameTeam)
 int Native_IsPointInRespawnRoom(Handle plugin, int nParams) {
 	if (IsNativeParamNullVector(1)) {
@@ -677,6 +706,37 @@ any Native_SetPlayerBurnDuration(Handle plugin, int numParams) {
 	SetEntDataFloat(client, pOffsSharedBurnDuration, duration);
 }
 
+// void(int client, int attacker, float duration, int weapon = INVALID_ENT_REFERENCE);
+any Native_IgnitePlayer(Handle plugin, int numParams) {
+	int client = GetNativeCell(1);
+	int attacker = GetNativeCell(2);
+	float duration = GetNativeCell(3);
+	int weapon = GetNativeCell(4);
+	
+	if (!(0 < client <= MaxClients)) {
+		return ThrowNativeError(SP_ERROR_NATIVE, "Client %d index is not valid", client);
+	} else if (!IsClientInGame(client)) {
+		return ThrowNativeError(SP_ERROR_NATIVE, "Client %d is not in game", client);
+	}
+	
+	if (attacker != INVALID_ENT_REFERENCE) {
+		if (!(0 < attacker <= MaxClients)) {
+			return ThrowNativeError(SP_ERROR_NATIVE, "Client %d index is not valid", attacker);
+		} else if (!IsClientInGame(attacker)) {
+			return ThrowNativeError(SP_ERROR_NATIVE, "Client %d is not in game", attacker);
+		}
+	}
+	
+	if (weapon != INVALID_ENT_REFERENCE) {
+		if (!IsValidEntity(weapon) || !IsEntityWeapon(weapon)) {
+			return ThrowNativeError(SP_ERROR_NATIVE, "Entity %d is not a valid weapon", weapon);
+		}
+	}
+	
+	SDKCall(g_SDKCallPlayerSharedBurn, GetPlayerSharedAddress(client), attacker, weapon,
+			duration);
+}
+
 // float(int client);
 any Native_GetPlayerRespawnTimeOverride(Handle plugin, int numParams) {
 	int client = GetNativeCell(1);
@@ -730,6 +790,10 @@ static Address GetConditionData(int client, TFCond cond) {
 	return pCondMemory + view_as<Address>(view_as<int>(cond) * sizeof_TFCondInfo);
 }
 
+static Address GetPlayerSharedAddress(int client) {
+	return GetEntityAddress(client) + FindSendPropInfo("CTFPlayer", "m_Shared");
+}
+
 static bool IsConditionValid(TFCond cond) {
 	return 0 <= view_as<any>(cond) < g_nConditions;
 }