p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Island: Use AgentConfiguration to get inaccessible subnets in ReportService

This commit is contained in:
Shreya Malviya 2022-07-14 13:39:33 +05:30
parent 57b97d09f4
commit b237d258e5
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
monkey/monkey_island/cc/services/reporting/report.py
View File

@ -11,9 +11,6 @@ from monkey_island.cc.database import mongo
from monkey_island.cc.models import Monkey from monkey_island.cc.models import Monkey
from monkey_island.cc.models.report import get_report, save_report from monkey_island.cc.models.report import get_report, save_report
from monkey_island.cc.repository import IAgentConfigurationRepository, ICredentialsRepository from monkey_island.cc.repository import IAgentConfigurationRepository, ICredentialsRepository
from monkey_island.cc.services.configuration.utils import (
get_config_network_segments_as_subnet_groups,
)
from monkey_island.cc.services.node import NodeService from monkey_island.cc.services.node import NodeService
from monkey_island.cc.services.reporting.exploitations.manual_exploitation import get_manual_monkeys from monkey_island.cc.services.reporting.exploitations.manual_exploitation import get_manual_monkeys
from monkey_island.cc.services.reporting.exploitations.monkey_exploitation import ( from monkey_island.cc.services.reporting.exploitations.monkey_exploitation import (
@ -325,8 +322,8 @@ class ReportService:
return cross_segment_issues return cross_segment_issues
@staticmethod @classmethod
def get_cross_segment_issues(): def get_cross_segment_issues(cls):
scans = mongo.db.telemetry.find( scans = mongo.db.telemetry.find(
{"telem_category": "scan"}, {"telem_category": "scan"},
{ {
@ -340,7 +337,8 @@ class ReportService:
cross_segment_issues = [] cross_segment_issues = []
# For now the feature is limited to 1 group. # For now the feature is limited to 1 group.
subnet_groups = get_config_network_segments_as_subnet_groups() agent_configuration = cls._agent_configuration_repository.get_configuration()
subnet_groups = agent_configuration.propagation.network_scan.targets.inaccessible_subnets
for subnet_group in subnet_groups: for subnet_group in subnet_groups:
cross_segment_issues += ReportService.get_cross_segment_issues_per_subnet_group( cross_segment_issues += ReportService.get_cross_segment_issues_per_subnet_group(