xref: /llvm-project/llvm/test/tools/llvm-reduce/Inputs/llvm-dis-and-filecheck.py (revision 363d99db49a1ded7d31c7253d203f737d269ad86)
1"""
2Script to disassembles a bitcode file and run FileCheck on the output with the
3provided arguments. The first 2 arguments are the paths to the llvm-dis and
4FileCheck binaries, followed by arguments to be passed to FileCheck. The last
5argument is the bitcode file to disassemble.
6
7Usage:
8    python llvm-dis-and-filecheck.py
9      <path to llvm-dis> <path to FileCheck>
10      [arguments passed to FileCheck] <path to bitcode file>
11
12"""
13
14
15import sys
16import os
17import subprocess
18
19llvm_dis = sys.argv[1]
20filecheck = sys.argv[2]
21filecheck_args = [
22    filecheck,
23]
24filecheck_args.extend(sys.argv[3:-1])
25bitcode_file = sys.argv[-1]
26ir_file = bitcode_file + ".ll"
27
28disassemble = subprocess.Popen([llvm_dis, "--preserve-ll-uselistorder", "-o", ir_file, bitcode_file])
29if os.path.exists(ir_file + ".0"):
30    ir_file = ir_file + ".0"
31
32disassemble.communicate()
33
34if disassemble.returncode != 0:
35    print("stderr:")
36    print(disassemble.stderr)
37    print("stdout:")
38    print(disassemble.stdout)
39    sys.exit(1)
40
41check = None
42with open(ir_file, "r") as ir:
43    check = subprocess.Popen(filecheck_args, stdin=ir, stdout=sys.stdout)
44check.communicate()
45sys.exit(check.returncode)
46