devbox
小丑西瓜9
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

Apache MINA SSHD

作者：小丑西瓜9 | 2024-04-13 13:16:27

赞

踩

Apache MINA SSHD

      

目录

1.远程登录

1.1密码登录

1.2密钥登录

2.执行命令

2.1 ChannelExec

2.2 ChannelShell

3.文件传输

 3.1 上传文件

3.2 下载文件

3.3 SftpFileSystem

       Apache MINA SSHD（Secure Shell Daemon）是基于Apache MINA（Multipurpose Infrastructure for Network Applications）开发的一个开源的Java库，专门用于提供SSH（Secure Shell）服务。SSH是一种网络协议，用于在不安全的网络环境中进行安全通信和远程操作：主要用于远程登录、文件传输、以及安全的命令执行等场景。

1.远程登录

1.1密码登录

  1. package com.yichenkeji.starter.ssh;
  2.  
  3. import lombok.extern.slf4j.Slf4j;
  4. import org.apache.sshd.client.SshClient;
  5. import org.apache.sshd.client.channel.ChannelExec;
  6. import org.apache.sshd.client.channel.ChannelShell;
  7. import org.apache.sshd.client.channel.ClientChannelEvent;
  8. import org.apache.sshd.client.session.ClientSession;
  9. import org.apache.sshd.sftp.client.SftpClientFactory;
  10. import org.apache.sshd.sftp.client.fs.SftpFileSystem;
  11.  
  12. import java.io.*;
  13. import java.nio.file.Files;
  14. import java.nio.file.Path;
  15. import java.nio.file.Paths;
  16. import java.security.KeyPair;
  17. import java.security.KeyPairGenerator;
  18. import java.util.EnumSet;
  19. import java.util.Set;
  20. import java.util.concurrent.TimeUnit;
  21.  
  22. /**
  23.  * Sshd工具类
  24.  */
  25. @Slf4j
  26. public class SshdTest {
  27. 	private SshClient client;
  28. 	private ClientSession session;
  29. 	public static void main(String[] args) throws Exception {
  30. 		
  31.  
  32. 	}
  33.  
  34.     /**
  35.      * 连接
  36. 	 * @param host
  37.      * @param port
  38.      * @param username
  39. 	 */
  40. 	private void connect(String host, int port, String username)  {
  41. 		client = SshClient.setUpDefaultClient();
  42. 		client.start();
  43. 		try  {
  44. 			session = client.connect(username, host, port)
  45. 					.verify(5000)
  46. 					.getSession();
  47.  
  48. 		} catch (IOException e) {
  49. 			throw new RuntimeException(e);
  50. 		}
  51.  
  52. 	}
  53.  
  54. 	/**
  55. 	 * 密码登录
  56. 	 * @param host
  57. 	 * @param port
  58. 	 * @param username
  59. 	 * @param password
  60. 	 */
  61. 	public SshdTest(String host, int port, String username, String password) {
  62. 		connect(host,port,username);
  63. 		try  {
  64. 			session.addPasswordIdentity(password); // for password-based authentication
  65.  
  66. 			if (session.auth().verify(5000).isFailure()) {
  67. 				throw new RuntimeException("验证失败");
  68. 			}
  69. 		} catch (IOException e) {
  70. 			throw new RuntimeException(e);
  71. 		}
  72.  
  73. 	}
  74.  
  75.  
  76.  
  77.  
  78. 	/**
  79. 	 * 关闭连接
  80. 	 */
  81. 	public void close(){
  82. 		log.info("关闭 SSH");
  83. 		closeSession();
  84. 		if(client != null){
  85. 			try {
  86. 				client.close();
  87. 			} catch (IOException e) {
  88. 				log.error(e.getMessage());
  89. 			}
  90. 		}
  91.  
  92. 	}
  93. 	private void closeSession() {
  94. 		if(session != null){
  95. 			try {
  96. 				session.close();
  97. 			} catch (IOException e) {
  98. 				log.error(e.getMessage());
  99. 			}
  100. 		}
  101. 	}
  102. }

