Skip to content
Open
Changes from 11 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
75 changes: 74 additions & 1 deletion crates/wdk-build/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ mod bindgen;

use cargo_metadata::MetadataCommand;
use serde::{Deserialize, Serialize};
use serde_json::{Value, from_value};
use thiserror::Error;

use crate::utils::detect_windows_sdk_version;
Expand All @@ -45,6 +46,9 @@ pub struct Config {
cpu_architecture: CpuArchitecture,
/// Build configuration of driver
pub driver_config: DriverConfig,
/// List of features enabled for `wdk-sys` in resolved dependency graph
Comment thread
Alan632 marked this conversation as resolved.
#[serde(default)]
Comment thread
wmmc88 marked this conversation as resolved.
Outdated
enabled_api_subsets: Vec<ApiSubset>,
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This should be a BTreeSet<ApiSubset> rather than a Vec. Each variant should appear at most once, and the three call sites that do iter().any(|f| f == &ApiSubset::Hid) collapse to contains(&ApiSubset::Hid). BTreeSet over HashSet keeps Config's Serialize/Deserialize round-trips deterministic (although we dont actually use this right now)

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done!

}

/// The driver type with its associated configuration parameters
Expand Down Expand Up @@ -316,7 +320,8 @@ rustflags = [\"-C\", \"target-feature=+crt-static\"]
}

