1.4.2

Cargo.toml

@@ -1,18 +1,18 @@
 [package]
 name = "W11Boost"
-version = "1.4.1"
+version = "1.4.2"
 edition = "2024"
 build = "build.rs"
 authors = ["felikcat"]
 
 [package.metadata.winres]
 CompanyName = "felikcat"
 FileDescription = "https://github.com/felikcat/W11Boost"
-FileVersion = "1.4.1.0"
+FileVersion = "1.4.2.0"
 LegalCopyright = "© 2024 felikcat, AGPL3 License"
 OriginalFilename = "W11Boost.exe"
 ProductName = "W11Boost"
-ProductVersion = "1.4.1.0"
+ProductVersion = "1.4.2.0"
 
 # Optimize for size.
 [profile.release]
@@ -24,7 +24,7 @@ codegen-units = 1
 [dependencies]
 fltk-theme = "0.7.3"
 curl = "0.4.47"
-chrono = "0.4.38"
+chrono = "0.4.39"
 widestring = "1.1.0"
 
 [dependencies.winsafe]
@@ -55,7 +55,7 @@ features = [
 ]
 
 [dependencies.fltk]
-version = "^1.4.36"
+version = "^1.5"
 features = [
     "use-ninja"
 ]

src/common.rs

@@ -1,11 +1,20 @@
 use chrono::{Datelike, Timelike, Utc};
 use fltk::app;
+use windows_sys::Win32::System::SystemServices::MAXIMUM_ALLOWED;
+use windows_sys::Win32::System::Threading::{OpenProcess, OpenProcessToken};
 use std::error::Error;
+use std::os::raw::c_void;
 use std::fs::{self, OpenOptions};
 use std::io::Write;
 use std::path::PathBuf;
+use std::ptr::null_mut;
 use windows::Win32::Foundation::ERROR_SUCCESS;
 use windows::Win32::System::Registry::{REG_DWORD, REG_SZ, RegCreateKeyW, RegSetValueExW};
+use windows_sys::Win32::Foundation::INVALID_HANDLE_VALUE;
+use windows_sys::Win32::Security::{
+    DuplicateTokenEx, SECURITY_ATTRIBUTES, SecurityImpersonation,
+    TOKEN_ALL_ACCESS, TOKEN_DUPLICATE, TokenImpersonation,
+};
 use windows::core::PCWSTR;
 use windows::{
     Win32::System::{
@@ -345,3 +354,31 @@ pub fn save_registry_gpo(
 
     Ok(())
 }
+
+pub fn create_access_token_from_pid(process_id: u32) -> Result<*mut c_void, Box<dyn Error>> {
+    let mut dup_token = INVALID_HANDLE_VALUE;
+    unsafe {
+        let process = OpenProcess(MAXIMUM_ALLOWED, 0, process_id);
+        if !process.is_null() {
+            let mut token = INVALID_HANDLE_VALUE;
+            OpenProcessToken(process, TOKEN_DUPLICATE, &mut token);
+
+            let attributes = SECURITY_ATTRIBUTES {
+                nLength: size_of::<SECURITY_ATTRIBUTES>() as u32,
+                lpSecurityDescriptor: null_mut(),
+                bInheritHandle: 0,
+            };
+
+            DuplicateTokenEx(
+                token,
+                TOKEN_ALL_ACCESS,
+                &attributes,
+                SecurityImpersonation,
+                TokenImpersonation,
+                &mut dup_token,
+            );
+        }
+    }
+
+    Ok(dup_token)
+}