1.2密钥登录

  1. package com.yichenkeji.starter.ssh;
  2.  
  3. import lombok.extern.slf4j.Slf4j;
  4. import org.apache.sshd.client.SshClient;
  5. import org.apache.sshd.client.channel.ChannelExec;
  6. import org.apache.sshd.client.channel.ChannelShell;
  7. import org.apache.sshd.client.channel.ClientChannelEvent;
  8. import org.apache.sshd.client.session.ClientSession;
  9. import org.apache.sshd.sftp.client.SftpClientFactory;
  10. import org.apache.sshd.sftp.client.fs.SftpFileSystem;
  11.  
  12. import java.io.*;
  13. import java.nio.file.Files;
  14. import java.nio.file.Path;
  15. import java.nio.file.Paths;
  16. import java.security.KeyPair;
  17. import java.security.KeyPairGenerator;
  18. import java.util.EnumSet;
  19. import java.util.Set;
  20. import java.util.concurrent.TimeUnit;
  21.  
  22. /**
  23.  * Sshd工具类
  24.  */
  25. @Slf4j
  26. public class SshdTest {
  27. 	private SshClient client;
  28. 	private ClientSession session;
  29. 	public static void main(String[] args) throws Exception {
  30.  
  31.  
  32. 	}
  33.  
  34.     /**
  35.      * 连接
  36. 	 * @param host
  37.      * @param port
  38.      * @param username
  39. 	 */
  40. 	private void connect(String host, int port, String username)  {
  41. 		client = SshClient.setUpDefaultClient();
  42. 		client.start();
  43. 		try  {
  44. 			session = client.connect(username, host, port)
  45. 					.verify(5000)
  46. 					.getSession();
  47.  
  48. 		} catch (IOException e) {
  49. 			throw new RuntimeException(e);
  50. 		}
  51.  
  52. 	}
  53.  
  54. 	/**
  55. 	 * 密钥登录
  56. 	 * @param host
  57. 	 * @param port
  58. 	 * @param username
  59. 	 */
  60. 	public SshdTest(String host, int port, String username) {
  61. 		connect(host,port,username);
  62. 		try  {
  63.  
  64. 			String privateKeyPath = System.getProperty("user.home") + "/.ssh/id_rsa";
  65. 			String privateKeyContent = new String(Files.readAllBytes(Paths.get(privateKeyPath)));
  66.  
  67. 			//获取密钥对
  68. 			KeyPairGenerator rsa = KeyPairGenerator.getInstance("RSA");
  69. 			KeyPair keyPair = rsa.generateKeyPair();
  70. 			ByteArrayOutputStream stream = new ByteArrayOutputStream();
  71. 			stream.write(privateKeyContent.getBytes());
  72. 			ObjectOutputStream o = new ObjectOutputStream(stream);
  73. 			o.writeObject(keyPair);
  74.  
  75. 			session.addPublicKeyIdentity(keyPair);
  76.  
  77.  
  78. 			if (session.auth().verify(5000).isFailure()) {
  79. 				throw new RuntimeException("验证失败");
  80. 			}
  81. 		} catch (Exception e) {
  82. 			throw new RuntimeException(e);
  83. 		}
  84.  
  85. 	}
  86.  
  87.  
  88.  
  89. 	/**
  90. 	 * 关闭连接
  91. 	 */
  92. 	public void close(){
  93. 		log.info("关闭 SSH");
  94. 		closeSession();
  95. 		if(client != null){
  96. 			try {
  97. 				client.close();
  98. 			} catch (IOException e) {
  99. 				log.error(e.getMessage());
  100. 			}
  101. 		}
  102.  
  103. 	}
  104. 	private void closeSession() {
  105. 		if(session != null){
  106. 			try {
  107. 				session.close();
  108. 			} catch (IOException e) {
  109. 				log.error(e.getMessage());
  110. 			}
  111. 		}
  112. 	}
  113. }

2.执行命令

