feat: detach analysis option

Author: vgoat21

reai_toolkit/features/__init__.py

@@ -6,6 +6,7 @@
 from reai_toolkit.features.match_current_function import MatchCurrentFunctionFeature
 from reai_toolkit.features.view_function_in_portal import ViewFunctionInPortalFeature
 from reai_toolkit.features.ai_decompiler import AIDecompilerFeature
+from reai_toolkit.features.detach import DetachAnalysisFeature
 __all__ = [
     'ConfigurationFeature', 
     'UploadFeature', 
@@ -15,4 +16,5 @@
     'MatchCurrentFunctionFeature', 
     'ViewFunctionInPortalFeature', 
     'AIDecompilerFeature'
+    'DetachAnalysisFeature'
     ] 

reai_toolkit/features/ai_decompiler/__init__.py

@@ -66,4 +66,4 @@ def on_dock_closed(self, visible):
             self.ai_decompiler.stop_address_tracking()
 
     def is_valid(self, bv: BinaryView, func):
-        return self.config.is_configured == True 
+        return self.config.is_configured == True and self.config.analysis_id is not None 

reai_toolkit/features/auto_unstrip/__init__.py

@@ -22,4 +22,7 @@ def show_auto_unstrip_dialog(self, bv: BinaryView):
         log_info("RevEng.AI | Opening AutoUnstrip dialog")
         dialog = AutoUnstripDialog(self.config, self.auto_unstrip, bv)
         dialog.exec_()
+
+    def is_valid(self, bv: BinaryView):
+        return self.config.is_configured == True and self.config.analysis_id is not None
 

reai_toolkit/features/choose_source/__init__.py

@@ -23,3 +23,6 @@ def show_choose_source_dialog(self, bv: BinaryView):
         dialog = ChooseSourceDialog(self.config, self.choose_source, bv)
         dialog.exec_()
 
+    def is_valid(self, bv: BinaryView):
+        return self.config.is_configured == True and self.config.analysis_id is None
+

reai_toolkit/features/configuration/config.py

@@ -190,4 +190,22 @@ def get_analysis_id(self, bv: BinaryView):
         if self.sha256 in all_analyses:
             return all_analyses[self.sha256]["analysis_id"]
         else:
-            return 0
+            return 0
+        
+
+    def reset_analysis_data(self, bv: BinaryView):
+        try:
+            self.analysis_id = None
+            self.binary_id = None
+            self.model_id = None
+            self.sha256 = get_sha256(bv.file.filename)
+            all_analyses = self.get_all_analyses()
+            if self.sha256 in all_analyses:
+                del all_analyses[self.sha256]
+                settings = Settings()
+                settings.set_json("revengai.all_analyses", json.dumps(all_analyses))
+                log_info(f"RevEng.AI | Reset analysis data for SHA256: {self.sha256[:8]}...")
+            return True, ""
+        except Exception as e:
+            log_error(f"RevEng.AI | Failed to reset analysis data: {str(e)}")
+            return False, str(e)

reai_toolkit/features/detach/__init__.py

@@ -0,0 +1,47 @@
+from reai_toolkit.utils import BaseAuthFeature
+from binaryninja import PluginCommand, log_info, BinaryView
+from PySide6.QtWidgets import QMessageBox
+
+class DetachAnalysisFeature(BaseAuthFeature):
+    def __init__(self, config=None):
+        super().__init__(config)
+        log_info("RevEng.AI | DetachAnalysis Feature initialized")
+
+    def register(self):
+        PluginCommand.register(
+            "RevEng.AI\\Analysis\\\u200b\u200bDetach Analysis",
+            "Detach from the current RevEng.AI analysis",
+            self.verify_detach,
+            self.is_valid
+        )
+        log_info("RevEng.AI | DetachAnalysis Feature registered")
+
+    def verify_detach(self, bv: BinaryView):
+        log_info("RevEng.AI | Requesting detach confirmation")
+        
+        reply = QMessageBox.question(
+            None,
+            "RevEng.AI - Detach Analysis",
+            "Are you sure you want to detach from the current RevEng.AI analysis?\n\n"
+            "This will disconnect this binary from its analysis on the RevEng.AI platform. "
+            "You can reconnect later by choosing a source again.",
+            QMessageBox.Yes | QMessageBox.No,
+            QMessageBox.No
+        )
+        
+        if reply == QMessageBox.Yes:
+            log_info("RevEng.AI | User confirmed detach")
+            self.config.reset_analysis_data(bv)
+            
+            QMessageBox.information(
+                None,
+                "RevEng.AI - Analysis Detached",
+                "Successfully detached from the RevEng.AI analysis.\n\n"
+                "You can process a new binary or choose a different source.",
+                QMessageBox.Ok
+            )
+        else:
+            log_info("RevEng.AI | User cancelled detach")
+
+    def is_valid(self, bv: BinaryView):
+        return self.config.is_configured == True and self.config.analysis_id is not None 

reai_toolkit/features/match_current_function/__init__.py

@@ -24,4 +24,4 @@ def show_match_current_function_dialog(self, bv: BinaryView, func):
         dialog.exec_() 
 
     def is_valid(self, bv: BinaryView, func):
-        return self.config.is_configured == True 
+        return self.config.is_configured == True and self.config.analysis_id is not None 

reai_toolkit/features/match_functions/__init__.py

@@ -22,3 +22,6 @@ def show_match_functions_dialog(self, bv: BinaryView):
         log_info("RevEng.AI | Opening MatchFunctions dialog")
         dialog = MatchFunctionsDialog(self.config, self.match_functions, bv)
         dialog.exec_() 
+
+    def is_valid(self, bv: BinaryView):
+        return self.config.is_configured == True and self.config.analysis_id is not None

reai_toolkit/features/upload/__init__.py

@@ -21,4 +21,7 @@ def register(self):
     def show_upload_dialog(self, bv: BinaryView):
         log_info("RevEng.AI | Opening process dialog")
         dialog = UploadDialog(self.config, self.uploader, bv)
-        dialog.exec()
+        dialog.exec()
+
+    def is_valid(self, bv: BinaryView):
+        return self.config.is_configured == True and self.config.analysis_id is None

reai_toolkit/features/view_function_in_portal/__init__.py

@@ -25,4 +25,4 @@ def show_match_current_function_dialog(self, bv: BinaryView, func):
         dialog.exec_() 
 
     def is_valid(self, bv: BinaryView, func):
-        return self.config.is_configured == True 
+        return self.config.is_configured == True and self.config.analysis_id is not None