p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Use ScanEvent for typehint in ScanEventHandler 

`event` gets passed to `_get_source_machine()`, which expects
`ScanEvent`, not `AbstractAgentEvent`
This commit is contained in:
Mike Salvatore 2022-10-07 07:43:10 -04:00
parent 4c026241ea
commit 0965b97d45
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/monkey_island/cc/agent_event_handlers/scan_event_handler.py
View File

@ -4,7 +4,7 @@ from typing import Union
from typing_extensions import TypeAlias 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 common.types import PortStatus, SocketAddress
from monkey_island.cc.models import CommunicationType, Machine, Node from monkey_island.cc.models import CommunicationType, Machine, Node
from monkey_island.cc.repository import ( from monkey_island.cc.repository import (
@ -75,7 +75,7 @@ class ScanEventHandler:
self._machine_repository.upsert_machine(machine) self._machine_repository.upsert_machine(machine)
return 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) machine = self._get_source_machine(event)
return self._node_repository.get_node_by_machine_id(machine.id) return self._node_repository.get_node_by_machine_id(machine.id)