Parallelization: fix wrong length for error handling

DataDog · Feb 10, 2022 · 02b8784 · 02b8784
1 parent 5d88ada
commit 02b8784
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd/stratus/detonate_cmd.go b/cmd/stratus/detonate_cmd.go
@@ -64,7 +64,7 @@ func doDetonateCmd(techniques []*stratus.AttackTechnique, cleanup bool) {
 	}
 	close(techniquesChan)
 
-	if hadError := handleErrorsChannel(errorsChan, len(techniquesChan)); hadError {
+	if hadError := handleErrorsChannel(errorsChan, len(techniques)); hadError {
 		os.Exit(1)
 	}
 }

diff --git a/cmd/stratus/warmup_cmd.go b/cmd/stratus/warmup_cmd.go
@@ -53,7 +53,7 @@ func doWarmupCmd(techniques []*stratus.AttackTechnique) {
 	}
 	close(techniquesChan)
 
-	if hadError := handleErrorsChannel(errorsChan, len(techniquesChan)); hadError {
+	if hadError := handleErrorsChannel(errorsChan, len(techniques)); hadError {
 		os.Exit(1)
 	}
 }