来源:https://blog.csdn.net/NoCortY/article/details/104772431

前言

最近由于项目需求,项目中需要实现一个WebSSH连接终端的功能,由于自己第一次做这类型功能,所以首先上了GitHub找了找有没有现成的轮子可以拿来直接用,当时看到了很多这方面的项目,例如:GateOne、webssh、shellinabox等。
这些项目都可以很好地实现webssh的功能,但是最终并没有采用,原因是在于这些底层大都是python写的,需要依赖很多文件,自己用的时候可以使用这种方案,快捷省事,但是做到项目中供用户使用时,总不能要求用户做到服务器中必须包含这些底层依赖,这显然不太合理,所以我决定自己动手写一个WebSSH的功能,并且作为一个独立的项目开源出来。(文末附项目开源地址)

技术选型

由于webssh需要实时数据交互,所以会选用长连接的WebSocket,为了开发的方便,框架选用SpringBoot,另外还自己了解了Java用户连接ssh的jsch和实现前端shell页面的xterm.js.所以,最终的技术选型就是 SpringBoot+Websocket+jsch+xterm.js。

导入依赖

  1. <parent>
  2. <groupId>org.springframework.boot</groupId>
  3. <artifactId>spring-boot-starter-parent</artifactId>
  4. <version>2.1.7.RELEASE</version>
  5. <relativePath /> <!-- lookup parent from repository -->
  6. </parent>
  7. <dependencies>
  8. <!-- Web相关 -->
  9. <dependency>
  10. <groupId>org.springframework.boot</groupId>
  11. <artifactId>spring-boot-starter-web</artifactId>
  12. </dependency>
  13. <!-- jsch支持 -->
  14. <dependency>
  15. <groupId>com.jcraft</groupId>
  16. <artifactId>jsch</artifactId>
  17. <version>0.1.54</version>
  18. </dependency>
  19. <!-- WebSocket 支持 -->
  20. <dependency>
  21. <groupId>org.springframework.boot</groupId>
  22. <artifactId>spring-boot-starter-websocket</artifactId>
  23. </dependency>
  24. <!-- 文件上传解析器 -->
  25. <dependency>
  26. <groupId>commons-io</groupId>
  27. <artifactId>commons-io</artifactId>
  28. <version>1.4</version>
  29. </dependency>
  30. <dependency>
  31. <groupId>commons-fileupload</groupId>
  32. <artifactId>commons-fileupload</artifactId>
  33. <version>1.3.1</version>
  34. </dependency>
  35. </dependencies>

一个简单的xterm案例

由于xterm是一个冷门技术,所以很多同学并没有这方面的知识支撑,我也是为了实现这个功能所以临时学的,所以在这给大家介绍一下。
xterm.js是一个基于WebSocket的容器,它可以帮助我们在前端实现命令行的样式。就像是我们平常再用SecureCRT或者XShell连接服务器时一样。下面是官网上的入门案例:

  1. <!doctype html>
  2. <html>
  3. <head>
  4. <link rel="stylesheet" href="node_modules/xterm/css/xterm.css" />
  5. <script src="node_modules/xterm/lib/xterm.js"></script>
  6. </head>
  7. <body>
  8. <div id="terminal"></div>
  9. <script>
  10. var term = new Terminal();
  11. term.open(document.getElementById('terminal'));
  12. term.write('Hello from \x1B[1;3;31mxterm.js\x1B[0m $ ')
  13. </script>
  14. </body>
  15. </html>

最终测试,页面就是下面这个样子:
WebSSH网页终端 - 图1可以看到页面已经出现了类似与shell的样式,那就根据这个继续深入,实现一个webssh。

后端实现

由于xterm只要只是实现了前端的样式,并不能真正地实现与服务器交互,与服务器交互主要还是靠我们Java后端来进行控制的,所以我们从后端开始,使用jsch+websocket实现这部分内容。微信搜索公众号:前端技术编程,回复:前端 领取资料 。

