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

Fix data manipulation bot tests

Browse Source
This commit is contained in:
Jake Walker
2023-11-24 15:53:07 +00:00
parent c5cfbb825a
commit e62ca22cb7
2 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -128,16 +128,16 @@ class DataManipulationBot(DatabaseClient):
# perform the attack # perform the attack
if not self.connected: if not self.connected:
self.connect() self.connect()
if self.connected: if self.connected:
self.query(self.payload) self.query(self.payload)
self.sys_log.info(f"{self.name} payload delivered: {self.payload}") self.sys_log.info(f"{self.name} payload delivered: {self.payload}")
attack_successful = True attack_successful = True
if attack_successful: if attack_successful:
self.sys_log.info(f"{self.name}: Data manipulation successful") self.sys_log.info(f"{self.name}: Data manipulation successful")
self.attack_stage = DataManipulationAttackStage.COMPLETE self.attack_stage = DataManipulationAttackStage.COMPLETE
else: else:
self.sys_log.info(f"{self.name}: Data manipulation failed") self.sys_log.info(f"{self.name}: Data manipulation failed")
self.attack_stage = DataManipulationAttackStage.FAILED self.attack_stage = DataManipulationAttackStage.FAILED
def run(self): def run(self):
""" """

2
tests/unit_tests/_primaite/_simulator/_system/_services/_red_services/test_data_manipulation_bot.py
View File

@@ -4,6 +4,7 @@ from primaite.simulator.network.hardware.base import Node
from primaite.simulator.network.networks import arcd_uc2_network from primaite.simulator.network.networks import arcd_uc2_network
from primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from primaite.simulator.system.applications.application import ApplicationOperatingState
from primaite.simulator.system.services.red_services.data_manipulation_bot import ( from primaite.simulator.system.services.red_services.data_manipulation_bot import (
DataManipulationAttackStage, DataManipulationAttackStage,
DataManipulationBot, DataManipulationBot,
@@ -64,6 +65,7 @@ def test_dm_bot_perform_data_manipulation_no_success(dm_bot):
def test_dm_bot_perform_data_manipulation_success(dm_bot): def test_dm_bot_perform_data_manipulation_success(dm_bot):
dm_bot.attack_stage = DataManipulationAttackStage.PORT_SCAN dm_bot.attack_stage = DataManipulationAttackStage.PORT_SCAN
dm_bot.operating_state = ApplicationOperatingState.RUNNING
dm_bot._perform_data_manipulation(p_of_success=1.0) dm_bot._perform_data_manipulation(p_of_success=1.0)