import json from typing import Any, Dict, List from contrib.descriptions import VulnDescriptionProvider from contrib.internal_types import ScanResult from contrib.report_builders import ReportBuilder class JsonReportBuilder(ReportBuilder): def __init__(self, description_provider: VulnDescriptionProvider): self.description_provider = description_provider self._buffer = {'ips': [], 'vulnerable': {}, 'not_vulnerable': {}} def init_report(self, start_date: str, nmap_command: str): self._buffer['start_date'] = start_date self._buffer['nmap_command'] = nmap_command def build(self) -> Any: return json.dumps(self._buffer) def add_vulnerable_services(self, scan_results: Dict[str, ScanResult]): for app_name, result in scan_results.items(): self._buffer['vulnerable'][app_name] = { 'vulnerabilities': [], 'locations': self._serialize_locations(result.locations) } for v in result.vulns: data = v.to_dict() description = self.description_provider.get_description(v.name, v.vuln_type) data['description'], data['url'] = description.text, description.url self._buffer['vulnerable'][app_name]['vulnerabilities'].append(data) def add_non_vulnerable_services(self, scan_results: Dict[str, ScanResult]): for app_name, result in scan_results.items(): self._buffer['not_vulnerable'][app_name] = { 'locations': self._serialize_locations(result.locations) } def add_ip_address(self, ip: str): self._buffer['ips'].append(ip) @staticmethod def _serialize_locations(locations: Dict[str, List[str]]): return {loc: [int(port) for port in ports] for loc, ports in locations.items()}