1use std::path::{Path, PathBuf};
6
7use anyhow::{Result, bail};
8
9#[derive(Debug, Clone)]
11pub struct RepoRef {
12 pub name: String,
15 pub upstream_url: String,
17 pub cache_dir: PathBuf,
19}
20
21pub fn repo_name_from_path(path: &str, suffix: &str) -> Option<String> {
24 let repo = path
25 .trim_start_matches('/')
26 .strip_suffix(suffix.trim_start_matches('/'))?;
27 Some(repo.trim_end_matches('/').to_string())
28}
29
30pub const INCOMING_SUFFIX: &str = ".__incoming__";
35
36pub const EVICTING_SUFFIX: &str = ".__evicting__";
41
42pub fn resolve(name: &str, upstream_base: &str, cache_root: &Path) -> Result<RepoRef> {
45 if name.is_empty() || name.contains('\0') || name.starts_with('/') {
46 bail!("invalid repo path: {name:?}");
47 }
48 for comp in name.split('/') {
49 if comp.is_empty() || comp == "." || comp == ".." {
50 bail!("invalid repo path component in {name:?}");
51 }
52 }
53 if name.contains(INCOMING_SUFFIX) || name.contains(EVICTING_SUFFIX) {
57 bail!("invalid repo path (reserved suffix) in {name:?}");
58 }
59 let cache_dir = cache_root.join(name);
60 if !cache_dir.starts_with(cache_root) {
62 bail!("repo path escapes cache root: {name:?}");
63 }
64 let upstream_url = format!("{}/{}", upstream_base.trim_end_matches('/'), name);
65 Ok(RepoRef {
66 name: name.to_string(),
67 upstream_url,
68 cache_dir,
69 })
70}
71
72#[cfg(test)]
73mod tests {
74 use super::*;
75
76 #[test]
77 fn strips_suffixes() {
78 assert_eq!(
79 repo_name_from_path("/group/team/foo.git/info/refs", "/info/refs").as_deref(),
80 Some("group/team/foo.git")
81 );
82 assert_eq!(
83 repo_name_from_path("/a/b.git/git-upload-pack", "/git-upload-pack").as_deref(),
84 Some("a/b.git")
85 );
86 assert_eq!(repo_name_from_path("/nope", "/info/refs"), None);
87 }
88
89 #[test]
90 fn rejects_traversal() {
91 let root = Path::new("/cache");
92 assert!(resolve("../etc/passwd", "https://up", root).is_err());
93 assert!(resolve("a/../../b", "https://up", root).is_err());
94 assert!(resolve("/abs", "https://up", root).is_err());
95 let ok = resolve("g/r.git", "https://up/", root).unwrap();
96 assert_eq!(ok.upstream_url, "https://up/g/r.git");
97 assert_eq!(ok.cache_dir, Path::new("/cache/g/r.git"));
98 }
99
100 #[test]
101 fn rejects_reserved_suffixes() {
102 let root = Path::new("/cache");
103 assert!(resolve(&format!("foo{INCOMING_SUFFIX}"), "https://up", root).is_err());
107 assert!(resolve(&format!("a/b{INCOMING_SUFFIX}"), "https://up", root).is_err());
108 assert!(resolve(&format!("foo{EVICTING_SUFFIX}"), "https://up", root).is_err());
109 assert!(resolve(&format!("a/b{EVICTING_SUFFIX}"), "https://up", root).is_err());
110 }
111}