2016-06-28 27 views
0

我使用apache mina sshd爲Junit測試目的設置了一個模擬的ssh服務器。由於apache mina的文檔相當不清楚,我無法弄清楚如何處理測試中的身份驗證問題。如何解決Junit測試中的jsch.addIdentity()問題?

代碼我想測試一下,基本上用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用於測試。如果我想通過使用我的資源文件夾中的給定密鑰文件來驗證所有用戶&主機對,我應該爲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 !!! "); 
} 
+0

提示:你的方法scpTo正在做**許多**事情的方式。我認真地推薦您查看Robert Martin的「Clean Code」...或者花一些時間觀看https://www.youtube.com/playlist?list=PLD0011D00849E1B79 ...您的代碼可以從中獲益。 .. – GhostCat

回答

0

你需要改變這一行: userAuthFactories.add(新UserAuthPassword.Factory()); 至 userAuthFactories.add(new UserAuthPublicKey.Factory());