From cfd3b167e41aba1b70838965e53dd7801b392e5b Mon Sep 17 00:00:00 2001
From: Sebastien Tricaud <sebastien@honeynet.org>
Date: Thu, 16 Apr 2020 08:55:12 -0700
Subject: [PATCH] Fix bug when we read data that does not exists but try to
 count anyway

---
 src/daemon/db.rs | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/src/daemon/db.rs b/src/daemon/db.rs
index cff4cd3..0c8e52f 100644
--- a/src/daemon/db.rs
+++ b/src/daemon/db.rs
@@ -108,13 +108,20 @@ impl Database {
         };
     }
     pub fn get_count(&mut self, path: &str, value: &str) -> u128 {
-        let valuestable = self.hashtable.get_mut(&path.to_string()).unwrap();
-        let attr = valuestable.get_mut(&value.to_string());
-        match attr {
-            Some(attr) => { return attr.count(); },
+        let valuestable = self.hashtable.get_mut(&path.to_string());
+        match valuestable {
+            Some(valuestable) => {
+                let attr = valuestable.get_mut(&value.to_string());
+                match attr {
+                    Some(attr) => { return attr.count(); },
+                    None => {
+                        return 0;
+                    },            
+                };
+            },
             None => {
                 return 0;
-            },            
+            },
         };
     }
     pub fn get_attr(&mut self, path: &str, value: &str, with_stats: bool, consensus_count: u128) -> String {