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 25from distutils.core import setup, Extension 26 27from distutils.command.build_ext import build_ext as _build_ext 28from distutils.command.install_lib import install_lib as _install_lib 29 30class build_ext(_build_ext): 31 def finalize_options(self): 32 _build_ext.finalize_options(self) 33 self.build_lib = build_lib 34 self.build_temp = build_tmp 35 36class install_lib(_install_lib): 37 def finalize_options(self): 38 _install_lib.finalize_options(self) 39 self.build_dir = build_lib 40 41 42cflags = getenv('CFLAGS', '').split() 43# switch off several checks (need to be at the end of cflags list) 44cflags += ['-fno-strict-aliasing', '-Wno-write-strings', '-Wno-unused-parameter', '-Wno-redundant-decls' ] 45if not cc_is_clang: 46 cflags += ['-Wno-cast-function-type' ] 47 48src_perf = getenv('srctree') + '/tools/perf' 49build_lib = getenv('PYTHON_EXTBUILD_LIB') 50build_tmp = getenv('PYTHON_EXTBUILD_TMP') 51libtraceevent = getenv('LIBTRACEEVENT') 52libapikfs = getenv('LIBAPI') 53libperf = getenv('LIBPERF') 54 55ext_sources = [f.strip() for f in open('util/python-ext-sources') 56 if len(f.strip()) > 0 and f[0] != '#'] 57 58# use full paths with source files 59ext_sources = list(map(lambda x: '%s/%s' % (src_perf, x) , ext_sources)) 60 61extra_libraries = [] 62if '-DHAVE_LIBNUMA_SUPPORT' in cflags: 63 extra_libraries = [ 'numa' ] 64if '-DHAVE_LIBCAP_SUPPORT' in cflags: 65 extra_libraries += [ 'cap' ] 66 67perf = Extension('perf', 68 sources = ext_sources, 69 include_dirs = ['util/include'], 70 libraries = extra_libraries, 71 extra_compile_args = cflags, 72 extra_objects = [libtraceevent, libapikfs, libperf], 73 ) 74 75setup(name='perf', 76 version='0.1', 77 description='Interface with the Linux profiling infrastructure', 78 author='Arnaldo Carvalho de Melo', 79 author_email='acme@redhat.com', 80 license='GPLv2', 81 url='http://perf.wiki.kernel.org', 82 ext_modules=[perf], 83 cmdclass={'build_ext': build_ext, 'install_lib': install_lib}) 84