YimMenu/src/hooks/protections/can_apply_data.cpp

#include "hooking.hpp"
#include "core/globals.hpp"
#include "base/CObject.hpp"
#include "entities/fwEntity.hpp"
#include "rage/netSyncDataNodeBase.hpp"
#include "rage/netSyncTree.hpp"
#include "gta/net_object_mgr.hpp"
#include "datanodes/door/CDoorCreationDataNode.hpp"
#include "datanodes/dynamic_entity/CDynamicEntityGameStateDataNode.hpp"
#include "datanodes/object/CObjectCreationDataNode.hpp"
#include "datanodes/ped/CPedAttachDataNode.hpp"
#include "datanodes/ped/CPedCreationDataNode.hpp"
#include "datanodes/ped/CPedInventoryDataNode.hpp"
#include "datanodes/pickup/CPickupCreationDataNode.hpp"
#include "datanodes/physical/CPhysicalAttachDataNode.hpp"
#include "datanodes/player/CPlayerAppearanceDataNode.hpp"
#include "datanodes/player/CPlayerCreationDataNode.hpp"
#include "datanodes/proximity_migrateable/CSectorDataNode.hpp"
#include "datanodes/vehicle/CVehicleCreationDataNode.hpp"
#include "datanodes/vehicle/CVehicleGadgetDataNode.hpp"
#include "datanodes/train/CTrainGameStateDataNode.hpp"
#include "network/netObject.hpp"
#include "base/CBaseModelInfo.hpp"
#include "vehicle/CVehicleModelInfo.hpp"
#include "util/model_info.hpp"
#include "network/CNetGamePlayer.hpp"
#include "util/notify.hpp"

