1# -*- Python -*-
2
3import os
4
5def get_required_attr(config, attr_name):
6  attr_value = getattr(config, attr_name, None)
7  if attr_value == None:
8    lit_config.fatal(
9      "No attribute %r in test configuration! You may need to run "
10      "tests from your build directory or add this attribute "
11      "to lit.site.cfg " % attr_name)
12  return attr_value
13
14# Setup source root.
15config.test_source_root = os.path.dirname(__file__)
16
17# Choose between standalone and UBSan+ASan modes.
18ubsan_lit_test_mode = get_required_attr(config, 'ubsan_lit_test_mode')
19if ubsan_lit_test_mode == "Standalone":
20  config.name = 'UBSan-Standalone-' + config.target_arch
21  config.available_features.add("ubsan-standalone")
22  clang_ubsan_cflags = []
23elif ubsan_lit_test_mode == "AddressSanitizer":
24  if config.host_os == 'Darwin':
25    # ubsan-asan doesn't yet work on Darwin,
26    # see http://llvm.org/bugs/show_bug.cgi?id=21112.
27    config.unsupported = True
28  config.name = 'UBSan-ASan-' + config.target_arch
29  config.available_features.add("ubsan-asan")
30  clang_ubsan_cflags = ["-fsanitize=address"]
31  config.environment['ASAN_OPTIONS'] = 'detect_leaks=0'
32else:
33  lit_config.fatal("Unknown UBSan test mode: %r" % ubsan_lit_test_mode)
34
35def build_invocation(compile_flags):
36  return " " + " ".join([config.clang] + compile_flags) + " "
37
38target_cflags = [get_required_attr(config, "target_cflags")]
39clang_ubsan_cflags += target_cflags
40clang_ubsan_cxxflags = config.cxx_mode_flags + clang_ubsan_cflags
41
42# Define %clang and %clangxx substitutions to use in test RUN lines.
43config.substitutions.append( ("%clang ", build_invocation(clang_ubsan_cflags)) )
44config.substitutions.append( ("%clangxx ", build_invocation(clang_ubsan_cxxflags)) )
45
46# Default test suffixes.
47config.suffixes = ['.c', '.cc', '.cpp']
48
49# Check that the host supports UndefinedBehaviorSanitizer tests
50if config.host_os not in ['Linux', 'Darwin', 'FreeBSD']:
51  config.unsupported = True
52
53# Allow tests to use REQUIRES=stable-runtime.  For use when you cannot use XFAIL
54# because the test hangs or fails on one configuration and not the other.
55if config.target_arch.startswith('arm') == False:
56  config.available_features.add('stable-runtime')
57