2.1 ChannelExec

        ChannelExec是Apache Mina SSHD中的一个类，它提供了一种在SSH连接上执行远程命令的方式，以及处理命令输入、输出、参数和状态的能力。它灵活性高、可扩展性强，适用于需要与远程服务器进行命令交互和执行的场景。

  1. package com.yichenkeji.starter.ssh;
  2.  
  3. import lombok.extern.slf4j.Slf4j;
  4. import org.apache.sshd.client.SshClient;
  5. import org.apache.sshd.client.channel.ChannelExec;
  6. import org.apache.sshd.client.channel.ChannelShell;
  7. import org.apache.sshd.client.channel.ClientChannelEvent;
  8. import org.apache.sshd.client.session.ClientSession;
  9. import org.apache.sshd.sftp.client.SftpClient;
  10. import org.apache.sshd.sftp.client.SftpClientFactory;
  11. import org.apache.sshd.sftp.client.fs.SftpFileSystem;
  12.  
  13. import java.io.*;
  14. import java.nio.file.Files;
  15. import java.nio.file.Path;
  16. import java.nio.file.Paths;
  17. import java.nio.file.StandardCopyOption;
  18. import java.security.KeyPair;
  19. import java.security.KeyPairGenerator;
  20. import java.util.EnumSet;
  21. import java.util.Set;
  22. import java.util.concurrent.TimeUnit;
  23.  
  24. /**
  25.  * Sshd工具类
  26.  */
  27. @Slf4j
  28. public class SshdTest {
  29. 	private SshClient client;
  30. 	private ClientSession session;
  31. 	public static void main(String[] args) throws Exception {
  32.  
  33. 		SshdTest sshUtil = new SshdTest("192.168.179.131",22,"weisx","123456");
  34. 		sshUtil.execCommand("pwd");
  35. 	}
  36. 	
  37.     /**
  38.      * 连接
  39. 	 * @param host
  40.      * @param port
  41.      * @param username
  42. 	 */
  43. 	private void connect(String host, int port, String username)  {
  44. 		client = SshClient.setUpDefaultClient();
  45. 		client.start();
  46. 		try  {
  47. 			session = client.connect(username, host, port)
  48. 					.verify(5000)
  49. 					.getSession();
  50.  
  51. 		} catch (IOException e) {
  52. 			throw new RuntimeException(e);
  53. 		}
  54.  
  55. 	}
  56.  
  57. 	/**
  58. 	 * 密码登录
  59. 	 * @param host
  60. 	 * @param port
  61. 	 * @param username
  62. 	 * @param password
  63. 	 */
  64. 	public SshdTest(String host, int port, String username, String password) {
  65. 		connect(host,port,username);
  66. 		try  {
  67. 			session.addPasswordIdentity(password); // for password-based authentication
  68.  
  69. 			if (session.auth().verify(5000).isFailure()) {
  70. 				throw new RuntimeException("验证失败");
  71. 			}
  72. 		} catch (IOException e) {
  73. 			throw new RuntimeException(e);
  74. 		}
  75.  
  76. 	}
  77. 	/**
  78. 	 * 密钥登录
  79. 	 * @param host
  80. 	 * @param port
  81. 	 * @param username
  82. 	 */
  83. 	public SshdTest(String host, int port, String username) {
  84. 		connect(host,port,username);
  85. 		try  {
  86.  
  87. 			String privateKeyPath = System.getProperty("user.home") + "/.ssh/id_rsa";
  88. 			String privateKeyContent = new String(Files.readAllBytes(Paths.get(privateKeyPath)));
  89.  
  90. 			//获取密钥对
  91. 			KeyPairGenerator rsa = KeyPairGenerator.getInstance("RSA");
  92. 			KeyPair keyPair = rsa.generateKeyPair();
  93. 			ByteArrayOutputStream stream = new ByteArrayOutputStream();
  94. 			stream.write(privateKeyContent.getBytes());
  95. 			ObjectOutputStream o = new ObjectOutputStream(stream);
  96. 			o.writeObject(keyPair);
  97.  
  98. 			session.addPublicKeyIdentity(keyPair);
  99.  
  100.  
  101. 			if (session.auth().verify(5000).isFailure()) {
  102. 				throw new RuntimeException("验证失败");
  103. 			}
  104. 		} catch (Exception e) {
  105. 			throw new RuntimeException(e);
  106. 		}
  107.  
  108. 	}
  109.  
  110. 	/**
  111. 	 * 执行命令
  112. 	 * @param command
  113. 	 * @return
  114. 	 */
  115. 	public String execCommand(String command){
  116. 		session.resetAuthTimeout();
  117. 		System.out.println("exe cmd: "+ command);
  118.  
  119. 		// 返回结果流
  120. 		ByteArrayOutputStream out = new ByteArrayOutputStream();
  121. 		// 错误信息
  122. 		ByteArrayOutputStream err = new ByteArrayOutputStream();
  123. 		try(ChannelExec channelExec = session.createExecChannel(command)) {
  124.  
  125. 			channelExec.setOut(out);
  126. 			channelExec.setErr(err);
  127. 			// 执行并等待
  128. 			channelExec.open();
  129.  
  130. 			Set<ClientChannelEvent> events =
  131. 					channelExec.waitFor(
  132. 							EnumSet.of(ClientChannelEvent.CLOSED),
  133. 							TimeUnit.SECONDS.toMillis(100000));
  134. 			// 检查请求是否超时
  135. 			if (events.contains(ClientChannelEvent.TIMEOUT)) {
  136. 				throw new RuntimeException("请求连接超时");
  137. 			}
  138. 			// 一般退出状态为0表示正常
  139. 			int exitStatus = channelExec.getExitStatus();
  140. 			if (exitStatus == 1) {
  141. 				log.info("exitStatus:{}",exitStatus);
  142. //				throw new RuntimeException("执行命令失败："+exitStatus);
  143. 			}
  144. 		} catch (Exception e) {
  145. 			throw new RuntimeException(e);
  146. 		}
  147. 		return out.toString().trim();
  148. 	}
  149.  
  150. 	/**
  151. 	 * 关闭连接
  152. 	 */
  153. 	public void close(){
  154. 		log.info("关闭 SSH");
  155. 		closeSession();
  156. 		if(client != null){
  157. 			try {
  158. 				client.close();
  159. 			} catch (IOException e) {
  160. 				log.error(e.getMessage());
  161. 			}
  162. 		}
  163.  
  164. 	}
  165. 	private void closeSession() {
  166. 		if(session != null){
  167. 			try {
  168. 				session.close();
  169. 			} catch (IOException e) {
  170. 				log.error(e.getMessage());
  171. 			}
  172. 		}
  173. 	}
  174. }

