Rewrite all root method with own su library

2025-09-06 06:36:58 +00:00 · 2016-08-25 05:58:15 +08:00
parent e18f4c843a
commit a5ea214553
10 changed files with 411 additions and 237 deletions
@@ -0,0 +1,172 @@
+package com.topjohnwu.magisk.utils;
+
+import java.io.DataOutputStream;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * Modified by topjohnwu, based on Chainfire's libsuperuser
+ */
+
+public class Shell {
+
+
+    public static String suPath;
+    // -1 = problematic/unknown issue; 0 = not rooted; 1 = properly rooted; 2 = improperly rooted;
+    public static int rootStatus = 0;
+
+    private static Process rootShell;
+    private static DataOutputStream rootSTDIN;
+    private static StreamGobbler rootSTDOUT;
+    private static List<String> rootOutList = new ArrayList<>();
+
+    static {
+        List<String> ret = sh("getprop magisk.supath");
+        if(!ret.isEmpty()) {
+            suPath = ret.get(0) + "/su";
+        } else {
+            suPath = "su";
+            rootStatus = 2;
+        }
+    }
+
+    public static boolean rootAccess() {
+        return rootStatus > 0;
+    }
+
+    public static void startRoot() {
+        rootStatus = rootStatus == 0 ? 1 : 2;
+
+        try {
+            rootShell = Runtime.getRuntime().exec(suPath);
+            rootSTDIN = new DataOutputStream(rootShell.getOutputStream());
+            rootSTDOUT = new StreamGobbler(rootShell.getInputStream(), rootOutList);
+            rootSTDOUT.start();
+        } catch (IOException e) {
+            // runtime error! No binary found!
+            rootStatus = 0;
+            return;
+        }
+
+
+        List<String> ret = su("echo -BOC-", "id");
+        if (ret == null) {
+            // Something wrong with root install, not enabled?
+            rootStatus = -1;
+            return;
+        }
+
+        for (String line : ret) {
+            if (line.contains("uid=")) {
+                // id command is working, let's see if we are actually root
+                rootStatus = line.contains("uid=0") ? rootStatus : -1;
+                return;
+            } else if (!line.contains("-BOC-")) {
+                rootStatus = -1;
+                return;
+            }
+        }
+    }
+
+    @Override
+    protected void finalize() throws Throwable {
+        super.finalize();
+        if (rootAccess()) {
+            rootSTDIN.write("exit\n".getBytes("UTF-8"));
+            rootSTDIN.flush();
+            rootSTDIN.flush();
+            rootShell.waitFor();
+            rootSTDIN.close();
+            rootSTDOUT.join();
+            rootShell.destroy();
+        }
+    }
+
+    public static List<String> sh(String... commands) {
+        List<String> res = Collections.synchronizedList(new ArrayList<String>());
+
+        try {
+            Process process = Runtime.getRuntime().exec("sh");
+            DataOutputStream STDIN = new DataOutputStream(process.getOutputStream());
+            StreamGobbler STDOUT = new StreamGobbler(process.getInputStream(), res);
+
+            STDOUT.start();
+
+            try {
+                for (String write : commands) {
+                    STDIN.write((write + "\n").getBytes("UTF-8"));
+                    STDIN.flush();
+                }
+                STDIN.write("exit\n".getBytes("UTF-8"));
+                STDIN.flush();
+            } catch (IOException e) {
+                if (!e.getMessage().contains("EPIPE")) {
+                    throw e;
+                }
+            }
+
+            process.waitFor();
+
+            try {
+                STDIN.close();
+            } catch (IOException e) {
+                // might be closed already
+            }
+            STDOUT.join();
+            process.destroy();
+
+        } catch (IOException | InterruptedException e) {
+            // shell probably not found
+            res = null;
+        }
+
+        return res;
+    }
+
+    public static List<String> su(String... commands) {
+
+        if(!Shell.rootAccess()) return null;
+
+        rootOutList.clear();
+
+        try {
+            try {
+                for (String write : commands) {
+                    rootSTDIN.write((write + "\n").getBytes("UTF-8"));
+                    rootSTDIN.flush();
+                }
+                rootSTDIN.write(("echo \'-done-\'\n").getBytes("UTF-8"));
+                rootSTDIN.flush();
+            } catch (IOException e) {
+                if (!e.getMessage().contains("EPIPE")) {
+                    throw e;
+                }
+            }
+
+            while (true) {
+                try {
+                    // Process terminated, it means the interactive shell cannot be initialized
+                    rootShell.exitValue();
+                    return null;
+                } catch (IllegalThreadStateException e) {
+                    // Process still running, gobble output until done
+                    if (rootOutList != null && !rootOutList.isEmpty()) {
+                        if (rootOutList.get(rootOutList.size() - 1).equals("-done-")) {
+                            rootOutList.remove(rootOutList.size() - 1);
+                            break;
+                        }
+                    }
+                    rootSTDOUT.join(100);
+                }
+            }
+
+        } catch (IOException | InterruptedException e) {
+            // shell probably not found
+            return null;
+        }
+
+        return new ArrayList<>(rootOutList);
+    }
+}