fix: 修复关闭SSH终端标签页时会话状态未更新的问题

server/global/gateway/gateway.go

@@ -0,0 +1,105 @@
+package gateway
+
+import (
+	"errors"
+	"fmt"
+	"net"
+	"next-terminal/server/common/term"
+	"os"
+	"sync"
+
+	"next-terminal/server/utils"
+
+	"golang.org/x/crypto/ssh"
+)
+
+// Gateway 接入网关
+type Gateway struct {
+	ID         string // 接入网关ID
+	IP         string
+	Port       int
+	Username   string
+	Password   string
+	PrivateKey string
+	Passphrase string
+	Connected  bool   // 是否已连接
+	Message    string // 失败原因
+	SshClient  *ssh.Client
+
+	mutex   sync.Mutex
+	tunnels map[string]*Tunnel
+}
+
+func (g *Gateway) OpenSshTunnel(id, ip string, port int) (exposedIP string, exposedPort int, err error) {
+	g.mutex.Lock()
+	defer g.mutex.Unlock()
+	if !g.Connected {
+		sshClient, err := term.NewSshClient(g.IP, g.Port, g.Username, g.Password, g.PrivateKey, g.Passphrase)
+		if err != nil {
+			g.Connected = false
+			g.Message = "接入网关不可用：" + err.Error()
+			return "", 0, errors.New(g.Message)
+		} else {
+			g.Connected = true
+			g.SshClient = sshClient
+			g.Message = "使用中"
+		}
+	}
+
+	localPort, err := utils.GetAvailablePort()
+	if err != nil {
+		return "", 0, err
+	}
+
+	hostname, err := os.Hostname()
+	if err != nil {
+		return "", 0, err
+	}
+
+	// debug
+	//hostname = "0.0.0.0"
+
+	localAddr := fmt.Sprintf("%s:%d", hostname, localPort)
+	listener, err := net.Listen("tcp", localAddr)
+	if err != nil {
+		return "", 0, err
+	}
+
+	tunnel := &Tunnel{
+		id:        id,
+		localHost: hostname,
+		//localHost:  "docker.for.mac.host.internal",
+		localPort:  localPort,
+		remoteHost: ip,
+		remotePort: port,
+		listener:   listener,
+	}
+	go tunnel.Open(g.SshClient)
+	g.tunnels[tunnel.id] = tunnel
+
+	return tunnel.localHost, tunnel.localPort, nil
+}
+
+func (g *Gateway) CloseSshTunnel(id string) {
+	g.mutex.Lock()
+	defer g.mutex.Unlock()
+	t := g.tunnels[id]
+	if t != nil {
+		t.Close()
+		delete(g.tunnels, id)
+	}
+
+	if len(g.tunnels) == 0 {
+		if g.SshClient != nil {
+			_ = g.SshClient.Close()
+		}
+		g.Connected = false
+		g.Message = "暂未使用"
+	}
+}
+
+func (g *Gateway) Close() {
+	for id := range g.tunnels {
+		g.CloseSshTunnel(id)
+	}
+}

server/global/gateway/manager.go

@@ -0,0 +1,50 @@
+package gateway
+
+import (
+	"sync"
+
+	"next-terminal/server/model"
+)
+
+type manager struct {
+	gateways sync.Map
+}
+
+func (m *manager) GetById(id string) *Gateway {
+	if val, ok := m.gateways.Load(id); ok {
+		return val.(*Gateway)
+	}
+	return nil
+}
+
+func (m *manager) Add(model *model.AccessGateway) *Gateway {
+	g := &Gateway{
+		ID:         model.ID,
+		IP:         model.IP,
+		Port:       model.Port,
+		Username:   model.Username,
+		Password:   model.Password,
+		PrivateKey: model.PrivateKey,
+		Passphrase: model.Passphrase,
+		Connected:  false,
+		SshClient:  nil,
+		Message:    "暂未使用",
+		tunnels:    make(map[string]*Tunnel),
+	}
+	m.gateways.Store(g.ID, g)
+	return g
+}
+
+func (m *manager) Del(id string) {
+	g := m.GetById(id)
+	if g != nil {
+		g.Close()
+	}
+	m.gateways.Delete(id)
+}
+
+var GlobalGatewayManager *manager
+
+func init() {
+	GlobalGatewayManager = &manager{}
+}

server/global/gateway/tunnel.go

@@ -0,0 +1,57 @@
+package gateway
+
+import (
+	"fmt"
+	"io"
+	"net"
+
+	"golang.org/x/crypto/ssh"
+)
+
+type Tunnel struct {
+	id                string // 唯一标识
+	localHost         string // 本地监听地址
+	localPort         int    // 本地端口
+	remoteHost        string // 远程连接地址
+	remotePort        int    // 远程端口
+	listener          net.Listener
+	localConnections  []net.Conn
+	remoteConnections []net.Conn
+}
+
+func (r *Tunnel) Open(sshClient *ssh.Client) {
+
+	for {
+		localConn, err := r.listener.Accept()
+		if err != nil {
+			return
+		}
+		r.localConnections = append(r.localConnections, localConn)
+
+		remoteAddr := fmt.Sprintf("%s:%d", r.remoteHost, r.remotePort)
+		remoteConn, err := sshClient.Dial("tcp", remoteAddr)
+		if err != nil {
+			return
+		}
+		r.remoteConnections = append(r.remoteConnections, remoteConn)
+
+		go copyConn(localConn, remoteConn)
+		go copyConn(remoteConn, localConn)
+	}
+}
+
+func (r *Tunnel) Close() {
+	for i := range r.localConnections {
+		_ = r.localConnections[i].Close()
+	}
+	r.localConnections = nil
+	for i := range r.remoteConnections {
+		_ = r.remoteConnections[i].Close()
+	}
+	r.remoteConnections = nil
+	_ = r.listener.Close()
+}
+
+func copyConn(writer, reader net.Conn) {
+	_, _ = io.Copy(writer, reader)
+}