119 lines
3.2 KiB
Go
119 lines
3.2 KiB
Go
package controller
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
"be.ems/src/framework/cmd"
|
|
"be.ems/src/framework/i18n"
|
|
"be.ems/src/framework/logger"
|
|
"be.ems/src/framework/reqctx"
|
|
"be.ems/src/framework/resp"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// SSH 终端
|
|
//
|
|
// GET /ssh?hostId=1&cols=80&rows=40
|
|
func (s *WSController) SSH(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
var query struct {
|
|
HostId int64 `form:"hostId" binding:"required"` // 连接主机ID
|
|
Cols int `form:"cols"` // 终端单行字符数
|
|
Rows int `form:"rows"` // 终端显示行数
|
|
}
|
|
if err := c.ShouldBindQuery(&query); err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(422001, errMsgs))
|
|
return
|
|
}
|
|
if query.Cols < 80 || query.Cols > 400 {
|
|
query.Cols = 80
|
|
}
|
|
if query.Rows < 40 || query.Rows > 1200 {
|
|
query.Rows = 40
|
|
}
|
|
|
|
// 登录用户信息
|
|
loginUser, err := reqctx.LoginUser(c)
|
|
if err != nil {
|
|
c.JSON(401, resp.CodeMsg(401002, i18n.TKey(language, err.Error())))
|
|
return
|
|
}
|
|
|
|
// neHost := neService.NewNeHost.FindById(query.HostId)
|
|
// if neHost.ID != query.HostId || neHost.HostType != "ssh" {
|
|
// // 没有可访问主机信息数据！
|
|
// c.JSON(200, resp.ErrMsg(i18n.TKey(language, "neHost.noData")))
|
|
// return
|
|
// }
|
|
|
|
// // 创建链接SSH客户端
|
|
// var connSSH ssh.ConnSSH
|
|
// neHost.CopyTo(&connSSH)
|
|
// var client *ssh.ConnSSH
|
|
// var clientErr error
|
|
// if neHost.AuthMode == "2" {
|
|
// client, clientErr = connSSH.NewClientByLocalPrivate()
|
|
// } else {
|
|
// client, clientErr = connSSH.NewClient()
|
|
// }
|
|
// if clientErr != nil {
|
|
// // 连接主机失败，请检查连接参数后重试
|
|
// c.JSON(200, resp.ErrMsg(i18n.TKey(language, "neHost.errByHostInfo")))
|
|
// return
|
|
// }
|
|
// defer client.Close()
|
|
|
|
// 创建SSH客户端会话
|
|
clientSession, err := cmd.NewClientSession(query.Cols, query.Rows)
|
|
if err != nil {
|
|
// 连接主机失败，请检查连接参数后重试
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "neHost.errByHostInfo")))
|
|
return
|
|
}
|
|
defer clientSession.Close()
|
|
|
|
// 将 HTTP 连接升级为 WebSocket 连接
|
|
wsConn := s.wsService.UpgraderWs(c.Writer, c.Request)
|
|
if wsConn == nil {
|
|
return
|
|
}
|
|
defer wsConn.Close()
|
|
|
|
wsClient := s.wsService.ClientCreate(loginUser.UserId, nil, wsConn, clientSession)
|
|
go s.wsService.ClientWriteListen(wsClient)
|
|
go s.wsService.ClientReadListen(wsClient, s.wsReceiveService.Shell)
|
|
|
|
// 实时读取SSH消息直接输出
|
|
msTicker := time.NewTicker(100 * time.Millisecond)
|
|
defer msTicker.Stop()
|
|
for {
|
|
select {
|
|
case ms := <-msTicker.C:
|
|
outputByte := clientSession.Read()
|
|
if len(outputByte) > 0 {
|
|
outputStr := string(outputByte)
|
|
msgByte, _ := json.Marshal(resp.Ok(map[string]any{
|
|
"requestId": fmt.Sprintf("ssh_%d_%d", query.HostId, ms.UnixMilli()),
|
|
"data": outputStr,
|
|
}))
|
|
wsClient.MsgChan <- msgByte
|
|
|
|
// 退出ssh登录
|
|
// if strings.LastIndex(outputStr, "logout\r\n") != -1 {
|
|
// time.Sleep(1 * time.Second)
|
|
// s.wsService.CloseClient(wsClient.ID)
|
|
// return
|
|
// }
|
|
}
|
|
case <-wsClient.StopChan: // 等待停止信号
|
|
s.wsService.ClientClose(wsClient.ID)
|
|
logger.Infof("ws Stop Client UID %d", wsClient.BindUid)
|
|
return
|
|
}
|
|
}
|
|
}
|