repos / pico

ssh.go

 1package pastes
 2
 3import (
 4	"context"
 5	"os"
 6	"os/signal"
 7	"syscall"
 8	"time"
 9
10	"github.com/picosh/pico/pkg/db/postgres"
11	"github.com/picosh/pico/pkg/filehandlers"
12	"github.com/picosh/pico/pkg/pssh"
13	"github.com/picosh/pico/pkg/send/auth"
14	"github.com/picosh/pico/pkg/send/list"
15	"github.com/picosh/pico/pkg/send/pipe"
16	"github.com/picosh/pico/pkg/send/protocols/rsync"
17	"github.com/picosh/pico/pkg/send/protocols/scp"
18	"github.com/picosh/pico/pkg/send/protocols/sftp"
19	"github.com/picosh/pico/pkg/shared"
20	"github.com/picosh/utils"
21)
22
23func StartSshServer() {
24	appName := "pastes-ssh"
25
26	host := utils.GetEnv("PASTES_HOST", "0.0.0.0")
27	port := utils.GetEnv("PASTES_SSH_PORT", "2222")
28	promPort := utils.GetEnv("PASTES_PROM_PORT", "9222")
29	cfg := NewConfigSite(appName)
30	logger := cfg.Logger
31
32	ctx, cancel := context.WithCancel(context.Background())
33	defer cancel()
34
35	dbh := postgres.NewDB(cfg.DbURL, cfg.Logger)
36	defer func() {
37		_ = dbh.Close()
38	}()
39	hooks := &FileHooks{
40		Cfg: cfg,
41		Db:  dbh,
42	}
43
44	fileMap := map[string]filehandlers.ReadWriteHandler{
45		"fallback": filehandlers.NewScpPostHandler(dbh, cfg, hooks),
46	}
47	handler := filehandlers.NewFileHandlerRouter(cfg, dbh, fileMap)
48	sshAuth := shared.NewSshAuthHandler(dbh, logger)
49
50	// Create a new SSH server
51	server, err := pssh.NewSSHServerWithConfig(
52		ctx,
53		logger,
54		appName,
55		host,
56		port,
57		promPort,
58		"ssh_data/term_info_ed25519",
59		sshAuth.PubkeyAuthHandler,
60		[]pssh.SSHServerMiddleware{
61			pipe.Middleware(handler, ""),
62			list.Middleware(handler),
63			scp.Middleware(handler),
64			rsync.Middleware(handler),
65			auth.Middleware(handler),
66			pssh.PtyMdw(pssh.DeprecatedNotice(), 200*time.Millisecond),
67			pssh.LogMiddleware(handler, dbh),
68		},
69		[]pssh.SSHServerMiddleware{
70			sftp.Middleware(handler),
71			pssh.LogMiddleware(handler, dbh),
72		},
73		nil,
74	)
75
76	if err != nil {
77		logger.Error("failed to create ssh server", "err", err.Error())
78		os.Exit(1)
79	}
80
81	done := make(chan os.Signal, 1)
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}