treewide: refactor mkdir and read_file

mkdir and read_file augment the error case with the path in the error
case. This is useful as we otherwise don't know which location failed.
This commit refactors the functions to take any argument that coerces
into a Path - which is implemented by many types.

mkdir now uses create_dir_all as it is not meant to fail if the
directory already exists.

Signed-off-by: Stefan Kerkmann <[email protected]>

Author: KarlK90

src/main.rs

@@ -5,7 +5,7 @@ use std::env;
 use std::env::current_exe;
 use std::ffi::CString;
 use std::fmt::Write as _;
-use std::fs::{create_dir, read_to_string, File, OpenOptions};
+use std::fs::{File, OpenOptions};
 use std::io;
 use std::io::Write as _;
 use std::os::fd::{AsFd, AsRawFd, RawFd};
@@ -25,6 +25,7 @@ use nix::unistd::{chdir, chroot, dup2, execv, unlink};
 use systemd::{mount_systemd, shutdown};
 #[cfg(feature = "usb9pfs")]
 use usbg_9pfs::prepare_9pfs_gadget;
+use util::read_file;
 
 mod cmdline;
 #[cfg(feature = "dmverity")]
@@ -34,22 +35,10 @@ mod mount;
 mod systemd;
 #[cfg(feature = "usb9pfs")]
 mod usbg_9pfs;
+mod util;
 
 type Result<T> = std::result::Result<T, Box<dyn std::error::Error>>;
 
-pub fn mkdir(dir: &str) -> Result<()> {
-    if let Err(e) = create_dir(dir) {
-        if e.kind() != io::ErrorKind::AlreadyExists {
-            return Err(format!("Failed to create {dir}: {e}",).into());
-        }
-    }
-    Ok(())
-}
-
-fn read_file(filename: &str) -> std::result::Result<String, String> {
-    read_to_string(filename).map_err(|e| format!("Failed to read {filename}: {e}"))
-}
-
 /*
  * Setup stdout/stderr. The kernel will create /dev/console in the
  * initramfs, so we can use that.

src/mount.rs

@@ -7,7 +7,8 @@ use log::debug;
 use nix::mount::{mount, MsFlags};
 
 use crate::cmdline::CmdlineOptions;
-use crate::{mkdir, Result};
+use crate::util::mkdir;
+use crate::Result;
 
 pub fn do_mount(
     src: Option<&str>,

src/systemd.rs

@@ -10,7 +10,8 @@ use nix::sys::reboot::{reboot, RebootMode};
 
 use crate::cmdline::CmdlineOptions;
 use crate::mount::do_mount;
-use crate::{mkdir, Result};
+use crate::util::mkdir;
+use crate::Result;
 
 pub fn mount_systemd(options: &mut CmdlineOptions) -> Result<()> {
     do_mount(

src/usbg_9pfs.rs

@@ -9,7 +9,8 @@ use log::debug;
 
 use crate::cmdline::CmdlineOptions;
 use crate::mount::mount_apivfs;
-use crate::{mkdir, Result};
+use crate::util::mkdir;
+use crate::Result;
 
 fn write_file<C: AsRef<[u8]>>(path: &str, content: C) -> Result<()> {
     write(path, content).map_err(|e| format!("Failed to write to {path}: {e}").into())

src/util.rs

@@ -0,0 +1,24 @@
+use std::{
+    fs::{create_dir_all, read_to_string},
+    path::Path,
+};
+
+use crate::Result;
+
+pub(crate) fn mkdir(dir: impl AsRef<Path>) -> Result<()> {
+    Ok(create_dir_all(dir.as_ref()).map_err(|e| {
+        format!(
+            "Failed to create directory {}: {e}",
+            dir.as_ref().to_string_lossy()
+        )
+    })?)
+}
+
+pub(crate) fn read_file(filename: impl AsRef<Path>) -> Result<String> {
+    Ok(read_to_string(filename.as_ref()).map_err(|e| {
+        format!(
+            "Failed to read {}: {e}",
+            filename.as_ref().to_string_lossy()
+        )
+    })?)
+}