Skip to content

Commit eb47e2f

Browse files
wmmc88wmmc88
committed
feat: expand wdk-sys coverage to include spb-related headers
1 parent 90a52c2 commit eb47e2f

File tree

13 files changed

+136
-18
lines changed

13 files changed

+136
-18
lines changed

‎.vscode/settings.json

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
{
22
"rust-analyzer.rustfmt.extraArgs": [
3-
"+nightly"
3+
"+nightly",
4+
"--all"
45
],
56
"rust-analyzer.rustfmt.rangeFormatting.enable": true,
67
"evenBetterToml.formatter.crlf": true,

‎crates/wdk-build/src/lib.rs

+16-1
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,8 @@ pub enum ApiSubset {
199199
Wdf,
200200
/// API subset for HID (Human Interface Device) drivers: <https://learn.microsoft.com/en-us/windows-hardware/drivers/ddi/_hid/>
201201
Hid,
202+
/// API subset required for SPB (Serial Peripheral Bus) drivers: <https://learn.microsoft.com/en-us/windows-hardware/drivers/ddi/_spb/>
203+
Spb,
202204
}
203205

204206
impl Default for Config {
@@ -643,7 +645,7 @@ impl Config {
643645
match api_subset {
644646
ApiSubset::Base => match &self.driver_config {
645647
DriverConfig::Wdm | DriverConfig::Kmdf(_) => {
646-
vec!["ntifs.h", "ntddk.h"]
648+
vec!["ntifs.h", "ntddk.h", "ntstrsafe.h"]
647649
}
648650
DriverConfig::Umdf(_) => {
649651
vec!["windows.h"]
@@ -667,6 +669,19 @@ impl Config {
667669
hid_headers.extend(["HidSpiCx/1.0/hidspicx.h"]);
668670
}
669671

672+
hid_headers
673+
}
674+
ApiSubset::Spb => {
675+
let mut hid_headers = vec!["spb.h", "reshub.h"];
676+
677+
if let DriverConfig::Wdm | DriverConfig::Kmdf(_) = self.driver_config {
678+
hid_headers.extend(["pwmutil.h"]);
679+
}
680+
681+
if let DriverConfig::Kmdf(_) = self.driver_config {
682+
hid_headers.extend(["spb/1.1/spbcx.h"]);
683+
}
684+
670685
hid_headers
671686
}
672687
}

‎crates/wdk-sys/Cargo.toml

+1
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ wdk-macros.workspace = true
3838
default = []
3939

4040
hid = []
41+
spb = []
4142

4243
nightly = ["wdk-macros/nightly"]
4344
test-stubs = []

‎crates/wdk-sys/build.rs

+43-4
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,7 @@ const BINDGEN_FILE_GENERATORS_TUPLES: &[(&str, GenerateFn)] = &[
159159
("base.rs", generate_base),
160160
("wdf.rs", generate_wdf),
161161
("hid.rs", generate_hid),
162+
("spb.rs", generate_spb),
162163
];
163164

164165
fn initialize_tracing() -> Result<(), ParseError> {
@@ -223,6 +224,8 @@ fn generate_constants(out_path: &Path, config: &Config) -> Result<(), ConfigErro
223224
ApiSubset::Wdf,
224225
#[cfg(feature = "hid")]
225226
ApiSubset::Hid,
227+
#[cfg(feature = "spb")]
228+
ApiSubset::Spb,
226229
]);
227230
trace!(header_contents = ?header_contents);
228231

@@ -245,6 +248,8 @@ fn generate_types(out_path: &Path, config: &Config) -> Result<(), ConfigError> {
245248
ApiSubset::Wdf,
246249
#[cfg(feature = "hid")]
247250
ApiSubset::Hid,
251+
#[cfg(feature = "spb")]
252+
ApiSubset::Spb,
248253
]);
249254
trace!(header_contents = ?header_contents);
250255

@@ -295,10 +300,6 @@ fn generate_wdf(out_path: &Path, config: &Config) -> Result<(), ConfigError> {
295300
.allowlist_file("(?i).*wdf.*");
296301
trace!(bindgen_builder = ?bindgen_builder);
297302

298-
// As of NI WDK, this may generate an empty file due to no non-type and non-var
299-
// items in the wdf headers(i.e. functions are all inlined). This step is
300-
// intentionally left here in case older/newer WDKs have non-inlined functions
301-
// or new WDKs may introduce non-inlined functions.
302303
Ok(bindgen_builder
303304
.generate()
304305
.expect("Bindings should succeed to generate")
@@ -348,6 +349,42 @@ fn generate_hid(out_path: &Path, config: &Config) -> Result<(), ConfigError> {
348349
}
349350
}
350351

352+
fn generate_spb(out_path: &Path, config: &Config) -> Result<(), ConfigError> {
353+
cfg_if::cfg_if! {
354+
if #[cfg(feature = "spb")] {
355+
info!("Generating bindings to WDK: spb.rs");
356+
357+
let header_contents = config.bindgen_header_contents([ApiSubset::Base, ApiSubset::Wdf, ApiSubset::Spb]);
358+
trace!(header_contents = ?header_contents);
359+
360+
let bindgen_builder = {
361+
let mut builder = bindgen::Builder::wdk_default(config)?
362+
.with_codegen_config((CodegenConfig::TYPES | CodegenConfig::VARS).complement())
363+
.header_contents("spb-input.h", &header_contents);
364+
365+
// Only allowlist files in the spb-specific files to avoid duplicate definitions
366+
for header_file in config.headers(ApiSubset::Spb)
367+
{
368+
builder = builder.allowlist_file(format!("(?i).*{header_file}.*"));
369+
}
370+
builder
371+
};
372+
trace!(bindgen_builder = ?bindgen_builder);
373+
374+
Ok(bindgen_builder
375+
.generate()
376+
.expect("Bindings should succeed to generate")
377+
.write_to_file(out_path.join("spb.rs"))?)
378+
} else {
379+
let _ = (out_path, config); // Silence unused variable warnings when spb feature is not enabled
380+
381+
info!(
382+
"Skipping spb.rs generation since spb feature is not enabled");
383+
Ok(())
384+
}
385+
}
386+
}
387+
351388
/// Generates a `wdf_function_table.rs` file in `OUT_DIR` which contains the
352389
/// definition of `WDF_FUNCTION_TABLE`. This is required to be generated here
353390
/// since the size of the table is derived from either a global symbol
@@ -517,6 +554,8 @@ fn main() -> anyhow::Result<()> {
517554
ApiSubset::Wdf,
518555
#[cfg(feature = "hid")]
519556
ApiSubset::Hid,
557+
#[cfg(feature = "spb")]
558+
ApiSubset::Spb,
520559
])
521560
.as_bytes(),
522561
)?;

