Main.java Source code

Introduction

Here is the source code for Main.java

Source

//package com.java2s;
//License from project: Open Source License 

import java.io.ByteArrayOutputStream;
import java.io.DataOutputStream;

import java.io.InputStream;

public class Main {
    private static int BUFF_LEN = 1024 * 100;
    private static Process sh = null;
    private static Process su = null;

    private static byte[] ExecuteCommand(String command, Boolean useroot, boolean forcenew) throws Exception {
        Process p;
        DataOutputStream stdin;
        InputStream stdout;
        ByteArrayOutputStream baos;
        int read;
        byte[] buffer;
        /*
         * If for any reason the command does not print anything we are stuck forever.
         * Make sure that we print SOMETHING ALWAYS!
         */
        command = "RESULT=$(" + command + "); if [[ $RESULT == '' ]]; then echo '#null#';else echo $RESULT;fi\n";

        p = getProcess(useroot, forcenew);

        stdin = new DataOutputStream(p.getOutputStream());
        stdout = p.getInputStream();
        buffer = new byte[BUFF_LEN];
        baos = new ByteArrayOutputStream();

        stdin.writeBytes(command);

        while (true) {
            read = stdout.read(buffer);
            baos.write(buffer, 0, read);
            if (read < BUFF_LEN) {
                //we have read everything
                break;
            }
        }
        if (forcenew) {
            stdin.writeBytes("exit\n");
            stdin.flush();
            stdin.close();
        }

        //p.waitFor();

        return baos.toByteArray();
    }

    private static Process getProcess(boolean useroot, boolean forcenew) throws Exception {

        if (forcenew && useroot)
            return Runtime.getRuntime().exec(new String[] { "su" });
        else if (forcenew && !useroot)
            return Runtime.getRuntime().exec(new String[] { "sh" });
        else if (useroot && su != null)
            return su;
        else if (useroot && su == null) {
            su = Runtime.getRuntime().exec(new String[] { "su" });
            return su;
        } else if (!useroot && sh != null)
            return sh;
        else if (!useroot && sh == null) {
            sh = Runtime.getRuntime().exec(new String[] { "sh" });
            return sh;
        }

        /* Shouldn't ever actually get here! */
        return Runtime.getRuntime().exec(new String[] { "sh" });

    }
}