WebSocket配置

由于消息实时推送到前端需要用到WebSocket,不了解WebSocket的同学可以先去自行了解一下,这里就不过多介绍了,我们直接开始进行WebSocket的配置。

  1. /**
  2. * @Description: websocket配置
  3. * @Author: NoCortY
  4. * @Date: 2020/3/8
  5. */
  6. @Configuration
  7. @EnableWebSocket
  8. public class WebSSHWebSocketConfig implements WebSocketConfigurer{
  9. @Autowired
  10. WebSSHWebSocketHandler webSSHWebSocketHandler;
  11. @Override
  12. public void registerWebSocketHandlers(WebSocketHandlerRegistry webSocketHandlerRegistry) {
  13. //socket通道
  14. //指定处理器和路径,并设置跨域
  15. webSocketHandlerRegistry.addHandler(webSSHWebSocketHandler, "/webssh")
  16. .addInterceptors(new WebSocketInterceptor())
  17. .setAllowedOrigins("*");
  18. }
  19. }

处理器(Handler)和拦截器(Interceptor)的实现

刚才我们完成了WebSocket的配置,并指定了一个处理器和拦截器。所以接下来就是处理器和拦截器的实现。拦截器:

  1. public class WebSocketInterceptor implements HandshakeInterceptor{
  2. /**
  3. * @Description: Handler处理前调用
  4. * @Param: [serverHttpRequest, serverHttpResponse, webSocketHandler, map]
  5. * @return: boolean
  6. * @Author: NoCortY
  7. * @Date: 2020/3/1
  8. */
  9. @Override
  10. public boolean beforeHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map<String, Object> map) throws Exception{
  11. if (serverHttpRequest instanceof ServletServerHttpRequest) {
  12. ServletServerHttpRequest request = (ServletServerHttpRequest) serverHttpRequest;
  13. //生成一个UUID,这里由于是独立的项目,没有用户模块,所以可以用随机的UUID
  14. //但是如果要集成到自己的项目中,需要将其改为自己识别用户的标识
  15. String uuid = UUID.randomUUID().toString().replace("-","");
  16. //将uuid放到websocketsession中
  17. map.put(ConstantPool.USER_UUID_KEY, uuid);
  18. return true;
  19. } else {
  20. return false;
  21. }
  22. }
  23. @Override
  24. public void afterHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Exception e){
  25. }
  26. }

