X-Git-Url: http://git.ieval.ro/?a=blobdiff_plain;f=src%2Fro%2Fieval%2Ffonbot%2FHeavy.java;h=b1f443c5f6067fb8a1366f2e51e71777d3504b32;hb=251aceaaa3379c65d66a975ff9e7524f7d2f1d73;hp=909e7ca1e866dbb0e3c5c3a3546d0172e219de0e;hpb=1b41fa24b0346319a5bec7f59f0e8815fb3462fa;p=fonbot.git

diff --git a/src/ro/ieval/fonbot/Heavy.java b/src/ro/ieval/fonbot/Heavy.java
index 909e7ca..b1f443c 100644
--- a/src/ro/ieval/fonbot/Heavy.java
+++ b/src/ro/ieval/fonbot/Heavy.java
@@ -3,11 +3,13 @@ package ro.ieval.fonbot;
 import static ro.ieval.fonbot.R.string.*;
 import static ro.ieval.fonbot.Utils.toNonNull;
 
+import java.io.BufferedReader;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.FileNotFoundException;
 import java.io.InputStream;
+import java.io.InputStreamReader;
 import java.io.IOException;
 import java.io.OutputStream;
 import java.lang.reflect.InvocationTargetException;
@@ -665,6 +667,12 @@ final class Heavy {
 		case TORCH:
 			Utils.sendMessage(context, replyTo, torch_help);
 			break;
+		case SH:
+			Utils.sendMessage(context, replyTo, sh_help);
+			break;
+		case ROOTSH:
+			Utils.sendMessage(context, replyTo, rootsh_help);
+			break;
 		}
 	}
 
@@ -1843,8 +1851,22 @@ final class Heavy {
 	 */
 	public static void reboot(final Context context, final Address replyTo, final @Nullable String reason) {
 		final PowerManager pm=(PowerManager) context.getSystemService(Context.POWER_SERVICE);
-		Utils.sendMessage(context, replyTo, rebooting);	
-		pm.reboot(reason);
+		Utils.sendMessage(context, replyTo, rebooting);
+		try {
+			pm.reboot(reason);
+		} catch (final Exception e){
+			e.printStackTrace();
+		}
+		try {
+			Runtime.getRuntime().exec(new String[]{
+					"su",
+					"-c",
+					"reboot"
+				}).waitFor();
+		} catch (final Exception e){
+			e.printStackTrace();
+		}
+		Utils.sendMessage(toNonNull(context), toNonNull(replyTo), reboot_failed);
 	}
 
 	/**
@@ -1913,4 +1935,30 @@ final class Heavy {
 	public static void getfile(final Context context, final Address replyTo, final String filename, final String hostname, final int port){
 		new GetfileExecutableRunnable(context, replyTo, filename, hostname, port).execute();
 	}
+
+	/**
+	 * Execute a command using a given shell and reply with the output.
+	 *
+	 * @param context Context instance
+	 * @param replyTo reply Address
+	 * @param shell The shell to execute with. Usually sh or su.
+	 * @param command The command to pass to the shell.
+	 */
+	public static void execute(final Context context, final Address replyTo, final String shell, final String command) {
+		try {
+			final Process proc = Runtime.getRuntime().exec(new String[]{
+					shell,
+					"-c",
+					command
+				});
+			final BufferedReader br = new BufferedReader (new InputStreamReader(proc.getInputStream()));
+			String line;
+			while((line = br.readLine()) != null)
+				Utils.sendMessage(context, replyTo, line);
+			proc.waitFor();
+		} catch (final Exception e){
+			Utils.sendMessage(context, replyTo, error_while_processing_command, e.getClass().getName(), e.getMessage());
+			Log.w(Heavy.class.getName(), "Error while processing command", e);
+		}
+	}
 }