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

Island: Build MongoCredentialsRepository with a mongo connection

This commit is contained in:
Ilija Lazoroski 2022-07-11 12:41:23 +02:00
parent 1b1bd7fcc2
commit efb12df483
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
monkey/monkey_island/cc/services/initialize.py
View File

@ -1,6 +1,8 @@
import logging import logging
from pathlib import Path from pathlib import Path
from pymongo import MongoClient
from common import DIContainer from common import DIContainer
from common.aws import AWSInstance from common.aws import AWSInstance
from common.configuration import ( from common.configuration import (
@ -29,6 +31,7 @@ from monkey_island.cc.server_utils.consts import MONKEY_ISLAND_ABS_PATH
from monkey_island.cc.services import AWSService, IslandModeService from monkey_island.cc.services import AWSService, IslandModeService
from monkey_island.cc.services.post_breach_files import PostBreachFilesService from monkey_island.cc.services.post_breach_files import PostBreachFilesService
from monkey_island.cc.services.run_local_monkey import LocalMonkeyRunService from monkey_island.cc.services.run_local_monkey import LocalMonkeyRunService
from monkey_island.cc.setup.mongo.mongo_setup import MONGO_URL
from . import AuthenticationService, JsonFileUserDatastore from . import AuthenticationService, JsonFileUserDatastore
from .reporting.report import ReportService from .reporting.report import ReportService
@ -75,9 +78,7 @@ def _register_repositories(container: DIContainer, data_dir: Path):
IAgentConfigurationRepository, container.resolve(FileAgentConfigurationRepository) IAgentConfigurationRepository, container.resolve(FileAgentConfigurationRepository)
) )
container.register_instance(ISimulationRepository, container.resolve(FileSimulationRepository)) container.register_instance(ISimulationRepository, container.resolve(FileSimulationRepository))
container.register_instance( container.register_instance(ICredentialsRepository, _build_mongo_credentials_repository())
ICredentialsRepository, container.resolve(MongoCredentialsRepository)
)
def _decorate_file_repository(file_repository: IFileRepository) -> IFileRepository: def _decorate_file_repository(file_repository: IFileRepository) -> IFileRepository:
@ -95,6 +96,13 @@ def _build_agent_binary_repository():
return agent_binary_repository return agent_binary_repository
def _build_mongo_credentials_repository():
mongo = MongoClient(MONGO_URL, serverSelectionTimeoutMS=100)
mongo_credentials_repository = MongoCredentialsRepository(mongo)
return mongo_credentials_repository
def _log_agent_binary_hashes(agent_binary_repository: IAgentBinaryRepository): def _log_agent_binary_hashes(agent_binary_repository: IAgentBinaryRepository):
""" """
Logs all the hashes of the agent executables for debbuging ease Logs all the hashes of the agent executables for debbuging ease