do so many things

src/util/consts.rs

@@ -0,0 +1,2 @@
+pub const ENHANCED: &str = "GTA5_Enhanced.exe";
+pub const LEGACY: &str = "GTA5.exe";

src/util/countdown.rs

@@ -1,15 +1,15 @@
 use std::time::{Duration, Instant};
 
 pub struct Countdown {
-    pub i: usize,
-    pub i_original: usize,
+    pub i: u64,
+    pub i_original: u64,
     pub i_string: String,
     pub interval: Instant,
     pub first_count: bool,
 }
 
 impl Countdown {
-    pub fn new(i: usize) -> Self {
+    pub fn new(i: u64) -> Self {
         Self {
             i,
             i_original: i,

src/util/elevate.rs

@@ -1,19 +0,0 @@
-use windows::{
-    Win32::UI::{Shell::ShellExecuteW, WindowsAndMessaging::SW_HIDE},
-    core::{HSTRING, PCWSTR},
-};
-
-pub fn elevate() {
-    let exe = std::env::current_exe().unwrap();
-    unsafe {
-        ShellExecuteW(
-            None,
-            &HSTRING::from("runas"),
-            &HSTRING::from(exe.as_path()),
-            PCWSTR::null(),
-            PCWSTR::null(),
-            SW_HIDE,
-        );
-    }
-    std::process::exit(0);
-}

src/util/elevation.rs

@@ -0,0 +1,45 @@
+use windows::{
+    Win32::Foundation::{CloseHandle, HANDLE},
+    Win32::Security::{GetTokenInformation, TOKEN_ELEVATION, TOKEN_QUERY, TokenElevation},
+    Win32::System::Threading::{GetCurrentProcess, OpenProcessToken},
+    Win32::UI::{Shell::ShellExecuteW, WindowsAndMessaging::SW_HIDE},
+    core::{HSTRING, PCWSTR},
+};
+
+pub fn elevate() {
+    let exe = std::env::current_exe().unwrap();
+    unsafe {
+        ShellExecuteW(
+            None,
+            &HSTRING::from("runas"),
+            &HSTRING::from(exe.as_path()),
+            PCWSTR::null(),
+            PCWSTR::null(),
+            SW_HIDE,
+        );
+    }
+    std::process::exit(0);
+}
+
+pub fn is_elevated() -> bool {
+    unsafe {
+        let mut token: HANDLE = HANDLE::default();
+        if OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &mut token).is_ok() {
+            let mut elevation = TOKEN_ELEVATION::default();
+            let size = std::mem::size_of::<TOKEN_ELEVATION>() as u32;
+            let mut ret_size = size;
+            let result = GetTokenInformation(
+                token,
+                TokenElevation,
+                Some(&mut elevation as *mut _ as *mut _),
+                size,
+                &mut ret_size,
+            );
+            CloseHandle(token).unwrap();
+            if result.is_ok() && elevation.TokenIsElevated != 0 {
+                return true;
+            }
+        }
+        false
+    }
+}

src/util/mod.rs

@@ -1,2 +1,3 @@
+pub mod consts;
 pub mod countdown;
-pub mod elevate;
+pub mod elevation;