source: other-projects/trunk/gs3-release-maker/tasks/sshtaskdef/src/mindbright/ssh/SSHExec.java@ 14627

/******************************************************************************
 *
 * Copyright (c) 1998,99 by Mindbright Technology AB, Stockholm, Sweden.
 *                 www.mindbright.se, [email protected]
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 *****************************************************************************
 * $Author: mats $
 * $Date: 2000/03/27 07:53:54 $
 * $Name: rel1-2-1 $
 *****************************************************************************/
package mindbright.ssh;

import java.io.*;
import java.net.Socket;

import mindbright.terminal.Terminal;
import mindbright.security.Cipher;

public class SSHExec extends SSHClientUserAdaptor implements SSHConsole {

    SSHClient        client;
    SSHInteractor    interactor = null;
    SSHClientUser    proxyUser  = null;
    SSHExecIndicator indicator  = null;
    SSHAuthenticator authenticator = null;

    SSHInteractor   ourInteractAdapter = null;

    File    cwd;
    boolean recursive;
    boolean verbose;

    String sshHost;

    PipedInputStream  inTop;
    PipedOutputStream inBottom;

    public SSHExec(String sshHost, int port, SSHAuthenticator authenticator,
           boolean verbose) throws IOException {
    super(sshHost, port);

    // OUCH: Note must be set before constructing SSHClient since
    // its constructor calls getInteractor to fetch this
    //
    ourInteractAdapter = new SSHInteractorAdapter() {
        public void open(SSHClient client) {
        SSHExec.this.open(client);
        }
        public void disconnected(SSHClient client, boolean graceful) {
        SSHExec.this.disconnected(client, graceful);
        }
        public void alert(String msg) {
        SSHExec.this.alert(msg);
        }
    };

    this.authenticator  = authenticator;
    client         = new SSHClient(authenticator, this);
    inTop          = new PipedInputStream();
    inBottom       = new PipedOutputStream(inTop);
    this.cwd       = cwd;
    this.verbose   = verbose;
    this.sshHost   = sshHost;

    client.setConsole(this);
    client.activateTunnels = false;
    }

    public void setInteractor(SSHInteractor interactor) {
    this.interactor = interactor;
    }

    public void setClientUser(SSHClientUser proxyUser) {
    this.proxyUser = proxyUser;
    }

    public void setIndicator(SSHExecIndicator indicator) {
    this.indicator = indicator;
    }

    public void abort() {
    interactor = null;
    indicator  = null;
    client.forcedDisconnect();
    }

    public void exec(String command) throws IOException {
    //String cmd = "scp -t " + (verbose ? "-v " : "") + authenticator.getUsername(this) + "@" + sshHost + " \"" + command + "\"";
    String cmd = command;
    //  System.out.println(cmd);
    client.doSingleCommand(cmd, true, 0);
    //  readResponse("After starting remote ssh exec");
    client.forcedDisconnect();

    }

    int readByte() throws IOException {
    return inTop.read();
    }

    String readString() throws IOException {
    byte[] buf = new byte[2048];
    int ch, i = 0;
    while(((ch = readByte()) != ((int)'\n')) && ch >= 0) {
        buf[i++] = (byte)ch;
    }
    if(ch == -1) {
        throw new EOFException();
    }
    if(buf[0] == (byte)'\n')
        throw new IOException("Unexpected <NL>");
    if(buf[0] == (byte)'\02' || buf[0] == (byte)'\01') {
        // !!!
        String errMsg = new String(buf, 1, i - 1);
        if(buf[0] == (byte)'\02')
        throw new IOException(errMsg);
        alert(errMsg);
        return null;
    }
    return new String(buf, 0, i);
    }

    void readResponse(String where) throws IOException {
    int r = readByte();
    if(r == 0) {
        // All is well, no error
        return;
    }
    String errMsg = readString();
    if(r == (byte)'\02')
        throw new IOException(errMsg);
    alert(errMsg);
    }

    public void stderrWriteString(byte[] str) {
    if(verbose) alert("Remote warning/error: " + new String(str));
    }

    public void stdoutWriteString(byte[] str) {
    try {
        inBottom.write(str);
    } catch(IOException e) {
        try {
        inBottom.close();
        } catch (IOException ee) {
        // !!!
        }
        alert("Error writing data to stdout-pipe");
    }
    }

    public Terminal getTerminal() {
    return null;
    }
    public void print(String str) {
    }
    public void println(String str) {
    }
    public void serverConnect(SSHChannelController controller, Cipher sndCipher) {
    }
    public void serverDisconnect(String reason) {
    }
    public boolean wantPTY() {
    return false;
    }
    public void open(SSHClient client) {
    if(indicator != null)
        indicator.connected(sshHost);
    }
    public void disconnected(SSHClient client, boolean graceful) {
    try {
        inBottom.close();
    } catch (IOException e) {
        // !!!
    }
    }
    public void alert(String msg) {
    if(interactor != null) {
        interactor.alert(msg);
    }
    }

    public Socket getProxyConnection() throws IOException {
    if(proxyUser != null) {
        return proxyUser.getProxyConnection();
    }
    return null;
    }

    public SSHInteractor getInteractor() {
    return ourInteractAdapter;
    }

}