reverted changed from hashbrown to std::collections

src/afterfact.rs

@@ -14,7 +14,7 @@ use lazy_static::lazy_static;
 use linked_hash_map::LinkedHashMap;
 
 use std::cmp::min;
-use std::collections::{HashMap, HashSet};
+use hashbrown::{HashMap, HashSet};
 use std::error::Error;
 
 use std::fs::File;
@@ -475,7 +475,7 @@ fn _print_detection_summary_by_date(
     let mut wtr = buf_wtr.buffer();
     wtr.set_color(ColorSpec::new().set_fg(None)).ok();
 
-    let level_full_map = HashMap::from([
+    let level_full_map = std::collections::HashMap::from([
         ("crit", "critical"),
         ("high", "high"),
         ("med ", "medium"),
@@ -524,7 +524,7 @@ fn _print_detection_summary_by_computer(
     let mut wtr = buf_wtr.buffer();
     wtr.set_color(ColorSpec::new().set_fg(None)).ok();
 
-    let level_full_map = HashMap::from([
+    let level_full_map = std::collections::HashMap::from([
         ("crit", "critical"),
         ("high", "high"),
         ("med ", "medium"),
@@ -590,7 +590,7 @@ mod tests {
     use chrono::{Local, TimeZone, Utc};
     use linked_hash_map::LinkedHashMap;
     use serde_json::Value;
-    use std::collections::HashMap;
+    use hashbrown::HashMap;
     use std::fs::File;
     use std::fs::{read_to_string, remove_file};
     use std::io;

src/detections/configs.rs

@@ -6,7 +6,7 @@ use chrono::{DateTime, Utc};
 use clap::{App, CommandFactory, Parser};
 use lazy_static::lazy_static;
 use regex::Regex;
-use std::collections::{HashMap, HashSet};
+use hashbrown::{HashMap, HashSet};
 use std::env::current_exe;
 use std::path::PathBuf;
 use std::sync::RwLock;
@@ -568,7 +568,7 @@ fn load_eventcode_info(path: &str) -> EventInfoConfig {
 mod tests {
     use crate::detections::configs;
     use chrono::{DateTime, Utc};
-    use std::collections::HashSet;
+    use hashbrown::HashSet;
 
     //     #[test]
     //     #[ignore]

src/detections/detection.rs

@@ -25,7 +25,7 @@ use crate::detections::utils::{get_serde_number_to_string, make_ascii_titlecase}
 use crate::filter;
 use crate::yaml::ParseYaml;
 use serde_json::Value;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::fmt::Write;
 use std::path::Path;
 

src/detections/message.rs

@@ -11,7 +11,7 @@ use lazy_static::lazy_static;
 use linked_hash_map::LinkedHashMap;
 use regex::Regex;
 use serde_json::Value;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::env;
 use std::fs::create_dir;
 use std::fs::File;
@@ -376,7 +376,7 @@ mod tests {
     use crate::detections::message::AlertMessage;
     use crate::detections::message::{parse_message, MESSAGES};
     use serde_json::Value;
-    use std::collections::HashMap;
+    use hashbrown::HashMap;
 
     use super::{create_output_filter_config, get_default_details};
 

src/detections/pivot.rs

@@ -1,6 +1,6 @@
 use lazy_static::lazy_static;
 use serde_json::Value;
-use std::collections::{HashMap, HashSet};
+use hashbrown::{HashMap, HashSet};
 use std::sync::RwLock;
 
 use crate::detections::configs;

src/detections/rule/condition_parser.rs

@@ -5,7 +5,7 @@ use self::selectionnodes::{
     AndSelectionNode, NotSelectionNode, OrSelectionNode, RefSelectionNode, SelectionNode,
 };
 use super::selectionnodes;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::sync::Arc;
 
 lazy_static! {

src/detections/rule/count.rs

@@ -7,7 +7,7 @@ use crate::detections::rule::AggResult;
 use crate::detections::rule::RuleNode;
 use chrono::{DateTime, TimeZone, Utc};
 use serde_json::Value;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::num::ParseIntError;
 use std::path::Path;
 
@@ -311,9 +311,10 @@ impl CountStrategy for FieldStrategy {
         }
 
         let value = &datas[idx as usize].field_record_value;
-        let key_val = self.value_2_cnt.get_mut(value);
+        let key_val = self.value_2_cnt.get_key_value_mut(value);
         if let Some(kv) = key_val {
-            *kv += 1;
+            let (_, val) = kv;
+            *val += 1;
         } else {
             self.value_2_cnt.insert(value.to_string(), 1);
         }
@@ -498,7 +499,7 @@ mod tests {
     use crate::detections::rule::create_rule;
     use crate::detections::rule::AggResult;
     use crate::detections::utils;
-    use std::collections::HashMap;
+    use hashbrown::HashMap;
 
     use chrono::{TimeZone, Utc};
     use yaml_rust::YamlLoader;

src/detections/rule/mod.rs

@@ -2,7 +2,7 @@ extern crate regex;
 
 use chrono::{DateTime, Utc};
 
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::{fmt::Debug, sync::Arc, vec};
 
 use yaml_rust::Yaml;

src/detections/utils.rs

@@ -4,7 +4,7 @@ extern crate regex;
 
 use crate::detections::configs;
 use crate::detections::configs::CURRENT_EXE_PATH;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::path::Path;
 use std::path::PathBuf;
 

src/filter.rs

@@ -3,7 +3,7 @@ use crate::detections::message::AlertMessage;
 use crate::detections::message::ERROR_LOG_STACK;
 use crate::detections::message::QUIET_ERRORS_FLAG;
 use regex::Regex;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::fs::File;
 use std::io::{BufRead, BufReader};
 

src/main.rs

@@ -27,7 +27,7 @@ use hayabusa::{detections::utils::write_color_buffer, filter};
 use hhmmss::Hhmmss;
 use pbr::ProgressBar;
 use serde_json::Value;
-use std::collections::{HashMap, HashSet};
+use hashbrown::{HashMap, HashSet};
 use std::ffi::{OsStr, OsString};
 use std::fmt::Display;
 use std::fmt::Write as _;

src/options/level_tuning.rs

@@ -2,7 +2,7 @@ use crate::detections::utils::write_color_buffer;
 use crate::detections::{configs, utils};
 use crate::filter::RuleExclude;
 use crate::yaml::ParseYaml;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::fs::{self, File};
 use std::io::Write;
 use termcolor::{BufferWriter, ColorChoice};

src/options/profile.rs

@@ -5,7 +5,7 @@ use crate::yaml;
 use lazy_static::lazy_static;
 use linked_hash_map::LinkedHashMap;
 use regex::RegexSet;
-use std::collections::HashSet;
+use hashbrown::HashSet;
 use std::fs::OpenOptions;
 use std::io::{BufWriter, Write};
 use std::path::Path;

src/options/update_rules.rs

@@ -8,7 +8,7 @@ use std::fs::{self};
 use std::path::Path;
 
 use std::cmp::Ordering;
-use std::collections::{HashMap, HashSet};
+use hashbrown::{HashMap, HashSet};
 
 use std::time::SystemTime;
 

src/timeline/statistics.rs

@@ -1,6 +1,6 @@
 use crate::detections::message::{LOGONSUMMARY_FLAG, STATISTICS_FLAG};
 use crate::detections::{detection::EvtxRecordInfo, utils};
-use std::collections::HashMap;
+use hashbrown::HashMap;
 
 #[derive(Debug)]
 pub struct EventStatistics {

src/timeline/timelines.rs

@@ -3,7 +3,7 @@ use crate::detections::{configs::CONFIG, detection::EvtxRecordInfo};
 use prettytable::{Cell, Row, Table};
 
 use super::statistics::EventStatistics;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 
 #[derive(Debug)]
 pub struct Timeline {

src/yaml.rs

@@ -6,7 +6,7 @@ use crate::detections::configs::EXCLUDE_STATUS;
 use crate::detections::message::AlertMessage;
 use crate::detections::message::{ERROR_LOG_STACK, QUIET_ERRORS_FLAG};
 use crate::filter::RuleExclude;
-use std::collections::HashMap;
+use hashbrown::HashMap;
 use std::ffi::OsStr;
 use std::fs;
 use std::io;
@@ -321,7 +321,7 @@ mod tests {
     use crate::filter;
     use crate::yaml;
     use crate::yaml::RuleExclude;
-    use std::collections::HashMap;
+    use hashbrown::HashMap;
     use std::path::Path;
     use yaml_rust::YamlLoader;