114 lines · plain
1include(CompilerRTCompile)2 3set(LIBFUZZER_UNITTEST_CFLAGS4 ${COMPILER_RT_UNITTEST_CFLAGS}5 ${COMPILER_RT_GTEST_CFLAGS}6 -I${COMPILER_RT_SOURCE_DIR}/lib/fuzzer7 -fno-rtti8 -O2)9 10if (APPLE)11 set(FUZZER_SUPPORTED_OS osx)12endif()13 14add_custom_target(FuzzerUnitTests)15set_target_properties(FuzzerUnitTests PROPERTIES FOLDER "Compiler-RT/Tests")16 17add_custom_target(FuzzedDataProviderUnitTests)18set_target_properties(FuzzedDataProviderUnitTests PROPERTIES FOLDER "Compiler-RT/Tests")19 20set(LIBFUZZER_UNITTEST_LINK_FLAGS ${COMPILER_RT_UNITTEST_LINK_FLAGS})21list(APPEND LIBFUZZER_UNITTEST_LINK_FLAGS --driver-mode=g++)22 23if(MSVC)24 list(APPEND LIBFUZZER_UNITTEST_LINK_FLAGS -Wl,-defaultlib:libcmt,-defaultlib:oldnames)25else()26 if (APPLE)27 list(APPEND LIBFUZZER_UNITTEST_CFLAGS -isysroot ${DARWIN_osx_SYSROOT})28 list(APPEND LIBFUZZER_UNITTEST_LINK_FLAGS -isysroot ${DARWIN_osx_SYSROOT})29 endif()30 list(APPEND LIBFUZZER_UNITTEST_LINK_FLAGS -lpthread)31endif()32 33if(CMAKE_SYSTEM_NAME STREQUAL "Linux" AND34 COMPILER_RT_LIBCXX_PATH AND35 COMPILER_RT_LIBCXXABI_PATH)36 list(APPEND LIBFUZZER_UNITTEST_CFLAGS -nostdinc++ -fno-exceptions)37 list(APPEND LIBFUZZER_UNITTEST_LINK_FLAGS -nostdlib++ -fno-exceptions)38 39 # When we use -nostdlib++, we remove the default C++ runtime which normally40 # provides the stack unwinding symbols (like __aeabi_unwind_cpp_pr0).41 # We must now manually find and link a suitable unwinder library.42 set(FUZZER_UNWINDER_LIBS)43 if(COMPILER_RT_USE_LLVM_UNWINDER)44 # Prefer LLVM's own libunwind.45 list(APPEND FUZZER_UNWINDER_LIBS ${COMPILER_RT_UNWINDER_LINK_LIBS})46 elseif(COMPILER_RT_HAS_GCC_S_LIB)47 # As a fallback, use the shared libgcc_s library.48 list(APPEND FUZZER_UNWINDER_LIBS -lgcc_s)49 elseif(COMPILER_RT_HAS_GCC_LIB)50 # As a final fallback, use the static libgcc library.51 list(APPEND FUZZER_UNWINDER_LIBS -lgcc)52 elseif(NOT COMPILER_RT_USE_BUILTINS_LIBRARY)53 # If no unwinder is found and we aren't using the builtins library54 message(FATAL_ERROR "Fuzzer tests require a suitable unwinder, but none was found.")55 endif()56 # Add the detected unwinder library to our link flags.57 list(APPEND LIBFUZZER_UNITTEST_LINK_FLAGS ${FUZZER_UNWINDER_LIBS})58 59endif()60 61if ("-fvisibility=hidden" IN_LIST LIBFUZZER_CFLAGS)62 # Match visibility settings.63 list(APPEND LIBFUZZER_UNITTEST_CFLAGS "-fvisibility=hidden")64endif()65 66if(COMPILER_RT_DEFAULT_TARGET_ARCH IN_LIST FUZZER_SUPPORTED_ARCH)67 # libFuzzer unit tests are only run on the host machine.68 set(arch ${COMPILER_RT_DEFAULT_TARGET_ARCH})69 70 set(LIBFUZZER_TEST_RUNTIME RTFuzzerTest.${arch})71 if(APPLE)72 set(LIBFUZZER_TEST_RUNTIME_OBJECTS73 $<TARGET_OBJECTS:RTfuzzer.osx>)74 else()75 set(LIBFUZZER_TEST_RUNTIME_OBJECTS76 $<TARGET_OBJECTS:RTfuzzer.${arch}>)77 endif()78 add_library(${LIBFUZZER_TEST_RUNTIME} STATIC79 ${LIBFUZZER_TEST_RUNTIME_OBJECTS})80 set_target_properties(${LIBFUZZER_TEST_RUNTIME} PROPERTIES81 ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}82 FOLDER "Compiler-RT/Tests/Runtime")83 84 if(CMAKE_SYSTEM_NAME STREQUAL "Linux" AND85 COMPILER_RT_LIBCXX_PATH AND86 COMPILER_RT_LIBCXXABI_PATH)87 file(GLOB libfuzzer_headers ../*.h)88 set(LIBFUZZER_TEST_RUNTIME_DEPS libcxx_fuzzer_${arch}-install-cmake326-workaround ${libfuzzer_headers})89 set(LIBFUZZER_TEST_RUNTIME_CFLAGS -isystem ${LIBCXX_${arch}_PREFIX}/include/c++/v1)90 set(LIBFUZZER_TEST_RUNTIME_LINK_FLAGS ${LIBCXX_${arch}_PREFIX}/lib/libc++.a)91 endif()92 93 set(FuzzerTestObjects)94 generate_compiler_rt_tests(FuzzerTestObjects95 FuzzerUnitTests "Fuzzer-${arch}-Test" ${arch}96 SOURCES FuzzerUnittest.cpp ${COMPILER_RT_GTEST_SOURCE}97 RUNTIME ${LIBFUZZER_TEST_RUNTIME}98 DEPS ${LIBFUZZER_TEST_RUNTIME_DEPS}99 CFLAGS ${LIBFUZZER_UNITTEST_CFLAGS} ${LIBFUZZER_TEST_RUNTIME_CFLAGS}100 LINK_FLAGS ${LIBFUZZER_UNITTEST_LINK_FLAGS} ${LIBFUZZER_TEST_RUNTIME_LINK_FLAGS})101 set_target_properties(FuzzerUnitTests PROPERTIES102 RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})103 104 set(FuzzedDataProviderTestObjects)105 generate_compiler_rt_tests(FuzzedDataProviderTestObjects106 FuzzedDataProviderUnitTests "FuzzerUtils-${arch}-Test" ${arch}107 SOURCES FuzzedDataProviderUnittest.cpp ${COMPILER_RT_GTEST_SOURCE}108 DEPS ${LIBFUZZER_TEST_RUNTIME_DEPS} ${COMPILER_RT_SOURCE_DIR}/include/fuzzer/FuzzedDataProvider.h109 CFLAGS ${LIBFUZZER_UNITTEST_CFLAGS} ${LIBFUZZER_TEST_RUNTIME_CFLAGS}110 LINK_FLAGS ${LIBFUZZER_UNITTEST_LINK_FLAGS} ${LIBFUZZER_TEST_RUNTIME_LINK_FLAGS})111 set_target_properties(FuzzedDataProviderUnitTests PROPERTIES112 RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})113endif()114