feat: add ssh server

backend/pkg/proto/ssh/handler/sider.go

@@ -0,0 +1,47 @@
+package handler
+
+import (
+	"fmt"
+
+	gossh "github.com/gliderlabs/ssh"
+
+	"github.com/veops/oneterm/pkg/logger"
+	"github.com/veops/oneterm/pkg/proto/ssh/config"
+)
+
+func RegisterMonitorSession(sessionId string, sess gossh.Session) {
+	_, ok := config.TotalHostSession.Load(sessionId)
+	if !ok {
+		return
+	}
+
+	config.TotalMonitors.LoadOrStore(sessionId, sess)
+
+	if _, ok := config.TotalMonitors.Load(sessionId); !ok {
+		config.TotalMonitors.Store(sessionId, sess)
+	}
+
+	<-sess.Context().Done()
+	DeleteMonitorSession(sessionId)
+}
+
+func DeleteMonitorSession(sessionId string) {
+	config.TotalMonitors.Delete(sessionId)
+}
+
+func getMonitorSession(sessionId string) gossh.Session {
+	if v, ok := config.TotalMonitors.Load(sessionId); ok {
+		return v.(gossh.Session)
+	}
+	return nil
+}
+
+func Monitor(sessionId string, p []byte) {
+	if s := getMonitorSession(sessionId); s != nil {
+		_, err := s.Write(p)
+		if err != nil {
+			logger.L.Error(fmt.Sprintf("moninor session %s failed: %s", sessionId, err.Error()))
+			return
+		}
+	}
+}