wanderlust 0.2.5

A powerful Windows PATH cleaner and healer. Automatically discovers tools, removes duplicates, and ensures system stability.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

//! UAC privilege check and admin relanch via ShellExecute "runas".

use log::info;
use std::ffi::CString;
use windows::Win32::Security::{GetTokenInformation, TOKEN_ELEVATION, TOKEN_QUERY, TokenElevation};
use windows::Win32::System::Threading::{GetCurrentProcess, OpenProcessToken};
use windows::Win32::UI::Shell::ShellExecuteA;
use windows::Win32::UI::WindowsAndMessaging::SW_SHOW;

pub fn is_elevated() -> bool {
    let mut token = windows::Win32::Foundation::HANDLE::default();
    unsafe {
        if OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &mut token).is_ok() {
            let mut elevation = TOKEN_ELEVATION::default();
            let mut size = 0;
            // GetTokenInformation is the Win32 API to read struct data from a token handle.
            if GetTokenInformation(
                token,
                TokenElevation,
                Some(&mut elevation as *mut _ as *mut _),
                std::mem::size_of::<TOKEN_ELEVATION>() as u32,
                &mut size,
            )
            .is_ok()
            {
                return elevation.TokenIsElevated != 0;
            }
        }
    }
    false
}

pub fn relaunch_as_admin() -> bool {
    if let Ok(exe_path) = std::env::current_exe() {
        // Safe conversion handling null bytes
        let exe_path_str = match CString::new(exe_path.to_string_lossy().as_bytes()) {
            Ok(s) => s,
            Err(_) => return false,
        };

        // Reconstruct the command line arguments
        let args: Vec<String> = std::env::args().skip(1).collect();
        let args_str = match CString::new(args.join(" ")) {
            Ok(s) => s,
            Err(_) => return false,
        };

        info!("Relaunching as admin: {:?} {:?}", exe_path, args);

        let operation = CString::new("runas").unwrap();

        unsafe {
            // ShellExecuteA executes an operation on a specified file.
            // "runas" is the magic verb that requests elevation.
            let result = ShellExecuteA(
                None, // Parent window (None = Desktop)
                windows::core::PCSTR(operation.as_ptr() as *const _),
                windows::core::PCSTR(exe_path_str.as_ptr() as *const _),
                windows::core::PCSTR(args_str.as_ptr() as *const _),
                windows::core::PCSTR(std::ptr::null()), // Working directory (NULL = current)
                SW_SHOW,                                // Show command normally
            );

            // ShellExecute returns an HINSTANCE > 32 on success.
            // Values <= 32 are error codes (e.g. SE_ERR_ACCESSDENIED).
            if result.0 as isize > 32 {
                return true;
            }
        }
    }
    false
}