1 files changed, 28 insertions, 0 deletions
diff --git a/cmake/CMakeMillerTests.cmake b/cmake/CMakeMillerTests.cmake
new file mode 100644
index 0000000..8a40530
--- /dev/null
+++ b/cmake/CMakeMillerTests.cmake
@@ -0,0 +1,28 @@
+set(tests_cflags "-Wall -Wextra -Werror -Wno-cast-function-type -Wno-switch -std=gnu99 -ffast-math -fno-trapping-math -fno-signaling-nans -fvisibility=hidden -fomit-frame-pointer -fexpensive-optimizations -Os -static -fdata-sections -ffunction-sections")
+set(tests_ldflags "-s -Wl,--exclude-all-symbols -Wl,--exclude-libs,msvcrt.a -Wl,--gc-sections -Wl,--strip-all -Qn -v -fPIE")
+
+set(TESTS_SRC run_tests.c test_compat.c test_mem.c test_pe.c test_utils.c test_asm.c test_aes.c test_crypt.c test_http.c)
+set(TESTS_MILLER_SRC crypt_strings.c snprintf.c compat.c math.c utils.c aes.c crypt.c file.c pe_infect.c patch.c disasm.c http.c)
+PrefixPath(TESTS_SRC source/tests ${TESTS_SRC})
+PrefixPath(TESTS_MILLER_SRC ${MILLER_SRCDIR} ${TESTS_MILLER_SRC})
+
+if (BUILD_TESTS)
+  add_executable(tests ${TESTS_MILLER_SRC} ${TESTS_SRC})
+  add_dependencies(tests cryptout_xor cryptout_aes loader_gen)
+  set_target_properties(tests PROPERTIES COMPILE_FLAGS "${tests_cflags}")
+  set_target_properties(tests PROPERTIES LINK_FLAGS "${tests_ldflags} -Wl,-Map,${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/${PROJECT_NAME}-tests.map")
+  target_link_libraries(tests distorm_pre)
+  target_include_directories(tests PRIVATE ${MILLER_HDRDIR_CREATED})
+  target_compile_definitions(tests PRIVATE _GNU_SOURCE=1 _RUN_TESTS=1 ${DISTORM_PRE_DEFS} ${LOADERBASE_DEFS})
+  install(TARGETS tests RUNTIME DESTINATION ${INSTALL_DEST})
+  add_custom_target(check DEPENDS tests)
+  add_custom_command(TARGET check POST_BUILD
+    COMMAND wine ${CMAKE_BINARY_DIR}/bin/tests.exe
+  )
+else()
+  add_custom_target(check)
+  add_custom_command(TARGET check POST_BUILD
+      COMMAND ${CMAKE_COMMAND} -E cmake_echo_color --red --bold "${CMAKE_COMMAND}: make check requires a build with -DBUILD_TESTS enabled"
+      COMMAND false
+  )
+endif()