1from os import getenv 2from subprocess import Popen, PIPE 3from re import sub 4 5cc = getenv("CC") 6cc_is_clang = b"clang version" in Popen([cc.split()[0], "-v"], stderr=PIPE).stderr.readline() 7 8def clang_has_option(option): 9 return [o for o in Popen([cc, option], stderr=PIPE).stderr.readlines() if b"unknown argument" in o] == [ ] 10 11if cc_is_clang: 12 from distutils.sysconfig import get_config_vars 13 vars = get_config_vars() 14 for var in ('CFLAGS', 'OPT'): 15 vars[var] = sub("-specs=[^ ]+", "", vars[var]) 16 if not clang_has_option("-mcet"): 17 vars[var] = sub("-mcet", "", vars[var]) 18 if not clang_has_option("-fcf-protection"): 19 vars[var] = sub("-fcf-protection", "", vars[var]) 20 if not clang_has_option("-fstack-clash-protection"): 21 vars[var] = sub("-fstack-clash-protection", "", vars[var]) 22 if not clang_has_option("-fstack-protector-strong"): 23 vars[var] = sub("-fstack-protector-strong", "", vars[var]) 24 if not clang_has_option("-fno-semantic-interposition"): 25 vars[var] = sub("-fno-semantic-interposition", "", vars[var]) 26 27from distutils.core import setup, Extension 28 29from distutils.command.build_ext import build_ext as _build_ext 30from distutils.command.install_lib import install_lib as _install_lib 31 32class build_ext(_build_ext): 33 def finalize_options(self): 34 _build_ext.finalize_options(self) 35 self.build_lib = build_lib 36 self.build_temp = build_tmp 37 38class install_lib(_install_lib): 39 def finalize_options(self): 40 _install_lib.finalize_options(self) 41 self.build_dir = build_lib 42 43 44cflags = getenv('CFLAGS', '').split() 45# switch off several checks (need to be at the end of cflags list) 46cflags += ['-fno-strict-aliasing', '-Wno-write-strings', '-Wno-unused-parameter', '-Wno-redundant-decls', '-DPYTHON_PERF' ] 47if not cc_is_clang: 48 cflags += ['-Wno-cast-function-type' ] 49 50src_perf = getenv('srctree') + '/tools/perf' 51build_lib = getenv('PYTHON_EXTBUILD_LIB') 52build_tmp = getenv('PYTHON_EXTBUILD_TMP') 53libtraceevent = getenv('LIBTRACEEVENT') 54libapikfs = getenv('LIBAPI') 55libperf = getenv('LIBPERF') 56 57ext_sources = [f.strip() for f in open('util/python-ext-sources') 58 if len(f.strip()) > 0 and f[0] != '#'] 59 60# use full paths with source files 61ext_sources = list(map(lambda x: '%s/%s' % (src_perf, x) , ext_sources)) 62 63extra_libraries = [] 64if '-DHAVE_LIBNUMA_SUPPORT' in cflags: 65 extra_libraries = [ 'numa' ] 66if '-DHAVE_LIBCAP_SUPPORT' in cflags: 67 extra_libraries += [ 'cap' ] 68 69perf = Extension('perf', 70 sources = ext_sources, 71 include_dirs = ['util/include'], 72 libraries = extra_libraries, 73 extra_compile_args = cflags, 74 extra_objects = [libtraceevent, libapikfs, libperf], 75 ) 76 77setup(name='perf', 78 version='0.1', 79 description='Interface with the Linux profiling infrastructure', 80 author='Arnaldo Carvalho de Melo', 81 author_email='acme@redhat.com', 82 license='GPLv2', 83 url='http://perf.wiki.kernel.org', 84 ext_modules=[perf], 85 cmdclass={'build_ext': build_ext, 'install_lib': install_lib}) 86