File: llvm-dis-and-filecheck.py

package info (click to toggle)
llvm-toolchain-16 1%3A16.0.6-15~deb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,634,792 kB
  • sloc: cpp: 6,179,261; ansic: 1,216,205; asm: 741,319; python: 196,614; objc: 75,325; f90: 49,640; lisp: 32,396; pascal: 12,286; sh: 9,394; perl: 7,442; ml: 5,494; awk: 3,523; makefile: 2,723; javascript: 1,206; xml: 886; fortran: 581; cs: 573
file content (43 lines) | stat: -rwxr-xr-x 1,134 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
"""
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, "-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)