SSHConnectivity.java

package com.yeokm1.nussocprintandroid.network;

import android.content.Context;
import android.util.Log;

import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import com.jcraft.jsch.SftpException;
import com.jcraft.jsch.SftpProgressMonitor;
import com.yeokm1.nussocprintandroid.R;

import java.io.IOException;
import java.io.InputStream;

/**
 * Created by yeokm1 on 6/10/2014.
 */
public class SSHConnectivity {

    private static final int portNumber = 22;
    private static final int timeout = 60000;
    private static final String TAG = "SSHConnectivity";

    private JSch jschSSHChannel;
    private Session session;
    private String hostname;
    private String username;
    private String password;
    private Context context;

    public SSHConnectivity(String hostname, String username, String password, Context context) {
        this.hostname = hostname;
        this.username = username;
        this.password = password;
        this.context = context;
        this.jschSSHChannel = new JSch();
    }


    public void connect() throws Exception {
        Log.i(TAG, "connect");
        try {
            session = jschSSHChannel.getSession(username, hostname, portNumber);
            session.setPassword(password);
            // To make things easier for user, I skip key check
            session.setConfig("StrictHostKeyChecking", "no");
            session.connect(timeout);

        } catch (JSchException e) {
            if (e.getMessage().equals("Auth fail")) {
                String wrongCredentials = context.getString(R.string.misc_wrong_credentials);
                throw new Exception(wrongCredentials);
            } else {
                throw e;
            }
        }
    }


    public void disconnect() {
        if (session != null) {
            Log.i(TAG, "disconnect");
            session.disconnect();
            session = null;
        }
    }


    public String runCommand(String command) throws Exception {
        Log.i(TAG + " runCommand", command);

        StringBuilder outputBuffer = new StringBuilder();

        Channel channel = session.openChannel("exec");
        ((ChannelExec) channel).setCommand(command);
        channel.connect();
        InputStream commandOutput = channel.getInputStream();
        int readByte = commandOutput.read();

        while (readByte != 0xffffffff) {
            outputBuffer.append((char) readByte);
            readByte = commandOutput.read();
        }

        channel.disconnect();

        String output = outputBuffer.toString();

        Log.i(TAG + " runCommand", command + ", output: " + output);
        return output;
    }

    public void uploadFile(InputStream toBePrinted, String directory, String filename, SftpProgressMonitor progressMonitor) throws SftpException, JSchException, IOException {
        if(session== null){
            throw new JSchException(context.getString(R.string.misc_connection_not_set_up));
        }

        Channel channel = session.openChannel("sftp");

        channel.connect();
        ChannelSftp channelSftp = (ChannelSftp)channel;

        try{
            channelSftp.mkdir(directory);
        } catch (SftpException e){
            //If cannot make directory, means directory already created
        }
        channelSftp.cd(directory);
        channelSftp.put(toBePrinted, filename, progressMonitor, ChannelSftp.OVERWRITE);
        channelSftp.disconnect();
    }


}