"""
Script to disassembles a bitcode file and run FileCheck on the output with the
provided arguments. The first 2 arguments are the paths to the llvm-dis and
FileCheck binaries, followed by arguments to be passed to FileCheck. The last
argument is the bitcode file to disassemble.
Usage:
python llvm-dis-and-filecheck.py
<path to llvm-dis> <path to FileCheck>
[arguments passed to FileCheck] <path to bitcode file>
"""
import sys
import os
import subprocess
llvm_dis = sys.argv[1]
filecheck = sys.argv[2]
filecheck_args = [
filecheck,
]
filecheck_args.extend(sys.argv[3:-1])
bitcode_file = sys.argv[-1]
ir_file = bitcode_file + ".ll"
disassemble = subprocess.Popen([llvm_dis, "--preserve-ll-uselistorder", "-o", ir_file, bitcode_file])
if os.path.exists(ir_file + ".0"):
ir_file = ir_file + ".0"
disassemble.communicate()
if disassemble.returncode != 0:
print("stderr:")
print(disassemble.stderr)
print("stdout:")
print(disassemble.stdout)
sys.exit(1)
check = None
with open(ir_file, "r") as ir:
check = subprocess.Popen(filecheck_args, stdin=ir, stdout=sys.stdout)
check.communicate()
sys.exit(check.returncode)