Skip to main content

lean_ctx/core/
pathutil.rs

1use std::path::{Path, PathBuf};
2
3/// Canonicalize a path and strip the Windows verbatim/extended-length prefix (`\\?\`)
4/// that `std::fs::canonicalize` adds on Windows. This prefix breaks many tools and
5/// string-based path comparisons.
6///
7/// On non-Windows platforms this is equivalent to `std::fs::canonicalize`.
8pub fn safe_canonicalize(path: &Path) -> std::io::Result<PathBuf> {
9    // TCC choke-point (#356): a launchd-standalone process (daemon/proxy/auto-
10    // updater, ppid 1) must never realpath a path under ~/Documents, ~/Desktop
11    // or ~/Downloads — the `stat` trips the macOS privacy prompt in lean-ctx's
12    // own name, and every release re-invalidates the grant (new cdhash), so it
13    // re-prompts forever. Heuristic call sites (project-root detection, session
14    // matching, path normalization, scan-root checks) all funnel through here,
15    // so guarding the sink protects them centrally instead of one opt-in check
16    // per call site. Return the path unchanged (lexical) rather than touching
17    // the filesystem. Security boundaries (PathJail) deliberately bypass this
18    // guard via `canonicalize_secure` — they only resolve paths the client
19    // explicitly asked to access, where a prompt is legitimate, and must keep
20    // resolving symlinks to detect jail escapes.
21    if !may_probe_path(path) {
22        return Ok(path.to_path_buf());
23    }
24    canonicalize_raw(path)
25}
26
27/// Raw realpath + Windows-verbatim strip, with **no** TCC guard. Internal sink
28/// shared by [`safe_canonicalize`] (which gates it behind `may_probe_path`) and
29/// [`canonicalize_secure`] (which never gates it).
30fn canonicalize_raw(path: &Path) -> std::io::Result<PathBuf> {
31    let canon = std::fs::canonicalize(path)?;
32    Ok(strip_verbatim(canon))
33}
34
35/// Like `safe_canonicalize` but returns the original path on failure.
36pub fn safe_canonicalize_or_self(path: &Path) -> PathBuf {
37    safe_canonicalize(path).unwrap_or_else(|_| path.to_path_buf())
38}
39
40/// SECURITY canonicalize: always resolves symlinks, even under ~/Documents in a
41/// launchd-standalone process. PathJail relies on this to detect symlink jail
42/// escapes (#356 must never weaken the security boundary). A standalone process
43/// only reaches here for a path the client *explicitly* asked to access, where a
44/// one-time TCC prompt is legitimate — unlike the self-initiated heuristic
45/// probes that [`safe_canonicalize`] suppresses.
46pub fn canonicalize_secure(path: &Path) -> std::io::Result<PathBuf> {
47    canonicalize_raw(path)
48}
49
50/// Like `canonicalize_secure` but returns the original path on failure.
51pub fn canonicalize_secure_or_self(path: &Path) -> PathBuf {
52    canonicalize_secure(path).unwrap_or_else(|_| path.to_path_buf())
53}
54
55/// Canonicalize with a timeout guard. Protects against hangs on WSL2 DrvFS,
56/// Windows reparse points, NFS, FUSE, sshfs, and other slow filesystems.
57/// Falls back to the original path if canonicalize doesn't complete within the timeout.
58/// Self-healing: after a timeout, subsequent calls to slow mounts skip the thread entirely.
59///
60/// Heuristic variant — honours the #356 TCC guard (see [`safe_canonicalize`]).
61pub fn safe_canonicalize_bounded(path: &Path, timeout_ms: u64) -> PathBuf {
62    canonicalize_bounded_with(path, timeout_ms, safe_canonicalize_or_self)
63}
64
65/// SECURITY variant of [`safe_canonicalize_bounded`] — bypasses the #356 TCC
66/// guard so PathJail keeps resolving symlinks to detect jail escapes. See
67/// [`canonicalize_secure`] for why a prompt here (explicit request) is legitimate.
68pub fn canonicalize_secure_bounded(path: &Path, timeout_ms: u64) -> PathBuf {
69    canonicalize_bounded_with(path, timeout_ms, canonicalize_secure_or_self)
70}
71
72/// Shared timeout machinery for the bounded canonicalizers. `resolve` selects
73/// the guarded (`safe_canonicalize_or_self`) or security (`canonicalize_secure_or_self`)
74/// sink so both variants get identical slow-mount/self-healing behaviour.
75fn canonicalize_bounded_with(
76    path: &Path,
77    timeout_ms: u64,
78    resolve: fn(&Path) -> PathBuf,
79) -> PathBuf {
80    use super::io_health;
81
82    let path_str = path.to_string_lossy();
83    if io_health::is_slow_mount(&path_str) && io_health::recent_freeze_count() > 0 {
84        return resolve(path);
85    }
86
87    let effective_timeout =
88        io_health::adaptive_timeout(std::time::Duration::from_millis(timeout_ms));
89
90    let path_owned = path.to_path_buf();
91    let (tx, rx) = std::sync::mpsc::channel();
92    let _ = std::thread::Builder::new()
93        .name("canonicalize-bounded".into())
94        .spawn(move || {
95            let _ = tx.send(resolve(&path_owned));
96        });
97    if let Ok(canonical) = rx.recv_timeout(effective_timeout) {
98        canonical
99    } else {
100        io_health::record_freeze();
101        tracing::warn!(
102            "[SECURITY] canonicalize timed out ({}ms) for {}; PathJail checks on \
103             uncanonicalized paths may be less reliable",
104            effective_timeout.as_millis(),
105            path.display()
106        );
107        path.to_path_buf()
108    }
109}
110
111/// Remove the `\\?\` / `//?/` verbatim prefix from a `PathBuf`.
112/// Handles both regular verbatim (`\\?\C:\...`) and UNC verbatim (`\\?\UNC\...`).
113pub fn strip_verbatim(path: PathBuf) -> PathBuf {
114    let s = path.to_string_lossy();
115    if let Some(stripped) = strip_verbatim_str(&s) {
116        PathBuf::from(stripped)
117    } else {
118        path
119    }
120}
121
122/// Remove the `\\?\` / `//?/` verbatim prefix from a path string.
123/// Returns `Some(cleaned)` if a prefix was found, `None` otherwise.
124pub fn strip_verbatim_str(path: &str) -> Option<String> {
125    let normalized = path.replace('\\', "/");
126
127    if let Some(rest) = normalized.strip_prefix("//?/UNC/") {
128        Some(format!("//{rest}"))
129    } else {
130        normalized
131            .strip_prefix("//?/")
132            .map(std::string::ToString::to_string)
133    }
134}
135
136/// MSYS2/Git Bash drive mapping: `/c/Users/...` -> `C:/Users/...`.
137///
138/// Returns `None` when the path does not carry a single-letter drive prefix.
139/// Callers must apply this **only on Windows hosts**: clients running under
140/// MSYS2/Git Bash hand POSIX-style drive paths to a native Windows lean-ctx.
141/// On Linux/macOS `/c/...` is a literal directory and must pass through
142/// untouched (GH #397 — the unconditional rewrite broke every `ctx_*` tool
143/// for Linux projects rooted under `/c/...` and similar paths).
144fn translate_msys_drive_prefix(p: &str) -> Option<String> {
145    if p.len() >= 3
146        && p.starts_with('/')
147        && p.as_bytes()[1].is_ascii_alphabetic()
148        && p.as_bytes()[2] == b'/'
149    {
150        let drive = p.as_bytes()[1].to_ascii_uppercase() as char;
151        Some(format!("{drive}:{}", &p[2..]))
152    } else {
153        None
154    }
155}
156
157/// Lexical (string-only) part of [`normalize_tool_path`]: MSYS2 drive prefix
158/// (Windows hosts only), separators, double slashes, trailing slash. Performs
159/// **no** filesystem access, so it is safe on persisted paths in
160/// TCC-standalone processes (launchd daemon, #356) and as a dedupe key where
161/// symlink resolution is not worth a `realpath` per entry.
162pub fn normalize_tool_path_lexical(path: &str) -> String {
163    let mut p = match strip_verbatim_str(path) {
164        Some(stripped) => stripped,
165        None => path.to_string(),
166    };
167
168    if cfg!(windows)
169        && let Some(translated) = translate_msys_drive_prefix(&p)
170    {
171        p = translated;
172    }
173
174    p = p.replace('\\', "/");
175
176    // Collapse double slashes (preserve UNC paths starting with //)
177    while p.contains("//") && !p.starts_with("//") {
178        p = p.replace("//", "/");
179    }
180
181    // Remove trailing slash (unless root like "/" or "C:/")
182    if p.len() > 1 && p.ends_with('/') && !p.ends_with(":/") {
183        p.pop();
184    }
185
186    p
187}
188
189/// Normalize paths from any client format to a consistent OS-native form.
190/// Handles MSYS2/Git Bash drive prefixes on Windows hosts
191/// (`/c/Users/...` -> `C:/Users/...`), mixed separators, double slashes, and
192/// trailing slashes. Uses forward slashes for consistency. On non-Windows
193/// hosts `/c/...` is a literal directory and passes through unchanged (#397).
194pub fn normalize_tool_path(path: &str) -> String {
195    let mut p = normalize_tool_path_lexical(path);
196
197    // Resolve symlinks for absolute paths to ensure cache key consistency.
198    // Skip relative paths (preserve "." / "../" as-is), root-only paths (/ or C:/),
199    // slow mounts (WSL DrvFS /mnt/) where canonicalize can hang, and paths a
200    // TCC-standalone process must not stat (launchd daemon + ~/Documents, #356).
201    // Uses safe_canonicalize to strip Windows \\?\ prefix.
202    let is_absolute = p.starts_with('/') || (p.len() >= 3 && p.as_bytes()[1] == b':');
203    let is_root_only = p == "/" || (p.len() <= 3 && p.ends_with('/') && is_absolute);
204    if is_absolute
205        && !is_root_only
206        && !crate::core::io_health::is_slow_mount(&p)
207        && may_probe_path(Path::new(&*p))
208        && let Ok(canonical) = safe_canonicalize(Path::new(&*p))
209    {
210        let canonical_str = canonical.to_string_lossy().replace('\\', "/");
211        if !canonical_str.is_empty() {
212            p = canonical_str;
213        }
214    }
215
216    p
217}
218
219/// Returns `true` if the directory is too broad to be a valid project root.
220/// Rejects home directory, filesystem root, `.` (bare CWD), and agent sandbox
221/// directories (`.claude`, `.codex`). Used to prevent writing project-scoped
222/// data (overlays, policies) into the global `~/.lean-ctx/` data directory.
223pub fn is_broad_or_unsafe_root(dir: &Path) -> bool {
224    if let Some(home) = dirs::home_dir()
225        && dir == home
226    {
227        return true;
228    }
229    let s = dir.to_string_lossy();
230    if s == "/" || s == "\\" || s == "." {
231        return true;
232    }
233    s.ends_with("/.claude")
234        || s.ends_with("/.codex")
235        || s.ends_with("/.codebuddy")
236        || s.contains("/.claude/")
237        || s.contains("/.codex/")
238        || s.contains("/.codebuddy/")
239}
240
241/// Well-known project markers used to identify project roots.
242pub const PROJECT_MARKERS: &[&str] = &[
243    ".git",
244    "Cargo.toml",
245    "package.json",
246    "go.mod",
247    "pyproject.toml",
248    "setup.py",
249    "pom.xml",
250    "build.gradle",
251    "Makefile",
252    "project.godot",
253    ".lean-ctx.toml",
254    ".planning",
255];
256
257/// Returns `true` if `dir` contains at least one known project marker.
258///
259/// TCC guard (#356): a launchd-owned process (daemon/proxy/auto-updater) must
260/// not stat marker files under `~/Documents` & co. — the probe itself pops the
261/// macOS privacy prompt. For those processes this conservatively reports
262/// "no marker" without touching the filesystem.
263pub fn has_project_marker(dir: &Path) -> bool {
264    if !may_probe_path(dir) {
265        return false;
266    }
267    PROJECT_MARKERS.iter().any(|m| dir.join(m).exists())
268}
269
270/// Returns `true` if the (lstat) metadata describes a symlink — or, on
271/// Windows, *any* reparse point (junctions, mount points, app-exec links).
272///
273/// Security boundaries must use this instead of `FileType::is_symlink`:
274/// Rust's `is_symlink()` reports `false` for NTFS junctions, which redirect
275/// exactly like directory symlinks and would otherwise bypass jail/TOCTOU
276/// checks on Windows (GL#442).
277pub fn is_symlink_or_reparse(meta: &std::fs::Metadata) -> bool {
278    if meta.file_type().is_symlink() {
279        return true;
280    }
281    #[cfg(windows)]
282    {
283        use std::os::windows::fs::MetadataExt;
284        const FILE_ATTRIBUTE_REPARSE_POINT: u32 = 0x0400;
285        return meta.file_attributes() & FILE_ATTRIBUTE_REPARSE_POINT != 0;
286    }
287    #[cfg(not(windows))]
288    false
289}
290
291/// Returns `true` if `dir` is the home directory or one of the macOS "magic"
292/// home subdirectories (`Documents`, `Desktop`, `Downloads`).
293///
294/// macOS guards these with TCC: the first time a process *enumerates or stats
295/// inside* one, the OS pops a privacy prompt ("lean-ctx would like to access
296/// files in your Documents folder", #356). They are also never valid project
297/// roots or multi-repo workspace parents, so scan heuristics should treat them
298/// as off-limits *without* calling `read_dir` (which is what trips the prompt).
299pub fn is_tcc_sensitive_home_dir(dir: &Path) -> bool {
300    let Some(home) = dirs::home_dir() else {
301        return false;
302    };
303    if dir == home {
304        return true;
305    }
306    if dir.parent() != Some(home.as_path()) {
307        return false;
308    }
309    matches!(
310        dir.file_name().and_then(|n| n.to_str()),
311        Some("Documents" | "Desktop" | "Downloads")
312    )
313}
314
315/// Returns `true` if `path` lies inside (or is) one of the macOS TCC-protected
316/// home folders (`~/Documents`, `~/Desktop`, `~/Downloads`). Pure string/path
317/// comparison — performs **no** filesystem access itself.
318///
319/// Unlike [`is_tcc_sensitive_home_dir`] (which only matches the magic dirs
320/// themselves), this also matches nested paths like `~/Documents/proj/src`,
321/// because *any* `stat` below the magic dir trips the TCC prompt (#356).
322pub fn is_under_tcc_protected_dir(path: &Path) -> bool {
323    if !cfg!(target_os = "macos") {
324        return false;
325    }
326    let Some(home) = dirs::home_dir() else {
327        return false;
328    };
329    ["Documents", "Desktop", "Downloads"]
330        .iter()
331        .any(|magic| path.starts_with(home.join(magic)))
332}
333
334/// Returns `true` when this process is its own TCC identity on macOS — i.e.
335/// it was started (or re-parented) by `launchd` rather than by a
336/// TCC-granted host like a terminal or an editor.
337///
338/// Context (#356): TCC permissions attach to the *responsible process*. The
339/// lean-ctx daemon/proxy LaunchAgents and the scheduled auto-updater run
340/// directly under `launchd` (ppid 1), so any `stat`/`read_dir` they perform
341/// under `~/Documents` pops the privacy prompt **in lean-ctx's own name** —
342/// and because every release replaces the ad-hoc-signed binary (new cdhash),
343/// a previously granted permission is invalidated on each update, re-prompting
344/// forever. Such processes must never probe TCC-protected paths on their own
345/// initiative. Child processes of a terminal or editor (MCP server, CLI)
346/// inherit their host's TCC grant and keep full functionality.
347pub fn process_is_tcc_standalone() -> bool {
348    #[cfg(target_os = "macos")]
349    {
350        // Deliberately uncached: getppid is a cheap syscall, the env override
351        // must stay testable within one process, and a daemonizing fork could
352        // change the answer after startup.
353        if let Ok(v) = std::env::var("LEAN_CTX_TCC_STANDALONE") {
354            match v.trim() {
355                "1" | "true" => return true,
356                "0" | "false" => return false,
357                _ => {}
358            }
359        }
360        // A process carrying the deny-~/Documents seatbelt sentinel is, by
361        // construction, a launchd-standalone descendant: the sentinel is set
362        // only by the LaunchAgent plist env and the self re-exec, and child
363        // processes inherit it. This catches a daemon the long-lived standalone
364        // proxy spawned via `start_daemon` (ppid = proxy, not 1), whose code-side
365        // path guards would otherwise stay off because `getppid()` is no longer
366        // 1. (#356)
367        if std::env::var_os(crate::core::tcc_guard_sandbox::SEATBELT_SENTINEL).is_some() {
368            return true;
369        }
370        // SAFETY: `getppid` takes no arguments and cannot fail.
371        (unsafe { libc::getppid() }) == 1
372    }
373    #[cfg(not(target_os = "macos"))]
374    {
375        false
376    }
377}
378
379/// Returns `true` when this process may `stat`/`read_dir`/`canonicalize`
380/// `path` without risking a macOS TCC privacy prompt in lean-ctx's name.
381///
382/// Heuristic call sites (project-marker probes, session/root matching) must
383/// consult this before touching paths from persisted state; security
384/// boundaries (PathJail) are exempt — they only ever canonicalize paths the
385/// client explicitly asked to access, in which case a prompt is legitimate.
386pub fn may_probe_path(path: &Path) -> bool {
387    !(process_is_tcc_standalone() && is_under_tcc_protected_dir(path))
388}
389
390/// Returns `true` if `dir` is a multi-repo workspace parent — i.e. it has at
391/// least 2 immediate child directories that each contain a project marker.
392pub fn has_multi_repo_children(dir: &Path) -> bool {
393    // Never enumerate the home dir or macOS TCC-protected dirs: read_dir there
394    // pops a macOS privacy prompt (#356) and they are never workspace parents.
395    // `is_tcc_sensitive_home_dir` only matches the magic dirs themselves;
396    // `!may_probe_path` additionally refuses *nested* paths like
397    // `~/Documents/proj` when this process is launchd-standalone.
398    if is_tcc_sensitive_home_dir(dir) || !may_probe_path(dir) {
399        return false;
400    }
401    let Ok(entries) = std::fs::read_dir(dir) else {
402        return false;
403    };
404    let count = entries
405        .filter_map(Result::ok)
406        .filter(|e| e.file_type().is_ok_and(|ft| ft.is_dir()))
407        .filter(|e| has_project_marker(&e.path()))
408        .take(2)
409        .count();
410    count >= 2
411}
412
413/// Returns `true` if `project_root` collides with the lean-ctx data directory.
414/// This prevents project-scoped files (overlays.json, policies.json) from being
415/// written into `~/.lean-ctx/` or `~/.config/lean-ctx/`.
416pub fn is_data_dir_collision(project_root: &Path) -> bool {
417    if is_broad_or_unsafe_root(project_root) {
418        return true;
419    }
420    if let Ok(data_dir) = crate::core::data_dir::lean_ctx_data_dir() {
421        let project_lean_ctx = project_root.join(".lean-ctx");
422        if project_lean_ctx == data_dir || data_dir.starts_with(&project_lean_ctx) {
423            return true;
424        }
425    }
426    false
427}
428
429/// Returns the project-scoped `.lean-ctx/` directory if the project root is safe.
430/// Returns `Err` if the project root collides with the global data directory.
431pub fn safe_project_data_dir(project_root: &Path) -> Result<PathBuf, String> {
432    if is_data_dir_collision(project_root) {
433        return Err(format!(
434            "project root {} collides with global data directory; \
435             skipping project-scoped write",
436            project_root.display()
437        ));
438    }
439    Ok(project_root.join(".lean-ctx"))
440}
441
442#[cfg(test)]
443mod tests {
444    use super::*;
445
446    #[test]
447    fn strip_regular_verbatim() {
448        let p = PathBuf::from(r"\\?\C:\Users\dev\project");
449        let result = strip_verbatim(p);
450        assert_eq!(result, PathBuf::from("C:/Users/dev/project"));
451    }
452
453    #[test]
454    fn tcc_sensitive_home_dir_matches_home_and_magic_dirs() {
455        let Some(home) = dirs::home_dir() else {
456            return;
457        };
458        // Home itself and the macOS magic dirs are off-limits (#356).
459        assert!(is_tcc_sensitive_home_dir(&home));
460        assert!(is_tcc_sensitive_home_dir(&home.join("Documents")));
461        assert!(is_tcc_sensitive_home_dir(&home.join("Desktop")));
462        assert!(is_tcc_sensitive_home_dir(&home.join("Downloads")));
463    }
464
465    #[test]
466    fn tcc_sensitive_home_dir_allows_real_projects() {
467        let Some(home) = dirs::home_dir() else {
468            return;
469        };
470        // A real project (even nested under Documents) and non-magic home children
471        // are scannable — only the bare magic dirs / home are blocked.
472        assert!(!is_tcc_sensitive_home_dir(
473            &home.join("Documents").join("my-project")
474        ));
475        assert!(!is_tcc_sensitive_home_dir(&home.join("code")));
476        assert!(!is_tcc_sensitive_home_dir(&home.join("Projects")));
477    }
478
479    #[test]
480    #[cfg(target_os = "macos")]
481    fn under_tcc_protected_dir_matches_nested_paths() {
482        let Some(home) = dirs::home_dir() else {
483            return;
484        };
485        // The magic dirs themselves and anything nested below them (#356).
486        assert!(is_under_tcc_protected_dir(&home.join("Documents")));
487        assert!(is_under_tcc_protected_dir(
488            &home.join("Documents/deep/nested/project")
489        ));
490        assert!(is_under_tcc_protected_dir(&home.join("Desktop/scratch")));
491        assert!(is_under_tcc_protected_dir(&home.join("Downloads/x.zip")));
492        // Home itself, siblings, and non-home paths are fine.
493        assert!(!is_under_tcc_protected_dir(&home));
494        assert!(!is_under_tcc_protected_dir(&home.join("code/project")));
495        assert!(!is_under_tcc_protected_dir(Path::new("/tmp/Documents")));
496    }
497
498    #[test]
499    #[cfg(target_os = "macos")]
500    #[serial_test::serial]
501    fn tcc_standalone_blocks_probes_under_protected_dirs() {
502        let Some(home) = dirs::home_dir() else {
503            return;
504        };
505        let doc_proj = home.join("Documents/some-project");
506
507        crate::test_env::set_var("LEAN_CTX_TCC_STANDALONE", "1");
508        assert!(process_is_tcc_standalone());
509        assert!(!may_probe_path(&doc_proj));
510        // Non-protected paths stay probeable even for standalone processes.
511        assert!(may_probe_path(Path::new("/tmp/some-project")));
512        // has_project_marker must refuse without touching the filesystem.
513        assert!(!has_project_marker(&doc_proj));
514
515        crate::test_env::set_var("LEAN_CTX_TCC_STANDALONE", "0");
516        assert!(!process_is_tcc_standalone());
517        assert!(may_probe_path(&doc_proj));
518        crate::test_env::remove_var("LEAN_CTX_TCC_STANDALONE");
519    }
520
521    #[test]
522    #[cfg(target_os = "macos")]
523    #[serial_test::serial]
524    fn tcc_standalone_detected_via_seatbelt_sentinel() {
525        let Some(home) = dirs::home_dir() else {
526            return;
527        };
528        let doc_proj = home.join("Documents/some-project");
529
530        // No explicit override: a process carrying the deny-~/Documents seatbelt
531        // sentinel (inherited from its sandboxed launchd parent) counts as
532        // standalone even when ppid != 1, so its heuristic probes stay
533        // suppressed — this is the proxy→daemon chain the ppid check missed. (#356)
534        crate::test_env::remove_var("LEAN_CTX_TCC_STANDALONE");
535        crate::test_env::set_var(crate::core::tcc_guard_sandbox::SEATBELT_SENTINEL, "1");
536        assert!(process_is_tcc_standalone());
537        assert!(!may_probe_path(&doc_proj));
538        crate::test_env::remove_var(crate::core::tcc_guard_sandbox::SEATBELT_SENTINEL);
539
540        // With neither override nor sentinel a normal test process (ppid != 1)
541        // is not standalone, so the sentinel is what flipped the result above.
542        assert!(!process_is_tcc_standalone());
543    }
544
545    #[test]
546    #[cfg(target_os = "macos")]
547    #[serial_test::serial]
548    fn tcc_standalone_skips_canonicalize_under_protected_dirs() {
549        let Some(home) = dirs::home_dir() else {
550            return;
551        };
552        // A path that does NOT exist under ~/Documents. With the TCC choke-point
553        // guard active, `safe_canonicalize` returns Ok(input) *without* calling
554        // `std::fs::canonicalize` (which would Err on a missing path) — proving
555        // the filesystem is never touched (#356). This is the structural fix:
556        // every heuristic canonicalize funnels through here.
557        let missing = home.join("Documents/lean-ctx-tcc-test-does-not-exist-xyzzy");
558
559        crate::test_env::set_var("LEAN_CTX_TCC_STANDALONE", "1");
560        let guarded = safe_canonicalize(&missing);
561        assert!(
562            guarded.is_ok(),
563            "standalone safe_canonicalize must short-circuit (no stat) under ~/Documents"
564        );
565        assert_eq!(guarded.unwrap(), missing);
566        assert_eq!(safe_canonicalize_or_self(&missing), missing);
567
568        // Outside the protected dirs the guard never engages, even when standalone.
569        let tmp_missing = Path::new("/tmp/lean-ctx-tcc-test-does-not-exist-xyzzy");
570        assert!(safe_canonicalize(tmp_missing).is_err());
571
572        // Without standalone the guard is inactive: a missing ~/Documents path
573        // Errs from the real `std::fs::canonicalize` as before.
574        crate::test_env::set_var("LEAN_CTX_TCC_STANDALONE", "0");
575        assert!(safe_canonicalize(&missing).is_err());
576
577        crate::test_env::remove_var("LEAN_CTX_TCC_STANDALONE");
578    }
579
580    #[test]
581    #[cfg(target_os = "macos")]
582    #[serial_test::serial]
583    fn canonicalize_secure_bypasses_tcc_guard_for_pathjail() {
584        // SECURITY counterpart to the test above (#356): PathJail must keep
585        // resolving symlinks even when standalone under ~/Documents, so the jail
586        // can detect escapes. `canonicalize_secure` therefore must NOT honour the
587        // guard — it always touches the filesystem. We prove that by feeding a
588        // missing ~/Documents path while standalone: the guarded path returns
589        // Ok(lexical) (no stat), while the secure path Errs (it did stat).
590        let Some(home) = dirs::home_dir() else {
591            return;
592        };
593        let missing = home.join("Documents/lean-ctx-secure-canon-does-not-exist-xyzzy");
594
595        crate::test_env::set_var("LEAN_CTX_TCC_STANDALONE", "1");
596        // Guarded sink short-circuits (no fs access).
597        assert_eq!(safe_canonicalize(&missing).unwrap(), missing);
598        // Security sink ignores the guard and actually stats -> Err on a missing
599        // path. If this ever returns Ok(lexical), the jail's symlink-escape
600        // detection has silently regressed under ~/Documents.
601        assert!(
602            canonicalize_secure(&missing).is_err(),
603            "canonicalize_secure must bypass the TCC guard and touch the filesystem"
604        );
605        assert_eq!(canonicalize_secure_or_self(&missing), missing);
606        crate::test_env::remove_var("LEAN_CTX_TCC_STANDALONE");
607    }
608
609    #[test]
610    fn strip_unc_verbatim() {
611        let p = PathBuf::from(r"\\?\UNC\server\share\dir");
612        let result = strip_verbatim(p);
613        assert_eq!(result, PathBuf::from("//server/share/dir"));
614    }
615
616    #[test]
617    fn no_prefix_unchanged() {
618        let p = PathBuf::from("/home/user/project");
619        let result = strip_verbatim(p.clone());
620        assert_eq!(result, p);
621    }
622
623    #[test]
624    fn windows_drive_unchanged() {
625        let p = PathBuf::from("C:/Users/dev");
626        let result = strip_verbatim(p.clone());
627        assert_eq!(result, p);
628    }
629
630    #[test]
631    fn strip_str_regular() {
632        assert_eq!(
633            strip_verbatim_str(r"\\?\E:\code\lean-ctx"),
634            Some("E:/code/lean-ctx".to_string())
635        );
636    }
637
638    #[test]
639    fn strip_str_unc() {
640        assert_eq!(
641            strip_verbatim_str(r"\\?\UNC\myserver\data"),
642            Some("//myserver/data".to_string())
643        );
644    }
645
646    #[test]
647    fn strip_str_forward_slash_variant() {
648        assert_eq!(
649            strip_verbatim_str("//?/C:/Users/dev"),
650            Some("C:/Users/dev".to_string())
651        );
652    }
653
654    #[test]
655    fn strip_str_no_prefix() {
656        assert_eq!(strip_verbatim_str("/home/user"), None);
657    }
658
659    #[test]
660    fn safe_canonicalize_or_self_nonexistent() {
661        let p = Path::new("/this/path/should/not/exist/xyzzy");
662        let result = safe_canonicalize_or_self(p);
663        assert_eq!(result, p.to_path_buf());
664    }
665
666    // The drive translation itself is platform-independent and testable
667    // everywhere; only its *application* is gated on Windows hosts (#397).
668    #[test]
669    fn msys_drive_prefix_translation() {
670        assert_eq!(
671            translate_msys_drive_prefix("/c/Users/ABC").as_deref(),
672            Some("C:/Users/ABC")
673        );
674        assert_eq!(
675            translate_msys_drive_prefix("/D/Program Files").as_deref(),
676            Some("D:/Program Files")
677        );
678        assert_eq!(translate_msys_drive_prefix("/usr/local/bin"), None);
679        assert_eq!(translate_msys_drive_prefix("/c"), None);
680        assert_eq!(translate_msys_drive_prefix("c/Users"), None);
681    }
682
683    #[cfg(windows)]
684    #[test]
685    fn normalize_msys_path_to_native() {
686        assert_eq!(
687            normalize_tool_path("/c/Users/ABC/AppData/lean-ctx"),
688            "C:/Users/ABC/AppData/lean-ctx"
689        );
690        assert_eq!(
691            normalize_tool_path("/D/Program Files/lean-ctx.exe"),
692            "D:/Program Files/lean-ctx.exe"
693        );
694    }
695
696    // GH #397: on Linux/macOS, /c/… is a literal directory — a Linux project
697    // rooted there must not be rewritten to a Windows drive path.
698    #[cfg(not(windows))]
699    #[test]
700    fn normalize_single_letter_unix_path_untouched() {
701        assert_eq!(
702            normalize_tool_path_lexical("/c/Users/me/proj"),
703            "/c/Users/me/proj"
704        );
705        assert_eq!(
706            normalize_tool_path_lexical("/x/projects/app/src"),
707            "/x/projects/app/src"
708        );
709    }
710
711    #[test]
712    fn normalize_native_windows_path_unchanged() {
713        assert_eq!(
714            normalize_tool_path("C:/Users/ABC/lean-ctx.exe"),
715            "C:/Users/ABC/lean-ctx.exe"
716        );
717    }
718
719    #[test]
720    fn normalize_backslash_windows_path() {
721        assert_eq!(
722            normalize_tool_path(r"C:\Users\ABC\lean-ctx.exe"),
723            "C:/Users/ABC/lean-ctx.exe"
724        );
725    }
726
727    #[test]
728    fn normalize_unix_path_unchanged() {
729        assert_eq!(
730            normalize_tool_path("/usr/local/bin/lean-ctx"),
731            "/usr/local/bin/lean-ctx"
732        );
733    }
734
735    #[test]
736    fn normalize_windows_path_with_spaces_and_backslashes() {
737        // The exact "paths with spaces" scenario reported on Windows (#324):
738        // backslashes are converted to forward slashes (so client render layers
739        // never escape-mangle them) while spaces in directory names survive.
740        assert_eq!(
741            normalize_tool_path(r"C:\Users\My Name\My Project\src\main.rs"),
742            "C:/Users/My Name/My Project/src/main.rs"
743        );
744        assert_eq!(
745            normalize_tool_path(r"C:\Program Files\app\config.toml"),
746            "C:/Program Files/app/config.toml"
747        );
748    }
749
750    #[test]
751    fn normalize_double_slashes() {
752        assert_eq!(
753            normalize_tool_path("C:/Users//ABC//lean-ctx"),
754            "C:/Users/ABC/lean-ctx"
755        );
756    }
757
758    #[test]
759    fn normalize_trailing_slash_removed() {
760        assert_eq!(normalize_tool_path("C:/Users/ABC/"), "C:/Users/ABC");
761        assert_eq!(
762            normalize_tool_path_lexical("/tmp/nonexistent-dir-xyzzy/"),
763            "/tmp/nonexistent-dir-xyzzy"
764        );
765    }
766
767    #[test]
768    fn normalize_root_slash_preserved() {
769        assert_eq!(normalize_tool_path("/"), "/");
770    }
771
772    #[test]
773    fn normalize_drive_root_preserved() {
774        assert_eq!(normalize_tool_path("C:/"), "C:/");
775    }
776
777    #[test]
778    fn normalize_verbatim_with_msys() {
779        assert_eq!(normalize_tool_path(r"\\?\C:\Users\dev"), "C:/Users/dev");
780    }
781
782    #[test]
783    fn broad_root_rejects_home() {
784        if let Some(home) = dirs::home_dir() {
785            assert!(is_broad_or_unsafe_root(&home));
786        }
787    }
788
789    #[test]
790    fn broad_root_rejects_filesystem_root() {
791        assert!(is_broad_or_unsafe_root(Path::new("/")));
792    }
793
794    #[test]
795    fn broad_root_rejects_dot() {
796        assert!(is_broad_or_unsafe_root(Path::new(".")));
797    }
798
799    #[test]
800    fn broad_root_rejects_agent_dirs() {
801        assert!(is_broad_or_unsafe_root(Path::new("/home/user/.claude")));
802        assert!(is_broad_or_unsafe_root(Path::new("/home/user/.codex")));
803    }
804
805    #[test]
806    fn broad_root_allows_project_subdir() {
807        let tmp = tempfile::tempdir().unwrap();
808        let subdir = tmp.path().join("my-project");
809        std::fs::create_dir_all(&subdir).unwrap();
810        assert!(!is_broad_or_unsafe_root(&subdir));
811    }
812
813    #[test]
814    fn broad_root_allows_home_subdirs() {
815        if let Some(home) = dirs::home_dir() {
816            let subdir = home.join("projects").join("my-app");
817            assert!(!is_broad_or_unsafe_root(&subdir));
818        }
819    }
820
821    #[test]
822    fn data_dir_collision_rejects_home() {
823        if let Some(home) = dirs::home_dir() {
824            assert!(is_data_dir_collision(&home));
825        }
826    }
827
828    #[test]
829    fn data_dir_collision_allows_normal_project() {
830        let tmp = tempfile::tempdir().unwrap();
831        let project = tmp.path().join("my-project");
832        std::fs::create_dir_all(&project).unwrap();
833        assert!(!is_data_dir_collision(&project));
834    }
835
836    #[test]
837    fn has_project_marker_detects_git() {
838        let tmp = tempfile::tempdir().unwrap();
839        let root = tmp.path().join("repo");
840        std::fs::create_dir_all(&root).unwrap();
841        assert!(!has_project_marker(&root));
842        std::fs::create_dir(root.join(".git")).unwrap();
843        assert!(has_project_marker(&root));
844    }
845
846    #[test]
847    fn has_project_marker_detects_cargo_toml() {
848        let tmp = tempfile::tempdir().unwrap();
849        let root = tmp.path().join("rust-project");
850        std::fs::create_dir_all(&root).unwrap();
851        std::fs::write(root.join("Cargo.toml"), "[package]").unwrap();
852        assert!(has_project_marker(&root));
853    }
854
855    #[test]
856    fn has_project_marker_detects_godot_project() {
857        let tmp = tempfile::tempdir().unwrap();
858        let root = tmp.path().join("godot-game");
859        std::fs::create_dir_all(&root).unwrap();
860        std::fs::write(root.join("project.godot"), "config_version=5\n").unwrap();
861        assert!(has_project_marker(&root));
862    }
863
864    #[test]
865    fn multi_repo_children_needs_two() {
866        let tmp = tempfile::tempdir().unwrap();
867        let parent = tmp.path().join("code");
868        std::fs::create_dir_all(&parent).unwrap();
869
870        // 0 repos → false
871        assert!(!has_multi_repo_children(&parent));
872
873        // 1 repo → false
874        let repo1 = parent.join("repo1");
875        std::fs::create_dir_all(repo1.join(".git")).unwrap();
876        assert!(!has_multi_repo_children(&parent));
877
878        // 2 repos → true
879        let repo2 = parent.join("repo2");
880        std::fs::create_dir_all(repo2.join(".git")).unwrap();
881        assert!(has_multi_repo_children(&parent));
882    }
883
884    #[test]
885    fn multi_repo_children_ignores_files() {
886        let tmp = tempfile::tempdir().unwrap();
887        let parent = tmp.path().join("mixed");
888        std::fs::create_dir_all(&parent).unwrap();
889
890        // One repo dir + one plain file with .git name (not a dir)
891        let repo1 = parent.join("repo1");
892        std::fs::create_dir_all(repo1.join(".git")).unwrap();
893        std::fs::write(parent.join("not-a-repo"), "file").unwrap();
894        assert!(!has_multi_repo_children(&parent));
895
896        // Add second actual repo
897        let repo2 = parent.join("repo2");
898        std::fs::create_dir_all(&repo2).unwrap();
899        std::fs::write(repo2.join("package.json"), "{}").unwrap();
900        assert!(has_multi_repo_children(&parent));
901    }
902
903    #[test]
904    fn multi_repo_children_nonexistent_dir() {
905        assert!(!has_multi_repo_children(Path::new("/nonexistent/path/xyz")));
906    }
907
908    #[test]
909    fn regular_file_is_not_symlink_or_reparse() {
910        let tmp = tempfile::tempdir().unwrap();
911        let file = tmp.path().join("plain.txt");
912        std::fs::write(&file, "x").unwrap();
913        let meta = std::fs::symlink_metadata(&file).unwrap();
914        assert!(!is_symlink_or_reparse(&meta));
915    }
916
917    #[cfg(unix)]
918    #[test]
919    fn unix_symlink_is_detected() {
920        let tmp = tempfile::tempdir().unwrap();
921        let target = tmp.path().join("target.txt");
922        std::fs::write(&target, "x").unwrap();
923        let link = tmp.path().join("link.txt");
924        std::os::unix::fs::symlink(&target, &link).unwrap();
925        let meta = std::fs::symlink_metadata(&link).unwrap();
926        assert!(is_symlink_or_reparse(&meta));
927    }
928
929    /// Runs in the windows-latest CI lane (GL#442). Symlink creation needs
930    /// either admin or Developer Mode — skip gracefully when unavailable.
931    #[cfg(windows)]
932    #[test]
933    fn windows_symlink_is_detected() {
934        let tmp = tempfile::tempdir().unwrap();
935        let target = tmp.path().join("target.txt");
936        std::fs::write(&target, "x").unwrap();
937        let link = tmp.path().join("link.txt");
938        if std::os::windows::fs::symlink_file(&target, &link).is_err() {
939            eprintln!("skipping: symlink creation not permitted on this runner");
940            return;
941        }
942        let meta = std::fs::symlink_metadata(&link).unwrap();
943        assert!(is_symlink_or_reparse(&meta));
944    }
945}