main.rs

// SPDX-License-Identifier: MIT OR Apache-2.0

#![allow(clippy::collapsible_if)]

mod arch;
mod cargo;
mod check_raw;
mod device_path;
mod disk;
mod net;
mod opt;
mod pipe;
mod platform;
mod qemu;
mod tpm;
mod util;

use crate::opt::{FmtOpt, TestOpt};
use anyhow::{Result, bail};
use arch::UefiArch;
use cargo::{Cargo, CargoAction, Feature, Package, TargetTypes};
use clap::Parser;
use itertools::Itertools;
use log::{LevelFilter, Metadata, Record};
use opt::{Action, BuildOpt, ClippyOpt, CovOpt, DocOpt, Opt, QemuOpt, TpmVersion};
use std::process::Command;
use util::run_cmd;

/// Builds various feature permutations that are valid for the `uefi` crate.
fn build_feature_permutations(opt: &BuildOpt) -> Result<()> {
    let all_package_features = Feature::package_features(Package::Uefi);
    for features in all_package_features.iter().powerset() {
        let features = features.iter().map(|f| **f).collect();

        let cargo = Cargo {
            action: CargoAction::Build,
            features,
            packages: vec![Package::Uefi],
            release: opt.build_mode.release,
            target: Some(*opt.target),
            warnings_as_errors: true,
            target_types: TargetTypes::BinsExamplesLib,
        };
        run_cmd(cargo.command()?)?;
    }

    Ok(())
}

fn build(opt: &BuildOpt) -> Result<()> {
    if opt.feature_permutations {
        return build_feature_permutations(opt);
    }

    let cargo = Cargo {
        action: CargoAction::Build,
        features: Feature::more_code(false, true),
        packages: Package::all_except_xtask(),
        release: opt.build_mode.release,
        target: Some(*opt.target),
        warnings_as_errors: false,
        target_types: TargetTypes::BinsExamplesLib,
    };
    run_cmd(cargo.command()?)
}

fn clippy(opt: &ClippyOpt) -> Result<()> {
    // Run clippy on all the UEFI packages.
    let cargo = Cargo {
        action: CargoAction::Clippy,
        features: Feature::more_code(false, true),
        packages: Package::all_except_xtask(),
        release: false,
        target: Some(*opt.target),
        warnings_as_errors: opt.warning.warnings_as_errors,
        target_types: TargetTypes::BinsExamplesLib,
    };
    run_cmd(cargo.command()?)?;

    // Run clippy on xtask.
    let cargo = Cargo {
        action: CargoAction::Clippy,
        features: Vec::new(),
        packages: vec![Package::Xtask],
        release: false,
        target: None,
        warnings_as_errors: opt.warning.warnings_as_errors,
        target_types: TargetTypes::Default,
    };
    run_cmd(cargo.command()?)
}

/// Generate a code coverage report.
fn code_coverage(opt: &CovOpt) -> Result<()> {
    if has_cmd("cargo-llvm-cov") {
        let cargo = Cargo {
            action: CargoAction::Coverage {
                lcov: opt.lcov,
                open: opt.open,
            },
            features: Feature::more_code(*opt.unstable, false),
            // Leave out uefi-macros; the compilation tests will just make
            // things slower without contributing anything to the coverage
            // report.
            packages: vec![Package::UefiRaw, Package::Uefi],
            release: false,
            target: None,
            warnings_as_errors: false,
            target_types: TargetTypes::Default,
        };
        run_cmd(cargo.command()?)
    } else {
        bail!("cargo-llvm-cov not found, see https://github.com/taiki-e/cargo-llvm-cov");
    }
}

/// Build docs.
fn doc(opt: &DocOpt) -> Result<()> {
    let cargo = Cargo {
        action: CargoAction::Doc {
            open: opt.open,
            document_private_items: opt.document_private_items,
        },
        features: Feature::more_code(*opt.unstable, true),
        packages: Package::published(),
        release: false,
        target: None,
        warnings_as_errors: opt.warning.warnings_as_errors,
        target_types: TargetTypes::Default,
    };
    run_cmd(cargo.command()?)
}

/// Run unit tests and doctests under Miri.
fn run_miri() -> Result<()> {
    let cargo = Cargo {
        action: CargoAction::Miri,
        features: [Feature::Alloc].into(),
        packages: [Package::Uefi].into(),
        release: false,
        target: None,
        warnings_as_errors: false,
        target_types: TargetTypes::Default,
    };
    run_cmd(cargo.command()?)
}

