如何在Junit测试中解决jsch.addIdentity()?

时间:2016-06-28 19:03:26

标签: junit jsch apache-mina

我使用apache mina sshd为Junit测试目的设置一个模拟的ssh服务器。由于apache mina的文档很不清楚,我无法弄清楚如何处理测试中的身份验证问题。

代码我想测试,基本上使用Jsch将文件从本地传输到远程。

public static void scpTo(String rfilePath, String lfilePath, String user, String host, String keyPath, int port) {
    FileInputStream fis=null;
    try {
        while(true) {
            String rfile = rfilePath;
            String lfile = lfilePath;

            JSch jsch = new JSch();
            jsch.addIdentity(keyPath);
            java.util.Properties config = new java.util.Properties();
            config.put("StrictHostKeyChecking", "no");

            Session session = jsch.getSession(user, host, port);
            session.setConfig(config);

            session.connect();

            // exec 'scp -t rfile' remotely
            String command = "scp " + " -t " + rfile;
            Channel channel = session.openChannel("exec");
            ((ChannelExec) channel).setCommand(command);

            // get I/O streams for remote scp
            OutputStream out = channel.getOutputStream();
            InputStream in = channel.getInputStream();

            channel.connect();

            if (checkAck(in) != 0) {
                System.exit(0);
            }

            File _lfile = new File(lfile);
            if(!_lfile.exists()) {
                System.err.println("Local file not existing");
            }

            //check file existing
            File _rfile = new File(rfile);
            if (_rfile.exists()) {
                System.out.println("Remote file already existed");
                break;
            }

            // send "C0644 filesize filename", where filename should not include '/'
            long filesize = _lfile.length();
            command = "C0644 " + filesize + " ";
            if (lfile.lastIndexOf('/') > 0) {
                command += lfile.substring(lfile.lastIndexOf('/') + 1);
            } else {
                command += lfile;
            }
            command += "\n";
            out.write(command.getBytes());
            out.flush();
            if (checkAck(in) != 0) {
                System.exit(0);
            }

            // send a content of lfile
            fis = new FileInputStream(lfile);
            byte[] buf = new byte[1024];
            while (true) {
                int len = fis.read(buf, 0, buf.length);
                if (len <= 0) break;
                out.write(buf, 0, len); //out.flush();
            }
            fis.close();
            fis = null;
            // send '\0'
            buf[0] = 0;
            out.write(buf, 0, 1);
            out.flush();
            if (checkAck(in) != 0) {
                System.exit(0);
            }
            out.close();

            channel.disconnect();
            session.disconnect();

            //System.exit(0);

            Thread.sleep(2000);
        }
    } catch (Exception e) {
        System.out.println(e);
        try {
            if (fis != null) fis.close();
        } catch (Exception ee) {
        }
    }
}

用于测试的setUp。如果我想验证所有用户和&amp;通过在我的资源文件夹中使用给定的密钥文件来主机对,我该怎么做setUp?对于当前的setUp,它将具有Auth失败错误。

@Before
public void setup() {
    user = "siyangli";
    host = "localhost";
    //pick a port not occupied for tesing
    port = 22998;
    sshd = SshServer.setUpDefaultServer();
    sshd.setPort(port);
    keyPath = "/Users/siyangli/.ssh/id_rsa";
    //it will change the key file??? do not know why, how to work around the authentication key??
    sshd.setKeyPairProvider(new FileKeyPairProvider(new String[]{"/Users/siyangli/.ssh/id_rsa"}));
    //sshd.setKeyPairProvider(new SimpleGeneratorHostKeyProvider(keyPath));
    sshd.setPasswordAuthenticator(new PasswordAuthenticator() {
        public boolean authenticate(String username, String password, ServerSession session) {
            return true;
        }
    });
    sshd.setPublickeyAuthenticator(new PublickeyAuthenticator() {
        @Override
        public boolean authenticate(String username, PublicKey key, ServerSession session) {
            return true;
        }
    });
    CommandFactory myCommandFactory = new CommandFactory() {
        public Command createCommand(String command) {
            System.out.println("Command: " + command);
            return null;
        }
    };
    sshd.setCommandFactory(new ScpCommandFactory(myCommandFactory));
    List<NamedFactory<UserAuth>> userAuthFactories = new ArrayList<NamedFactory<UserAuth>>();
    userAuthFactories.add(new UserAuthPassword.Factory());
    sshd.setUserAuthFactories(userAuthFactories);
    List<NamedFactory<Command>> namedFactoryList = new ArrayList<NamedFactory<Command>>();
    namedFactoryList.add(new SftpSubsystem.Factory()); sshd.setSubsystemFactories(namedFactoryList);
    try {
        sshd.start();
    } catch (IOException e) {
        System.out.println(e.getMessage());
        e.printStackTrace();
    }
    System.out.println("Finished setup !!! ");
}

1 个答案:

答案 0 :(得分:0)

您需要更改此行: userAuthFactories.add(new UserAuthPassword.Factory()); 至 userAuthFactories.add(new UserAuthPublicKey.Factory());