ksud: Add omitted files

userspace/ksud/src/event.rs

@@ -0,0 +1,113 @@
+use std::path::{Path, PathBuf};
+
+use crate::{defs, utils::mount_image};
+use anyhow::{bail, Result};
+use subprocess::Exec;
+
+pub fn on_post_data_fs() -> Result<()> {
+    let module_update_img = defs::MODULE_UPDATE_IMG;
+    let module_img = defs::MODULE_IMG;
+    let module_dir = defs::MODULE_DIR;
+    let module_update_flag = Path::new(defs::WORKING_DIR).join(defs::UPDATE_FILE_NAME);
+
+    // modules.img is the default image
+    let mut target_update_img = &module_img;
+
+    if Path::new(module_update_img).exists() && module_update_flag.exists() {
+        // if modules_update.img exists, and the the flag indicate this is an update
+        // this make sure that if the update failed, we will fallback to the old image
+        // if we boot succeed, we will rename the modules_update.img to modules.img #on_boot_complete
+        target_update_img = &module_update_img;
+        // And we should delete the flag immediately
+        std::fs::remove_file(module_update_flag)?;
+    }
+
+    if !Path::new(target_update_img).exists() {
+        // no image exist, do nothing for module!
+        return Ok(());
+    }
+
+    let module_path = Path::new(module_dir);
+    if !module_path.exists() {
+        std::fs::create_dir_all(module_path)?;
+    }
+
+    println!("mount {} to {}", target_update_img, module_dir);
+    mount_image(target_update_img, module_dir)?;
+
+    // construct overlay mount params
+    let dir = std::fs::read_dir(module_dir);
+    let Ok(dir) = dir else {
+        bail!("open {} failed", defs::MODULE_DIR);
+    };
+
+    let mut lowerdir: Vec<String> = Vec::new();
+    for entry in dir.flatten() {
+        let module = entry.path();
+        if !module.is_dir() {
+            continue;
+        }
+        let disabled = module.join(defs::DISABLE_FILE_NAME).exists();
+        if disabled {
+            println!("module: {} is disabled, ignore!", module.display());
+            continue;
+        }
+
+        let mut module_system = PathBuf::new();
+        module_system.push(&module);
+        module_system.push("system");
+
+        if !module_system.as_path().exists() {
+            println!(
+                "module: {} has no system overlay, ignore!",
+                module.display()
+            );
+            continue;
+        }
+        lowerdir.push(format!("{}", module_system.display()));
+    }
+
+    if lowerdir.is_empty() {
+        println!("lowerdir is empty, ignore!");
+        return Ok(());
+    }
+
+    // add /system as the last lowerdir
+    lowerdir.push(String::from("/system"));
+
+    let lowerdir = lowerdir.join(":");
+    println!("lowerdir: {}", lowerdir);
+
+    let mount_args = format!(
+        "mount -t overlay overlay -o ro,lowerdir={} /system",
+        lowerdir
+    );
+    let result = Exec::shell(mount_args).join()?;
+    if !result.success() {
+        println!("mount overlay failed");
+    }
+
+    Ok(())
+}
+
+pub fn on_boot_completed() -> Result<()> {
+    let module_update_img = Path::new(defs::MODULE_UPDATE_IMG);
+    let module_img = Path::new(defs::MODULE_IMG);
+    if module_update_img.exists() {
+        // this is a update and we successfully booted
+        std::fs::rename(module_update_img, module_img)?;
+    }
+    Ok(())
+}
+
+pub fn daemon() -> Result<()> {
+    Ok(())
+}
+
+pub fn install() -> Result<()> {
+    let src = "/proc/self/exe";
+    let dst = defs::DAEMON_PATH;
+
+    std::fs::copy(src, dst)?;
+    Ok(())
+}