[分享]CMake MinGW OLLVM编译Windows驱动-编程技术-看雪-安全社区|安全招聘|kanxue.com

[分享]CMake MinGW OLLVM编译Windows驱动

发表于: 2025-6-4 09:02 666

[分享]CMake MinGW OLLVM编译Windows驱动

mb_vmdpdaoc 活跃值

2025-6-4 09:02

666

是我从FindWDK项目上面改的，原版只能用VS，看Pull Request有人发了MinGW改版，我自己测试原版MinGW是可以用，但是到OLLVM上面去就不行了，我就再改了一版

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

262

263

264

265

266

267

268

269

270

271

272

273

274

275

276

277

278

279

280

281

282

283

284

285

286

287

288

289

290

291

292

293

294

295

296

297

298

299

300

301

302

303

304

305

306

307

308

309

310

311

312

313

314

315

316

# Redistribution and use is allowed under the OSI-approved 3-clause BSD license.
# Copyright (c) 2018 Sergey Podobry (sergey.podobry at gmail.com). All rights reserved.
 
#.rst:
# FindWDK
# ----------
#
# This module searches for the installed Windows Development Kit (WDK) and
# exposes commands for creating kernel drivers and kernel libraries.
#
# Output variables:
# - `WDK_FOUND` -- if false, do not try to use WDK
# - `WDK_ROOT` -- where WDK is installed
# - `WDK_VERSION` -- the version of the selected WDK
# - `WDK_WINVER` -- the WINVER used for kernel drivers and libraries
#        (default value is `0x0601` and can be changed per target or globally)
# - `WDK_NTDDI_VERSION` -- the NTDDI_VERSION used for kernel drivers and libraries,
#                          if not set, the value will be automatically calculated by WINVER
#        (default value is left blank and can be changed per target or globally)
#
# Example usage:
#
#   find_package(WDK REQUIRED)
#
#   wdk_add_library(KmdfCppLib STATIC KMDF 1.15
#       KmdfCppLib.h
#       KmdfCppLib.cpp
#       )
#   target_include_directories(KmdfCppLib INTERFACE .)
#
#   wdk_add_driver(KmdfCppDriver KMDF 1.15
#       Main.cpp
#       )
#   target_link_libraries(KmdfCppDriver KmdfCppLib)
#
 
if (DEFINED ENV{WDKContentRoot})
    file(GLOB WDK_NTDDK_FILES
            "$ENV{WDKContentRoot}/Include/*/km/ntddk.h" # WDK 10
            "$ENV{WDKContentRoot}/Include/km/ntddk.h" # WDK 8.0, 8.1
            )
else ()
    file(GLOB WDK_NTDDK_FILES
            "C:/Program Files (x86)/Windows Kits/*/Include/*/km/ntddk.h" # WDK 10
            "C:/Program Files (x86)/Windows Kits/*/Include/km/ntddk.h" # WDK 8.0, 8.1
            )
endif ()
 
if (WDK_NTDDK_FILES)
    if (NOT CMAKE_VERSION VERSION_LESS 3.18.0)
        list(SORT WDK_NTDDK_FILES COMPARE NATURAL) # sort to use the latest available WDK
    endif ()
    list(GET WDK_NTDDK_FILES -1 WDK_LATEST_NTDDK_FILE)
endif ()
 
include(FindPackageHandleStandardArgs)
find_package_handle_standard_args(WDK REQUIRED_VARS WDK_LATEST_NTDDK_FILE)
 
if (MINGW)
    get_filename_component(MINGW_BIN_DIR ${CMAKE_RC_COMPILER} DIRECTORY)
    get_filename_component(MINGW_DIR ${MINGW_BIN_DIR} DIRECTORY)
    if (REACTOS_SDK)
        set(REACTOS_DDK "${REACTOS_SDK}/include/ddk" CACHE PATH "The path to ReactOS DDK")
        set(REACTOS_CRT_HEADERS "${REACTOS_SDK}/include/CRT" CACHE PATH "The path to ReactOS CRT headers")
        set(REACTOS_PSDK_HEADERS "${REACTOS_SDK}/include/psdk" CACHE PATH "The path to ReactOS PSDK headers")
        set(REACTOS_XDK_HEADERS "${REACTOS_SDK}/include/xdk" CACHE PATH "The path to ReactOS XDK headers")
    endif (REACTOS_SDK)
    message(STATUS "MINGW_DIR ${MINGW_DIR}")
 
    file(GLOB MINGW_INCLUDE_EXCPT_HS "${MINGW_DIR}/*/include/excpt.h")
    list(GET MINGW_INCLUDE_EXCPT_HS -1 MINGW_INCLUDE_EXCPT_H)
    get_filename_component(MINGW_INCLUDE_DIR ${MINGW_INCLUDE_EXCPT_H} DIRECTORY)
 
    set(MINGW_INCLUDE_DIR "${MINGW_INCLUDE_DIR}" CACHE PATH "The path to MinGW include")
    set(MINGW_DDK "${MINGW_INCLUDE_DIR}/ddk" CACHE PATH "The path to MinGW DDK")
 
    if (NOT MINGW_DDK)
        message(FATAL_ERROR "MINGW_DDK not found!")
        return()
    else ()
        message(STATUS "MINGW_DDK_ROOT: " ${MINGW_DDK})
    endif ()
