Common: Add PropagationEvent to agent_events

This commit is contained in:
Ilija Lazoroski 2022-10-03 13:51:39 +02:00
parent 25f12305f5
commit 3389915399
3 changed files with 27 additions and 4 deletions

View File

@ -2,3 +2,4 @@ from .abstract_agent_event import AbstractAgentEvent
from .credentials_stolen_events import CredentialsStolenEvent from .credentials_stolen_events import CredentialsStolenEvent
from .ping_scan_event import PingScanEvent from .ping_scan_event import PingScanEvent
from .tcp_scan_event import TCPScanEvent from .tcp_scan_event import TCPScanEvent
from .propagation_event import PropagationEvent

View File

@ -0,0 +1,22 @@
from ipaddress import IPv4Address
from pydantic import Field
from . import AbstractAgentEvent
class PropagationEvent(AbstractAgentEvent):
"""
An event that occurs when the Agent propagates on a host
Attributes:
:param target: IP address of the propagated system
:param success: Status of the propagation
:param exploiter_name: Name of the exploiter that propagated
:param error_message: Message if an error occurs during propagation
"""
target: IPv4Address
success: bool
exploiter_name: str
error_message: str = Field(default="")

View File

@ -7,7 +7,7 @@ from common.agent_configuration.agent_sub_configurations import (
CustomPBAConfiguration, CustomPBAConfiguration,
ScanTargetConfiguration, ScanTargetConfiguration,
) )
from common.agent_events import PingScanEvent, TCPScanEvent from common.agent_events import PingScanEvent, PropagationEvent, TCPScanEvent
from common.credentials import Credentials, LMHash, NTHash from common.credentials import Credentials, LMHash, NTHash
from common.types import NetworkPort from common.types import NetworkPort
from infection_monkey.exploit.log4shell_utils.ldap_server import LDAPServerFactory from infection_monkey.exploit.log4shell_utils.ldap_server import LDAPServerFactory
@ -311,13 +311,13 @@ IAgentLogRepository
IAgentLogRepository.upsert_agent_log IAgentLogRepository.upsert_agent_log
IAgentLogRepository.get_agent_log IAgentLogRepository.get_agent_log
# TODO: Remove once #2268 is closed
PingScanEvent
# TODO: Remove once #2267 is closed # TODO: Remove once #2267 is closed
TCPScanEvent TCPScanEvent
TCPScanEvent.port_status TCPScanEvent.port_status
# TODO: Remove once #2269 is close
PropagationEvent
# pydantic base models # pydantic base models
underscore_attrs_are_private underscore_attrs_are_private
extra extra