main.py

import gradio as gr
import subprocess
import tempfile

description = """This is a space testing a method for evaluating the quality of decompilation.

Currently unhandled features:
* Global references
* Function calls
* Compiler and flag selection
"""


def compile(source):
    # Create a temporary file for the C source code
    with tempfile.NamedTemporaryFile(suffix=".c", delete=False) as temp_c_file:
        temp_c_file.write(source.encode())
        temp_c_file_name = temp_c_file.name

    # Create a temporary file for the object file
    with tempfile.NamedTemporaryFile(suffix=".o", delete=False) as temp_o_file:
        temp_o_file_name = temp_o_file.name

    # Compile the C file to an object file
    subprocess.run(["cc", "-c", temp_c_file_name, "-o", temp_o_file_name], check=True)

    # Create a temporary file for the raw bytes
    with tempfile.NamedTemporaryFile(suffix=".raw", delete=True) as raw_bytes_file:
        subprocess.run(
            [
                "objcopy",
                "--only-section",
                ".text",
                "-O",
                "binary",
                temp_o_file_name,
                raw_bytes_file.name,
            ]
        )
        compiled_bytes = raw_bytes_file.read()

    return compiled_bytes


def predict(source):
    bytes = compile(source)
    return bytes, 1.0


def run():
    demo = gr.Interface(
        fn=predict,
        description=description,
        inputs=gr.Textbox(lines=10, label="Decompiled C Source Code"),
        outputs=[gr.Textbox(), gr.Number()],
    )

    demo.launch(server_name="0.0.0.0", server_port=7860)


run()