From 135130a42daabf2407d7a608433727274e961ced Mon Sep 17 00:00:00 2001 From: alirezade Date: Thu, 28 Dec 2023 00:04:23 +0100 Subject: [PATCH] fix --- src/AttackerRemote.py | 29 ++++++++++++++++------------- 1 file changed, 16 insertions(+), 13 deletions(-) diff --git a/src/AttackerRemote.py b/src/AttackerRemote.py index 4846e94..3c8722f 100644 --- a/src/AttackerRemote.py +++ b/src/AttackerRemote.py @@ -8,19 +8,7 @@ import queue -def find_device_address(device_name): - if device_name.lower() == 'plc1': - return '192.168.0.11' - elif device_name.lower() == 'plc2': - target = '192.168.0.12' - elif device_name.lower() == 'hmi1': - target = '192.168.0.21' - - elif device_name.lower() == 'hmi2': - target = '192.168.0.22' - else: - raise Exception(f'target:({device_name}) is not recognized!') class AttackerRemote(AttackerBase): @@ -100,7 +88,7 @@ def process_messages(self, msg): if mode.lower() == 'link': target_1 = self.find_tag_in_msg(msg, 'target1') target_2 = self.find_tag_in_msg(msg, 'target2') - target = target_1 + "," + target_2 + target = self.find_device_address(target_1) + "," + self.find_device_address(target_2) self._replay_scapy_attack(target=target, timeout=timeout, replay_count=replay) else: raise Exception(f"attack type: ({attack}) is not recognized!") @@ -113,6 +101,21 @@ def find_tag_in_msg(msg, tag): raise Exception(f'Cannot find tag name: ({tag}) in message!') return msg[tag] + @staticmethod + def find_device_address(device_name): + if device_name.lower() == 'plc1': + return '192.168.0.11' + elif device_name.lower() == 'plc2': + target = '192.168.0.12' + + elif device_name.lower() == 'hmi1': + target = '192.168.0.21' + + elif device_name.lower() == 'hmi2': + target = '192.168.0.22' + else: + raise Exception(f'target:({device_name}) is not recognized!') + if __name__ == '__main__': attackerRemote = AttackerRemote("input/sample_mqtt_connection.txt")