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

crates/wdk-build/src/lib.rs

@@ -45,6 +45,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: Vec<String>,
 }
 
 /// The driver type with its associated configuration parameters
@@ -397,6 +399,7 @@ impl Default for Config {
             ),
             driver_config: DriverConfig::Wdm,
             cpu_architecture: utils::detect_cpu_architecture_in_build_script(),
+            enabled: Vec::new(),
         }
     }
 }
@@ -447,6 +450,32 @@ 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.clone(),
+            );
+
         // 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 +491,7 @@ impl Config {
 
         Ok(Self {
             driver_config: wdk_metadata.driver_model,
+            enabled: wdk_sys_enabled_features,
             ..Default::default()
         })
     }
@@ -1157,6 +1187,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.contains(&"hid".to_string()) {
+                    println!("cargo::rustc-cdylib-link-arg=VhfKm.lib");
+                }
             }
             DriverConfig::Kmdf(_) => {
                 // Emit KMDF-specific libraries to link to
@@ -1186,6 +1223,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.contains(&"hid".to_string()) {
+                    println!("cargo::rustc-cdylib-link-arg=VhfKm.lib");
+                }
             }
             DriverConfig::Umdf(umdf_config) => {
                 // Emit UMDF-specific libraries to link to
@@ -1200,6 +1244,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.contains(&"hid".to_string()) {
+                    println!("cargo::rustc-cdylib-link-arg=VhfUm.lib");
+                }
             }
         }