feat: add ui dialog for config options

Author: SmoothHacker

__init__.py

@@ -1,15 +1,24 @@
 import os
 
+import binaryninja as bn
 from binaryninja import BinaryView, Function
 from binaryninja import CoreSymbol, Logger
 from binaryninja import PluginCommand
 from binaryninja import SymbolType, SymbolBinding
 from binaryninja import TypeLibrary
 
 
-def create_type_library(log: Logger, bv: BinaryView, func_list: list[Function]) -> TypeLibrary:
+def create_type_library(log: Logger, bv: BinaryView, func_list: list[Function], config: dict) -> TypeLibrary:
     typelib = TypeLibrary.new(bv.arch, os.path.basename(bv.file.filename))
     typelib.add_platform(bv.platform)
+
+    if "alternate_names" not in config:
+        name_list = [name.strip() for name in config["alternate_names"].split(";")]
+        for name in name_list:
+            typelib.add_alternate_name(name)
+
+    if "dependency_name" not in config:
+        typelib.dependency_name = config["dependency_name"]
     log.log_debug(f"Exporting {len(func_list)} functions to a type library")
     for func in func_list:
         bv.export_object_to_library(typelib, func.name, func.function_type)
@@ -28,8 +37,23 @@ def get_funcs_from_syms(log: Logger, bv: BinaryView, func_syms: list[CoreSymbol]
     return func_list
 
 
+def get_config_options(bv: BinaryView):
+    alternate_names = bn.TextLineField("Alternative Names (optional):", "lib_musl.so;lib_musl.so.5")
+    export_path = bn.TextLineField("Path to store type library:", f"{bv.file.filename}.bntl")
+    dependency_name = bn.TextLineField("Dependency Name (optional):")
+    bn.get_form_input([alternate_names, export_path, dependency_name], "Export as Type Library Options")
+
+    config = {"alternate_names": alternate_names.result, "export_path": export_path.result,
+              "dependency_name": dependency_name.result}
+    return config
+
+
 def export_functions(bv: BinaryView):
     log = bv.create_logger("TypeLib_Exporter")
+    config = get_config_options(bv)
+    if not os.path.exists(config['export_path']):
+        log.log_error("Please specify a path to export the type library")
+        return
 
     func_list = bv.get_symbols_of_type(SymbolType.FunctionSymbol)
     export_func_syms = [sym for sym in func_list
@@ -38,10 +62,10 @@ def export_functions(bv: BinaryView):
     export_funcs = get_funcs_from_syms(log, bv, export_func_syms)
     log.log_debug(f"Discovered {len(export_funcs)} exported functions")
 
-    typelib = create_type_library(log, bv, export_funcs)
+    typelib = create_type_library(log, bv, export_funcs, config)
     typelib.finalize()
-    log.log_info(f"Exported {len(export_funcs)} functions to {os.path.basename(bv.file.filename)}.bntl")
-    typelib.write_to_file(f"~/{os.path.basename(bv.file.filename)}.bntl")
+    log.log_info(f"Exported {len(export_funcs)} functions to {config['export_path']}")
+    typelib.write_to_file(config['export_path'])
     return