处理器:

  1. /**
  2. * @Description: WebSSH的WebSocket处理器
  3. * @Author: NoCortY
  4. * @Date: 2020/3/8
  5. */
  6. @Component
  7. public class WebSSHWebSocketHandler implements WebSocketHandler{
  8. @Autowired
  9. private WebSSHService webSSHService;
  10. private Logger logger = LoggerFactory.getLogger(WebSSHWebSocketHandler.class);
  11. /**
  12. * @Description: 用户连接上WebSocket的回调
  13. * @Param: [webSocketSession]
  14. * @return: void
  15. * @Author: Object
  16. * @Date: 2020/3/8
  17. */
  18. @Override
  19. public void afterConnectionEstablished(WebSocketSession webSocketSession) throws Exception{
  20. logger.info("用户:{},连接WebSSH", webSocketSession.getAttributes().get(ConstantPool.USER_UUID_KEY));
  21. //调用初始化连接
  22. webSSHService.initConnection(webSocketSession);
  23. }
  24. /**
  25. * @Description: 收到消息的回调
  26. * @Param: [webSocketSession, webSocketMessage]
  27. * @return: void
  28. * @Author: NoCortY
  29. * @Date: 2020/3/8
  30. */
  31. @Override
  32. public void handleMessage(WebSocketSession webSocketSession, WebSocketMessage<?> webSocketMessage) throws Exception{
  33. if (webSocketMessage instanceof TextMessage) {
  34. logger.info("用户:{},发送命令:{}", webSocketSession.getAttributes().get(ConstantPool.USER_UUID_KEY), webSocketMessage.toString());
  35. //调用service接收消息
  36. webSSHService.recvHandle(((TextMessage) webSocketMessage).getPayload(), webSocketSession);
  37. } else if (webSocketMessage instanceof BinaryMessage) {
  38. } else if (webSocketMessage instanceof PongMessage) {
  39. } else {
  40. System.out.println("Unexpected WebSocket message type: " + webSocketMessage);
  41. }
  42. }
  43. /**
  44. * @Description: 出现错误的回调
  45. * @Param: [webSocketSession, throwable]
  46. * @return: void
  47. * @Author: Object
  48. * @Date: 2020/3/8
  49. */
  50. @Override
  51. public void handleTransportError(WebSocketSession webSocketSession, Throwable throwable) throws Exception{
  52. logger.error("数据传输错误");
  53. }
  54. /**
  55. * @Description: 连接关闭的回调
  56. * @Param: [webSocketSession, closeStatus]
  57. * @return: void
  58. * @Author: NoCortY
  59. * @Date: 2020/3/8
  60. */
  61. @Override
  62. public void afterConnectionClosed(WebSocketSession webSocketSession, CloseStatus closeStatus) throws Exception{
  63. logger.info("用户:{}断开webssh连接", String.valueOf(webSocketSession.getAttributes().get(ConstantPool.USER_UUID_KEY)));
  64. //调用service关闭连接
  65. webSSHService.close(webSocketSession);
  66. }
  67. @Override
  68. public boolean supportsPartialMessages(){
  69. return false;
  70. }
  71. }

需要注意的是,我在拦截器中加入的用户标识是使用了随机的UUID,这是因为作为一个独立的websocket项目,没有用户模块,如果需要将这个项目集成到自己的项目中,需要修改这部分代码,将其改为自己项目中识别一个用户所用的用户标识。

WebSSH的业务逻辑实现(核心)

刚才我们实现了websocket的配置,都是一些死代码,实现了接口再根据自身需求即可实现,现在我们将进行后端主要业务逻辑的实现,在实现这个逻辑之前,我们先来想想,WebSSH,我们主要想要呈现一个什么效果。我这里做了一个总结:
1.首先我们得先连接上终端(初始化连接)
2.其次我们的服务端需要处理来自前端的消息(接收并处理前端消息)
3.我们需要将终端返回的消息回写到前端(数据回写前端)
4.关闭连接
根据这四个需求,我们先定义一个接口,这样可以让需求明了起来。

  1. /**
  2. * @Description: WebSSH的业务逻辑
  3. * @Author: NoCortY
  4. * @Date: 2020/3/7
  5. */
  6. public interface WebSSHService{
  7. /**
  8. * @Description: 初始化ssh连接
  9. * @Param:
  10. * @return:
  11. * @Author: NoCortY
  12. * @Date: 2020/3/7
  13. */
  14. public void initConnection(WebSocketSession session);
  15. /**
  16. * @Description: 处理客户段发的数据
  17. * @Param:
  18. * @return:
  19. * @Author: NoCortY
  20. * @Date: 2020/3/7
  21. */
  22. public void recvHandle(String buffer, WebSocketSession session);
  23. /**
  24. * @Description: 数据写回前端 for websocket
  25. * @Param:
  26. * @return:
  27. * @Author: NoCortY
  28. * @Date: 2020/3/7
  29. */
  30. public void sendMessage(WebSocketSession session, byte[] buffer) throws IOException;
  31. /**
  32. * @Description: 关闭连接
  33. * @Param:
  34. * @return:
  35. * @Author: NoCortY
  36. * @Date: 2020/3/7
  37. */
  38. public void close(WebSocketSession session);
  39. }

现在我们可以根据这个接口去实现我们定义的功能了。**

