9ef5d45b2c
This patch addresses some comments on http://reviews.llvm.org/D7019. - Move compiler.py to test/libcxx from test/libcxx/test. - Make CXXCompiler.target None by default. - Use `{}` instead of `dict()` to initialize an empty dict. - Pass the -fsanitize options to both the compile and link commands. git-svn-id: https://llvm.org/svn/llvm-project/libcxx/trunk@226575 91177308-0d34-0410-b5e6-96231b3b80d8
114 lines
4.2 KiB
Python
114 lines
4.2 KiB
Python
|
|
import lit.util
|
|
|
|
|
|
class CXXCompiler(object):
|
|
def __init__(self, path, flags=[], compile_flags=[], link_flags=[], use_ccache=False):
|
|
self.path = path
|
|
self.flags = list(flags)
|
|
self.compile_flags = list(compile_flags)
|
|
self.link_flags = list(link_flags)
|
|
self.use_ccache = use_ccache
|
|
self.type = None
|
|
self.version = None
|
|
self._initTypeAndVersion()
|
|
|
|
def _initTypeAndVersion(self):
|
|
# Get compiler type and version
|
|
macros = self.dumpMacros()
|
|
if macros is None:
|
|
return
|
|
compiler_type = None
|
|
major_ver = minor_ver = patchlevel = None
|
|
if '__clang__' in macros.keys():
|
|
compiler_type = 'clang'
|
|
# Treat apple's llvm fork differently.
|
|
if '__apple_build_version__' in macros.keys():
|
|
compiler_type = 'apple-clang'
|
|
major_ver = macros['__clang_major__']
|
|
minor_ver = macros['__clang_minor__']
|
|
patchlevel = macros['__clang_patchlevel__']
|
|
elif '__GNUC__' in macros.keys():
|
|
compiler_type = 'gcc'
|
|
major_ver = macros['__GNUC__']
|
|
minor_ver = macros['__GNUC_MINOR__']
|
|
patchlevel = macros['__GNUC_PATCHLEVEL__']
|
|
self.type = compiler_type
|
|
self.version = (major_ver, minor_ver, patchlevel)
|
|
|
|
def _basicCmd(self, infiles, out, is_link=False):
|
|
cmd = []
|
|
if self.use_ccache and not is_link:
|
|
cmd += ['ccache']
|
|
cmd += [self.path]
|
|
if out is not None:
|
|
cmd += ['-o', out]
|
|
if isinstance(infiles, list):
|
|
cmd += infiles
|
|
elif isinstance(infiles, str):
|
|
cmd += [infiles]
|
|
else:
|
|
raise TypeError('infiles must be a string or list')
|
|
return cmd
|
|
|
|
def preprocessCmd(self, infiles, out=None, flags=[]):
|
|
cmd = self._basicCmd(infiles, out) + ['-x', 'c++', '-E']
|
|
cmd += self.flags + self.compile_flags + flags
|
|
return cmd
|
|
|
|
def compileCmd(self, infiles, out=None, flags=[]):
|
|
cmd = self._basicCmd(infiles, out) + ['-x', 'c++', '-c']
|
|
cmd += self.flags + self.compile_flags + flags
|
|
return cmd
|
|
|
|
def linkCmd(self, infiles, out=None, flags=[]):
|
|
cmd = self._basicCmd(infiles, out, is_link=True)
|
|
cmd += self.flags + self.link_flags + flags
|
|
return cmd
|
|
|
|
def compileLinkCmd(self, infiles, out=None, flags=[]):
|
|
cmd = self._basicCmd(infiles, out, is_link=True) + ['-x', 'c++']
|
|
cmd += self.flags + self.compile_flags + self.link_flags + flags
|
|
return cmd
|
|
|
|
def preprocess(self, infiles, out=None, flags=[], env=None, cwd=None):
|
|
cmd = self.preprocessCmd(infiles, out, flags)
|
|
out, err, rc = lit.util.executeCommand(cmd, env=env, cwd=cwd)
|
|
return cmd, out, err, rc
|
|
|
|
def compile(self, infiles, out=None, flags=[], env=None, cwd=None):
|
|
cmd = self.compileCmd(infiles, out, flags)
|
|
out, err, rc = lit.util.executeCommand(cmd, env=env, cwd=cwd)
|
|
return cmd, out, err, rc
|
|
|
|
def link(self, infiles, out=None, flags=[], env=None, cwd=None):
|
|
cmd = self.linkCmd(infiles, out, flags)
|
|
out, err, rc = lit.util.executeCommand(cmd, env=env, cwd=cwd)
|
|
return cmd, out, err, rc
|
|
|
|
def compileLink(self, infiles, out=None, flags=[], env=None, cwd=None):
|
|
cmd = self.compileLinkCmd(infiles, out, flags)
|
|
out, err, rc = lit.util.executeCommand(cmd, env=env, cwd=cwd)
|
|
return cmd, out, err, rc
|
|
|
|
def dumpMacros(self, infiles=None, flags=[], env=None, cwd=None):
|
|
if infiles is None:
|
|
infiles = '/dev/null'
|
|
flags = ['-dM'] + flags
|
|
cmd, out, err, rc = self.preprocess(infiles, flags=flags, env=env,
|
|
cwd=cwd)
|
|
if rc != 0:
|
|
return None
|
|
parsed_macros = {}
|
|
lines = [l.strip() for l in out.split('\n') if l.strip()]
|
|
for l in lines:
|
|
assert l.startswith('#define ')
|
|
l = l[len('#define '):]
|
|
macro, _, value = l.partition(' ')
|
|
parsed_macros[macro] = value
|
|
return parsed_macros
|
|
|
|
def getTriple(self):
|
|
cmd = [self.path] + self.flags + ['-dumpmachine']
|
|
return lit.util.capture(cmd).strip()
|