llvm-project

1
# -*- Python -*-
2

3
import os
4

5
# Setup config name.
6
config.name = "GWP-ASan" + config.name_suffix
7

8
# Setup source root.
9
config.test_source_root = os.path.dirname(__file__)
10

11
# Test suffixes.
12
config.suffixes = [".c", ".cpp", ".test"]
13

14
# C & CXX flags.
15
c_flags = [config.target_cflags]
16

17
cxx_flags = c_flags + config.cxx_mode_flags + ["-std=c++14"]
18

19
libscudo_standalone = os.path.join(
20
    config.compiler_rt_libdir, "libclang_rt.scudo_standalone%s.a" % config.target_suffix
21
)
22
libscudo_standalone_cxx = os.path.join(
23
    config.compiler_rt_libdir,
24
    "libclang_rt.scudo_standalone_cxx%s.a" % config.target_suffix,
25
)
26

27
scudo_link_flags = [
28
    "-pthread",
29
    "-Wl,--whole-archive",
30
    libscudo_standalone,
31
    "-Wl,--no-whole-archive",
32
]
33
scudo_link_cxx_flags = [
34
    "-Wl,--whole-archive",
35
    libscudo_standalone_cxx,
36
    "-Wl,--no-whole-archive",
37
]
38

39
# -rdynamic is necessary for online function symbolization.
40
gwp_asan_flags = ["-rdynamic"] + scudo_link_flags
41

42

43
def build_invocation(compile_flags):
44
    return " " + " ".join([config.clang] + compile_flags) + " "
45

46

47
# Add substitutions.
48
config.substitutions.append(("%clang ", build_invocation(c_flags)))
49
config.substitutions.append(
50
    ("%clang_gwp_asan ", build_invocation(c_flags + gwp_asan_flags))
51
)
52
config.substitutions.append(
53
    (
54
        "%clangxx_gwp_asan ",
55
        build_invocation(cxx_flags + gwp_asan_flags + scudo_link_cxx_flags),
56
    )
57
)
58

59
# Platform-specific default GWP_ASAN for lit tests. Ensure that GWP-ASan is
60
# enabled and that it samples every allocation.
61
default_gwp_asan_options = "GWP_ASAN_Enabled=1:GWP_ASAN_SampleRate=1"
62

63
config.environment["SCUDO_OPTIONS"] = default_gwp_asan_options
64
default_gwp_asan_options += ":"
65
config.substitutions.append(
66
    ("%env_scudo_options=", "env SCUDO_OPTIONS=" + default_gwp_asan_options)
67
)
68

69
# GWP-ASan tests are currently supported on Linux only.
70
if config.host_os not in ["Linux"]:
71
    config.unsupported = True
72