From 2c73209516c2abd399bfeafe16ef37ce73e6ce43 Mon Sep 17 00:00:00 2001
From: skyanth <skyanth@gmail.com>
Date: Wed, 29 Mar 2017 11:40:42 +0200
Subject: [PATCH] implemented finding count placeholder (cont.)

---
 xml/dtd/pentestreport.xsd  | 2 +-
 xml/xslt/placeholders.xslt | 9 ++++++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/xml/dtd/pentestreport.xsd b/xml/dtd/pentestreport.xsd
index d4c965a..b2c1b69 100644
--- a/xml/dtd/pentestreport.xsd
+++ b/xml/dtd/pentestreport.xsd
@@ -384,7 +384,7 @@
       <xs:element name="p_reportdue"/>
       <xs:element name="finding_count">
         <xs:complexType>
-          <xs:attribute ref="threatLevel" use="required"/>
+          <xs:attribute ref="threatLevel" use="optional"/>
         </xs:complexType>
       </xs:element>
     </xs:choice>
diff --git a/xml/xslt/placeholders.xslt b/xml/xslt/placeholders.xslt
index 338c6d2..4cd00bc 100644
--- a/xml/xslt/placeholders.xslt
+++ b/xml/xslt/placeholders.xslt
@@ -349,7 +349,14 @@
 
     <xsl:template match="finding_count">
         <xsl:param name="threatLevel" select="@threatLevel"/>
-            <xsl:value-of select="count(//finding[@threatLevel=$threatLevel])" />
+        <xsl:choose>
+            <xsl:when test="@threatLevel">
+                <xsl:value-of select="count(//finding[@threatLevel = $threatLevel])"/>
+            </xsl:when>
+            <xsl:otherwise>
+                <xsl:value-of select="count(//finding)"/>
+            </xsl:otherwise>
+        </xsl:choose>
     </xsl:template>
 
     <xsl:template name="checkPlaceholder">