diff --git a/monkey/monkey_island/cc/models/__init__.py b/monkey/monkey_island/cc/models/__init__.py index 65e63fe14..ca4078faa 100644 --- a/monkey/monkey_island/cc/models/__init__.py +++ b/monkey/monkey_island/cc/models/__init__.py @@ -15,3 +15,4 @@ from .communication_type import CommunicationType from .node import Node from common.types import AgentID from .agent import Agent +from .agent_signals import AgentSignals diff --git a/monkey/monkey_island/cc/models/agent_signals.py b/monkey/monkey_island/cc/models/agent_signals.py new file mode 100644 index 000000000..37af7b4c1 --- /dev/null +++ b/monkey/monkey_island/cc/models/agent_signals.py @@ -0,0 +1,8 @@ +from datetime import datetime +from typing import Optional + +from common.base_models import InfectionMonkeyBaseModel + + +class AgentSignals(InfectionMonkeyBaseModel): + terminate: Optional[datetime] diff --git a/vulture_allowlist.py b/vulture_allowlist.py index 43292e49e..742450c50 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, Simulation +from monkey_island.cc.models import AgentSignals, 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