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.py " % attr_name)
12  return attr_value
13
14# Setup config name.
15config.name = 'UBSan-' + config.name_suffix
16
17# Setup source root.
18config.test_source_root = os.path.dirname(__file__)
19
20default_ubsan_opts = list(config.default_sanitizer_opts)
21# Choose between standalone and UBSan+ASan modes.
22ubsan_lit_test_mode = get_required_attr(config, 'ubsan_lit_test_mode')
23if ubsan_lit_test_mode == "Standalone":
24  config.available_features.add("ubsan-standalone")
25  clang_ubsan_cflags = []
26elif ubsan_lit_test_mode == "StandaloneStatic":
27  config.available_features.add("ubsan-standalone-static")
28  clang_ubsan_cflags = ['-static-libsan']
29elif ubsan_lit_test_mode == "AddressSanitizer":
30  config.available_features.add("ubsan-asan")
31  clang_ubsan_cflags = ["-fsanitize=address"]
32  default_ubsan_opts += ['detect_leaks=0']
33elif ubsan_lit_test_mode == "MemorySanitizer":
34  config.available_features.add("ubsan-msan")
35  clang_ubsan_cflags = ["-fsanitize=memory"]
36elif ubsan_lit_test_mode == "ThreadSanitizer":
37  config.available_features.add("ubsan-tsan")
38  clang_ubsan_cflags = ["-fsanitize=thread"]
39else:
40  lit_config.fatal("Unknown UBSan test mode: %r" % ubsan_lit_test_mode)
41
42# Platform-specific default for lit tests.
43if config.target_arch == 's390x':
44  # On SystemZ we need -mbackchain to make the fast unwinder work.
45  clang_ubsan_cflags.append("-mbackchain")
46
47default_ubsan_opts_str = ':'.join(default_ubsan_opts)
48if default_ubsan_opts_str:
49  config.environment['UBSAN_OPTIONS'] = default_ubsan_opts_str
50  default_ubsan_opts_str += ':'
51# Substitution to setup UBSAN_OPTIONS in portable way.
52config.substitutions.append(('%env_ubsan_opts=',
53                             'env UBSAN_OPTIONS=' + default_ubsan_opts_str))
54
55def build_invocation(compile_flags):
56  return " " + " ".join([config.clang] + compile_flags) + " "
57
58target_cflags = [get_required_attr(config, "target_cflags")]
59clang_ubsan_cflags += target_cflags
60clang_ubsan_cxxflags = config.cxx_mode_flags + clang_ubsan_cflags
61
62# Define %clang and %clangxx substitutions to use in test RUN lines.
63config.substitutions.append( ("%clang ", build_invocation(clang_ubsan_cflags)) )
64config.substitutions.append( ("%clangxx ", build_invocation(clang_ubsan_cxxflags)) )
65config.substitutions.append( ("%gmlt ", " ".join(config.debug_info_flags) + " ") )
66
67# Default test suffixes.
68config.suffixes = ['.c', '.cpp', '.m']
69
70# Check that the host supports UndefinedBehaviorSanitizer tests
71if config.host_os not in ['Linux', 'Darwin', 'FreeBSD', 'Windows', 'NetBSD', 'SunOS', 'OpenBSD']:
72  config.unsupported = True
73
74config.available_features.add('arch=' + config.target_arch)
75
76config.excludes = ['Inputs']
77
78if ubsan_lit_test_mode in ['AddressSanitizer', 'MemorySanitizer', 'ThreadSanitizer']:
79  if not config.parallelism_group:
80    config.parallelism_group = 'shadow-memory'
81  if config.host_os == 'NetBSD':
82    config.substitutions.insert(0, ('%run', config.netbsd_noaslr_prefix))
83