4.CMake设置Windows驱动程序配置

1. CMake设置windows驱动开发配置（MSVC）

编写CMakelists.txt文件

cmake_minimum_required(VERSION 3.15)

# 设置项目名称
project(MyWindowsDriver)

# 设置驱动项目类型
set(CMAKE_SYSTEM_NAME Windows)
set(CMAKE_SYSTEM_VERSION 10.0)

# 设置WDK路径
set(WDK_ROOT "C:/Program Files (x86)/Windows Kits/10")
set(WDK_VERSION "10.0.22621.0")
set(WDF_VERSION "1.15")

# 找到WDK包含目录
set(WDK_INCLUDE_DIRS
    "${WDK_ROOT}/Include/${WDK_VERSION}/km"
    "${WDK_ROOT}/Include/${WDK_VERSION}/km/crt"
    "${WDK_ROOT}/Include/${WDK_VERSION}/shared"
    "${WDK_ROOT}/Include/wdf/kmdf/${WDF_VERSION}"
    "${WDK_ROOT}/Include/wdf/umdf/${WDF_VERSION}"
)

# 找到WDK库目录
if(CMAKE_SIZEOF_VOID_P EQUAL 8)
    set(WDK_LIB_DIR "${WDK_ROOT}/Lib/${WDK_VERSION}/km/x64")
    set(WDF_LIB_DIR "${WDK_ROOT}/Lib/wdf/kmdf/x64/${WDF_VERSION}")
else()
    set(WDK_LIB_DIR "${WDK_ROOT}/Lib/${WDK_VERSION}/km/x86")
    set(WDF_LIB_DIR "${WDK_ROOT}/Lib/wdf/kmdf/x86/${WDF_VERSION}")
endif()

# 设置驱动源文件
set(DRIVER_SOURCES
    entry.c
    src/vtbase/vtbase.c
)

# 使用驱动类型的库
add_library(${PROJECT_NAME} SHARED ${DRIVER_SOURCES})

# 设置包含目录
target_include_directories(${PROJECT_NAME} PRIVATE
    ${CMAKE_CURRENT_SOURCE_DIR}
    ${WDK_INCLUDE_DIRS}
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)

# 设置编译标志
target_compile_definitions(${PROJECT_NAME} PRIVATE
    _AMD64_
    _WIN64
    DRIVER
    NDEBUG
    WINNT=1
)

# 设置链接标志
target_link_libraries(${PROJECT_NAME} PRIVATE
    "${WDK_LIB_DIR}/ntoskrnl.lib"
    "${WDK_LIB_DIR}/hal.lib"
    "${WDK_LIB_DIR}/wmilib.lib"
    "${WDF_LIB_DIR}/WdfDriverEntry.lib"
    "${WDF_LIB_DIR}/WdfLdr.lib"
)

# 设置输出属性
set_target_properties(${PROJECT_NAME} PROPERTIES
    SUFFIX ".sys"
    PREFIX ""
    COMPILE_PDB_NAME "${PROJECT_NAME}"
    COMPILE_PDB_OUTPUT_DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}"
    ENABLE_EXPORTS TRUE
)
if(MSVC)
# 设置编译选项
target_compile_options(${PROJECT_NAME} PRIVATE
    /W4
    /WX-
    /Od        # Disable optimization for debugging
    /Oi        # Enable intrinsic functions
    /GF        # Enable string pooling
    /GS-       # Disable buffer security checks
    /Gy        # Enable function-level linking
    /Zc:wchar_t
    /Zc:forScope
    /Zc:inline
    /kernel    # Kernel mode compilation
    /GR-       # Disable RTTI
    /MT        # Use static runtime library
    /Oy        # Enable frame pointer omission
    /RTC-      # Disable runtime checks explicitly
)

# Disable runtime checks at definition level
target_compile_definitions(${PROJECT_NAME} PRIVATE
    _NO_CRT_STDIO_INLINE
    _NO_DEBUG_HEAP
    _NDEBUG
)

# Set debug information format and disable runtime checks in debug mode
set(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} /Z7 /RTC-")
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /Z7 /RTC-")

# Override CMake's default debug flags to remove /RTC1
string(REPLACE "/RTC1" "" CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG}")
string(REPLACE "/RTC1" "" CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG}")
endif()
if(MSVC)
# 设置链接标志
set_target_properties(${PROJECT_NAME} PROPERTIES
    LINK_FLAGS "/DRIVER /SUBSYSTEM:NATIVE /ENTRY:DriverEntry /MANIFEST:NO /NODEFAULTLIB /IGNORE:4210"
)

# 添加额外的库
target_link_libraries(${PROJECT_NAME} PRIVATE
    "${WDK_LIB_DIR}/ntoskrnl.lib"
    "${WDK_LIB_DIR}/hal.lib"
    "${WDK_LIB_DIR}/wmilib.lib"
    "${WDK_LIB_DIR}/bufferoverflowk.lib"
    "${WDK_LIB_DIR}/libcntpr.lib"
    "${WDF_LIB_DIR}/WdfDriverEntry.lib"
    "${WDF_LIB_DIR}/WdfLdr.lib"
)
endif()
# 禁用清单生成的另一种方式
if(MSVC)
    # 禁用清单生成
    set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} /MANIFEST:NO")
    set(CMAKE_SHARED_LINKER_FLAGS "${CMAKE_SHARED_LINKER_FLAGS} /MANIFEST:NO")
    set(CMAKE_MODULE_LINKER_FLAGS "${CMAKE_MODULE_LINKER_FLAGS} /MANIFEST:NO")
endif()
if(MSVC)
# 处理源文件编码问题
set_target_properties(${PROJECT_NAME} PROPERTIES
    COMPILE_FLAGS "/utf-8"
)
endif()

编写代码

enty.c

#include <ntddk.h>

//卸载驱动函数
VOID DriverUnload(PDRIVER_OBJECT pDriver)
{
    pDriver;
    //打印函数
    KdPrint(("驱动卸载成功\r\n"));
}

//加载驱动函数
NTSTATUS DriverEntry(PDRIVER_OBJECT pDriver, PUNICODE_STRING pReg)
{
    pDriver;
    pReg;
    //打印函数
    KdPrint(("驱动加载成功\r\n"));
    KdPrint(("我的第一个驱动程序\r\n"));

    //指定卸载驱动函数
    pDriver->DriverUnload = DriverUnload;

    //返回值 STATUS_SUCCESS 代表成功
    return STATUS_SUCCESS;
}

使用命令编译

1
2
3

cd build
cmake -A x64 ..
cmake --build . --config Release -j8

打开x64 Native Tools Command Prompt for VS 2022编译

#需要打开x64 Native Tools Command Prompt for VS 2022
cd D:\test\CC++\cmake\FirstDriver
mkdir D:\test\CC++\cmake\FirstDriver\build
cd D:\test\CC++\cmake\FirstDriver\build
cmake -S D:\test\CC++\cmake\FirstDriver -B D:\test\CC++\cmake\FirstDriver\build -G "Visual Studio 17 2022" -A x64
msbuild D:/test/CC++/cmake/FirstDriver/build/MyWindowsDriver.sln