diff --git a/monkey/monkey_island/cc/models/simulation.py b/monkey/monkey_island/cc/models/simulation.py index d04bee76b..3f52ae370 100644 --- a/monkey/monkey_island/cc/models/simulation.py +++ b/monkey/monkey_island/cc/models/simulation.py @@ -1,6 +1,8 @@ from __future__ import annotations +from datetime import datetime from enum import Enum +from typing import Optional from common.base_models import InfectionMonkeyBaseModel @@ -13,3 +15,4 @@ class IslandMode(Enum): class Simulation(InfectionMonkeyBaseModel): mode: IslandMode = IslandMode.UNSET + terminate_signal_time: Optional[datetime] = None diff --git a/vulture_allowlist.py b/vulture_allowlist.py index ef9613fca..43292e49e 100644 --- a/vulture_allowlist.py +++ b/vulture_allowlist.py @@ -11,7 +11,7 @@ from common.credentials import Credentials, LMHash, NTHash from common.types import SocketAddress from infection_monkey.exploit.log4shell_utils.ldap_server import LDAPServerFactory from monkey_island.cc.event_queue import IslandEventTopic, PyPubSubIslandEventQueue -from monkey_island.cc.models import Report +from monkey_island.cc.models import Report, Simulation from monkey_island.cc.models.networkmap import Arc, NetworkMap from monkey_island.cc.repository import MongoAgentRepository, MongoMachineRepository from monkey_island.cc.repository.attack.IMitigationsRepository import IMitigationsRepository @@ -328,3 +328,5 @@ CC_TUNNEL # TODO: Remove after #2323 SocketAddress + +Simulation.terminate_signal_time