1  # -*- Python -*-
 2  
 3  # Common configuration for running leak detection tests under LSan/ASan.
 4  
 5  import os
 6  
 7  def get_required_attr(config, attr_name):
 8    attr_value = getattr(config, attr_name, None)
 9    if attr_value == None:
10      lit_config.fatal(
11        "No attribute %r in test configuration! You may need to run "
12        "tests from your build directory or add this attribute "
13        "to lit.site.cfg " % attr_name)
14    return attr_value
15  
16  # Setup source root.
17  config.test_source_root = os.path.dirname(__file__)
18  
19  # Choose between standalone and LSan+ASan modes.
20  lsan_lit_test_mode = get_required_attr(config, 'lsan_lit_test_mode')
21  if lsan_lit_test_mode == "Standalone":
22    config.name = "LeakSanitizer-Standalone"
23    lsan_cflags = ["-fsanitize=leak"]
24  elif lsan_lit_test_mode == "AddressSanitizer":
25    config.name = "LeakSanitizer-AddressSanitizer"
26    lsan_cflags = ["-fsanitize=address"]
27    config.available_features.add('asan')
28  else:
29    lit_config.fatal("Unknown LSan test mode: %r" % lsan_lit_test_mode)
30  
31  clang_cflags = ["-O0", "-m64"] + config.debug_info_flags
32  clang_cxxflags = config.cxx_mode_flags + clang_cflags
33  clang_lsan_cflags = clang_cflags + lsan_cflags
34  clang_lsan_cxxflags = clang_cxxflags + lsan_cflags
35  
36  config.clang_cflags = clang_cflags
37  config.clang_cxxflags = clang_cxxflags
38  
39  def build_invocation(compile_flags):
40    return " " + " ".join([config.clang] + compile_flags) + " "
41  
42  config.substitutions.append( ("%clang ", build_invocation(clang_cflags)) )
43  config.substitutions.append( ("%clangxx ", build_invocation(clang_cxxflags)) )
44  config.substitutions.append( ("%clang_lsan ", build_invocation(clang_lsan_cflags)) )
45  config.substitutions.append( ("%clangxx_lsan ", build_invocation(clang_lsan_cxxflags)) )
46  
47  # LeakSanitizer tests are currently supported on x86-64 Linux and mips64 Linux only.
48  if config.host_os not in ['Linux'] or config.host_arch not in ['x86_64', 'mips64']:
49    config.unsupported = True
50  
51  config.suffixes = ['.c', '.cc', '.cpp']