随时更改System.in并以编程方式读取System.out(Jsch)

时间:2018-08-23 14:08:35

标签: java ssh inputstream jsch

在我的代码中,我试图通过SSH在远程服务器上运行一些命令。 这些命令必须相互构建,但后面要有逻辑。这意味着: 当命令a的输出包含“ retcode 0”时,则执行命令b。其他命令c

我发现无法将这种逻辑实现为几个“ exec”命令。似乎每个“执行人员”都有自己的流程,因此我无法继续以前的工作。有了一个“ exec”,我就可以传递所有命令都将被执行的命令列表,因此那里没有逻辑。因此,我决定对Jsch使用“ shell”。 (如果有办法使用exec,我会很高兴的)

基于jcraft的示例,我编写了以下代码:

import com.jcraft.jsch.*;
import java.io.ByteArrayInputStream;
import java.io.InputStream;

public class Main {
    public static void main(String[] args) {
        try {
            JSch jsch = new JSch();
            String user = "sshuser";
            String host = "localhost";
            Session session = jsch.getSession(user, host, 22);
            String passwd = "password";
            session.setPassword(passwd);
            session.setConfig("StrictHostKeyChecking", "no");

            //session.connect();
            session.connect(30000);   // making a connection with timeout.
            Channel channel = session.openChannel("shell");

            // Enable agent-forwarding.
            ((ChannelShell)channel).setAgentForwarding(true);


            // Set Streams
            channel.setInputStream(System.in);
            channel.setOutputStream(System.out);

            channel.connect(3 * 1000);
        } catch (Exception e) {
            System.out.println(e);
        }
    }
}

基本上,这给了我作为人类确切的机会去做我想做的事情。我可以在System.in中输入命令,返回值将打印到System.out。我可以阅读它,决定要使用它做什么,然后输入下一个命令。该命令将在我之前的位置准确执行,所以一切都很好。

现在我必须找到一种通过java做到这一点的方法。我找到了一种通过修订字符串输入第一个命令的方法:

[...]
InputStream testInput = new ByteArrayInputStream( "dir \n".getBytes("UTF-8") );
// Set Streams
channel.setInputStream(testInput);
[...]

但是在那之后,我找不到发送下一个的方法(即使不读取输出,这也是第一步)。

所以,我的问题是,有没有办法通过Java代码设置System.in,它将直接通过Jsch发送此消息(System.setIn()对我不起作用)或另一种更改方式动态输入字符串,以便通过Jsch进行传输?

感谢您的时间!

1 个答案:

答案 0 :(得分:1)

多亏了Martin Prikryl的意见,我找到了解决方案。 我用Telnet而不是我的实际应用程序创建了一个小示例。基本原理是一样的,我认为它更有用,因为当它不是基于特定软件时,更多的人可以尝试并使用它。

import com.jcraft.jsch.*;
import java.io.*;

public class Main {
    public static void main(String[] args) {
        OutputStream out = null;
        Session session = null;
        try {
            JSch jsch = new JSch();
            String user = "sshuser";
            String host = "localhost";
            session = jsch.getSession(user, host, 22);
            String passwd = "password";
            session.setPassword(passwd);
            session.setConfig("StrictHostKeyChecking", "no");

            // vars and objects used later
            String lineSeperator = System.getProperty("line.separator");
            StringBuilder sb = new StringBuilder();
            Main main = new Main();

            //session.connect();
            session.connect(30000);   // making a connection with timeout.
            ChannelExec channel = (ChannelExec) session.openChannel("exec");
            InputStream in = channel.getInputStream();
            BufferedReader reader = new BufferedReader(new InputStreamReader(in));

            // start telnet session
            channel.setCommand("telnet 192.168.222.128 -l sshuser");
            out = channel.getOutputStream();
            channel.connect();
            // wait a little bit for telnet to be ready to take the input
            Thread.sleep(500);
            // pass the password
            out.write(("password\n").getBytes());
            out.write(("\n").getBytes());
            Thread.sleep(500);
            // flush reader, very important!
            out.flush();

            // Read from Bufferreader until the current line contains a specific string
            // For my real application it would be "---       END", for this example i
            // used something from the last line my machine returns. Very important that this string
            // appears on every possible output, or you will stuck in a while loop!
            //
            // Tried it with while((reader.readline())!=null) but this ends in a infinity loop too.
            // Since in my application there is an String that always get returned i didn't look it further up
            String responeFromLogin = main.readOutput("security updates.", reader, lineSeperator, sb);

            // Working with the response, in this example a simple fail-->Exception, success --> progress
            if (responeFromLogin.contains("Login incorrect")) {
                throw new Exception("Failed: Login");
            }
            System.out.println("Login Successfull");

            // Log in was successful, so lets do the next command, basiclly the same routine again
            out.write(("dir\n").getBytes());
            Thread.sleep(500);
            out.flush();
            // Again, not bulletproofed in this example
            String responseFromHelp = main.readOutput("examples.desktop", reader, lineSeperator, sb);
            if (!responseFromHelp.contains("test")) {
                throw new Exception("Failed: Help");
            }
            System.out.println("Folder Found");



        } catch (InterruptedException e1) {
            e1.printStackTrace();
        } catch (IOException e1) {
            e1.printStackTrace();
        } catch (JSchException e1) {
            e1.printStackTrace();
        } catch (Exception e1) {
            e1.printStackTrace();
        } finally {
            try {
                if (out != null) {
                    out.flush();
                }
            } catch (IOException e1) {
                e1.printStackTrace();
            }
            System.out.println("_________________________");
            System.out.println("I am done");
            if (session != null) {
                session.disconnect();
            }
        }
    }

    public String readOutput(String endString, BufferedReader reader, String lineSeperator, StringBuilder sb) {
        String line;
        String returnString = "Error";

        while (true) {
            try {
                line = reader.readLine();
                if (line.contains(endString)) {
                    sb.append(line).append(lineSeperator);
                    returnString = sb.toString();
                    break;
                } else {
                    sb.append(line).append(lineSeperator);
                }
            } catch (IOException e) {
                returnString = "Error";
                e.printStackTrace();
            }
        }
        return returnString;
    }
}