repos / pico

sftp.go

 1package sftp
 2
 3import (
 4	"errors"
 5	"fmt"
 6	"io"
 7
 8	"github.com/picosh/pico/pkg/pssh"
 9	"github.com/picosh/pico/pkg/send/utils"
10	"github.com/pkg/sftp"
11)
12
13// func SSHOption(writeHandler utils.CopyFromClientHandler) ssh.Option {
14// 	return func(server *ssh.Server) error {
15// 		if server.SubsystemHandlers == nil {
16// 			server.SubsystemHandlers = map[string]ssh.SubsystemHandler{}
17// 		}
18
19// 		server.SubsystemHandlers["sftp"] = SubsystemHandler(writeHandler)
20// 		return nil
21// 	}
22// }
23
24func Middleware(writeHandler utils.CopyFromClientHandler) pssh.SSHServerMiddleware {
25	return func(next pssh.SSHServerHandler) pssh.SSHServerHandler {
26		return func(session *pssh.SSHServerConnSession) error {
27			logger := writeHandler.GetLogger(session).With(
28				"sftp", true,
29			)
30
31			defer func() {
32				if r := recover(); r != nil {
33					logger.Error("error running sftp middleware", "err", r)
34					fmt.Fprintln(session, "error running sftp middleware, check the flags you are using")
35				}
36			}()
37
38			err := writeHandler.Validate(session)
39			if err != nil {
40				fmt.Fprintln(session.Stderr(), err)
41				return err
42			}
43
44			handler := &handlererr{
45				Handler: &handler{
46					session:      session,
47					writeHandler: writeHandler,
48				},
49			}
50
51			handlers := sftp.Handlers{
52				FilePut:  handler,
53				FileList: handler,
54				FileGet:  handler,
55				FileCmd:  handler,
56			}
57
58			requestServer := sftp.NewRequestServer(session, handlers)
59
60			err = requestServer.Serve()
61			if err != nil && !errors.Is(err, io.EOF) {
62				fmt.Fprintln(session.Stderr(), err)
63				logger.Error("Error serving sftp subsystem", "err", err)
64			}
65
66			return err
67		}
68	}
69}