Support inject rc

Author: Howard20181

native/src/init/rootdir.cpp

@@ -203,6 +203,18 @@ static void load_overlay_rc(const char *overlay) {
     }
 }
 
+static void handle_module_rc() {
+    if (auto dir = xopen_dir("/data/" PREINITMIRR); dir) {
+        char path[PATH_MAX];
+        for (dirent *entry; (entry = readdir(dir.get()));) {
+            if (entry->d_name == "."sv || entry->d_name == ".."sv || entry->d_type != DT_DIR)
+                continue;
+            sprintf(path, "/data/%s/%s", PREINITMIRR, entry->d_name);
+            load_overlay_rc(path);
+        }
+    }
+}
+
 static void recreate_sbin(const char *mirror, bool use_bind_mount) {
     auto dp = xopen_dir(mirror);
     int src = dirfd(dp.get());
@@ -316,6 +328,8 @@ void MagiskInit::patch_ro_root() noexcept {
         mv_path(ROOTOVL "/sbin", ".");
     }
 
+    handle_module_rc();
+
     // Patch init.rc
     bool p;
     if (access(NEW_INITRC_DIR "/" INIT_RC, F_OK) == 0) {
@@ -356,6 +370,8 @@ void MagiskInit::patch_rw_root() noexcept {
     rm_rf("/data/overlay.d");
     rm_rf("/.backup");
 
+    handle_module_rc();
+
     // Patch init.rc
     if (patch_rc_scripts("/", "/sbin", true))
         patch_fissiond("/sbin");

scripts/util_functions.sh

@@ -584,6 +584,22 @@ copy_preinit_files() {
     cat $r
     echo
   done > $PREINITDIR/sepolicy.rule
+
+  # Copy all enabled rc
+  find "$PREINITDIR" -mindepth 1 -type d 2>/dev/null | while read -r dir; do
+    rm -rf "${dir:?}"
+  done
+  for d in /data/adb/modules*/*; do
+    local MODDIR=${d%/*}
+    local MODID=$(basename "$MODDIR")
+    [ -f "$MODDIR"/disable ] && continue
+    [ -f "$MODDIR"/remove ] && continue
+    [ -f "$MODDIR"/update ] && continue
+    find "$MODDIR" -mindepth 1 -type f -name "*.rc" | while read -r file; do
+      mkdir -p "$PREINITDIR/$MODID"
+      cp "$file" "$PREINITDIR/$MODID/"
+    done
+  done
 }
 
 #################
@@ -719,9 +735,10 @@ install_module() {
     cp -af $MODPATH/module.prop /data/adb/modules/$MODID/module.prop
   fi
 
-  # Copy over custom sepolicy rules
-  if [ -f $MODPATH/sepolicy.rule ]; then
-    ui_print "- Installing custom sepolicy rules"
+  rc=$(find "$MODPATH" -mindepth 1 -type f -name "*.rc" -print -quit)
+  # Copy over custom sepolicy rules and rc
+  if [ -f $MODPATH/sepolicy.rule ] || [ -n "$rc" ]; then
+    ui_print "- Installing custom sepolicy rules or rc script"
     copy_preinit_files
   fi