From c85c175416dc525f1cfd7d853438c69e322af0a4 Mon Sep 17 00:00:00 2001
From: adrien <adrien@malingrey.fr>
Date: Thu, 11 May 2023 12:09:05 +0200
Subject: [PATCH] improve smb share size

---
 nse/smb-shares-size.nse | 11 ++++++-----
 results.xsl             | 10 +++++-----
 2 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/nse/smb-shares-size.nse b/nse/smb-shares-size.nse
index 57571cb..70a3cd5 100644
--- a/nse/smb-shares-size.nse
+++ b/nse/smb-shares-size.nse
@@ -43,13 +43,14 @@ action = function(host)
   -- Get more information on each share
   for i = 1, #shares, 1 do
     local share = shares[i]
-
-    local status, result = get_share_info(host, share)
-    response[share] = result
+    if (share ~= nil) then
+      local status, result = get_share_info(host, share)
+      if (status) then
+        response[share] = result
+      end
+    end
   end
 
-  --table.sort(response)
-
   return response
 end
 
diff --git a/results.xsl b/results.xsl
index 155ce9d..818c4fa 100644
--- a/results.xsl
+++ b/results.xsl
@@ -34,11 +34,11 @@
     margin: auto;
 }
 
-.share-size.primary {
-    --bg: #1678c2;
+.button.share-size {
+    --bg: #21ba45;
 }
 
-.share-size.item {
+.item.share-size {
     --bg: white;
 }
 
@@ -118,7 +118,7 @@
     <div class="column">
         <xsl:variable name="status">
             <xsl:choose>
-                <xsl:when test="$scannedHost/status/@state='up'">info</xsl:when>
+                <xsl:when test="$scannedHost/status/@state='up'">success</xsl:when>
                 <xsl:otherwise>error</xsl:otherwise>
             </xsl:choose>
         </xsl:variable>
@@ -153,7 +153,7 @@
             <xsl:when test="$scannedPort/script[@id='http-info']/elem[@key='status']>=500">red</xsl:when>
             <xsl:when test="$scannedPort/script[@id='http-info']/elem[@key='status']>=400">yellow</xsl:when>
             <xsl:when test="$scannedPort/state/@state='filtered'">yellow</xsl:when>
-            <xsl:when test="$scannedPort/state/@state='open'">primary</xsl:when>
+            <xsl:when test="$scannedPort/state/@state='open'">green</xsl:when>
             <xsl:otherwise>red</xsl:otherwise>
         </xsl:choose>
     </xsl:variable>