-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: single instance check on windows (#26)
* deps: update * feat: single instance on windows * docs: add info about restarts in the readme
- Loading branch information
Showing
9 changed files
with
196 additions
and
31 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -3,3 +3,4 @@ pub mod elevate; | |
| pub mod message_box; | ||
| pub mod path; | ||
| mod pwstr; | ||
| pub mod single_instance; | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,99 @@ | ||
| use crate::pwstr::ManagedPwstr; | ||
| use std::{env, mem, ptr}; | ||
| use windows::{ | ||
| core::{Error, Result, HRESULT}, | ||
| Win32::{ | ||
| Foundation::{ | ||
| CloseHandle, GetLastError, ERROR_ALREADY_EXISTS, ERROR_NOT_FOUND, ERROR_SUCCESS, | ||
| HANDLE, MAX_PATH, | ||
| }, | ||
| System::{ | ||
| ProcessStatus::{K32EnumProcesses, K32GetModuleFileNameExW}, | ||
| Threading::{ | ||
| CreateMutexW, OpenProcess, TerminateProcess, PROCESS_QUERY_INFORMATION, | ||
| PROCESS_TERMINATE, | ||
| }, | ||
| }, | ||
| }, | ||
| }; | ||
|
|
||
| /// Tries to create and lock on a system wide mutex | ||
| /// | ||
| /// If the mutex is already locked, then another application locked the mutex and another instance is already running. | ||
| /// | ||
| /// Returns `false` if another instance is already running, and `true` if we are the only instance running. | ||
| pub fn try_create_new_instance(unique_instance_id: &str) -> bool { | ||
| // make sure this is a handle and not a result so we get a compiler error when the metadata changes | ||
| let _: HANDLE = unsafe { CreateMutexW(ptr::null(), true, unique_instance_id) }; | ||
| unsafe { | ||
| match GetLastError() { | ||
| ERROR_SUCCESS => true, | ||
| ERROR_ALREADY_EXISTS => false, | ||
| x => { | ||
| eprintln!("Unexpected error - {:?}", x); | ||
| false | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| pub fn kill_other_instances_of_this_application() -> Result<()> { | ||
| let mut path_buf = ManagedPwstr::alloc(MAX_PATH as usize + 1); | ||
| let this_path = ManagedPwstr::from(env::current_exe().unwrap().into_os_string()); | ||
|
|
||
| let (processes, n_processes) = get_all_processes()?; | ||
| let pid = match processes | ||
| .into_iter() | ||
| .take(n_processes as usize) | ||
| .find(|pid| cmp_path(*pid, &this_path, &mut path_buf)) | ||
| { | ||
| Some(pid) => pid, | ||
| None => return Err(Error::from(HRESULT::from(ERROR_NOT_FOUND))), | ||
| }; | ||
| unsafe { | ||
| let handle = OpenProcess(PROCESS_TERMINATE, None, pid); | ||
| if handle.is_invalid() { | ||
| return Err(GetLastError().into()); | ||
| } | ||
|
|
||
| if !TerminateProcess(handle, u32::MAX).as_bool() { | ||
| return Err(GetLastError().into()); | ||
| } | ||
| CloseHandle(handle); | ||
| } | ||
|
|
||
| Ok(()) | ||
| } | ||
|
|
||
| fn get_all_processes() -> Result<(Vec<u32>, u32)> { | ||
| let mut buf = vec![0u32; 1024]; | ||
| let mut returned_bytes = 0; | ||
| unsafe { | ||
| if !K32EnumProcesses( | ||
| buf.as_mut_ptr(), | ||
| (mem::size_of::<u32>() * buf.len()) as u32, | ||
| &mut returned_bytes, | ||
| ) | ||
| .as_bool() | ||
| { | ||
| return Err(GetLastError().into()); | ||
| } | ||
| } | ||
| Ok((buf, returned_bytes / (mem::size_of::<u32>() as u32))) | ||
| } | ||
|
|
||
| fn cmp_path(pid: u32, path: &ManagedPwstr, path_buf: &mut ManagedPwstr) -> bool { | ||
| unsafe { | ||
| let proc = OpenProcess(PROCESS_QUERY_INFORMATION, false, pid); | ||
| if proc.is_invalid() { | ||
| return false; | ||
| } | ||
| let chars = K32GetModuleFileNameExW(proc, None, path_buf.as_mut_slice()); | ||
| CloseHandle(proc); | ||
| if chars == 0 { | ||
| false | ||
| } else { | ||
| path == path_buf | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters