98 lines · python
1# -*- Python -*-2 3import os4 5 6def get_required_attr(config, attr_name):7 attr_value = getattr(config, attr_name, None)8 if attr_value is None:9 lit_config.fatal(10 "No attribute %r in test configuration! You may need to run "11 "tests from your build directory or add this attribute "12 "to lit.site.cfg.py " % attr_name13 )14 return attr_value15 16 17# Setup config name.18config.name = "UBSan-" + config.name_suffix19 20# Setup source root.21config.test_source_root = os.path.dirname(__file__)22 23default_ubsan_opts = list(config.default_sanitizer_opts)24# Choose between standalone and UBSan+ASan modes.25ubsan_lit_test_mode = get_required_attr(config, "ubsan_lit_test_mode")26if ubsan_lit_test_mode == "Standalone":27 config.available_features.add("ubsan-standalone")28 clang_ubsan_cflags = []29elif ubsan_lit_test_mode == "StandaloneStatic":30 config.available_features.add("ubsan-standalone-static")31 clang_ubsan_cflags = ["-static-libsan"]32elif ubsan_lit_test_mode == "AddressSanitizer":33 config.available_features.add("ubsan-asan")34 clang_ubsan_cflags = ["-fsanitize=address"]35 default_ubsan_opts += ["detect_leaks=0"]36elif ubsan_lit_test_mode == "MemorySanitizer":37 config.available_features.add("ubsan-msan")38 clang_ubsan_cflags = ["-fsanitize=memory"]39elif ubsan_lit_test_mode == "ThreadSanitizer":40 config.available_features.add("ubsan-tsan")41 clang_ubsan_cflags = ["-fsanitize=thread"]42elif ubsan_lit_test_mode == "TypeSanitizer":43 config.available_features.add("ubsan-tysan")44 clang_ubsan_cflags = ["-fsanitize=type"]45else:46 lit_config.fatal("Unknown UBSan test mode: %r" % ubsan_lit_test_mode)47 48# Platform-specific default for lit tests.49if config.target_arch == "s390x":50 # On SystemZ we need -mbackchain to make the fast unwinder work.51 clang_ubsan_cflags.append("-mbackchain")52 53default_ubsan_opts_str = ":".join(default_ubsan_opts)54if default_ubsan_opts_str:55 config.environment["UBSAN_OPTIONS"] = default_ubsan_opts_str56 default_ubsan_opts_str += ":"57# Substitution to setup UBSAN_OPTIONS in portable way.58config.substitutions.append(59 ("%env_ubsan_opts=", "env UBSAN_OPTIONS=" + default_ubsan_opts_str)60)61 62 63def build_invocation(compile_flags):64 return " " + " ".join([config.clang] + compile_flags) + " "65 66 67target_cflags = [get_required_attr(config, "target_cflags")]68clang_ubsan_cflags += target_cflags69clang_ubsan_cxxflags = config.cxx_mode_flags + clang_ubsan_cflags70 71# Define %clang and %clangxx substitutions to use in test RUN lines.72config.substitutions.append(("%clang ", build_invocation(clang_ubsan_cflags)))73config.substitutions.append(("%clangxx ", build_invocation(clang_ubsan_cxxflags)))74config.substitutions.append(("%gmlt ", " ".join(config.debug_info_flags) + " "))75 76# Default test suffixes.77config.suffixes = [".c", ".cpp", ".m"]78 79# Check that the host supports UndefinedBehaviorSanitizer tests80if config.target_os not in [81 "Linux",82 "Darwin",83 "FreeBSD",84 "Windows",85 "NetBSD",86 "SunOS",87 "OpenBSD",88]:89 config.unsupported = True90 91config.excludes = ["Inputs"]92 93if ubsan_lit_test_mode in ["AddressSanitizer", "MemorySanitizer", "ThreadSanitizer"]:94 if not config.parallelism_group:95 config.parallelism_group = "shadow-memory"96 if config.target_os == "NetBSD":97 config.substitutions.insert(0, ("%run", config.netbsd_noaslr_prefix))98