CMakeLists.txt

cmake_minimum_required(VERSION 3.16)

project(netdata_windows_driver VERSION 1.0.0 LANGUAGES C)

include(GNUInstallDirs)

set(NETDATA_DRIVER_WINDOWS_ENV FALSE)
if(WIN32 OR MINGW OR MSYS OR CMAKE_SYSTEM_NAME STREQUAL "MSYS")
    set(NETDATA_DRIVER_WINDOWS_ENV TRUE)
endif()

if(DEFINED BINDIR)
    set(NETDATA_DRIVER_INSTALL_DIR "${BINDIR}")
else()
    set(NETDATA_DRIVER_INSTALL_DIR "${CMAKE_INSTALL_BINDIR}")
endif()

set(NETDATA_DRIVER_SOURCES
    src/driver/netdata_driver.c
    src/driver/netdata_driver.h
)

set(NETDATA_DRIVER_FILE "${CMAKE_CURRENT_BINARY_DIR}/netdata_driver.sys")
set(NETDATA_DRIVER_FILE_INF "${CMAKE_CURRENT_BINARY_DIR}/netdata_driver.inf")
set(NETDATA_DRIVER_FILE_CAT "${CMAKE_CURRENT_BINARY_DIR}/netdata_driver.cat")
set(NETDATA_DRIVER_INF2CAT_DIR "${CMAKE_CURRENT_BINARY_DIR}/inf2cat")
set(NETDATA_DRIVER_INF2CAT_FILE_SYS "${NETDATA_DRIVER_INF2CAT_DIR}/netdata_driver.sys")
set(NETDATA_DRIVER_INF2CAT_FILE_INF "${NETDATA_DRIVER_INF2CAT_DIR}/netdata_driver.inf")
set(NETDATA_DRIVER_INF2CAT_FILE_CAT "${NETDATA_DRIVER_INF2CAT_DIR}/netdata_driver.cat")
set(NETDATA_DRIVER_MSI_VERSION "${PROJECT_VERSION}")
set(NETDATA_DRIVER_INF2CAT_OS "10_X64" CACHE STRING
    "Comma-separated Inf2Cat OS identifier list used to generate netdata_driver.cat"
)

set(NETDATA_DRIVER_PACKAGING_ASSETS
    BackGround.bmp
    LICENSE.txt
    NetdataWhite.ico
    REDISTRIBUTED.md
    Top.bmp
    eula.rtf
)

file(TO_NATIVE_PATH "${NETDATA_DRIVER_FILE}" NETDATA_DRIVER_FILE_NATIVE)
file(TO_NATIVE_PATH "${NETDATA_DRIVER_FILE_INF}" NETDATA_DRIVER_FILE_INF_NATIVE)
file(TO_NATIVE_PATH "${NETDATA_DRIVER_FILE_CAT}" NETDATA_DRIVER_FILE_CAT_NATIVE)

configure_file(src/driver/netdata_driver.inf netdata_driver.inf COPYONLY)
configure_file(packaging/windows/netdata_driver.wxs.in netdata_driver.wxs @ONLY)

foreach(asset IN LISTS NETDATA_DRIVER_PACKAGING_ASSETS)
    configure_file(
        "${CMAKE_CURRENT_SOURCE_DIR}/packaging/windows/${asset}"
        "${CMAKE_CURRENT_BINARY_DIR}/${asset}"
        COPYONLY
    )
endforeach()

set(_netdata_driver_default_gnu_ddk_include_dir "/mingw64/include/ddk")

foreach(_netdata_driver_gnu_ddk_candidate
    "/mingw64/include/ddk"
    "/clang64/include/ddk"
    "/ucrt64/include/ddk"
    "/usr/include/w32api/ddk"
)
    if(EXISTS "${_netdata_driver_gnu_ddk_candidate}/ntddk.h")
        set(_netdata_driver_default_gnu_ddk_include_dir "${_netdata_driver_gnu_ddk_candidate}")
        break()
    endif()
