TheRock/compiler/amd-llvm/openmp/runtime/tools/check-execstack.py

#!/usr/bin/env python3

#
# //===----------------------------------------------------------------------===//
# //
# // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# // See https://llvm.org/LICENSE.txt for license information.
# // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
# //
# //===----------------------------------------------------------------------===//
#

import argparse
import os
import re
import sys
from libomputils import ScriptError, error, print_error_line, execute_command


def is_stack_executable_readelf(library):
    """Returns true if stack of library file is executable"""
    r = execute_command(["readelf", "-l", "-W", library])
    if r.returncode != 0:
        error("{} failed".format(r.command))
    stack_lines = []
    for line in r.stdout.split(os.linesep):
        if re.search("STACK", line):
            stack_lines.append(line.strip())
    if not stack_lines:
        error("{}: Not stack segment found".format(library))
    if len(stack_lines) > 1:
        error("{}: More than one stack segment found".format(library))
    h = r"0x[0-9a-fA-F]+"
    m = re.search(
        r"((GNU_)?STACK)\s+({0})\s+({0})\s+({0})\s+({0})\s+({0})"
        " ([R ][W ][E ])".format(h),
        stack_lines[0],
    )
    if not m:
        error("{}: Cannot parse stack segment line".format(library))
    if m:
        flags = m.group(8)
        if "E" in flags:
            return True
    return False


def main():
    parser = argparse.ArgumentParser(
        description="Check library does not have" " executable stack"
    )
    parser.add_argument("library", help="The library file to check")
    commandArgs = parser.parse_args()
    if is_stack_executable_readelf(commandArgs.library):
        error("{}: Stack is executable".format(commandArgs.library))


if __name__ == "__main__":
    try:
        main()
    except ScriptError as e:
        print_error_line(str(e))
        sys.exit(1)

# end of file