replay_file_path

Author: t-contreras

modules/ansible/roles/data_replay/tasks/main.yml

@@ -2,7 +2,7 @@
 
 - name: Upload replay
   copy:
-    src: ../../{{ file_name }}
+    src: "{{ file_name }}"
     dest: /tmp/data.log
 
 - name: Call oneshot import

modules/aws_controller.py

@@ -396,6 +396,10 @@ def dump(self, dump_name, search, earliest, latest) -> None:
         self.logger.info("[Completed]")
 
     def replay(self, file_name, index, sourcetype, source) -> None:
+        ### check if input log file is afile path or just a file name
+        ### if file name ,assume it is in current working dir of attack_range.py
+        if Path(file_name).parent == Path("."):
+            file_name = str(Path("../..")/file_name)
         ansible_vars = {}
         ansible_vars["file_name"] = file_name
         ansible_vars["ansible_user"] = "ubuntu"

modules/azure_controller.py

@@ -12,7 +12,7 @@
 from modules.attack_range_controller import AttackRangeController
 from modules.art_simulation_controller import ArtSimulationController
 from modules.purplesharp_simulation_controller import PurplesharpSimulationController
-
+from pathlib import Path
 
 class AzureController(AttackRangeController):
 
@@ -287,6 +287,10 @@ def dump(self, dump_name, search, earliest, latest) -> None:
         self.logger.info("[Completed]")
 
     def replay(self, file_name, index, sourcetype, source) -> None:
+        ### check if input log file is afile path or just a file name
+        ### if file name ,assume it is in current working dir of attack_range.py
+        if Path(file_name).parent == Path("."):
+            file_name = str(Path("../..")/file_name)
         ansible_vars = {}
         ansible_vars["file_name"] = file_name
         ansible_vars["ansible_user"] = "ubuntu"