namespace big
{
	constexpr auto nodes = std::to_array<std::pair<Hash, uint64_t>>({
		{RAGE_JOAAT("CAutomobileCreationDataNode"), 0x1b088e0 + 64},
		{RAGE_JOAAT("CBikeGameStateDataNode"), 0x1b08a40 + 64},
		{RAGE_JOAAT("CBoatGameStateDataNode"), 0x1b08a80 + 64},
		{RAGE_JOAAT("CDoorCreationDataNode"), 0x1b08be0 + 64},
		{RAGE_JOAAT("CDoorMovementDataNode"), 0x1b09000 + 64},
		{RAGE_JOAAT("CDoorScriptGameStateDataNode"), 0x1b08ea0 + 64},
		{RAGE_JOAAT("CDoorScriptInfoDataNode"), 0x1b08d40 + 64},
		{RAGE_JOAAT("CDynamicEntityGameStateDataNode"), 0x1b06450 + 64},
		{RAGE_JOAAT("CEntityOrientationDataNode"), 0x1b05df0 + 64},
		{RAGE_JOAAT("CEntityScriptGameStateDataNode"), 0x1b05c90 + 64},
		{RAGE_JOAAT("CEntityScriptInfoDataNode"), 0x1b05b30 + 64},
		{RAGE_JOAAT("CGlobalFlagsDataNode"), 0x1b057f0 + 64},
		{RAGE_JOAAT("CHeliControlDataNode"), 0x1b08180 + 64},
		{RAGE_JOAAT("CHeliHealthDataNode"), 0x1b06f60 + 64},
		{RAGE_JOAAT("CMigrationDataNode"), 0x1b05af0 + 64},
		{RAGE_JOAAT("CObjectCreationDataNode"), 0x1b096e0 + 64},
		{RAGE_JOAAT("CObjectGameStateDataNode"), 0x1b09840 + 64},
		{RAGE_JOAAT("CObjectOrientationNode"), 0x1b09c60 + 64},
		{RAGE_JOAAT("CObjectScriptGameStateDataNode"), 0x1b099a0 + 64},
		{RAGE_JOAAT("CObjectSectorPosNode"), 0x1b09b00 + 64},
		{RAGE_JOAAT("CPedAIDataNode"), 0x1b0ae80 + 64},
		{RAGE_JOAAT("CPedAppearanceDataNode"), 0x1b0ad20 + 64},
		{RAGE_JOAAT("CPedAttachDataNode"), 0x1b0a900 + 64},
		{RAGE_JOAAT("CPedComponentReservationDataNode"), 0x1b0a380 + 64},
		{RAGE_JOAAT("CPedCreationDataNode"), 0x1b09ca0 + 64},
		{RAGE_JOAAT("CPedGameStateDataNode"), 0x1b09f60 + 64},
		{RAGE_JOAAT("CPedHealthDataNode"), 0x1b0a7a0 + 64},
		{RAGE_JOAAT("CPedInventoryDataNode"), 0x1b0b2a0 + 64},
		{RAGE_JOAAT("CPedMovementDataNode"), 0x1b0abc0 + 64},
		{RAGE_JOAAT("CPedMovementGroupDataNode"), 0x1b0aa60 + 64},
		{RAGE_JOAAT("CPedOrientationDataNode"), 0x1b0a640 + 64},
		{RAGE_JOAAT("CPedScriptCreationDataNode"), 0x1b09e00 + 64},
		{RAGE_JOAAT("CPedScriptGameStateDataNode"), 0x1b0a4e0 + 64},
		{RAGE_JOAAT("CPedSectorPosMapNode"), 0x1b05ab0 + 64},
		{RAGE_JOAAT("CPedSectorPosNavMeshNode"), 0x1b0a220 + 64},
		{RAGE_JOAAT("CPedTaskSequenceDataNode"), 0x1b0b400 + 64},
		{RAGE_JOAAT("CPedTaskSpecificDataNode"), 0x1b0b140 + 64},
		{RAGE_JOAAT("CPedTaskTreeDataNode"), 0x1b0afe0 + 64},
		{RAGE_JOAAT("CPhysicalAngVelocityDataNode"), 0x1b06f20 + 64},
		{RAGE_JOAAT("CPhysicalAttachDataNode"), 0x1b06fa0 + 64},
		{RAGE_JOAAT("CPhysicalGameStateDataNode"), 0x1b06b00 + 64},
		{RAGE_JOAAT("CPhysicalHealthDataNode"), 0x1b06f60 + 64},
		{RAGE_JOAAT("CPhysicalMigrationDataNode"), 0x1b07100 + 64},
		{RAGE_JOAAT("CPhysicalScriptGameStateDataNode"), 0x1b06c60 + 64},
		{RAGE_JOAAT("CPhysicalScriptMigrationDataNode"), 0x1b07140 + 64},
		{RAGE_JOAAT("CPhysicalVelocityDataNode"), 0x1b06dc0 + 64},
		{RAGE_JOAAT("CPickupCreationDataNode"), 0x1b0b560 + 64},
		{RAGE_JOAAT("CPickupPlacementCreationDataNode"), 0x1b0b860 + 64},
		{RAGE_JOAAT("CPickupPlacementStateDataNode"), 0x1b0b9c0 + 64},
		{RAGE_JOAAT("CPickupScriptGameStateNode"), 0x1b0b820 + 64},
		{RAGE_JOAAT("CPickupSectorPosNode"), 0x1b0b6c0 + 64},
		{RAGE_JOAAT("CPlaneControlDataNode"), 0x1b08180 + 64},
		{RAGE_JOAAT("CPlaneGameStateDataNode"), 0x1b0c380 + 64},
		{RAGE_JOAAT("CPlayerAmbientModelStreamingNode"), 0x1b0c040 + 64},
		{RAGE_JOAAT("CPlayerAppearanceDataNode"), 0x1b0bc20 + 64},
		{RAGE_JOAAT("CPlayerCameraDataNode"), 0x1b0bba0 + 64},
		{RAGE_JOAAT("CPlayerCreationDataNode"), 0x1b0ba00 + 64},
		{RAGE_JOAAT("CPlayerExtendedGameStateNode"), 0x1b0c300 + 64},
		{RAGE_JOAAT("CPlayerGameStateDataNode"), 0x1b0bbe0 + 64},
		{RAGE_JOAAT("CPlayerGamerDataNode"), 0x1b0c1a0 + 64},
		{RAGE_JOAAT("CPlayerPedGroupDataNode"), 0x1b0bd80 + 64},
		{RAGE_JOAAT("CPlayerSectorPosNode"), 0x1b0bb60 + 64},
		{RAGE_JOAAT("CPlayerWantedAndLOSDataNode"), 0x1b0bee0 + 64},
		{RAGE_JOAAT("CSectorDataNode"), 0x1b05950 + 64},
		{RAGE_JOAAT("CSectorPositionDataNode"), 0x1b05ab0 + 64},
		{RAGE_JOAAT("CSubmarineControlDataNode"), 0x1b08180 + 64},
		{RAGE_JOAAT("CSubmarineGameStateDataNode"), 0x1b092c0 + 64},
		{RAGE_JOAAT("CTrainGameStateDataNode"), 0x1b0c340 + 64},
		{RAGE_JOAAT("CVehicleAngVelocityDataNode"), 0x1b06f20 + 64},
		{RAGE_JOAAT("CVehicleAppearanceDataNode"), 0x1b082e0 + 64},
		{RAGE_JOAAT("CVehicleComponentReservationDataNode"), 0x1b085e0 + 64},
		{RAGE_JOAAT("CVehicleControlDataNode"), 0x1b08180 + 64},
		{RAGE_JOAAT("CVehicleCreationDataNode"), 0x1b07940 + 64},
		{RAGE_JOAAT("CVehicleDamageStatusDataNode"), 0x1b08440 + 64},
		{RAGE_JOAAT("CVehicleGadgetDataNode"), 0x1b08780 + 64},
		{RAGE_JOAAT("CVehicleGameStateDataNode"), 0x1b07c00 + 64},
		{RAGE_JOAAT("CVehicleHealthDataNode"), 0x1b07ec0 + 64},
		{RAGE_JOAAT("CVehicleProximityMigrationDataNode"), 0x1b08740 + 64},
		{RAGE_JOAAT("CVehicleScriptGameStateDataNode"), 0x1b07d60 + 64},
		{RAGE_JOAAT("CVehicleSteeringDataNode"), 0x1b08020 + 64},
		{RAGE_JOAAT("CVehicleTaskDataNode"), 0x1b085a0 + 64} });

