Merge branch 'feature/code_refactor'

src/detections/detection.rs

@@ -8,7 +8,6 @@ use crate::models::event;
 use evtx::EvtxParser;
 use quick_xml::de::DeError;
 use std::collections::BTreeMap;
-use std::collections::HashMap;
 
 #[derive(Debug)]
 pub struct Detection {

src/detections/security.rs

@@ -42,56 +42,51 @@ impl Security {
     // Special privileges assigned to new logon (possible admin access)
     //
     fn se_debug_privilege(&mut self, event_data: HashMap<String, String>) {
-        match event_data.get("PrivilegeList") {
-            Some(privileage_list) => {
-                match privileage_list.find("SeDebugPrivilege") {
-                    Some(_data) => {
-                        // alert_all_adminが有効であれば、標準出力して知らせる
-                        // DeepBlueCLIでは必ず0になっていて、基本的には表示されない。
-                        if self.alert_all_admin == 1 {
-                            println!("Logon with SeDebugPrivilege (admin access)");
-                            println!("Username:{}", event_data["SubjectUserName"]);
-                            println!("Domain:{}", event_data["SubjectDomainName"]);
-                            println!("User SID:{}", event_data["SubjectUserSid"]);
-                            println!("Domain:{}", event_data["PrivilegeList"]);
-                        }
 
-                        self.total_admin_logons += 1;
+        if let Some(privileage_list) = event_data.get("PrivilegeList") {
+            if let Some(_data) = privileage_list.find("SeDebugPrivilege") {
+                // alert_all_adminが有効であれば、標準出力して知らせる
+                // DeepBlueCLIでは必ず0になっていて、基本的には表示されない。
+                if self.alert_all_admin == 1 {
+                    println!("Logon with SeDebugPrivilege (admin access)");
+                    println!("Username:{}", event_data["SubjectUserName"]);
+                    println!("Domain:{}", event_data["SubjectDomainName"]);
+                    println!("User SID:{}", event_data["SubjectUserSid"]);
+                    println!("Domain:{}", event_data["PrivilegeList"]);
+                }
 
-                        // admin_logons配列にusernameが含まれているか確認
+                self.total_admin_logons += 1;
-                        match self.admin_logons.get(&event_data["SubjectUserName"]) {
-                            Some(sid) => {
-                                // 含まれていれば、マルチユーザが管理者としてログインしているか確認
-                                // マルチログオンのデータをセット
-                                if event_data["SubjectUserName"] != event_data["SubjectUserSid"] {
-                                    // One username with multiple admin logon SIDs
-                                    self.multiple_admin_logons
-                                        .insert(event_data["SubjectUserName"].to_string(), 1);
 
-                                    let mut count_hash: HashMap<String, i32> = HashMap::new();
+                // admin_logons配列にusernameが含まれているか確認
-                                    count_hash.insert(
+                match self.admin_logons.get(&event_data["SubjectUserName"]) {
-                                        event_data["SubjectUserSid"].to_string(),
+                    Some(sid) => {
-                                        sid[&event_data["SubjectUserSid"]] + 1,
+                        // 含まれていれば、マルチユーザが管理者としてログインしているか確認
-                                    );
+                        // マルチログオンのデータをセット
-                                    self.admin_logons.insert(
+                        if event_data["SubjectUserName"] != event_data["SubjectUserSid"] {
-                                        event_data["SubjectUserName"].to_string(),
+                            // One username with multiple admin logon SIDs
-                                        count_hash,
+                            self.multiple_admin_logons
-                                    );
+                                .insert(event_data["SubjectUserName"].to_string(), 1);
-                                }
+
-                            }
+                            let mut count_hash: HashMap<String, i32> = HashMap::new();
-                            None => {
+                            count_hash.insert(
-                                // admin_logons配列にセットUserNameとSIDとカウンタをセット
+                                event_data["SubjectUserSid"].to_string(),
-                                let mut count_hash: HashMap<String, i32> = HashMap::new();
+                                sid[&event_data["SubjectUserSid"]] + 1,
-                                count_hash.insert(event_data["SubjectUserSid"].to_string(), 1);
+                            );
-                                self.admin_logons
+                            self.admin_logons.insert(
-                                    .insert(event_data["SubjectUserName"].to_string(), count_hash);
+                                event_data["SubjectUserName"].to_string(),
-                            }
+                                count_hash,
+                            );
                         }
                     }
-                    None => (),
+                    None => {
+                        // admin_logons配列にセットUserNameとSIDとカウンタをセット
+                        let mut count_hash: HashMap<String, i32> = HashMap::new();
+                        count_hash.insert(event_data["SubjectUserSid"].to_string(), 1);
+                        self.admin_logons
+                            .insert(event_data["SubjectUserName"].to_string(), count_hash);
+                    }
                 }
             }
-            None => (),
         }
     }
 }

src/main.rs

@@ -4,7 +4,7 @@ extern crate serde;
 use clap::{App, AppSettings, Arg};
 use evtx::EvtxParser;
 use quick_xml::de::DeError;
-use std::{env, path::PathBuf, process};
+use std::{path::PathBuf, process};
 use yamato_event_analyzer::detections::detection;
 
 fn build_app() -> clap::App<'static, 'static> {
@@ -38,9 +38,8 @@ fn main() -> Result<(), DeError> {
     let args = build_app().get_matches();
     let filepath: Option<&str> = args.value_of("filepath");
 
-    match filepath {
+    if let Some(filepath) = filepath {
-        Some(filepath) => parse_file(filepath),
+        parse_file(filepath);
-        None => (),
     }
 
     Ok(())

src/models/event.rs

@@ -85,12 +85,11 @@ impl Evtx {
     // 文字列データを取得する
     //
     fn get_string(v: &Data) -> String {
-        match &v.text {
+        let mut ret = "".to_string();
-            Some(text) => {
+        if let Some(text) = &v.text {
-                return text.to_string();
+            ret = text.to_string();
-            }
-            _ => return "".to_string(),
         }
+        return ret;
     }
 
     //
@@ -99,21 +98,14 @@ impl Evtx {
     pub fn parse_event_data(&self) -> HashMap<String, String> {
         let mut values = HashMap::new();
 
-        match &self.event_data {
+        if let Some(event_data) = &self.event_data {
-            Some(event_data) => match &event_data.data {
+            if let Some(data) = &event_data.data {
-                Some(data) => {
+                for v in data.iter() {
-                    for v in data.iter() {
+                    if let Some(name) = &v.name {
-                        match &v.name {
+                        values.insert(name.to_string(), Evtx::get_string(v));
-                            Some(name) => {
-                                values.insert(name.to_string(), Evtx::get_string(v));
-                            }
-                            None => (),
-                        }
                     }
                 }
-                None => (),
+            }
-            },
-            None => (),
         }
 
         values