From 7d66a73307239baa2ad20651087f93fefd8ee405 Mon Sep 17 00:00:00 2001
From: Matevz Erzen <matevz.erzen@xlab.si>
Date: Fri, 20 May 2022 14:48:56 +0200
Subject: [PATCH] Additional logging

---
 wazuh_evidence_collector/wazuh_evidence_collector.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/wazuh_evidence_collector/wazuh_evidence_collector.py b/wazuh_evidence_collector/wazuh_evidence_collector.py
index 3d5c7c9..fc7962d 100644
--- a/wazuh_evidence_collector/wazuh_evidence_collector.py
+++ b/wazuh_evidence_collector/wazuh_evidence_collector.py
@@ -68,7 +68,12 @@ def main():
 
 # Wrapper function that runs all the checks (for every manager/agent)
 def run_collector():
-    checker = Checker(wc, es, LOGGER) if not DUMMY_WAZUH_MANAGER else DemoChecker()
+    if DUMMY_WAZUH_MANAGER is False:
+        LOGGER.debug("Using Checker to evaluate Wazuh logs.")
+        checker = Checker(wc, es, LOGGER)
+    else:
+        LOGGER.debug("Simulating evidence generation using Demo Checker.")
+        checker = DemoChecker()
 
     # Get list of all agent ids (including manager's)
     def get_agents(wc):
@@ -87,10 +92,11 @@ def run_collector():
 
     ae_req_list = []
 
+    LOGGER.debug("Generating evidences for " + str(len(agent_list)) + " agents...")
     for agent in agent_list:
         ae_req_list.append(generate_evidence(agent, checker))
 
-    # TODO:
+    LOGGER.debug("Forwarding " + str(len(ae_req_list)) + " evidences...")
     for ae_req in ae_req_list:
         forwarder.send_evidence(ae_req, oauth_client.get_token())
         print_evidence(LOGGER, ae_req.evidence)
-- 
GitLab