1include(CheckCXXCompilerFlag) 2include(CompilerRTCompile) 3include(CompilerRTLink) 4 5include_directories(..) 6include_directories(../..) 7 8set(MSAN_LIBCXX_CFLAGS 9 -fsanitize=memory 10 -fsanitize-memory-track-origins 11 -Wno-pedantic 12 -Xclang -fdepfile-entry=${COMPILER_RT_OUTPUT_DIR}/share/msan_blacklist.txt 13 ) 14 15# Unittest sources and build flags. 16set(MSAN_UNITTEST_SOURCES 17 msan_test.cpp 18 msan_test_main.cpp 19 ) 20set(MSAN_LOADABLE_SOURCE 21 msan_loadable.cpp 22 ) 23set(MSAN_UNITTEST_HEADERS 24 msan_test_config.h 25 ../../../include/sanitizer/msan_interface.h 26 ) 27set(MSAN_UNITTEST_COMMON_CFLAGS 28 -nostdinc++ 29 ${COMPILER_RT_UNITTEST_CFLAGS} 30 ${COMPILER_RT_GTEST_CFLAGS} 31 -I${COMPILER_RT_SOURCE_DIR}/include 32 -I${COMPILER_RT_SOURCE_DIR}/lib 33 -I${COMPILER_RT_SOURCE_DIR}/lib/msan 34 -g 35 -O2 36 -fno-omit-frame-pointer 37 -mno-omit-leaf-frame-pointer 38 -Wno-deprecated-declarations 39 -Wno-unused-variable 40 -Wno-zero-length-array 41 -Wno-uninitialized 42 -Werror=sign-compare 43 -Wno-gnu-zero-variadic-macro-arguments 44) 45# Remove -stdlib= which is unused when passing -nostdinc++. 46string(REGEX REPLACE "-stdlib=[a-zA-Z+]*" "" CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS}) 47 48set(MSAN_UNITTEST_INSTRUMENTED_CFLAGS 49 ${MSAN_UNITTEST_COMMON_CFLAGS} 50 -fsanitize=memory 51 -fsanitize-memory-track-origins 52 -mllvm -msan-keep-going=1 53) 54set(MSAN_UNITTEST_LINK_FLAGS 55 ${COMPILER_RT_UNITTEST_LINK_FLAGS} 56 -fsanitize=memory 57 # Don't need -stdlib=libc++ because we explicitly list libc++.a in the linker 58 # inputs. 59) 60 61append_list_if(COMPILER_RT_HAS_LIBDL -ldl MSAN_UNITTEST_LINK_FLAGS) 62 63macro(msan_compile obj_list source arch kind cflags) 64 sanitizer_test_compile( 65 ${obj_list} ${source} ${arch} 66 KIND ${kind} 67 COMPILE_DEPS ${MSAN_UNITTEST_HEADERS} 68 DEPS gtest msan 69 CFLAGS -isystem ${CMAKE_CURRENT_BINARY_DIR}/../libcxx_msan_${arch}/include/c++/v1 70 ${MSAN_UNITTEST_INSTRUMENTED_CFLAGS} ${cflags} 71 ) 72endmacro() 73 74macro(msan_link_shared so_list so_name arch kind) 75 cmake_parse_arguments(SOURCE "" "" "OBJECTS;LINK_FLAGS;DEPS" ${ARGN}) 76 set(output_so "${CMAKE_CURRENT_BINARY_DIR}/${so_name}.${arch}${kind}.so") 77 get_target_flags_for_arch(${arch} TARGET_LINK_FLAGS) 78 if(NOT COMPILER_RT_STANDALONE_BUILD) 79 list(APPEND SOURCE_DEPS msan) 80 endif() 81 clang_link_shared(${output_so} 82 OBJECTS ${SOURCE_OBJECTS} 83 LINK_FLAGS ${COMPILER_RT_UNITTEST_LINK_FLAGS} ${TARGET_LINK_FLAGS} ${SOURCE_LINK_FLAGS} 84 DEPS ${SOURCE_DEPS}) 85 list(APPEND ${so_list} ${output_so}) 86endmacro() 87 88# Main MemorySanitizer unit tests. 89add_custom_target(MsanUnitTests) 90set_target_properties(MsanUnitTests PROPERTIES FOLDER "MSan unit tests") 91 92# Adds MSan unit tests and benchmarks for architecture. 93macro(add_msan_tests_for_arch arch kind cflags) 94 # Build gtest instrumented with MSan. 95 set(MSAN_INST_GTEST) 96 msan_compile(MSAN_INST_GTEST ${COMPILER_RT_GTEST_SOURCE} ${arch} "${kind}" 97 "${cflags}") 98 99 # Instrumented tests. 100 set(MSAN_INST_TEST_OBJECTS) 101 foreach (SOURCE ${MSAN_UNITTEST_SOURCES}) 102 msan_compile(MSAN_INST_TEST_OBJECTS ${SOURCE} ${arch} "${kind}" "${cflags}") 103 endforeach(SOURCE) 104 105 # Instrumented loadable module objects. 106 set(MSAN_INST_LOADABLE_OBJECTS) 107 msan_compile(MSAN_INST_LOADABLE_OBJECTS ${MSAN_LOADABLE_SOURCE} ${arch} "${kind}" 108 "-fPIC;${cflags}") 109 110 # Instrumented loadable library tests. 111 set(MSAN_LOADABLE_SO) 112 msan_link_shared(MSAN_LOADABLE_SO "libmsan_loadable" ${arch} "${kind}" 113 OBJECTS ${MSAN_INST_LOADABLE_OBJECTS} 114 DEPS ${MSAN_INST_LOADABLE_OBJECTS}) 115 116 set(MSAN_TEST_OBJECTS ${MSAN_INST_TEST_OBJECTS} ${MSAN_INST_GTEST}) 117 set(MSAN_TEST_DEPS ${MSAN_TEST_OBJECTS} libcxx_msan_${arch}-build 118 ${MSAN_LOADABLE_SO} 119 "${MSAN_LIBCXX_DIR}/libc++.a" "${MSAN_LIBCXX_DIR}/libc++abi.a") 120 if(NOT COMPILER_RT_STANDALONE_BUILD) 121 list(APPEND MSAN_TEST_DEPS msan) 122 endif() 123 get_target_flags_for_arch(${arch} TARGET_LINK_FLAGS) 124 add_compiler_rt_test(MsanUnitTests "Msan-${arch}${kind}-Test" ${arch} 125 OBJECTS ${MSAN_TEST_OBJECTS} "${MSAN_LIBCXX_DIR}/libc++.a" "${MSAN_LIBCXX_DIR}/libc++abi.a" 126 DEPS ${MSAN_TEST_DEPS} 127 LINK_FLAGS ${MSAN_UNITTEST_LINK_FLAGS} 128 ${TARGET_LINK_FLAGS}) 129endmacro() 130 131# We should only build MSan unit tests if we can build instrumented libcxx. 132if(COMPILER_RT_CAN_EXECUTE_TESTS AND 133 COMPILER_RT_LIBCXX_PATH AND 134 COMPILER_RT_LIBCXXABI_PATH) 135 foreach(arch ${MSAN_SUPPORTED_ARCH}) 136 get_target_flags_for_arch(${arch} TARGET_CFLAGS) 137 set(LIBCXX_PREFIX ${CMAKE_CURRENT_BINARY_DIR}/../libcxx_msan_${arch}) 138 add_custom_libcxx(libcxx_msan_${arch} ${LIBCXX_PREFIX} 139 DEPS ${MSAN_RUNTIME_LIBRARIES} 140 CFLAGS ${MSAN_LIBCXX_CFLAGS} ${TARGET_CFLAGS} 141 USE_TOOLCHAIN) 142 set(MSAN_LIBCXX_DIR ${LIBCXX_PREFIX}/lib/) 143 144 add_msan_tests_for_arch(${arch} "" "") 145 add_msan_tests_for_arch(${arch} "-with-call" 146 "-mllvm;-msan-instrumentation-with-call-threshold=0") 147 endforeach() 148endif() 149