	constexpr uint32_t crash_peds[] = { RAGE_JOAAT("slod_human"), RAGE_JOAAT("slod_small_quadped"), RAGE_JOAAT("slod_large_quadped") };

	constexpr uint32_t crash_vehicles[] = { RAGE_JOAAT("arbitergt"), RAGE_JOAAT("astron2"), RAGE_JOAAT("cyclone2"), RAGE_JOAAT("ignus2"), RAGE_JOAAT("s95") };

	constexpr uint32_t crash_objects[] = { RAGE_JOAAT("prop_dummy_01"), RAGE_JOAAT("prop_dummy_car"), RAGE_JOAAT("prop_dummy_light"), RAGE_JOAAT("prop_dummy_plane"),
		RAGE_JOAAT("prop_distantcar_night"), RAGE_JOAAT("prop_distantcar_day"), RAGE_JOAAT("hei_bh1_08_details4_em_night"),
		RAGE_JOAAT("dt1_18_sq_night_slod"), RAGE_JOAAT("ss1_12_night_slod"), -1288391198, RAGE_JOAAT("h4_prop_bush_bgnvla_med_01"), RAGE_JOAAT("h4_prop_bush_bgnvla_lrg_01"),
		RAGE_JOAAT("h4_prop_bush_buddleia_low_01"), RAGE_JOAAT("h4_prop_bush_ear_aa"), RAGE_JOAAT("h4_prop_bush_ear_ab"), RAGE_JOAAT("h4_prop_bush_fern_low_01"),
		RAGE_JOAAT("h4_prop_bush_fern_tall_cc"), RAGE_JOAAT("h4_prop_bush_mang_ad"), RAGE_JOAAT("h4_prop_bush_mang_low_aa"), RAGE_JOAAT("h4_prop_bush_mang_low_ab"),
		RAGE_JOAAT("h4_prop_bush_seagrape_low_01"), RAGE_JOAAT("prop_h4_ground_cover"), RAGE_JOAAT("h4_prop_weed_groundcover_01"), RAGE_JOAAT("h4_prop_grass_med_01"),
		RAGE_JOAAT("h4_prop_grass_tropical_lush_01"), RAGE_JOAAT("h4_prop_grass_wiregrass_01"), RAGE_JOAAT("h4_prop_weed_01_plant"), RAGE_JOAAT("h4_prop_weed_01_row"),
		RAGE_JOAAT("urbanweeds02_l1"), RAGE_JOAAT("proc_forest_grass01"), RAGE_JOAAT("prop_small_bushyba"), RAGE_JOAAT("v_res_d_dildo_a"), RAGE_JOAAT("v_res_d_dildo_b"), RAGE_JOAAT("v_res_d_dildo_c"),
		RAGE_JOAAT("v_res_d_dildo_d"), RAGE_JOAAT("v_res_d_dildo_e"), RAGE_JOAAT("v_res_d_dildo_f"), RAGE_JOAAT("v_res_skateboard"), RAGE_JOAAT("prop_battery_01"), RAGE_JOAAT("prop_barbell_01"),
		RAGE_JOAAT("prop_barbell_02"), RAGE_JOAAT("prop_bandsaw_01"), RAGE_JOAAT("prop_bbq_3"), RAGE_JOAAT("v_med_curtainsnewcloth2"), RAGE_JOAAT("bh1_07_flagpoles"),
		92962485 };

