use clang to try parsing

README.md

@@ -9,4 +9,4 @@ license: apache-2.0
 ---
 
 This is a space that runs ReSym's Field Helper code in their own Docker
-container.
+container.  You probably should never use it directly.

main.py

@@ -1,8 +1,9 @@
-import tempfile
+import frontmatter
 import gradio as gr
 import json
-import os
 import subprocess
+import tempfile
+
 
 def predict(decompiled_code):
 
@@ -14,10 +15,29 @@ def predict(decompiled_code):
         with open(code_file_name, "w") as f:
             f.write('#include "/home/ReSym/clang-parser/defs.hh"\n' + decompiled_code)
 
-        # subprocess.check_output(["python", "/home/ReSym/process_data/prep_decompiled.py", code_file_name, file_save_dir, parsed_save_dir])
-        output = subprocess.run(["/home/ReSym/clang-parser/build/field_access", code_file_name, field_file_name], text=True, capture_output=True)
+        # First use clang to see if parsing failed
+        output = subprocess.run([
+            "/usr/lib/llvm-12/bin/clang",
+            "-c",
+            code_file_name,
+            "-o",
+            "/dev/null"
+        ],
+            text=True,
+            capture_output=True,
+        )
+        assert output.returncode == 0, f"Clang failed to parse: {output}"
+
+        output = subprocess.run(
+            [
+                "/home/ReSym/clang-parser/build/field_access",
+                code_file_name,
+                field_file_name,
+            ],
+            text=True,
+            capture_output=True,
+        )
         print(output)
-        assert output.returncode == 0
 
         try:
             field_data = open(field_file_name).read()
@@ -25,11 +45,13 @@ def predict(decompiled_code):
             field_data = "[]"
         return field_data
 
+
 def run():
     demo = gr.Interface(
         fn=predict,
-        inputs=gr.Text(),
-        outputs=gr.JSON(),
+        inputs=gr.Textbox(label="Hex-Rays Decompilation", lines=10),
+        outputs=[gr.JSON(label="Field access"), gr.Textbox(label="Standard error")],
+        description=frontmatter.load("README.md").content,
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860, debug=True)

requirements.txt

@@ -1,2 +1,2 @@
 gradio
-
+python-frontmatter