fix(agent/agentssh): wait for sessions to exit (coder#8008)

mafredri · web-flow · commit 74fdcb1ace5a · 2023-06-13T17:52:31.000Z
diff --git 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
 }

Original file line number	Diff line number	Diff line change
`@@ -648,9 +648,11 @@ func (s *Server) trackSession(ss ssh.Session, add bool) (ok bool) {`
`648`	`648`	`// Server closed.`
`649`	`649`	`return false`
`650`	`650`	`}`
	`651`	`+ s.wg.Add(1)`
`651`	`652`	`s.sessions[ss] = struct{}{}`
`652`	`653`	`return true`
`653`	`654`	`}`
	`655`	`+ s.wg.Done()`
`654`	`656`	`delete(s.sessions, ss)`
`655`	`657`	`return true`
`656`	`658`	`}`