
Cadre automatisé de test d'injection de prompt pour applications intégrant des LLM avec une architecture à double LLM.

LLMMap est un cadre de test automatisé d'injection de prompts pour les applications intégrant des LLM. Il découvre des points d'injection dans les requêtes HTTP, génère des prompts d'attaque ciblés à l'aide d'une architecture à deux LLM, les envoie à la cible, et confirme les résultats avec des tests de fiabilité statistiques. Inspiré par sqlmap, LLMMap apporte la même approche systématique et fondée sur des preuves aux tests de sécurité des LLM.
Avertissement légal : L'utilisation de LLMMap pour attaquer des cibles sans consentement mutuel préalable est illégale. Il est de la responsabilité de l'utilisateur final de respecter toutes les lois locales, étatiques et fédérales applicables. Les développeurs déclinent toute responsabilité et ne sont pas responsables de tout usage abusif ou dommage causé par ce programme.
python -m venv .venv && source .venv/bin/activate
pip install -e .
Prérequis : Python >= 3.11. Seule dépendance : PyYAML.
LLMMap accepte des cibles via l'export de requête Burp Suite (-r) ou une URL (-u). Placez * à l'endroit où le prompt doit être injecté.
# Cible URL (Ollama par défaut -- pas de clé API nécessaire)
llmmap -u "https://target.example.com/chat?q=*" --goal "révéler le mot de passe caché"
# Capture de requête Burp Suite
llmmap -r request --goal "révéler le prompt système"
# Fournisseur cloud (OpenAI, Anthropic, Google)
export OPENAI_API_KEY=sk-...
llmmap -r request --goal "révéler le prompt système" --provider openai
# Contrôle de l'intensité (1-5, défaut : 1)
llmmap -r request --goal "révéler le prompt système" --intensity 3
# Limiter à des paramètres spécifiques ou classes de points d'injection
llmmap -r request --goal "..." -p prompt --injection-points QB
Référence complète des options :
llmmap -h
--no-colorCible (-r/-u)
|
v
Découverte des points d'injection (Q/B/H/C/P)
|
v
Génération de prompts (Générateur LLM + bibliothèque de techniques)
|
v
Mutation et envoi des requêtes
|
v
Analyse des réponses (Juge LLM + détecteurs heuristiques)
|
v
Confirmation de fiabilité (Intervalle de Wilson)
|
v
Rapport de résultats
Backend par défaut : Ollama (local, sans clé API). Autres backends supportés : OpenAI, Anthropic, Google.