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 :

   root@kitploit:~

   # 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 :

   root@kitploit:~

   pip install psutil
   

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

   root@kitploit:~

   # 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 :

   root@kitploit:~

   # 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 :

   root@kitploit:~

   # 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 :
     root@kitploit:~

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

2. Démarrer le service Ollama :

   root@kitploit:~

   ollama serve
   

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

   root@kitploit:~

   # 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 :

   root@kitploit:~

   # 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

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

Accessible via Edit → VulChat → Control Panel ou Ctrl+Alt+P :

• Changer de fournisseur d'IA : basculer entre OpenAI, Gemini et Ollama
• Sélection du modèle : choisir des modèles spécifiques pour chaque fournisseur
• Informations d'état : afficher la configuration actuelle et la disponibilité des fournisseurs
• Guide de configuration : obtenir des instructions d'installation détaillées

Raccourcis clavier

Mode sans interface (headless)

Pour l'analyse automatisée et l'intégration CI/CD :

# Scan all functions for vulnerabilities
ida64.exe -A -S"BinAIVulHunter.py --scan-all --output results.json" binary.exe

# Decompile all functions with caching
ida64.exe -A -S"BinAIVulHunter.py --decompile-all --cache-dir ./cache" binary.exe

# Custom batch settings
ida64.exe -A -S"BinAIVulHunter.py --scan-all --batch-size 5 --function-pause 3 --batch-pause 15" binary.exe

Paramètres du mode headless

🛡️ Catégories de vulnérabilités

BinAIVulHunter analyse le code selon la taxonomie CWE-699 du développement logiciel :

📚 Exemples

Exemple 1 : Analyse de vulnérabilité de base

# Navigate to a function in IDA Pro
# Press Ctrl+Alt+V or right-click → "Find Possible Vulnerability"

# Example output:
"""
POTENTIAL VULNERABILITIES:

1. Buffer overflow in strcpy operation
   CWE: CWE-120 | Severity: High
   Mitigation: Replace strcpy with strncpy and validate buffer sizes

2. Integer overflow in arithmetic operation
   CWE: CWE-190 | Severity: Medium  
   Mitigation: Add overflow checks before arithmetic operations
"""

Exemple 2 : Analyse par lots

# Command line batch scan
ida64.exe -A -S"BinAIVulHunter.py --scan-all --output vuln_report.json" malware.exe

# Results in vuln_report.json:
{
  "scan_results": {
    "total_functions": 245,
    "vulnerable_functions": 12,
    "functions": [
      {
        "name": "sub_401000",
        "address": "0x401000", 
        "vulnerabilities": [
          {
            "description": "Unsafe string function used",
            "cwe_id": "CWE-120",
            "severity": "High",
            "mitigation": "Use safe string functions"
          }
        ]
      }
    ]
  }
}

Exemple 3 : Changement de fournisseur d'IA

# Set environment variables
os.environ['VULCHAT_PROVIDER'] = 'ollama'
os.environ['OLLAMA_MODEL'] = 'codellama'

# Or use Control Panel in IDA Pro:
# Edit → VulChat → Control Panel → Provider Menu → Ollama

🔧 Dépannage

Problèmes courants

1. Erreur « Fournisseur non disponible » :

   root@kitploit:~

   # Check environment variables
   echo %OPENAI_API_KEY%
   echo %GEMINI_API_KEY%
   
   # Verify API keys are valid
   # Restart IDA Pro after setting variables
   

2. Échec de connexion à Ollama :

   root@kitploit:~

   # Check if Ollama is running
   curl http://localhost:11434/api/tags
   
   # Start Ollama if needed
   ollama serve
   
   # Verify models are installed
   ollama list
   

3. Problèmes de mémoire lors du traitement par lots :

   root@kitploit:~

   # Reduce batch size in Control Panel or headless mode
   --batch-size 1 --function-pause 10
   

4. Limites de débit de l'API :

   root@kitploit:~

   # Increase pause times
   --function-pause 10 --batch-pause 30
   
   # Or switch to Ollama for unlimited local processing
   

Informations de débogage

Consultez la fenêtre de sortie d'IDA Pro pour obtenir des journaux détaillés :

• État de l'initialisation du fournisseur
• Informations sur les requêtes/réponses API
• Statistiques d'utilisation de la mémoire
• Messages d'erreur et traces de pile

🏆 Modèles d'IA recommandés

Pour l'analyse de sécurité

Comparaison des performances

🤝 Contribution

Nous accueillons volontiers les contributions ! Veuillez consulter nos Directives de contribution pour plus de détails.

Configuration de développement

1. Forkez le dépôt
2. Clonez votre fork :
   root@kitploit:~

   git clone https://github.com/yourusername/BinAIVulHunter.git
   

3. Installez les dépendances de développement :
   root@kitploit:~

   pip install -e .[dev]
   

4. Effectuez vos modifications
5. Soumettez une demande d'extraction (pull request)

Domaines de contribution

• Intégrations de fournisseurs d'IA supplémentaires
• Modèles de détection de vulnérabilités améliorés
• Prise en charge de nouvelles catégories CWE
• Optimisations des performances
• Améliorations de la documentation
• Élargissement de la couverture des tests

📄 Licence

Ce projet est sous licence MIT - voir le fichier LICENSE pour plus de détails.

🔗 Liens

• IDA Pro : https://www.hex-rays.com/products/ida/
• OpenAI API : https://platform.openai.com/
• Google Gemini : https://aistudio.google.com/
• Ollama : https://ollama.ai/
• Base de données CWE : https://cwe.mitre.org/

🙏 Remerciements

• Hex-Rays pour l'excellente plateforme IDA Pro
• Les équipes d'OpenAI, Google et Ollama pour leurs technologies d'IA
• MITRE pour la taxonomie complète des CWE
• La communauté de la rétro-ingénierie et de la cybersécurité

⚠️ Avertissement : Cet outil est destiné uniquement à la recherche éducative et légitime en matière de sécurité. Assurez-vous toujours d'avoir l'autorisation appropriée avant d'analyser un logiciel. L'analyse générée par l'IA doit être vérifiée manuellement et ne remplace pas l'examen par un expert humain.