forked from p15670423/monkey
Island: Patch T1003 with a callable class
This commit is contained in:
parent
06a64c14d7
commit
885f0565a8
|
@ -41,6 +41,7 @@ from monkey_island.cc.services.telemetry.processing.processing import (
|
|||
TELEMETRY_CATEGORY_TO_PROCESSING_FUNC,
|
||||
)
|
||||
from monkey_island.cc.setup.mongo.mongo_setup import MONGO_URL
|
||||
from monkey_island.services.atack.technique_reports.T1003 import T1003, T1003GetReportData
|
||||
|
||||
from . import AuthenticationService
|
||||
from .reporting.report import ReportService
|
||||
|
@ -64,6 +65,8 @@ def initialize_services(data_dir: Path) -> DIContainer:
|
|||
_register_repositories(container, data_dir)
|
||||
_register_services(container)
|
||||
|
||||
_dirty_hacks(container)
|
||||
|
||||
# Note: A hack to resolve credentials parser
|
||||
# It changes telemetry processing function, this will be refactored!
|
||||
_patch_credentials_parser(container)
|
||||
|
@ -157,3 +160,12 @@ def _patch_credentials_parser(container: DIContainer):
|
|||
TELEMETRY_CATEGORY_TO_PROCESSING_FUNC[TelemCategoryEnum.CREDENTIALS] = container.resolve(
|
||||
CredentialsParser
|
||||
)
|
||||
|
||||
|
||||
def _dirty_hacks(container: DIContainer):
|
||||
# A dirty hacks function that patches some of the things that
|
||||
# are needed at the current point
|
||||
|
||||
# Patches attack technique T1003 which is a static class
|
||||
# but it needs stolen credentials from the database
|
||||
T1003.get_report_data = container.resolve(T1003GetReportData)
|
||||
|
|
Loading…
Reference in New Issue