fixed relative path issue, caused by chroot

src/bin/pf2bodyfile/main.rs

@@ -1,26 +1,67 @@
 mod cli;
-use std::path::Path;
 
 use cli::Cli;
 use dfir_toolkit::common::bodyfile::Bodyfile3Line;
 use dfir_toolkit::common::FancyParser;
 use forensic_rs::prelude::*;
 use frnsc_prefetch::prelude::*;
+use log::{error, warn};
+use std::path::Path;
 
 fn main() -> anyhow::Result<()> {
     let cli = Cli::parse_cli();
 
+    if cli.prefetch_files().iter().any(|f| !f.path().exists()) {
+        anyhow::bail!("some files you specified do not exist");
+    }
     if cli.prefetch_files().iter().any(|f| !f.path().is_file()) {
-        anyhow::bail!("{} you must specify a file", env!("CARGO_BIN_NAME"));
+        anyhow::bail!("some paths you specified are no files");
     }
-    let mut fs = ChRootFileSystem::new(Path::new("."), Box::new(StdVirtualFS::new()));
+
+    let vfs = Box::new(StdVirtualFS::new());
 
     for input in cli.prefetch_files().iter() {
+        match input.parent() {
+            Some(parent) => {
+                let mut fs = ChRootFileSystem::new(parent, vfs.clone());
+                if let Some(pf_os_filename) = input.path().file_name() {
+                    if let Some(pf_filename) = pf_os_filename.to_str() {
+                        let pf_file = read_prefetch_file(
+                            pf_filename,
+                            fs.open(Path::new(&pf_filename.to_string()))?,
+                        )?;
 
-        let pf_file_name = input.path().file_name().unwrap().to_string_lossy();
-        let pf_file = read_prefetch_file(&pf_file_name, fs.open(input.path())?)?;
+                        pf_file.display_prefetch_file(pf_filename, *cli.include_metrics())?;
+                    } else {
+                        error!("invalid Unicode characters in filename: '{pf_os_filename:?}'")
+                    }
+                } else {
+                    warn!("unable to handle directories; you must specify concrete file names");
+                }
+            }
+            None => {
+                error!("specified path has no parent: {input}")
+            }
+        }
+    }
+    Ok(())
+}
 
-        for time in pf_file.last_run_times {
+trait DisplayPrefetchFile {
+    fn display_prefetch_file(
+        &self,
+        pf_file_name: &str,
+        include_metrics: bool,
+    ) -> anyhow::Result<()>;
+}
+
+impl DisplayPrefetchFile for PrefetchFile {
+    fn display_prefetch_file(
+        &self,
+        pf_file_name: &str,
+        include_metrics: bool,
+    ) -> anyhow::Result<()> {
+        for time in &self.last_run_times {
             let accessed =
                 winstructs::timestamp::WinTimestamp::new(&time.filetime().to_le_bytes())?
                     .to_datetime()
@@ -29,25 +70,24 @@ fn main() -> anyhow::Result<()> {
             let bf_line = Bodyfile3Line::new()
                 .with_owned_name(format!(
                     "Prefetch: run '{}' (run {} times, read from '{pf_file_name}')",
-                    pf_file.name, pf_file.run_count
+                    self.name, self.run_count
                 ))
                 .with_atime(accessed);
             println!("{bf_line}");
 
-            if *cli.include_metrics() {
-                for metric in &pf_file.metrics {
+            if include_metrics {
+                for metric in &self.metrics {
                     let mf = &metric.file;
                     let bf_line = Bodyfile3Line::new()
                         .with_owned_name(format!(
                             "Prefetch: running '{} possibly loaded '{mf}', read from '{pf_file_name}')",
-                            pf_file.name
+                            self.name
                         ))
                         .with_atime(accessed);
                     println!("{bf_line}");
                 }
             }
         }
+        Ok(())
     }
-    Ok(())
 }
-