61 lines · python
1import lit.formats2import os3 4config.name = "cfi" + config.name_suffix5config.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 hosts21 config.unsupported = True22elif 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 = True55 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