[DEV] Added isRoot check if process is started with su command

manneohlund · Oct 2, 2017 · aba16f7 · aba16f7
1 parent 6ad683b
commit aba16f7
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/library/src/main/java/ashell/ShellFactory.java b/library/src/main/java/ashell/ShellFactory.java
@@ -22,16 +22,22 @@
 
 public class ShellFactory {
 
+    private boolean isRoot = false;
     private AShell shell;
 
-    private ShellFactory(AShell shell) {
+    public ShellFactory(AShell shell, boolean isRoot) {
         this.shell = shell;
+        this.isRoot = isRoot;
     }
 
     public AShell getShell() {
         return shell;
     }
 
+    public boolean isRoot() {
+        return isRoot;
+    }
+
     @SuppressWarnings("unchecked") // Single-interface proxy creation guarded by parameter safety.
     public <T> T create(final Class<T> shellModel) {
         //Utils.validateServiceInterface(service);
@@ -97,12 +103,13 @@ public ShellFactory build() throws IOException {
         }
 
         public ShellFactory build(boolean runAsSuperuser) throws IOException {
-
+            boolean isRoot = false;
             if (processBuilder == null) {
                 processBuilder = ProcessUtils.getDefaultProcessBuilder("sh");
                 if (runAsSuperuser) {
                     try {
                         processBuilder = ProcessUtils.getDefaultProcessBuilder("su");
+                        isRoot = true;
                     } catch (Exception e) {
                         e.printStackTrace();
                     }
@@ -134,7 +141,7 @@ public ShellFactory build(boolean runAsSuperuser) throws IOException {
 
             shell.start();
 
-            return new ShellFactory(shell);
+            return new ShellFactory(shell, isRoot);
         }
     }
 }