From 0965b97d455d415fcf202e32ee4a07684896be61 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Fri, 7 Oct 2022 07:43:10 -0400 Subject: [PATCH] Island: Use ScanEvent for typehint in ScanEventHandler `event` gets passed to `_get_source_machine()`, which expects `ScanEvent`, not `AbstractAgentEvent` --- .../cc/agent_event_handlers/scan_event_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/monkey_island/cc/agent_event_handlers/scan_event_handler.py b/monkey/monkey_island/cc/agent_event_handlers/scan_event_handler.py index 3a345e012..16acca23b 100644 --- a/monkey/monkey_island/cc/agent_event_handlers/scan_event_handler.py +++ b/monkey/monkey_island/cc/agent_event_handlers/scan_event_handler.py @@ -4,7 +4,7 @@ from typing import Union from typing_extensions import TypeAlias -from common.agent_events import AbstractAgentEvent, PingScanEvent, TCPScanEvent +from common.agent_events import PingScanEvent, TCPScanEvent from common.types import PortStatus, SocketAddress from monkey_island.cc.models import CommunicationType, Machine, Node from monkey_island.cc.repository import ( @@ -75,7 +75,7 @@ class ScanEventHandler: self._machine_repository.upsert_machine(machine) return machine - def _get_source_node(self, event: AbstractAgentEvent) -> Node: + def _get_source_node(self, event: ScanEvent) -> Node: machine = self._get_source_machine(event) return self._node_repository.get_node_by_machine_id(machine.id)