feat: add linker args for VHF lib to wdk-build

crates/wdk-build/src/lib.rs

@@ -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;
@@ -45,6 +46,8 @@ 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
+    enabled_api_subsets: Vec<ApiSubset>,
 }
 
 /// The driver type with its associated configuration parameters
@@ -316,7 +319,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,
@@ -397,6 +401,7 @@ impl Default for Config {
             ),
             driver_config: DriverConfig::Wdm,
             cpu_architecture: utils::detect_cpu_architecture_in_build_script(),
+            enabled_api_subsets: Vec::new(),
         }
     }
 }
@@ -447,6 +452,37 @@ 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 = wdk_sys_package_id
+            .and_then(|id| {
+                cargo_metadata
+                    .resolve
+                    .as_ref()
+                    .and_then(|resolve| resolve.nodes.iter().find(|node| node.id == *id))
+            })
+            .map_or_else(
+                || {
+                    tracing::warn!(
+                        "Could not detect wdk-sys features from cargo metadata resolve. \
+                         Feature-dependent libraries will not be linked automatically."
+                    );
+                    Vec::new()
+                },
+                |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)
@@ -462,6 +498,7 @@ impl Config {
 
         Ok(Self {
             driver_config: wdk_metadata.driver_model,
+            enabled_api_subsets: wdk_sys_enabled_features,
             ..Default::default()
         })
     }
@@ -1157,6 +1194,17 @@ 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
+                    .iter()
+                    .any(|f| f == &ApiSubset::Hid)
+                {
+                    println!("cargo::rustc-cdylib-link-arg=VhfKm.lib");
+                }
             }
             DriverConfig::Kmdf(_) => {
                 // Emit KMDF-specific libraries to link to
@@ -1186,6 +1234,17 @@ 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
+                    .iter()
+                    .any(|f| f == &ApiSubset::Hid)
+                {
+                    println!("cargo::rustc-cdylib-link-arg=VhfKm.lib");
+                }
             }
             DriverConfig::Umdf(umdf_config) => {
                 // Emit UMDF-specific libraries to link to
@@ -1200,6 +1259,17 @@ 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
+                    .iter()
+                    .any(|f| f == &ApiSubset::Hid)
+                {
+                    println!("cargo::rustc-cdylib-link-arg=VhfUm.lib");
+                }
             }
         }