diff --git a/src/main.rs b/src/main.rs index ade1ad82..335cfc76 100644 --- a/src/main.rs +++ b/src/main.rs @@ -73,7 +73,7 @@ fn detect_files(evtx_files: Vec) { let evnt_records = evtx_to_jsons(&evtx_files); let mut tl = Timeline::new(); - tl.start(&evnt_records); + tl.start(&evtx_files, &evnt_records); let mut detection = detection::Detection::new(); &detection.start(evnt_records); diff --git a/src/timeline/statistics.rs b/src/timeline/statistics.rs index f0cfe1c1..fb1c415e 100644 --- a/src/timeline/statistics.rs +++ b/src/timeline/statistics.rs @@ -1,3 +1,5 @@ +use std::path::PathBuf; + use crate::detections::{configs, detection::EvtxRecordInfo}; #[derive(Debug)] @@ -18,7 +20,11 @@ impl EventStatistics { // recordからEventIDを取得するには、detection::utils::get_event_value()という関数があるので、それを使うと便利かもしれません。 // 現状では、この関数の戻り値として返すVecを表示するコードは実装していません。 - pub fn start(&mut self, _records: &Vec) -> Vec { + pub fn start( + &mut self, + evtx_files: &Vec, + records: &Vec, + ) -> Vec { // 引数でstatisticsオプションが指定されている時だけ、統計情報を出力する。 if !configs::CONFIG .read() diff --git a/src/timeline/timeline.rs b/src/timeline/timeline.rs index 1ac008d6..ad4f02fc 100644 --- a/src/timeline/timeline.rs +++ b/src/timeline/timeline.rs @@ -1,3 +1,5 @@ +use std::path::PathBuf; + use crate::detections::detection::EvtxRecordInfo; use super::statistics::EventStatistics; @@ -10,8 +12,8 @@ impl Timeline { return Timeline {}; } - pub fn start(&mut self, records: &Vec) { + pub fn start(&mut self, evtx_files: &Vec, records: &Vec) { let mut statistic = EventStatistics::new(); - statistic.start(records); + statistic.start(evtx_files, records); } }