else (MINGW)
    if (NOT WDK_LATEST_NTDDK_FILE)
        return()
    endif ()
endif (MINGW)
 
get_filename_component(WDK_ROOT ${WDK_LATEST_NTDDK_FILE} DIRECTORY)
get_filename_component(WDK_ROOT ${WDK_ROOT} DIRECTORY)
get_filename_component(WDK_VERSION ${WDK_ROOT} NAME)
get_filename_component(WDK_ROOT ${WDK_ROOT} DIRECTORY)
if (NOT WDK_ROOT MATCHES ".*/[0-9][0-9.]*$") # WDK 10 has a deeper nesting level
    get_filename_component(WDK_ROOT ${WDK_ROOT} DIRECTORY) # go up once more
    set(WDK_LIB_VERSION "${WDK_VERSION}")
    set(WDK_INC_VERSION "${WDK_VERSION}")
else () # WDK 8.0, 8.1
    set(WDK_INC_VERSION "")
    foreach (VERSION winv6.3 win8 win7)
        if (EXISTS "${WDK_ROOT}/Lib/${VERSION}/")
            set(WDK_LIB_VERSION "${VERSION}")
            break()
        endif ()
    endforeach ()
    set(WDK_VERSION "${WDK_LIB_VERSION}")
endif ()
 
message(STATUS "WDK_ROOT: " ${WDK_ROOT})
message(STATUS "WDK_VERSION: " ${WDK_VERSION})
 
set(WDK_WINVER "0x0601" CACHE STRING "Default WINVER for WDK targets")
set(WDK_NTDDI_VERSION "" CACHE STRING "Specified NTDDI_VERSION for WDK targets if needed")
 
set(WDK_ADDITIONAL_FLAGS_FILE "${CMAKE_CURRENT_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/wdkflags.h")
file(WRITE ${WDK_ADDITIONAL_FLAGS_FILE} "#pragma runtime_checks(\"suc\", off)")
 
if (MINGW)
    set(WDK_COMPILE_FLAGS
            "-shared"
            "-nostartfiles"
            "-nostdlib"
            "-lntoskrnl"
            "-lhal"
            "-lndis"
            "-mdll"
            "-fno-rtti"
            "-mrtd" # __stdcall by default
            "-fpack-struct=8" # set struct alignment
            "-fms-extensions" # microsoft visual c++ compatibility
            "-fdelayed-template-parsing"
            "-fms-compatibility"
            "-D_MSC_VER=1300"
            "-masm=intel" #MSVC headers use intel syntax
            #"-std=c++17"
            )
    include_directories(SYSTEM "${LIB_DIR}/Include")
    #set(WDK_COMPILE_FLAGS
    #    "/kernel"  # create kernel mode binary
    #    "/FIwarning.h" # disable warnings in WDK headers
    #    "/FI${WDK_ADDITIONAL_FLAGS_FILE}" # include file to disable RTC
    #    )
elseif (MSVC)
    set(WDK_COMPILE_FLAGS
            "/Zp8" # set struct alignment
            "/GF"  # enable string pooling
            "/GR-" # disable RTTI
            "/Gz" # __stdcall by default
            "/kernel"  # create kernel mode binary
            "/FIwarning.h" # disable warnings in WDK headers
            "/FI${WDK_ADDITIONAL_FLAGS_FILE}" # include file to disable RTC
            )
endif (MINGW)
 
 
set(WDK_COMPILE_DEFINITIONS "WINNT=1")
set(WDK_COMPILE_DEFINITIONS_DEBUG "MSC_NOOPT;DEPRECATE_DDK_FUNCTIONS=1;DBG=1")
 
