1 files changed, 200 insertions, 0 deletions
diff --git a/source/sourcemod/scripting/movementapi/stocks.sp b/source/sourcemod/scripting/movementapi/stocks.sp
new file mode 100644
index 0000000..e080efc
--- /dev/null
+++ b/source/sourcemod/scripting/movementapi/stocks.sp
@@ -0,0 +1,200 @@
+stock void GameMove_SetVelocity(Address addr, float velocity[3])
+{
+	if (velocity[0] != velocity[0] || velocity[1] != velocity[1] || velocity[2] != velocity[2])
+	{
+		return;
+	}
+	static int mvOffset;
+	static int velocityOffset;
+	if (!mvOffset)
+	{
+		char buffer[8];
+		if (!gH_GameData.GetKeyValue("CGameMovement::mv", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CGameMovement::mv offset.");
+			return;
+		}
+		mvOffset = StringToInt(buffer);
+		if (!gH_GameData.GetKeyValue("CMoveData::m_vecVelocity", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CMoveData::m_vecVelocity offset.");
+			return;
+		}
+		velocityOffset = StringToInt(buffer);
+	}
+
+	Address mvAddress = view_as<Address>(LoadFromAddress(view_as<Address>(view_as<int>(addr) + mvOffset), NumberType_Int32));
+	// TODO: idk if raw cast works here or not
+	StoreToAddress(view_as<Address>(view_as<int>(mvAddress) + velocityOffset), view_as<int>(velocity[0]), NumberType_Int32);
+	StoreToAddress(view_as<Address>(view_as<int>(mvAddress) + velocityOffset + 4), view_as<int>(velocity[1]), NumberType_Int32);
+	StoreToAddress(view_as<Address>(view_as<int>(mvAddress) + velocityOffset + 8), view_as<int>(velocity[2]), NumberType_Int32);
+}
+
+stock void GameMove_SetOrigin(Address addr, float origin[3])
+{
+	if (origin[0] != origin[0] || origin[1] != origin[1] || origin[2] != origin[2])
+	{
+		return;
+	}
+	static int mvOffset;
+	static int originOffset;
+	if (!mvOffset)
+	{
+		char buffer[8];
+		if (!gH_GameData.GetKeyValue("CGameMovement::mv", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CGameMovement::mv offset.");
+			return;
+		}
+		mvOffset = StringToInt(buffer);
+		if (!gH_GameData.GetKeyValue("CMoveData::m_vecAbsOrigin", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CMoveData::m_vecAbsOrigin offset.");
+			return;
+		}
+		originOffset = StringToInt(buffer);
+	}
+
+	Address mvAddress = view_as<Address>(LoadFromAddress(view_as<Address>(view_as<int>(addr) + mvOffset), NumberType_Int32));
+	// TODO: idk if raw cast works here or not
+	StoreToAddress(view_as<Address>(view_as<int>(mvAddress) + originOffset), view_as<int>(origin[0]), NumberType_Int32);
+	StoreToAddress(view_as<Address>(view_as<int>(mvAddress) + originOffset + 4), view_as<int>(origin[1]), NumberType_Int32);
+	StoreToAddress(view_as<Address>(view_as<int>(mvAddress) + originOffset + 8), view_as<int>(origin[2]), NumberType_Int32);
+}
+
+stock void GameMove_GetVelocity(Address addr, float result[3])
+{
+	static int mvOffset;
+	static int velocityOffset;
+	if (!mvOffset)
+	{
+		char buffer[8];
+		if (!gH_GameData.GetKeyValue("CGameMovement::mv", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CGameMovement::mv offset.");
+			return;
+		}
+		mvOffset = StringToInt(buffer);
+		if (!gH_GameData.GetKeyValue("CMoveData::m_vecVelocity", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CMoveData::m_vecVelocity offset.");
+			return;
+		}
+		velocityOffset = StringToInt(buffer);
+	}
+
+	Address mvAddress = view_as<Address>(LoadFromAddress(view_as<Address>(view_as<int>(addr) + mvOffset), NumberType_Int32));
+	result[0] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + velocityOffset), NumberType_Int32));
+	result[1] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + velocityOffset + 4), NumberType_Int32));
+	result[2] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + velocityOffset + 8), NumberType_Int32));
+}
+
+stock void GameMove_GetOrigin(Address addr, float result[3])
+{
+	static int mvOffset;
+	static int originOffset;
+	if (!mvOffset)
+	{
+		char buffer[8];
+		if (!gH_GameData.GetKeyValue("CGameMovement::mv", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CGameMovement::mv offset.");
+			return;
+		}
+		mvOffset = StringToInt(buffer);
+		if (!gH_GameData.GetKeyValue("CMoveData::m_vecAbsOrigin", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CMoveData::m_vecAbsOrigin offset.");
+			return;
+		}
+		originOffset = StringToInt(buffer);
+	}
+	
+	Address mvAddress = view_as<Address>(LoadFromAddress(view_as<Address>(view_as<int>(addr) + mvOffset), NumberType_Int32));
+	result[0] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + originOffset), NumberType_Int32));
+	result[1] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + originOffset + 4), NumberType_Int32));
+	result[2] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + originOffset + 8), NumberType_Int32));
+}
+
+stock void GameMove_GetEyeAngles(Address addr, float result[3])
+{
+	static int mvOffset;
+	static int viewAngleOffset;
+	if (!mvOffset)
+	{
+		char buffer[8];
+		if (!gH_GameData.GetKeyValue("CGameMovement::mv", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CGameMovement::mv offset.");
+			return;
+		}
+		mvOffset = StringToInt(buffer);
+		if (!gH_GameData.GetKeyValue("CMoveData::m_viewAngleOffset", buffer, sizeof(buffer)))
+		{
+			ThrowError("Failed to get CMoveData::m_viewAngleOffset offset.");
+			return;
+		}
+		viewAngleOffset = StringToInt(buffer);
+	}
+	
+	Address mvAddress = view_as<Address>(LoadFromAddress(view_as<Address>(view_as<int>(addr) + mvOffset), NumberType_Int32));
+	result[0] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + viewAngleOffset), NumberType_Int32));
+	result[1] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + viewAngleOffset + 4), NumberType_Int32));
+	result[2] = view_as<float>(LoadFromAddress(view_as<Address>(view_as<int>(mvAddress) + viewAngleOffset + 8), NumberType_Int32));
+}
+
+stock int GetEntityFromAddress(Address pEntity) {
+	static int offs_RefEHandle;
+	if (offs_RefEHandle) {
+		return EntRefToEntIndex(LoadFromAddress(pEntity + view_as<Address>(offs_RefEHandle), NumberType_Int32) | (1 << 31));
+	}
+
+	// if we don't have it already, attempt to lookup offset based on SDK information
+	// CWorld is derived from CBaseEntity so it should have both offsets
+	int offs_angRotation = FindDataMapInfo(0, "m_angRotation"),
+			offs_vecViewOffset = FindDataMapInfo(0, "m_vecViewOffset");
+	if (offs_angRotation == -1) {
+		ThrowError("Could not find offset for ((CBaseEntity) CWorld)::m_angRotation");
+	} else if (offs_vecViewOffset == -1) {
+		ThrowError("Could not find offset for ((CBaseEntity) CWorld)::m_vecViewOffset");
+	} else if ((offs_angRotation + 0x0C) != (offs_vecViewOffset - 0x04)) {
+		char game[32];
+		GetGameFolderName(game, sizeof(game));
+		ThrowError("Could not confirm offset of CBaseEntity::m_RefEHandle "
+				... "(incorrect assumption for game '%s'?)", game);
+	}
+	
+	// offset seems right, cache it for the next call
+	offs_RefEHandle = offs_angRotation + 0x0C;
+	return GetEntityFromAddress(pEntity);
+}
+
+stock int GetClientFromGameMovementAddress(Address addr) 
+{
+	char buffer[8];
+	if (!gH_GameData.GetKeyValue("CGameMovement::player", buffer, sizeof(buffer)))
+	{
+		ThrowError("Failed to get CGameMovement::player offset.");
+		return -1;
+	}
+	int offset = StringToInt(buffer);
+	Address playerAddr = view_as<Address>(LoadFromAddress(view_as<Address>(view_as<int>(addr) + offset), NumberType_Int32));
+	return GetEntityFromAddress(playerAddr);
+}
+
+stock void HookGameMovementFunction(DynamicDetour handle, char[] fName, DHookCallback preCallback, DHookCallback postCallback)
+{
+	handle = DynamicDetour.FromConf(gH_GameData, fName);
+	if (!handle)
+	{
+		SetFailState("Failed to find %s config", fName);
+	}
+	if (!handle.Enable(Hook_Pre, preCallback))
+	{
+		SetFailState("Failed to enable detour on %s", fName);
+	}
+	if (!handle.Enable(Hook_Post, postCallback))
+	{
+		SetFailState("Failed to enable detour on %s", fName);
+	}
+}