release v1.2.0

server/global/gateway/gateway.go

@@ -0,0 +1,129 @@
+package gateway
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"net"
+	"os"
+
+	"next-terminal/server/config"
+	"next-terminal/server/utils"
+
+	"golang.org/x/crypto/ssh"
+)
+
+// Gateway 接入网关
+type Gateway struct {
+	ID        string // 接入网关ID
+	Connected bool   // 是否已连接
+	LocalHost string // 隧道映射到本地的IP地址
+	SshClient *ssh.Client
+	Message   string // 失败原因
+
+	tunnels map[string]*Tunnel
+
+	Add  chan *Tunnel
+	Del  chan string
+	exit chan bool
+}
+
+func NewGateway(id, localhost string, connected bool, message string, client *ssh.Client) *Gateway {
+	return &Gateway{
+		ID:        id,
+		LocalHost: localhost,
+		Connected: connected,
+		Message:   message,
+		SshClient: client,
+		Add:       make(chan *Tunnel),
+		Del:       make(chan string),
+		tunnels:   map[string]*Tunnel{},
+		exit:      make(chan bool, 1),
+	}
+}
+
+func (g *Gateway) Run() {
+	for {
+		select {
+		case t := <-g.Add:
+			g.tunnels[t.ID] = t
+			go t.Run()
+		case k := <-g.Del:
+			if _, ok := g.tunnels[k]; ok {
+				g.tunnels[k].Close()
+				delete(g.tunnels, k)
+			}
+		case <-g.exit:
+			return
+		}
+	}
+}
+
+func (g *Gateway) Close() {
+	g.exit <- true
+	if g.SshClient != nil {
+		_ = g.SshClient.Close()
+	}
+	if len(g.tunnels) > 0 {
+		for _, tunnel := range g.tunnels {
+			tunnel.Close()
+		}
+	}
+}
+
+func (g *Gateway) OpenSshTunnel(id, ip string, port int) (exposedIP string, exposedPort int, err error) {
+	if !g.Connected {
+		return "", 0, errors.New(g.Message)
+	}
+
+	localPort, err := utils.GetAvailablePort()
+	if err != nil {
+		return "", 0, err
+	}
+	localHost := g.LocalHost
+	if localHost == "" {
+		if config.GlobalCfg.Container {
+			localIp, err := utils.GetLocalIp()
+			if err != nil {
+				hostname, err := os.Hostname()
+				if err != nil {
+					return "", 0, err
+				} else {
+					localHost = hostname
+				}
+			} else {
+				localHost = localIp
+			}
+		} else {
+			localHost = "localhost"
+		}
+	}
+
+	localAddr := fmt.Sprintf("%s:%d", localHost, localPort)
+	listener, err := net.Listen("tcp", localAddr)
+	if err != nil {
+		return "", 0, err
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	tunnel := &Tunnel{
+		ID:         id,
+		LocalHost:  g.LocalHost,
+		LocalPort:  localPort,
+		Gateway:    g,
+		RemoteHost: ip,
+		RemotePort: port,
+		ctx:        ctx,
+		cancel:     cancel,
+		listener:   listener,
+	}
+	g.Add <- tunnel
+
+	return tunnel.LocalHost, tunnel.LocalPort, nil
+}
+
+func (g Gateway) CloseSshTunnel(id string) {
+	if g.tunnels[id] != nil {
+		g.tunnels[id].Close()
+	}
+}

server/global/gateway/manager.go

@@ -0,0 +1,42 @@
+package gateway
+
+type Manager struct {
+	gateways map[string]*Gateway
+
+	Add chan *Gateway
+	Del chan string
+}
+
+func NewManager() *Manager {
+	return &Manager{
+		Add:      make(chan *Gateway),
+		Del:      make(chan string),
+		gateways: map[string]*Gateway{},
+	}
+}
+
+func (m *Manager) Run() {
+	for {
+		select {
+		case g := <-m.Add:
+			m.gateways[g.ID] = g
+			go g.Run()
+		case k := <-m.Del:
+			if _, ok := m.gateways[k]; ok {
+				m.gateways[k].Close()
+				delete(m.gateways, k)
+			}
+		}
+	}
+}
+
+func (m Manager) GetById(id string) *Gateway {
+	return m.gateways[id]
+}
+
+var GlobalGatewayManager *Manager
+
+func init() {
+	GlobalGatewayManager = NewManager()
+	go GlobalGatewayManager.Run()
+}

server/global/gateway/tunnel.go

@@ -0,0 +1,63 @@
+package gateway
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"net"
+
+	"next-terminal/server/log"
+)
+
+type Tunnel struct {
+	ID         string // 唯一标识
+	LocalHost  string // 本地监听地址
+	LocalPort  int    // 本地端口
+	RemoteHost string // 远程连接地址
+	RemotePort int    // 远程端口
+	Gateway    *Gateway
+	ctx        context.Context
+	cancel     context.CancelFunc
+	listener   net.Listener
+	err        error
+}
+
+func (r *Tunnel) Run() {
+	localAddr := fmt.Sprintf("%s:%d", r.LocalHost, r.LocalPort)
+	log.Debugf("等待客户端访问 [%v] ...", localAddr)
+	localConn, err := r.listener.Accept()
+	if err != nil {
+		r.err = err
+		return
+	}
+
+	log.Debugf("客户端 [%v] 已连接至 [%v]", localConn.RemoteAddr().String(), localAddr)
+	remoteAddr := fmt.Sprintf("%s:%d", r.RemoteHost, r.RemotePort)
+	log.Debugf("连接远程主机 [%v] ...", remoteAddr)
+	remoteConn, err := r.Gateway.SshClient.Dial("tcp", remoteAddr)
+	if err != nil {
+		log.Debugf("连接远程主机 [%v] 失败", remoteAddr)
+		r.err = err
+		return
+	}
+
+	log.Debugf("连接远程主机 [%v] 成功", remoteAddr)
+	go copyConn(localConn, remoteConn)
+	go copyConn(remoteConn, localConn)
+	log.Debugf("开始转发数据 [%v]->[%v]", localAddr, remoteAddr)
+	go func() {
+		<-r.ctx.Done()
+		_ = r.listener.Close()
+		_ = localConn.Close()
+		_ = remoteConn.Close()
+		log.Debugf("SSH隧道 [%v]-[%v] 已关闭", localAddr, remoteAddr)
+	}()
+}
+
+func (r Tunnel) Close() {
+	r.cancel()
+}
+
+func copyConn(writer, reader net.Conn) {
+	_, _ = io.Copy(writer, reader)
+}