From 3b0a2adf296791bdbd0c1b0fba6399910c1fac82 Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Sun, 17 Aug 2025 04:08:24 +0800 Subject: Context fixes --- forged/internal/incoming/ssh/ssh.go | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) (limited to 'forged/internal/incoming/ssh') diff --git a/forged/internal/incoming/ssh/ssh.go b/forged/internal/incoming/ssh/ssh.go index b0269e5..efa24f1 100644 --- a/forged/internal/incoming/ssh/ssh.go +++ b/forged/internal/incoming/ssh/ssh.go @@ -1,6 +1,7 @@ package ssh import ( + "context" "fmt" "os" @@ -53,8 +54,19 @@ func New(config Config) (server *Server, err error) { return } -func (server *Server) Run() (err error) { +func (server *Server) Run(ctx context.Context) (err error) { listener, err := misc.Listen(server.net, server.addr) + defer func() { + _ = listener.Close() + }() + + go func() { + <-ctx.Done() + _ = server.gliderServer.Close() + _ = listener.Close() // unnecessary? + // TODO: Log the error + }() + if err = server.gliderServer.Serve(listener); err != nil { return fmt.Errorf("serve SSH: %w", err) } -- cgit v1.2.3