mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 11:27:17 +08:00
ddfcd3e45c
* Initial plan * fix: 修复 PTY Websocket 客户端断开连接后命令仍在后台运行的问题 Co-authored-by: devhaozi <115467771+devhaozi@users.noreply.github.com> * fix: 解决连接中断后命令不被杀死的问题 --------- Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: devhaozi <115467771+devhaozi@users.noreply.github.com> Co-authored-by: 耗子 <haozi@loli.email>
120 lines
2.3 KiB
Go
120 lines
2.3 KiB
Go
package ssh
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"time"
|
||
|
||
"github.com/coder/websocket"
|
||
"golang.org/x/crypto/ssh"
|
||
)
|
||
|
||
type MessageResize struct {
|
||
Resize bool `json:"resize"`
|
||
Columns int `json:"columns"`
|
||
Rows int `json:"rows"`
|
||
}
|
||
|
||
type Turn struct {
|
||
ctx context.Context
|
||
stdin io.WriteCloser
|
||
session *ssh.Session
|
||
ws *websocket.Conn
|
||
}
|
||
|
||
func NewTurn(ctx context.Context, ws *websocket.Conn, client *ssh.Client) (*Turn, error) {
|
||
sess, err := client.NewSession()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
stdin, err := sess.StdinPipe()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
turn := &Turn{ctx: ctx, stdin: stdin, session: sess, ws: ws}
|
||
sess.Stdout = turn
|
||
sess.Stderr = turn
|
||
|
||
modes := ssh.TerminalModes{
|
||
ssh.ECHO: 1,
|
||
ssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaud
|
||
ssh.TTY_OP_OSPEED: 14400, // output speed = 14.4kbaud
|
||
}
|
||
if err = sess.RequestPty("xterm", 150, 80, modes); err != nil {
|
||
return nil, err
|
||
}
|
||
if err = sess.Shell(); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return turn, nil
|
||
}
|
||
|
||
func (t *Turn) Write(p []byte) (n int, err error) {
|
||
if err = t.ws.Write(t.ctx, websocket.MessageText, p); err != nil {
|
||
return 0, err
|
||
}
|
||
return len(p), nil
|
||
}
|
||
|
||
func (t *Turn) Close() {
|
||
_ = t.stdin.Close()
|
||
_ = t.session.Signal(ssh.SIGTERM)
|
||
|
||
// 等待最多 10 秒
|
||
done := make(chan struct{})
|
||
go func() {
|
||
_ = t.session.Wait()
|
||
close(done)
|
||
}()
|
||
|
||
select {
|
||
case <-done:
|
||
// 会话已退出
|
||
case <-time.After(10 * time.Second):
|
||
// 超时，KILL
|
||
_ = t.session.Signal(ssh.SIGKILL)
|
||
}
|
||
|
||
_ = t.session.Close()
|
||
}
|
||
|
||
func (t *Turn) Handle(ctx context.Context) error {
|
||
var resize MessageResize
|
||
|
||
for {
|
||
select {
|
||
case <-ctx.Done():
|
||
return ctx.Err()
|
||
default:
|
||
_, data, err := t.ws.Read(ctx)
|
||
if err != nil {
|
||
// 通常是客户端关闭连接
|
||
return fmt.Errorf("reading ws message err: %v", err)
|
||
}
|
||
|
||
// 判断是否是 resize 消息
|
||
if err = json.Unmarshal(data, &resize); err == nil {
|
||
if resize.Resize && resize.Columns > 0 && resize.Rows > 0 {
|
||
if err = t.session.WindowChange(resize.Rows, resize.Columns); err != nil {
|
||
return fmt.Errorf("change window size err: %v", err)
|
||
}
|
||
}
|
||
continue
|
||
}
|
||
|
||
if _, err = t.stdin.Write(data); err != nil {
|
||
return fmt.Errorf("writing ws message to stdin err: %v", err)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
func (t *Turn) Wait() {
|
||
_ = t.session.Wait()
|
||
}
|