VulnAgent-X Prototipo di Ricerca

VulnAgent-X è un prototipo di rilevamento di vulnerabilità/difetti multi-agente orientato alla riproduzione di paper di ricerca. Accetta in input un repository locale o un diff e produce in output findings strutturati, catena di prove, risultati di localizzazione, confidenza e log degli esperimenti.

Funzionalità principali

• Modalità di input: repo path o unified diff
• Flusso di lavoro: screening -> context expansion -> scheduler -> router -> experts -> sceptic -> verification(stub) -> evidence fusion
• Campi di output:
  • issue_type
  • location(file + line range)
  • evidence_summary
  • confidence
  • severity
  • optional_cwe
  • fix_hint
  • evidence_chain
  • counter_evidence
• Interfacce: CLI + FastAPI
• Esperimenti riproducibili: pytest / mypy / ruff + supporto all'esecuzione con Docker

Panoramica del flusso di lavoro

1. screening: screening rapido delle aree sospette (regole + segnali dai metadati)
2. context_expansion: recupero del "contesto minimo sufficiente" (estrazione di finestre basate sulle posizioni sospette)
3. scheduler: strategie di escalation basate su confidenza e rischio (early_exit / expert_review / verification)
4. router_agent: seleziona l'Agent esperto per ogni area sospetta
5. semantic/security/logic: forniscono affermazioni strutturate e prove da diverse prospettive
6. sceptic_agent: genera controprove e segnali di penalizzazione
7. verification: verifica dinamica opzionale (attualmente implementazione segnaposto sicura)
8. evidence_fusion: fusione unificata delle prove e output dei findings finali

Guida all'installazione e all'esecuzione

1) Preparazione dell'ambiente

Requisiti: Python 3.11+ (attualmente funziona anche su versioni più recenti)

cd /Users/xiaolu/Documents/Python_code/vulnAgentX
python3 -m venv .venv
source .venv/bin/activate
python -m pip install -e '.[dev]'

2) Utilizzo della CLI

Analizza un intero repository:

.venv/bin/vulnagentx analyze --repo /path/to/repo --output json

Analizza un file di diff:

.venv/bin/vulnagentx analyze --diff-file /path/to/patch.diff --output json

Output sintetico:

.venv/bin/vulnagentx analyze --repo /path/to/repo --output summary

3) Utilizzo dell'API

Avvia il servizio:

.venv/bin/uvicorn vulnagentx.app.api:app --reload

Health check:

curl http://127.0.0.1:8000/health

Avvia un'analisi:

curl -X POST http://127.0.0.1:8000/analyze \
  -H "Content-Type: application/json" \
  -d '{"repo_path":"/path/to/repo"}'

4) Esecuzione con Docker

docker compose -f docker/docker-compose.yml up --build

5) Controllo di qualità e test

.venv/bin/ruff check src tests
.venv/bin/mypy src
.venv/bin/pytest

Esempio di output

{
  "run_id": "...",
  "findings": [
    {
      "issue_type": "command_injection",
      "location": {"file_path": "app.py", "start_line": 42, "end_line": 42},
      "evidence_summary": "Command execution surface detected...",
      "confidence": 0.87,
      "severity": "critical",
      "optional_cwe": "CWE-78",
      "fix_hint": "Avoid shell command composition...",
      "source_agents": ["security_agent", "semantic_agent"],
      "evidence_chain": [],
      "counter_evidence": []
    }
  ],
  "metrics": {
    "runtime_seconds": 0.07
  },
  "logs": []
}

Descrizione dei file (file per file)

File nella directory radice e file di progetto

Codice sorgente principale (src/vulnagentx)

File di test (tests)

Moduli avanzati implementati in questa versione

• Adapter LLM reali: OpenAI + Ollama locale + commutazione automatica tramite factory
• Tree-sitter AST e grafo del codice reali: analisi delle relazioni tra funzioni/import/chiamate, con supporto all'espansione del contesto tramite file vicini
• Set di regole Semgrep reale: file di regole integrato e collegato alla fase di screening
• Catena di esecuzione in sandbox per verification: pianificazione dei compiti, esecuzione limitata, controllo del timeout e scrittura strutturata dei risultati
• Valutazione sui dataset e ablazione: caricatori di dati, moduli di metriche, script per esperimenti e ablazione