11adb55b1SChris Appleimport os 21adb55b1SChris Apple 31adb55b1SChris Apple# Setup config name. 41adb55b1SChris Appleconfig.name = "RTSAN" + config.name_suffix 51adb55b1SChris Apple 6*aa43f3abSChris Apple 7*aa43f3abSChris Appledefault_rtsan_opts = "atexit_sleep_ms=0" 8*aa43f3abSChris Apple 9*aa43f3abSChris Appleif config.host_os == "Darwin": 10*aa43f3abSChris Apple # On Darwin, we default to `abort_on_error=1`, which would make tests run 11*aa43f3abSChris Apple # much slower. Let's override this and run lit tests with 'abort_on_error=0'. 12*aa43f3abSChris Apple default_rtsan_opts += ":abort_on_error=0" 13*aa43f3abSChris Apple 14*aa43f3abSChris Appleif default_rtsan_opts: 15*aa43f3abSChris Apple config.environment["RTSAN_OPTIONS"] = default_rtsan_opts 16*aa43f3abSChris Apple default_rtsan_opts += ":" 17*aa43f3abSChris Apple 18*aa43f3abSChris Appleconfig.substitutions.append( 19*aa43f3abSChris Apple ("%env_rtsan_opts=", "env RTSAN_OPTIONS=" + default_rtsan_opts) 20*aa43f3abSChris Apple) 21*aa43f3abSChris Apple 221adb55b1SChris Apple# Setup source root. 231adb55b1SChris Appleconfig.test_source_root = os.path.dirname(__file__) 241adb55b1SChris Apple 251adb55b1SChris Apple# Setup default compiler flags use with -frtsan-instrument option. 261adb55b1SChris Appleclang_rtsan_cflags = ["-frtsan-instrument", config.target_cflags] 271adb55b1SChris Apple 281adb55b1SChris Appleclang_rtsan_cxxflags = config.cxx_mode_flags + clang_rtsan_cflags 291adb55b1SChris Apple 301adb55b1SChris Apple 311adb55b1SChris Appledef build_invocation(compile_flags): 321adb55b1SChris Apple return " " + " ".join([config.clang] + compile_flags) + " " 331adb55b1SChris Apple 341adb55b1SChris Apple 351adb55b1SChris Apple# Assume that llvm-rtsan is in the config.llvm_tools_dir. 361adb55b1SChris Applellvm_rtsan = os.path.join(config.llvm_tools_dir, "llvm-rtsan") 371adb55b1SChris Apple 381adb55b1SChris Apple# Setup substitutions. 391adb55b1SChris Appleif config.host_os == "Linux": 401adb55b1SChris Apple libdl_flag = "-ldl" 411adb55b1SChris Appleelse: 421adb55b1SChris Apple libdl_flag = "" 431adb55b1SChris Apple 441adb55b1SChris Appleconfig.substitutions.append(("%clang ", build_invocation([config.target_cflags]))) 451adb55b1SChris Appleconfig.substitutions.append( 461adb55b1SChris Apple ("%clangxx ", build_invocation(config.cxx_mode_flags + [config.target_cflags])) 471adb55b1SChris Apple) 481adb55b1SChris Appleconfig.substitutions.append(("%clang_rtsan ", build_invocation(clang_rtsan_cflags))) 491adb55b1SChris Appleconfig.substitutions.append(("%clangxx_rtsan", build_invocation(clang_rtsan_cxxflags))) 501adb55b1SChris Appleconfig.substitutions.append(("%llvm_rtsan", llvm_rtsan)) 511adb55b1SChris Apple 521adb55b1SChris Apple# Default test suffixes. 531adb55b1SChris Appleconfig.suffixes = [".c", ".cpp"] 541adb55b1SChris Apple 551adb55b1SChris Appleif config.host_os not in ["Darwin", "FreeBSD", "Linux", "NetBSD", "OpenBSD"]: 561adb55b1SChris Apple config.unsupported = True 571adb55b1SChris Appleelif "64" not in config.host_arch: 581adb55b1SChris Apple if "arm" in config.host_arch: 591adb55b1SChris Apple if "-mthumb" in config.target_cflags: 601adb55b1SChris Apple config.unsupported = True 611adb55b1SChris Apple else: 621adb55b1SChris Apple config.unsupported = True 631adb55b1SChris Apple 641adb55b1SChris Appleif config.host_os == "NetBSD": 651adb55b1SChris Apple config.substitutions.insert(0, ("%run", config.netbsd_nomprotect_prefix)) 66