Browse Source

Merge branch 'abbreviated_label_fix' of stefan.schmidt/ID2T-toolkit into master

Jens Keim 5 years ago
parent
commit
2ccba7db81
1 changed files with 2 additions and 2 deletions
  1. 2 2
      code/Core/AttackController.py

+ 2 - 2
code/Core/AttackController.py

@@ -104,8 +104,6 @@ class AttackController:
         :param seed: random seed for param generation
         :return: None
         """
-        attack_name = self.choose_attack(attack_name)
-
         print("\nCreating attack instance of \033[1m" + attack_name + "\033[0m")
         # Load attack class
         attack_module = importlib.import_module("Attack." + attack_name)
@@ -137,6 +135,8 @@ class AttackController:
         :param time: Measure packet generation time or not.
         :return: The file path to the created pcap file.
         """
+        attack = self.choose_attack(attack)
+
         self.create_attack(attack, self.seed)
 
         print("Validating and adding attack parameters.")