oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2064: turn on everything when node is turned on

Browse Source
This commit is contained in:
Czar Echavez
2023-11-23 22:10:53 +00:00
parent f0fc6518a0
commit 2ce03e0262
7 changed files with 126 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/primaite/simulator/system/services/red_services/data_manipulation_bot.py
View File

@@ -42,10 +42,13 @@ class DataManipulationBot(DatabaseClient):
if self.server_ip_address and self.payload:
self.sys_log.info(f"{self.name}: Attempting to start the {self.name}")
super().run()
if not self.connected:
self.connect()
if self.connected:
self.query(self.payload)
self.sys_log.info(f"{self.name} payload delivered: {self.payload}")
else:
self.sys_log.error(f"Failed to start the {self.name} as it requires both a target_ip_address and payload.")
def attack(self):
"""Run the datab manipulation attack."""
if not self.connected:
self.connect()
if self.connected:
self.query(self.payload)
self.sys_log.info(f"{self.name} payload delivered: {self.payload}")