mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-30 05:26:06 +00:00
34ee3d91a8
Normally, the run.py wrapper script runs the child processes in a clean environment, with only the environment variables available that are passed via the --env parameter. However, the COMSPEC and TEMP variables are kind of necessary when running some tests; COMSPEC is necessary for finding the interpreter when executing commands via std::system(). Before f1a96de1bc8db527b5eb820c36c17e275900ca2b, tests were executed via an intermediate shell which implicitly readded the COMSPEC variable. The TEMP variable allows temp files to be placed in a sensible location; if unset, they're placed in the default temp fallback of C:\Windows instead. Differential Revision: https://reviews.llvm.org/D97452
57 lines
2.0 KiB
Python
Executable File
57 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#===----------------------------------------------------------------------===##
|
|
#
|
|
# 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
|
|
#
|
|
#===----------------------------------------------------------------------===##
|
|
|
|
"""run.py is a utility for running a program.
|
|
|
|
It can perform code signing, forward arguments to the program, and return the
|
|
program's error code.
|
|
"""
|
|
|
|
import argparse
|
|
import os
|
|
import platform
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--execdir', type=str, required=True)
|
|
parser.add_argument('--codesign_identity', type=str, required=False, default=None)
|
|
parser.add_argument('--env', type=str, nargs='*', required=False, default=dict())
|
|
parser.add_argument("command", nargs=argparse.ONE_OR_MORE)
|
|
args = parser.parse_args()
|
|
commandLine = args.command
|
|
|
|
# Do any necessary codesigning.
|
|
if args.codesign_identity:
|
|
exe = commandLine[0]
|
|
rc = subprocess.call(['xcrun', 'codesign', '-f', '-s', args.codesign_identity, exe], env={})
|
|
if rc != 0:
|
|
sys.stderr.write('Failed to codesign: ' + exe)
|
|
return rc
|
|
|
|
# Extract environment variables into a dictionary
|
|
env = {k : v for (k, v) in map(lambda s: s.split('=', 1), args.env)}
|
|
if platform.system() == 'Windows':
|
|
# Pass some extra variables through on Windows:
|
|
# COMSPEC is needed for running subprocesses via std::system().
|
|
if 'COMSPEC' in os.environ:
|
|
env['COMSPEC'] = os.environ.get('COMSPEC')
|
|
# TEMP is needed for placing temp files in a sensible directory.
|
|
if 'TEMP' in os.environ:
|
|
env['TEMP'] = os.environ.get('TEMP')
|
|
|
|
# Run the command line with the given environment in the execution directory.
|
|
return subprocess.call(commandLine, cwd=args.execdir, env=env, shell=False)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
exit(main())
|