ksud: add debug insmod

Author: Ylarod

userspace/ksud/src/cli.rs

@@ -173,6 +173,12 @@ enum Debug {
         path: PathBuf,
     },
 
+    /// Load a kernel module from disk
+    Insmod {
+        /// kernel module path
+        module: PathBuf,
+    },
+
     /// Process mark management
     Mark {
         #[command(subcommand)]
@@ -637,6 +643,7 @@ pub fn run() -> Result<()> {
                 let data = assets::get_asset_data(&name)?;
                 utils::ensure_binary(&path, &data, false)
             }
+            Debug::Insmod { module } => debug::insmod(&module),
             Debug::Mark { command } => match command {
                 MarkCommand::Get { pid } => debug::mark_get(pid),
                 MarkCommand::Mark { pid } => debug::mark_set(pid),

userspace/ksud/src/debug.rs

@@ -1,5 +1,6 @@
 use anyhow::{Context, Ok, Result, bail, ensure};
 use std::{
+    fs,
     path::{Path, PathBuf},
     process::Command,
 };
@@ -49,6 +50,20 @@ pub fn set_manager(pkg: &str) -> Result<()> {
     Ok(())
 }
 
+pub fn insmod(module: &Path) -> Result<()> {
+    let module = module
+        .canonicalize()
+        .with_context(|| format!("resolve module path failed: {}", module.display()))?;
+    let module_data =
+        fs::read(&module).with_context(|| format!("read module failed: {}", module.display()))?;
+
+    ksuinit::load_module(&module_data)
+        .with_context(|| format!("load module failed: {}", module.display()))?;
+
+    println!("Loaded kernel module: {}", module.display());
+    Ok(())
+}
+
 /// Get mark status for a process
 pub fn mark_get(pid: i32) -> Result<()> {
     let result = ksucalls::mark_get(pid)?;