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

#2725: add fix duration to application and service configuration

Browse Source
This commit is contained in:
Czar Echavez
2024-07-02 15:52:18 +01:00
parent eee7a68f28
commit 6a72f6af42
5 changed files with 616 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/primaite/game/game.py
View File

@@ -297,6 +297,11 @@ class PrimaiteGame:
new_node.software_manager.install(SERVICE_TYPES_MAPPING[service_type])
new_service = new_node.software_manager.software[service_type]
# fixing duration for the service
fix_duration = service_cfg.get("options", {}).get("fix_duration", None)
if fix_duration:
new_service.fixing_duration = fix_duration
# start the service
new_service.start()
else:
@@ -336,6 +341,11 @@ class PrimaiteGame:
if application_type in APPLICATION_TYPES_MAPPING:
new_node.software_manager.install(APPLICATION_TYPES_MAPPING[application_type])
new_application = new_node.software_manager.software[application_type]
# fixing duration for the application
fix_duration = application_cfg.get("options", {}).get("fix_duration", None)
if fix_duration:
new_application.fixing_duration = fix_duration
else:
msg = f"Configuration contains an invalid application type: {application_type}"
_LOGGER.error(msg)
@@ -358,7 +368,7 @@ class PrimaiteGame:
if "options" in application_cfg:
opt = application_cfg["options"]
new_application.configure(
server_ip_address=IPv4Address(opt.get("server_ip")),
server_ip_address=IPv4Address(opt.get("server_ip")) if opt.get("server_ip") else None,
server_password=opt.get("server_password"),
payload=opt.get("payload", "ENCRYPT"),
)

2
src/primaite/simulator/system/applications/red_applications/ransomware_script.py
View File

@@ -88,7 +88,7 @@ class RansomwareScript(Application):
def configure(
self,
server_ip_address: IPv4Address,
server_ip_address: Optional[IPv4Address] = None,
server_password: Optional[str] = None,
payload: Optional[str] = None,
):