/// Build uefi-test-runner and run it in QEMU.
fn run_vm_tests(opt: &QemuOpt) -> Result<()> {
    let mut features = vec![];

    // Enable the DebugSupport test on supported platforms. Not available on
    // AARCH64 since edk2 commit f4213fed34.
    if *opt.target != UefiArch::AArch64 {
        features.push(Feature::DebugSupport);
    }

    // Enable the PXE test unless networking is disabled
    if !opt.disable_network {
        features.push(Feature::Pxe);
    }

    // Enable TPM tests if a TPM device is present.
    match opt.tpm {
        Some(TpmVersion::V1) => features.push(Feature::TpmV1),
        Some(TpmVersion::V2) => features.push(Feature::TpmV2),
        None => {}
    }

    // Enable the multi-processor test if not targeting AARCH64, and if KVM is
    // available. KVM is available on Linux generally, but not in our CI.
    if *opt.target != UefiArch::AArch64 && platform::is_linux() && !opt.ci {
        features.push(Feature::MultiProcessor);
    }

    // Enable `unstable` if requested.
    if *opt.unstable {
        features.push(Feature::TestUnstable);
    }

    // Build uefi-test-runner.
    let cargo = Cargo {
        action: CargoAction::Build,
        features,
        packages: vec![Package::UefiTestRunner],
        release: opt.build_mode.release,
        target: Some(*opt.target),
        warnings_as_errors: false,
        target_types: TargetTypes::BinsExamples,
    };
    run_cmd(cargo.command()?)?;

    qemu::run_qemu(*opt.target, opt)
}

/// 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(test_opt: &TestOpt) -> Result<()> {
    // Run xtask tests.
    let cargo = Cargo {
        action: CargoAction::Test,
        features: Vec::new(),
        packages: vec![Package::Xtask],
        release: false,
        target: None,
        warnings_as_errors: false,
        target_types: TargetTypes::Default,
    };
    run_cmd(cargo.command()?)?;

    let mut packages = vec![Package::UefiRaw, Package::Uefi];
    if !test_opt.skip_macro_tests {
        packages.push(Package::UefiMacros);
    }

    // Run uefi-rs and uefi-macros tests with `unstable` feature.
    let cargo = Cargo {
        action: CargoAction::Test,
        // Some tests may behave differently depending on the unstable feature.
        features: Feature::more_code(*test_opt.unstable, false),
        packages,
        release: false,
        // Use the host target so that tests can run without a VM.
        target: None,
        warnings_as_errors: false,
        target_types: TargetTypes::Default,
    };
    run_cmd(cargo.command()?)
}

/// Formats the project: nix, rust, and yml.
fn run_fmt_project(fmt_opt: &FmtOpt) -> Result<()> {
    let mut any_errors = false;

    {
        eprintln!("Formatting: file headers");

        match format_file_headers(fmt_opt) {
            Ok(_) => {
                eprintln!("✅ expected file headers present")
            }
            Err(e) => {
                if fmt_opt.check {
                    eprintln!("❌ {e:#?}");
                } else {
                    eprintln!("❌ rust formatter failed: {e:#?}");
                }
                any_errors = true;
            }
        }
    }

    // fmt rust
    {
        eprintln!("Formatting: rust");
        let mut command = Command::new("cargo");
        command.arg("fmt");
        if fmt_opt.check {
            command.arg("--check");
        }
        command
            .arg("--all")
            .arg("--")
            .arg("--config")
            .arg("imports_granularity=Module");

        match run_cmd(command) {
            Ok(_) => {
                eprintln!("✅ rust files formatted")
            }
            Err(e) => {
                if fmt_opt.check {
                    eprintln!("❌ rust files do not pass check");
                } else {
                    eprintln!("❌ rust formatter failed: {e:#?}");
                }
                any_errors = true;
            }
        }
    }

    // fmt yml
    if has_cmd("yamlfmt") {
        eprintln!("Formatting: yml");
        let mut command = Command::new("yamlfmt");
        if fmt_opt.check {
            command.arg("-lint");
        }
        // We only have yml files here.
        command.arg(".github");

        match run_cmd(command) {
            Ok(_) => {
                eprintln!("✅ yml files formatted")
            }
            Err(e) => {
                if fmt_opt.check {
                    eprintln!("❌ yml files do not pass check");
                } else {
                    eprintln!("❌ yml formatter failed: {e:#?}");
                }
                any_errors = true;
            }
        }
    } else {
        eprintln!("Formatting: yml - SKIPPED");
    }

    // fmt nix (by passing files as arguments)
    // `find . -name "*.nix" -type f -exec nix fmt {} \+`
    if has_cmd("find") && has_cmd("nixfmt") {
        eprintln!("Formatting: nix");
        let mut command = Command::new("find");
        command.arg(".");
        command.arg("-name");
        command.arg("*.nix");
        command.arg("-type");
        command.arg("f");
        command.arg("-exec");
        command.arg("nixfmt");
        if fmt_opt.check {
            command.arg("--check");
        }
        command.arg("{}");
        command.arg("+");

        match run_cmd(command) {
            Ok(_) => {
                eprintln!("✅ nix files formatted")
            }
            Err(e) => {
                if fmt_opt.check {
                    eprintln!("❌ nix files do not pass check");
                } else {
                    eprintln!("❌ nix formatter failed: {e:#?}");
                }
                any_errors = true;
            }
        }
    } else {
        eprintln!("Formatting: nix - SKIPPED");
    }

    if any_errors {
        bail!("one or more formatting errors");
    }

    Ok(())
}

