java串連SSH伺服器並執行shell命令

package com.flyingzl.ssh;import java.util.ArrayList
import java.util.Hashtable
import java.util.List
import org.apache.log4j.Logger
import org.apache.oro.text.regex.MalformedPatternException
import com.jcraft.jsch.ChannelShell
import com.jcraft.jsch.JSch
import com.jcraft.jsch.Session
import com.jcraft.jsch.UserInfo
import expect4j.Closure
import expect4j.Expect4j
import expect4j.ExpectState
import expect4j.matches.EofMatch
import expect4j.matches.Match
import expect4j.matches.RegExpMatch
import expect4j.matches.TimeoutMatch
public class Shell {    private static Logger log = Logger.getLogger(Shell.class)
        private Session session
    private ChannelShell channel
    private static Expect4j expect = null
    private static final long defaultTimeOut = 1000
    private StringBuffer buffer=new StringBuffer()
        public static final int COMMAND_EXECUTION_SUCCESS_OPCODE = -2
    public static final String BACKSLASH_R = "\r"
    public static final String BACKSLASH_N = "\n"
    public static final String COLON_CHAR = ":"
    public static String ENTER_CHARACTER = BACKSLASH_R
    public static final int SSH_PORT = 22
        //正則匹配，用於處理伺服器返回的結果    public static String[] linuxPromptRegEx = new String[] { "~]#", "~#", "#",            ":~#", "/$", ">" }
        public static String[] errorMsg=new String[]{"could not acquire the config lock "}
        //ssh伺服器的ip地址    private String ip
    //ssh伺服器的登入連接埠    private int port
    //ssh伺服器的登入使用者名稱    private String user
    //ssh伺服器的登入密碼    private String password
        public Shell(String ip,int port,String user,String password) {        this.ip=ip
        this.port=port
        this.user=user
        this.password=password
        expect = getExpect()
    }        /**     * 關閉SSH遠端連線     */    public void disconnect(){        if(channel!=null){            channel.disconnect()
        }        if(session!=null){            session.disconnect()
        }    }        /**     * 擷取伺服器返回的資訊     * @return 服務端的執行結果     */    public String getResponse(){        return buffer.toString()
    }        //獲得Expect4j對象，該對用可以往SSH發送命令請求    private Expect4j getExpect() {        try {            log.debug(String.format("Start logging to
%s@%s:%s",user,ip,port))
            JSch jsch = new JSch()
            session = jsch.getSession(user, ip, port)
            session.setPassword(password)
            Hashtable<String, String> config = new Hashtable<String, String>()
            config.put("StrictHostKeyChecking", "no")
            session.setConfig(config)
            localUserInfo ui = new localUserInfo()
            session.setUserInfo(ui)
            session.connect()
            channel = (ChannelShell) session.openChannel("shell")
            Expect4j expect = new Expect4j(channel.getInputStream(), channel                    .getOutputStream())
            channel.connect()
            log.debug(String.format("Logging to %s@%s:%s successfully!",user,ip,port))
            return expect
        } catch (Exception ex) {            log.error("Connect to "+ip+":"+port+"failed,please check your username and password!")
            ex.printStackTrace()
        }        return null
    }    /**     * 執行配置命令     * @param commands 要執行的命令，為字元數組     * @return 執行是否成功     */    public boolean executeCommands(String[] commands) {        //如果expect返回為0，說明登入沒有成功        if(expect==null){            return false
        }                log.debug("----------Running commands are listed as follows:----------")
        for(String command:commands){            log.debug(command)
        }        log.debug("----------End----------")
                Closure closure = new Closure() {            public void run(ExpectState expectState) throws Exception {                buffer.append(expectState.getBuffer())
// buffer is string                                                        // buffer for appending                                                        // output of executed                                                        // command               
expectState.exp_continue()
                            }        }
        List<Match> lstPattern = new ArrayList<Match>()
        String[] regEx = linuxPromptRegEx
        if (regEx != null && regEx.length > 0) {            synchronized (regEx) {                for (String regexElement : regEx) {// list of regx like, :>, />                                                    // etc. it is possible                                                   
// command prompts of your                                                    // remote machine                    try {                        RegExpMatch mat = new RegExpMatch(regexElement, closure)
                        lstPattern.add(mat)
                    } catch (MalformedPatternException e) {                        return false
                    } catch (Exception e) {                        return false
                    }                }                lstPattern.add(new EofMatch(new Closure() { // should cause                                                            // entire page to be                                                            // collected                           
public void run(ExpectState state) {                            }                        }))
                lstPattern.add(new TimeoutMatch(defaultTimeOut, new Closure() {                    public void run(ExpectState state) {                    }                }))
            }        }        try {            boolean isSuccess = true
            for (String strCmd : commands){                isSuccess = isSuccess(lstPattern, strCmd)
            }            //防止最後一個命令執行不了            isSuccess = !checkResult(expect.expect(lstPattern))
                        //找不到錯誤資訊標示成功            String response=buffer.toString().toLowerCase()
            for(String msg:errorMsg){                if(response.indexOf(msg)>-1){                    return false
                }            }                        return isSuccess
        } catch (Exception ex) {            ex.printStackTrace()
            return false
        }    }    //檢查執行是否成功    private boolean isSuccess(List<Match> objPattern, String strCommandPattern) {        try {            boolean isFailed = checkResult(expect.expect(objPattern))
            if (!isFailed) {                expect.send(strCommandPattern)
                expect.send("\r")
                return true
            }            return false
        } catch (MalformedPatternException ex) {            return false
        } catch (Exception ex) {            return false
        }    }    //檢查執行返回的狀態    private boolean checkResult(int intRetVal) {        if (intRetVal == COMMAND_EXECUTION_SUCCESS_OPCODE) {            return true
        }        return false
    }        //登入SSH時的控制資訊    //設定不提示輸入密碼、不顯示登入資訊等    public static class localUserInfo implements UserInfo {        String passwd
        public String getPassword() {            return passwd
        }        public boolean promptYesNo(String str) {            return true
        }        public String getPassphrase() {            return null
        }        public boolean promptPassphrase(String message) {            return true
        }        public boolean promptPassword(String message) {            return true
        }        public void showMessage(String message) {                    }    }}

 

轉載http://www.cnblogs.com/flyingzl/articles/2145032.html