67 lines
1.7 KiB
Rust
67 lines
1.7 KiB
Rust
use crate::{features, gui::App, util::countdown::Countdown};
|
|
use std::time::{Duration, Instant};
|
|
use sysinfo::System;
|
|
use windows::Win32::{
|
|
Foundation::{CloseHandle, HANDLE, NTSTATUS},
|
|
System::Threading::{OpenProcess, PROCESS_SUSPEND_RESUME},
|
|
};
|
|
|
|
pub const INTERVAL: Duration = Duration::from_secs(10);
|
|
const ENHANCED: &str = "GTA5_Enhanced.exe";
|
|
const LEGACY: &str = "GTA5.exe";
|
|
|
|
#[derive(Debug)]
|
|
pub struct EmptySession {
|
|
pub disabled: bool,
|
|
pub interval: Instant,
|
|
pub countdown: Countdown,
|
|
}
|
|
|
|
impl Default for EmptySession {
|
|
fn default() -> Self {
|
|
Self {
|
|
disabled: false,
|
|
interval: Instant::now(),
|
|
countdown: Countdown::new(features::empty_session::INTERVAL.as_secs()),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[link(name = "ntdll")]
|
|
unsafe extern "system" {
|
|
pub unsafe fn NtSuspendProcess(ProcessHandle: HANDLE) -> NTSTATUS;
|
|
pub unsafe fn NtResumeProcess(ProcessHandle: HANDLE) -> NTSTATUS;
|
|
}
|
|
|
|
fn get_gta_pid(sysinfo: &mut System) -> u32 {
|
|
sysinfo.refresh_all();
|
|
if let Some((pid, _)) = sysinfo
|
|
.processes()
|
|
.iter()
|
|
.find(|(_, p)| p.name() == ENHANCED || p.name() == LEGACY)
|
|
{
|
|
return pid.as_u32();
|
|
};
|
|
u32::MAX
|
|
}
|
|
|
|
pub fn activate(app: &mut App) {
|
|
let pid = get_gta_pid(&mut app.sysinfo);
|
|
if pid == u32::MAX {
|
|
return;
|
|
}
|
|
unsafe {
|
|
app.game_handle = OpenProcess(PROCESS_SUSPEND_RESUME, false, pid).unwrap();
|
|
let _ = NtSuspendProcess(app.game_handle);
|
|
}
|
|
}
|
|
|
|
pub fn deactivate(app: &mut App) {
|
|
unsafe {
|
|
if !app.game_handle.is_invalid() {
|
|
let _ = NtResumeProcess(app.game_handle);
|
|
let _ = CloseHandle(app.game_handle);
|
|
}
|
|
}
|
|
}
|