From e37d0148e14d55bf82ab0f8ec6befe6c9c58db16 Mon Sep 17 00:00:00 2001
From: tiann <twsxtd@gmail.com>
Date: Thu, 5 Jan 2023 14:10:38 +0800
Subject: [PATCH] ksud: Add support for post-fs-data and system.prop

---
 userspace/ksud/src/event.rs  |  3 ++-
 userspace/ksud/src/module.rs | 46 +++++++++++++++++++++++++++++++++++-
 2 files changed, 47 insertions(+), 2 deletions(-)

diff --git a/userspace/ksud/src/event.rs b/userspace/ksud/src/event.rs
index 4db8683e..a288eca2 100644
--- a/userspace/ksud/src/event.rs
+++ b/userspace/ksud/src/event.rs
@@ -92,7 +92,8 @@ pub fn on_post_data_fs() -> Result<()> {
 
     // module mounted, exec modules post-fs-data scripts
     // todo: Add timeout
-    crate::module::exec_post_fs_data()?;
+    let _ = crate::module::exec_post_fs_data();
+    let _ = crate::module::load_system_prop();
 
     Ok(())
 }
diff --git a/userspace/ksud/src/module.rs b/userspace/ksud/src/module.rs
index f795ff34..92a74ce1 100644
--- a/userspace/ksud/src/module.rs
+++ b/userspace/ksud/src/module.rs
@@ -7,7 +7,7 @@ use std::{
     io::Cursor,
     path::{Path, PathBuf},
     process::{Command, Stdio},
-    str::FromStr,
+    str::FromStr, os::unix::prelude::PermissionsExt,
 };
 use subprocess::Exec;
 use zip_extensions::*;
@@ -155,6 +155,50 @@ pub fn exec_post_fs_data() -> Result<()> {
     Ok(())
 }
 
+const RESETPROP: &[u8] = include_bytes!("./magisk64");
+const RESETPROP_PATH: &str = concatcp!(defs::WORKING_DIR, "/magisk64");
+
+fn ensure_resetprop() -> Result<()> {
+    if Path::new(RESETPROP_PATH).exists() {
+        return Ok(());
+    }
+    std::fs::write(RESETPROP_PATH, RESETPROP)?;
+    std::fs::set_permissions(RESETPROP_PATH, std::fs::Permissions::from_mode(0o755))?;
+    Ok(())
+}
+
+pub fn load_system_prop() -> Result<()> {
+
+    ensure_resetprop()?;
+
+    let modules_dir = Path::new(defs::MODULE_DIR);
+    let dir = std::fs::read_dir(modules_dir)?;
+    for entry in dir.flatten() {
+        let path = entry.path();
+        let disabled = path.join(defs::DISABLE_FILE_NAME);
+        if disabled.exists() {
+            println!("{} is disabled, skip", path.display());
+            continue;
+        }
+
+        let system_prop = path.join("system.prop");
+        if !system_prop.exists() {
+            continue;
+        }
+        println!("load {} system.prop", path.display());
+
+        // resetprop --file system.prop
+        Command::new(RESETPROP_PATH)
+            .arg("resetprop")
+            .arg("--file")
+            .arg(&system_prop)
+            .status()
+            .with_context(|| format!("Failed to exec {}", system_prop.display()))?;
+    }
+
+    Ok(())
+}
+
 pub fn install_module(zip: String) -> Result<()> {
     ensure_boot_completed()?;