syz-manager: restore crash collection in snapshotting mode

We should have been closing only the context of the Pool.Loop().

1 files changed, 6 insertions, 3 deletions

diff --git a/syz-manager/manager.go b/syz-manager/manager.go
index 8156ac436..ec4fc5dd4 100644
--- a/syz-manager/manager.go
+++ b/syz-manager/manager.go
@@ -312,13 +312,16 @@ func RunManager(mode Mode, cfg *mgrconfig.Config) {
 		<-vm.Shutdown
 		return
 	}
-	ctx, cancel := context.WithCancel(vm.ShutdownCtx())
-	mgr.loopStop = cancel
 	mgr.pool = vm.NewDispatcher(mgr.vmPool, mgr.fuzzerInstance)
 	mgr.reproMgr = newReproManager(mgr, mgr.vmPool.Count()-mgr.cfg.FuzzingVMs, mgr.cfg.DashboardOnlyRepro)
+	ctx := vm.ShutdownCtx()
 	go mgr.processFuzzingResults(ctx)
 	go mgr.reproMgr.Loop(ctx)
-	mgr.pool.Loop(ctx)
+
+	loopCtx, cancel := context.WithCancel(ctx)
+	mgr.loopStop = cancel
+	mgr.pool.Loop(loopCtx)
+
 	if cfg.Snapshot {
 		log.Logf(0, "starting VMs for snapshot mode")
 		mgr.serv.Close()