From aa65c53a95ad33b356a588ed054b9e0a0dfaf3cc Mon Sep 17 00:00:00 2001 From: Jake Walker Date: Tue, 21 Nov 2023 15:09:51 +0000 Subject: [PATCH] Pass probability of success through to functions --- .../system/services/red_services/data_manipulation_bot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/primaite/simulator/system/services/red_services/data_manipulation_bot.py b/src/primaite/simulator/system/services/red_services/data_manipulation_bot.py index 35ea413a..5e4e2d3f 100644 --- a/src/primaite/simulator/system/services/red_services/data_manipulation_bot.py +++ b/src/primaite/simulator/system/services/red_services/data_manipulation_bot.py @@ -135,8 +135,8 @@ class DataManipulationBot(DatabaseClient): if self.server_ip_address and self.payload and self.operating_state: self.sys_log.info(f"{self.name}: Running") self._logon() - self._perform_port_scan() - self._perform_data_manipulation() + self._perform_port_scan(p_of_success=self.execution_definition.port_scan_p_of_success) + self._perform_data_manipulation(p_of_success=self.execution_definition.data_manipulation_p_of_success) else: self.sys_log.error(f"{self.name}: Failed to start as it requires both a target_ip_address and payload.")