Island: Handle StorageError in handle_scan_data

This commit is contained in:
Kekoa Kaaikala 2022-09-27 18:42:06 +00:00
parent 304dfbd21f
commit eacd426969
1 changed files with 2 additions and 1 deletions

View File

@ -8,6 +8,7 @@ from monkey_island.cc.repository import (
IMachineRepository, IMachineRepository,
INodeRepository, INodeRepository,
RetrievalError, RetrievalError,
StorageError,
UnknownRecordError, UnknownRecordError,
) )
@ -52,5 +53,5 @@ class handle_scan_data:
self._node_repository.upsert_communication( self._node_repository.upsert_communication(
src_machine.id, dest_machine.id, CommunicationType.SCANNED src_machine.id, dest_machine.id, CommunicationType.SCANNED
) )
except (RetrievalError, UnknownRecordError) as err: except (RetrievalError, StorageError, UnknownRecordError) as err:
logger.error(f"Unable to process scan data: {err}") logger.error(f"Unable to process scan data: {err}")