JavaTuer/src/de/hacksaar/javatuer/TyshellClient.java

   1 package de.hacksaar.javatuer;
   2
   3 import com.jcraft.jsch.JSchException;
   4 import com.jcraft.jsch.UserInfo;
   5
   6 import java.io.IOException;
   7 import java.io.InputStreamReader;
   8 import java.io.OutputStreamWriter;
   9
  10 public class TyshellClient {
  11         private static final String TAG = "TyshellClient";
  12         public static final char END_OF_COMMAND = '\n';
  13         private final String hostname;
  14         private final int port;
  15         private final TuerLogging log;
  16         private SshClient client;
  17         private InputStreamReader inputStream;
  18         private OutputStreamWriter outputStream;
  19
  20         public TyshellClient(String hostname, int port) {
  21                 this(hostname, port, new DummyLogging());
  22         }
  23
  24         public TyshellClient(String hostname, int port, TuerLogging log) {
  25                 this.log = log;
  26                 this.hostname = hostname;
  27                 this.port = port;
  28         }
  29
  30         public void connect(String username, String password) {
  31                 try {
  32                         client = new SshClient(hostname, username, port, log);
  33                         client.login(password);
  34                         inputStream = new InputStreamReader(client.getInputStream());
  35                         outputStream = new OutputStreamWriter(client.getOutputStream());
  36                 } catch (JSchException | IOException e) {
  37                         log.exception(TAG, e);
  38                 }
  39         }
  40
  41         public void connect(String username, String keyFile, String passphrase) {
  42                 try {
  43                         client = new SshClient(hostname, username, port, log);
  44                         client.addPrivateKey(keyFile, passphrase);
  45                         client.login();
  46                         inputStream = new InputStreamReader(client.getInputStream());
  47                         outputStream = new OutputStreamWriter(client.getOutputStream());
  48                 } catch (JSchException | IOException e) {
  49                         log.exception(TAG, e);
  50                 }
  51         }
  52
  53         public void connect(String username, String keyFile, UserInfo interactiveLogin) {
  54                 try {
  55                         client = new SshClient(hostname, username, port, log);
  56                         client.addPrivateKey(keyFile);
  57                         client.login(interactiveLogin);
  58                         inputStream = new InputStreamReader(client.getInputStream());
  59                         outputStream = new OutputStreamWriter(client.getOutputStream());
  60                 } catch (JSchException | IOException e) {
  61                         log.exception(TAG, e);
  62                 }
  63         }
  64
  65         public void disconnect() {
  66                 try {
  67                         inputStream.close();
  68                         outputStream.close();
  69                 } catch (IOException e) {
  70                         log.exception(TAG, e);
  71                 }
  72                 client.disconnect();
  73         }
  74
  75         boolean isConnected() {
  76                 return (client.isConnected() && inputStream != null && outputStream != null);
  77         }
  78
  79         public InputStreamReader getInputStream() {
  80                 return inputStream;
  81         }
  82
  83         public void sendCommand(String command) {
  84                 if (isConnected()) {
  85                         try {
  86                                 assert outputStream != null;
  87                                 log.debug(TAG, "Sending: " + command);
  88                                 outputStream.write((command + END_OF_COMMAND).toCharArray());
  89                                 outputStream.flush();
  90                         } catch (IOException e) {
  91                                 log.exception(TAG, e);
  92                         }
  93                 }
  94         }
  95 }