73 lines
2 KiB
Rust
73 lines
2 KiB
Rust
use std::env;
|
|
use std::ffi::OsString;
|
|
use std::os::windows::ffi::OsStringExt;
|
|
use std::path::PathBuf;
|
|
use std::ptr;
|
|
use std::slice;
|
|
|
|
use windows_sys::Win32::Foundation::S_OK;
|
|
use windows_sys::Win32::System::Com::CoTaskMemFree;
|
|
use windows_sys::Win32::UI::Shell::{FOLDERID_Profile, SHGetKnownFolderPath, KF_FLAG_DONT_VERIFY};
|
|
|
|
pub fn home_dir_inner() -> Option<PathBuf> {
|
|
env::var_os("USERPROFILE")
|
|
.filter(|s| !s.is_empty())
|
|
.map(PathBuf::from)
|
|
.or_else(home_dir_crt)
|
|
}
|
|
|
|
#[cfg(not(target_vendor = "uwp"))]
|
|
fn home_dir_crt() -> Option<PathBuf> {
|
|
unsafe {
|
|
let mut path = ptr::null_mut();
|
|
match SHGetKnownFolderPath(&FOLDERID_Profile, KF_FLAG_DONT_VERIFY as u32, 0, &mut path) {
|
|
S_OK => {
|
|
let path_slice = slice::from_raw_parts(path, wcslen(path));
|
|
let s = OsString::from_wide(&path_slice);
|
|
CoTaskMemFree(path.cast());
|
|
Some(PathBuf::from(s))
|
|
}
|
|
_ => {
|
|
// Free any allocated memory even on failure. A null ptr is a no-op for `CoTaskMemFree`.
|
|
CoTaskMemFree(path.cast());
|
|
None
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(target_vendor = "uwp")]
|
|
fn home_dir_crt() -> Option<PathBuf> {
|
|
None
|
|
}
|
|
|
|
extern "C" {
|
|
fn wcslen(buf: *const u16) -> usize;
|
|
}
|
|
|
|
#[cfg(not(target_vendor = "uwp"))]
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::home_dir_inner;
|
|
use std::env;
|
|
use std::ops::Deref;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
#[test]
|
|
fn test_with_without() {
|
|
let olduserprofile = env::var_os("USERPROFILE").unwrap();
|
|
|
|
env::remove_var("HOME");
|
|
env::remove_var("USERPROFILE");
|
|
|
|
assert_eq!(home_dir_inner(), Some(PathBuf::from(olduserprofile)));
|
|
|
|
let home = Path::new(r"C:\Users\foo tar baz");
|
|
|
|
env::set_var("HOME", home.as_os_str());
|
|
assert_ne!(home_dir_inner().as_ref().map(Deref::deref), Some(home));
|
|
|
|
env::set_var("USERPROFILE", home.as_os_str());
|
|
assert_eq!(home_dir_inner().as_ref().map(Deref::deref), Some(home));
|
|
}
|
|
}
|