102 Commits
Author SHA1 Message Date
futile aeb99992bb 0.10.0 2025-11-07 12:26:12 +00:00
futile 8ab6e14560 show git commit hash as monospace 2025-11-07 11:59:57 +00:00
futile b9ebdfdb5c update force close game button width to fit new font 2025-11-07 11:56:26 +00:00
futile 2804812691 replace Ubuntu fonts with Inter 2025-11-07 11:51:38 +00:00
futile 606fb7e191 update readme 2025-11-07 11:37:13 +00:00
futile e2a779b7f5 redo links in about stage a bit 2025-11-07 11:35:56 +00:00
futile a8b04bb72b move colour constants from gui::colours to util::consts::colours 2025-11-07 00:34:42 +00:00
futile c34cf187d5 ensure persistent_state is easier and more infallible to add fields to 2025-11-06 23:55:04 +00:00
futile f07421e75a use newer syntax for cargo instructions in build.rs 2025-11-06 08:53:09 +00:00
futile 8b4a455dc1 make empty_session::activate simply return a bool 2025-11-06 08:45:50 +00:00
futile a1300f8d76 lowkey improve logging 2025-11-06 08:41:53 +00:00
futile 9e623219a5 shorten some paths in main.rs 2025-11-06 05:07:49 +00:00
futile 09e19b6e16 remove hardly useful "windows compilation only" warning 2025-11-06 05:06:57 +00:00
futile 4498c45965 apply clippy suggestion 2025-11-06 04:20:40 +00:00
futile af8a5b1746 persist anti afk toggle 2025-11-06 04:17:03 +00:00
futile 560d9659f4 make git hash embedding more reliable 2025-11-06 03:06:25 +00:00
futile bf086d0442 amend comment 2025-11-06 02:38:13 +00:00
futile a6ea8ad8e2 ignore return of NtResumeProcess in empty_session::deactivate to reduce crashing 2025-11-06 02:37:26 +00:00
futile 4c00d1846b add git commit hash when hovering version info 2025-11-05 00:33:17 +00:00
futile e5e335f4f3 apply clippy suggestion 2025-11-05 00:10:39 +00:00
futile 3ab5987b67 use more unwraps instead of let _ = 2025-11-05 00:08:53 +00:00
futile 2030796337 add description to Cargo.toml after making sure tauri-winres doesn't use it 2025-11-04 23:08:59 +00:00
futile 2edb4f492a get rid of justfile it sucks 2025-11-04 23:08:31 +00:00
futile 114292d912 use tauri-winres instead of winresource & specify some more fields in the windows resource in build.rs 2025-11-04 22:56:03 +00:00
futile 8adc1f49ad add license and author information to Cargo.toml 2025-11-04 20:15:09 +00:00
futile bb27334f57 0.9.0 2025-11-03 07:43:40 +00:00
futile 1e13905885 update codebase for github 2025-11-03 07:13:00 +00:00
futile 8fc73b2992 update readme for github 2025-11-03 06:56:28 +00:00
futile 43b72995dd create github workflow to build and release on tag push 2025-11-03 06:42:22 +00:00
futile d3473d46e3 create github workflow to rebuild and rerelease all existing tags 2025-11-03 05:26:46 +00:00
futile 1363f5b125 use match instead of if for LaunchVersion checks 2025-10-25 09:35:31 +01:00
futile d15b8070ec improve strings in util::system_info 2025-10-11 03:32:39 +01:00
futile efccd6ed50 update readme 2025-09-13 16:58:52 +01:00
futile 5c73ee7bd5 improve (tighten) scoping of unsafe blocks 2025-09-13 16:50:44 +01:00
futile 445b63fa4c small system_info improvements 2025-09-13 16:39:16 +01:00
futile 941868dcc8 use &[c_char] instead of &[i8] in c_char_arr_to_string 2025-09-13 12:02:19 +01:00
futile 762a3b9b26 update readme 2025-09-13 00:34:48 +01:00
futile 0cb211c103 0.8.0 2025-09-12 05:17:47 +01:00
futile 90b32c5c5a this couldn't hurt to be create_dir_all instead of create_dir 2025-09-12 05:12:11 +01:00
futile 71c3c69ea2 a few clippy fixes 2025-09-12 05:04:12 +01:00
futile b553e418f5 reduce clutter in system_info 2025-09-12 01:21:56 +01:00
futile ab026c0bc6 adapt codebase to own, new system info impl 2025-09-12 01:07:10 +01:00
futile 32e21f58a1 create a clone of sysinfo focused on only our needs 2025-09-12 00:52:19 +01:00
futile b0c339cc5f update readme 2025-09-09 14:09:17 +01:00
futile c2584b5d01 reimplement "update checking" in the form of a simple button that goes to latest release 2025-09-09 14:07:46 +01:00
futile 3322f9582f update readme 2025-09-09 03:16:00 +01:00
futile 5348a1d6ee rip out update checking (too unreliable) 2025-09-09 03:15:01 +01:00
futile 05ce68cb47 update readme 2025-09-09 03:03:36 +01:00
futile f63d24ea43 improve reliability and logging when running into game access denied issues 2025-09-09 03:03:32 +01:00
futile 356d5f17e5 me when i don't smoke the drugs that make you write retarded code 2025-09-08 14:32:36 +01:00
futile bc5e80aaca update readme 2025-09-06 21:49:44 +01:00
futile e050231692 update readme 2025-09-06 12:26:38 +01:00
futile 6e33536dfe 0.7.0 2025-09-06 12:11:39 +01:00
futile c36a41f919 remove extra release optimizations to save sanity on compile times 2025-09-06 12:11:20 +01:00
futile 660e7a874f improve reliability of panic hook by moving it to main and always creating its storage path 2025-09-06 12:10:38 +01:00
futile 72abda7352 partial fix some weird colouring that was introduced in egui 0.32 (?) 2025-09-05 17:12:38 +01:00
futile ebf704c6f4 update readme to reflect prior changes 2025-09-05 16:10:47 +01:00
futile 5c5e6d60a1 add option to launch legacy version of the game & fix epic games launch urls 2025-09-05 16:09:24 +01:00
futile 98a6545b27 update dependencies i think 2025-09-05 14:26:27 +01:00
futile a9a965462a random small fixes 2025-09-05 13:24:16 +01:00
futile 81b4b26f96 small grammar fix on non-windows compilation error 2025-09-04 09:44:07 +01:00
futile 06df466d24 switch back to ureq from nyquest for http client
note: even if this raises binary size by a significant amount, it should in theory be more reliable, so it is a worthy sacrifice.
2025-09-04 07:41:54 +01:00
futile b8c444dc5b move theme refresh button to debug menu 2025-09-04 06:21:27 +01:00
futile e848ca8d79 GET OUT 2025-09-04 05:36:00 +01:00
futile 0a831580e2 use reworked logging to log errors in util::meta::default 2025-05-16 10:10:16 +01:00
futile 10e69eaac3 rework (error) logging 2025-05-16 10:03:48 +01:00
futile 9dbfbc4419 rework storage paths a bit 2025-05-13 09:40:45 +01:00
futile ef0b6de6ba improve reliability of util::codeberg::get_latest_release 2025-05-13 09:40:18 +01:00
futile e7470bd694 update readme 2025-05-12 08:00:25 +01:00
futile 936f683385 update readme 2025-05-12 07:50:00 +01:00
futile b3158f71fb 0.6.0 2025-05-10 07:04:59 +01:00
futile 070e477fd4 derive default on some shit instead of manually 2025-05-10 06:57:26 +01:00
futile 7745bba10e improve styling in relation to windows theme 2025-05-10 06:44:58 +01:00
futile 8d7a0d12b9 don't force capture backtrace in panic hook & otherwise improve gui::run 2025-05-04 18:36:32 +01:00
futile b325ebb03c clippy stuff 2025-05-04 18:21:55 +01:00
futile 38c6402a12 split ui of App::show_debug_viewport into App::add_debug_viewport_contents 2025-05-04 18:01:50 +01:00
futile 2d42c0c3c4 improve util::codeberg 2025-05-04 17:52:50 +01:00
futile a423721aee bump nyquest and stop using .json() workaround 2025-05-04 17:52:06 +01:00
futile f3436c8ddc 0.5.4 2025-05-03 13:12:47 +01:00
futile afcf0a87c0 get sysinfo from crates again, instead of git repo 2025-05-03 13:12:14 +01:00
futile 73d610949f use nyquest instead of ureq for http client 2025-05-03 13:10:40 +01:00
futile 7f358d9adb change how module roots are organized 2025-05-01 11:23:27 +01:00
futile 7caa5421f7 eliminate chrono dependency by using unix timestamp in panic logs 2025-04-30 01:36:17 +01:00
futile ecac0d54aa eliminate dirs dependency by checking LOCALAPPDATA envvar instead 2025-04-30 00:53:27 +01:00
futile a84401ca33 fail compilation if compiling outside of windows 2025-04-30 00:48:44 +01:00
futile 1bcd6b2c7d fix duplicate inputs bug in features::anti_afk::send 2025-04-26 05:47:10 +01:00
futile 1dbdc0e4cb remove redundant method to_color32 2025-04-26 04:42:29 +01:00
futile 3610e44a08 create trait UiExt 2025-04-26 04:37:10 +01:00
futile f5fd22d35c redesign if_failed_return_to_unblocked to if_failed_return_to_boolean 2025-04-26 04:23:37 +01:00
futile 022828b802 update readme 2025-04-26 04:07:39 +01:00
futile 262d2d0720 update readme 2025-04-25 17:18:38 +00:00
futile 2326edf97b update readme with a guide 2025-04-24 23:43:52 +01:00
futile 301be82837 update screenshot in readme 2025-04-24 13:54:39 +01:00
futile 0d7dde929d 0.5.3 2025-04-24 13:24:35 +01:00
futile 3ab802fd56 make RED, YELLOW, and GREEN slightly brighter 2025-04-23 14:33:08 +01:00
futile 53030cc5f7 add debug menu feature to change blocked_status of GameNetworking 2025-04-23 14:32:29 +01:00
futile 383224c2fa update readme 2025-04-23 14:31:50 +01:00
futile 575a6a4f1d update readme 2025-04-22 20:03:09 +01:00
futile 17579de277 update readme 2025-04-22 19:57:09 +01:00
futile 00727b9913 add compilation instructions to the repo 2025-04-22 14:34:37 +01:00
futile 1f6fc22525 check for app elevation in Default of Flags 2025-04-21 12:38:20 +01:00
futile 27b09be76e update screenshot in readme 2025-04-20 12:06:57 +01:00
34 changed files with 1468 additions and 1309 deletions
+31
View File
@@ -0,0 +1,31 @@
name: Create release for pushed tag
on:
push:
tags:
- "*"
jobs:
release-on-tag:
runs-on: windows-latest
permissions:
contents: write
steps:
- uses: actions/checkout@v5
- uses: dtolnay/rust-toolchain@stable
- name: Create release
shell: bash
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
TAG: ${{ github.ref_name }}
run: |
if gh release view "$TAG" >/dev/null 2>&1; then
echo "A release already exists for $TAG, skipping"
else
cargo build --release
prev_tag=$(gh release list --limit 100 --json tagName --jq '.[0].tagName')
notes="**Full Changelog**: https://github.com/${{ github.repository }}/compare/${prev_tag}...${TAG}"
gh release create "$TAG" target/release/gta-tools.exe \
--title "$TAG" \
--notes "$notes"
fi
@@ -0,0 +1,43 @@
name: Create releases for existing tags
on:
workflow_dispatch:
jobs:
release-retroactively:
runs-on: windows-latest
permissions:
contents: write
steps:
- name: Checkout repository
uses: actions/checkout@v5
with:
fetch-depth: 0
- name: Install Rust toolchain
uses: dtolnay/rust-toolchain@stable
- name: Get tags and create release for each
shell: bash
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
run: |
mapfile -t tags < <(git tag -l)
echo "${#tags[@]} tags found."
prev_tag=""
for tag in "${tags[@]}"; do
if gh release view "$tag" >/dev/null 2>&1; then
echo "A release already exists for $tag, skipping"
else
echo "Proceeding with tag $tag"
git checkout "$tag"
cargo build --release
if [ -n "$prev_tag" ]; then
notes="**Full Changelog**: https://github.com/${{ github.repository }}/compare/${prev_tag}...${tag}"
else
notes="**Full Changelog**: https://github.com/${{ github.repository }}/commits/${tag}"
fi
gh release create "$tag" target/release/gta-tools.exe \
--title "$tag" \
--notes "$notes"
fi
prev_tag="$tag"
done
+1
View File
@@ -1,2 +1,3 @@
/target /target
.vscode .vscode
.idea
Generated
+706 -914
View File
File diff suppressed because it is too large Load Diff
+20 -22
View File
@@ -1,44 +1,42 @@
[package] [package]
name = "gta-tools" name = "gta-tools"
version = "0.5.2" version = "0.10.0"
edition = "2024" edition = "2024"
license = "Apache-2.0"
authors = ["futile <git@futile.eu>"]
description = "A toolset of convenient things for GTA V Online."
[dependencies] [dependencies]
catppuccin-egui = { version = "5.5.0", default-features = false, features = [ catppuccin-egui = { version = "5.6.0", default-features = false, features = [
"egui31", "egui32",
] } ] }
chrono = { version = "0.4.40", default-features = false, features = ["clock"] } eframe = { version = "0.32.2", default-features = false, features = [
dirs = "6.0.0"
eframe = { version = "0.31.1", default-features = false, features = [
"glow", "glow",
"default_fonts", "default_fonts",
] } ] }
egui_extras = { version = "0.31.1", default-features = false, features = [ egui_extras = { version = "0.32.2", default-features = false, features = [
"image", "image",
"svg", "svg",
] } ] }
image = { version = "0.25.6", default-features = false, features = ["png"] } humantime = "2.3.0"
image = { version = "0.25.8", default-features = false, features = ["png"] }
log = "0.4.28"
open = "5.3.2" open = "5.3.2"
semver = "1.0.26"
serde = { version = "1.0.219", features = ["derive"] } serde = { version = "1.0.219", features = ["derive"] }
serde_json = "1.0.140" serde_json = "1.0.143"
strum = { version = "0.27.1", features = ["derive"] } strum = { version = "0.27.2", features = ["derive"] }
sysinfo = { git = "https://github.com/GuillaumeGomez/sysinfo", rev = "bcbb7b2b389b8800ff457773f48b130e710458fc" } windows = { version = "0.62.0", features = [
ureq = { version = "3.0.11", features = ["json"] } "Win32_NetworkManagement_WindowsFirewall",
windows = { version = "0.61.1", features = [ "Win32_Security",
"Win32_UI_Input_KeyboardAndMouse", "Win32_System_Com",
"Win32_System_Diagnostics_ToolHelp",
"Win32_System_Threading", "Win32_System_Threading",
"Win32_UI_Input_KeyboardAndMouse",
"Win32_UI_Shell", "Win32_UI_Shell",
"Win32_UI_WindowsAndMessaging", "Win32_UI_WindowsAndMessaging",
"Win32_Security",
"Win32_NetworkManagement_WindowsFirewall",
] } ] }
winreg = "0.55.0" winreg = "0.55.0"
[build-dependencies] [build-dependencies]
static_vcruntime = "2.0.0" static_vcruntime = "2.0.0"
winresource = "0.1.20" tauri-winres = "0.3.3"
[profile.release]
lto = true
codegen-units = 1
+66 -1
View File
@@ -1,4 +1,69 @@
# GTA Tools # GTA Tools
A toolset of convenient things for GTA V Online. A toolset of convenient things for GTA V Online.
![](https://i.vgy.me/s6bf1g.png) <picture>
<source srcset="https://i.vgy.me/M4sOHh.png" media="(prefers-color-scheme: dark)">
<img src="https://i.vgy.me/mpO9uc.png">
</picture>
## Installing
**Option 1** — <ins>Download</ins>
Download the latest release [here](https://github.com/elituf/gta-tools/releases/latest/download/gta-tools.exe) and place it somewhere convenient for you, such as Documents. You could then make a shortcut titled "GTA Tools", and pin it to taskbar or Start.
**Option 2** — <ins>Build from source</ins>
You will need the Rust toolchain, which can be obtained [here](https://rustup.rs). Follow the instructions of its installer. Once you have Rust installed, clone this repo and navigate to it. At this point, you should probably `git checkout x.x.x`, where `x.x.x` is the latest tag. You can then run `cargo build --release`. Once you do that, you can use the binary located at `.\target\release\gta-tools.exe` in the same way as **Option 1**.
## Guide
Every feature of GTA Tools is Legacy/Enhanced-agnostic. Some functionality of GTA Tools requires administrator access. If necessary, GTA Tools can either be started as admin manually, or, the user can simply use the <kbd>Elevate</kbd> button to relaunch GTA Tools as admin.
It is recommended to always use an up-to-date version of GTA Tools from [releases](https://github.com/elituf/gta-tools/releases). You can also easily access this repository by going to the <kbd>About</kbd> page of GTA Tools and clicking the GitHub button beside the version number.
#### Game
This section is quite simple.
There is a <kbd>Launch</kbd> feature, which will start your game on the chosen launcher and game version.
There is also a <kbd>Force close game</kbd> feature, which simply kills all game processes. This button requires a second press after the first one for confirmation. This *does not* touch Rockstar Games Launcher or any other processes, only ones named `GTA5_Enhanced.exe` or `GTA5.exe`.
#### Session
This section also has two features.
The first being <kbd>Empty current session</kbd>, which "removes everyone" from your session by suspending your game for 10 seconds, and then resuming it. This method is exceedingly similar to the Resource Monitor method, but made convenient in one button. This can be useful in numerous ways:
* You are stuck loading into a session
* You are stuck loading into an interior
* You are about to get griefed while doing a freemode mission
* You want to quickly get a "new" session
The other feature is <kbd>Anti AFK</kbd>, which, when toggled, prevents you from getting kicked for idling by pressing keys on your keyboard every 60 seconds. At present, those keys are `VK_NUMPAD4` & `VK_NUMPAD6`, due to being a relatively good balance of keys that the game actually registers but don't interrupt gameplay too much if they activate while you're playing. You *must* be tabbed into GTA V for <kbd>Anti AFK</kbd> to work. While toggled, <kbd>Anti AFK</kbd> will only activate when all of the following conditions are true:
* The `Grand Theft Auto V` window is currently focused
* None of the designated keys are currently being pressed[^1]
* The mouse cursor is not currently visible[^2]
#### Network
The last section has a single feature, <kbd>Game's network access</kbd>, with two buttons, <kbd>Block</kbd> and <kbd>Unblock</kbd>, and a coloured indicator dot. This feature requires administrator, and blocks/unblocks GTA V's entire internet access using the Windows Firewall. The coloured indicator dot has these meanings:
* **Green** - the game is connected to the internet
* **Yellow** - the game could not be found running, and thus was not blocked
* **Red** - the game is blocked from the internet
This feature is primarily useful for **replay glitching**, which is an exploit that allows you to complete any heist/mission finale, get the money for it, and then be able to do it again right away. Here's how you can do a replay glitch:
1. Get to the finale of a heist/contract/mission series, such as **The Contract: Dr. Dre**
2. Complete the finale as normal
3. Right *after* the **HEIST/MISSION PASSED** screen you get when you complete a finale (basically, when your cut is being shown), <kbd>Block</kbd> the game's internet access
4. After a few seconds, you will be shown an error screen and kicked to the main menu
5. At this point, <kbd>Unblock</kbd> the game's internet access and load back into online
6. Finally, spend any amount of money, and you should receive the finale cut, but also be able to go play the finale again immediately
## Issues
- I have noticed that on my current Windows 10 install, when not elevated (administrator), the <kbd>Force close game</kbd> and <kbd>Empty current session</kbd> features can fail due to being denied access. This was tested by other people and is not guaranteed to happen. If this does happen to you, I recommend using GTA Tools always in elevated mode. Check "Always start elevated" in the Settings tab.
[^1]: Otherwise, if you are, for example, using those keys to fly a plane, <kbd>Anti AFK</kbd> activating could cancel the input.
[^2]: Specifically, the Windows mouse cursor, not the in-game mouse cursor. This check is mostly used to prevent typing while you are in the Rockstar overlay.
Binary file not shown.
Binary file not shown.
Binary file not shown.
+17 -1
View File
@@ -1,7 +1,23 @@
fn main() { fn main() {
static_vcruntime::metabuild(); static_vcruntime::metabuild();
winresource::WindowsResource::new() tauri_winres::WindowsResource::new()
.set("FileDescription", "GTA Tools")
.set("ProductName", "GTA Tools")
.set("LegalCopyright", "futile <git@futile.eu>")
.set_language(0x0009)
.set_icon("assets/icon.ico") .set_icon("assets/icon.ico")
.compile() .compile()
.unwrap(); .unwrap();
embed_latest_git_hash();
}
fn embed_latest_git_hash() {
let git_rev_parse = std::process::Command::new("git")
.args(["rev-parse", "--short=8", "HEAD"])
.output()
.unwrap();
let git_hash = String::from_utf8(git_rev_parse.stdout).unwrap();
println!("cargo::rustc-env=LATEST_GIT_COMMIT_HASH={git_hash}");
println!("cargo::rerun-if-changed=.git/refs/heads/main");
println!("cargo::rerun-if-changed=.git/HEAD");
} }
-6
View File
@@ -1,6 +0,0 @@
set windows-shell := ["pwsh", "-NoLogo", "-NoProfileLoadTime", "-Command"]
install:
cargo build --release
cp .\target\release\gta-tools.exe ~\.cargo\bin
cp .\target\release\gta-tools.exe ~\Documents
+2 -2
View File
@@ -42,7 +42,7 @@ impl AntiAfk {
fn send(vk_codes: &[VIRTUAL_KEY]) { fn send(vk_codes: &[VIRTUAL_KEY]) {
let mut inputs = Vec::new(); let mut inputs = Vec::new();
for &vk_code in vk_codes { for &vk_code in vk_codes {
let scan_code = unsafe { MapVirtualKeyW(u32::from(vk_code.0), MAPVK_VK_TO_VSC) as u16 }; let scan_code = unsafe { MapVirtualKeyW(u32::from(vk_code.0), MAPVK_VK_TO_VSC) } as u16;
for event in [KEYBD_EVENT_FLAGS(0), KEYEVENTF_KEYUP] { for event in [KEYBD_EVENT_FLAGS(0), KEYEVENTF_KEYUP] {
let mut input = INPUT { let mut input = INPUT {
r#type: INPUT_KEYBOARD, r#type: INPUT_KEYBOARD,
@@ -57,6 +57,6 @@ fn send(vk_codes: &[VIRTUAL_KEY]) {
}; };
inputs.push(input); inputs.push(input);
} }
unsafe { SendInput(&inputs, size_of::<INPUT>() as i32) };
} }
unsafe { SendInput(&inputs, size_of::<INPUT>() as i32) };
} }
+22 -23
View File
@@ -1,11 +1,11 @@
use crate::util::{ use crate::util::{
consts::game::{EXE_ENHANCED, EXE_LEGACY}, consts::game::{EXE_ENHANCED, EXE_LEGACY},
countdown::Countdown, countdown::Countdown,
system_info::SystemInfo,
}; };
use std::time::{Duration, Instant}; use std::time::{Duration, Instant};
use sysinfo::System;
use windows::Win32::{ use windows::Win32::{
Foundation::{CloseHandle, HANDLE, NTSTATUS}, Foundation::{HANDLE, NTSTATUS},
System::Threading::{OpenProcess, PROCESS_SUSPEND_RESUME}, System::Threading::{OpenProcess, PROCESS_SUSPEND_RESUME},
}; };
@@ -48,34 +48,33 @@ unsafe extern "system" {
unsafe fn NtResumeProcess(ProcessHandle: HANDLE) -> NTSTATUS; unsafe fn NtResumeProcess(ProcessHandle: HANDLE) -> NTSTATUS;
} }
fn get_gta_pid(sysinfo: &mut System) -> u32 { fn get_gta_pid(system_info: &mut SystemInfo) -> Option<u32> {
sysinfo.refresh_all(); system_info.refresh();
if let Some((pid, _)) = sysinfo system_info
.processes() .processes()
.iter() .iter()
.find(|(_, p)| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY) .find(|p| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY)
{ .map(|p| p.pid())
return pid.as_u32();
}
u32::MAX
} }
pub fn activate(game_handle: &mut HANDLE, sysinfo: &mut System) { pub fn activate(game_handle: &mut HANDLE, system_info: &mut SystemInfo) -> bool {
let pid = get_gta_pid(sysinfo); let Some(pid) = get_gta_pid(system_info) else {
if pid == u32::MAX { return false;
return; };
} match unsafe { OpenProcess(PROCESS_SUSPEND_RESUME, false, pid) } {
unsafe { Ok(handle) => *game_handle = handle,
*game_handle = OpenProcess(PROCESS_SUSPEND_RESUME, false, pid).unwrap(); Err(why) => {
let _ = NtSuspendProcess(*game_handle); log::error!("failed to suspend game for empty session:\n{why}");
return false;
}
} }
unsafe { NtSuspendProcess(*game_handle) }.unwrap();
true
} }
pub fn deactivate(game_handle: &mut HANDLE) { pub fn deactivate(game_handle: &mut HANDLE) {
unsafe { if !game_handle.is_invalid() {
if !game_handle.is_invalid() { // ignoring the return because this function behaves very weirdly
let _ = NtResumeProcess(*game_handle); let _ = unsafe { NtResumeProcess(*game_handle) };
let _ = CloseHandle(*game_handle);
}
} }
} }
+16 -12
View File
@@ -1,6 +1,8 @@
use crate::util::consts::game::{EXE_ENHANCED, EXE_LEGACY}; use crate::util::{
consts::game::{EXE_ENHANCED, EXE_LEGACY},
system_info::SystemInfo,
};
use std::time::{Duration, Instant}; use std::time::{Duration, Instant};
use sysinfo::System;
const INTERVAL: Duration = Duration::from_secs(3); const INTERVAL: Duration = Duration::from_secs(3);
@@ -24,7 +26,7 @@ impl Default for ForceClose {
} }
impl ForceClose { impl ForceClose {
pub fn prompt(&mut self, force_close_button_clicked: bool, sysinfo: &mut System) { pub fn prompt(&mut self, force_close_button_clicked: bool, system_info: &mut SystemInfo) {
if force_close_button_clicked && !self.counting { if force_close_button_clicked && !self.counting {
self.button_text = "Are you sure?".to_owned(); self.button_text = "Are you sure?".to_owned();
self.timer = Instant::now(); self.timer = Instant::now();
@@ -34,7 +36,7 @@ impl ForceClose {
if self.counting && self.timer.elapsed() >= INTERVAL { if self.counting && self.timer.elapsed() >= INTERVAL {
self.reset(); self.reset();
} else if force_close_button_clicked && !self.current_frame { } else if force_close_button_clicked && !self.current_frame {
activate(sysinfo); activate(system_info);
self.reset(); self.reset();
} }
self.finish_current_frame(); self.finish_current_frame();
@@ -44,21 +46,23 @@ impl ForceClose {
*self = Self::default(); *self = Self::default();
} }
fn finish_current_frame(&mut self) { const fn finish_current_frame(&mut self) {
if self.current_frame { if self.current_frame {
self.current_frame = false; self.current_frame = false;
} }
} }
} }
fn activate(sysinfo: &mut System) { fn activate(system_info: &mut SystemInfo) {
sysinfo.refresh_all(); system_info.refresh();
sysinfo system_info
.processes() .processes()
.iter() .iter()
.filter(|(_, p)| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY) .filter(|p| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY)
.for_each(|(_, p)| { .for_each(|p| {
p.kill(); if !p.kill() {
log::error!("failed to force close game, probably due to access denied");
}
}); });
sysinfo.refresh_all(); system_info.refresh();
} }
+37 -42
View File
@@ -1,9 +1,12 @@
use crate::util::consts::game::{EXE_ENHANCED, EXE_LEGACY}; use crate::util::{
consts::game::{EXE_ENHANCED, EXE_LEGACY},
system_info::SystemInfo,
};
use std::{ use std::{
path::Path, path::Path,
time::{Duration, Instant}, time::{Duration, Instant},
}; };
use sysinfo::System; use strum::{Display, EnumIter};
use windows::{ use windows::{
Win32::{ Win32::{
NetworkManagement::WindowsFirewall::{ NetworkManagement::WindowsFirewall::{
@@ -23,7 +26,7 @@ const FILTER_NAME_OUT: &str = "[GTA Tools] Block all outbound traffic for GTA V"
const INTERVAL: Duration = Duration::from_secs(3); const INTERVAL: Duration = Duration::from_secs(3);
#[derive(Clone, Copy, Debug, PartialEq, Eq)] #[derive(Clone, Copy, Debug, Display, PartialEq, Eq, EnumIter)]
pub enum BlockedStatus { pub enum BlockedStatus {
Blocked, Blocked,
Failed, Failed,
@@ -61,63 +64,59 @@ impl Default for GameNetworking {
impl Drop for GameNetworking { impl Drop for GameNetworking {
fn drop(&mut self) { fn drop(&mut self) {
unsafe { if self.com_initialized {
if self.com_initialized { unsafe { CoUninitialize() };
CoUninitialize();
}
} }
} }
} }
impl GameNetworking { impl GameNetworking {
pub fn block_all(&mut self, sysinfo: &mut System) { pub fn block_all(&mut self, system_info: &mut SystemInfo) {
let Some(exe_path) = get_game_exe_path(sysinfo) else { let Some(exe_path) = get_game_exe_path(system_info) else {
self.blocked_status = BlockedStatus::Failed; self.blocked_status = BlockedStatus::Failed;
return; return;
}; };
let policy: INetFwPolicy2 = let policy: INetFwPolicy2 =
unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER).unwrap() }; unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER) }.unwrap();
let rules = unsafe { policy.Rules().unwrap() }; let rules = unsafe { policy.Rules() }.unwrap();
let exe_path = BSTR::from(exe_path.to_string_lossy().to_string()); let exe_path = BSTR::from(exe_path.to_string_lossy().to_string());
for filter in [ for filter in [
(FILTER_NAME_IN, NET_FW_RULE_DIR_IN), (FILTER_NAME_IN, NET_FW_RULE_DIR_IN),
(FILTER_NAME_OUT, NET_FW_RULE_DIR_OUT), (FILTER_NAME_OUT, NET_FW_RULE_DIR_OUT),
] { ] {
let _ = unsafe { rules.Remove(&BSTR::from(filter.0)) }; unsafe { rules.Remove(&BSTR::from(filter.0)) }.unwrap();
unsafe { let rule: INetFwRule =
let rule: INetFwRule = unsafe { CoCreateInstance(&NetFwRule, None, CLSCTX_INPROC_SERVER) }.unwrap();
CoCreateInstance(&NetFwRule, None, CLSCTX_INPROC_SERVER).unwrap(); unsafe { rule.SetName(&BSTR::from(filter.0)) }.unwrap();
rule.SetName(&BSTR::from(filter.0)).unwrap(); unsafe { rule.SetApplicationName(&exe_path) }.unwrap();
rule.SetApplicationName(&exe_path).unwrap(); unsafe { rule.SetDirection(filter.1) }.unwrap();
rule.SetDirection(filter.1).unwrap(); unsafe { rule.SetEnabled(true.into()) }.unwrap();
rule.SetEnabled(true.into()).unwrap(); unsafe { rule.SetAction(NET_FW_ACTION_BLOCK) }.unwrap();
rule.SetAction(NET_FW_ACTION_BLOCK).unwrap(); unsafe { rule.SetProtocol(NET_FW_IP_PROTOCOL_ANY.0) }.unwrap();
rule.SetProtocol(NET_FW_IP_PROTOCOL_ANY.0).unwrap(); unsafe { rules.Add(&rule) }.unwrap();
rules.Add(&rule).unwrap();
}
} }
self.blocked_status = Self::is_blocked().into(); self.blocked_status = Self::is_blocked().into();
} }
pub fn unblock_all(&mut self) { pub fn unblock_all(&mut self) {
let policy: INetFwPolicy2 = let policy: INetFwPolicy2 =
unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER).unwrap() }; unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER) }.unwrap();
let rules = unsafe { policy.Rules().unwrap() }; let rules = unsafe { policy.Rules() }.unwrap();
unsafe { rules.Remove(&BSTR::from(FILTER_NAME_IN)).unwrap() }; unsafe { rules.Remove(&BSTR::from(FILTER_NAME_IN)) }.unwrap();
unsafe { rules.Remove(&BSTR::from(FILTER_NAME_OUT)).unwrap() }; unsafe { rules.Remove(&BSTR::from(FILTER_NAME_OUT)) }.unwrap();
self.blocked_status = Self::is_blocked().into(); self.blocked_status = Self::is_blocked().into();
} }
fn is_blocked() -> bool { fn is_blocked() -> bool {
let policy: INetFwPolicy2 = let policy: INetFwPolicy2 =
unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER).unwrap() }; unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER) }.unwrap();
let rules = unsafe { policy.Rules().unwrap() }; let rules = unsafe { policy.Rules() }.unwrap();
let in_rule_exists = unsafe { rules.Item(&BSTR::from(FILTER_NAME_IN)).is_ok() }; let in_rule_exists = unsafe { rules.Item(&BSTR::from(FILTER_NAME_IN)) }.is_ok();
let out_rule_exists = unsafe { rules.Item(&BSTR::from(FILTER_NAME_OUT)).is_ok() }; let out_rule_exists = unsafe { rules.Item(&BSTR::from(FILTER_NAME_OUT)) }.is_ok();
in_rule_exists || out_rule_exists in_rule_exists || out_rule_exists
} }
pub fn if_failed_return_to_unblocked(&mut self) { pub fn if_failed_return_to_boolean(&mut self) {
if self.blocked_status == BlockedStatus::Failed && !self.counting { if self.blocked_status == BlockedStatus::Failed && !self.counting {
self.counting = true; self.counting = true;
self.timer = Instant::now(); self.timer = Instant::now();
@@ -127,20 +126,16 @@ impl GameNetworking {
&& self.timer.elapsed() >= INTERVAL && self.timer.elapsed() >= INTERVAL
{ {
self.counting = false; self.counting = false;
self.blocked_status = BlockedStatus::Unblocked; self.blocked_status = Self::is_blocked().into();
} }
} }
} }
fn get_game_exe_path(sysinfo: &mut System) -> Option<&Path> { fn get_game_exe_path(system_info: &mut SystemInfo) -> Option<&Path> {
sysinfo.refresh_all(); system_info.refresh();
if let Some((_, process)) = sysinfo system_info
.processes() .processes()
.iter() .iter()
.find(|(_, p)| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY) .find(|p| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY)
{ .and_then(|p| p.exe())
process.exe()
} else {
None
}
} }
+22 -8
View File
@@ -1,3 +1,4 @@
use crate::gui::settings::LaunchVersion;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::{path::PathBuf, process::Command}; use std::{path::PathBuf, process::Command};
use strum::{Display, EnumIter}; use strum::{Display, EnumIter};
@@ -18,16 +19,22 @@ pub struct Launch {
pub selected: Platform, pub selected: Platform,
} }
pub fn launch(platform: &Platform) { pub fn launch(platform: &Platform, version: &LaunchVersion) {
match platform { match platform {
Platform::Steam => { Platform::Steam => {
let _ = open::that_detached("steam://run/3240220"); let steam_url = match version {
LaunchVersion::Enhanced => "steam://run/3240220",
LaunchVersion::Legacy => "steam://run/271590",
};
open::that_detached(steam_url).unwrap();
} }
Platform::Rockstar => { Platform::Rockstar => {
let hklm = RegKey::predef(HKEY_LOCAL_MACHINE); let hklm = RegKey::predef(HKEY_LOCAL_MACHINE);
let Ok(gta_v_enhanced) = let rockstar_url = match version {
hklm.open_subkey(r"SOFTWARE\WOW6432Node\Rockstar Games\GTAV Enhanced") LaunchVersion::Enhanced => r"SOFTWARE\WOW6432Node\Rockstar Games\GTAV Enhanced",
else { LaunchVersion::Legacy => r"SOFTWARE\WOW6432Node\Rockstar Games\Grand Theft Auto V",
};
let Ok(gta_v_enhanced) = hklm.open_subkey(rockstar_url) else {
return; return;
}; };
let Ok(install_folder): Result<String, std::io::Error> = let Ok(install_folder): Result<String, std::io::Error> =
@@ -37,12 +44,19 @@ pub fn launch(platform: &Platform) {
}; };
let mut play_gtav_path = PathBuf::from(install_folder); let mut play_gtav_path = PathBuf::from(install_folder);
play_gtav_path.push("PlayGTAV.exe"); play_gtav_path.push("PlayGTAV.exe");
// ignoring the return because if it errors that means GTA isn't installed via Rockstar
let _ = Command::new(play_gtav_path).spawn(); let _ = Command::new(play_gtav_path).spawn();
} }
Platform::Epic => { Platform::Epic => {
let _ = open::that_detached( let epic_url = match version {
"com.epicgames.launcher://apps/331226ba7c944720baa99103cb1fe80c?action=launch&silent=true", LaunchVersion::Enhanced => {
); "com.epicgames.launcher://apps/8769e24080ea413b8ebca3f1b8c50951?action=launch&silent=true"
}
LaunchVersion::Legacy => {
"com.epicgames.launcher://apps/9d2d0eb64d5c44529cece33fe2a46482?action=launch&silent=true"
}
};
open::that_detached(epic_url).unwrap();
} }
} }
} }
+2 -2
View File
@@ -1,6 +1,6 @@
mod app; pub mod app;
mod colours;
mod debug; mod debug;
pub mod run; pub mod run;
pub mod settings; pub mod settings;
mod tools; mod tools;
mod ui_ext;
+65 -43
View File
@@ -1,7 +1,12 @@
use crate::{ use crate::{
features, features,
gui::{settings::Settings, tools}, gui::{settings::Settings, tools, ui_ext::UiExt},
util::{consts::game::WINDOW_TITLE, meta::Meta, persistent_state::PersistentState, win}, util::{
consts::{colours, game::WINDOW_TITLE},
persistent_state::PersistentState,
system_info::SystemInfo,
win,
},
}; };
use eframe::egui; use eframe::egui;
use std::time::{Duration, Instant}; use std::time::{Duration, Instant};
@@ -17,25 +22,34 @@ enum Stage {
About, About,
} }
#[derive(Debug, Default)] #[derive(Debug)]
pub struct Flags { pub struct Flags {
pub elevated: bool, pub elevated: bool,
pub debug: bool, pub debug: bool,
closing: bool, closing: bool,
} }
impl Default for Flags {
fn default() -> Self {
Self {
elevated: win::is_elevated(),
debug: false,
closing: false,
}
}
}
#[derive(Debug, Default)] #[derive(Debug, Default)]
pub struct App { pub struct App {
pub meta: Meta,
pub settings: Settings, pub settings: Settings,
stage: Stage, stage: Stage,
pub flags: Flags, pub flags: Flags,
pub sysinfo: sysinfo::System, pub system_info: SystemInfo,
game_handle: windows::Win32::Foundation::HANDLE, game_handle: windows::Win32::Foundation::HANDLE,
pub anti_afk: features::anti_afk::AntiAfk, pub anti_afk: features::anti_afk::AntiAfk,
empty_session: features::empty_session::EmptySession, empty_session: features::empty_session::EmptySession,
force_close: features::force_close::ForceClose, force_close: features::force_close::ForceClose,
game_networking: features::game_networking::GameNetworking, pub game_networking: features::game_networking::GameNetworking,
pub launch: features::launch::Launch, pub launch: features::launch::Launch,
} }
@@ -47,7 +61,7 @@ impl eframe::App for App {
.exact_height(25.0) .exact_height(25.0)
.show(ctx, |ui| { .show(ctx, |ui| {
ui.with_layout(egui::Layout::left_to_right(egui::Align::Center), |ui| { ui.with_layout(egui::Layout::left_to_right(egui::Align::Center), |ui| {
tools::build_menu::<Stage>(ui, &mut self.stage); ui.build_menu(&mut self.stage);
ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| { ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
let button = ui let button = ui
.add_enabled(!self.flags.elevated, egui::Button::new("Elevate")) .add_enabled(!self.flags.elevated, egui::Button::new("Elevate"))
@@ -82,34 +96,38 @@ impl eframe::App for App {
impl App { impl App {
fn show_game_section(&mut self, _ctx: &egui::Context, ui: &mut egui::Ui) { fn show_game_section(&mut self, _ctx: &egui::Context, ui: &mut egui::Ui) {
tools::header(ui, "Game"); ui.header("Game");
ui.horizontal(|ui| { ui.horizontal(|ui| {
if ui.button("Launch").clicked() { if ui.button("Launch").clicked() {
features::launch::launch(&self.launch.selected); features::launch::launch(&self.launch.selected, &self.settings.launch_version);
} }
egui::ComboBox::from_id_salt("Launch") egui::ComboBox::from_id_salt("Launch")
.selected_text(self.launch.selected.to_string()) .selected_text(self.launch.selected.to_string())
.width(120.0) .width(120.0)
.show_ui(ui, |ui| { .show_ui(ui, |ui| {
tools::build_menu(ui, &mut self.launch.selected); ui.build_menu(&mut self.launch.selected);
}); });
}); });
let force_close_button = ui.add_sized( let force_close_button = ui.add_sized(
[104.0, 0.0], [108.0, 0.0],
egui::Button::new(&self.force_close.button_text), egui::Button::new(&self.force_close.button_text),
); );
self.force_close self.force_close
.prompt(force_close_button.clicked(), &mut self.sysinfo); .prompt(force_close_button.clicked(), &mut self.system_info);
} }
fn show_session_section(&mut self, _ctx: &egui::Context, ui: &mut egui::Ui) { fn show_session_section(&mut self, _ctx: &egui::Context, ui: &mut egui::Ui) {
tools::header(ui, "Session"); ui.header("Session");
ui.add_enabled_ui(!self.empty_session.disabled, |ui| { ui.add_enabled_ui(!self.empty_session.disabled, |ui| {
ui.horizontal(|ui| { ui.horizontal(|ui| {
if ui.button("Empty current session").clicked() { if ui.button("Empty current session").clicked()
&& features::empty_session::activate(
&mut self.game_handle,
&mut self.system_info,
)
{
self.empty_session.interval = Instant::now(); self.empty_session.interval = Instant::now();
self.empty_session.disabled = true; self.empty_session.disabled = true;
features::empty_session::activate(&mut self.game_handle, &mut self.sysinfo);
} }
ui.label(&self.empty_session.countdown.i_string); ui.label(&self.empty_session.countdown.i_string);
}); });
@@ -134,7 +152,7 @@ impl App {
} }
fn show_network_section(&mut self, _ctx: &egui::Context, ui: &mut egui::Ui) { fn show_network_section(&mut self, _ctx: &egui::Context, ui: &mut egui::Ui) {
tools::header(ui, "Network"); ui.header("Network");
egui::Frame::new() egui::Frame::new()
.outer_margin(egui::vec2(0.0, -2.0)) .outer_margin(egui::vec2(0.0, -2.0))
.show(ui, |ui| { .show(ui, |ui| {
@@ -142,13 +160,11 @@ impl App {
let label = ui.horizontal(|ui| { let label = ui.horizontal(|ui| {
let label = ui.label("Game's network access"); let label = ui.label("Game's network access");
ui.add_space(1.0); ui.add_space(1.0);
ui.add( ui.create_indicator_dot(self.game_networking.blocked_status)
egui::Image::new(egui::include_image!("../../assets/circle.svg")) .on_hover_text(
.max_size([4.0, 4.0].into()) "This turns yellow if GTA Tools\ncannot find your game.",
.tint(self.game_networking.blocked_status.to_color32()), );
) self.game_networking.if_failed_return_to_boolean();
.on_hover_text("This turns yellow if GTA Tools\ncannot find your game.");
self.game_networking.if_failed_return_to_unblocked();
label label
}); });
ui.horizontal(|ui| { ui.horizontal(|ui| {
@@ -159,7 +175,7 @@ impl App {
.add_sized([button_width, 18.0], egui::Button::new("Block")) .add_sized([button_width, 18.0], egui::Button::new("Block"))
.clicked() .clicked()
{ {
self.game_networking.block_all(&mut self.sysinfo); self.game_networking.block_all(&mut self.system_info);
} }
if ui if ui
.add_sized([button_width, 18.0], egui::Button::new("Unblock")) .add_sized([button_width, 18.0], egui::Button::new("Unblock"))
@@ -182,13 +198,21 @@ impl App {
} }
fn show_settings_stage(&mut self, ctx: &egui::Context, ui: &mut egui::Ui) { fn show_settings_stage(&mut self, ctx: &egui::Context, ui: &mut egui::Ui) {
ui.horizontal(|ui| {
ui.label("Launch version");
egui::ComboBox::from_id_salt("Launch version")
.selected_text(self.settings.launch_version.to_string())
.show_ui(ui, |ui| {
ui.build_menu(&mut self.settings.launch_version);
});
});
ui.horizontal(|ui| { ui.horizontal(|ui| {
let selection = self.settings.theme; let selection = self.settings.theme;
ui.label("Theme"); ui.label("Theme");
egui::ComboBox::from_id_salt("Theme") egui::ComboBox::from_id_salt("Theme")
.selected_text(self.settings.theme.to_string()) .selected_text(self.settings.theme.to_string())
.show_ui(ui, |ui| { .show_ui(ui, |ui| {
tools::build_menu(ui, &mut self.settings.theme); ui.build_menu(&mut self.settings.theme);
}); });
if selection != self.settings.theme { if selection != self.settings.theme {
catppuccin_egui::set_theme(ctx, self.settings.theme.into()); catppuccin_egui::set_theme(ctx, self.settings.theme.into());
@@ -203,28 +227,25 @@ impl App {
ui.horizontal(|ui| { ui.horizontal(|ui| {
ui.spacing_mut().item_spacing.x = 0.0; ui.spacing_mut().item_spacing.x = 0.0;
ui.label("with "); ui.label("with ");
ui.hyperlink_to("", "https://codeberg.org/futile/gta-tools"); ui.scope(|ui| {
ui.style_mut().visuals.override_text_color = Some(colours::RED);
ui.label("");
});
ui.label(" from "); ui.label(" from ");
ui.hyperlink_to("futile", "http://futile.eu"); ui.hyperlink_to("futile", "https://futile.eu");
}); });
ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| { ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
ui.label(format!( ui.style_mut().spacing.button_padding = egui::Vec2::new(4.0, 0.0);
"v{} {}", if ui.button("").on_hover_text("View source code").clicked() {
self.meta.current_version, open::that("https://github.com/elituf/gta-tools").unwrap();
if cfg!(debug_assertions) { "(dev)" } else { "" }
));
let button = ui.add_enabled_ui(self.meta.newer_version_available, |ui| {
ui.style_mut().spacing.button_padding = egui::Vec2::new(3.0, 0.0);
ui.button("")
.on_disabled_hover_text("Already up to date.")
.on_hover_text(format!(
"New version available! ({})",
self.meta.latest_release.version
))
});
if button.inner.clicked() {
open::that(&self.meta.latest_release.download_url).unwrap();
} }
if cfg!(debug_assertions) {
ui.label("(dev)");
}
ui.label(format!("v{}", env!("CARGO_PKG_VERSION")))
.on_hover_text(
egui::RichText::new(env!("LATEST_GIT_COMMIT_HASH")).monospace(),
);
}); });
}); });
ui.add(egui::Image::new(egui::include_image!( ui.add(egui::Image::new(egui::include_image!(
@@ -239,6 +260,7 @@ impl Drop for App {
// save any persistent state to config file // save any persistent state to config file
PersistentState { PersistentState {
launcher: self.launch.selected, launcher: self.launch.selected,
anti_afk_enabled: self.anti_afk.enabled,
settings: self.settings.clone(), settings: self.settings.clone(),
} }
.set(); .set();
-22
View File
@@ -1,22 +0,0 @@
use crate::features::game_networking::BlockedStatus;
use eframe::egui;
pub const RED: egui::Color32 = egui::Color32::from_rgb(255, 128, 128);
pub const YELLOW: egui::Color32 = egui::Color32::from_rgb(255, 255, 128);
pub const GREEN: egui::Color32 = egui::Color32::from_rgb(128, 255, 128);
impl From<BlockedStatus> for egui::Color32 {
fn from(value: BlockedStatus) -> Self {
match value {
BlockedStatus::Blocked => RED,
BlockedStatus::Failed => YELLOW,
BlockedStatus::Unblocked => GREEN,
}
}
}
impl BlockedStatus {
pub fn to_color32(self) -> egui::Color32 {
self.into()
}
}
+54 -49
View File
@@ -2,15 +2,67 @@ use crate::{
gui::{ gui::{
app::{App, WINDOW_SIZE}, app::{App, WINDOW_SIZE},
tools, tools,
ui_ext::UiExt,
}, },
util::consts::{ util::consts::{
APP_STORAGE_PATH,
game::{EXE_ENHANCED, EXE_LEGACY}, game::{EXE_ENHANCED, EXE_LEGACY},
path,
}, },
}; };
use eframe::egui; use eframe::egui;
impl App { impl App {
fn add_debug_viewport_contents(&mut self, ui: &mut egui::Ui) {
ui.collapsing("misc", |ui| {
if ui.button("open storage path").clicked() {
open::that_detached(path::APP_STORAGE.as_path()).unwrap();
}
ui.scope(|ui| {
use windows::Win32::UI::WindowsAndMessaging::{
GetForegroundWindow, GetWindowTextW,
};
let current_title = {
let mut buffer = [0; 512];
let hwnd = unsafe { GetForegroundWindow() };
let length = unsafe { GetWindowTextW(hwnd, &mut buffer) };
String::from_utf16_lossy(&buffer[..length as usize])
};
ui.label(format!("focused: \"{current_title}\""));
});
ui.horizontal(|ui| {
ui.label("blocked_status");
egui::ComboBox::from_id_salt("blocked_status")
.selected_text(self.game_networking.blocked_status.to_string())
.show_ui(ui, |ui| {
ui.build_menu(&mut self.game_networking.blocked_status);
});
});
if ui.add(egui::Button::new("force refresh theme")).clicked() {
catppuccin_egui::set_theme(ui.ctx(), self.settings.theme.into());
}
});
ui.collapsing("anti afk", |ui| {
ui.label(format!(
"timer: {}",
self.anti_afk.interval.elapsed().as_secs()
));
ui.label(format!("can activate: {}", self.anti_afk.can_activate()));
});
ui.collapsing("system info", |ui| {
if ui.button("refresh").clicked() {
self.system_info.refresh();
}
let pid = self
.system_info
.processes()
.iter()
.find(|p| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY)
.map_or_else(|| "no pid found!".to_owned(), |p| p.pid().to_string());
ui.label(format!("gta pid: {pid}"));
});
ui.collapsing("app state", |ui| ui.label(format!("{self:#?}")));
}
pub fn show_debug_viewport(&mut self, ctx: &egui::Context) { pub fn show_debug_viewport(&mut self, ctx: &egui::Context) {
let main = ctx.input(|i| i.viewport().outer_rect.unwrap_or(egui::Rect::EVERYTHING)); let main = ctx.input(|i| i.viewport().outer_rect.unwrap_or(egui::Rect::EVERYTHING));
let builder = egui::ViewportBuilder::default() let builder = egui::ViewportBuilder::default()
@@ -30,54 +82,7 @@ impl App {
egui::CentralPanel::default().show(ctx, |ui| { egui::CentralPanel::default().show(ctx, |ui| {
egui::ScrollArea::both() egui::ScrollArea::both()
.auto_shrink([false, true]) .auto_shrink([false, true])
.show(ui, |ui| { .show(ui, |ui| self.add_debug_viewport_contents(ui));
ui.collapsing("misc", |ui| {
if ui.button("open storage path").clicked() {
open::that_detached(APP_STORAGE_PATH.as_path()).unwrap();
}
ui.checkbox(
&mut self.meta.newer_version_available,
"spoof new version available",
)
.on_hover_text("(this could already be checked if\nthere actually IS a new version)");
ui.scope(|ui| {
use windows::Win32::UI::WindowsAndMessaging::{GetForegroundWindow, GetWindowTextW};
let mut buffer = [0; 512];
let current_title = unsafe {
let hwnd = GetForegroundWindow();
let length = GetWindowTextW(hwnd, &mut buffer);
String::from_utf16_lossy(&buffer[..length as usize])
};
ui.label(format!("focused: \"{current_title}\""));
});
});
ui.collapsing("anti afk", |ui| {
ui.label(format!(
"timer: {}",
self.anti_afk.interval.elapsed().as_secs()
));
ui.label(format!(
"can activate: {}",
self.anti_afk.can_activate()
));
});
ui.collapsing("sysinfo", |ui| {
if ui.button("refresh all").clicked() {
self.sysinfo.refresh_all();
}
let pid = self
.sysinfo
.processes()
.iter()
.find(|(_, p)| p.name() == EXE_ENHANCED || p.name() == EXE_LEGACY)
.map_or_else(
|| "no pid found!".to_owned(),
|(pid, _)| pid.as_u32().to_string(),
);
ui.label(format!("gta pid: {pid}"));
});
ui.collapsing("app state", |ui| ui.label(format!("{self:#?}")));
});
}); });
}, },
); );
+25 -32
View File
@@ -3,62 +3,55 @@ use crate::{
app::{App, WINDOW_SIZE}, app::{App, WINDOW_SIZE},
tools, tools,
}, },
util::{self, consts::APP_STORAGE_PATH, persistent_state::PersistentState}, util::{persistent_state::PersistentState, win},
}; };
use eframe::egui; use eframe::egui;
use std::{fs::File, io::Write};
fn panic_hook(panic_info: &std::panic::PanicHookInfo<'_>) {
let log_path = APP_STORAGE_PATH.join("panic.log");
let mut file = File::options()
.create(true)
.append(true)
.open(log_path)
.unwrap();
let timestamp = chrono::Local::now().to_rfc3339_opts(chrono::SecondsFormat::Secs, false);
let backtrace = std::backtrace::Backtrace::force_capture();
let message = format!("[{timestamp}]\n{panic_info}\nstack backtrace:\n{backtrace}\n");
file.write_all(message.as_bytes()).unwrap();
}
fn app_creator( fn app_creator(
cc: &eframe::CreationContext<'_>, cc: &eframe::CreationContext<'_>,
) -> Result<Box<dyn eframe::App>, Box<dyn std::error::Error + Send + Sync>> { ) -> Result<Box<dyn eframe::App>, Box<dyn std::error::Error + Send + Sync>> {
// use our own panic hook which logs all panics to a file
std::panic::set_hook(Box::new(panic_hook));
// initialize App early to modify some things before returning it // initialize App early to modify some things before returning it
let mut app = Box::<App>::default(); let mut app = Box::new(App::default());
// load previously selected launch platform & settings from persistent state // load previously selected launch platform & settings from persistent state
if let Some(persistent_state) = PersistentState::get() { if let Some(persistent_state) = PersistentState::get() {
app.launch.selected = persistent_state.launcher; persistent_state.apply_to(&mut app);
app.settings = persistent_state.settings;
} }
// check if we're elevated. if not, and the user wants an elevated launch - relaunch elevated // check if we're elevated. if not, and the user wants an elevated launch - relaunch elevated
let elevated = util::win::is_elevated(); if !app.flags.elevated && app.settings.start_elevated {
if app.settings.start_elevated && !elevated { win::elevate(win::ElevationExitMethod::Forced);
util::win::elevate(util::win::ElevationExitMethod::Forced);
} }
app.flags.elevated = elevated; // refresh system info because it initializes with nothing
// refresh sysinfo because it initializes with nothing app.system_info.refresh();
app.sysinfo.refresh_all();
// enable image loading support in egui // enable image loading support in egui
egui_extras::install_image_loaders(&cc.egui_ctx); egui_extras::install_image_loaders(&cc.egui_ctx);
// set our initial theme, from earlier loaded settings // set our initial theme, from earlier loaded settings. we set the egui theme
// to dark here to work around system theme based switching of the egui style
cc.egui_ctx.set_theme(egui::Theme::Dark);
catppuccin_egui::set_theme(&cc.egui_ctx, app.settings.theme.into()); catppuccin_egui::set_theme(&cc.egui_ctx, app.settings.theme.into());
// apply some global styling that we like // apply some global styling that we like
cc.egui_ctx.style_mut(|style| { cc.egui_ctx.all_styles_mut(|style| {
style.spacing.item_spacing = egui::vec2(4.0, 4.0); style.spacing.item_spacing = egui::vec2(4.0, 4.0);
style.interaction.selectable_labels = false; style.interaction.selectable_labels = false;
style.visuals.hyperlink_color = catppuccin_egui::Theme::from(app.settings.theme).text;
}); });
// load any extra fonts that we need // load any extra fonts that we need
let mut fonts = egui::FontDefinitions::default(); let mut fonts = egui::FontDefinitions::default();
fonts.font_data.insert( fonts.font_data.insert(
"Ubuntu-Regular".to_owned(), "Inter 18pt Light".to_owned(),
egui::FontData::from_static(include_bytes!("../../assets/Ubuntu-Regular.ttf")).into(), egui::FontData::from_static(include_bytes!("../../assets/Inter_18pt-Light.ttf")).into(),
);
fonts
.families
.get_mut(&egui::FontFamily::Proportional)
.unwrap()
.insert(0, "Inter 18pt Light".to_owned());
fonts.font_data.insert(
"Inter 18pt Regular".to_owned(),
egui::FontData::from_static(include_bytes!("../../assets/Inter_18pt-Regular.ttf")).into(),
); );
fonts.families.insert( fonts.families.insert(
egui::FontFamily::Name("Ubuntu-Regular".into()), egui::FontFamily::Name("Inter 18pt Regular".into()),
vec!["Ubuntu-Regular".to_owned()], vec!["Inter 18pt Regular".to_owned()],
); );
cc.egui_ctx.set_fonts(fonts); cc.egui_ctx.set_fonts(fonts);
// finally return the App // finally return the App
+21 -11
View File
@@ -1,8 +1,19 @@
use crate::util::win;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use strum::{Display, EnumIter}; use strum::{Display, EnumIter};
#[derive(Clone, Copy, Debug, Display, PartialEq, Eq, Serialize, Deserialize, EnumIter)] #[derive(Clone, Copy, Debug, Default, Display, PartialEq, Eq, Serialize, Deserialize, EnumIter)]
pub enum LaunchVersion {
#[default]
Enhanced,
Legacy,
}
#[derive(Clone, Copy, Debug, Default, Display, PartialEq, Eq, Serialize, Deserialize, EnumIter)]
pub enum Theme { pub enum Theme {
#[default]
#[strum(to_string = "Auto")]
Auto,
#[strum(to_string = "Catppuccin Latte")] #[strum(to_string = "Catppuccin Latte")]
CatppuccinLatte, CatppuccinLatte,
#[strum(to_string = "Catppuccin Frappe")] #[strum(to_string = "Catppuccin Frappe")]
@@ -16,6 +27,13 @@ pub enum Theme {
impl From<Theme> for catppuccin_egui::Theme { impl From<Theme> for catppuccin_egui::Theme {
fn from(val: Theme) -> Self { fn from(val: Theme) -> Self {
match val { match val {
Theme::Auto => {
if win::is_system_theme_dark() {
catppuccin_egui::MOCHA
} else {
catppuccin_egui::LATTE
}
}
Theme::CatppuccinLatte => catppuccin_egui::LATTE, Theme::CatppuccinLatte => catppuccin_egui::LATTE,
Theme::CatppuccinFrappe => catppuccin_egui::FRAPPE, Theme::CatppuccinFrappe => catppuccin_egui::FRAPPE,
Theme::CatppuccinMacchiato => catppuccin_egui::MACCHIATO, Theme::CatppuccinMacchiato => catppuccin_egui::MACCHIATO,
@@ -24,17 +42,9 @@ impl From<Theme> for catppuccin_egui::Theme {
} }
} }
#[derive(Clone, Debug, Serialize, Deserialize)] #[derive(Clone, Debug, Default, Serialize, Deserialize)]
pub struct Settings { pub struct Settings {
pub launch_version: LaunchVersion,
pub theme: Theme, pub theme: Theme,
pub start_elevated: bool, pub start_elevated: bool,
} }
impl Default for Settings {
fn default() -> Self {
Self {
theme: Theme::CatppuccinMocha,
start_elevated: false,
}
}
}
-19
View File
@@ -11,25 +11,6 @@ pub fn load_icon() -> egui::IconData {
} }
} }
pub fn header(ui: &mut egui::Ui, text: &str) {
ui.horizontal(|ui| {
ui.label(egui::RichText::new(text).font(egui::FontId::new(
12.5,
egui::FontFamily::Name("Ubuntu-Regular".into()),
)));
ui.add(egui::Separator::default().horizontal());
});
}
pub fn build_menu<E>(ui: &mut egui::Ui, current_value: &mut E)
where
E: strum::IntoEnumIterator + std::fmt::Display + std::cmp::PartialEq + Copy,
{
E::iter().for_each(|variant| {
ui.selectable_value(current_value, variant, variant.to_string());
});
}
pub fn debug_keycombo_pressed(ctx: &egui::Context) -> bool { pub fn debug_keycombo_pressed(ctx: &egui::Context) -> bool {
ctx.input(|i| i.modifiers.all() && i.key_pressed(egui::Key::D)) ctx.input(|i| i.modifiers.all() && i.key_pressed(egui::Key::D))
} }
+38
View File
@@ -0,0 +1,38 @@
use eframe::egui;
pub trait UiExt {
fn header(&mut self, text: &str);
fn build_menu<E>(&mut self, current_value: &mut E)
where
E: strum::IntoEnumIterator + std::fmt::Display + std::cmp::PartialEq + Copy;
fn create_indicator_dot(&mut self, colour: impl Into<egui::Color32>) -> egui::Response;
}
impl UiExt for egui::Ui {
fn header(&mut self, text: &str) {
self.horizontal(|ui| {
ui.label(egui::RichText::new(text).font(egui::FontId::new(
12.5,
egui::FontFamily::Name("Inter 18pt Regular".into()),
)));
ui.add(egui::Separator::default().horizontal());
});
}
fn build_menu<E>(&mut self, current_value: &mut E)
where
E: strum::IntoEnumIterator + std::fmt::Display + std::cmp::PartialEq + Copy,
{
E::iter().for_each(|variant| {
self.selectable_value(current_value, variant, variant.to_string());
});
}
fn create_indicator_dot(&mut self, colour: impl Into<egui::Color32>) -> egui::Response {
self.add(
egui::Image::new(egui::include_image!("../../assets/circle.svg"))
.max_size([4.0, 4.0].into())
.tint(colour),
)
}
}
+18
View File
@@ -4,6 +4,24 @@ mod features;
mod gui; mod gui;
mod util; mod util;
fn init_storage() {
if !util::consts::path::APP_STORAGE.exists() {
std::fs::create_dir_all(util::consts::path::APP_STORAGE.as_path()).unwrap();
}
}
fn panic_hook(panic_info: &std::panic::PanicHookInfo<'_>) {
let backtrace = std::backtrace::Backtrace::capture();
let mut message = format!("{panic_info}");
if backtrace.status() == std::backtrace::BacktraceStatus::Captured {
message += &format!("\nstack backtrace:\n{backtrace}");
}
log::error!("{message}");
}
fn main() { fn main() {
init_storage();
util::logging::Logger::init(log::LevelFilter::Info);
std::panic::set_hook(Box::new(panic_hook));
gui::run::run(); gui::run::run();
} }
+2 -2
View File
@@ -1,6 +1,6 @@
mod codeberg;
pub mod consts; pub mod consts;
pub mod countdown; pub mod countdown;
pub mod meta; pub mod logging;
pub mod persistent_state; pub mod persistent_state;
pub mod system_info;
pub mod win; pub mod win;
-35
View File
@@ -1,35 +0,0 @@
use semver::Version;
const APP_USER_AGENT: &str = concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"));
const CODEBERG_ENDPOINT_ROOT: &str = "https://codeberg.org/api/v1";
#[derive(Debug)]
pub struct Release {
pub version: Version,
pub download_url: String,
}
impl Default for Release {
fn default() -> Self {
Self {
version: Version::parse(env!("CARGO_PKG_VERSION")).unwrap(),
download_url: String::new(),
}
}
}
pub fn get_latest_release() -> Option<Release> {
let request_url = format!("{CODEBERG_ENDPOINT_ROOT}/repos/futile/gta-tools/releases/latest");
let ureq: ureq::Agent = ureq::Agent::config_builder()
.user_agent(APP_USER_AGENT)
.build()
.into();
let mut response = ureq.get(request_url).call().ok()?;
let json = response.body_mut().read_json::<serde_json::Value>().ok()?;
let tag_name = json["tag_name"].as_str()?;
let browser_download_url = json["assets"][0]["browser_download_url"].as_str()?;
Some(Release {
version: Version::parse(tag_name).expect("expected a valid semver pattern"),
download_url: String::from(browser_download_url),
})
}
+28 -4
View File
@@ -1,10 +1,34 @@
use std::{path::PathBuf, sync::LazyLock}; pub mod path {
use std::{env, path::PathBuf, sync::LazyLock};
pub static APP_STORAGE_PATH: LazyLock<PathBuf> = pub static APP_STORAGE: LazyLock<PathBuf> = LazyLock::new(|| {
LazyLock::new(|| dirs::config_local_dir().unwrap().join("GTA Tools")); PathBuf::from(env::var("LOCALAPPDATA").unwrap_or_else(|_| String::from(".")))
.join("GTA Tools")
});
pub static APP_CONFIG: LazyLock<PathBuf> = LazyLock::new(|| APP_STORAGE.join("config.json"));
pub static APP_LOG: LazyLock<PathBuf> = LazyLock::new(|| APP_STORAGE.join("gta-tools.log"));
}
pub mod game { pub mod game {
pub const EXE_ENHANCED: &str = "GTA5_Enhanced.exe"; pub const EXE_ENHANCED: &str = "GTA5_Enhanced.exe";
pub const EXE_LEGACY: &str = "GTA5.exe"; pub const EXE_LEGACY: &str = "GTA5.exe";
pub const WINDOW_TITLE: &str = "Grand Theft Auto V"; pub const WINDOW_TITLE: &str = "Grand Theft Auto V";
} }
pub mod colours {
use crate::features::game_networking::BlockedStatus;
use eframe::egui;
pub const RED: egui::Color32 = egui::Color32::from_rgb(255, 96, 96);
pub const YELLOW: egui::Color32 = egui::Color32::from_rgb(255, 255, 96);
pub const GREEN: egui::Color32 = egui::Color32::from_rgb(96, 255, 96);
impl From<BlockedStatus> for egui::Color32 {
fn from(value: BlockedStatus) -> Self {
match value {
BlockedStatus::Blocked => RED,
BlockedStatus::Failed => YELLOW,
BlockedStatus::Unblocked => GREEN,
}
}
}
}
+64
View File
@@ -0,0 +1,64 @@
use std::{
fs::File,
io::Write,
sync::{LazyLock, Mutex},
time::SystemTime,
};
pub struct Logger {
file: Mutex<File>,
}
static LOGGER: LazyLock<Logger> = LazyLock::new(|| Logger {
file: File::options()
.create(true)
.append(true)
.open(crate::util::consts::path::APP_LOG.as_path())
.unwrap()
.into(),
});
impl log::Log for Logger {
fn enabled(&self, metadata: &log::Metadata) -> bool {
metadata.level() <= log::Level::Info
}
fn flush(&self) {}
fn log(&self, record: &log::Record) {
if self.enabled(record.metadata()) {
self.log_to_stderr(record);
self.log_to_file(record);
}
}
}
impl Logger {
pub fn init(level: log::LevelFilter) {
log::set_logger(&*LOGGER)
.map(|()| log::set_max_level(level))
.unwrap();
}
fn log_to_file(&self, record: &log::Record) {
let mut file = self.file.lock().unwrap();
write!(
file,
"[{}][{}]\n{}\n\n",
humantime::format_rfc3339_seconds(SystemTime::now()),
record.level(),
record.args()
)
.unwrap();
file.flush().unwrap();
}
fn log_to_stderr(&self, record: &log::Record) {
eprintln!(
"[{}][{}]\n{}",
humantime::format_rfc3339_seconds(SystemTime::now()),
record.level(),
record.args()
)
}
}
-25
View File
@@ -1,25 +0,0 @@
use crate::util::codeberg;
use semver::Version;
#[derive(Debug)]
pub struct Meta {
pub current_version: Version,
pub latest_release: codeberg::Release,
pub newer_version_available: bool,
}
impl Default for Meta {
fn default() -> Self {
let current_version = Version::parse(env!("CARGO_PKG_VERSION")).unwrap();
let latest_release = codeberg::get_latest_release().unwrap_or_default();
let newer_version_available = matches!(
&current_version.cmp_precedence(&latest_release.version),
std::cmp::Ordering::Less
);
Self {
current_version,
latest_release,
newer_version_available,
}
}
}
+19 -12
View File
@@ -1,35 +1,42 @@
use crate::{features::launch::Platform, gui::settings::Settings, util::consts::APP_STORAGE_PATH}; use crate::{
features::launch::Platform,
gui::{app, settings::Settings},
util::consts::path,
};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::{ use std::{
fs::{self, File}, fs::{self, File},
io::Write, io::Write,
path::PathBuf,
sync::LazyLock,
}; };
static CONFIG_PATH: LazyLock<PathBuf> = LazyLock::new(|| APP_STORAGE_PATH.join("config.json"));
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
pub struct PersistentState { pub struct PersistentState {
pub launcher: Platform, pub launcher: Platform,
pub anti_afk_enabled: bool,
pub settings: Settings, pub settings: Settings,
} }
impl PersistentState { impl PersistentState {
pub fn get() -> Option<Self> { pub fn get() -> Option<Self> {
fs::read_to_string(CONFIG_PATH.as_path()) fs::read_to_string(path::APP_CONFIG.as_path())
.ok() .ok()
.and_then(|config| serde_json::from_str::<Self>(&config).ok()) .and_then(|config| serde_json::from_str::<Self>(&config).ok())
} }
pub fn set(&self) { pub fn set(&self) {
let config_path = CONFIG_PATH.as_path(); let mut config_file = File::create(path::APP_CONFIG.as_path()).unwrap();
let config_path_parent = config_path.parent().unwrap();
if !config_path_parent.exists() {
fs::create_dir(config_path_parent).unwrap();
}
let mut config_file = File::create(config_path).unwrap();
let json = serde_json::to_string_pretty(&self).unwrap(); let json = serde_json::to_string_pretty(&self).unwrap();
config_file.write_all(json.as_bytes()).unwrap(); config_file.write_all(json.as_bytes()).unwrap();
} }
pub fn apply_to(self, app: &mut app::App) {
let Self {
launcher,
anti_afk_enabled,
settings,
} = self;
app.launch.selected = launcher;
app.anti_afk.enabled = anti_afk_enabled;
app.settings = settings;
}
} }
+117
View File
@@ -0,0 +1,117 @@
use std::{
ffi::{OsStr, OsString},
os::windows::{ffi::OsStringExt, process::CommandExt},
path::{Path, PathBuf},
process::Command,
};
use windows::{
Win32::System::{
Diagnostics::ToolHelp::{
CreateToolhelp32Snapshot, PROCESSENTRY32W, Process32FirstW, Process32NextW,
TH32CS_SNAPPROCESS,
},
Threading::{
CREATE_NO_WINDOW, OpenProcess, PROCESS_NAME_WIN32, PROCESS_QUERY_LIMITED_INFORMATION,
QueryFullProcessImageNameW,
},
},
core::PWSTR,
};
#[derive(Clone, Debug)]
pub struct Process {
pid: u32,
name: OsString,
exe: Option<PathBuf>,
}
impl Process {
pub const fn pid(&self) -> u32 {
self.pid
}
pub fn name(&self) -> &OsStr {
&self.name
}
pub fn exe(&self) -> Option<&Path> {
self.exe.as_deref()
}
pub fn kill(&self) -> bool {
let mut taskkill = Command::new("taskkill.exe");
taskkill.creation_flags(CREATE_NO_WINDOW.0);
taskkill.arg("/F").arg("/PID").arg(self.pid.to_string());
match taskkill.output() {
Ok(output) => output.status.success(),
Err(_) => false,
}
}
}
#[derive(Debug, Default)]
pub struct SystemInfo {
processes: Vec<Process>,
}
impl SystemInfo {
pub fn refresh(&mut self) {
let mut processes = Vec::new();
let snapshot_handle = unsafe { CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0) }.unwrap();
let mut process_entry = PROCESSENTRY32W {
dwSize: size_of::<PROCESSENTRY32W>() as u32,
..Default::default()
};
unsafe { Process32FirstW(snapshot_handle, &raw mut process_entry) }.unwrap();
let exe_full_path = get_exe_full_path(&process_entry);
processes.push(Process {
pid: process_entry.th32ProcessID,
name: wide_array_to_os_string(&process_entry.szExeFile),
exe: exe_full_path,
});
while unsafe { Process32NextW(snapshot_handle, &raw mut process_entry) }.is_ok() {
let exe_full_path = get_exe_full_path(&process_entry);
processes.push(Process {
pid: process_entry.th32ProcessID,
name: wide_array_to_os_string(&process_entry.szExeFile),
exe: exe_full_path,
});
}
self.processes = processes;
}
pub fn processes(&self) -> &[Process] {
&self.processes
}
}
fn get_exe_full_path(process_entry: &PROCESSENTRY32W) -> Option<PathBuf> {
let process_handle_result = unsafe {
OpenProcess(
PROCESS_QUERY_LIMITED_INFORMATION,
false,
process_entry.th32ProcessID,
)
};
process_handle_result.map_or(None, |process_handle| {
let mut exe_name = [0u16; 260];
let mut dw_size = exe_name.len() as u32;
let image_name_result = unsafe {
QueryFullProcessImageNameW(
process_handle,
PROCESS_NAME_WIN32,
PWSTR(exe_name.as_mut_ptr()),
&raw mut dw_size,
)
};
match image_name_result {
Ok(()) => Some(PathBuf::from(wide_array_to_os_string(&exe_name))),
Err(_) => None,
}
})
}
fn wide_array_to_os_string(wide: &[u16]) -> OsString {
let null_pos = wide.iter().position(|&x| x == 0).unwrap_or(wide.len());
OsString::from_wide(&wide[..null_pos])
}
+32 -22
View File
@@ -25,25 +25,21 @@ pub fn is_cursor_visible() -> bool {
cbSize: u32::try_from(std::mem::size_of::<CURSORINFO>()).unwrap(), cbSize: u32::try_from(std::mem::size_of::<CURSORINFO>()).unwrap(),
..Default::default() ..Default::default()
}; };
unsafe { unsafe { GetCursorInfo(&raw mut ci) }.unwrap();
GetCursorInfo(&mut ci).unwrap();
}
ci.flags == CURSOR_SHOWING ci.flags == CURSOR_SHOWING
} }
pub fn is_window_focused(target_title: &str) -> bool { pub fn is_window_focused(target_title: &str) -> bool {
let mut buffer = [0; 512]; let mut buffer = [0; 512];
unsafe { let hwnd = unsafe { GetForegroundWindow() };
let hwnd = GetForegroundWindow(); let length = unsafe { GetWindowTextW(hwnd, &mut buffer) };
let length = GetWindowTextW(hwnd, &mut buffer); let current_title = String::from_utf16_lossy(&buffer[..length as usize]);
let current_title = String::from_utf16_lossy(&buffer[..length as usize]); current_title == target_title
current_title == target_title
}
} }
pub fn is_any_key_pressed(keys: &[VIRTUAL_KEY]) -> bool { pub fn is_any_key_pressed(keys: &[VIRTUAL_KEY]) -> bool {
keys.iter() keys.iter()
.any(|&key| unsafe { (GetAsyncKeyState(i32::from(key.0)) & i16::MIN) != 0 }) .any(|&key| unsafe { GetAsyncKeyState(i32::from(key.0)) } & i16::MIN != 0)
} }
pub fn elevate(closing: ElevationExitMethod) { pub fn elevate(closing: ElevationExitMethod) {
@@ -66,20 +62,34 @@ pub fn elevate(closing: ElevationExitMethod) {
pub fn is_elevated() -> bool { pub fn is_elevated() -> bool {
let mut token: HANDLE = HANDLE::default(); let mut token: HANDLE = HANDLE::default();
unsafe { if unsafe { OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &raw mut token) }.is_err() {
if OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &mut token).is_err() { return false;
return false; }
} let mut elevation = TOKEN_ELEVATION::default();
let mut elevation = TOKEN_ELEVATION::default(); let mut size = u32::try_from(std::mem::size_of::<TOKEN_ELEVATION>()).unwrap();
let mut size = u32::try_from(std::mem::size_of::<TOKEN_ELEVATION>()).unwrap(); let result = unsafe {
let result = GetTokenInformation( GetTokenInformation(
token, token,
TokenElevation, TokenElevation,
Some((&raw mut elevation).cast()), Some((&raw mut elevation).cast()),
size, size,
&mut size, &raw mut size,
); )
CloseHandle(token).unwrap(); };
result.is_ok() && elevation.TokenIsElevated != 0 unsafe { CloseHandle(token) }.unwrap();
} result.is_ok() && elevation.TokenIsElevated != 0
}
pub fn is_system_theme_dark() -> bool {
use winreg::RegKey;
let hkcu = RegKey::predef(winreg::enums::HKEY_CURRENT_USER);
let Ok(subkey) =
hkcu.open_subkey("Software\\Microsoft\\Windows\\CurrentVersion\\Themes\\Personalize")
else {
return true;
};
let Ok(dword): Result<u32, std::io::Error> = subkey.get_value("AppsUseLightTheme") else {
return true;
};
dword != 1
} }