| from datetime import datetime |
| from collections import Counter |
| from typing import List |
|
|
| from threat_kb import get_threats_for_component |
| from detector import Detection |
|
|
| SEVERIDADE_ORDEM = {"Crítica": 0, "Alta": 1, "Média": 2, "Baixa": 3} |
|
|
|
|
| def _dedup_components(detections: List[Detection]) -> Counter: |
| """Conta ocorrências únicas de cada tipo de componente (ignora bounding boxes duplicadas).""" |
| return Counter(d.label for d in detections) |
|
|
|
|
| def build_markdown_report(detections: List[Detection], demo_mode: bool = False, |
| sistema_nome: str = "Sistema Analisado") -> str: |
| counts = _dedup_components(detections) |
| data_geracao = datetime.now().strftime("%d/%m/%Y %H:%M") |
|
|
| linhas = [] |
| linhas.append(f"# Relatório de Modelagem de Ameaças (STRIDE)") |
| linhas.append(f"**Sistema:** {sistema_nome} ") |
| linhas.append(f"**Data de geração:** {data_geracao} ") |
| linhas.append(f"**Gerado por:** pipeline de detecção de componentes + base de conhecimento STRIDE ") |
| if demo_mode: |
| linhas.append("\n> ⚠️ **Modo demonstração**: nenhum modelo treinado foi encontrado em " |
| "`models/best.pt`. Os componentes abaixo foram estimados por uma " |
| "heurística de cor/contorno apenas para fins de demonstração da interface. " |
| "Execute `train.py` para obter detecções reais do modelo supervisionado.\n") |
| linhas.append("\n---\n") |
|
|
| if not counts: |
| linhas.append("Nenhum componente foi identificado no diagrama enviado. " |
| "Verifique a qualidade/resolução da imagem e tente novamente.") |
| return "\n".join(linhas) |
|
|
| linhas.append("## 1. Componentes Identificados\n") |
| linhas.append("| Componente | Ocorrências |") |
| linhas.append("|---|---|") |
| for label, qtd in sorted(counts.items(), key=lambda x: -x[1]): |
| linhas.append(f"| {label.replace('_', ' ').title()} | {qtd} |") |
|
|
| linhas.append("\n## 2. Ameaças por Componente (STRIDE)\n") |
|
|
| total_ameacas = 0 |
| for label in counts: |
| threats = get_threats_for_component(label) |
| if not threats: |
| continue |
| linhas.append(f"### {label.replace('_', ' ').title()}\n") |
| threats_ordenadas = sorted(threats, key=lambda t: SEVERIDADE_ORDEM.get(t.severidade, 9)) |
| for t in threats_ordenadas: |
| total_ameacas += 1 |
| linhas.append(f"**[{t.categoria}] {t.nome_categoria} — Severidade: {t.severidade}**") |
| linhas.append(f"- *Ameaça:* {t.descricao}") |
| linhas.append(f"- *Contramedida recomendada:* {t.contramedida}\n") |
|
|
| linhas.append("## 3. Resumo Executivo\n") |
| n_criticas = sum(1 for label in counts for t in get_threats_for_component(label) if t.severidade == "Crítica") |
| n_altas = sum(1 for label in counts for t in get_threats_for_component(label) if t.severidade == "Alta") |
| linhas.append(f"- Componentes identificados: **{len(counts)}**") |
| linhas.append(f"- Ameaças mapeadas: **{total_ameacas}**") |
| linhas.append(f"- Ameaças de severidade **Crítica**: **{n_criticas}**") |
| linhas.append(f"- Ameaças de severidade **Alta**: **{n_altas}**") |
| linhas.append("\n> Este relatório é gerado automaticamente por um MVP do fluxo e deve ser " |
| "revisado por um especialista em segurança antes de decisões de mitigação.") |
|
|
| return "\n".join(linhas) |
|
|
|
|
| def save_pdf_report(markdown_text: str, output_path: str = "relatorio_stride.pdf"): |
| """Converte o relatório (texto simples, sem renderização rica de markdown) para PDF.""" |
| from fpdf import FPDF |
|
|
| pdf = FPDF() |
| pdf.set_auto_page_break(auto=True, margin=15) |
| pdf.add_page() |
| pdf.set_font("Helvetica", size=11) |
|
|
| for raw_line in markdown_text.split("\n"): |
| line = raw_line.replace("**", "").replace("### ", "").replace("## ", "").replace("# ", "") |
| line = line.replace("*Ameaça:*", "Ameaca:").replace("*Contramedida recomendada:*", "Contramedida:") |
| |
| |
| line = line.encode("latin-1", errors="ignore").decode("latin-1") |
| if line.strip() == "": |
| pdf.ln(4) |
| continue |
| pdf.set_x(pdf.l_margin) |
| pdf.multi_cell(0, 6, line) |
| pdf.output(output_path) |
| return output_path |
|
|
