#50: Check .reg files for ad hoc Wine prefixes

mtkennerly · Feb 17, 2021 · 7c1b5a1 · 7c1b5a1
1 parent 78c0727
commit 7c1b5a1
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/src/prelude.rs b/src/prelude.rs
@@ -363,21 +363,26 @@ pub fn scan_game_for_backup(
     #[allow(unused_mut)]
     let mut found_registry_keys = std::collections::HashSet::new();
 
+    let mut paths_to_check = std::collections::HashSet::<StrictPath>::new();
+
     // Add a dummy root for checking paths without `<root>`.
     let mut roots_to_check: Vec<RootsConfig> = vec![RootsConfig {
         path: StrictPath::new(SKIP.to_string()),
         store: Store::Other,
     }];
     roots_to_check.extend(roots.iter().cloned());
+
     if let Some(wp) = wine_prefix {
         roots_to_check.push(RootsConfig {
             path: wp.clone(),
             store: Store::OtherWine,
         });
+        paths_to_check.insert(StrictPath::relative(
+            format!("{}/*.reg", wp.interpret()),
+            Some(manifest_dir.interpret()),
+        ));
     }
 
-    let mut paths_to_check = std::collections::HashSet::<StrictPath>::new();
-
     for root in &roots_to_check {
         if root.path.raw().trim().is_empty() {
             continue;
@@ -1038,6 +1043,11 @@ mod tests {
                         size: 0,
                         original_path: None,
                     },
+                    ScannedFile {
+                        path: StrictPath::new(format!("{}/tests/wine-prefix/user.reg", repo())),
+                        size: 92,
+                        original_path: None,
+                    },
                 },
                 found_registry_keys: hashset! {},
                 registry_file: None,

diff --git a/tests/wine-prefix/user.reg b/tests/wine-prefix/user.reg
@@ -0,0 +1,4 @@
+Windows Registry Editor Version 5.00
+
+[HKEY_CURRENT_USER\Software\Ludusavi]
+"foo"="bar"