Support checks vuls (#121)

* support GetSubstepsWithVulnerabilities * minor * minor fix * small fix
kubescape · Jul 20, 2023 · 04dd5f1 · 04dd5f1
1 parent 8334eca
commit 04dd5f1
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/reporthandling/attacktrack/v1alpha1/attacktrackmethods.go b/reporthandling/attacktrack/v1alpha1/attacktrackmethods.go
@@ -290,10 +290,11 @@ func (handler *AttackTrackAllPathsHandler) GenerateAttackTrackFromPaths(paths []
 // filterNodesWithControls - filters out nodes that do not have controls
 func (handler *AttackTrackAllPathsHandler) filterNodesWithControls(step IAttackTrackStep, paths [][]IAttackTrackStep) *AttackTrackStep {
 	filteredStep := AttackTrackStep{
-		Name:        step.GetName(),
-		Description: step.GetDescription(),
-		SubSteps:    nil,
-		Controls:    step.GetControls(),
+		Name:                  step.GetName(),
+		Description:           step.GetDescription(),
+		SubSteps:              nil,
+		Controls:              step.GetControls(),
+		ChecksVulnerabilities: step.DoesCheckVulnerabilities(),
 	}
 
 	if step.Length() == 0 {