if (CMAKE_SIZEOF_VOID_P EQUAL 4)
    list(APPEND WDK_COMPILE_DEFINITIONS "_X86_=1;i386=1;STD_CALL")
    set(WDK_PLATFORM "x86")
elseif (CMAKE_SIZEOF_VOID_P EQUAL 8)
    list(APPEND WDK_COMPILE_DEFINITIONS "_WIN64;_AMD64_;AMD64")
    set(WDK_PLATFORM "x64")
else ()
    message(FATAL_ERROR "Unsupported architecture")
endif ()
 
if (MINGW)
    string(CONCAT WDK_LINK_FLAGS
            "--subsystem,native "
            "--image-base,0x10000 "
            "--file-alignment,0x1000 "
            "--section-alignment,0x1000 "
            "--entry,_DriverEntry@8 "
            )
elseif (MSVC)
    string(CONCAT WDK_LINK_FLAGS
            "/MANIFEST:NO " #
            "/DRIVER " #
            "/OPT:REF " #
            "/INCREMENTAL:NO " #
            "/OPT:ICF " #
            "/SUBSYSTEM:NATIVE " #
            "/MERGE:_TEXT=.text;_PAGE=PAGE " #
            "/NODEFAULTLIB " # do not link default CRT
            "/SECTION:INIT,d " #
            "/VERSION:10.0 " #
            )
elseif (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
    string(CONCAT WDK_LINK_FLAGS
            "-Wl,/MANIFEST:NO "
            "-Wl,/DRIVER "
            "-Wl,/OPT:REF "
            "-Wl,/INCREMENTAL:NO "
            "-Wl,/OPT:ICF "
            "-Wl,/SUBSYSTEM:NATIVE "
            "-Wl,/MERGE:_TEXT=.text;_PAGE=PAGE "
            "-Wl,/NODEFAULTLIB "
            "-Wl,/VERSION:10.0 "
            )
endif (MINGW)
 
# Generate imported targets for WDK lib files
file(GLOB WDK_LIBRARIES "${WDK_ROOT}/Lib/${WDK_LIB_VERSION}/km/${WDK_PLATFORM}/*.lib")
foreach (LIBRARY IN LISTS WDK_LIBRARIES)
    get_filename_component(LIBRARY_NAME ${LIBRARY} NAME_WE)
    string(TOUPPER ${LIBRARY_NAME} LIBRARY_NAME)
    add_library(WDK::${LIBRARY_NAME} INTERFACE IMPORTED)
    set_property(TARGET WDK::${LIBRARY_NAME} PROPERTY INTERFACE_LINK_LIBRARIES ${LIBRARY})
endforeach (LIBRARY)
unset(WDK_LIBRARIES)
 
function(wdk_add_driver _target)
    cmake_parse_arguments(WDK "" "KMDF;WINVER;NTDDI_VERSION" "" ${ARGN})
 
    add_executable(${_target} ${WDK_UNPARSED_ARGUMENTS})
 
    set_target_properties(${_target} PROPERTIES SUFFIX ".sys")
 
    if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
        set_target_properties(${_target} PROPERTIES
                COMPILE_OPTIONS "${WDK_COMPILE_FLAGS}"
                COMPILE_FLAGS "${CMAKE_CXX_FLAGS_RELEASE}"
                )
    else ()
        set_target_properties(${_target} PROPERTIES COMPILE_OPTIONS "${WDK_COMPILE_FLAGS}")
    endif ()
 
    set_target_properties(${_target} PROPERTIES COMPILE_DEFINITIONS
            "${WDK_COMPILE_DEFINITIONS};$<$<CONFIG:Debug>:${WDK_COMPILE_DEFINITIONS_DEBUG}>;_WIN32_WINNT=${WDK_WINVER}"
            )
    set_target_properties(${_target} PROPERTIES LINK_FLAGS "${WDK_LINK_FLAGS}")
    if (WDK_NTDDI_VERSION)
        target_compile_definitions(${_target} PRIVATE NTDDI_VERSION=${WDK_NTDDI_VERSION})
    endif ()
 
    target_include_directories(${_target} SYSTEM PRIVATE
            "${WDK_ROOT}/Include/${WDK_INC_VERSION}/shared"
            "${WDK_ROOT}/Include/${WDK_INC_VERSION}/km"
            "${WDK_ROOT}/Include/${WDK_INC_VERSION}/km/crt"
            )
 
    target_link_libraries(${_target} WDK::NTOSKRNL WDK::HAL WDK::BUFFEROVERFLOWK WDK::WMILIB)
 
    if (CMAKE_SIZEOF_VOID_P EQUAL 4)
        target_link_libraries(${_target} WDK::MEMCMP)
    endif ()
 
    if (DEFINED WDK_KMDF)
        target_include_directories(${_target} SYSTEM PRIVATE "${WDK_ROOT}/Include/wdf/kmdf/${WDK_KMDF}")
        target_link_libraries(${_target}
                "${WDK_ROOT}/Lib/wdf/kmdf/${WDK_PLATFORM}/${WDK_KMDF}/WdfDriverEntry.lib"
                "${WDK_ROOT}/Lib/wdf/kmdf/${WDK_PLATFORM}/${WDK_KMDF}/WdfLdr.lib"
                )
 
        if (CMAKE_SIZEOF_VOID_P EQUAL 4)
            if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "-Wl,/ENTRY:FxDriverEntry@8")
            else ()
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "/ENTRY:FxDriverEntry@8")
            endif ()
        elseif (CMAKE_SIZEOF_VOID_P EQUAL 8)
            if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "-Wl,/ENTRY:FxDriverEntry")
            else ()
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "/ENTRY:FxDriverEntry")
            endif ()
        endif ()
    else ()
        if (CMAKE_SIZEOF_VOID_P EQUAL 4)
            if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "-Wl,/ENTRY:GsDriverEntry@8")
            else ()
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "/ENTRY:GsDriverEntry@8")
            endif ()
        elseif (CMAKE_SIZEOF_VOID_P EQUAL 8)
            if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "-Wl,/ENTRY:GsDriverEntry")
            else ()
                set_property(TARGET ${_target} APPEND_STRING PROPERTY LINK_FLAGS "/ENTRY:GsDriverEntry")
            endif ()
        endif ()
    endif ()