1. 初始化连接
由于我们的底层是依赖jsch实现的,所以这里是需要使用jsch去建立连接的。而所谓初始化连接,实际上就是将我们所需要的连接信息,保存在一个Map中,这里并不进行任何的真实连接操作。
为什么这里不直接进行连接?因为这里前端只是连接上了WebSocket,但是我们还需要前端给我们发来linux终端的用户名和密码,没有这些信息,我们是无法进行连接的。

  1. public void initConnection(WebSocketSession session) {
  2. JSch jSch = new JSch();
  3. SSHConnectInfo sshConnectInfo = new SSHConnectInfo();
  4. sshConnectInfo.setjSch(jSch);
  5. sshConnectInfo.setWebSocketSession(session);
  6. String uuid = String.valueOf(session.getAttributes().get(ConstantPool.USER_UUID_KEY));
  7. //将这个ssh连接信息放入map中
  8. sshMap.put(uuid, sshConnectInfo);
  9. }

2. 处理客户端发送的数据
在这一步骤中,我们会分为两个分支。第一个分支:如果客户端发来的是终端的用户名和密码等信息,那么我们进行终端的连接。第二个分支:如果客户端发来的是操作终端的命令,那么我们就直接转发到终端并且获取终端的执行结果。具体代码实现:

  1. public void recvHandle(String buffer, WebSocketSession session) {
  2. ObjectMapper objectMapper = new ObjectMapper();
  3. WebSSHData webSSHData = null;
  4. try {
  5. //转换前端发送的JSON
  6. webSSHData = objectMapper.readValue(buffer, WebSSHData.class);
  7. } catch (IOException e) {
  8. logger.error("Json转换异常");
  9. logger.error("异常信息:{}", e.getMessage());
  10. return;
  11. }
  12. //获取刚才设置的随机的uuid
  13. String userId = String.valueOf(session.getAttributes().get(ConstantPool.USER_UUID_KEY));
  14. if (ConstantPool.WEBSSH_OPERATE_CONNECT.equals(webSSHData.getOperate())) {
  15. //如果是连接请求
  16. //找到刚才存储的ssh连接对象
  17. SSHConnectInfo sshConnectInfo = (SSHConnectInfo) sshMap.get(userId);
  18. //启动线程异步处理
  19. WebSSHData finalWebSSHData = webSSHData;
  20. executorService.execute(new Runnable() {
  21. @Override
  22. public void run() {
  23. try {
  24. //连接到终端
  25. connectToSSH(sshConnectInfo, finalWebSSHData, session);
  26. } catch (JSchException | IOException e) {
  27. logger.error("webssh连接异常");
  28. logger.error("异常信息:{}", e.getMessage());
  29. close(session);
  30. }
  31. }
  32. });
  33. } else if (ConstantPool.WEBSSH_OPERATE_COMMAND.equals(webSSHData.getOperate())) {
  34. //如果是发送命令的请求
  35. String command = webSSHData.getCommand();
  36. SSHConnectInfo sshConnectInfo = (SSHConnectInfo) sshMap.get(userId);
  37. if (sshConnectInfo != null) {
  38. try {
  39. //发送命令到终端
  40. transToSSH(sshConnectInfo.getChannel(), command);
  41. } catch (IOException e) {
  42. logger.error("webssh连接异常");
  43. logger.error("异常信息:{}", e.getMessage());
  44. close(session);
  45. }
  46. }
  47. } else {
  48. logger.error("不支持的操作");
  49. close(session);
  50. }
  51. }

3. 数据通过websocket发送到前端

  1. public void sendMessage(WebSocketSession session, byte[] buffer) throws IOException {
  2. session.sendMessage(new TextMessage(buffer));
  3. }

