fix: auto-unstrip renaming

Author: robertmarsal

reai_toolkit/features/auto_unstrip/auto_unstrip.py

@@ -83,7 +83,14 @@ def rename_functions(self, bv: BinaryView, options: List[Dict]):
         try:
             renamed_count = 0
             for option in options:
-                if rename_function_util(bv, option['virtual_address'], option['suggested_name']):
+                if rename_function_util(
+                        self.config,
+                        bv,
+                        option['virtual_address'],
+                        option['suggested_name'],
+                        option['suggested_mangled_name'],
+                        option['source_function_id'],
+                ):
                     renamed_count += 1
             return True, f"Successfully renamed {renamed_count} functions"
         except Exception as e:
@@ -122,15 +129,28 @@ def auto_unstrip(self, bv: BinaryView):
                 else:
                     raise Exception(api_response.error)
             for match in matches:
+                print(match, flush=True)
                 try:
                     if self.cancelled.is_set():
                         return False, "Operation cancelled"
 
                     function = get_function_by_addr_util(bv, match.function_vaddr)
-                    results.append({"virtual_address": match.function_vaddr, "current_name": function.name, "suggested_name": match.suggested_demangled_name})
+                    results.append({
+                        "virtual_address": match.function_vaddr,
+                        "current_name": function.name,
+                        "suggested_name": match.suggested_demangled_name,
+                        "suggested_mangled_name": match.suggested_name,
+                        "source_function_id": match.function_id,
+                    })
                 except Exception as e:
                     log_error(f"RevEng.AI | Error getting function by address {match.function_vaddr}: {str(e)}")
-                    results.append({"virtual_address": match.function_vaddr, "current_name": "N/A", "suggested_name": match.suggested_demangled_name})
+                    results.append({
+                        "virtual_address": match.function_vaddr,
+                        "current_name": "N/A",
+                        "suggested_name": match.suggested_demangled_name,
+                        "suggested_mangled_name": match.suggested_name,
+                        "source_function_id": match.function_id,
+                    })
 
             if self.cancelled.is_set():
                 return False, "Operation cancelled"

reai_toolkit/features/auto_unstrip/auto_unstrip_dialog.py

@@ -145,7 +145,7 @@ def populate_results_table(self, results):
             select_item.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsSelectable)
             select_item.setCheckState(Qt.Checked)
             self.results_table.setItem(i, 0, select_item)
-            
+
             columns = [
                 (1, "virtual_address", lambda x: f"0x{x:x}" if isinstance(x, int) else str(x)),
                 (2, "current_name", lambda x: str(x)),

reai_toolkit/features/view_analysis/__init__.py

@@ -8,7 +8,7 @@ def __init__(self, config=None):
 
     def register(self):
         PluginCommand.register(
-            "RevEng.AI\\Analysis\\\u200b\u200bView Analysis",
+            "RevEng.AI\\Analysis\\\u200b\u200bView in portal",
             "View the current analysis in the RevEng.AI portal",
             self.open_analysis,
             self.is_valid