dev

main.py

@@ -1,7 +1,57 @@
 import gradio as gr
+import shutil
+import subprocess
+import tempfile
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+def new_binary(file):
+
+    with tempfile.TemporaryDirectory() as TEMPDIR, tempfile.TemporaryDirectory() as OUTPUTDIR:
+        shutil.copy2(file.name, TEMPDIR)
+
+        # python3 generate.py --ghidra PATH_TO_GHIDRA_ANALYZEHEADLESS -t NUM_THREADS -n [NUM_FILES|None] -b BINARIES_DIR -o OUTPUT_DIR
+        output = subprocess.check_output(f"python /DIRTY/dataset-gen-ghidra/generate.py --ghidra /ghidra/support/analyzeHeadless -t 1 -b {TEMPDIR} -o {OUTPUT_DIR}")
+        print(output)
+
+
+
+with gr.Blocks() as demo:
+
+    all_dis_state = gr.State()
+
+    gr.Markdown(
+        """
+    # DIRTY-Ghidra Testing
+    First, upload a binary.
+    """
+    )
+
+    file_widget = gr.File(label="Binary file")
+
+    def file_change_fn(file, progress=gr.Progress()):
+
+        print("hi")
+        new_binary(file)
+
+        if file is None:
+            return {
+                    #col: gr.update(visible=False),
+                    all_dis_state: None}
+        else:
+
+            #fun_data = {42: 2, 43: 3}
+            progress(0, desc="Disassembling executable")
+            #fun_data = get_all_dis(file.name)
+
+            #addrs = ["%#x" % addr for addr in fun_data.keys()]
+
+            return {
+                    #col: gr.update(visible=True),
+                    #fun_dropdown: gr.Dropdown.update(choices=addrs, value=addrs[0]),
+                    #all_dis_state: fun_data
+                    }
+        
+    file_widget.change(file_change_fn, file_widget)
+
+demo.queue()
 demo.launch(server_name="0.0.0.0", server_port=7860)