BinAIVulHunter

Un plugin IDA Pro pour aider à trouver des vulnérabilités dans les binaires.

Utilisez le décompilateur HexRays d'IDA Pro avec OpenAI(ChatGPT) pour trouver d'éventuelles vulnérabilités dans les binaires.

Avertissement : les réponses obtenues en cherchant des vulnérabilités dans les binaires à l'aide d'une IA peuvent contenir des faux positifs. Cependant, cela a fonctionné dans de nombreux CTF auxquels j'ai participé (simple/moyen... certains difficiles).

Inspiré par Gepetto : https://github.com/JusticeRage/Gepetto

Installation :

Placez le script Python dans le répertoire des plugins d'IDA Pro.

Utilisation :

Faites un clic droit sur le code décompilé, sélectionnez « Rechercher une vulnérabilité possible dans la fonction »

Mis à jour avec la création d'un exemple d'exploit Python (Parfois bon, parfois mauvais)

Mis à jour avec les fonctions de renommage et d'explication de Gepetto

BinAIVulHunter - Plugin d'analyse de vulnérabilités propulsé par l'IA pour IDA Pro

BinAIVulHunter est un plugin avancé pour IDA Pro qui exploite plusieurs fournisseurs d'IA (OpenAI GPT, Google Gemini et Ollama) pour effectuer une analyse intelligente des vulnérabilités, une explication du code et une évaluation de la sécurité des fonctions décompilées.

🚀 Fonctionnalités

• Prise en charge multi-fournisseurs d'IA : choisissez entre OpenAI GPT, Google Gemini ou Ollama
• Détection de vulnérabilités : analyse automatique basée sur la taxonomie CWE (Common Weakness Enumeration)
• Explication du code : analyse et explication des fonctions décompilées par l'IA
• Renommage des variables : suggestions intelligentes pour de meilleurs noms de variables et de fonctions
• Traitement par lots : analyse de toutes les fonctions d'un binaire avec gestion de la mémoire
• Intégration des références CWE : recherche CWE intégrée et catégorisation détaillée des vulnérabilités
• Mode sans interface (headless) : prise en charge en ligne de commande pour l'analyse automatisée
• Prise en charge de l'IA locale : analyse respectueuse de la vie privée avec Ollama (sans dépendance au cloud)

📋 Table des matières

• Installation
• Configuration des fournisseurs d'IA
  • OpenAI
  • Google Gemini
  • Ollama (IA locale)
• Configuration
• Utilisation
  • Mode interactif
  • Menu contextuel
  • Panneau de contrôle
  • Mode sans interface (headless)
• Catégories de vulnérabilités
• Exemples
• Dépannage
• Contribution
• Licence

🛠 Installation

1. Télécharger le plugin :

   # Save BinAIVulHunter.py to your IDA Pro plugins directory
   # Typically: C:\Program Files\IDA Pro X.X\plugins\
   

2. Installer les dépendances de base :

   pip install psutil
   

3. Installer les bibliothèques des fournisseurs d'IA (choisissez-en une ou plusieurs) :

   # For OpenAI
   pip install openai
   
   # For Google Gemini
   pip install google-generativeai
   
   # For Ollama (local AI)
   pip install requests
   

4. Redémarrer IDA Pro pour charger le plugin

🤖 Configuration des fournisseurs d'IA

OpenAI

1. Obtenir une clé API :

   • Visitez OpenAI API Keys
   • Créez une nouvelle clé API

2. Définir la variable d'environnement :

   # Windows Command Prompt
   set OPENAI_API_KEY=sk-your-openai-key-here
   
   # PowerShell
   $env:OPENAI_API_KEY="sk-your-openai-key-here"
   
   # Linux/Mac
   export OPENAI_API_KEY="sk-your-openai-key-here"
   

3. Redémarrer IDA Pro

Google Gemini

1. Obtenir une clé API :

   • Visitez Google AI Studio
   • Créez une nouvelle clé API

2. Définir la variable d'environnement :

   # Windows Command Prompt
   set GEMINI_API_KEY=your-gemini-key-here
   
   # PowerShell
   $env:GEMINI_API_KEY="your-gemini-key-here"
   
   # Linux/Mac
   export GEMINI_API_KEY="your-gemini-key-here"
   

3. Redémarrer IDA Pro

Ollama (IA locale)

1. Installer Ollama :

   • Visitez Ollama.ai et téléchargez le programme d'installation
   • Ou utilisez les gestionnaires de paquets :

     # Linux
     curl -fsSL https://ollama.ai/install.sh | sh
     
     # macOS
     brew install ollama
     

2. Démarrer le service Ollama :

   ollama serve
   

3. Télécharger les modèles d'IA :

   # For general code analysis
   ollama pull llama2
   
   # For coding tasks (recommended)
   ollama pull codellama
   
   # Other specialized models
   ollama pull mistral
   ollama pull deepseek-coder
   

4. Variables d'environnement facultatives :

   # Custom Ollama server (if not localhost:11434)
   set OLLAMA_BASE_URL=http://your-server:11434
   
   # Set default model
   set OLLAMA_MODEL=codellama
   

⚙️ Configuration

Variables d'environnement

Variable	Description	Défaut	Requis
OPENAI_API_KEY	Clé API OpenAI	Aucune	Pour OpenAI
GEMINI_API_KEY	Clé API Google Gemini	Aucune	Pour Gemini
OLLAMA_BASE_URL	URL du serveur Ollama	http://localhost:11434	Non
OLLAMA_MODEL	Modèle Ollama par défaut	llama2	Non
VULCHAT_PROVIDER	Fournisseur d'IA à utiliser	openai	Non
VULCHAT_MODEL	Nom du modèle spécifique	Défaut du fournisseur	Non
OPENAI_BASE_URL	Point de terminaison OpenAI personnalisé	API officielle	Non

Sélection du fournisseur

Définissez votre fournisseur d'IA préféré :

# Use OpenAI (default)
set VULCHAT_PROVIDER=openai

# Use Google Gemini
set VULCHAT_PROVIDER=gemini

# Use Ollama (local)
set VULCHAT_PROVIDER=ollama

🎯 Utilisation

Mode interactif

1. Ouvrez IDA Pro avec votre fichier binaire
2. Accédez à la vue du décompilateur (F5)
3. Accédez à VulChat via :
   • Menu : Edit → VulChat
   • Menu contextuel : clic droit dans la vue du décompilateur
   • Raccourcis clavier (voir ci-dessous)

Menu contextuel

Faites un clic droit dans la vue du décompilateur pour accéder à :

• Rechercher une vulnérabilité possible - Analyser la fonction actuelle pour détecter des problèmes de sécurité
• Expliquer - Obtenir une explication par l'IA du comportement de la fonction
• Renommer les variables - Obtenir des suggestions intelligentes de noms de variables
• Générer des entrées de test sûres - Créer des cas de test pour la fonction
• Tout analyser - Analyse par lots des vulnérabilités de l'ensemble du binaire
• Informations CWE - Consulter les détails des vulnérabilités CWE
• Panneau de contrôle - Changer de fournisseur d'IA et configurer les paramètres

Panneau de contrôle