2.2 ChannelShell

        ChannelShell是Apache Mina SSHD中的一个用于使用交互式shell的类，它提供了在远程服务器上执行交互式命令和脚本，并与其进行实时交互的功能。与 ChannelExec 相比，可以更灵活地与远程服务器进行交互和执行复杂的命令。

  1. package com.yichenkeji.starter.ssh;
  2.  
  3. import lombok.extern.slf4j.Slf4j;
  4. import org.apache.sshd.client.SshClient;
  5. import org.apache.sshd.client.channel.ChannelExec;
  6. import org.apache.sshd.client.channel.ChannelShell;
  7. import org.apache.sshd.client.channel.ClientChannelEvent;
  8. import org.apache.sshd.client.session.ClientSession;
  9. import org.apache.sshd.sftp.client.SftpClient;
  10. import org.apache.sshd.sftp.client.SftpClientFactory;
  11. import org.apache.sshd.sftp.client.fs.SftpFileSystem;
  12.  
  13. import java.io.*;
  14. import java.nio.file.Files;
  15. import java.nio.file.Path;
  16. import java.nio.file.Paths;
  17. import java.nio.file.StandardCopyOption;
  18. import java.security.KeyPair;
  19. import java.security.KeyPairGenerator;
  20. import java.util.EnumSet;
  21. import java.util.Set;
  22. import java.util.concurrent.TimeUnit;
  23.  
  24. /**
  25.  * Sshd工具类
  26.  */
  27. @Slf4j
  28. public class SshdTest {
  29. 	private SshClient client;
  30. 	private ClientSession session;
  31. 	public static void main(String[] args) throws Exception {
  32.  
  33. 		SshdTest sshUtil = new SshdTest("192.168.179.131",22,"weisx","123456");
  34. 		String[] commends ={"pwd","cd ","pwd"};
  35. 		sshUtil.execCommand(commends);
  36. 	}
  37.  
  38.     /**
  39.      * 连接
  40. 	 * @param host
  41.      * @param port
  42.      * @param username
  43. 	 */
  44. 	private void connect(String host, int port, String username)  {
  45. 		client = SshClient.setUpDefaultClient();
  46. 		client.start();
  47. 		try  {
  48. 			session = client.connect(username, host, port)
  49. 					.verify(5000)
  50. 					.getSession();
  51.  
  52. 		} catch (IOException e) {
  53. 			throw new RuntimeException(e);
  54. 		}
  55.  
  56. 	}
  57.  
  58. 	/**
  59. 	 * 密码登录
  60. 	 * @param host
  61. 	 * @param port
  62. 	 * @param username
  63. 	 * @param password
  64. 	 */
  65. 	public SshdTest(String host, int port, String username, String password) {
  66. 		connect(host,port,username);
  67. 		try  {
  68. 			session.addPasswordIdentity(password); // for password-based authentication
  69.  
  70. 			if (session.auth().verify(5000).isFailure()) {
  71. 				throw new RuntimeException("验证失败");
  72. 			}
  73. 		} catch (IOException e) {
  74. 			throw new RuntimeException(e);
  75. 		}
  76.  
  77. 	}
  78. 	/**
  79. 	 * 密钥登录
  80. 	 * @param host
  81. 	 * @param port
  82. 	 * @param username
  83. 	 */
  84. 	public SshdTest(String host, int port, String username) {
  85. 		connect(host,port,username);
  86. 		try  {
  87.  
  88. 			String privateKeyPath = System.getProperty("user.home") + "/.ssh/id_rsa";
  89. 			String privateKeyContent = new String(Files.readAllBytes(Paths.get(privateKeyPath)));
  90.  
  91. 			//获取密钥对
  92. 			KeyPairGenerator rsa = KeyPairGenerator.getInstance("RSA");
  93. 			KeyPair keyPair = rsa.generateKeyPair();
  94. 			ByteArrayOutputStream stream = new ByteArrayOutputStream();
  95. 			stream.write(privateKeyContent.getBytes());
  96. 			ObjectOutputStream o = new ObjectOutputStream(stream);
  97. 			o.writeObject(keyPair);
  98.  
  99. 			session.addPublicKeyIdentity(keyPair);
  100.  
  101.  
  102. 			if (session.auth().verify(5000).isFailure()) {
  103. 				throw new RuntimeException("验证失败");
  104. 			}
  105. 		} catch (Exception e) {
  106. 			throw new RuntimeException(e);
  107. 		}
  108.  
  109. 	}
  110.  
  111. 	/**
  112. 	 * 执行命令
  113. 	 * @param commands
  114. 	 * @return
  115. 	 */
  116. 	public String execCommand(String[] commands){
  117. 		session.resetAuthTimeout();
  118. 		System.out.println("exe cmd: "+ String.join(";",commands));
  119. 		// 命令返回的结果
  120. 		StringBuffer resultBuf = new StringBuffer("");
  121. 		try (ChannelShell channel = session.createShellChannel()) {
  122. 			channel.open().verify(5 , TimeUnit.SECONDS);
  123. 			// 执行命令
  124. 			try (PrintStream out = new PrintStream(channel.getInvertedIn())) {
  125. 				for (String command:commands){
  126. 					out.println(command);
  127. 					out.flush();
  128. 				}
  129.  
  130. 				out.println("exit");
  131. 				out.flush();
  132. 			}
  133. 			//channel.waitFor(Collections.singleton(ClientChannelEvent.CLOSED), 0);
  134. 			try (BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(channel.getInvertedOut()))) {
  135. 				String line;
  136. 				while ((line = bufferedReader.readLine()) != null) {
  137. 					System.out.println(line);
  138. 					resultBuf.append(line);
  139. 				}
  140. 			}
  141. 		} catch (IOException e) {
  142. 			throw new RuntimeException(e);
  143. 		}
  144. 		return resultBuf.toString().trim();
  145. 	}
  146.  
  147. 	/**
  148. 	 * 关闭连接
  149. 	 */
  150. 	public void close(){
  151. 		log.info("关闭 SSH");
  152. 		closeSession();
  153. 		if(client != null){
  154. 			try {
  155. 				client.close();
  156. 			} catch (IOException e) {
  157. 				log.error(e.getMessage());
  158. 			}
  159. 		}
  160.  
  161. 	}
  162. 	private void closeSession() {
  163. 		if(session != null){
  164. 			try {
  165. 				session.close();
  166. 			} catch (IOException e) {
  167. 				log.error(e.getMessage());
  168. 			}
  169. 		}
  170. 	}
  171. }

