xref: /llvm-project/compiler-rt/test/cfi/lit.cfg.py (revision f98ee40f4b5d7474fc67e82824bf6abbaedb7b1c) 
1import lit.formats
2import os
3
4config.name = "cfi" + config.name_suffix
5config.suffixes = [".c", ".cpp", ".test"]
6config.test_source_root = os.path.dirname(__file__)
7
8
9def build_invocation(compile_flags):
10    return " " + " ".join([config.clang] + compile_flags) + " "
11
12
13clang = build_invocation([config.target_cflags])
14clangxx = build_invocation([config.target_cflags] + config.cxx_mode_flags)
15
16config.substitutions.append((r"%clang ", clang + " "))
17config.substitutions.append((r"%clangxx ", clangxx + " "))
18
19if "darwin" in config.available_features:
20    # -fsanitize=cfi is not supported on Darwin hosts
21    config.unsupported = True
22elif config.lto_supported:
23    clang_cfi = clang + "-fsanitize=cfi "
24
25    if config.cfi_lit_test_mode == "Devirt":
26        config.available_features.add("devirt")
27        clang_cfi += "-fwhole-program-vtables "
28        config.substitutions.append((r"%expect_crash_unless_devirt ", ""))
29    else:
30        config.substitutions.append(
31            (r"%expect_crash_unless_devirt ", config.expect_crash)
32        )
33
34    cxx = " ".join(config.cxx_mode_flags) + " "
35    diag = "-fno-sanitize-trap=cfi -fsanitize-recover=cfi "
36    non_dso = "-fvisibility=hidden "
37    dso = "-fsanitize-cfi-cross-dso -fvisibility=default "
38    if config.android:
39        dso += "-include " + config.test_source_root + "/cross-dso/util/cfi_stubs.h "
40    config.substitutions.append((r"%clang_cfi ", clang_cfi + non_dso))
41    config.substitutions.append((r"%clangxx_cfi ", clang_cfi + cxx + non_dso))
42    config.substitutions.append((r"%clang_cfi_diag ", clang_cfi + non_dso + diag))
43    config.substitutions.append(
44        (r"%clangxx_cfi_diag ", clang_cfi + cxx + non_dso + diag)
45    )
46    config.substitutions.append((r"%clangxx_cfi_dso ", clang_cfi + cxx + dso))
47    config.substitutions.append(
48        (r"%clangxx_cfi_dso_diag ", clang_cfi + cxx + dso + diag)
49    )
50    config.substitutions.append(
51        (r"%debug_info_flags", " ".join(config.debug_info_flags))
52    )
53else:
54    config.unsupported = True
55
56if config.default_sanitizer_opts:
57    config.environment["UBSAN_OPTIONS"] = ":".join(config.default_sanitizer_opts)
58
59if lit_config.params.get("check_supported", None) and config.unsupported:
60    raise BaseException("Tests unsupported")
61