Stephen Hines | 2d1fdb2 | 2014-05-28 23:58:16 -0700 | [diff] [blame] | 1 | # -*- Python -*- |
| 2 | |
| 3 | # Common configuration for running leak detection tests under LSan/ASan. |
| 4 | |
| 5 | import os |
| 6 | |
| 7 | def get_required_attr(config, attr_name): |
| 8 | attr_value = getattr(config, attr_name, None) |
| 9 | if attr_value == None: |
| 10 | lit_config.fatal( |
| 11 | "No attribute %r in test configuration! You may need to run " |
| 12 | "tests from your build directory or add this attribute " |
| 13 | "to lit.site.cfg " % attr_name) |
| 14 | return attr_value |
| 15 | |
| 16 | # Setup source root. |
| 17 | config.test_source_root = os.path.dirname(__file__) |
| 18 | |
| 19 | # Choose between standalone and LSan+ASan modes. |
| 20 | lsan_lit_test_mode = get_required_attr(config, 'lsan_lit_test_mode') |
| 21 | if lsan_lit_test_mode == "Standalone": |
| 22 | config.name = "LeakSanitizer-Standalone" |
| 23 | lsan_cflags = ["-fsanitize=leak"] |
| 24 | elif lsan_lit_test_mode == "AddressSanitizer": |
| 25 | config.name = "LeakSanitizer-AddressSanitizer" |
| 26 | lsan_cflags = ["-fsanitize=address"] |
| 27 | config.available_features.add('asan') |
Stephen Hines | 2d1fdb2 | 2014-05-28 23:58:16 -0700 | [diff] [blame] | 28 | else: |
| 29 | lit_config.fatal("Unknown LSan test mode: %r" % lsan_lit_test_mode) |
Pirama Arumuga Nainar | c58a436 | 2016-09-19 23:00:23 -0700 | [diff] [blame] | 30 | config.name += config.name_suffix |
Stephen Hines | 2d1fdb2 | 2014-05-28 23:58:16 -0700 | [diff] [blame] | 31 | |
Pirama Arumuga Nainar | c58a436 | 2016-09-19 23:00:23 -0700 | [diff] [blame] | 32 | clang_cflags = ["-O0", config.target_cflags] + config.debug_info_flags |
Stephen Hines | 2d1fdb2 | 2014-05-28 23:58:16 -0700 | [diff] [blame] | 33 | clang_cxxflags = config.cxx_mode_flags + clang_cflags |
| 34 | clang_lsan_cflags = clang_cflags + lsan_cflags |
| 35 | clang_lsan_cxxflags = clang_cxxflags + lsan_cflags |
| 36 | |
| 37 | config.clang_cflags = clang_cflags |
| 38 | config.clang_cxxflags = clang_cxxflags |
| 39 | |
| 40 | def build_invocation(compile_flags): |
| 41 | return " " + " ".join([config.clang] + compile_flags) + " " |
| 42 | |
| 43 | config.substitutions.append( ("%clang ", build_invocation(clang_cflags)) ) |
| 44 | config.substitutions.append( ("%clangxx ", build_invocation(clang_cxxflags)) ) |
| 45 | config.substitutions.append( ("%clang_lsan ", build_invocation(clang_lsan_cflags)) ) |
| 46 | config.substitutions.append( ("%clangxx_lsan ", build_invocation(clang_lsan_cxxflags)) ) |
| 47 | |
Stephen Hines | 86277eb | 2015-03-23 12:06:32 -0700 | [diff] [blame] | 48 | # LeakSanitizer tests are currently supported on x86-64 Linux and mips64 Linux only. |
| 49 | if config.host_os not in ['Linux'] or config.host_arch not in ['x86_64', 'mips64']: |
Stephen Hines | 2d1fdb2 | 2014-05-28 23:58:16 -0700 | [diff] [blame] | 50 | config.unsupported = True |
| 51 | |
| 52 | config.suffixes = ['.c', '.cc', '.cpp'] |