mod.rs

use uefi::boot::{self, OpenProtocolParams};
use uefi::prelude::*;
use uefi::proto::loaded_image::LoadedImage;
use uefi::{proto, Identify};

pub fn test(st: &mut SystemTable<Boot>) {
    info!("Testing various protocols");

    console::test(st);

    let bt = st.boot_services();
    find_protocol(bt);
    test_protocols_per_handle(bt);
    test_protocols_per_handle_freestanding();
    test_test_protocol_freestanding();

    debug::test(bt);
    device_path::test(bt);
    driver::test(bt);
    load::test(bt);
    loaded_image::test(bt);
    media::test(bt);
    network::test(bt);
    pi::test(bt);
    rng::test(bt);
    shell_params::test(bt);
    string::test(bt);
    misc::test(bt);

    #[cfg(any(
        target_arch = "x86",
        target_arch = "x86_64",
        target_arch = "arm",
        target_arch = "aarch64"
    ))]
    shim::test(bt);
    tcg::test(bt);
}

fn find_protocol(bt: &BootServices) {
    let handles = bt
        .find_handles::<proto::console::text::Output>()
        .expect("Failed to retrieve list of handles");

    assert!(
        !handles.is_empty(),
        "There should be at least one implementation of Simple Text Output (stdout)"
    );
}

fn test_protocols_per_handle(bt: &BootServices) {
    let pph = bt
        .protocols_per_handle(bt.image_handle())
        .expect("Failed to get protocols for image handle");

    info!("Image handle has {} protocols", pph.len());

    // Check that one of the image's protocols is `LoadedImage`.
    assert!(pph.iter().any(|guid| **guid == LoadedImage::GUID));
}

fn test_protocols_per_handle_freestanding() {
    let pph = boot::protocols_per_handle(boot::image_handle()).unwrap();
    info!("Image handle has {} protocols", pph.len());
    // Check that one of the image's protocols is `LoadedImage`.
    assert!(pph.iter().any(|guid| **guid == LoadedImage::GUID));
}

fn test_test_protocol_freestanding() {
    assert!(boot::test_protocol::<LoadedImage>(OpenProtocolParams {
        handle: boot::image_handle(),
        agent: boot::image_handle(),
        controller: None,
    })
    .unwrap());
}

mod console;
mod debug;
mod device_path;
mod driver;
mod load;
mod loaded_image;
mod media;
mod misc;
mod network;
mod pi;
mod rng;
mod shell_params;
#[cfg(any(
    target_arch = "x86",
    target_arch = "x86_64",
    target_arch = "arm",
    target_arch = "aarch64"
))]
mod shim;
mod string;
mod tcg;