mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-12-11 23:19:09 +08:00
100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
|
package terminal
|
||
|
|
||
|
import (
|
||
|
"encoding/base64"
|
||
|
"encoding/json"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/1Panel-dev/1Panel/backend/global"
|
||
|
"github.com/gorilla/websocket"
|
||
|
"github.com/pkg/errors"
|
||
|
)
|
||
|
|
||
|
type LocalWsSession struct {
|
||
|
slave *LocalCommand
|
||
|
wsConn *websocket.Conn
|
||
|
|
||
|
writeMutex sync.Mutex
|
||
|
}
|
||
|
|
||
|
func NewLocalWsSession(cols, rows int, wsConn *websocket.Conn, slave *LocalCommand) (*LocalWsSession, error) {
|
||
|
if err := slave.ResizeTerminal(cols, rows); err != nil {
|
||
|
global.LOG.Errorf("ssh pty change windows size failed, err: %v", err)
|
||
|
}
|
||
|
|
||
|
return &LocalWsSession{
|
||
|
slave: slave,
|
||
|
wsConn: wsConn,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func (sws *LocalWsSession) Start(quitChan chan bool) {
|
||
|
go sws.handleSlaveEvent(quitChan)
|
||
|
go sws.receiveWsMsg(quitChan)
|
||
|
}
|
||
|
|
||
|
func (sws *LocalWsSession) handleSlaveEvent(exitCh chan bool) {
|
||
|
defer setQuit(exitCh)
|
||
|
|
||
|
buffer := make([]byte, 1024)
|
||
|
for {
|
||
|
select {
|
||
|
case <-exitCh:
|
||
|
return
|
||
|
default:
|
||
|
n, _ := sws.slave.Read(buffer)
|
||
|
_ = sws.masterWrite(buffer[:n])
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (sws *LocalWsSession) masterWrite(data []byte) error {
|
||
|
sws.writeMutex.Lock()
|
||
|
defer sws.writeMutex.Unlock()
|
||
|
err := sws.wsConn.WriteMessage(websocket.TextMessage, data)
|
||
|
if err != nil {
|
||
|
return errors.Wrapf(err, "failed to write to master")
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (sws *LocalWsSession) receiveWsMsg(exitCh chan bool) {
|
||
|
wsConn := sws.wsConn
|
||
|
defer setQuit(exitCh)
|
||
|
for {
|
||
|
select {
|
||
|
case <-exitCh:
|
||
|
return
|
||
|
default:
|
||
|
_, wsData, err := wsConn.ReadMessage()
|
||
|
if err != nil {
|
||
|
global.LOG.Errorf("reading webSocket message failed, err: %v", err)
|
||
|
return
|
||
|
}
|
||
|
msgObj := wsMsg{}
|
||
|
_ = json.Unmarshal(wsData, &msgObj)
|
||
|
switch msgObj.Type {
|
||
|
case wsMsgResize:
|
||
|
if msgObj.Cols > 0 && msgObj.Rows > 0 {
|
||
|
if err := sws.slave.ResizeTerminal(msgObj.Cols, msgObj.Rows); err != nil {
|
||
|
global.LOG.Errorf("ssh pty change windows size failed, err: %v", err)
|
||
|
}
|
||
|
}
|
||
|
case wsMsgCmd:
|
||
|
decodeBytes, err := base64.StdEncoding.DecodeString(msgObj.Cmd)
|
||
|
if err != nil {
|
||
|
global.LOG.Errorf("websock cmd string base64 decoding failed, err: %v", err)
|
||
|
}
|
||
|
sws.sendWebsocketInputCommandToSshSessionStdinPipe(decodeBytes)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (sws *LocalWsSession) sendWebsocketInputCommandToSshSessionStdinPipe(cmdBytes []byte) {
|
||
|
if _, err := sws.slave.Write(cmdBytes); err != nil {
|
||
|
global.LOG.Errorf("ws cmd bytes write to ssh.stdin pipe failed, err: %v", err)
|
||
|
}
|
||
|
}
|