91 lines
3.1 KiB
Rust
91 lines
3.1 KiB
Rust
|
use memflow::{types::Address, mem::MemoryView};
|
||
|
|
use anyhow::Result;
|
||
|
|
use num_traits::FromPrimitive;
|
||
|
|
|
||
|
|
use crate::{dma::CheatCtx, structs::{Vec3, communication::PlayerType}, sdk::{cs2dumper, structs::{BaseEntity, MemoryClass, TeamID}}};
|
||
|
|
|
||
|
|
use super::CPlayerPawn;
|
||
|
|
|
||
|
|
pub struct CPlayerController(Address);
|
||
|
|
|
||
|
|
impl MemoryClass for CPlayerController {
|
||
|
|
fn ptr(&self) -> Address {
|
||
|
|
self.0
|
||
|
|
}
|
||
|
|
|
||
|
|
fn new(ptr: Address) -> Self {
|
||
|
|
Self(ptr)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl BaseEntity for CPlayerController {
|
||
|
|
fn from_index(ctx: &mut CheatCtx, entity_list: Address, index: i32) -> Result<Option<CPlayerController>> {
|
||
|
|
let list_entry = ctx.process.read_addr64(entity_list + 8 * (index >> 9) + 16)?;
|
||
|
|
if list_entry.is_null() && !list_entry.is_valid() {
|
||
|
|
return Ok(None);
|
||
|
|
}
|
||
|
|
|
||
|
|
let player_ptr = ctx.process.read_addr64(list_entry + 120 * (index & 0x1FF))?;
|
||
|
|
if player_ptr.is_null() && !player_ptr.is_valid() {
|
||
|
|
return Ok(None);
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(Some(Self::new(player_ptr)))
|
||
|
|
}
|
||
|
|
|
||
|
|
fn pos(&self, ctx: &mut CheatCtx) -> anyhow::Result<Vec3> {
|
||
|
|
let node = ctx.process.read_addr64(self.0 + cs2dumper::client::C_BaseEntity::m_pGameSceneNode)?;
|
||
|
|
Ok(ctx.process.read(node + cs2dumper::client::CGameSceneNode::m_vecAbsOrigin)?)
|
||
|
|
}
|
||
|
|
|
||
|
|
fn class_name(&self, ctx: &mut CheatCtx) -> anyhow::Result<String> {
|
||
|
|
let entity_identity_ptr = ctx.process.read_addr64(self.0 + cs2dumper::client::CEntityInstance::m_pEntity)?;
|
||
|
|
let class_name_ptr = ctx.process.read_addr64(entity_identity_ptr + cs2dumper::client::CEntityIdentity::m_designerName)?;
|
||
|
|
Ok(ctx.process.read_char_string_n(class_name_ptr, 32)?)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl CPlayerController {
|
||
|
|
pub fn get_pawn(&self, ctx: &mut CheatCtx, entity_list: Address) -> Result<Option<CPlayerPawn>> {
|
||
|
|
let uhandle = ctx.process.read(self.0 + cs2dumper::client::CCSPlayerController::m_hPlayerPawn)?;
|
||
|
|
CPlayerPawn::from_uhandle(ctx, entity_list, uhandle)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Technically something that should be in the BaseEntity trait, but we are only gonna use it on CPlayerController
|
||
|
|
pub fn get_team(&self, ctx: &mut CheatCtx) -> Result<Option<TeamID>> {
|
||
|
|
let team_num: i32 = ctx.process.read(self.0 + cs2dumper::client::C_BaseEntity::m_iTeamNum)?;
|
||
|
|
Ok(TeamID::from_i32(team_num))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn get_player_type(&self, ctx: &mut CheatCtx, local: &CPlayerController) -> Result<Option<PlayerType>> {
|
||
|
|
if self.0 == local.0 {
|
||
|
|
return Ok(Some(PlayerType::Local))
|
||
|
|
}
|
||
|
|
|
||
|
|
let team = {
|
||
|
|
match self.get_team(ctx)? {
|
||
|
|
Some(t) => t,
|
||
|
|
None => { return Ok(None) },
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
let local_team = {
|
||
|
|
match local.get_team(ctx)? {
|
||
|
|
Some(t) => t,
|
||
|
|
None => { return Ok(None) },
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
let player_type = {
|
||
|
|
if team == TeamID::Spectator {
|
||
|
|
PlayerType::Spectator
|
||
|
|
} else if team != local_team {
|
||
|
|
PlayerType::Enemy
|
||
|
|
} else {
|
||
|
|
PlayerType::Team
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
Ok(Some(player_type))
|
||
|
|
}
|
||
|
|
}
|