3.文件传输

        SftpClient是基于Apache MINA SSHD库的一个SFTP（SSH File Transfer Protocol）客户端实现，它提供了一套方便且易于使用的API，用于在Java应用程序中与远程SFTP服务器进行文件传输和管理。

 3.1 上传文件

  1. package com.yichenkeji.starter.ssh;
  2.  
  3. import lombok.extern.slf4j.Slf4j;
  4. import org.apache.sshd.client.SshClient;
  5. import org.apache.sshd.client.channel.ChannelExec;
  6. import org.apache.sshd.client.channel.ChannelShell;
  7. import org.apache.sshd.client.channel.ClientChannelEvent;
  8. import org.apache.sshd.client.session.ClientSession;
  9. import org.apache.sshd.sftp.client.SftpClient;
  10. import org.apache.sshd.sftp.client.SftpClientFactory;
  11. import org.apache.sshd.sftp.client.fs.SftpFileSystem;
  12. import org.apache.sshd.sftp.client.fs.SftpPath;
  13.  
  14. import java.io.*;
  15. import java.nio.file.Files;
  16. import java.nio.file.Path;
  17. import java.nio.file.Paths;
  18. import java.nio.file.StandardCopyOption;
  19. import java.nio.file.attribute.PosixFilePermissions;
  20. import java.security.KeyPair;
  21. import java.security.KeyPairGenerator;
  22. import java.util.EnumSet;
  23. import java.util.List;
  24. import java.util.Set;
  25. import java.util.concurrent.TimeUnit;
  26. import java.util.stream.Collectors;
  27. import java.util.stream.Stream;
  28.  
  29. /**
  30.  * Sshd工具类
  31.  *
  32.  * https://github.com/apache/mina-sshd/blob/master/docs/sftp.md
  33.  */
  34. @Slf4j
  35. public class SshdTest {
  36. 	private SshClient client;
  37. 	private ClientSession session;
  38. 	public static void main(String[] args) throws Exception {
  39.  
  40. 		SshdTest sshUtil = new SshdTest("192.168.179.131",22,"weisx","123456");
  41. 		sshUtil.uploadFile(new File("F:\\24\\tmp.txt"),"/home/weisx/");
  42. 	}
  43.  
  44.     /**
  45.      * 连接
  46. 	 * @param host
  47.      * @param port
  48.      * @param username
  49. 	 */
  50. 	private void connect(String host, int port, String username)  {
  51. 		client = SshClient.setUpDefaultClient();
  52. 		client.start();
  53. 		try  {
  54. 			session = client.connect(username, host, port)
  55. 					.verify(5000)
  56. 					.getSession();
  57.  
  58. 		} catch (IOException e) {
  59. 			throw new RuntimeException(e);
  60. 		}
  61.  
  62. 	}
  63.  
  64. 	/**
  65. 	 * 密码登录
  66. 	 * @param host
  67. 	 * @param port
  68. 	 * @param username
  69. 	 * @param password
  70. 	 */
  71. 	public SshdTest(String host, int port, String username, String password) {
  72. 		connect(host,port,username);
  73. 		try  {
  74. 			session.addPasswordIdentity(password); // for password-based authentication
  75.  
  76. 			if (session.auth().verify(5000).isFailure()) {
  77. 				throw new RuntimeException("验证失败");
  78. 			}
  79. 		} catch (IOException e) {
  80. 			throw new RuntimeException(e);
  81. 		}
  82.  
  83. 	}
  84. 	/**
  85. 	 * 密钥登录
  86. 	 * @param host
  87. 	 * @param port
  88. 	 * @param username
  89. 	 */
  90. 	public SshdTest(String host, int port, String username) {
  91. 		connect(host,port,username);
  92. 		try  {
  93.  
  94. 			String privateKeyPath = System.getProperty("user.home") + "/.ssh/id_rsa";
  95. 			String privateKeyContent = new String(Files.readAllBytes(Paths.get(privateKeyPath)));
  96.  
  97. 			//获取密钥对
  98. 			KeyPairGenerator rsa = KeyPairGenerator.getInstance("RSA");
  99. 			KeyPair keyPair = rsa.generateKeyPair();
  100. 			ByteArrayOutputStream stream = new ByteArrayOutputStream();
  101. 			stream.write(privateKeyContent.getBytes());
  102. 			ObjectOutputStream o = new ObjectOutputStream(stream);
  103. 			o.writeObject(keyPair);
  104.  
  105. 			session.addPublicKeyIdentity(keyPair);
  106.  
  107.  
  108. 			if (session.auth().verify(5000).isFailure()) {
  109. 				throw new RuntimeException("验证失败");
  110. 			}
  111. 		} catch (Exception e) {
  112. 			throw new RuntimeException(e);
  113. 		}
  114.  
  115. 	}
  116.  
  117. 	/**
  118. 	 * 上传文件
  119. 	 * @param uploadFile 上传的文件
  120. 	 * @param remotePath 远程目录
  121. 	 */
  122. 	public  String uploadFile(File uploadFile,String remotePath) {
  123. 		try(SftpClient  sftpClient = SftpClientFactory.instance().createSftpClient(session);
  124. 			OutputStream outputStream = sftpClient.write(remotePath)
  125. 		) {
  126. 			Files.copy(uploadFile.toPath(), outputStream);
  127. 			return remotePath;
  128. 		} catch (IOException e) {
  129. 			throw new RuntimeException(e);
  130. 		}
  131. 	}
  132. 	/**
  133. 	 * 关闭连接
  134. 	 */
  135. 	public void close(){
  136. 		log.info("关闭 SSH");
  137. 		closeSession();
  138. 		if(client != null){
  139. 			try {
  140. 				client.close();
  141. 			} catch (IOException e) {
  142. 				log.error(e.getMessage());
  143. 			}
  144. 		}
  145.  
  146. 	}
  147. 	private void closeSession() {
  148. 		if(session != null){
  149. 			try {
  150. 				session.close();
  151. 			} catch (IOException e) {
  152. 				log.error(e.getMessage());
  153. 			}
  154. 		}
  155. 	}
  156. }

