relocs

main.py

@@ -68,20 +68,27 @@ def compile(compiler, flags, source):
 
     # Disassemble the object file
     disassembly = subprocess.run(
-        ["objdump", "-dr", temp_o_file_name],
+        ["objdump", "-r", temp_o_file_name],
         capture_output=True,
         text=True
     ).stdout
     disassembly = trim_objdump(disassembly)
 
+    # Relocs
+    relocs = subprocess.run(
+        ["objdump", "-r", temp_o_file_name],
+        capture_output=True,
+        text=True
+    ).stdout
+
     if result.returncode == 0:
-        return compiled_bytes, compile_output, disassembly
+        return relocs, compiled_bytes, compile_output, disassembly
     else:
-        return None, compile_output, disassembly
+        return None, None, compile_output, disassembly
 
 def predict(target_bytes, source, compiler, flags, disasm_arch, disasm_options):
     target_bytes = bytes.fromhex(target_bytes)
-    compiled_bytes, compile_output, compiled_disassembly = compile(compiler, flags, source)
+    compiled_relocs, compiled_bytes, compile_output, compiled_disassembly = compile(compiler, flags, source)
     target_disassembly = disassemble_bytes(target_bytes, disasm_arch, disasm_options)
 
     if compiled_bytes is not None:
@@ -91,6 +98,7 @@ def predict(target_bytes, source, compiler, flags, disasm_arch, disasm_options):
             editdistance.eval(compiled_bytes, target_bytes),
             compile_output,
             compiled_disassembly,
+            compiled_relocs,
             target_disassembly
         )
     else:
@@ -100,6 +108,7 @@ def predict(target_bytes, source, compiler, flags, disasm_arch, disasm_options):
             -1,
             compile_output,
             compiled_disassembly,
+            compiled_relocs,
             target_disassembly
         )
 
@@ -130,6 +139,7 @@ def run():
             gr.Number(label="Edit distance (lower is better)"),
             gr.Textbox(label="Compiler Output"),
             gr.Textbox(label="Compiled Disassembly"),
+            gr.Textbox(label="Compiled relocations"),
             gr.Textbox(label="Target Disassembly"),
         ],
     )