2026-08-23 00:33:53 -04:00
|
|
|
//! The one place `#[cfg]` lives. Two rules: every function is defined for
|
|
|
|
|
//! every target, so callers never wrap a call site in `#[cfg]`; and anything
|
|
|
|
|
//! decidable from a string is split out and testable on Linux ([`is_unc_string`]).
|
2026-08-02 19:04:30 -04:00
|
|
|
|
|
|
|
|
use std::ffi::OsString;
|
|
|
|
|
use std::path::{Component, Path, PathBuf};
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// The user's home directory. On Windows `%USERPROFILE%` is checked **first**:
|
|
|
|
|
/// Git Bash and MSYS2 export `HOME` as a POSIX path (`/c/Users/me`) that no
|
|
|
|
|
/// Win32 API can open.
|
2026-08-02 19:04:30 -04:00
|
|
|
pub fn home_dir() -> Option<OsString> {
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
|
|
|
|
if let Some(profile) = std::env::var_os("USERPROFILE") {
|
|
|
|
|
return Some(profile);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
std::env::var_os("HOME").or_else(|| std::env::var_os("USERPROFILE"))
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Spelled out so this compiles and tests on Linux; checked against the real header below.
|
2026-08-05 18:05:04 -04:00
|
|
|
const FILE_ATTRIBUTE_HIDDEN: u32 = 0x2;
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
const _: () = assert!(
|
|
|
|
|
FILE_ATTRIBUTE_HIDDEN == windows_sys::Win32::Storage::FileSystem::FILE_ATTRIBUTE_HIDDEN
|
|
|
|
|
);
|
|
|
|
|
|
2026-08-09 16:25:43 -04:00
|
|
|
/// `FILE_ATTRIBUTE_HIDDEN` and nothing else. `FILE_ATTRIBUTE_SYSTEM` must not
|
2026-08-23 00:33:53 -04:00
|
|
|
/// be part of this test: Windows honours a folder's `desktop.ini` only if it
|
|
|
|
|
/// carries Read-only or System, so cloud-sync clients set System on their
|
|
|
|
|
/// plainly visible sync roots purely for a branded icon — including it pruned
|
|
|
|
|
/// whole cloud folders from the index. Everything System was meant to catch
|
|
|
|
|
/// (`$RECYCLE.BIN`, `pagefile.sys`, …) is Hidden **and** System anyway.
|
2026-08-05 18:05:04 -04:00
|
|
|
#[cfg_attr(not(windows), allow(dead_code))]
|
|
|
|
|
pub(crate) fn attributes_are_hidden(attributes: u32) -> bool {
|
|
|
|
|
attributes & FILE_ATTRIBUTE_HIDDEN != 0
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Only the attribute case is worth the walk's log line: an attribute the
|
|
|
|
|
/// user cannot see has no discoverability.
|
2026-08-05 18:05:04 -04:00
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
|
|
|
pub enum HiddenReason {
|
|
|
|
|
DotPrefix,
|
|
|
|
|
Attribute,
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Whether a directory entry counts as hidden, and why. Unix: a leading dot.
|
|
|
|
|
/// Windows: a leading dot **or** `FILE_ATTRIBUTE_HIDDEN` (never System; see
|
|
|
|
|
/// [`attributes_are_hidden`]). `meta` must report the entry **itself**, never
|
|
|
|
|
/// a link target — call sites disagreeing churns the index every cycle — and
|
|
|
|
|
/// is a closure because on Unix it is never called (an extra `lstat` per entry).
|
2026-08-05 18:05:04 -04:00
|
|
|
pub fn entry_hidden_reason<F>(name: &str, meta: F) -> Option<HiddenReason>
|
2026-08-02 19:04:30 -04:00
|
|
|
where
|
|
|
|
|
F: FnOnce() -> Option<std::fs::Metadata>,
|
|
|
|
|
{
|
|
|
|
|
if name.starts_with('.') {
|
2026-08-05 18:05:04 -04:00
|
|
|
return Some(HiddenReason::DotPrefix);
|
2026-08-02 19:04:30 -04:00
|
|
|
}
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::windows::fs::MetadataExt;
|
|
|
|
|
if let Some(m) = meta() {
|
2026-08-05 18:05:04 -04:00
|
|
|
if attributes_are_hidden(m.file_attributes()) {
|
|
|
|
|
return Some(HiddenReason::Attribute);
|
|
|
|
|
}
|
2026-08-02 19:04:30 -04:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
#[cfg(not(windows))]
|
|
|
|
|
{
|
|
|
|
|
let _ = meta;
|
|
|
|
|
}
|
2026-08-05 18:05:04 -04:00
|
|
|
None
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub fn entry_is_hidden<F>(name: &str, meta: F) -> bool
|
|
|
|
|
where
|
|
|
|
|
F: FnOnce() -> Option<std::fs::Metadata>,
|
|
|
|
|
{
|
|
|
|
|
entry_hidden_reason(name, meta).is_some()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const FILE_ATTRIBUTE_REPARSE_POINT: u32 = 0x400;
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
const _: () = assert!(
|
|
|
|
|
FILE_ATTRIBUTE_REPARSE_POINT
|
|
|
|
|
== windows_sys::Win32::Storage::FileSystem::FILE_ATTRIBUTE_REPARSE_POINT
|
|
|
|
|
);
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// The attributes a dehydrated cloud file carries: `OFFLINE` (0x1000) is the
|
|
|
|
|
/// old tape-archive bit OneDrive reused; `RECALL_ON_OPEN` (0x40000) marks a
|
|
|
|
|
/// file whose *metadata* is local but whose data is not;
|
|
|
|
|
/// `RECALL_ON_DATA_ACCESS` (0x400000) is the modern Files-On-Demand
|
|
|
|
|
/// placeholder. Any one of them means a read pulls the file over the network.
|
2026-08-05 18:05:04 -04:00
|
|
|
const FILE_ATTRIBUTE_OFFLINE: u32 = 0x1000;
|
|
|
|
|
const FILE_ATTRIBUTE_RECALL_ON_OPEN: u32 = 0x4_0000;
|
|
|
|
|
const FILE_ATTRIBUTE_RECALL_ON_DATA_ACCESS: u32 = 0x40_0000;
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
const _: () = {
|
|
|
|
|
use windows_sys::Win32::Storage::FileSystem as fs_attrs;
|
|
|
|
|
assert!(FILE_ATTRIBUTE_OFFLINE == fs_attrs::FILE_ATTRIBUTE_OFFLINE);
|
|
|
|
|
assert!(FILE_ATTRIBUTE_RECALL_ON_OPEN == fs_attrs::FILE_ATTRIBUTE_RECALL_ON_OPEN);
|
|
|
|
|
assert!(FILE_ATTRIBUTE_RECALL_ON_DATA_ACCESS == fs_attrs::FILE_ATTRIBUTE_RECALL_ON_DATA_ACCESS);
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
/// Whether reading this file's contents would pull it down from the cloud.
|
2026-08-23 00:33:53 -04:00
|
|
|
/// A placeholder has real metadata and no data; the walk hashes the head of
|
|
|
|
|
/// every new or changed file, so getting this wrong quietly downloads the
|
2026-08-09 16:25:43 -04:00
|
|
|
/// user's whole cloud drive. Always `false` off Windows.
|
2026-08-05 18:05:04 -04:00
|
|
|
pub fn is_cloud_placeholder(meta: &std::fs::Metadata) -> bool {
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::windows::fs::MetadataExt;
|
|
|
|
|
return attributes_are_dehydrated(meta.file_attributes());
|
|
|
|
|
}
|
|
|
|
|
#[cfg(not(windows))]
|
|
|
|
|
{
|
|
|
|
|
let _ = meta;
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg_attr(not(windows), allow(dead_code))]
|
|
|
|
|
pub(crate) fn attributes_are_dehydrated(attributes: u32) -> bool {
|
|
|
|
|
attributes
|
|
|
|
|
& (FILE_ATTRIBUTE_OFFLINE
|
|
|
|
|
| FILE_ATTRIBUTE_RECALL_ON_OPEN
|
|
|
|
|
| FILE_ATTRIBUTE_RECALL_ON_DATA_ACCESS)
|
|
|
|
|
!= 0
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-09 16:25:43 -04:00
|
|
|
/// *Not* the same question as `FileType::is_symlink`, which additionally
|
|
|
|
|
/// requires the name-surrogate bit: see [`entry_cached_metadata`].
|
2026-08-05 18:05:04 -04:00
|
|
|
#[cfg_attr(not(windows), allow(dead_code))]
|
|
|
|
|
pub(crate) fn attributes_are_reparse_point(attributes: u32) -> bool {
|
|
|
|
|
attributes & FILE_ATTRIBUTE_REPARSE_POINT != 0
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Metadata a directory read already handed back. On Windows `FindNextFileW`
|
|
|
|
|
/// returns size, timestamps and attributes with every name and
|
|
|
|
|
/// `DirEntry::metadata` is a copy of that buffer; `fs::metadata(path)` is a
|
|
|
|
|
/// whole extra open/stat/close through every antivirus minifilter.
|
|
|
|
|
/// Uninhabited elsewhere, so `Option<CachedMetadata>` is zero-sized off Windows.
|
2026-08-05 18:05:04 -04:00
|
|
|
#[cfg(windows)]
|
|
|
|
|
pub(crate) type CachedMetadata = std::fs::Metadata;
|
|
|
|
|
#[cfg(not(windows))]
|
|
|
|
|
pub(crate) type CachedMetadata = std::convert::Infallible;
|
|
|
|
|
|
|
|
|
|
#[cfg(not(windows))]
|
|
|
|
|
const _: () = assert!(std::mem::size_of::<Option<CachedMetadata>>() == 0);
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Cached entry metadata, but only where trusting it is indistinguishable
|
|
|
|
|
/// from a fresh `stat`. `None` for any reparse point, and that is the whole
|
|
|
|
|
/// subtlety: `fs::metadata` *follows* a reparse point while the cached buffer
|
|
|
|
|
/// describes the link itself, and the tags std does not classify as symlinks
|
|
|
|
|
/// (`IO_REPARSE_TAG_APPEXECLINK`, OneDrive's `IO_REPARSE_TAG_CLOUD_*`) reach
|
|
|
|
|
/// the walk's ordinary file arm where the two answers differ. Hence the raw
|
|
|
|
|
/// `FILE_ATTRIBUTE_REPARSE_POINT` test, not `is_symlink`, which also wants
|
|
|
|
|
/// the name-surrogate bit `0x20000000` — the test that lets those tags through.
|
2026-08-05 18:05:04 -04:00
|
|
|
pub(crate) fn entry_cached_metadata<F>(meta: F) -> Option<CachedMetadata>
|
|
|
|
|
where
|
|
|
|
|
F: FnOnce() -> Option<std::fs::Metadata>,
|
|
|
|
|
{
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::windows::fs::MetadataExt;
|
|
|
|
|
if let Some(m) = meta().filter(|m| !attributes_are_reparse_point(m.file_attributes())) {
|
|
|
|
|
return Some(m);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
#[cfg(not(windows))]
|
|
|
|
|
{
|
|
|
|
|
let _ = meta;
|
|
|
|
|
}
|
|
|
|
|
None
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
pub(crate) fn metadata_or_stat(
|
|
|
|
|
path: &Path,
|
|
|
|
|
cached: Option<CachedMetadata>,
|
|
|
|
|
) -> std::io::Result<std::fs::Metadata> {
|
|
|
|
|
match cached {
|
|
|
|
|
Some(m) => Ok(m),
|
|
|
|
|
None => std::fs::metadata(path),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(not(windows))]
|
|
|
|
|
pub(crate) fn metadata_or_stat(
|
|
|
|
|
path: &Path,
|
|
|
|
|
_cached: Option<CachedMetadata>,
|
|
|
|
|
) -> std::io::Result<std::fs::Metadata> {
|
|
|
|
|
std::fs::metadata(path)
|
2026-08-02 19:04:30 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Whether `path` has a hidden component *below* the root that contains it.
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Components at or above a root are exempt because the walkers exempt their
|
|
|
|
|
/// root too; the two must agree or the index churns — a certainty on Windows,
|
|
|
|
|
/// where `AppData` is Hidden and roots under it are routine. `roots` match by
|
|
|
|
|
/// whole components; a path under no known root is checked in full.
|
2026-08-02 19:04:30 -04:00
|
|
|
pub fn path_has_hidden_component_under(path: &Path, roots: &[PathBuf]) -> bool {
|
|
|
|
|
// Innermost containing root wins: with both `/data` and `/data/.cache`
|
|
|
|
|
// configured, a file under the latter is only judged below `.cache`.
|
|
|
|
|
let base = roots
|
|
|
|
|
.iter()
|
|
|
|
|
.filter(|r| path.starts_with(r))
|
|
|
|
|
.max_by_key(|r| r.components().count());
|
|
|
|
|
|
|
|
|
|
let (mut current, tail) = match base {
|
|
|
|
|
Some(root) => match path.strip_prefix(root) {
|
|
|
|
|
Ok(tail) => (root.clone(), tail),
|
|
|
|
|
Err(_) => (PathBuf::new(), path),
|
|
|
|
|
},
|
|
|
|
|
None => (PathBuf::new(), path),
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
for component in tail.components() {
|
|
|
|
|
current.push(component);
|
|
|
|
|
if let Component::Normal(name) = component {
|
|
|
|
|
let name = name.to_string_lossy();
|
2026-08-05 18:05:04 -04:00
|
|
|
// `symlink_metadata`, not `metadata`: each component is judged as
|
2026-08-23 00:33:53 -04:00
|
|
|
// itself, which is what the walkers do.
|
2026-08-05 18:05:04 -04:00
|
|
|
if entry_is_hidden(&name, || std::fs::symlink_metadata(¤t).ok()) {
|
2026-08-02 19:04:30 -04:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Whether `s` names a UNC path, in either spelling. Only *called* on Windows.
|
2026-08-02 19:04:30 -04:00
|
|
|
#[cfg_attr(not(windows), allow(dead_code))]
|
|
|
|
|
pub(crate) fn is_unc_string(s: &str) -> bool {
|
|
|
|
|
s.starts_with(r"\\?\UNC\") || (s.starts_with(r"\\") && !s.starts_with(r"\\?\"))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(target_os = "linux")]
|
|
|
|
|
const NETWORK_FS_TYPES: [&str; 8] = [
|
2026-08-04 03:27:05 -04:00
|
|
|
"cifs",
|
|
|
|
|
"smb3",
|
|
|
|
|
"smbfs",
|
|
|
|
|
"nfs",
|
|
|
|
|
"nfs4",
|
|
|
|
|
"afs",
|
|
|
|
|
"fuse.sshfs",
|
|
|
|
|
"9p",
|
2026-08-02 19:04:30 -04:00
|
|
|
];
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// The longest `/proc/mounts` mount point that is a prefix of `path` is the
|
|
|
|
|
/// one that actually serves it.
|
2026-08-02 19:04:30 -04:00
|
|
|
#[cfg(target_os = "linux")]
|
|
|
|
|
pub(crate) fn is_network_path(path: &Path) -> bool {
|
|
|
|
|
let Ok(mounts) = std::fs::read_to_string("/proc/mounts") else {
|
|
|
|
|
return false;
|
|
|
|
|
};
|
|
|
|
|
let target = std::fs::canonicalize(path).unwrap_or_else(|_| path.to_path_buf());
|
|
|
|
|
|
|
|
|
|
let mut best: Option<(usize, bool)> = None;
|
|
|
|
|
for line in mounts.lines() {
|
|
|
|
|
let mut fields = line.split_whitespace();
|
|
|
|
|
let (Some(_dev), Some(point), Some(fstype)) = (fields.next(), fields.next(), fields.next())
|
|
|
|
|
else {
|
|
|
|
|
continue;
|
|
|
|
|
};
|
|
|
|
|
// `/proc/mounts` octal-escapes spaces and a few other characters.
|
|
|
|
|
let point = point.replace("\\040", " ");
|
|
|
|
|
let point = Path::new(&point);
|
|
|
|
|
if !target.starts_with(point) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
let depth = point.components().count();
|
|
|
|
|
let is_network = NETWORK_FS_TYPES.contains(&fstype);
|
|
|
|
|
if best.is_none_or(|(d, _)| depth > d) {
|
|
|
|
|
best = Some((depth, is_network));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
best.is_some_and(|(_, is_network)| is_network)
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// UNC needs no syscall; a *mapped drive letter* is indistinguishable from a
|
2026-08-09 16:25:43 -04:00
|
|
|
/// local disk by string inspection — `GetDriveTypeW` is the only way to tell.
|
2026-08-02 19:04:30 -04:00
|
|
|
#[cfg(windows)]
|
|
|
|
|
pub(crate) fn is_network_path(path: &Path) -> bool {
|
|
|
|
|
use std::os::windows::ffi::OsStrExt;
|
|
|
|
|
use windows_sys::Win32::Storage::FileSystem::GetDriveTypeW;
|
|
|
|
|
use windows_sys::Win32::System::WindowsProgramming::DRIVE_REMOTE;
|
|
|
|
|
|
|
|
|
|
let s = path.to_string_lossy();
|
|
|
|
|
if is_unc_string(&s) {
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// GetDriveTypeW wants a root ("Z:\"), not an arbitrary path.
|
|
|
|
|
let Some(root) = path.components().next() else {
|
|
|
|
|
return false;
|
|
|
|
|
};
|
|
|
|
|
let Component::Prefix(prefix) = root else {
|
|
|
|
|
return false;
|
|
|
|
|
};
|
|
|
|
|
let mut wide: Vec<u16> = prefix.as_os_str().encode_wide().collect();
|
|
|
|
|
wide.push(b'\\' as u16);
|
|
|
|
|
wide.push(0);
|
|
|
|
|
unsafe { GetDriveTypeW(wide.as_ptr()) == DRIVE_REMOTE }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(not(any(target_os = "linux", windows)))]
|
|
|
|
|
pub(crate) fn is_network_path(_path: &Path) -> bool {
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Whether the notification backend covers a whole tree from one watch on its
|
|
|
|
|
/// root. `false` (inotify): one descriptor covers one directory, so the
|
|
|
|
|
/// caller registers every directory itself — which lets it skip ignored and
|
|
|
|
|
/// hidden subtrees. `true` (`ReadDirectoryChangesW`): one handle covers the
|
|
|
|
|
/// subtree, and notify allocates a 16 KiB buffer *inline per watch*, so
|
|
|
|
|
/// per-directory registration would ask for gigabytes; pruning moves to the
|
|
|
|
|
/// event path. macOS FSEvents is also recursive but stays per-directory.
|
2026-08-02 19:04:30 -04:00
|
|
|
pub const WATCH_ROOTS_RECURSIVELY: bool = cfg!(windows);
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// What ignore patterns compile against: on Windows and macOS `node_modules`
|
2026-08-09 16:25:43 -04:00
|
|
|
/// has to exclude `Node_Modules`, and on Linux it must not.
|
2026-08-05 18:05:04 -04:00
|
|
|
pub const PATHS_ARE_CASE_INSENSITIVE: bool = cfg!(any(windows, target_os = "macos"));
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Drop the **calling thread** to background scheduling priority (the GUI
|
|
|
|
|
/// shares this process). Best-effort and idempotent. **CPU only, on every
|
|
|
|
|
/// platform**: Windows' `THREAD_MODE_BACKGROUND_BEGIN` drops I/O to a tier
|
|
|
|
|
/// the kernel actively rate-limits, and Linux's `nice` has no I/O half —
|
|
|
|
|
/// don't reintroduce it.
|
2026-08-03 03:06:19 -04:00
|
|
|
pub fn set_background_priority() {
|
|
|
|
|
#[cfg(target_os = "linux")]
|
|
|
|
|
{
|
|
|
|
|
// Linux schedules per task, so `nice` moves this thread alone.
|
|
|
|
|
unsafe { libc::nice(10) };
|
|
|
|
|
}
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
|
|
|
|
use windows_sys::Win32::System::Threading::{
|
2026-08-05 18:05:04 -04:00
|
|
|
GetCurrentThread, SetThreadPriority, THREAD_PRIORITY_BELOW_NORMAL,
|
2026-08-03 03:06:19 -04:00
|
|
|
};
|
2026-08-05 18:05:04 -04:00
|
|
|
unsafe { SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_BELOW_NORMAL) };
|
2026-08-03 03:06:19 -04:00
|
|
|
}
|
2026-08-23 00:33:53 -04:00
|
|
|
// Elsewhere (macOS, BSD): `nice` applies to the whole process — skip.
|
2026-08-03 03:06:19 -04:00
|
|
|
}
|
2026-08-05 19:17:11 -04:00
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// glibc **caches freed thread stacks** (`stack_cache_maxsize`, 40 MiB
|
|
|
|
|
/// default) and a cached stack keeps its dirty pages for the life of the
|
|
|
|
|
/// process; 512 KiB is ample. Document parsing recurses on untrusted input —
|
|
|
|
|
/// if a malformed document ever overflows this, raise it and add a depth limit.
|
2026-08-05 19:17:11 -04:00
|
|
|
const WORKER_STACK_SIZE: usize = 512 * 1024;
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Spawn a named worker thread with [`WORKER_STACK_SIZE`]; panics like
|
|
|
|
|
/// `thread::spawn` if it cannot.
|
2026-08-05 19:17:11 -04:00
|
|
|
pub fn spawn_worker<F, T>(name: &str, f: F) -> std::thread::JoinHandle<T>
|
|
|
|
|
where
|
|
|
|
|
F: FnOnce() -> T + Send + 'static,
|
|
|
|
|
T: Send + 'static,
|
|
|
|
|
{
|
|
|
|
|
std::thread::Builder::new()
|
|
|
|
|
.name(name.to_string())
|
|
|
|
|
.stack_size(WORKER_STACK_SIZE)
|
|
|
|
|
.spawn(f)
|
|
|
|
|
.expect("spawn worker thread")
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Return free heap pages to the kernel. glibc's `free` keeps chunks on arena
|
|
|
|
|
/// free lists, so a transient peak stays in RSS for the life of the process;
|
|
|
|
|
/// `malloc_trim(0)` walks *every* arena, reclaiming other threads' leavings
|
|
|
|
|
/// too, and costs milliseconds — it must not go anywhere hot. Idempotent.
|
2026-08-05 19:17:11 -04:00
|
|
|
pub fn release_free_heap() {
|
|
|
|
|
#[cfg(all(target_os = "linux", target_env = "gnu"))]
|
|
|
|
|
{
|
2026-08-23 00:33:53 -04:00
|
|
|
// SAFETY: callable from any thread; glibc takes the arena locks itself.
|
2026-08-05 19:17:11 -04:00
|
|
|
unsafe { libc::malloc_trim(0) };
|
|
|
|
|
}
|
2026-08-23 00:33:53 -04:00
|
|
|
// Elsewhere: `malloc_trim` is a glibc extension; musl frees to the kernel.
|
2026-08-05 19:17:11 -04:00
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Create `dir` and its parents, readable only by their owner:
|
|
|
|
|
/// `create_dir_all` leaves the mode to the umask (022 ⇒ world `+rx`) and
|
|
|
|
|
/// these directories hold the index and config. Only directories *created
|
|
|
|
|
/// here* are narrowed — a user who chose `~/Documents` as their data
|
|
|
|
|
/// directory did not ask for it to be locked down.
|
2026-08-20 00:13:10 -04:00
|
|
|
pub fn create_dir_private(dir: &Path) -> std::io::Result<()> {
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::unix::fs::DirBuilderExt;
|
|
|
|
|
std::fs::DirBuilder::new()
|
|
|
|
|
.recursive(true)
|
|
|
|
|
.mode(0o700)
|
|
|
|
|
.create(dir)
|
|
|
|
|
}
|
|
|
|
|
#[cfg(not(unix))]
|
|
|
|
|
{
|
2026-08-23 00:33:53 -04:00
|
|
|
// Windows: the user-profile ACL already excludes other users.
|
2026-08-20 00:13:10 -04:00
|
|
|
std::fs::create_dir_all(dir)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Narrow `path` to owner-only access, best effort. SQLite creates its
|
|
|
|
|
/// database 0644 and copies that mode to the `-wal`/`-shm` it derives, so
|
|
|
|
|
/// narrowing the main file before anything else opens covers all three; the
|
|
|
|
|
/// index is a strictly larger secret than any single file it was read from.
|
|
|
|
|
/// Failure is ignored: a FAT stick or share has no permissions to set.
|
2026-08-20 00:13:10 -04:00
|
|
|
pub fn restrict_to_owner(path: &Path) {
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::unix::fs::PermissionsExt;
|
|
|
|
|
let _ = std::fs::set_permissions(path, std::fs::Permissions::from_mode(0o600));
|
|
|
|
|
}
|
|
|
|
|
#[cfg(not(unix))]
|
|
|
|
|
{
|
|
|
|
|
let _ = path;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Open `path` for reading, refusing anything that is not a regular file. A
|
|
|
|
|
/// rename since the caller's `stat` can put a FIFO or device there, and
|
|
|
|
|
/// `open` on one blocks in the kernel until a writer appears —
|
|
|
|
|
/// uninterruptibly, past any stop flag — stranding a walk worker and the
|
|
|
|
|
/// pool behind it. `O_NONBLOCK` makes the open return; `fstat` on the
|
|
|
|
|
/// descriptor then answers un-swappably. Both are free on a regular file.
|
2026-08-20 00:13:10 -04:00
|
|
|
pub fn open_regular_file(path: &Path) -> std::io::Result<std::fs::File> {
|
|
|
|
|
let mut opts = std::fs::OpenOptions::new();
|
|
|
|
|
opts.read(true);
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::unix::fs::OpenOptionsExt;
|
|
|
|
|
opts.custom_flags(libc::O_NONBLOCK);
|
|
|
|
|
}
|
|
|
|
|
let file = opts.open(path)?;
|
|
|
|
|
if !file.metadata()?.file_type().is_file() {
|
|
|
|
|
return Err(std::io::Error::new(
|
|
|
|
|
std::io::ErrorKind::InvalidInput,
|
|
|
|
|
"not a regular file",
|
|
|
|
|
));
|
|
|
|
|
}
|
|
|
|
|
Ok(file)
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-20 02:34:08 -04:00
|
|
|
/// Why [`IndexLock::acquire`] did not hand back a lock.
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
|
pub enum LockError {
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Another process holds it. `pid` is for the message only — never decides.
|
2026-08-20 02:34:08 -04:00
|
|
|
Held { pid: Option<u32> },
|
2026-08-23 00:33:53 -04:00
|
|
|
/// The filesystem does not do locks. The caller must carry on regardless.
|
2026-08-20 02:34:08 -04:00
|
|
|
Unsupported(String),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Proof that this process, and no other, owns the index at a given path.
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Dropping it, or exiting, releases the lock.
|
2026-08-20 02:34:08 -04:00
|
|
|
#[derive(Debug)]
|
|
|
|
|
pub struct IndexLock {
|
2026-08-23 00:33:53 -04:00
|
|
|
/// The lock lives on this open file description; keeping it alive is the mechanism.
|
2026-08-20 02:34:08 -04:00
|
|
|
_file: std::fs::File,
|
|
|
|
|
path: PathBuf,
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Never dropped: the kernel releases the lock however the process goes away.
|
2026-08-20 02:34:08 -04:00
|
|
|
static HELD_LOCK: std::sync::Mutex<Option<IndexLock>> = std::sync::Mutex::new(None);
|
|
|
|
|
|
|
|
|
|
impl IndexLock {
|
2026-08-23 00:33:53 -04:00
|
|
|
/// The lock file for the index at `db_path`. Its own name, never the
|
|
|
|
|
/// database or a SQLite sidecar: a `flock` of ours on an inode SQLite
|
|
|
|
|
/// also locks would be a second protocol on one file, and on Unix our
|
|
|
|
|
/// `close` of it would cancel SQLite's locks.
|
2026-08-20 02:34:08 -04:00
|
|
|
pub fn path_for(db_path: &Path) -> PathBuf {
|
|
|
|
|
let name = db_path
|
|
|
|
|
.file_name()
|
|
|
|
|
.and_then(|s| s.to_str())
|
|
|
|
|
.unwrap_or("index.sqlite");
|
|
|
|
|
db_path.with_file_name(format!("{}.lock", name))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Take the index lock, or report who has it.
|
|
|
|
|
///
|
2026-08-23 00:33:53 -04:00
|
|
|
/// **The guard is the kernel's lock, never the file's existence**: the
|
|
|
|
|
/// kernel drops it however the process dies, so a leftover `.lock` file is
|
|
|
|
|
/// inert and nothing here may ever branch on its presence — a stale-PID
|
|
|
|
|
/// scheme would strand the user behind a crash. [`LockError::Unsupported`]
|
|
|
|
|
/// means the filesystem could not answer, not that the lock is taken;
|
|
|
|
|
/// callers **must** start anyway.
|
2026-08-20 02:34:08 -04:00
|
|
|
pub fn acquire(db_path: &Path) -> Result<IndexLock, LockError> {
|
|
|
|
|
let path = IndexLock::path_for(db_path);
|
|
|
|
|
if let Some(dir) = path.parent() {
|
|
|
|
|
if !dir.as_os_str().is_empty() {
|
|
|
|
|
let _ = create_dir_private(dir);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
let file = match std::fs::OpenOptions::new()
|
|
|
|
|
.read(true)
|
|
|
|
|
.write(true)
|
|
|
|
|
.create(true)
|
|
|
|
|
.truncate(false)
|
|
|
|
|
.open(&path)
|
|
|
|
|
{
|
|
|
|
|
Ok(f) => f,
|
2026-08-23 00:33:53 -04:00
|
|
|
// An unwritable index directory is not a second instance: carry on.
|
2026-08-20 02:34:08 -04:00
|
|
|
Err(e) => return Err(LockError::Unsupported(format!("{}: {}", path.display(), e))),
|
|
|
|
|
};
|
|
|
|
|
lock_exclusive_nonblocking(&file).map_err(|e| match e {
|
|
|
|
|
LockAttempt::Held => LockError::Held {
|
|
|
|
|
pid: read_recorded_pid(&path),
|
|
|
|
|
},
|
|
|
|
|
LockAttempt::Unsupported(msg) => LockError::Unsupported(msg),
|
|
|
|
|
})?;
|
|
|
|
|
let lock = IndexLock {
|
|
|
|
|
_file: file,
|
|
|
|
|
path: path.clone(),
|
|
|
|
|
};
|
|
|
|
|
lock.record_holder();
|
|
|
|
|
Ok(lock)
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// [`IndexLock::acquire`] into [`HELD_LOCK`]; what a frontend calls at startup.
|
2026-08-20 02:34:08 -04:00
|
|
|
pub fn hold(db_path: &Path) -> Result<(), LockError> {
|
|
|
|
|
let lock = IndexLock::acquire(db_path)?;
|
|
|
|
|
*crate::lock_ok(&HELD_LOCK) = Some(lock);
|
|
|
|
|
Ok(())
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Move the held lock onto the index at `db_path`. **The new lock is
|
|
|
|
|
/// taken before the old one is let go**, so a refusal leaves this process
|
|
|
|
|
/// holding what it held. [`LockError::Unsupported`] means the move
|
|
|
|
|
/// *happened* and the new path simply cannot be locked; carry on.
|
2026-08-20 02:34:08 -04:00
|
|
|
pub fn move_to(db_path: &Path) -> Result<(), LockError> {
|
|
|
|
|
let mut slot = crate::lock_ok(&HELD_LOCK);
|
|
|
|
|
// Short-circuit before acquiring: `flock` conflicts with itself across
|
|
|
|
|
// two open file descriptions even inside one process, so re-taking the
|
|
|
|
|
// path we already hold would report itself as `Held`.
|
|
|
|
|
if slot
|
|
|
|
|
.as_ref()
|
|
|
|
|
.is_some_and(|held| held.path == IndexLock::path_for(db_path))
|
|
|
|
|
{
|
|
|
|
|
return Ok(());
|
|
|
|
|
}
|
|
|
|
|
match IndexLock::acquire(db_path) {
|
|
|
|
|
Ok(lock) => {
|
|
|
|
|
*slot = Some(lock);
|
|
|
|
|
Ok(())
|
|
|
|
|
}
|
|
|
|
|
Err(LockError::Unsupported(why)) => {
|
|
|
|
|
*slot = None;
|
|
|
|
|
Err(LockError::Unsupported(why))
|
|
|
|
|
}
|
|
|
|
|
Err(e) => Err(e),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn record_holder(&self) {
|
|
|
|
|
use std::io::Write;
|
|
|
|
|
// A fresh handle: writing through `_file` would move the shared file
|
2026-08-23 00:33:53 -04:00
|
|
|
// offset. That a *second* handle can write here at all is why the
|
|
|
|
|
// Windows lock byte sits at [`LOCK_BYTE_OFFSET`] rather than offset 0
|
|
|
|
|
// — its byte-range locks are mandatory and per-handle, so a lock over
|
|
|
|
|
// this byte would block our own write.
|
2026-08-20 02:34:08 -04:00
|
|
|
let written = std::fs::OpenOptions::new()
|
|
|
|
|
.write(true)
|
|
|
|
|
.truncate(true)
|
|
|
|
|
.open(&self.path)
|
|
|
|
|
.and_then(|mut f| write!(f, "{}", std::process::id()));
|
|
|
|
|
if let Err(e) = written {
|
|
|
|
|
crate::log_warn!(
|
|
|
|
|
"could not record the lock holder in {}: {}",
|
|
|
|
|
self.path.display(),
|
|
|
|
|
e
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// `None` whenever the file is absent, empty or unparseable — all ordinary.
|
2026-08-20 02:34:08 -04:00
|
|
|
fn read_recorded_pid(path: &Path) -> Option<u32> {
|
|
|
|
|
std::fs::read_to_string(path).ok()?.trim().parse().ok()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
enum LockAttempt {
|
|
|
|
|
Held,
|
|
|
|
|
Unsupported(String),
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Nowhere near the PID at offset 0. The Unix arm has no equivalent: `flock`
|
|
|
|
|
/// locks the open file description, not a byte range.
|
2026-08-20 02:34:08 -04:00
|
|
|
#[cfg(windows)]
|
|
|
|
|
const LOCK_BYTE_OFFSET: u64 = 1 << 63;
|
|
|
|
|
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
|
fn lock_exclusive_nonblocking(file: &std::fs::File) -> Result<(), LockAttempt> {
|
|
|
|
|
use std::os::unix::io::AsRawFd;
|
|
|
|
|
|
|
|
|
|
// `flock`, not `fcntl`: a POSIX record lock would be cancelled by any
|
2026-08-23 00:33:53 -04:00
|
|
|
// `close` this process makes on the same inode — the hazard documented in
|
|
|
|
|
// `file_handling::index_file_set`. A `flock` belongs to the open file
|
|
|
|
|
// description and is immune to it.
|
2026-08-20 02:34:08 -04:00
|
|
|
let rc = unsafe { libc::flock(file.as_raw_fd(), libc::LOCK_EX | libc::LOCK_NB) };
|
|
|
|
|
if rc == 0 {
|
|
|
|
|
return Ok(());
|
|
|
|
|
}
|
|
|
|
|
let err = std::io::Error::last_os_error();
|
|
|
|
|
match err.raw_os_error() {
|
|
|
|
|
Some(libc::EWOULDBLOCK) => Err(LockAttempt::Held),
|
|
|
|
|
_ => Err(LockAttempt::Unsupported(err.to_string())),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
fn lock_exclusive_nonblocking(file: &std::fs::File) -> Result<(), LockAttempt> {
|
|
|
|
|
use std::os::windows::io::AsRawHandle;
|
|
|
|
|
use windows_sys::Win32::Foundation::ERROR_LOCK_VIOLATION;
|
|
|
|
|
use windows_sys::Win32::Storage::FileSystem::{
|
|
|
|
|
LockFileEx, LOCKFILE_EXCLUSIVE_LOCK, LOCKFILE_FAIL_IMMEDIATELY,
|
|
|
|
|
};
|
|
|
|
|
use windows_sys::Win32::System::IO::{OVERLAPPED, OVERLAPPED_0, OVERLAPPED_0_0};
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
// The locked byte sits at [`LOCK_BYTE_OFFSET`] — **not** at offset 0,
|
|
|
|
|
// where the PID is written. Windows byte-range locks are *mandatory* and
|
|
|
|
|
// belong to the file object: a lock covering offset 0 would make our own
|
|
|
|
|
// `record_holder` (a second handle) and `read_recorded_pid` fail with
|
|
|
|
|
// `ERROR_LOCK_VIOLATION`. Locking beyond end-of-file is explicitly legal
|
|
|
|
|
// and is the conventional way to use a file as a semaphore.
|
2026-08-20 02:34:08 -04:00
|
|
|
let mut overlapped: OVERLAPPED = unsafe { std::mem::zeroed() };
|
|
|
|
|
overlapped.Anonymous = OVERLAPPED_0 {
|
|
|
|
|
Anonymous: OVERLAPPED_0_0 {
|
|
|
|
|
Offset: LOCK_BYTE_OFFSET as u32,
|
|
|
|
|
OffsetHigh: (LOCK_BYTE_OFFSET >> 32) as u32,
|
|
|
|
|
},
|
|
|
|
|
};
|
|
|
|
|
let ok = unsafe {
|
|
|
|
|
LockFileEx(
|
|
|
|
|
file.as_raw_handle() as _,
|
|
|
|
|
LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY,
|
|
|
|
|
0,
|
|
|
|
|
1,
|
|
|
|
|
0,
|
|
|
|
|
&mut overlapped,
|
|
|
|
|
)
|
|
|
|
|
};
|
|
|
|
|
if ok != 0 {
|
|
|
|
|
return Ok(());
|
|
|
|
|
}
|
|
|
|
|
let err = std::io::Error::last_os_error();
|
|
|
|
|
match err.raw_os_error() {
|
|
|
|
|
Some(code) if code == ERROR_LOCK_VIOLATION as i32 => Err(LockAttempt::Held),
|
|
|
|
|
_ => Err(LockAttempt::Unsupported(err.to_string())),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Bytes free to this user on the filesystem holding `path`. `None` is
|
|
|
|
|
/// "unknown", never "zero". The nearest existing ancestor is what gets asked.
|
2026-08-20 02:34:08 -04:00
|
|
|
pub fn available_space(path: &Path) -> Option<u64> {
|
|
|
|
|
let mut probe = path;
|
|
|
|
|
loop {
|
|
|
|
|
if probe.exists() {
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
probe = probe.parent()?;
|
|
|
|
|
}
|
|
|
|
|
available_space_of_existing(probe)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
|
fn available_space_of_existing(path: &Path) -> Option<u64> {
|
|
|
|
|
use std::ffi::CString;
|
|
|
|
|
use std::os::unix::ffi::OsStrExt;
|
|
|
|
|
|
|
|
|
|
let c_path = CString::new(path.as_os_str().as_bytes()).ok()?;
|
|
|
|
|
let mut stat: libc::statvfs = unsafe { std::mem::zeroed() };
|
|
|
|
|
if unsafe { libc::statvfs(c_path.as_ptr(), &mut stat) } != 0 {
|
|
|
|
|
return None;
|
|
|
|
|
}
|
2026-08-23 00:33:53 -04:00
|
|
|
// `f_bavail`, not `f_bfree`: the difference is root's reserve. `f_frsize`
|
|
|
|
|
// is the fragment size the block counts are in — `f_bsize` is the
|
|
|
|
|
// preferred I/O size and is the wrong multiplier.
|
2026-08-20 02:34:08 -04:00
|
|
|
let frsize = if stat.f_frsize > 0 {
|
|
|
|
|
stat.f_frsize
|
|
|
|
|
} else {
|
|
|
|
|
stat.f_bsize
|
|
|
|
|
};
|
|
|
|
|
(stat.f_bavail as u64).checked_mul(frsize as u64)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
fn available_space_of_existing(path: &Path) -> Option<u64> {
|
|
|
|
|
use std::os::windows::ffi::OsStrExt;
|
|
|
|
|
use windows_sys::Win32::Storage::FileSystem::GetDiskFreeSpaceExW;
|
|
|
|
|
|
|
|
|
|
let wide: Vec<u16> = path
|
|
|
|
|
.as_os_str()
|
|
|
|
|
.encode_wide()
|
|
|
|
|
.chain(std::iter::once(0))
|
|
|
|
|
.collect();
|
|
|
|
|
let mut free_to_caller: u64 = 0;
|
|
|
|
|
let ok = unsafe {
|
|
|
|
|
GetDiskFreeSpaceExW(
|
|
|
|
|
wide.as_ptr(),
|
|
|
|
|
&mut free_to_caller,
|
|
|
|
|
std::ptr::null_mut(),
|
|
|
|
|
std::ptr::null_mut(),
|
|
|
|
|
)
|
|
|
|
|
};
|
|
|
|
|
(ok != 0).then_some(free_to_caller)
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-02 19:04:30 -04:00
|
|
|
#[cfg(windows)]
|
|
|
|
|
const REMOVE_RETRY_BUDGET: std::time::Duration = std::time::Duration::from_millis(500);
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// `fs::remove_file`, retried briefly on Windows, which returns a sharing
|
|
|
|
|
/// violation while *any* handle is open — most often an antivirus scanner
|
|
|
|
|
/// reading the file microseconds after we closed it. Unix `unlink` needs no retry.
|
2026-08-02 19:04:30 -04:00
|
|
|
pub fn remove_file_retrying(path: &Path) -> std::io::Result<()> {
|
|
|
|
|
#[cfg(not(windows))]
|
|
|
|
|
{
|
|
|
|
|
std::fs::remove_file(path)
|
|
|
|
|
}
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
|
|
|
|
let deadline = std::time::Instant::now() + REMOVE_RETRY_BUDGET;
|
|
|
|
|
loop {
|
|
|
|
|
match std::fs::remove_file(path) {
|
|
|
|
|
Ok(()) => return Ok(()),
|
|
|
|
|
Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Err(e),
|
|
|
|
|
Err(e) => {
|
|
|
|
|
if std::time::Instant::now() >= deadline {
|
|
|
|
|
return Err(e);
|
|
|
|
|
}
|
|
|
|
|
std::thread::sleep(std::time::Duration::from_millis(25));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-08-23 00:33:53 -04:00
|
|
|
/// Deny read access to `dir`, for tests (exposed: `tests/full_index.rs` is a
|
|
|
|
|
/// separate crate). Windows uses `icacls`: a deny ACE binds even the owner
|
|
|
|
|
/// until [`restore_read`] rewrites it; neither call needs elevation.
|
2026-08-02 19:04:30 -04:00
|
|
|
#[doc(hidden)]
|
|
|
|
|
pub fn deny_read(dir: &Path) -> std::io::Result<()> {
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::unix::fs::PermissionsExt;
|
|
|
|
|
std::fs::set_permissions(dir, std::fs::Permissions::from_mode(0o000))
|
|
|
|
|
}
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
2026-08-04 03:27:05 -04:00
|
|
|
icacls(
|
|
|
|
|
dir,
|
|
|
|
|
&["/deny", &format!("{}:(OI)(CI)(RD)", current_user()?)],
|
|
|
|
|
)
|
2026-08-02 19:04:30 -04:00
|
|
|
}
|
|
|
|
|
#[cfg(not(any(unix, windows)))]
|
|
|
|
|
{
|
|
|
|
|
let _ = dir;
|
|
|
|
|
Err(std::io::Error::new(
|
|
|
|
|
std::io::ErrorKind::Unsupported,
|
|
|
|
|
"deny_read is not supported on this target",
|
|
|
|
|
))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Undo [`deny_read`] so the directory can be cleaned up.
|
|
|
|
|
#[doc(hidden)]
|
|
|
|
|
pub fn restore_read(dir: &Path) -> std::io::Result<()> {
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
|
{
|
|
|
|
|
use std::os::unix::fs::PermissionsExt;
|
|
|
|
|
std::fs::set_permissions(dir, std::fs::Permissions::from_mode(0o755))
|
|
|
|
|
}
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
{
|
|
|
|
|
icacls(dir, &["/remove:d", ¤t_user()?])
|
|
|
|
|
}
|
|
|
|
|
#[cfg(not(any(unix, windows)))]
|
|
|
|
|
{
|
|
|
|
|
let _ = dir;
|
|
|
|
|
Ok(())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
fn current_user() -> std::io::Result<String> {
|
|
|
|
|
match (std::env::var("USERDOMAIN"), std::env::var("USERNAME")) {
|
|
|
|
|
(Ok(domain), Ok(user)) => Ok(format!("{}\\{}", domain, user)),
|
|
|
|
|
(_, Ok(user)) => Ok(user),
|
|
|
|
|
_ => Err(std::io::Error::other("USERNAME is not set")),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
|
fn icacls(dir: &Path, args: &[&str]) -> std::io::Result<()> {
|
|
|
|
|
let out = std::process::Command::new("icacls")
|
|
|
|
|
.arg(dir)
|
|
|
|
|
.args(args)
|
|
|
|
|
.output()?;
|
|
|
|
|
if out.status.success() {
|
|
|
|
|
Ok(())
|
|
|
|
|
} else {
|
|
|
|
|
Err(std::io::Error::other(format!(
|
|
|
|
|
"icacls {}: {}",
|
|
|
|
|
dir.display(),
|
|
|
|
|
String::from_utf8_lossy(&out.stderr).trim()
|
|
|
|
|
)))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(test)]
|
2026-08-09 16:25:43 -04:00
|
|
|
#[path = "platform_tests.rs"]
|
|
|
|
|
mod tests;
|