4. 关闭连接

  1. public void close(WebSocketSession session) {
  2. //获取随机生成的uuid
  3. String userId = String.valueOf(session.getAttributes().get(ConstantPool.USER_UUID_KEY));
  4. SSHConnectInfo sshConnectInfo = (SSHConnectInfo) sshMap.get(userId);
  5. if (sshConnectInfo != null) {
  6. //断开连接
  7. if (sshConnectInfo.getChannel() != null) sshConnectInfo.getChannel().disconnect();
  8. //map中移除该ssh连接信息
  9. sshMap.remove(userId);
  10. }
  11. }

至此,我们的整个后端实现就结束了,由于篇幅有限,这里将一些操作封装成了方法,就不做过多展示了,重点讲逻辑实现的思路吧。接下来我们将进行前端的实现。

前端实现

前端工作主要分为这么几个步骤:

  1. 页面的实现
  2. 连接WebSocket并完成数据的接收并回写
  3. 数据的发送

所以我们一步一步来实现它。

页面实现

页面的实现很简单,我们只不过需要在一整个屏幕上都显示终端那种大黑屏幕,所以我们并不用写什么样式,只需要创建一个div,之后将terminal实例通过xterm放到这个div中,就可以实现了。

  1. <!doctype html>
  2. <html>
  3. <head>
  4. <title>WebSSH</title>
  5. <link rel="stylesheet" href="../css/xterm.css" />
  6. </head>
  7. <body>
  8. <div id="terminal" style="width: 100%;height: 100%"></div>
  9. <script src="../lib/jquery-3.4.1/jquery-3.4.1.min.js"></script>
  10. <script src="../js/xterm.js" charset="utf-8"></script>
  11. <script src="../js/webssh.js" charset="utf-8"></script>
  12. <script src="../js/base64.js" charset="utf-8"></script>
  13. </body>
  14. </html>

连接WebSocket并完成数据的发送、接收、回写

  1. openTerminal( {
  2. //这里的内容可以写死,但是要整合到项目中时,需要通过参数的方式传入,可以动态连接某个终端。
  3. operate:'connect',
  4. host: 'ip地址',
  5. port: '端口号',
  6. username: '用户名',
  7. password: '密码'
  8. });
  9. function openTerminal(options){
  10. var client = new WSSHClient();
  11. var term = new Terminal({
  12. cols: 97,
  13. rows: 37,
  14. cursorBlink: true, // 光标闪烁
  15. cursorStyle: "block", // 光标样式 null | 'block' | 'underline' | 'bar'
  16. scrollback: 800, //回滚
  17. tabStopWidth: 8, //制表宽度
  18. screenKeys: true
  19. });
  20. term.on('data', function (data) {
  21. //键盘输入时的回调函数
  22. client.sendClientData(data);
  23. });
  24. term.open(document.getElementById('terminal'));
  25. //在页面上显示连接中...
  26. term.write('Connecting...');
  27. //执行连接操作
  28. client.connect({
  29. onError: function (error) {
  30. //连接失败回调
  31. term.write('Error: ' + error + '\r\n');
  32. },
  33. onConnect: function () {
  34. //连接成功回调
  35. client.sendInitData(options);
  36. },
  37. onClose: function () {
  38. //连接关闭回调
  39. term.write("\rconnection closed");
  40. },
  41. onData: function (data) {
  42. //收到数据时回调
  43. term.write(data);
  44. }
  45. });
  46. }

效果展示

连接

WebSSH网页终端 - 图2

连接成功

WebSSH网页终端 - 图3

命令操作

ls命令:
WebSSH网页终端 - 图4
vim编辑器:
WebSSH网页终端 - 图5
top命令:
WebSSH网页终端 - 图6

结语

这样我们就完成了一个webssh项目的实现,没有依赖其它任何的组件,后端完全使用Java实现,由于用了SpringBoot,非常容易部署。
但是,我们还可以对这个项目进行扩展,比如新增上传或下载文件,就像Xftp一样,可以很方便地拖拽式上传下载文件。
Github项目开源地址:https://github.com/NoCortY/WebSSH