Remove miniccc client if VM is no longer present

activeshadow · Sep 11, 2023 · b2d0da5 · b2d0da5
1 parent eb0f874
commit b2d0da5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cmd/miniccc/mux.go b/cmd/miniccc/mux.go
@@ -57,6 +57,7 @@ func mux(done chan struct{}) {
 			log.Warn("server connection lost: resetting client")
 
 			// server connection lost, so reset client
+			time.Sleep(5 * time.Second)
 			resetClient()
 			return
 		}

diff --git a/internal/ron/server.go b/internal/ron/server.go
@@ -1128,6 +1128,15 @@ func (s *Server) responseHandler() {
 	for cin := range s.responses {
 		log.Debug("responseHandler: %v", cin.UUID)
 
+		s.clientLock.Lock()
+		_, ok := s.vms[cin.UUID]
+		s.clientLock.Unlock()
+
+		if !ok {
+			s.removeClient(cin.UUID)
+			continue
+		}
+
 		// update all the client fields
 		s.updateClient(cin)