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