jsudo.java

package javaforce.utils;

/*
 * Created : Apr 13, 2012
 */
import java.io.*;
import java.util.*;
import javax.swing.*;

import javaforce.*;

public class jsudo {

  public static void main(String args[]) {
    if (args.length == 0) {
      System.out.println("Usage : jsudo command [args]");
      System.exit(1);
    }
    if (args[0].equals("--ask")) {
      String pass = GetPassword.getPassword(null);
      if (pass == null) {
        pass = "";
      }
      System.out.println(pass);
      System.exit(0);
    }
    try {
      //add "sudo -A"
      String cmd[] = new String[args.length + 2];
      cmd[0] = "sudo";
      cmd[1] = "-A";
      System.arraycopy(args, 0, cmd, 2, args.length);
      ProcessBuilder pb = new ProcessBuilder(cmd);
      Map<String, String> env = pb.environment();
      env.put("SUDO_ASKPASS", "/usr/bin/jsudo-ask");
      Process p = pb.start();
      OutputStream os = p.getOutputStream();
      InputStream err = p.getErrorStream();
      InputStream is = p.getInputStream();
      //System.in -> os
      new Relay(System.in, os).start();
      //is -> System.out
      new Relay(is, System.out).start();
      //err -> System.err
      new Relay(err, System.err).start();
      p.waitFor();
      //BUG : wait for io completion???
      System.exit(p.exitValue());
    } catch (Exception e) {
      e.printStackTrace();
      System.exit(1);
    }
  }

  public static class Relay extends Thread {

    private InputStream is;
    private OutputStream os;

    public Relay(InputStream is, OutputStream os) {
      this.is = is;
      this.os = os;
    }

    public void run() {
      byte buf[] = new byte[1024];
      try {
        while (true) {
          int len = is.read(buf);
          if (len > 0) {
            os.write(buf, 0, len);
          }
        }
      } catch (Exception e) {
      }
    }
  }
}