package com.fastech.utils;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;

import org.apache.log4j.Logger;

import ch.ethz.ssh2.Connection;
import ch.ethz.ssh2.Session;
import ch.ethz.ssh2.StreamGobbler;

public class SSHconnector {
public static final Logger log = Logger.getLogger(SSHconnector.class);
private Connection connection;
private Session session = null;
private static final int TIME_OUT = 30 * 1000;
/**
* 連接服務器
* */
public boolean connect(String ip,int port) {
connection = new Connection(ip,port);//獲取連接
try {
//connection
connection.connect(null, TIME_OUT, TIME_OUT);

} catch (IOException e) {
e.printStackTrace();
log.info("->SSHConnection connect failure: "+e);
return false;
}
log.info("Connect "+ip+":"+port+" Success!");
return true;
}
/**
* 執行命令
* */
public String sender(String command) {
try {
session = connection.openSession();
} catch (IOException e1) {
// TODO Auto-generated catch block
e1.printStackTrace();
}
StringBuffer sb = new StringBuffer();
InputStream inputstream = null;
InputStream inputstreamError = null;
BufferedReader br = null;
BufferedReader brError = null;
try {
session.execCommand(command);
inputstream = new StreamGobbler(session.getStdout());
inputstreamError = new StreamGobbler(session.getStderr());
br = new BufferedReader(new InputStreamReader(inputstream,"utf-8"));
brError = new BufferedReader(new InputStreamReader(inputstreamError,"utf-8"));
String line = "";
while ((line = br.readLine()) != null) {
sb.append(line + "\n");
}
boolean errorFlag = false;
while ((line = brError.readLine()) != null) {
if (!errorFlag) sb.append("Result:8 ");
sb.append(line + ";\n");
errorFlag = true;

}


} catch (IOException e) {
log.info("->SSHConnection send command failure: "+e);
e.printStackTrace();
} finally {
try {
if (br != null) {
br.close();
}
if (brError != null) {
brError.close();
}
if (inputstream != null) {
inputstream.close();
}
if (inputstreamError != null) {

inputstreamError.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
if (sb.toString() != null && !"".equals(sb.toString())) {
String result = sb.substring(0, sb.length() - 1);
if (!result.contains("Result:")) {
return "Result:4 " + result + ";";
}
return result;//截取最後一個\n;
}

return null;
}
/**
* 關閉連接
* */
public void destory() {
if (session != null) {
session.close();
}
if (connection != null) {
connection.close();
}
}
/**
* 登錄服務器
* */
public Boolean login(String username,String password) {
try {
boolean isAuthenticate = connection.authenticateWithPassword(username,password);

if (!isAuthenticate) {
log.info("->SSHConnection login failure: ");
}
} catch (Exception e) {
log.info("->SSHConnection login failure: "+e);
e.printStackTrace();
return false;
}
return true;
}
}

SSH連接服務器