/// Subset of APIs in the Windows Driver Kit
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
#[derive(Debug, Copy, Clone, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub enum ApiSubset {
/// API subset typically required for all Windows drivers
Base,
Expand Down Expand Up @@ -397,6 +402,7 @@ impl Default for Config {
),
driver_config: DriverConfig::Wdm,
cpu_architecture: utils::detect_cpu_architecture_in_build_script(),
enabled_api_subsets: Vec::new(),
}
}
}
Expand Down Expand Up @@ -447,6 +453,41 @@ impl Config {
.exec()?;
let wdk_metadata = metadata::Wdk::try_from(&cargo_metadata)?;

// Find the `wdk-sys` package's PackageId in `cargo_metadata`
let wdk_sys_package_id = cargo_metadata
.packages
.iter()
.find(|pkg| pkg.name == "wdk-sys")
.map(|pkg| &pkg.id);
// Extract the features enabled for `wdk-sys`
// Produces an empty Vec if `wdk-sys` is not found in the dependency graph
let wdk_sys_enabled_features = match wdk_sys_package_id {
None => {
// wdk-sys not in dependency graph
Vec::new()
}
Some(id) => cargo_metadata
.resolve
.as_ref()
.and_then(|resolve| resolve.nodes.iter().find(|node| node.id == *id))
.map_or_else(
Comment thread
wmmc88 marked this conversation as resolved.
Outdated
Comment thread
wmmc88 marked this conversation as resolved.
Outdated
|| {
tracing::warn!(
"wdk-sys was found in packages but its features could not be \
determined from cargo metadata resolve. Feature-dependent libraries \
will not be linked automatically."
);
Vec::new()
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

the per-feature from_value(Value::String(f.clone())) is doing an unnecessary JSON round-trip and allocates/clones for every feature. Driving ApiSubset's existing Deserialize impl directly from &str via serde::de::IntoDeserializer avoids both:

use serde::de::{IntoDeserializer, value::{Error as ValueError, StrDeserializer}};

node.features
    .iter()
    .filter_map(|f| {
        ApiSubset::deserialize(f.as_str().into_deserializer()).ok()
    })
    .collect()

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

actually, this should go further and actually filter some known set of cargo features that aren't APISubsets (ie. default, nightly), instead of relying on a deserialization failure

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

To clarify the second point - is the desire to have a set of known features (aka all wdk-sys features) and warn the user if something not in that set is found in the metadata?

If we have a set of known "non ApiSubset" features that we filter out without notifying the user about it, then that sounds similar to silently filtering out via deserialization failure?

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we'd have a set of known non-api-subset features, and if we encounter one of those, it should hard error. anything thats not an api-subset feature and not in the known list should be treated as an unknown feature and we should bail. this is to help make sure at compile time that when there are new api subsets added, they map cleanly to the enum

Copy link
Copy Markdown
Contributor Author

@Alan632 Alan632 May 9, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

wait, we should hard error on encountering non-api-subset features? (aka default, nightly, test-stubs) Shouldn't we just filter those out and only hard error on anything that isn't listed in the [features] section in wdk-sys's Cargo.toml? Otherwise build will fail if we enable a feature like default or nightly in wdk-sys

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I went ahead and changed the implementation to hard error on found features that are neither in the non-api-subset-features list and the ApiSubset enum (eg. an unknown 'new' feature from wdk-sys), which I think is what we're both trying to express 😅 let's discuss offline if needed!

},
|node| {
node.features
.iter()
.filter_map(|f| from_value::<ApiSubset>(Value::String(f.clone())).ok())
.collect()
},
),
};

// Force rebuilds if any of the manifest files change (ex. if wdk metadata
// section is modified)
for manifest_path in metadata::iter_manifest_paths(cargo_metadata)
Expand All @@ -462,6 +503,7 @@ impl Config {

Ok(Self {
driver_config: wdk_metadata.driver_model,
enabled_api_subsets: wdk_sys_enabled_features,
..Default::default()
})
}
Expand Down Expand Up @@ -1157,6 +1199,13 @@ impl Config {
// provides no way to set a symbol's name without also exporting the symbol:
// https://github.com/rust-lang/rust/issues/67399
println!("cargo::rustc-cdylib-link-arg=/IGNORE:4216");

// Link against VhfKm.lib (Virtual HID Framework, kernel-mode) when
// the "hid" feature is enabled in wdk-sys. Required by drivers that
// create virtual HID devices.
if self.enabled_api_subsets.contains(&ApiSubset::Hid) {
println!("cargo::rustc-cdylib-link-arg=VhfKm.lib");
Comment on lines +1222 to +1226
}
Comment on lines +1222 to +1227
Comment on lines +1222 to +1227
Comment on lines +1222 to +1227
}
DriverConfig::Kmdf(_) => {
// Emit KMDF-specific libraries to link to
Expand Down Expand Up @@ -1186,6 +1235,13 @@ impl Config {
// Ignore `LNK4257: object file was not compiled for kernel mode; the image
// might not run` since `rustc` has no support for `/KERNEL`
println!("cargo::rustc-cdylib-link-arg=/IGNORE:4257");

// Link against VhfKm.lib (Virtual HID Framework, kernel-mode) when
// the "hid" feature is enabled in wdk-sys. Required by drivers that
// create virtual HID devices.
if self.enabled_api_subsets.contains(&ApiSubset::Hid) {
println!("cargo::rustc-cdylib-link-arg=VhfKm.lib");
Comment on lines +1258 to +1262
}
Comment thread
Alan632 marked this conversation as resolved.
Comment on lines +1258 to +1263
Comment on lines +1258 to +1263
Comment on lines +1258 to +1263
}
DriverConfig::Umdf(umdf_config) => {
// Emit UMDF-specific libraries to link to
Expand All @@ -1200,6 +1256,13 @@ impl Config {

// Linker arguments derived from WindowsDriver.UserMode.props in Ni(22H2) WDK
println!("cargo::rustc-cdylib-link-arg=/SUBSYSTEM:WINDOWS");

// Link against VhfUm.lib (Virtual HID Framework, user-mode) when
// the "hid" feature is enabled in wdk-sys. Required by drivers that
// create virtual HID devices.
if self.enabled_api_subsets.contains(&ApiSubset::Hid) {
println!("cargo::rustc-cdylib-link-arg=VhfUm.lib");
Comment on lines +1279 to +1283
}
Comment on lines +1279 to +1284
Comment on lines +1279 to +1284
Comment on lines +1279 to +1284
}
}

Expand Down Expand Up @@ -1826,18 +1889,22 @@ mod tests {
#[cfg(assert_matches_stabilized)]
assert_matches!(config.driver_config, DriverConfig::Wdm);
assert_eq!(config.cpu_architecture, CpuArchitecture::Amd64);
assert!(config.enabled_api_subsets.is_empty());
}

#[test]
fn wdm_config() {
let config = with_env(&[("CARGO_CFG_TARGET_ARCH", Some("x86_64"))], || Config {
driver_config: DriverConfig::Wdm,
enabled_api_subsets: vec![ApiSubset::Hid],
..Config::default()
});

#[cfg(assert_matches_stabilized)]
assert_matches!(config.driver_config, DriverConfig::Wdm);
assert_eq!(config.cpu_architecture, CpuArchitecture::Amd64);
assert!(config.enabled_api_subsets.contains(&ApiSubset::Hid));
assert!(!config.enabled_api_subsets.contains(&ApiSubset::Gpio));
}

#[test]
Expand Down Expand Up @@ -1867,6 +1934,7 @@ mod tests {
target_kmdf_version_minor: 15,
minimum_kmdf_version_minor: None,
}),
enabled_api_subsets: vec![ApiSubset::Hid, ApiSubset::Gpio],
..Config::default()
});

Expand All @@ -1880,6 +1948,8 @@ mod tests {
})
);
assert_eq!(config.cpu_architecture, CpuArchitecture::Amd64);
assert!(config.enabled_api_subsets.contains(&ApiSubset::Hid));
assert!(config.enabled_api_subsets.contains(&ApiSubset::Gpio));
}

#[test]
Expand Down Expand Up @@ -1909,6 +1979,7 @@ mod tests {
target_umdf_version_minor: 15,
minimum_umdf_version_minor: None,
}),
enabled_api_subsets: vec![ApiSubset::Usb],
..Config::default()
});

Expand All @@ -1922,6 +1993,8 @@ mod tests {
})
);
assert_eq!(config.cpu_architecture, CpuArchitecture::Arm64);
assert!(!config.enabled_api_subsets.contains(&ApiSubset::Hid));
assert!(config.enabled_api_subsets.contains(&ApiSubset::Usb));
}

#[test]
Expand Down
Loading