corrects bug if finding is nil

SerpicoProject · Jul 21, 2017 · 238b819 · 238b819
1 parent f27fdb0
commit 238b819
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/views/findings_list.haml b/views/findings_list.haml
@@ -79,6 +79,7 @@
                 vulns = Hash.new 0
                 if @dread
                   @findings.each do |finding|
+                    next unless finding
                     if finding.dread_total > 39
                       vulns["critical"] += 1
                     elsif finding.dread_total < 40 && finding.dread_total > 24
@@ -92,6 +93,7 @@
                   labels = {"label_1" => "Critical", "label_2" => "Severe", "label_3" => "Moderate", "label_4" => "Low" }
                 elsif @cvss
                   @findings.each do |finding|
+                    next unless finding
                     if finding.cvss_total >= 7.0
                       vulns["severe"] += 1
                     elsif finding.cvss_total >= 4.0 and finding.cvss_total <= 6.9
@@ -103,6 +105,7 @@
                   labels = {"label_1" => "Critical (N/A)", "label_2" => "Severe", "label_3" => "Moderate",  "label_4" => "Low"}
                 elsif @cvssv3
                   @findings.each do |finding|
+                    next unless finding
                     if finding.cvss_total >= 9.0
                       vulns["critical"] += 1
                     elsif finding.cvss_total >= 7.0 and finding.cvss_total <= 8.9
@@ -116,6 +119,7 @@
                   labels = {"label_1" => "Critical", "label_2" => "Severe", "label_3" => "Moderate", "label_4" => "Low" }
                 else
                   @findings.each do |finding|
+                    next unless finding
                     if finding.risk == 4
                       vulns["critical"] += 1
                     elsif finding.risk == 3