xtask: add switch for stable and unstable functionality

Author: phip1611

xtask/src/cargo.rs

@@ -51,6 +51,7 @@ pub enum Feature {
     Logger,
     PanicOnLoggerErrors,
     Unstable,
+    UnstableAlloc,
 
     // `uefi-services` features.
     PanicHandler,
@@ -69,6 +70,7 @@ impl Feature {
             Self::Logger => "logger",
             Self::PanicOnLoggerErrors => "panic-on-logger-errors",
             Self::Unstable => "unstable",
+            Self::UnstableAlloc => "unstable_alloc",
 
             Self::PanicHandler => "uefi-services/panic_handler",
             Self::Qemu => "uefi-services/qemu",
@@ -95,8 +97,18 @@ impl Feature {
     }
 
     /// Set of features that enables more code in the root uefi crate.
-    pub fn more_code() -> Vec<Self> {
-        vec![Self::GlobalAllocator, Self::Alloc, Self::Logger]
+    /// # Parameters
+    /// - `include_unstable` - add all functionality behind the `unstable` feature
+    /// - `runtime_features` - add all functionality that effect the runtime of Rust
+    pub fn more_code(include_unstable: bool, runtime_features: bool) -> Vec<Self> {
+        let mut base_features = vec![Self::Alloc, Self::Logger];
+        if include_unstable {
+            base_features.extend([Self::Unstable, Self::UnstableAlloc])
+        }
+        if runtime_features {
+            base_features.extend([Self::GlobalAllocator])
+        }
+        base_features
     }
 
     fn comma_separated_string(features: &[Feature]) -> String {
@@ -307,8 +319,20 @@ mod tests {
     #[test]
     fn test_comma_separated_features() {
         assert_eq!(
-            Feature::comma_separated_string(&Feature::more_code()),
-            "global_allocator,alloc,logger"
+            Feature::comma_separated_string(&Feature::more_code(false, false)),
+            "alloc,logger"
+        );
+        assert_eq!(
+            Feature::comma_separated_string(&Feature::more_code(false, true)),
+            "alloc,logger,global_allocator"
+        );
+        assert_eq!(
+            Feature::comma_separated_string(&Feature::more_code(true, false)),
+            "alloc,logger,unstable,unstable_alloc"
+        );
+        assert_eq!(
+            Feature::comma_separated_string(&Feature::more_code(true, true)),
+            "alloc,logger,unstable,unstable_alloc,global_allocator"
         );
     }
 

xtask/src/main.rs

@@ -9,6 +9,7 @@ mod platform;
 mod qemu;
 mod util;
 
+use crate::opt::TestOpt;
 use anyhow::Result;
 use cargo::{fix_nested_cargo_env, Cargo, CargoAction, Feature, Package, TargetTypes};
 use clap::Parser;
@@ -47,7 +48,7 @@ fn build(opt: &BuildOpt) -> Result<()> {
 
     let cargo = Cargo {
         action: CargoAction::Build,
-        features: Feature::more_code(),
+        features: Feature::more_code(opt.include_unstable, true),
         packages: Package::all_except_xtask(),
         release: opt.build_mode.release,
         target: Some(*opt.target),
@@ -61,7 +62,8 @@ fn clippy(opt: &ClippyOpt) -> Result<()> {
     // Run clippy on all the UEFI packages.
     let cargo = Cargo {
         action: CargoAction::Clippy,
-        features: Feature::more_code(),
+        // for all possible features
+        features: Feature::more_code(true, true),
         packages: Package::all_except_xtask(),
         release: false,
         target: Some(*opt.target),
@@ -87,7 +89,8 @@ fn clippy(opt: &ClippyOpt) -> Result<()> {
 fn doc(opt: &DocOpt) -> Result<()> {
     let cargo = Cargo {
         action: CargoAction::Doc { open: opt.open },
-        features: Feature::more_code(),
+        // for all possible features
+        features: Feature::more_code(true, true),
         packages: Package::published(),
         release: false,
         target: None,
@@ -140,7 +143,7 @@ fn run_vm_tests(opt: &QemuOpt) -> Result<()> {
 /// Run unit tests and doctests on the host. Most of uefi-rs is tested
 /// with VM tests, but a few things like macros and data types can be
 /// tested with regular tests.
-fn run_host_tests() -> Result<()> {
+fn run_host_tests(test_opt: &TestOpt) -> Result<()> {
     // Run xtask tests.
     let cargo = Cargo {
         action: CargoAction::Test,
@@ -156,7 +159,8 @@ fn run_host_tests() -> Result<()> {
     // Run uefi-rs and uefi-macros tests.
     let cargo = Cargo {
         action: CargoAction::Test,
-        features: vec![Feature::Alloc],
+        // No runtime features as it is not possible to test a #[global_allocator] in Rust tests
+        features: Feature::more_code(test_opt.include_unstable, false),
         // Don't test uefi-services (or the packages that depend on it)
         // as it has lang items that conflict with `std`.
         packages: vec![Package::Uefi, Package::UefiMacros],
@@ -219,7 +223,7 @@ fn main() -> Result<()> {
         Action::GenCode(gen_opt) => device_path::gen_code(gen_opt),
         Action::Miri(_) => run_miri(),
         Action::Run(qemu_opt) => run_vm_tests(qemu_opt),
-        Action::Test(_) => run_host_tests(),
+        Action::Test(test_opt) => run_host_tests(test_opt),
         Action::TestLatestRelease(_) => test_latest_release(),
     }
 }

xtask/src/opt.rs

@@ -63,6 +63,11 @@ pub struct BuildOpt {
     #[clap(flatten)]
     pub build_mode: BuildModeOpt,
 
+    /// Include all features behind the "unstable" gate. uefi-rs must build without unstable
+    /// functionality on stable (eventually) and with it in our nightly MSRV.
+    #[clap(long, action)]
+    pub include_unstable: bool,
+
     /// Build multiple times to check that different feature
     /// combinations work.
     #[clap(long, action)]
@@ -145,7 +150,12 @@ pub struct QemuOpt {
 
 /// Run unit tests and doctests on the host.
 #[derive(Debug, Parser)]
-pub struct TestOpt;
+pub struct TestOpt {
+    /// Include all features behind the "unstable" gate. uefi-rs must build without unstable
+    /// functionality on stable (eventually) and with it in our nightly MSRV.
+    #[clap(long, action)]
+    pub include_unstable: bool,
+}
 
 /// Build the template against the crates.io packages.
 #[derive(Debug, Parser)]