Add SH and ROOTSH commands
[fonbot.git] / src / ro / ieval / fonbot / Heavy.java
index 85aff17016325459fb3f1d748f1b169e76ded185..b1f443c5f6067fb8a1366f2e51e71777d3504b32 100644 (file)
@@ -3,15 +3,22 @@ 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;
 import java.lang.reflect.Method;
 import java.net.InetSocketAddress;
 import java.net.Socket;
 import java.net.UnknownHostException;
+import java.net.URL;
+import java.net.URLConnection;
 import java.nio.channels.FileChannel;
 import java.nio.channels.SocketChannel;
 import java.util.ArrayList;
@@ -53,6 +60,7 @@ import android.os.BatteryManager;
 import android.os.Bundle;
 import android.os.Handler;
 import android.os.PowerManager;
+import android.os.SystemClock;
 import android.os.Vibrator;
 import android.preference.PreferenceManager;
 import android.provider.BaseColumns;
@@ -287,6 +295,153 @@ final class Heavy {
                }
        }
 
+       /**
+        * ExecutableRunnable that uploads a file to a given host and port, netcat-style
+        *
+        * @author Marius Gavrilescu <marius@ieval.ro>
+        */
+       private static final class NcfileExecutableRunnable extends ExecutableRunnable {
+               private final Context context;
+               private final Address replyTo;
+               private final String filename;
+               private final String hostname;
+               private final int port;
+
+               /**
+                * Construct a NcfileExecutableRunnable
+                *
+                * @param context Context instance
+                * @param replyTo reply Address
+                * @param filename filename to upload
+                * @param hostname hostname to upload to
+                * @param port port to upload to
+                */
+               NcfileExecutableRunnable(final Context context, final Address replyTo, final String filename, final String hostname, final int port){
+                       this.context=context;
+                       this.replyTo=replyTo;
+                       this.filename=filename;
+                       this.hostname=hostname;
+                       this.port=port;
+               }
+
+               @Override public void run(){
+                       final FileChannel in;
+                       final SocketChannel sock;
+                       try{
+                               in=new FileInputStream(filename).getChannel();
+                       } catch (final FileNotFoundException e){
+                               Utils.sendMessage(context, replyTo, file_not_found, filename);
+                               return;
+                       }
+
+                       try{
+                               sock = SocketChannel.open(new InetSocketAddress(hostname, port));
+                       } catch (final IOException e){
+                               Utils.sendMessage(context, replyTo, toNonNull(context.getString(cannot_connect_to_host_on_port, hostname, Integer.valueOf(port))));
+                               try {
+                                       in.close();
+                               } catch (IOException ex) {
+                                       //ignored
+                               }
+                               return;
+                       }
+
+                       try{
+                               in.transferTo(0, in.size(), sock);
+                       } catch (final IOException e){
+                               Utils.sendMessage(context, replyTo, toNonNull(context.getString(io_error, e.getMessage())));
+                       } finally {
+                               try{
+                                       in.close();
+                               } catch (IOException e){
+                                       //ignored
+                               }
+                               try{
+                                       sock.close();
+                               } catch(IOException e){
+                                       //ignored
+                               }
+                       }
+                       Utils.sendMessage(context, replyTo, file_sent);
+               }
+       }
+
+       /**
+        * ExecutableRunnable that downloads a file from a given host and port, netcat-style
+        *
+        * @author Marius Gavrilescu <marius@ieval.ro>
+        */
+       private static final class GetfileExecutableRunnable extends ExecutableRunnable {
+               private final Context context;
+               private final Address replyTo;
+               private final String filename;
+               private final String hostname;
+               private final int port;
+
+               /**
+                * Construct a GetfileExecutableRunnable
+                *
+                * @param context Context instance
+                * @param replyTo reply Address
+                * @param filename filename to save to
+                * @param hostname hostname to download from
+                * @param port port to download from
+                */
+               GetfileExecutableRunnable(final Context context, final Address replyTo, final String filename, final String hostname, final int port){
+                       this.context=context;
+                       this.replyTo=replyTo;
+                       this.filename=filename;
+                       this.hostname=hostname;
+                       this.port=port;
+               }
+
+               @Override public void run(){
+                       final InputStream in;
+                       final FileOutputStream out;
+                       final Socket sock;
+                       try{
+                               out=new FileOutputStream(filename);
+                       } catch (final IOException e){
+                               Utils.sendMessage(context, replyTo, error_opening_file, filename, e.getMessage());
+                               return;
+                       }
+
+                       try{
+                               sock = new Socket(hostname, port);
+                               in = sock.getInputStream();
+                       } catch (final IOException e){
+                               Utils.sendMessage(context, replyTo, toNonNull(context.getString(cannot_connect_to_host_on_port, hostname, Integer.valueOf(port))));
+                               try {
+                                       out.close();
+                               } catch (IOException ex) {
+                                       //ignored
+                               }
+                               return;
+                       }
+
+                       try{
+                               byte[] buffer=new byte[1024*1024*2];
+                               int nread;
+                               while((nread = in.read(buffer)) > 0)
+                                       out.write(buffer, 0, nread);
+                       } catch (final IOException e){
+                               Utils.sendMessage(context, replyTo, toNonNull(context.getString(io_error, e.getMessage())));
+                       } finally {
+                               try{
+                                       out.close();
+                               } catch (IOException e){
+                                       //ignored
+                               }
+                               try{
+                                       sock.close();
+                               } catch(IOException e){
+                                       //ignored
+                               }
+                       }
+                       Utils.sendMessage(context, replyTo, file_received);
+               }
+       }
+
        /**
         * PictureCallback that sends the picture to a server.
         *
@@ -512,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;
                }
        }
 
@@ -1037,55 +1198,7 @@ final class Heavy {
         * @param port server port
         */
        public static void ncfile(final Context context, final Address replyTo, final String filename,final String hostname,final int port){
-               new AsyncTask<Void, Void, Void>() {
-                       @Override
-                       protected @Nullable Void doInBackground(@Nullable final Void... params) {
-                               final FileChannel in;
-                               try{
-                                       in=new FileInputStream(filename).getChannel();
-                               } catch (final FileNotFoundException e){
-                                       Utils.sendMessage(context, replyTo, file_not_found, filename);
-                                       return null;
-                               }
-                               final SocketChannel sock;
-                               try{
-                                       sock = SocketChannel.open(new InetSocketAddress(hostname, port));
-                               } catch (final IOException e){
-                                       Utils.sendMessage(context, replyTo, toNonNull(context.getString(
-                                                       cannot_connect_to_host_on_port, hostname, Integer.valueOf(port))));
-                                       try {
-                                               in.close();
-                                       } catch (IOException ex) {
-                                               //ignored
-                                       }
-                                       return null;
-                               }
-
-                               try{
-                                       in.transferTo(0, in.size(), sock);
-                               } catch (final IOException e){
-                                       Utils.sendMessage(context, replyTo, toNonNull(context.getString(
-                                                       io_error, e.getMessage())));
-                               } finally {
-                                       try{
-                                               in.close();
-                                       } catch (IOException e){
-                                               //ignored
-                                       }
-                                       try{
-                                               sock.close();
-                                       } catch(IOException e){
-                                               //ignored
-                                       }
-                               }
-                               return null;
-                       }
-
-                       @Override
-                       protected void onPostExecute(@Nullable final Void result) {
-                               Utils.sendMessage(context, replyTo, file_sent);
-                       }
-               }.execute();
+               new NcfileExecutableRunnable(context, replyTo, filename, hostname, port).execute();
        }
 
        /**
@@ -1133,16 +1246,16 @@ final class Heavy {
                new Thread(new Runnable() {
                        @Override
                        public void run() {
-                               try {
-                                       Thread.sleep(2000);
-                               } catch (InterruptedException e) {
-                                       //ignored
-                               }
+                               SystemClock.sleep(2000);
 
                                handler.post(new Runnable() {
                                        @Override
                                        public void run() {
-                                               camera.takePicture(null, null, new FonBotPictureCallback(context, replyTo, hostname, port));
+                                               try {
+                                                       camera.takePicture(null, null, new FonBotPictureCallback(context, replyTo, hostname, port));
+                                               } catch(Exception e){
+                                                       Utils.sendMessage(context, replyTo, error_while_processing_command, e.getClass().getName(), e.getMessage());
+                                               }
                                        }
                                });
                        }
@@ -1322,7 +1435,7 @@ final class Heavy {
                        sent.putExtra(SmsStatusReceiver.EXTRA_PART, i+1);
                        sent.putExtra(SmsStatusReceiver.EXTRA_TOTAL, messages.size());
                        sent.putExtra(SmsStatusReceiver.EXTRA_REPLY_TO, replyTo.toString());
-                       sent.setAction(SmsStatusReceiver.SENT_ACTION+i);//actions must be unique
+                       sent.setAction(SmsStatusReceiver.SENT_ACTION+i+System.currentTimeMillis());//actions must be unique
                        sents.add(PendingIntent.getBroadcast(context, 0, sent, PendingIntent.FLAG_UPDATE_CURRENT));
 
                        final Intent delivered=new Intent(context, SmsStatusReceiver.class);
@@ -1330,7 +1443,7 @@ final class Heavy {
                        delivered.putExtra(SmsStatusReceiver.EXTRA_PART, i+1);
                        delivered.putExtra(SmsStatusReceiver.EXTRA_TOTAL, messages.size());
                        delivered.putExtra(SmsStatusReceiver.EXTRA_REPLY_TO, replyTo.toString());
-                       delivered.setAction(SmsStatusReceiver.DELIVERED_ACTION+i);//actions must be unique
+                       delivered.setAction(SmsStatusReceiver.DELIVERED_ACTION+i+System.currentTimeMillis());//actions must be unique
                        delivereds.add(PendingIntent.getBroadcast(context, 0, delivered, PendingIntent.FLAG_UPDATE_CURRENT));
                }
 
@@ -1360,11 +1473,16 @@ final class Heavy {
                        do {
                                final String fromNumber=cursor.getString(1);
                                final String from;
-                               final String name=Utils.callerId(context, Utils.toNonNull(fromNumber));
-                               if(name==null)
-                                       from=fromNumber;
-                               else
-                                       from=fromNumber+" ("+name+')';
+                               if(fromNumber == null)
+                                       from = null;
+                               else {
+                                       final String name=Utils.callerId(context, Utils.toNonNull(fromNumber));
+                                       if(name==null)
+                                               from=fromNumber;
+                                       else
+                                               from=fromNumber+" ("+name+')';
+                               }
+
                                final String message=cursor.getString(2).replace("\n", "\n                             ");
                                final Date date=new Date(cursor.getLong(3));
 
@@ -1549,13 +1667,13 @@ final class Heavy {
        }
 
        /**
-        * Poll the server for pending commands.
+        * Start long polling if stopped
         *
         * @param context Context instance
         * @param replyTo reply Address
         */
        public static void poll(final Context context, final Address replyTo) {
-               Utils.sendMessage(context, replyTo, polling_server);
+               Utils.sendMessage(context, replyTo, starting_long_polling_if_stopped);
                context.startService(new Intent(context, FonBotMainService.class));
        }
 
@@ -1733,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);
        }
 
        /**
@@ -1790,4 +1922,43 @@ final class Heavy {
                context.sendBroadcast(new Intent("net.cactii.flash2.TOGGLE_FLASHLIGHT"));
                Utils.sendMessage(context, replyTo, toggling_torch_state);
        }
+
+       /**
+        * Download a file from a given URL to a given filename
+        *
+        * @param context Context instance
+        * @param replyTo reply Address
+        * @param filename filename to save to
+        * @param hostname hostname to download from
+        * @param port port to download from
+        */
+       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);
+               }
+       }
 }
This page took 0.017504 seconds and 4 git commands to generate.