Set TMP_PATH env programmatically (JingMatrix#105)

By doing so, we can invoke the binary `bin/zygisk-ptrace64` directly.

The wrapper scirpt `zygisk-ctl.sh` is thus removed.

Author: JingMatrix

loader/build.gradle.kts

@@ -8,6 +8,7 @@ plugins {
 val verCode: Int by rootProject.extra
 val verName: String by rootProject.extra
 val commitHash: String by rootProject.extra
+val workDirectory: String by rootProject.extra
 
 fun Project.findInPath(executable: String, property: String): String? {
     val pathEnv = System.getenv("PATH")
@@ -29,7 +30,8 @@ val defaultCFlags = arrayOf(
     "-Wall", "-Wextra",
     "-fno-rtti", "-fno-exceptions",
     "-fno-stack-protector", "-fomit-frame-pointer",
-    "-Wno-builtin-macro-redefined", "-D__FILE__=__FILE_NAME__"
+    "-Wno-builtin-macro-redefined", "-D__FILE__=__FILE_NAME__",
+    "-DWORK_DIRECTORY='\"${workDirectory}\"'"
 )
 
 val releaseFlags = arrayOf(

loader/src/common/daemon.cpp

@@ -10,7 +10,10 @@
 namespace zygiskd {
 static std::string TMP_PATH;
 
-void Init(const char *path) { TMP_PATH = path; }
+void Init(const char *path) {
+    TMP_PATH = path;
+    setenv("TMP_PATH", TMP_PATH.data(), 0);
+}
 
 std::string GetTmpPath() { return TMP_PATH; }
 

loader/src/ptracer/main.cpp

@@ -18,6 +18,8 @@
 // Use string_view literals for efficient, allocation-free string comparisons.
 using namespace std::string_view_literals;
 
+const char *const kWorkDirectory = WORK_DIRECTORY;
+
 // The main entry point for the monitoring process.
 void init_monitor() {
     LOGI("NeoZygisk %s", ZKSU_VERSION);
@@ -72,7 +74,7 @@ static int handle_version();
  */
 int main(int argc, char **argv) {
     // This initialization is for the daemon's internal logic, not for CLI output.
-    zygiskd::Init(getenv("TMP_PATH"));
+    zygiskd::Init(kWorkDirectory);
 
     if (argc < 2) {
         print_usage(argv[0]);

module/src/customize.sh

@@ -110,13 +110,11 @@ extract "$ZIPFILE" 'module.prop'     "$MODPATH"
 extract "$ZIPFILE" 'post-fs-data.sh' "$MODPATH"
 extract "$ZIPFILE" 'service.sh'      "$MODPATH"
 extract "$ZIPFILE" 'uninstall.sh'      "$MODPATH"
-extract "$ZIPFILE" 'zygisk-ctl.sh'   "$MODPATH"
 mv "$TMPDIR/sepolicy.rule" "$MODPATH"
 
 mkdir "$MODPATH/bin"
 mkdir "$MODPATH/lib"
 mkdir "$MODPATH/lib64"
-mv "$MODPATH/zygisk-ctl.sh" "$MODPATH/bin/zygisk-ctl"
 
 if [ "$ARCH" = "x86" ]; then
   ui_print "- Extracting x86 libraries"
@@ -148,6 +146,7 @@ elif [ "$ARCH" = "arm64" ]; then
   mv "$MODPATH/bin/libzygisk_ptrace.so" "$MODPATH/bin/zygisk-ptrace64"
 fi
 
+sleep 1
 ui_print "- Setting permissions"
 set_perm_recursive "$MODPATH/bin" 0 0 0755 0755
 set_perm_recursive "$MODPATH/lib" 0 0 0755 0644 u:object_r:system_lib_file:s0

module/src/post-fs-data.sh

@@ -26,7 +26,7 @@ create_sys_perm() {
   chcon u:object_r:system_file:s0 $1
 }
 
-export TMP_PATH=@WORK_DIRECTORY@
+TMP_PATH=@WORK_DIRECTORY@
 
 if [ -d $TMP_PATH ]; then
   rm -rf $TMP_PATH