From 5e659a215726fed2dcc4cd98f679b08dcb28a817 Mon Sep 17 00:00:00 2001
From: alirezade <alirezade@WPPC1N9X16>
Date: Wed, 31 Jan 2024 10:40:25 +0100
Subject: [PATCH] fix memory leak error

---
 src/AgentProcessor.py | 18 +++++++++---------
 src/Helper.py         |  7 ++++---
 start.sh              |  1 +
 3 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/src/AgentProcessor.py b/src/AgentProcessor.py
index b802744..4aeef19 100644
--- a/src/AgentProcessor.py
+++ b/src/AgentProcessor.py
@@ -9,15 +9,15 @@ class AgentProcessor:
 
     def __init__(self, predictor_address, attacks_address, file_address, server_connection):
         self.__processes = []
-        annotator = ProcessAnnotator(predictor_address, attacks_address)
-        flow_file_writer = ProcessFlowFileWriter(file_address)
-        status_sender = ProcessStatusSenderMQTT(server_connection, voting_interval=Config.StatusSender.voting_interval)
-        flow_sender = ProcessFlowSenderMQTT(server_connection)
-
-        self.__processes.append(annotator)
-        self.__processes.append(flow_file_writer)
-        if annotator.is_prediction_enabled():
-            self.__processes.append(status_sender)
+        # annotator = ProcessAnnotator(predictor_address, attacks_address)
+        # flow_file_writer = ProcessFlowFileWriter(file_address)
+        # status_sender = ProcessStatusSenderMQTT(server_connection, voting_interval=Config.StatusSender.voting_interval)
+        # flow_sender = ProcessFlowSenderMQTT(server_connection)
+        #
+        # self.__processes.append(annotator)
+        # self.__processes.append(flow_file_writer)
+        # if annotator.is_prediction_enabled():
+        #     self.__processes.append(status_sender)
         #self.__processes.append(flow_sender)
 
     def process(self, flow):
diff --git a/src/Helper.py b/src/Helper.py
index 6d616ef..c2bdfa5 100644
--- a/src/Helper.py
+++ b/src/Helper.py
@@ -117,11 +117,12 @@ def configure_log_files (directory, separate_event_log):
 
     @staticmethod
     def log(text,  level):
-
-        msg = f"{text} [{inspect.currentframe().f_back}]"
+        frame = f"[{inspect.currentframe().f_back}]"
+        msg = f"{text} {frame}"
+        print_msg = f"{Log.log_colors[level]}[{logging.getLevelName(level)}] {text}\033[0m {frame}"
 
         logger = Log.event_logger if Log.event_logger else logging
         logger.log(level, msg)
 
-        print_msg = Log.log_colors[level] + f'[{logging.getLevelName(level)}] {msg}' + '\033[0m'
+
         print(print_msg)
diff --git a/start.sh b/start.sh
index f56c823..7ed0226 100755
--- a/start.sh
+++ b/start.sh
@@ -9,6 +9,7 @@ printStep(){
 
 printStep "Getting last changes pull request"
 sudo git pull
+sleep 1
 
 printStep "Check Requirement.txt"
 pip install -r requirements.txt