initial commit

Author: anisiosts

README.md

@@ -0,0 +1,35 @@
+# <img src="./images/logo.png" width=20> RevEng.AI Binary Ninja Plugin
+
+A Binary Ninja plugin for integrating with the RevEng.AI platform for binary analysis and function renaming.
+
+## Features
+
+- Upload binaries for analysis to RevEng.AI platform
+- Download and check analysis status
+- Rename functions based on similar functions found in other binaries
+- Batch analyze entire binaries for function renaming
+- Configuration management for API settings
+
+## Installation
+
+1. Ensure you have Python 3.9 or later installed
+2. Install the required dependencies:
+   ```bash
+   pip install -r requirements.txt
+   ```
+3. Copy the `revengai_bn` directory to your Binary Ninja plugins directory:
+   - Linux: `~/.binaryninja/plugins/`
+   - Windows: `%APPDATA%\Binary Ninja\plugins\`
+   - macOS: `~/Library/Application Support/Binary Ninja/plugins/`
+
+
+## Requirements
+
+- Binary Ninja 5.0 or later
+- Python 3.9 or later
+- Internet connection for API access
+- RevEng.AI API key
+
+## License
+
+This plugin is released under the GPL-2.0 license. 

__init__.py

@@ -0,0 +1,5 @@
+from binaryninja import log_info
+from .revengai import RevengAIPlugin
+
+plugin = RevengAIPlugin()
+log_info("RevEng.AI Plugin loaded successfully")

features/__init__.py

@@ -0,0 +1,4 @@
+from .configuration import ConfigurationFeature
+from .upload import UploadFeature
+
+__all__ = ['ConfigurationFeature', 'UploadFeature'] 

features/configuration/__init__.py

@@ -0,0 +1,24 @@
+from binaryninja import PluginCommand, log_info
+from .config import Config
+from .config_dialog import ApiKeyWizard
+
+class ConfigurationFeature:
+    def __init__(self):
+        self.config = Config()
+        log_info("RevEng.AI Configuration Feature initialized")
+        
+    def register(self):
+        PluginCommand.register(
+            "RevEng.AI\\Configure",
+            "Configure RevEng.AI settings",
+            self.show_configuration
+        )
+        log_info("RevEng.AI Configuration Feature registered")
+        
+    def show_configuration(self, bv):
+        log_info("Opening RevEng.AI configuration wizard")
+        wizard = ApiKeyWizard(self.config)
+        wizard.exec_()
+        
+    def get_config(self):
+        return self.config

features/configuration/config.py

@@ -0,0 +1,69 @@
+from binaryninja import Settings, log_info
+from reait.api import re_conf, RE_authentication
+
+class Config:
+    def __init__(self):
+        self.settings = Settings()
+        self.api_key = ""
+        self.host = ""
+        self.current_analysis = None
+        self._load_config()
+        
+
+    def _load_config(self):
+        settings = Settings()
+        settings.register_group("revengai", "RevEng.AI")
+        settings.register_setting("revengai.host", 
+            '{"title" : "Host URL",\
+            "type" : "string",\
+            "default" : "https://api.reveng.ai",\
+            "description" : "RevEng.AI Host URL"}')
+        settings.register_setting("revengai.api_key", 
+            '{"title" : "API Key",\
+            "type" : "string",\
+            "default" : "",\
+            "description" : "API Key"}')
+        settings.register_setting("revengai.current_analysis", 
+            '{"title" : "Current Analysis ID",\
+            "type" : "string",\
+            "default" : "",\
+            "description" : "Current Analysis ID"}')
+            
+        self.host = settings.get_string("revengai.host", None)
+        self.api_key = settings.get_string("revengai.api_key", None)
+        self.current_analysis = settings.get_string("revengai.current_analysis", None)
+
+        re_conf["apikey"] = self.api_key
+        re_conf["host"] = self.host
+        
+
+    def save_config(self) -> bool:
+        log_info(f"Saving configuration: {self.host} {self.api_key[:4]}...")
+        if self.current_analysis:
+            settings.set_string("revengai.current_analysis", self.current_analysis)
+        
+        try:
+            re_conf["apikey"] = self.api_key
+            re_conf["host"] = self.host
+            log_info(f"Saving configuration: {re_conf['apikey']} {re_conf['host']}")
+            RE_authentication()
+            log_info(f"Authentication successful")
+            settings = Settings()
+            settings.set_string("revengai.host", self.host)
+            settings.set_string("revengai.api_key", self.api_key)
+            return True
+        
+        except Exception as e:
+            log_info(f"Failed to save API key: {str(e)}")
+            return False
+        
+        
+    def clear_config(self):
+        self.api_key = ""
+        self.host = ""
+        self.current_analysis = None
+        self.save_config() 
+
+
+    def is_configured(self):
+        return bool(self.api_key and self.host) 

features/configuration/config_dialog.py

@@ -0,0 +1,179 @@
+from PySide6.QtWidgets import (QDialog, QVBoxLayout, QHBoxLayout, QLabel, 
+                                 QLineEdit, QPushButton, QMessageBox, QProgressDialog, QProgressBar)
+from PySide6.QtCore import Qt
+from PySide6.QtGui import QPixmap
+from binaryninja import log_info, log_error, log_warn
+import os
+
+class ApiKeyWizard(QDialog):
+    def __init__(self, config):
+        super().__init__()
+        self.config = config
+        self.init_ui()
+        
+    def init_ui(self):
+        self.setWindowTitle("RevEng.AI Configuration Wizard")
+        self.setMinimumWidth(500)
+        
+        layout = QVBoxLayout()
+
+        header_layout = QHBoxLayout()
+        
+        logo_label = QLabel()
+        logo_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "images", "logo.png")
+        if os.path.exists(logo_path):
+            pixmap = QPixmap(logo_path)
+            pixmap = pixmap.scaled(100, 100, Qt.KeepAspectRatio, Qt.SmoothTransformation)
+            logo_label.setPixmap(pixmap)
+        else:
+            log_info("RevEng.AI logo not found at: " + logo_path)
+        header_layout.addWidget(logo_label)
+        
+        welcome_layout = QVBoxLayout()
+        title_label = QLabel("Welcome to RevEng.AI!")
+        title_label.setStyleSheet("font-size: 18px; font-weight: bold;")
+        description_label = QLabel(
+            "To get started, you'll need to configure your API key and host URL.\n"
+            "You can get your API key from your RevEng.AI account settings."
+        )
+        description_label.setWordWrap(True)
+        welcome_layout.addWidget(title_label)
+        welcome_layout.addWidget(description_label)
+        header_layout.addLayout(welcome_layout, stretch=1)
+        
+        layout.addLayout(header_layout)
+        layout.addSpacing(20)
+        
+        api_key_label = QLabel("API Key:")
+        api_key_label.setStyleSheet("font-weight: bold;")
+        self.api_key_input = QLineEdit()
+        self.api_key_input.setPlaceholderText("Enter your RevEng.AI API key")
+        self.api_key_input.setText(self.config.api_key if self.config.api_key else "")
+
+        layout.addWidget(api_key_label)
+        layout.addWidget(self.api_key_input)
+        layout.addSpacing(10)
+        
+        host_label = QLabel("Host URL:")
+        host_label.setStyleSheet("font-weight: bold;")
+        self.host_input = QLineEdit()
+        self.host_input.setPlaceholderText("Enter the RevEng.AI host URL")
+        self.host_input.setText(self.config.host if self.config.host else "")
+
+        layout.addWidget(host_label)
+        layout.addWidget(self.host_input)
+        layout.addSpacing(20)
+        
+        button_layout = QHBoxLayout()
+        self.save_button = QPushButton("Save Configuration")
+        self.save_button.setStyleSheet("""
+            QPushButton {
+                background-color: #007bff;
+                color: white;
+                padding: 8px 16px;
+                border-radius: 4px;
+            }
+            QPushButton:hover {
+                background-color: #4400ff;
+            }
+        """)
+        self.save_button.clicked.connect(self.save_config)
+        self.cancel_button = QPushButton("Cancel")
+        self.cancel_button.setStyleSheet("""
+            QPushButton {
+                padding: 8px 16px;
+                border-radius: 4px;
+            }
+        """)
+        self.cancel_button.clicked.connect(self.reject)
+        
+        button_layout.addWidget(self.save_button)
+        button_layout.addWidget(self.cancel_button)
+        layout.addLayout(button_layout)
+        
+        self.setLayout(layout)
+        
+    def save_config(self):
+        api_key = self.api_key_input.text().strip()
+        host = self.host_input.text().strip()
+
+        if not api_key:
+            log_warn("API Key field is empty")
+            QMessageBox.warning(
+                self, 
+                "RevEng.AI Configuration", 
+                "API Key is required to use RevEng.AI services.",
+                QMessageBox.Ok
+            )
+            return
+            
+        if not host:
+            log_warn("Host URL field is empty")
+            QMessageBox.warning(
+                self, 
+                "RevEng.AI Configuration", 
+                "Host URL is required to connect to RevEng.AI services.",
+                QMessageBox.Ok
+            )
+            return
+            
+        try:
+            progress = QProgressDialog("Testing API key...", None, 0, 0, self)
+            progress.setWindowTitle("RevEng.AI Configuration")
+            progress.setWindowModality(Qt.WindowModal)
+            progress.setCancelButton(None)
+            progress.setMinimumWidth(400)
+            progress.setMinimumHeight(100)
+            progress.findChild(QProgressBar).setMinimumWidth(250)
+            progress.findChild(QProgressBar).setMinimumHeight(20)
+            progress.setStyleSheet("""
+                QProgressDialog {
+                    background-color: white;
+                    color: black;
+                }
+                QProgressBar {
+                    border: 1px solid #cccccc;
+                    border-radius: 4px;
+                    text-align: center;
+                    background-color: #f0f0f0;
+                    min-width: 250px;
+                    min-height: 20px;
+                }
+                QProgressBar::chunk {
+                    background-color: #007bff;
+                    border-radius: 3px;
+                }
+                QLabel {
+                    color: black;
+                    font-size: 13px;
+                    padding: 10px;
+                }
+            """)
+            progress.show()
+
+            self.config.api_key = api_key
+            self.config.host = host
+            key_try = self.config.save_config()
+            
+            progress.close()
+
+            if not key_try:
+                raise Exception("API key not valid or host not reachable.")
+            
+            log_info("RevEng.AI configuration saved successfully!")
+            QMessageBox.information(
+                self, 
+                "RevEng.AI Configuration",
+                "Configuration saved successfully!\nYou can now start using RevEng.AI services.",
+                QMessageBox.Ok
+            )
+            self.accept()
+
+        except Exception as e:
+            log_error(f"Failed to save RevEng.AI configuration: {str(e)}")
+            QMessageBox.critical(
+                self, 
+                "RevEng.AI Configuration Error",
+                f"Failed to save configuration: {str(e)}",
+                QMessageBox.Ok
+            ) 

features/upload/__init__.py

@@ -0,0 +1,25 @@
+from binaryninja import PluginCommand, log_info, log_error
+from .upload import BinaryUploader
+from .upload_dialog import UploadDialog
+from ...utils.config_validator import validate_config
+
+class UploadFeature:
+    def __init__(self, config=None):
+        self.config = config
+        self.uploader = BinaryUploader(config)
+        log_info("RevEng.AI Process Feature initialized")
+        
+    def register(self):
+        PluginCommand.register(
+            "RevEng.AI\\Process Binary",
+            "Process current binary to RevEng.AI for analysis",
+            self.show_upload_dialog
+        )
+        log_info("RevEng.AI Process feature registered")
+        
+    def show_upload_dialog(self, bv):
+        if not validate_config(self.config):
+            return
+        dialog = UploadDialog(self.config, self.uploader, bv)
+        if dialog.exec() == UploadDialog.Accepted:
+            self.uploader.upload_binary(bv, dialog.get_upload_options())

features/upload/upload.py

@@ -0,0 +1,32 @@
+from binaryninja import BinaryView, log_info, log_error
+from reait.api import RE_models
+
+class BinaryUploader:
+    def __init__(self, config):
+        self.config = config
+
+    
+    def get_models(self):
+        try:
+            models = RE_models().json()
+            return set([model["model_name"] for model in models["models"]])
+        except Exception as e:
+            log_error(f"Failed to get models: {str(e)}")
+            return set()
+                
+
+    def upload_binary(self, bv: BinaryView, options: dict):
+        try:
+            
+            # TODO: Implement actual upload logic here
+            # This will involve:
+            # 1. Getting the binary data from bv
+            # 2. Use the RE_upload_binary function to upload the binary
+            # 3. Handling the response
+            
+            log_info(f"Binary {bv.file.filename} uploaded successfully with options: {options}")
+            return True
+            
+        except Exception as e:
+            log_error(f"Failed to upload binary: {str(e)}")
+            return False