repos / pico

ssh.go

 1package feeds
 2
 3import (
 4	"context"
 5	"os"
 6	"os/signal"
 7	"syscall"
 8
 9	"github.com/picosh/pico/pkg/db/postgres"
10	"github.com/picosh/pico/pkg/filehandlers"
11	"github.com/picosh/pico/pkg/pssh"
12	"github.com/picosh/pico/pkg/send/auth"
13	"github.com/picosh/pico/pkg/send/list"
14	"github.com/picosh/pico/pkg/send/pipe"
15	"github.com/picosh/pico/pkg/send/protocols/rsync"
16	"github.com/picosh/pico/pkg/send/protocols/scp"
17	"github.com/picosh/pico/pkg/send/protocols/sftp"
18	"github.com/picosh/pico/pkg/shared"
19	"github.com/picosh/utils"
20)
21
22func StartSshServer() {
23	appName := "feeds-ssh"
24
25	host := utils.GetEnv("FEEDS_HOST", "0.0.0.0")
26	port := utils.GetEnv("FEEDS_SSH_PORT", "2222")
27	promPort := utils.GetEnv("FEEDS_PROM_PORT", "9222")
28	cfg := NewConfigSite(appName)
29	logger := cfg.Logger
30
31	ctx, cancel := context.WithCancel(context.Background())
32	defer cancel()
33
34	dbh := postgres.NewDB(cfg.DbURL, cfg.Logger)
35	defer dbh.Close()
36
37	hooks := &FeedHooks{
38		Cfg: cfg,
39		Db:  dbh,
40	}
41
42	fileMap := map[string]filehandlers.ReadWriteHandler{
43		"fallback": filehandlers.NewScpPostHandler(dbh, cfg, hooks),
44	}
45	handler := filehandlers.NewFileHandlerRouter(cfg, dbh, fileMap)
46
47	sshAuth := shared.NewSshAuthHandler(dbh, logger)
48
49	// Create a new SSH server
50	server, err := pssh.NewSSHServerWithConfig(
51		ctx,
52		logger,
53		appName,
54		host,
55		port,
56		promPort,
57		"ssh_data/term_info_ed25519",
58		sshAuth.PubkeyAuthHandler,
59		[]pssh.SSHServerMiddleware{
60			pipe.Middleware(handler, ".txt"),
61			list.Middleware(handler),
62			scp.Middleware(handler),
63			rsync.Middleware(handler),
64			auth.Middleware(handler),
65			Middleware(dbh, cfg),
66			pssh.LogMiddleware(handler, dbh),
67		},
68		[]pssh.SSHServerMiddleware{
69			sftp.Middleware(handler),
70			pssh.LogMiddleware(handler, dbh),
71		},
72		nil,
73	)
74
75	if err != nil {
76		logger.Error("failed to create ssh server", "err", err.Error())
77		os.Exit(1)
78	}
79
80	done := make(chan os.Signal, 1)
81
82	signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
83	logger.Info("Starting SSH server", "addr", server.Config.ListenAddr)
84	go func() {
85		if err = server.ListenAndServe(); err != nil {
86			logger.Error("serve", "err", err.Error())
87			os.Exit(1)
88		}
89	}()
90
91	exit := func() {
92		logger.Info("stopping ssh server")
93		cancel()
94	}
95
96	<-done
97	exit()
98}