fix: ensure SSH connections close on client disconnect

server/handler.go

@@ -11,7 +11,11 @@ func (s *Server) handleConnection(conn net.Conn) {
 	sshConn, chans, forwardingReqs, err := ssh.NewServerConn(conn, s.Config)
 	if err != nil {
 		log.Printf("failed to establish SSH connection: %v", err)
-		conn.Close()
+		err := conn.Close()
+		if err != nil {
+			log.Printf("failed to close SSH connection: %v", err)
+			return
+		}
 		return
 	}
 
@@ -21,5 +25,12 @@ func (s *Server) handleConnection(conn net.Conn) {
 	for ch := range chans {
 		newSession.ChannelChan <- ch
 	}
+
+	defer func(newSession *session.Session) {
+		err := newSession.Close()
+		if err != nil {
+			log.Printf("failed to close session: %v", err)
+		}
+	}(newSession)
 	return
 }