	inline CObject* get_game_object(rage::netObject* object)
	{
		return *(CObject**)((__int64)object + 0x50);
	}

	inline bool is_crash_ped(uint32_t model)
	{
		for (auto iterator : crash_peds)
			if (iterator == model)
				return true;
		return false;
	}

	inline bool is_crash_vehicle(uint32_t model)
	{
		for (auto iterator : crash_vehicles)
			if (iterator == model)
				return true;
		return false;
	}

	inline bool is_crash_object(uint32_t model)
	{
		for (auto iterator : crash_objects)
			if (iterator == model)
				return true;
		return false;
	}

	inline bool is_attachment_infinite(rage::CDynamicEntity* object, uint16_t attached_to_net_id)
	{
		if (object == nullptr)
			return false;

		while (object = g_pointers->m_get_entity_attached_to(object))
		{
			if (object->m_net_object && object->m_net_object->m_object_id == attached_to_net_id)
				return true;
		}

		return false;
	}

	bool check_node(rage::netSyncNodeBase* node, CNetGamePlayer* sender, rage::netObject* object)
	{
		if (node->IsParentNode())
		{
			for (auto child = node->m_first_child; child; child = child->m_next_sibling)
			{
				if (check_node(child, sender, object))
					return true;
			}
		}
		else if (node->IsDataNode())
		{
			auto offset_to_address = [](uint64_t off)
			{
				static auto base_addr = reinterpret_cast<uint64_t>(GetModuleHandle(nullptr));
				return base_addr + off;
			};

			auto vtable = *(void**)node;
			Hash node_hash = 0;

			for (const auto& n : nodes)
			{
				if ((void*)offset_to_address(n.second) == vtable)
				{
					node_hash = n.first;
					break;
				}
			}

			switch (node_hash)
			{
				case RAGE_JOAAT("CDoorCreationDataNode"):
				{
					const auto creation_node = (CDoorCreationDataNode*)(node);
					if (is_crash_object(creation_node->m_model))
					{
						notify::crash_blocked(sender, "invalid door model");
						return true;
					}
					break;
				}
				case RAGE_JOAAT("CPickupCreationDataNode"):
				{
					const auto creation_node = (CPickupCreationDataNode*)(node);
					if (is_crash_object(creation_node->m_custom_model))
					{
						notify::crash_blocked(sender, "invalid pickup model");
						return true;
					}
					break;
				}
				case RAGE_JOAAT("CPhysicalAttachDataNode"):
				{
					const auto attach_node = (CPhysicalAttachDataNode*)(node);

					// TODO: Find a better method to avoid false positives
					auto model_hash = get_game_object(object) ? get_game_object(object)->m_model_info->m_hash : 0;
					if (attach_node->m_attached && attach_node->m_attached_to == object->m_object_id && (model_hash != RAGE_JOAAT("hauler2") && model_hash != RAGE_JOAAT("phantom3")))
					{
						// notify::crash_blocked(sender, "infinite physical attachment");
						return true;
					}
					else if (attach_node->m_attached && is_attachment_infinite((rage::CDynamicEntity*)get_game_object(object), attach_node->m_attached_to))
					{
						// notify::crash_blocked(sender, "recursive infinite physical attachment");
						return true;
					}

					break;
				}
				case RAGE_JOAAT("CPedCreationDataNode"):
				{
					const auto creation_node = (CPedCreationDataNode*)(node);
					if (is_crash_ped(creation_node->m_model))
					{
						notify::crash_blocked(sender, "invalid ped model");
						return true;
					}
					else if (creation_node->m_has_prop && is_crash_object(creation_node->m_prop_model))
					{
						notify::crash_blocked(sender, "invalid ped prop model");
						return true;
					}
					break;
				}
				case RAGE_JOAAT("CPedAttachDataNode"):
				{
					const auto attach_node = (CPedAttachDataNode*)(node);
					if (attach_node->m_attached && attach_node->m_attached_to == object->m_object_id)
					{
						notify::crash_blocked(sender, "infinite ped attachment");
						return true;
					}
					else if (attach_node->m_attached && is_attachment_infinite(get_game_object(object), attach_node->m_attached_to))
					{
						notify::crash_blocked(sender, "recursive infinite ped attachment");
						return true;
					}

					break;
				}
				case RAGE_JOAAT("CObjectCreationDataNode"):
				{
					const auto creation_node = (CObjectCreationDataNode*)(node);
					if (is_crash_object(creation_node->m_model))
					{
						notify::crash_blocked(sender, "invalid object model");
						return true;
					}
					break;
				}
				case RAGE_JOAAT("CPlayerAppearanceDataNode"):
				{
					const auto player_appearance_node = (CPlayerAppearanceDataNode*)(node);
					if (is_crash_ped(player_appearance_node->m_model_hash))
					{
						notify::crash_blocked(sender, "invalid player model (appearance node)");
						return true;
					}
					break;
				}
				case RAGE_JOAAT("CPlayerCreationDataNode"):
				{
					const auto player_creation_node = (CPlayerCreationDataNode*)(node);
					if (is_crash_ped(player_creation_node->m_model))
					{
						notify::crash_blocked(sender, "invalid player model (creation node)");
						return true;
					}
					break;
				}
				case RAGE_JOAAT("CSectorDataNode"):
				{
					const auto sector_node = (CSectorDataNode*)(node);
					if (sector_node->m_pos_x == 712 || sector_node->m_pos_y == 712 || sector_node->m_pos_z == 712)
					{
						notify::crash_blocked(sender, "invalid sector position");
						return true;
					}
					break;
				}
			}
		}
		return false;
	}

	bool hooks::can_apply_data(rage::netSyncTree* tree, rage::netObject* object)
	{
		if (tree->m_child_node_count && check_node(tree->m_sync_node, g.m_syncing_player, object))
		{
			return false;
		}

		return g_hooking->get_original<hooks::can_apply_data>()(tree, object);
	}
}