‎crates/wdk-sys/src/hid.rs

+6-3
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,12 @@
1515
generate documentation for their bindings"
1616
)]
1717
mod bindings {
18-
// allow wildcards for types module since underlying c code relies on all
19-
// type definitions being in scope
20-
#[allow(clippy::wildcard_imports)]
18+
#[allow(
19+
clippy::wildcard_imports,
20+
reason = "the underlying c code relies on all type definitions being in scope, which \
21+
results in the bindgen generated code relying on the generated types being in \
22+
scope as well"
23+
)]
2124
use crate::types::*;
2225

2326
include!(concat!(env!("OUT_DIR"), "/hid.rs"));

‎crates/wdk-sys/src/lib.rs

+10
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,16 @@ pub mod windows;
4141
))]
4242
pub mod hid;
4343

44+
#[cfg(all(
45+
any(
46+
driver_model__driver_type = "WDM",
47+
driver_model__driver_type = "KMDF",
48+
driver_model__driver_type = "UMDF"
49+
),
50+
feature = "spb"
51+
))]
52+
pub mod spb;
53+
4454
#[cfg(feature = "test-stubs")]
4555
pub mod test_stubs;
4656

‎crates/wdk-sys/src/ntddk.rs

+6-3
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,12 @@ pub use bindings::*;
1111

1212
#[allow(missing_docs)]
1313
mod bindings {
14-
// allow wildcards for types module since underlying c code relies on all
15-
// type definitions being in scope
16-
#[allow(clippy::wildcard_imports)]
14+
#[allow(
15+
clippy::wildcard_imports,
16+
reason = "the underlying c code relies on all type definitions being in scope, which \
17+
results in the bindgen generated code relying on the generated types being in \
18+
scope as well"
19+
)]
1720
use crate::types::*;
1821