3.2 下载文件

  1. package com.yichenkeji.starter.ssh;
  2.  
  3. import lombok.extern.slf4j.Slf4j;
  4. import org.apache.sshd.client.SshClient;
  5. import org.apache.sshd.client.channel.ChannelExec;
  6. import org.apache.sshd.client.channel.ChannelShell;
  7. import org.apache.sshd.client.channel.ClientChannelEvent;
  8. import org.apache.sshd.client.session.ClientSession;
  9. import org.apache.sshd.sftp.client.SftpClient;
  10. import org.apache.sshd.sftp.client.SftpClientFactory;
  11. import org.apache.sshd.sftp.client.fs.SftpFileSystem;
  12. import org.apache.sshd.sftp.client.fs.SftpPath;
  13.  
  14. import java.io.*;
  15. import java.nio.file.Files;
  16. import java.nio.file.Path;
  17. import java.nio.file.Paths;
  18. import java.nio.file.StandardCopyOption;
  19. import java.nio.file.attribute.PosixFilePermissions;
  20. import java.security.KeyPair;
  21. import java.security.KeyPairGenerator;
  22. import java.util.EnumSet;
  23. import java.util.List;
  24. import java.util.Set;
  25. import java.util.concurrent.TimeUnit;
  26. import java.util.stream.Collectors;
  27. import java.util.stream.Stream;
  28.  
  29. /**
  30.  * Sshd工具类
  31.  *
  32.  * https://github.com/apache/mina-sshd/blob/master/docs/sftp.md
  33.  */
  34. @Slf4j
  35. public class SshdTest {
  36. 	private SshClient client;
  37. 	private ClientSession session;
  38. 	public static void main(String[] args) throws Exception {
  39.  
  40. 		SshdTest sshUtil = new SshdTest("192.168.179.131",22,"weisx","123456");
  41. 		sshUtil.downloadFile("/home/weisx/tmp.txt","F:\\24\\");
  42. 	}
  43.  
  44.     /**
  45.      * 连接
  46. 	 * @param host
  47.      * @param port
  48.      * @param username
  49. 	 */
  50. 	private void connect(String host, int port, String username)  {
  51. 		client = SshClient.setUpDefaultClient();
  52. 		client.start();
  53. 		try  {
  54. 			session = client.connect(username, host, port)
  55. 					.verify(5000)
  56. 					.getSession();
  57.  
  58. 		} catch (IOException e) {
  59. 			throw new RuntimeException(e);
  60. 		}
  61.  
  62. 	}
  63.  
  64. 	/**
  65. 	 * 密码登录
  66. 	 * @param host
  67. 	 * @param port
  68. 	 * @param username
  69. 	 * @param password
  70. 	 */
  71. 	public SshdTest(String host, int port, String username, String password) {
  72. 		connect(host,port,username);
  73. 		try  {
  74. 			session.addPasswordIdentity(password); // for password-based authentication
  75.  
  76. 			if (session.auth().verify(5000).isFailure()) {
  77. 				throw new RuntimeException("验证失败");
  78. 			}
  79. 		} catch (IOException e) {
  80. 			throw new RuntimeException(e);
  81. 		}
  82.  
  83. 	}
  84. 	/**
  85. 	 * 密钥登录
  86. 	 * @param host
  87. 	 * @param port
  88. 	 * @param username
  89. 	 */
  90. 	public SshdTest(String host, int port, String username) {
  91. 		connect(host,port,username);
  92. 		try  {
  93.  
  94. 			String privateKeyPath = System.getProperty("user.home") + "/.ssh/id_rsa";
  95. 			String privateKeyContent = new String(Files.readAllBytes(Paths.get(privateKeyPath)));
  96.  
  97. 			//获取密钥对
  98. 			KeyPairGenerator rsa = KeyPairGenerator.getInstance("RSA");
  99. 			KeyPair keyPair = rsa.generateKeyPair();
  100. 			ByteArrayOutputStream stream = new ByteArrayOutputStream();
  101. 			stream.write(privateKeyContent.getBytes());
  102. 			ObjectOutputStream o = new ObjectOutputStream(stream);
  103. 			o.writeObject(keyPair);
  104.  
  105. 			session.addPublicKeyIdentity(keyPair);
  106.  
  107.  
  108. 			if (session.auth().verify(5000).isFailure()) {
  109. 				throw new RuntimeException("验证失败");
  110. 			}
  111. 		} catch (Exception e) {
  112. 			throw new RuntimeException(e);
  113. 		}
  114.  
  115. 	}
  116.  
  117. 	/**
  118. 	 * 下载文件
  119. 	 * @param remoteFilePath 下载的文件
  120. 	 * @param savePath 保存的目录
  121. 	 */
  122. 	public  void downloadFile(String remoteFilePath,String savePath) {
  123. 		try(SftpClient  sftpClient = SftpClientFactory.instance().createSftpClient(session);
  124. 			InputStream inputStream = sftpClient.read(remoteFilePath)
  125. 		) {
  126. 			File destFile = new File(savePath);
  127. 			Files.copy(inputStream , destFile.toPath() , StandardCopyOption.REPLACE_EXISTING);
  128. 		} catch (IOException e) {
  129. 			throw new RuntimeException(e);
  130. 		}
  131. 	}
  132. 	/**
  133. 	 * 关闭连接
  134. 	 */
  135. 	public void close(){
  136. 		log.info("关闭 SSH");
  137. 		closeSession();
  138. 		if(client != null){
  139. 			try {
  140. 				client.close();
  141. 			} catch (IOException e) {
  142. 				log.error(e.getMessage());
  143. 			}
  144. 		}
  145.  
  146. 	}
  147. 	private void closeSession() {
  148. 		if(session != null){
  149. 			try {
  150. 				session.close();
  151. 			} catch (IOException e) {
  152. 				log.error(e.getMessage());
  153. 			}
  154. 		}
  155. 	}
  156. }

