p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Use NodeUpdateFacade in ScanEventHandler

This commit is contained in:
Mike Salvatore 2022-10-07 09:51:44 -04:00
parent e77932f7d6
commit e876682d84
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,7 +15,7 @@ from monkey_island.cc.repository import (
UnknownRecordError, UnknownRecordError,
) )
from .utils import get_or_create_target_machine from .node_update_facade import NodeUpdateFacade
ScanEvent: TypeAlias = Union[PingScanEvent, TCPScanEvent] ScanEvent: TypeAlias = Union[PingScanEvent, TCPScanEvent]
@ -33,6 +33,7 @@ class ScanEventHandler:
machine_repository: IMachineRepository, machine_repository: IMachineRepository,
node_repository: INodeRepository, node_repository: INodeRepository,
): ):
self._node_update_facade = NodeUpdateFacade(machine_repository)
self._agent_repository = agent_repository self._agent_repository = agent_repository
self._machine_repository = machine_repository self._machine_repository = machine_repository
self._node_repository = node_repository self._node_repository = node_repository
@ -66,7 +67,7 @@ class ScanEventHandler:
logger.exception("Unable to process tcp scan data") logger.exception("Unable to process tcp scan data")
def _get_target_machine(self, event: ScanEvent) -> Machine: def _get_target_machine(self, event: ScanEvent) -> Machine:
return get_or_create_target_machine(self._machine_repository, event.target) return self._node_update_facade.get_or_create_target_machine(event.target)
def _get_source_node(self, event: ScanEvent) -> Node: def _get_source_node(self, event: ScanEvent) -> Node:
machine = self._get_source_machine(event) machine = self._get_source_machine(event)