1# Ported from Clickhouse: https://github.com/ClickHouse/ClickHouse/blob/master/cmake/sanitize.cmake
2
3option (SANITIZE "Enable sanitizer: address, memory, undefined" "")
4set (SAN_FLAGS "${SAN_FLAGS} -g -fno-omit-frame-pointer -DSANITIZER")
5# O1 is normally set by clang, and -Og by gcc
6if (COMPILER_GCC)
7    set (SAN_FLAGS "${SAN_FLAGS} -Og")
8else ()
9    set (SAN_FLAGS "${SAN_FLAGS} -O1")
10endif ()
11if (SANITIZE)
12    if (ENABLE_JEMALLOC MATCHES "ON")
13        message (STATUS "Jemalloc support is useless in case of build with sanitizers")
14        set (ENABLE_JEMALLOC "OFF")
15    endif ()
16    if (SANITIZE STREQUAL "address")
17        set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SAN_FLAGS} -fsanitize=address -fsanitize-address-use-after-scope")
18        set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${SAN_FLAGS} -fsanitize=address -fsanitize-address-use-after-scope")
19        set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -fsanitize=address -fsanitize-address-use-after-scope")
20        if (COMPILER_GCC)
21            set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -static-libasan")
22            set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -static-libasan")
23            set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -static-libasan")
24        endif ()
25
26    elseif (SANITIZE STREQUAL "memory")
27        set (MSAN_FLAGS "-fsanitize=memory -fsanitize-memory-track-origins -fno-optimize-sibling-calls")
28
29        set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SAN_FLAGS} ${MSAN_FLAGS}")
30        set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${SAN_FLAGS} ${MSAN_FLAGS}")
31        set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -fsanitize=memory")
32
33        if (COMPILER_GCC)
34            set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -static-libmsan")
35            set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -static-libmsan")
36            set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -static-libmsan")
37        endif ()
38
39    elseif (SANITIZE STREQUAL "undefined")
40        set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SAN_FLAGS} -fsanitize=undefined -fno-sanitize-recover=all")
41        set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${SAN_FLAGS} -fsanitize=undefined -fno-sanitize-recover=all")
42        set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -fsanitize=undefined")
43
44        if (COMPILER_GCC)
45            set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -static-libubsan")
46            set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -static-libubsan")
47            set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -static-libubsan")
48        endif ()
49    else ()
50        message (FATAL_ERROR "Unknown sanitizer type: ${SANITIZE}")
51    endif ()
52    message (STATUS "Add sanitizer: ${SANITIZE}")
53    # Disable sanitizing on make stage e.g. for snowball compiler
54    set (ENV{ASAN_OPTIONS} "detect_leaks=0")
55    message (STATUS "Sanitizer CFLAGS: ${CMAKE_C_FLAGS} ${CMAKE_C_FLAGS_${CMAKE_BUILD_TYPE_UC}}")
56    message (STATUS "Sanitizer CXXFLAGS: ${CMAKE_CXX_FLAGS} ${CMAKE_C_FLAGS_${CMAKE_BUILD_TYPE_UC}}")
57endif()