3.3 SftpFileSystem

        SftpFileSystem是Apache Mina SSHD中实现的基于VFS框架的SFTP文件系统，它供了一套统一的API和一些额外的高级功能，方便用户访问和操作远程文件系统，适用于复杂的文件系统操作和要求更高级功能的场景。

  1. package com.yichenkeji.starter.ssh;
  2.  
  3. import lombok.extern.slf4j.Slf4j;
  4. import org.apache.sshd.client.SshClient;
  5. import org.apache.sshd.client.channel.ChannelExec;
  6. import org.apache.sshd.client.channel.ChannelShell;
  7. import org.apache.sshd.client.channel.ClientChannelEvent;
  8. import org.apache.sshd.client.session.ClientSession;
  9. import org.apache.sshd.sftp.client.SftpClient;
  10. import org.apache.sshd.sftp.client.SftpClientFactory;
  11. import org.apache.sshd.sftp.client.fs.SftpFileSystem;
  12. import org.apache.sshd.sftp.client.fs.SftpPath;
  13.  
  14. import java.io.*;
  15. import java.nio.file.Files;
  16. import java.nio.file.Path;
  17. import java.nio.file.Paths;
  18. import java.nio.file.StandardCopyOption;
  19. import java.nio.file.attribute.PosixFilePermissions;
  20. import java.security.KeyPair;
  21. import java.security.KeyPairGenerator;
  22. import java.util.EnumSet;
  23. import java.util.List;
  24. import java.util.Set;
  25. import java.util.concurrent.TimeUnit;
  26. import java.util.stream.Collectors;
  27. import java.util.stream.Stream;
  28.  
  29. /**
  30.  * Sshd工具类
  31.  *
  32.  * https://github.com/apache/mina-sshd/blob/master/docs/sftp.md
  33.  */
  34. @Slf4j
  35. public class SshdTest {
  36. 	private SshClient client;
  37. 	private ClientSession session;
  38. 	public static void main(String[] args) throws Exception {
  39.  
  40. 		SshdTest sshUtil = new SshdTest("192.168.179.131",22,"weisx","123456");
  41. 		sshUtil.fileMeta("/home/weisx/tmp.txt");
  42. 	}
  43.  
  44.     /**
  45.      * 连接
  46. 	 * @param host
  47.      * @param port
  48.      * @param username
  49. 	 */
  50. 	private void connect(String host, int port, String username)  {
  51. 		client = SshClient.setUpDefaultClient();
  52. 		client.start();
  53. 		try  {
  54. 			session = client.connect(username, host, port)
  55. 					.verify(5000)
  56. 					.getSession();
  57.  
  58. 		} catch (IOException e) {
  59. 			throw new RuntimeException(e);
  60. 		}
  61.  
  62. 	}
  63.  
  64. 	/**
  65. 	 * 密码登录
  66. 	 * @param host
  67. 	 * @param port
  68. 	 * @param username
  69. 	 * @param password
  70. 	 */
  71. 	public SshdTest(String host, int port, String username, String password) {
  72. 		connect(host,port,username);
  73. 		try  {
  74. 			session.addPasswordIdentity(password); // for password-based authentication
  75.  
  76. 			if (session.auth().verify(5000).isFailure()) {
  77. 				throw new RuntimeException("验证失败");
  78. 			}
  79. 		} catch (IOException e) {
  80. 			throw new RuntimeException(e);
  81. 		}
  82.  
  83. 	}
  84. 	/**
  85. 	 * 密钥登录
  86. 	 * @param host
  87. 	 * @param port
  88. 	 * @param username
  89. 	 */
  90. 	public SshdTest(String host, int port, String username) {
  91. 		connect(host,port,username);
  92. 		try  {
  93.  
  94. 			String privateKeyPath = System.getProperty("user.home") + "/.ssh/id_rsa";
  95. 			String privateKeyContent = new String(Files.readAllBytes(Paths.get(privateKeyPath)));
  96.  
  97. 			//获取密钥对
  98. 			KeyPairGenerator rsa = KeyPairGenerator.getInstance("RSA");
  99. 			KeyPair keyPair = rsa.generateKeyPair();
  100. 			ByteArrayOutputStream stream = new ByteArrayOutputStream();
  101. 			stream.write(privateKeyContent.getBytes());
  102. 			ObjectOutputStream o = new ObjectOutputStream(stream);
  103. 			o.writeObject(keyPair);
  104.  
  105. 			session.addPublicKeyIdentity(keyPair);
  106.  
  107.  
  108. 			if (session.auth().verify(5000).isFailure()) {
  109. 				throw new RuntimeException("验证失败");
  110. 			}
  111. 		} catch (Exception e) {
  112. 			throw new RuntimeException(e);
  113. 		}
  114.  
  115. 	}
  116.  
  117. 	/**
  118. 	 * 获取文件信息
  119. 	 * @param filePath 文件路径
  120. 	 */
  121. 	public  void fileMeta(String filePath) {
  122. 		try(SftpFileSystem sftp = SftpClientFactory.instance().createSftpFileSystem(session)) {
  123.  
  124. 			SftpPath path = sftp.getPath(filePath);
  125. 			System.out.println("文件名：" + path.getFileName());
  126. 			System.out.println("文件大小：" + Files.size(path));
  127. 			System.out.println("最后更新时间：" + Files.getLastModifiedTime(path));
  128. 			//如果是目录，则可以列出目录下全部文件
  129. 			try (Stream<Path> stream = Files.list(path)) {
  130. 				List<Path> paths = stream.collect(Collectors.toList());
  131. 				for (Path p : paths) {
  132. 					if (Files.isDirectory(p)) {
  133. 						System.out.println("文件夹：" + p.getFileName());
  134. 					} else {
  135. 						System.out.println("文件：  " + p.getFileName());
  136. 					}
  137. 				}
  138. 			}
  139.  
  140. 		} catch (IOException e) {
  141. 			throw new RuntimeException(e);
  142. 		}
  143. 	}
  144. 	/**
  145. 	 * 关闭连接
  146. 	 */
  147. 	public void close(){
  148. 		log.info("关闭 SSH");
  149. 		closeSession();
  150. 		if(client != null){
  151. 			try {
  152. 				client.close();
  153. 			} catch (IOException e) {
  154. 				log.error(e.getMessage());
  155. 			}
  156. 		}
  157.  
  158. 	}
  159. 	private void closeSession() {
  160. 		if(session != null){
  161. 			try {
  162. 				session.close();
  163. 			} catch (IOException e) {
  164. 				log.error(e.getMessage());
  165. 			}
  166. 		}
  167. 	}
  168. }

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小丑西瓜9/article/detail/416622
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号