1from os import getenv, path
  2from subprocess import Popen, PIPE
  3from re import sub
  4
  5cc = getenv("CC")
  6
  7# Check if CC has options, as is the case in yocto, where it uses CC="cc --sysroot..."
  8cc_tokens = cc.split()
  9if len(cc_tokens) > 1:
 10    cc = cc_tokens[0]
 11    cc_options = " ".join([str(e) for e in cc_tokens[1:]]) + " "
 12else:
 13    cc_options = ""
 14
 15cc_is_clang = b"clang version" in Popen([cc, "-v"], stderr=PIPE).stderr.readline()
 16src_feature_tests  = getenv('srctree') + '/tools/build/feature'
 17
 18def clang_has_option(option):
 19    cc_output = Popen([cc, cc_options + option, path.join(src_feature_tests, "test-hello.c") ], stderr=PIPE).stderr.readlines()
 20    return [o for o in cc_output if ((b"unknown argument" in o) or (b"is not supported" in o))] == [ ]
 21
 22if cc_is_clang:
 23    from sysconfig import get_config_vars
 24    vars = get_config_vars()
 25    for var in ('CFLAGS', 'OPT'):
 26        vars[var] = sub("-specs=[^ ]+", "", vars[var])
 27        if not clang_has_option("-mcet"):
 28            vars[var] = sub("-mcet", "", vars[var])
 29        if not clang_has_option("-fcf-protection"):
 30            vars[var] = sub("-fcf-protection", "", vars[var])
 31        if not clang_has_option("-fstack-clash-protection"):
 32            vars[var] = sub("-fstack-clash-protection", "", vars[var])
 33        if not clang_has_option("-fstack-protector-strong"):
 34            vars[var] = sub("-fstack-protector-strong", "", vars[var])
 35        if not clang_has_option("-fno-semantic-interposition"):
 36            vars[var] = sub("-fno-semantic-interposition", "", vars[var])
 37        if not clang_has_option("-ffat-lto-objects"):
 38            vars[var] = sub("-ffat-lto-objects", "", vars[var])
 39        if not clang_has_option("-ftree-loop-distribute-patterns"):
 40            vars[var] = sub("-ftree-loop-distribute-patterns", "", vars[var])
 41        if not clang_has_option("-gno-variable-location-views"):
 42            vars[var] = sub("-gno-variable-location-views", "", vars[var])
 43
 44from setuptools import setup, Extension
 45
 46from setuptools.command.build_ext   import build_ext   as _build_ext
 47from setuptools.command.install_lib import install_lib as _install_lib
 48
 49class build_ext(_build_ext):
 50    def finalize_options(self):
 51        _build_ext.finalize_options(self)
 52        self.build_lib  = build_lib
 53        self.build_temp = build_tmp
 54
 55class install_lib(_install_lib):
 56    def finalize_options(self):
 57        _install_lib.finalize_options(self)
 58        self.build_dir = build_lib
 59
 60
 61cflags = getenv('CFLAGS', '').split()
 62# switch off several checks (need to be at the end of cflags list)
 63cflags += ['-fno-strict-aliasing', '-Wno-write-strings', '-Wno-unused-parameter', '-Wno-redundant-decls', '-DPYTHON_PERF' ]
 64if cc_is_clang:
 65    cflags += ["-Wno-unused-command-line-argument" ]
 66else:
 67    cflags += ['-Wno-cast-function-type' ]
 68
 69# The python headers have mixed code with declarations (decls after asserts, for instance)
 70cflags += [ "-Wno-declaration-after-statement" ]
 71
 72src_perf  = getenv('srctree') + '/tools/perf'
 73build_lib = getenv('PYTHON_EXTBUILD_LIB')
 74build_tmp = getenv('PYTHON_EXTBUILD_TMP')
 75libtraceevent = getenv('LIBTRACEEVENT')
 76libapikfs = getenv('LIBAPI')
 77libperf = getenv('LIBPERF')
 78
 79ext_sources = [f.strip() for f in open('util/python-ext-sources')
 80				if len(f.strip()) > 0 and f[0] != '#']
 81
 82extra_libraries = []
 83
 84if '-DHAVE_LIBTRACEEVENT' in cflags:
 85    extra_libraries += [ 'traceevent' ]
 86else:
 87    ext_sources.remove('util/trace-event.c')
 88    ext_sources.remove('util/trace-event-parse.c')
 89
 90# use full paths with source files
 91ext_sources = list(map(lambda x: '%s/%s' % (src_perf, x) , ext_sources))
 92
 93if '-DHAVE_LIBNUMA_SUPPORT' in cflags:
 94    extra_libraries += [ 'numa' ]
 95if '-DHAVE_LIBCAP_SUPPORT' in cflags:
 96    extra_libraries += [ 'cap' ]
 97
 98perf = Extension('perf',
 99		  sources = ext_sources,
100		  include_dirs = ['util/include'],
101		  libraries = extra_libraries,
102		  extra_compile_args = cflags,
103		  extra_objects = [ x for x in [libtraceevent, libapikfs, libperf]
104                                    if x is not None],
105                 )
106
107setup(name='perf',
108      version='0.1',
109      description='Interface with the Linux profiling infrastructure',
110      author='Arnaldo Carvalho de Melo',
111      author_email='acme@redhat.com',
112      license='GPLv2',
113      url='http://perf.wiki.kernel.org',
114      ext_modules=[perf],
115      cmdclass={'build_ext': build_ext, 'install_lib': install_lib})