Llm4decompile 6.7b V1.5

🚀 LLM4Decompile

LLM4Decompile旨在將x86彙編指令反編譯為C語言代碼。新發布的V1.5系列使用了更大的數據集（150億個標記）進行訓練，最大標記長度為4096，與之前的模型相比，性能有顯著提升（最高提升100%）。

🚀 快速開始

LLM4Decompile是一個用於將x86彙編指令反編譯為C語言代碼的工具。新發布的V1.5系列模型在性能上有顯著提升。你可以通過以下鏈接訪問其Github倉庫：

• Github倉庫地址：LLM4Decompile

✨ 主要特性

• 更大的訓練數據集：V1.5系列使用了150億個標記的數據集進行訓練。
• 更長的標記長度：最大標記長度達到4096。
• 顯著的性能提升：與之前的模型相比，性能最高提升100%。

📊 評估結果

💻 使用示例

基礎用法

以下是使用我們模型的示例（V1.5版本修訂）。 注意：請將func0替換為你要反編譯的函數名。

預處理：將C代碼編譯為二進制文件，並將二進制文件反彙編為彙編指令。

import subprocess
import os

OPT = ["O0", "O1", "O2", "O3"]
fileName = 'samples/sample' #'path/to/file'
for opt_state in OPT:
    output_file = fileName +'_' + opt_state
    input_file = fileName+'.c'
    compile_command = f'gcc -o {output_file}.o {input_file} -{opt_state} -lm'#compile the code with GCC on Linux
    subprocess.run(compile_command, shell=True, check=True)
    compile_command = f'objdump -d {output_file}.o > {output_file}.s'#disassemble the binary file into assembly instructions
    subprocess.run(compile_command, shell=True, check=True)
    
    input_asm = ''
    with open(output_file+'.s') as f:#asm file
        asm= f.read()
        if '<'+'func0'+'>:' not in asm: #IMPORTANT replace func0 with the function name
            raise ValueError("compile fails")
        asm = '<'+'func0'+'>:' + asm.split('<'+'func0'+'>:')[-1].split('\n\n')[0] #IMPORTANT replace func0 with the function name
        asm_clean = ""
        asm_sp = asm.split("\n")
        for tmp in asm_sp:
            if len(tmp.split("\t"))<3 and '00' in tmp:
                continue
            idx = min(
                len(tmp.split("\t")) - 1, 2
            )
            tmp_asm = "\t".join(tmp.split("\t")[idx:])  # remove the binary code
            tmp_asm = tmp_asm.split("#")[0].strip()  # remove the comments
            asm_clean += tmp_asm + "\n"
    input_asm = asm_clean.strip()
    before = f"# This is the assembly code:\n"#prompt
    after = "\n# What is the source code?\n"#prompt
    input_asm_prompt = before+input_asm.strip()+after
    with open(fileName +'_' + opt_state +'.asm','w',encoding='utf-8') as f:
        f.write(input_asm_prompt)

反編譯：使用LLM4Decompile將彙編指令翻譯為C代碼。

from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

model_path = 'LLM4Binary/llm4decompile-6.7b-v1.5' # V1.5 Model
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()

with open(fileName +'_' + OPT[0] +'.asm','r') as f:#optimization level O0
    asm_func = f.read()
inputs = tokenizer(asm_func, return_tensors="pt").to(model.device)
with torch.no_grad():
    outputs = model.generate(**inputs, max_new_tokens=4000)
c_func_decompile = tokenizer.decode(outputs[0][len(inputs[0]):-1])

with open(fileName +'.c','r') as f:#original file
    func = f.read()

print(f'original function:\n{func}')# Note we only decompile one function, where the original file may contain multiple functions
print(f'decompiled function:\n{c_func_decompile}')

📄 許可證

本代碼倉庫遵循MIT許可證。

📞 聯繫我們

如果您有任何問題，請在倉庫中提出issue。