From 07a76ceff5b486658eb612f0c8388e79681497d1 Mon Sep 17 00:00:00 2001
From: Mathias Fredriksson <mafredri@gmail.com>
Date: Tue, 13 Jun 2023 17:40:19 +0000
Subject: [PATCH] fix(agent/agentssh): wait for sessions to exit
---
agent/agentssh/agentssh.go | 2 ++
1 file changed, 2 insertions(+)
diff --git a/agent/agentssh/agentssh.go b/agent/agentssh/agentssh.go
index fcf562a888293..6f39d116acf00 100644
--- a/agent/agentssh/agentssh.go
+++ b/agent/agentssh/agentssh.go
@@ -648,9 +648,11 @@ func (s *Server) trackSession(ss ssh.Session, add bool) (ok bool) {
// Server closed.
return false
}
+ s.wg.Add(1)
s.sessions[ss] = struct{}{}
return true
}
+ s.wg.Done()
delete(s.sessions, ss)
return true
}