extra args

main.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import logging
 import os
+import shlex
 import tempfile
 import subprocess
 import pyghidra
@@ -20,39 +21,8 @@ def get_functions(file):
         program = flat_api.getCurrentProgram()
         function_addrs = [(f.getName(), f.getEntryPoint().getOffset()) for f in program.getFunctionManager().getFunctions(True)]
 
-        # o = subprocess.run(["/code/GhidraFunctionCPPExporter/export.bash", file, "output_dir", TEMP_DIR], shell=False, capture_output=True, encoding="utf8")
-
-        # print(o.stdout)
-        # print(o.stderr)
-
-        # if (o.returncode != 0):
-        #     raise Exception(f"Ghidra export failed with return code {o.returncode}: {o.stderr}")
-
-        # # Show files in TEMP_DIR
-        # print("Files in TEMP_DIR:")
-        # for f in os.listdir(TEMP_DIR):
-        #     print(f)
-
     return function_addrs
 
-
-# def decomp_create_prompt(input_data: str) -> str:
-#     before = "# This is the decompiled code:\n"
-#     after = "\n# What is the source code?\n"
-#     prompt = before + input_data.strip() + after
-
-#     return prompt
-
-
-# pipe = pipeline(model="ejschwartz/decaf-v1-22b-4bit", return_full_text=False)
-# pipe.model.to("cuda")
-
-# def generate(text):
-#     print(f"Generating text... {text}")
-#     prompt = decomp_create_prompt(text)
-#     return pipe(prompt, max_new_tokens=2000)[0]['generated_text']
-
-
 with gr.Blocks() as demo:
 
     state = gr.State()
@@ -85,6 +55,8 @@ This is a space to experiment with GhidraFunctionCPPExporter, a Ghidra scripts t
                     """
         )
 
+        extra_args = gr.Textbox(label="Extra args to export.bash", placeholder="emit_type_definitions false", value="")
+
         with gr.Row(visible=True) as result:
             disassembly = gr.Code(
                 label="Disassembly", lines=20,
@@ -135,8 +107,8 @@ This is a space to experiment with GhidraFunctionCPPExporter, a Ghidra scripts t
                         "addrs": addrs},
             }
         
-    @fun_dropdown.change(inputs=[fun_dropdown, state], outputs=[disassembly, original_decompile])
-    def function_change_fn(selected_fun, state, progress=gr.Progress()):
+    @fun_dropdown.change(inputs=[fun_dropdown, state, extra_args], outputs=[disassembly, original_decompile])
+    def function_change_fn(selected_fun, state, extra_args, progress=gr.Progress()):
 
         with tempfile.TemporaryDirectory() as TEMP_DIR:
 
@@ -151,7 +123,7 @@ This is a space to experiment with GhidraFunctionCPPExporter, a Ghidra scripts t
                 disassembly_str = "\n".join([i.toString() for i in listing.getInstructions(func.getBody(), True)])
 
 
-            o = subprocess.run(["/code/GhidraFunctionCPPExporter/export.bash", state['file'], "base_name", "file", "address_set_str", hex(selected_fun), "output_dir", TEMP_DIR], shell=False, capture_output=True, encoding="utf8")
+            o = subprocess.run(["/code/GhidraFunctionCPPExporter/export.bash", state['file'], "base_name", "file", "address_set_str", hex(selected_fun), "output_dir", TEMP_DIR] + shlex.split(extra_args), shell=False, capture_output=True, encoding="utf8")
 
             print(o.stdout)
             print(o.stderr)