114 lines · python
1# -*- Python -*-2 3import os4import platform5import re6 7import lit.formats8 9 10def get_required_attr(config, attr_name):11 attr_value = getattr(config, attr_name, None)12 if attr_value is None:13 lit_config.fatal(14 "No attribute %r in test configuration! You may need to run "15 "tests from your build directory or add this attribute "16 "to lit.site.cfg.py " % attr_name17 )18 return attr_value19 20 21# Setup config name.22config.name = "MemProfiler" + config.name_suffix23 24# Platform-specific default MEMPROF_OPTIONS for lit tests.25default_memprof_opts = list(config.default_sanitizer_opts)26 27default_memprof_opts_str = ":".join(default_memprof_opts)28if default_memprof_opts_str:29 config.environment["MEMPROF_OPTIONS"] = default_memprof_opts_str30 default_memprof_opts_str += ":"31config.substitutions.append(32 ("%env_memprof_opts=", "env MEMPROF_OPTIONS=" + default_memprof_opts_str)33)34 35# Setup source root.36config.test_source_root = os.path.dirname(__file__)37 38libdl_flag = "-ldl"39 40# Setup default compiler flags used with -fmemory-profile option.41# FIXME: Review the set of required flags and check if it can be reduced.42target_cflags = [get_required_attr(config, "target_cflags")]43target_cxxflags = config.cxx_mode_flags + target_cflags44clang_memprof_static_cflags = (45 [46 "-fmemory-profile",47 "-mno-omit-leaf-frame-pointer",48 "-fno-omit-frame-pointer",49 "-fno-optimize-sibling-calls",50 ]51 + config.debug_info_flags52 + target_cflags53)54clang_memprof_static_cxxflags = config.cxx_mode_flags + clang_memprof_static_cflags55 56memprof_dynamic_flags = []57if config.memprof_dynamic:58 memprof_dynamic_flags = ["-shared-libsan"]59 config.available_features.add("memprof-dynamic-runtime")60else:61 config.available_features.add("memprof-static-runtime")62clang_memprof_cflags = clang_memprof_static_cflags + memprof_dynamic_flags63clang_memprof_cxxflags = clang_memprof_static_cxxflags + memprof_dynamic_flags64 65 66def build_invocation(compile_flags):67 return " " + " ".join([config.clang] + compile_flags) + " "68 69 70config.substitutions.append(("%clang ", build_invocation(target_cflags)))71config.substitutions.append(("%clangxx ", build_invocation(target_cxxflags)))72config.substitutions.append(("%clang_memprof ", build_invocation(clang_memprof_cflags)))73config.substitutions.append(74 ("%clangxx_memprof ", build_invocation(clang_memprof_cxxflags))75)76if config.memprof_dynamic:77 shared_libmemprof_path = os.path.join(78 config.compiler_rt_libdir,79 "libclang_rt.memprof{}.so".format(config.target_suffix),80 )81 config.substitutions.append(("%shared_libmemprof", shared_libmemprof_path))82 config.substitutions.append(83 ("%clang_memprof_static ", build_invocation(clang_memprof_static_cflags))84 )85 config.substitutions.append(86 ("%clangxx_memprof_static ", build_invocation(clang_memprof_static_cxxflags))87 )88 89config.substitutions.append(("%libdl", libdl_flag))90 91config.available_features.add("memprof-" + config.bits + "-bits")92 93config.available_features.add("fast-unwinder-works")94 95# Set LD_LIBRARY_PATH to pick dynamic runtime up properly.96new_ld_library_path = os.path.pathsep.join(97 (config.compiler_rt_libdir, config.environment.get("LD_LIBRARY_PATH", ""))98)99config.environment["LD_LIBRARY_PATH"] = new_ld_library_path100 101# Default test suffixes.102config.suffixes = [".c", ".cpp"]103 104config.substitutions.append(("%fPIC", "-fPIC"))105config.substitutions.append(("%fPIE", "-fPIE"))106config.substitutions.append(("%pie", "-pie"))107 108# Only run the tests on supported OSs.109if config.target_os not in ["Linux"]:110 config.unsupported = True111 112if not config.parallelism_group:113 config.parallelism_group = "shadow-memory"114