Island: Add AgentSignals model

This commit is contained in:
Ilija Lazoroski 2022-09-22 12:02:47 +02:00 committed by Mike Salvatore
parent 560d941885
commit 721cc29559
3 changed files with 10 additions and 1 deletions

View File

@ -15,3 +15,4 @@ from .communication_type import CommunicationType
from .node import Node from .node import Node
from common.types import AgentID from common.types import AgentID
from .agent import Agent from .agent import Agent
from .agent_signals import AgentSignals

View File

@ -0,0 +1,8 @@
from datetime import datetime
from typing import Optional
from common.base_models import InfectionMonkeyBaseModel
class AgentSignals(InfectionMonkeyBaseModel):
terminate: Optional[datetime]

View File

@ -11,7 +11,7 @@ from common.credentials import Credentials, LMHash, NTHash
from common.types import SocketAddress from common.types import SocketAddress
from infection_monkey.exploit.log4shell_utils.ldap_server import LDAPServerFactory from infection_monkey.exploit.log4shell_utils.ldap_server import LDAPServerFactory
from monkey_island.cc.event_queue import IslandEventTopic, PyPubSubIslandEventQueue 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.models.networkmap import Arc, NetworkMap
from monkey_island.cc.repository import MongoAgentRepository, MongoMachineRepository from monkey_island.cc.repository import MongoAgentRepository, MongoMachineRepository
from monkey_island.cc.repository.attack.IMitigationsRepository import IMitigationsRepository from monkey_island.cc.repository.attack.IMitigationsRepository import IMitigationsRepository