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