Linux服务器批处理远程命令执行工具

package test.core; import java.io.BufferedReader; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.io.OutputStream; import java.util.Vector; import org.junit.Test; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import test.base.KeterSSHBaseTest; import com.jcraft.jsch.Channel; import com.jcraft.jsch.ChannelSftp; import com.jcraft.jsch.JSch; import com.jcraft.jsch.Session; import com.jcraft.jsch.SftpException; public class CommandTest extends KeterSSHBaseTest { /** * Logger for this class */ private static final Logger logger = LoggerFactory.getLogger(CommandTest.class); @Test public void test() throws Exception { // logger.info(runBash("id"));//ok // logger.info(runBash("pwd"));//ok // logger.info(runBash("ssh 192.168.1.100 echo haha"));//ok // String cmd_scp="scp /opt/keter/keter-framework/framework-todo.txt 192.168.1.100:/tmp/haha"; // runBash(cmd_scp); // sshSftp("192.168.1.100","root","1qaz!QAZ",-1);//ok sshSftp("192.168.1.100","~/.ssh/authorized_keys"); } public String runBash(String runCommand) throws InterruptedException { String result = null; // String returnVal = null; try { Runtime r = Runtime.getRuntime(); Process p = r.exec(runCommand); // Process p = r.exec("/bin/bash -c "+runCommand); BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream())); String inputLine; while ((inputLine = in.readLine()) != null) { // System.out.println(inputLine); result += inputLine; // returnVal = inputLine; } if(p.waitFor()!=0){ logger.error("command [{}] executed error!",runCommand); } in.close(); } catch (IOException e) { System.out.println(e); } return result; } public static void sshSftp(String ip, String user, String psw ,int port) throws Exception{ Session session = null; Channel channel = null; JSch jsch = new JSch(); if(port <=0){ //连接服务器，采用默认端口 session = jsch.getSession(user, ip); }else{ //采用指定的端口连接服务器 session = jsch.getSession(user, ip ,port); } //如果服务器连接不上，则抛出异常 if (session == null) { throw new Exception("session is null"); } //设置登陆主机的密码 session.setPassword(psw);//设置密码 //设置第一次登陆的时候提示，可选值：(ask | yes | no) session.setConfig("StrictHostKeyChecking", "no"); //设置登陆超时时间 session.connect(30000); try { //创建sftp通信通道 channel = (Channel) session.openChannel("sftp"); channel.connect(1000); ChannelSftp sftp = (ChannelSftp) channel; //进入服务器指定的文件夹 sftp.cd("/opt/keter"); //列出服务器指定的文件列表 Vector v = sftp.ls("*.txt"); for(int i=0;i<v.size();i++){ logger.info("server file is:[{}]",v.get(i)); } //以下代码实现从本地上传一个文件到服务器，如果要实现下载，对换以下流就可以了 putFile(sftp,"/opt/keter/keter-framework/framework-todo.txt","/opt/haha/haha.txt"); } catch (Exception e) { e.printStackTrace(); } finally { session.disconnect(); channel.disconnect(); } } /** * 利用JSch包实现SFTP下载、上传文件 * @param ip 主机IP * @param user 主机登陆用户名 * @param psw 主机登陆密码 * @param port 主机ssh2登陆端口，如果取默认值(默认值22)，传-1 * @param privateKey 密钥文件路径 * @param passphrase 密钥的密码 * */ public static void sshSftp(String ip, String privateKey) throws Exception{ Session session = null; Channel channel = null; JSch jsch = new JSch(); //设置不带口令的密钥 jsch.addIdentity(privateKey); //连接服务器，采用默认端口 session = jsch.getSession("root", ip); //如果服务器连接不上，则抛出异常 if (session == null) { throw new Exception("session is null"); } //设置登陆主机的密码 // session.setPassword(psw);//设置密码 //设置第一次登陆的时候提示，可选值：(ask | yes | no) session.setConfig("StrictHostKeyChecking", "no"); //设置登陆超时时间 session.connect(30000); try { //创建sftp通信通道 channel = (Channel) session.openChannel("sftp"); channel.connect(1000); ChannelSftp sftp = (ChannelSftp) channel; //以下代码实现从本地上传一个文件到服务器，如果要实现下载，对换以下流就可以了 putFile(sftp,"c:/desktop.ini","/opt/haha/desktop"); } catch (Exception e) { e.printStackTrace(); } finally { session.disconnect(); channel.disconnect(); } } public static void putFile(ChannelSftp sftp,String localFile,String remoteFile) throws SftpException, FileNotFoundException, IOException { InputStream instream = new FileInputStream(localFile); OutputStream outstream = sftp.put(remoteFile); byte b[] = new byte[1024]; int n; while ((n = instream.read(b)) != -1) { outstream.write(b, 0, n); } outstream.flush(); outstream.close(); instream.close(); // //进入服务器指定的文件夹 // sftp.cd("domains"); // // //列出服务器指定的文件列表 // Vector v = sftp.ls("*.txt"); // for(int i=0;i<v.size();i++){ // System.out.println(v.get(i)); // } } }