endforeach()

set(NETDATA_DRIVER_GNU_DDK_INCLUDE_DIR "${_netdata_driver_default_gnu_ddk_include_dir}" CACHE PATH
    "DDK include directory for GNU-style Windows toolchains such as MSYS2 MinGW"
)

set(NETDATA_DRIVER_ENTRY_POINT "DriverEntry")

if(CMAKE_SIZEOF_VOID_P EQUAL 4)
    if(MSVC)
        set(NETDATA_DRIVER_ENTRY_POINT "_DriverEntry@8")
    else()
        set(NETDATA_DRIVER_ENTRY_POINT "DriverEntry@8")
    endif()
endif()

add_library(netdata_driver SHARED ${NETDATA_DRIVER_SOURCES})
set_target_properties(netdata_driver PROPERTIES
    LIBRARY_OUTPUT_NAME "netdata_driver"
    PREFIX ""
    SUFFIX ".sys"
)

if(NETDATA_DRIVER_WINDOWS_ENV)
    find_program(NETDATA_DRIVER_INF2CAT_EXECUTABLE NAMES Inf2Cat inf2cat)

    if(NOT NETDATA_DRIVER_INF2CAT_EXECUTABLE)
        message(FATAL_ERROR
            "Inf2Cat was not found. Install the Windows Driver Kit and ensure Inf2Cat is in PATH "
            "or set NETDATA_DRIVER_INF2CAT_EXECUTABLE explicitly."
        )
    endif()

    if(MSVC)
        file(TO_NATIVE_PATH "${NETDATA_DRIVER_INF2CAT_DIR}" NETDATA_DRIVER_INF2CAT_DRIVER_DIR_NATIVE)

        add_custom_command(
            OUTPUT "${NETDATA_DRIVER_FILE_CAT}"
            BYPRODUCTS
                "${NETDATA_DRIVER_INF2CAT_FILE_SYS}"
                "${NETDATA_DRIVER_INF2CAT_FILE_INF}"
                "${NETDATA_DRIVER_INF2CAT_FILE_CAT}"
            COMMAND "${CMAKE_COMMAND}" -E make_directory "${NETDATA_DRIVER_INF2CAT_DIR}"
            COMMAND "${CMAKE_COMMAND}" -E copy_if_different "${NETDATA_DRIVER_FILE}" "${NETDATA_DRIVER_INF2CAT_FILE_SYS}"
            COMMAND "${CMAKE_COMMAND}" -E copy_if_different "${NETDATA_DRIVER_FILE_INF}" "${NETDATA_DRIVER_INF2CAT_FILE_INF}"
            COMMAND "${NETDATA_DRIVER_INF2CAT_EXECUTABLE}"
                "/driver:${NETDATA_DRIVER_INF2CAT_DRIVER_DIR_NATIVE}"
                "/os:${NETDATA_DRIVER_INF2CAT_OS}"
                /verbose
            COMMAND "${CMAKE_COMMAND}" -E copy_if_different "${NETDATA_DRIVER_INF2CAT_FILE_CAT}" "${NETDATA_DRIVER_FILE_CAT}"
            DEPENDS
                netdata_driver
                "${NETDATA_DRIVER_FILE_INF}"
            WORKING_DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}"
            COMMENT "Generating netdata_driver.cat with Inf2Cat"
            VERBATIM
        )
    else()
        add_custom_command(
            OUTPUT "${NETDATA_DRIVER_FILE_CAT}"
            BYPRODUCTS
                "${NETDATA_DRIVER_INF2CAT_FILE_SYS}"
                "${NETDATA_DRIVER_INF2CAT_FILE_INF}"
                "${NETDATA_DRIVER_INF2CAT_FILE_CAT}"
            COMMAND "${CMAKE_COMMAND}" -E make_directory "${NETDATA_DRIVER_INF2CAT_DIR}"
            COMMAND "${CMAKE_COMMAND}" -E copy_if_different "${NETDATA_DRIVER_FILE}" "${NETDATA_DRIVER_INF2CAT_FILE_SYS}"
            COMMAND "${CMAKE_COMMAND}" -E copy_if_different "${NETDATA_DRIVER_FILE_INF}" "${NETDATA_DRIVER_INF2CAT_FILE_INF}"
            COMMAND bash "${CMAKE_CURRENT_SOURCE_DIR}/packaging/run-inf2cat.sh"
                "${NETDATA_DRIVER_INF2CAT_EXECUTABLE}"
                "${NETDATA_DRIVER_INF2CAT_DIR}"
                "${NETDATA_DRIVER_INF2CAT_OS}"
                "${NETDATA_DRIVER_INF2CAT_FILE_CAT}"
            COMMAND "${CMAKE_COMMAND}" -E copy_if_different "${NETDATA_DRIVER_INF2CAT_FILE_CAT}" "${NETDATA_DRIVER_FILE_CAT}"
            DEPENDS
                netdata_driver
                "${NETDATA_DRIVER_FILE_INF}"
            WORKING_DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}"
            COMMENT "Generating netdata_driver.cat with Inf2Cat"
            VERBATIM
        )
    endif()

    add_custom_target(netdata_driver_catalog ALL
        DEPENDS "${NETDATA_DRIVER_FILE_CAT}"
    )
