f98ee40f4b
This is an ongoing series of commits that are reformatting our Python code. This catches the last of the python files to reformat. Since they where so few I bunched them together. Reformatting is done with `black`. If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run git checkout --ours <yourfile> and then reformat it with black. If you run into any problems, post to discourse about it and we will try to help. RFC Thread below: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, #libc, Mordante, sivachandra Differential Revision: https://reviews.llvm.org/D150784
138 lines
5.0 KiB
Python
138 lines
5.0 KiB
Python
# -*- Python -*-
|
|
|
|
# Common configuration for running leak detection tests under LSan/ASan.
|
|
|
|
import os
|
|
import re
|
|
|
|
import lit.util
|
|
|
|
|
|
def get_required_attr(config, attr_name):
|
|
attr_value = getattr(config, attr_name, None)
|
|
if attr_value == None:
|
|
lit_config.fatal(
|
|
"No attribute %r in test configuration! You may need to run "
|
|
"tests from your build directory or add this attribute "
|
|
"to lit.site.cfg.py " % attr_name
|
|
)
|
|
return attr_value
|
|
|
|
|
|
# Setup source root.
|
|
config.test_source_root = os.path.dirname(__file__)
|
|
|
|
# Choose between standalone and LSan+(ASan|HWAsan) modes.
|
|
lsan_lit_test_mode = get_required_attr(config, "lsan_lit_test_mode")
|
|
target_arch = getattr(config, "target_arch", None)
|
|
|
|
if lsan_lit_test_mode == "Standalone":
|
|
config.name = "LeakSanitizer-Standalone"
|
|
lsan_cflags = ["-fsanitize=leak"]
|
|
config.available_features.add("lsan-standalone")
|
|
elif lsan_lit_test_mode == "AddressSanitizer":
|
|
config.name = "LeakSanitizer-AddressSanitizer"
|
|
lsan_cflags = ["-fsanitize=address"]
|
|
config.available_features.add("asan")
|
|
if config.host_os == "NetBSD":
|
|
config.substitutions.insert(0, ("%run", config.netbsd_noaslr_prefix))
|
|
elif lsan_lit_test_mode == "HWAddressSanitizer":
|
|
config.name = "LeakSanitizer-HWAddressSanitizer"
|
|
lsan_cflags = ["-fsanitize=hwaddress", "-fuse-ld=lld"]
|
|
if target_arch == "x86_64":
|
|
lsan_cflags = lsan_cflags + ["-fsanitize-hwaddress-experimental-aliasing"]
|
|
config.available_features.add("hwasan")
|
|
if config.host_os == "NetBSD":
|
|
config.substitutions.insert(0, ("%run", config.netbsd_noaslr_prefix))
|
|
else:
|
|
lit_config.fatal("Unknown LSan test mode: %r" % lsan_lit_test_mode)
|
|
config.name += config.name_suffix
|
|
|
|
# Platform-specific default LSAN_OPTIONS for lit tests.
|
|
default_common_opts_str = ":".join(list(config.default_sanitizer_opts))
|
|
default_lsan_opts = default_common_opts_str + ":detect_leaks=1"
|
|
if config.host_os == "Darwin":
|
|
# On Darwin, we default to `abort_on_error=1`, which would make tests run
|
|
# much slower. Let's override this and run lit tests with 'abort_on_error=0'.
|
|
# Also, make sure we do not overwhelm the syslog while testing.
|
|
default_lsan_opts += ":abort_on_error=0"
|
|
default_lsan_opts += ":log_to_syslog=0"
|
|
|
|
if default_lsan_opts:
|
|
config.environment["LSAN_OPTIONS"] = default_lsan_opts
|
|
default_lsan_opts += ":"
|
|
config.substitutions.append(
|
|
("%env_lsan_opts=", "env LSAN_OPTIONS=" + default_lsan_opts)
|
|
)
|
|
|
|
if lit.util.which("strace"):
|
|
config.available_features.add("strace")
|
|
|
|
clang_cflags = ["-O0", config.target_cflags] + config.debug_info_flags
|
|
if config.android:
|
|
clang_cflags = clang_cflags + ["-fno-emulated-tls"]
|
|
clang_cxxflags = config.cxx_mode_flags + clang_cflags
|
|
lsan_incdir = config.test_source_root + "/../"
|
|
clang_lsan_cflags = clang_cflags + lsan_cflags + ["-I%s" % lsan_incdir]
|
|
clang_lsan_cxxflags = clang_cxxflags + lsan_cflags + ["-I%s" % lsan_incdir]
|
|
|
|
config.clang_cflags = clang_cflags
|
|
config.clang_cxxflags = clang_cxxflags
|
|
|
|
|
|
def build_invocation(compile_flags):
|
|
return " " + " ".join([config.clang] + compile_flags) + " "
|
|
|
|
|
|
config.substitutions.append(("%clang ", build_invocation(clang_cflags)))
|
|
config.substitutions.append(("%clangxx ", build_invocation(clang_cxxflags)))
|
|
config.substitutions.append(("%clang_lsan ", build_invocation(clang_lsan_cflags)))
|
|
config.substitutions.append(("%clangxx_lsan ", build_invocation(clang_lsan_cxxflags)))
|
|
config.substitutions.append(("%clang_hwasan ", build_invocation(clang_lsan_cflags)))
|
|
config.substitutions.append(("%clangxx_hwasan ", build_invocation(clang_lsan_cxxflags)))
|
|
|
|
|
|
# LeakSanitizer tests are currently supported on
|
|
# Android{aarch64, x86, x86_64}, x86-64 Linux, PowerPC64 Linux, arm Linux, mips64 Linux, s390x Linux, loongarch64 Linux and x86_64 Darwin.
|
|
supported_android = (
|
|
config.android
|
|
and config.target_arch in ["x86_64", "i386", "aarch64"]
|
|
and "android-thread-properties-api" in config.available_features
|
|
)
|
|
supported_linux = (
|
|
(not config.android)
|
|
and config.host_os == "Linux"
|
|
and config.host_arch
|
|
in [
|
|
"aarch64",
|
|
"x86_64",
|
|
"ppc64",
|
|
"ppc64le",
|
|
"mips64",
|
|
"riscv64",
|
|
"arm",
|
|
"armhf",
|
|
"armv7l",
|
|
"s390x",
|
|
"loongarch64",
|
|
]
|
|
)
|
|
supported_darwin = config.host_os == "Darwin" and config.target_arch in ["x86_64"]
|
|
supported_netbsd = config.host_os == "NetBSD" and config.target_arch in [
|
|
"x86_64",
|
|
"i386",
|
|
]
|
|
if not (supported_android or supported_linux or supported_darwin or supported_netbsd):
|
|
config.unsupported = True
|
|
|
|
# Don't support Thumb due to broken fast unwinder
|
|
if re.search("mthumb", config.target_cflags) is not None:
|
|
config.unsupported = True
|
|
|
|
# HWASAN tests require lld because without D65857, ld.bfd and ld.gold would
|
|
# generate a corrupted binary. Mark them unsupported if lld is not available.
|
|
if "hwasan" in config.available_features and not config.has_lld:
|
|
config.unsupported = True
|
|
|
|
config.suffixes = [".c", ".cpp", ".mm"]
|