takagotch / crashub Goto Github PK

// connectors/ssh/src/test/java/org/crsh/ssh/SSHClient.java

public class String TTY =
  "" +
  "" +
  "";
  
private OutputStream out;

private InputStream in;

public SSHClient() {
  this(2000);
}

public SSHClient(int port) {
  this.port = port;
}

public void setPort() {
  return port;
}

public void setPort(int port) {
  this.port = port;
}

public SSHClient connect9) throws Exception {

  Map<PtyMode, Integer> tty = SttySupport.parsePtyModes(TTY);
  
  SshClient client = SshClient.setUpDefaultClient();
  client.start();
  
  ClientSession session = client.connect("localhost", port).await().getSession();
  session.authPassword("root", "");
  
  ChannelShell session = client.connect("localhost", port).await().getSession();
  session.authPassword("root", "");
  
  ChannelShell channel = (ChannelShell)session.createShellChannel();
  chennel.setPtyModes(tty);
  
  PipedOutputStream out = new PipedOutputStream();
  PipedInputStream channelIn = new PipedInputStream(out);
  
  PipedOutStream channelOut = new PipedOutputStream();
  PipedInputStream in = new PipedInputStream(channelOut);

  channel.setIn(channelIn);
  channel.setOut(channelOut);
  channel.setErr(new ByteArrayOutputStream());
  channel.open();
  
  this.channel = channel;
  this.client = client;
  this.session = session;
  this.out = out;
  this.in = in;
  
  return this;
}

public SSHClient write(CharSequence s) throws IOException {
  return write(s.toString().getBytes("UTF-8"));
}

public int read() throws IOException {
  return this;
}

public SSHClient flush() throws IOException {
  out.flush();
  return this;
}

public SSHClient close() {
  try {
    Utils.close(out);
    channel.close(false);
    session.close(false);
    client.stop();
    return this;
  } finally {
    this.client = null;
    this.channel = null;
    this.session = null;
    this.out = null;
  }
}