| include(CheckCXXCompilerFlag) |
| include(CompilerRTCompile) |
| include(CompilerRTLink) |
| |
| include_directories(..) |
| include_directories(../..) |
| |
| # Instrumented libcxx sources and build flags. |
| file(GLOB MSAN_LIBCXX_SOURCES ${MSAN_LIBCXX_PATH}/src/*.cpp) |
| set(MSAN_LIBCXX_CFLAGS |
| -I${MSAN_LIBCXX_PATH}/include |
| -fsanitize=memory |
| -fsanitize-memory-track-origins |
| -fPIC |
| -Wno-\#warnings |
| -g |
| -O2 |
| -std=c++0x |
| -fstrict-aliasing |
| -fno-exceptions |
| -nostdinc++ |
| -fno-omit-frame-pointer |
| -mno-omit-leaf-frame-pointer) |
| set(MSAN_LIBCXX_LINK_FLAGS |
| -nodefaultlibs |
| -lpthread |
| -lrt |
| -lc |
| -lstdc++ |
| -fsanitize=memory) |
| |
| # Unittest sources and build flags. |
| set(MSAN_UNITTEST_SOURCES msan_test.cc msan_test_main.cc) |
| set(MSAN_LOADABLE_SOURCE msan_loadable.cc) |
| set(MSAN_UNITTEST_HEADERS |
| msan_test_config.h |
| msandr_test_so.h |
| ../../../include/sanitizer/msan_interface.h |
| ) |
| set(MSANDR_UNITTEST_SOURCE msandr_test_so.cc) |
| set(MSAN_UNITTEST_COMMON_CFLAGS |
| -I${MSAN_LIBCXX_PATH}/include |
| ${COMPILER_RT_GTEST_INCLUDE_CFLAGS} |
| -I${COMPILER_RT_SOURCE_DIR}/include |
| -I${COMPILER_RT_SOURCE_DIR}/lib |
| -I${COMPILER_RT_SOURCE_DIR}/lib/msan |
| -std=c++0x |
| -stdlib=libc++ |
| -g |
| -O2 |
| -fno-exceptions |
| -fno-omit-frame-pointer |
| -mno-omit-leaf-frame-pointer |
| ) |
| set(MSAN_UNITTEST_INSTRUMENTED_CFLAGS |
| ${MSAN_UNITTEST_COMMON_CFLAGS} |
| -fsanitize=memory |
| -fsanitize-memory-track-origins |
| -mllvm -msan-keep-going=1 |
| ) |
| set(MSAN_UNITTEST_LINK_FLAGS |
| -fsanitize=memory |
| -ldl |
| # FIXME: we build libcxx without cxxabi and need libstdc++ to provide it. |
| -lstdc++ |
| ) |
| set(MSAN_LOADABLE_LINK_FLAGS |
| -fsanitize=memory |
| -shared |
| ) |
| |
| # Compile source for the given architecture, using compiler |
| # options in ${ARGN}, and add it to the object list. |
| macro(msan_compile obj_list source arch) |
| get_filename_component(basename ${source} NAME) |
| set(output_obj "${basename}.${arch}.o") |
| get_target_flags_for_arch(${arch} TARGET_CFLAGS) |
| clang_compile(${output_obj} ${source} |
| CFLAGS ${ARGN} ${TARGET_CFLAGS} |
| DEPS gtest ${MSAN_RUNTIME_LIBRARIES} ${MSAN_UNITTEST_HEADERS}) |
| list(APPEND ${obj_list} ${output_obj}) |
| endmacro() |
| |
| macro(msan_link_shared so_list so_name arch) |
| parse_arguments(SOURCE "OBJECTS;LINKFLAGS;DEPS" "" ${ARGN}) |
| set(output_so "${CMAKE_CURRENT_BINARY_DIR}/${so_name}.${arch}.so") |
| get_target_flags_for_arch(${arch} TARGET_LINKFLAGS) |
| clang_link_shared(${output_so} |
| OBJECTS ${SOURCE_OBJECTS} |
| LINKFLAGS ${TARGET_LINKFLAGS} ${SOURCE_LINKFLAGS} |
| DEPS ${SOURCE_DEPS}) |
| list(APPEND ${so_list} ${output_so}) |
| endmacro() |
| |
| # Link MSan unit test for a given architecture from a set |
| # of objects in ${ARGN}. |
| macro(add_msan_test test_suite test_name arch) |
| get_target_flags_for_arch(${arch} TARGET_LINK_FLAGS) |
| add_compiler_rt_test(${test_suite} ${test_name} |
| OBJECTS ${ARGN} |
| DEPS ${MSAN_RUNTIME_LIBRARIES} ${ARGN} |
| ${MSAN_LOADABLE_SO} |
| LINK_FLAGS ${MSAN_UNITTEST_LINK_FLAGS} |
| ${TARGET_LINK_FLAGS} |
| "-Wl,-rpath=${CMAKE_CURRENT_BINARY_DIR}") |
| endmacro() |
| |
| # Main MemorySanitizer unit tests. |
| add_custom_target(MsanUnitTests) |
| set_target_properties(MsanUnitTests PROPERTIES FOLDER "MSan unit tests") |
| |
| # Adds MSan unit tests and benchmarks for architecture. |
| macro(add_msan_tests_for_arch arch) |
| # Build gtest instrumented with MSan. |
| set(MSAN_INST_GTEST) |
| msan_compile(MSAN_INST_GTEST ${COMPILER_RT_GTEST_SOURCE} ${arch} |
| ${MSAN_UNITTEST_INSTRUMENTED_CFLAGS}) |
| |
| # Build libcxx instrumented with MSan. |
| set(MSAN_INST_LIBCXX_OBJECTS) |
| foreach(SOURCE ${MSAN_LIBCXX_SOURCES}) |
| msan_compile(MSAN_INST_LIBCXX_OBJECTS ${SOURCE} ${arch} |
| ${MSAN_LIBCXX_CFLAGS}) |
| endforeach(SOURCE) |
| |
| set(MSAN_INST_LIBCXX) |
| msan_link_shared(MSAN_INST_LIBCXX "libcxx" ${arch} |
| OBJECTS ${MSAN_INST_LIBCXX_OBJECTS} |
| LINKFLAGS ${MSAN_LIBCXX_LINK_FLAGS} |
| DEPS ${MSAN_INST_LIBCXX_OBJECTS} ${MSAN_RUNTIME_LIBRARIES}) |
| |
| # Instrumented tests. |
| set(MSAN_INST_TEST_OBJECTS) |
| foreach (SOURCE ${MSAN_UNITTEST_SOURCES}) |
| msan_compile(MSAN_INST_TEST_OBJECTS ${SOURCE} ${arch} |
| ${MSAN_UNITTEST_INSTRUMENTED_CFLAGS}) |
| endforeach(SOURCE) |
| |
| # Instrumented loadable module objects. |
| set(MSAN_INST_LOADABLE_OBJECTS) |
| msan_compile(MSAN_INST_LOADABLE_OBJECTS ${MSAN_LOADABLE_SOURCE} ${arch} |
| ${MSAN_UNITTEST_INSTRUMENTED_CFLAGS}) |
| |
| # Uninstrumented shared object for MSanDR tests. |
| set(MSANDR_TEST_OBJECTS) |
| msan_compile(MSANDR_TEST_OBJECTS ${MSANDR_UNITTEST_SOURCE} ${arch} |
| ${MSAN_UNITTEST_COMMON_CFLAGS}) |
| |
| # Instrumented loadable library tests. |
| set(MSAN_LOADABLE_SO) |
| msan_link_shared(MSAN_LOADABLE_SO "libmsan_loadable" ${arch} |
| OBJECTS ${MSAN_INST_LOADABLE_OBJECTS} |
| DEPS ${MSAN_INST_LOADABLE_OBJECTS} ${MSAN_RUNTIME_LIBRARIES}) |
| |
| # Uninstrumented shared library tests. |
| set(MSANDR_TEST_SO) |
| msan_link_shared(MSANDR_TEST_SO "libmsandr_test" ${arch} |
| OBJECTS ${MSANDR_TEST_OBJECTS} |
| DEPS ${MSANDR_TEST_OBJECTS} ${MSAN_RUNTIME_LIBRARIES}) |
| |
| # Link everything together. |
| add_msan_test(MsanUnitTests "Msan-${arch}-Test" ${arch} |
| ${MSAN_INST_TEST_OBJECTS} ${MSAN_INST_GTEST} |
| ${MSAN_INST_LIBCXX} ${MSANDR_TEST_SO}) |
| endmacro() |
| |
| if(COMPILER_RT_CAN_EXECUTE_TESTS AND MSAN_CAN_INSTRUMENT_LIBCXX) |
| if(CAN_TARGET_x86_64) |
| add_msan_tests_for_arch(x86_64) |
| endif() |
| endif() |