216cceeb8e5e3e4015abe19dacd345398fb27303

Author
Ayman Bagabas <ayman.bagabas@gmail.com>
Committer
Ayman Bagabas <ayman.bagabas@gmail.com>
Date

Message

fix: graceful shutdown

Diff

 1diff --git a/server/server.go b/server/server.go
 2index cbf40fd66a1f4aacd8e74c9677e05bbb52496d23..fed56f3588a4f037a50ec01d2741a73f07bf5d1f 100644
 3--- a/server/server.go
 4+++ b/server/server.go
 5@@ -67,12 +67,18 @@ func (srv *Server) Reload() error {
 6 
 7 // Start starts the SSH server.
 8 func (srv *Server) Start() error {
 9-	return srv.SSHServer.ListenAndServe()
10+	if err := srv.SSHServer.ListenAndServe(); err != ssh.ErrServerClosed {
11+		return err
12+	}
13+	return nil
14 }
15 
16 // Serve serves the SSH server using the provided listener.
17 func (srv *Server) Serve(l net.Listener) error {
18-	return srv.SSHServer.Serve(l)
19+	if err := srv.SSHServer.Serve(l); err != ssh.ErrServerClosed {
20+		return err
21+	}
22+	return nil
23 }
24 
25 // Shutdown lets the server gracefully shutdown.