release v1.2.0
This commit is contained in:
dushixiang
13
server/global/cache/cache.go
vendored
Normal file
13
server/global/cache/cache.go
vendored
Normal file
@ -0,0 +1,13 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/patrickmn/go-cache"
|
||||
)
|
||||
|
||||
var GlobalCache *cache.Cache
|
||||
|
||||
func init() {
|
||||
GlobalCache = cache.New(5*time.Minute, 10*time.Minute)
|
||||
}
|
||||
16
server/global/cron/cron.go
Normal file
16
server/global/cron/cron.go
Normal file
@ -0,0 +1,16 @@
|
||||
package cron
|
||||
|
||||
import "github.com/robfig/cron/v3"
|
||||
|
||||
var GlobalCron *cron.Cron
|
||||
|
||||
type Job cron.Job
|
||||
|
||||
func init() {
|
||||
GlobalCron = cron.New(cron.WithSeconds())
|
||||
GlobalCron.Start()
|
||||
}
|
||||
|
||||
func JobId(jobId int) cron.EntryID {
|
||||
return cron.EntryID(jobId)
|
||||
}
|
||||
129
server/global/gateway/gateway.go
Normal file
129
server/global/gateway/gateway.go
Normal file
@ -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()
|
||||
}
|
||||
}
|
||||
42
server/global/gateway/manager.go
Normal file
42
server/global/gateway/manager.go
Normal file
@ -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()
|
||||
}
|
||||
63
server/global/gateway/tunnel.go
Normal file
63
server/global/gateway/tunnel.go
Normal file
@ -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)
|
||||
}
|
||||
70
server/global/security/security.go
Normal file
70
server/global/security/security.go
Normal file
@ -0,0 +1,70 @@
|
||||
package security
|
||||
|
||||
import "sort"
|
||||
|
||||
type Security struct {
|
||||
ID string
|
||||
Rule string
|
||||
IP string
|
||||
Priority int64 // 越小优先级越高
|
||||
}
|
||||
|
||||
type Manager struct {
|
||||
securities map[string]*Security
|
||||
values []*Security
|
||||
|
||||
Add chan *Security
|
||||
Del chan string
|
||||
}
|
||||
|
||||
func NewManager() *Manager {
|
||||
return &Manager{
|
||||
Add: make(chan *Security),
|
||||
Del: make(chan string),
|
||||
securities: map[string]*Security{},
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) Run() {
|
||||
for {
|
||||
select {
|
||||
case s := <-m.Add:
|
||||
m.securities[s.ID] = s
|
||||
m.LoadData()
|
||||
case s := <-m.Del:
|
||||
if _, ok := m.securities[s]; ok {
|
||||
delete(m.securities, s)
|
||||
m.LoadData()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) Clear() {
|
||||
m.securities = map[string]*Security{}
|
||||
}
|
||||
|
||||
func (m *Manager) LoadData() {
|
||||
var values []*Security
|
||||
for _, security := range m.securities {
|
||||
values = append(values, security)
|
||||
}
|
||||
|
||||
sort.Slice(values, func(i, j int) bool {
|
||||
// 优先级数字越小代表优先级越高，因此此处用小于号
|
||||
return values[i].Priority < values[j].Priority
|
||||
})
|
||||
|
||||
m.values = values
|
||||
}
|
||||
|
||||
func (m Manager) Values() []*Security {
|
||||
return m.values
|
||||
}
|
||||
|
||||
var GlobalSecurityManager *Manager
|
||||
|
||||
func init() {
|
||||
GlobalSecurityManager = NewManager()
|
||||
go GlobalSecurityManager.Run()
|
||||
}
|
||||
97
server/global/session/session.go
Normal file
97
server/global/session/session.go
Normal file
@ -0,0 +1,97 @@
|
||||
package session
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"next-terminal/server/guacd"
|
||||
"next-terminal/server/term"
|
||||
|
||||
"github.com/gorilla/websocket"
|
||||
)
|
||||
|
||||
type Session struct {
|
||||
ID string
|
||||
Protocol string
|
||||
Mode string
|
||||
WebSocket *websocket.Conn
|
||||
GuacdTunnel *guacd.Tunnel
|
||||
NextTerminal *term.NextTerminal
|
||||
Observer *Manager
|
||||
}
|
||||
|
||||
type Manager struct {
|
||||
id string
|
||||
sessions map[string]*Session
|
||||
|
||||
Add chan *Session
|
||||
Del chan string
|
||||
exit chan bool
|
||||
}
|
||||
|
||||
func NewManager() *Manager {
|
||||
return &Manager{
|
||||
Add: make(chan *Session),
|
||||
Del: make(chan string),
|
||||
sessions: map[string]*Session{},
|
||||
exit: make(chan bool, 1),
|
||||
}
|
||||
}
|
||||
|
||||
func NewObserver(id string) *Manager {
|
||||
return &Manager{
|
||||
id: id,
|
||||
Add: make(chan *Session),
|
||||
Del: make(chan string),
|
||||
sessions: map[string]*Session{},
|
||||
exit: make(chan bool, 1),
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) Run() {
|
||||
defer fmt.Printf("Session Manager %v End\n", m.id)
|
||||
fmt.Printf("Session Manager %v Run\n", m.id)
|
||||
for {
|
||||
select {
|
||||
case s := <-m.Add:
|
||||
m.sessions[s.ID] = s
|
||||
case k := <-m.Del:
|
||||
if _, ok := m.sessions[k]; ok {
|
||||
ss := m.sessions[k]
|
||||
if ss.GuacdTunnel != nil {
|
||||
_ = ss.GuacdTunnel.Close()
|
||||
}
|
||||
if ss.NextTerminal != nil {
|
||||
_ = ss.NextTerminal.Close()
|
||||
}
|
||||
|
||||
if ss.WebSocket != nil {
|
||||
_ = ss.WebSocket.Close()
|
||||
}
|
||||
if ss.Observer != nil {
|
||||
ss.Observer.Close()
|
||||
}
|
||||
delete(m.sessions, k)
|
||||
}
|
||||
case <-m.exit:
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) Close() {
|
||||
m.exit <- true
|
||||
}
|
||||
|
||||
func (m Manager) GetById(id string) *Session {
|
||||
return m.sessions[id]
|
||||
}
|
||||
|
||||
func (m Manager) All() map[string]*Session {
|
||||
return m.sessions
|
||||
}
|
||||
|
||||
var GlobalSessionManager *Manager
|
||||
|
||||
func init() {
|
||||
GlobalSessionManager = NewManager()
|
||||
go GlobalSessionManager.Run()
|
||||
}
|
||||
Reference in New Issue
Block a user