endfunction()
 
function(wdk_add_library _target)
    cmake_parse_arguments(WDK "" "KMDF;WINVER;NTDDI_VERSION" "" ${ARGN})
 
    add_library(${_target} ${WDK_UNPARSED_ARGUMENTS})
 
    if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
        set_target_properties(${_target} PROPERTIES
                COMPILE_OPTIONS "${WDK_COMPILE_FLAGS}"
                COMPILE_FLAGS "${CMAKE_CXX_FLAGS_RELEASE}"
                )
    else ()
        set_target_properties(${_target} PROPERTIES COMPILE_OPTIONS "${WDK_COMPILE_FLAGS}")
    endif ()
 
    set_target_properties(${_target} PROPERTIES COMPILE_DEFINITIONS
            "${WDK_COMPILE_DEFINITIONS};$<$<CONFIG:Debug>:${WDK_COMPILE_DEFINITIONS_DEBUG};>_WIN32_WINNT=${WDK_WINVER}"
            )
    if (WDK_NTDDI_VERSION)
        target_compile_definitions(${_target} PRIVATE NTDDI_VERSION=${WDK_NTDDI_VERSION})
    endif ()
 
    target_include_directories(${_target} SYSTEM PRIVATE
            "${WDK_ROOT}/Include/${WDK_INC_VERSION}/shared"
            "${WDK_ROOT}/Include/${WDK_INC_VERSION}/km"
            "${WDK_ROOT}/Include/${WDK_INC_VERSION}/km/crt"
            )
 
    if (DEFINED WDK_KMDF)
        target_include_directories(${_target} SYSTEM PRIVATE "${WDK_ROOT}/Include/wdf/kmdf/${WDK_KMDF}")
    endif ()
endfunction()

[培训]科锐逆向工程师培训第53期2025年7月8日开班！

#驱动开发 #开源分享 #开发技巧

收藏・6

免费・2

支持

最新回复 (1)
青丝梦雪币： 1553 活跃值： (2894) 能力值： ( LV5，RANK：60 ) 在线值：发帖 9 回帖 154 粉丝 23 关注私信	青丝梦 1 2 楼 mark 2025-6-8 20:33 0
	游客登录 \| 注册方可回帖回帖表情雪币赚取及消费高级回复

mb_vmdpdaoc

发帖

回帖

RANK

关注

私信

他的文章

[分享]CMake MinGW OLLVM编译Windows驱动 667

关于我们

联系我们

企业服务

看雪公众号