/// Check that SPDX file headers are present.
///
/// If not in check mode, add any missing headers.
fn format_file_headers(fmt_opt: &FmtOpt) -> Result<()> {
    const EXPECTED_HEADER: &str = "// SPDX-License-Identifier: MIT OR Apache-2.0\n\n";

    // Path prefixes that should not be checked/formatted.
    const EXCLUDE_PATH_PREFIXES: &[&str] = &[
        // A user copying the template or uefi-std-example does not need to use
        // our license.
        "template/src/main.rs",
        "uefi-std-example/src/main.rs",
        // This directory contains short code snippets used in `trybuild` tests,
        // no license needed.
        "uefi-macros/tests/ui/",
    ];

    // Recursively get Rust files
    let mut cmd = Command::new("git");
    cmd.args(["ls-files", "*.rs"]);
    let output = cmd.output()?;
    if !output.status.success() {
        bail!("command failed: {}", output.status);
    }
    let mut paths: Vec<&str> = std::str::from_utf8(&output.stdout)?.lines().collect();

    // Filter out excluded paths.
    paths.retain(|path| {
        !EXCLUDE_PATH_PREFIXES
            .iter()
            .any(|prefix| path.starts_with(prefix))
    });

    // Paths that are missing the file header (only used in check mode).
    let mut missing = Vec::new();

    // Format or check each path.
    for path in paths {
        let text = fs_err::read_to_string(path)?;
        if text.starts_with(EXPECTED_HEADER) {
            // File header is present, nothing to do.
            continue;
        }

        if fmt_opt.check {
            // Add to the list of paths missing file headers.
            missing.push(path);
        } else {
            // Rewrite the file to prepend the header.
            let text = EXPECTED_HEADER.to_owned() + &text;
            fs_err::write(path, text)?;
        }
    }

    if fmt_opt.check && !missing.is_empty() {
        bail!("expected header missing from {}", missing.join(", "));
    }

    Ok(())
}

fn has_cmd(target_cmd: &str) -> bool {
    #[cfg(target_os = "windows")]
    let mut cmd = Command::new("where");
    #[cfg(target_family = "unix")]
    let mut cmd = Command::new("which");
    cmd.arg(target_cmd);
    run_cmd(cmd).is_ok()
}

fn install_logger() {
    struct Logger;

    impl log::Log for Logger {
        fn enabled(&self, _: &Metadata) -> bool {
            true
        }

        fn log(&self, record: &Record) {
            println!("[{}] {}", record.level(), record.args());
        }

        fn flush(&self) {}
    }

    static LOGGER: Logger = Logger;

    log::set_logger(&LOGGER)
        .map(|()| log::set_max_level(LevelFilter::Info))
        .unwrap();
}

fn main() -> Result<()> {
    let opt = Opt::parse();

    install_logger();

    match &opt.action {
        Action::Build(build_opt) => build(build_opt),
        Action::CheckRaw(_) => check_raw::check_raw(),
        Action::Clippy(clippy_opt) => clippy(clippy_opt),
        Action::Cov(cov_opt) => code_coverage(cov_opt),
        Action::Doc(doc_opt) => doc(doc_opt),
        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(test_opt) => run_host_tests(test_opt),
        Action::Fmt(fmt_opt) => run_fmt_project(fmt_opt),
    }
}