endif()

target_include_directories(netdata_driver PRIVATE
    ${CMAKE_CURRENT_SOURCE_DIR}/src/driver
)

if(NETDATA_DRIVER_WINDOWS_ENV AND CMAKE_C_COMPILER_ID MATCHES "GNU|Clang" AND NOT MSVC)
    if(NOT EXISTS "${NETDATA_DRIVER_GNU_DDK_INCLUDE_DIR}/ntddk.h")
        message(FATAL_ERROR
            "GNU-style Windows builds require DDK headers. "
            "Set NETDATA_DRIVER_GNU_DDK_INCLUDE_DIR to the directory containing ntddk.h."
        )
    endif()

    target_include_directories(netdata_driver BEFORE PRIVATE
        "${NETDATA_DRIVER_GNU_DDK_INCLUDE_DIR}"
    )

    target_compile_options(netdata_driver PRIVATE
        -Wall
        -Wextra
        -Werror
        -ffreestanding
        -fno-exceptions
        -fno-stack-protector
        -fshort-wchar
        -mno-red-zone
    )

    target_link_options(netdata_driver PRIVATE
        "-Wl,--entry,${NETDATA_DRIVER_ENTRY_POINT}"
        -nostdlib
        -Wl,--subsystem,native
        -Wl,--image-base,0x10000
    )

    target_link_libraries(netdata_driver PRIVATE
        kernel32
        ntoskrnl
    )
elseif(MSVC)
    target_compile_options(netdata_driver PRIVATE
        /W4
        /WX
        /GS-
    )

    target_link_options(netdata_driver PRIVATE
        /DRIVER
        /NODEFAULTLIB
        /SUBSYSTEM:NATIVE
        "/ENTRY:${NETDATA_DRIVER_ENTRY_POINT}"
        /BASE:0x10000
    )

    target_link_libraries(netdata_driver PRIVATE
        ntoskrnl
    )
endif()

install(FILES "${NETDATA_DRIVER_FILE}"
    COMPONENT netdata_driver
    DESTINATION "${NETDATA_DRIVER_INSTALL_DIR}"
)

install(FILES "${NETDATA_DRIVER_FILE_INF}"
    COMPONENT netdata_driver_inf
    DESTINATION "${NETDATA_DRIVER_INSTALL_DIR}"
)

if(NETDATA_DRIVER_WINDOWS_ENV)
    install(FILES "${NETDATA_DRIVER_FILE_CAT}"
        COMPONENT netdata_driver_cat
        DESTINATION "${NETDATA_DRIVER_INSTALL_DIR}"
    )
endif()