1922
include!(concat!(env!("OUT_DIR"), "/ntddk.rs"));

‎crates/wdk-sys/src/spb.rs

+37
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
// Copyright (c) Microsoft Corporation
2+
// License: MIT OR Apache-2.0
3+
4+
//! Direct FFI bindings to SPB APIs from the Windows Driver Kit (WDK)
5+
//!
6+
//! This module contains all bindings to functions, constants, methods,
7+
//! constructors and destructors in the following headers: `spb.h`, `spbcx.h`,
8+
//! `reshub.h`, `pwmutil.h`. Types are not included in this module, but are
9+
//! available in the top-level `wdk_sys` module.
10+
11+
#[allow(
12+
missing_docs,
13+
reason = "most items in the WDK headers have no inline documentation, so bindgen is unable to \
14+
generate documentation for their bindings"
15+
)]
16+
mod bindings {
17+
#[allow(
18+
clippy::wildcard_imports,
19+
reason = "the underlying c code relies on all type definitions being in scope, which \
20+
results in the bindgen generated code relying on the generated types being in \
21+
scope as well"
22+
)]
23+
#[allow(
24+
unused_imports,
25+
reason = "in certain versions of the WDK, there are no functions related to SPB that can \
26+
be generated by bindgen, so these types are unused "
27+
)]
28+
use crate::types::*;
29+
30+
include!(concat!(env!("OUT_DIR"), "/spb.rs"));
31+
}
32+
#[allow(
33+
unused_imports,
34+
reason = "in certain versions of the WDK, there are no functions related to SPB that can be \
35+
generated by bindgen, so the `bindings` module is empty"
36+
)]
37+
pub use bindings::*;

‎crates/wdk-sys/src/wdf.rs

+6-3
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,12 @@ pub use bindings::*;
1212
#[allow(missing_docs)]
1313
#[allow(clippy::unreadable_literal)]
1414
mod bindings {
15-
// allow wildcards for types module since underlying c code relies on all
16-
// type definitions being in scope
17-
#[allow(clippy::wildcard_imports)]
15+
#[allow(
16+
clippy::wildcard_imports,
17+
reason = "the underlying c code relies on all type definitions being in scope, which \
18+
results in the bindgen generated code relying on the generated types being in \
19+
scope as well"
20+
)]
1821
use crate::types::*;
1922

2023
include!(concat!(env!("OUT_DIR"), "/wdf.rs"));

‎crates/wdk-sys/src/windows.rs

+6-3
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,12 @@ pub use bindings::*;
1111

1212
#[allow(missing_docs)]
1313
mod bindings {
14-
// allow wildcards for types module since underlying c code relies on all
15-
// type definitions being in scope
16-
#[allow(clippy::wildcard_imports)]
14+
#[allow(
15+
clippy::wildcard_imports,
16+
reason = "the underlying c code relies on all type definitions being in scope, which \
17+
results in the bindgen generated code relying on the generated types being in \
18+
scope as well"
19+
)]
1720
use crate::types::*;
1821

1922
include!(concat!(env!("OUT_DIR"), "/windows.rs"));

‎examples/sample-kmdf-driver/Cargo.toml

+1
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ wdk-sys = { path = "../../crates/wdk-sys", version = "0.3.0" }
3333
default = []
3434

3535
hid = ["wdk-sys/hid"]
36+
spb = ["wdk-sys/spb"]
3637

3738
nightly = ["wdk/nightly", "wdk-sys/nightly"]
3839

‎examples/sample-umdf-driver/Cargo.toml

+1
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ wdk-sys = { path = "../../crates/wdk-sys", version = "0.3.0" }
3131
default = []
3232

3333
hid = ["wdk-sys/hid"]
34+
spb = ["wdk-sys/spb"]
3435

3536
nightly = ["wdk/nightly", "wdk-sys/nightly"]
3637

‎examples/sample-wdm-driver/Cargo.toml

+1
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ wdk-sys = { path = "../../crates/wdk-sys", version = "0.3.0" }
3232
default = []
3333

3434
hid = ["wdk-sys/hid"]
35+
spb = ["wdk-sys/spb"]
3536

3637
nightly = ["wdk/nightly", "wdk-sys/nightly"]
3738

0 commit comments

Comments
 (0)