mirror of
https://github.com/edera-dev/sprout.git
synced 2025-12-19 20:40:17 +00:00
Compare commits
22 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
b9b34394cd
|
|||
| 3c467fef65 | |||
| cfe7088970 | |||
|
9d3a022e08
|
|||
|
cc90199d61
|
|||
|
cff55322fc
|
|||
|
d2bef03437
|
|||
|
37ab0406bb
|
|||
|
a77be3c282
|
|||
|
87d608366f
|
|||
|
e7d2438e5f
|
|||
|
f82d24a206
|
|||
|
fe714cc411
|
|||
|
000e3ea6c7
|
|||
|
44de1bebd3
|
|||
|
4d62ac2ce2
|
|||
|
|
8d87fcfd2d | ||
|
1a2b103675
|
|||
|
5e1bc4659b
|
|||
|
27c82d24d0
|
|||
|
e2aa19174e
|
|||
|
fb342e12cb
|
42
.github/ISSUE_TEMPLATE/bug-report.yml
vendored
Normal file
42
.github/ISSUE_TEMPLATE/bug-report.yml
vendored
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
name: Report a bug
|
||||||
|
description: File a bug report.
|
||||||
|
title: "bug: "
|
||||||
|
labels: ["bug", "triage"]
|
||||||
|
type: bug
|
||||||
|
assignees:
|
||||||
|
- edera-dev/engineering
|
||||||
|
body:
|
||||||
|
- type: markdown
|
||||||
|
attributes:
|
||||||
|
value: |
|
||||||
|
Thanks for taking the time to fill out this bug report!
|
||||||
|
- type: input
|
||||||
|
id: version
|
||||||
|
attributes:
|
||||||
|
label: Version / Commit
|
||||||
|
description: What version of Sprout are you running?
|
||||||
|
validations:
|
||||||
|
required: true
|
||||||
|
- type: textarea
|
||||||
|
id: what-happened
|
||||||
|
attributes:
|
||||||
|
label: What happened?
|
||||||
|
description: Tell us what you expected to happen.
|
||||||
|
placeholder: Tell us what you see!
|
||||||
|
value: "A bug happened!"
|
||||||
|
validations:
|
||||||
|
required: true
|
||||||
|
- type: textarea
|
||||||
|
id: logs
|
||||||
|
attributes:
|
||||||
|
label: Log output
|
||||||
|
description: Please provide any relevant log output.
|
||||||
|
render: log
|
||||||
|
- type: checkboxes
|
||||||
|
id: terms
|
||||||
|
attributes:
|
||||||
|
label: Code of Conduct
|
||||||
|
description: By submitting this report, you agree to follow our [Code of Conduct](https://github.com/edera-dev/sprout/blob/main/CODE_OF_CONDUCT.md).
|
||||||
|
options:
|
||||||
|
- label: I agree to follow the Sprout Code of Conduct.
|
||||||
|
required: true
|
||||||
6
.github/ISSUE_TEMPLATE/config.yml
vendored
Normal file
6
.github/ISSUE_TEMPLATE/config.yml
vendored
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
blank_issues_enabled: false
|
||||||
|
contact_links:
|
||||||
|
- name: Ask a question
|
||||||
|
url: https://github.com/edera-dev/sprout/discussions
|
||||||
|
about: Please ask and answer questions here.
|
||||||
|
# Note that GitHub will automatically display our security policy in the new issue form.
|
||||||
29
.github/ISSUE_TEMPLATE/feature-request.yml
vendored
Normal file
29
.github/ISSUE_TEMPLATE/feature-request.yml
vendored
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
name: Request a feature
|
||||||
|
description: Request a feature.
|
||||||
|
title: "want: "
|
||||||
|
labels: ["enhancement", "triage"]
|
||||||
|
type: feature
|
||||||
|
assignees:
|
||||||
|
- edera-dev/engineering
|
||||||
|
body:
|
||||||
|
- type: markdown
|
||||||
|
attributes:
|
||||||
|
value: |
|
||||||
|
Thanks for taking the time to fill out a feature request!
|
||||||
|
- type: textarea
|
||||||
|
id: description
|
||||||
|
attributes:
|
||||||
|
label: Description
|
||||||
|
description:
|
||||||
|
placeholder: Tell us what you want to see!
|
||||||
|
value: "Your hopes and dreams here!"
|
||||||
|
validations:
|
||||||
|
required: true
|
||||||
|
- type: checkboxes
|
||||||
|
id: terms
|
||||||
|
attributes:
|
||||||
|
label: Code of Conduct
|
||||||
|
description: By submitting this report, you agree to follow our [Code of Conduct](https://github.com/edera-dev/sprout/blob/main/CODE_OF_CONDUCT.md).
|
||||||
|
options:
|
||||||
|
- label: I agree to follow the Sprout Code of Conduct.
|
||||||
|
required: true
|
||||||
@@ -10,7 +10,7 @@ permissions:
|
|||||||
contents: read # Needed to checkout the repository.
|
contents: read # Needed to checkout the repository.
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}"
|
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ github.sha }}"
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
@@ -12,7 +12,7 @@ permissions:
|
|||||||
contents: read # Needed to checkout the repository.
|
contents: read # Needed to checkout the repository.
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}"
|
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ github.sha }}"
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
@@ -12,7 +12,7 @@ permissions:
|
|||||||
contents: read # Needed to checkout the repository.
|
contents: read # Needed to checkout the repository.
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}"
|
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ github.sha }}"
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
@@ -19,7 +19,7 @@ permissions:
|
|||||||
contents: read # Needed to checkout the repository.
|
contents: read # Needed to checkout the repository.
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}"
|
group: "${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ github.sha }}"
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
2
Cargo.lock
generated
2
Cargo.lock
generated
@@ -116,7 +116,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "edera-sprout"
|
name = "edera-sprout"
|
||||||
version = "0.0.14"
|
version = "0.0.15"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"image",
|
"image",
|
||||||
|
|||||||
@@ -2,7 +2,7 @@
|
|||||||
name = "edera-sprout"
|
name = "edera-sprout"
|
||||||
description = "Modern UEFI bootloader"
|
description = "Modern UEFI bootloader"
|
||||||
license = "Apache-2.0"
|
license = "Apache-2.0"
|
||||||
version = "0.0.14"
|
version = "0.0.15"
|
||||||
homepage = "https://sprout.edera.dev"
|
homepage = "https://sprout.edera.dev"
|
||||||
repository = "https://github.com/edera-dev/sprout"
|
repository = "https://github.com/edera-dev/sprout"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|||||||
@@ -2,7 +2,7 @@
|
|||||||
ARG RUST_PROFILE=release
|
ARG RUST_PROFILE=release
|
||||||
ARG RUST_TARGET_SUBDIR=release
|
ARG RUST_TARGET_SUBDIR=release
|
||||||
|
|
||||||
FROM --platform=$BUILDPLATFORM rustlang/rust:nightly-alpine@sha256:7cba2edabb6ba0e92cd806cd1e0acae99d50f63e5b9c9ad842766d13c896d68c AS build
|
FROM --platform=$BUILDPLATFORM rustlang/rust:nightly-alpine@sha256:34532121803db17008af0cdc4e2e1210466cb257cc9d3840dac42d706640fee5 AS build
|
||||||
RUN apk --no-cache add musl-dev busybox-static
|
RUN apk --no-cache add musl-dev busybox-static
|
||||||
ARG RUST_PROFILE
|
ARG RUST_PROFILE
|
||||||
RUN adduser -S -s /bin/sh build
|
RUN adduser -S -s /bin/sh build
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
use crate::context::SproutContext;
|
use crate::context::SproutContext;
|
||||||
|
use crate::integrations::bootloader_interface::BootloaderInterface;
|
||||||
use crate::utils;
|
use crate::utils;
|
||||||
use crate::utils::media_loader::MediaLoaderHandle;
|
use crate::utils::media_loader::MediaLoaderHandle;
|
||||||
use crate::utils::media_loader::constants::linux::LINUX_EFI_INITRD_MEDIA_GUID;
|
use crate::utils::media_loader::constants::linux::LINUX_EFI_INITRD_MEDIA_GUID;
|
||||||
@@ -102,6 +103,10 @@ pub fn chainload(context: Rc<SproutContext>, configuration: &ChainloadConfigurat
|
|||||||
initrd_handle = Some(handle);
|
initrd_handle = Some(handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Mark execution of an entry in the bootloader interface.
|
||||||
|
BootloaderInterface::mark_exec(context.root().timer())
|
||||||
|
.context("unable to mark execution of boot entry in bootloader interface")?;
|
||||||
|
|
||||||
// Start the loaded image.
|
// Start the loaded image.
|
||||||
// This call might return, or it may pass full control to another image that will never return.
|
// This call might return, or it may pass full control to another image that will never return.
|
||||||
// Capture the result to ensure we can return an error if the image fails to start, but only
|
// Capture the result to ensure we can return an error if the image fails to start, but only
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
use crate::actions::ActionDeclaration;
|
use crate::actions::ActionDeclaration;
|
||||||
use crate::options::SproutOptions;
|
use crate::options::SproutOptions;
|
||||||
|
use crate::platform::timer::PlatformTimer;
|
||||||
use anyhow::anyhow;
|
use anyhow::anyhow;
|
||||||
use anyhow::{Result, bail};
|
use anyhow::{Result, bail};
|
||||||
use std::cmp::Reverse;
|
use std::cmp::Reverse;
|
||||||
@@ -12,22 +13,29 @@ const CONTEXT_FINALIZE_ITERATION_LIMIT: usize = 100;
|
|||||||
|
|
||||||
/// Declares a root context for Sprout.
|
/// Declares a root context for Sprout.
|
||||||
/// This contains data that needs to be shared across Sprout.
|
/// This contains data that needs to be shared across Sprout.
|
||||||
#[derive(Default)]
|
|
||||||
pub struct RootContext {
|
pub struct RootContext {
|
||||||
/// The actions that are available in Sprout.
|
/// The actions that are available in Sprout.
|
||||||
actions: BTreeMap<String, ActionDeclaration>,
|
actions: BTreeMap<String, ActionDeclaration>,
|
||||||
/// The device path of the loaded Sprout image.
|
/// The device path of the loaded Sprout image.
|
||||||
loaded_image_path: Option<Box<DevicePath>>,
|
loaded_image_path: Option<Box<DevicePath>>,
|
||||||
|
/// Platform timer started at the beginning of the boot process.
|
||||||
|
timer: PlatformTimer,
|
||||||
/// The global options of Sprout.
|
/// The global options of Sprout.
|
||||||
options: SproutOptions,
|
options: SproutOptions,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl RootContext {
|
impl RootContext {
|
||||||
/// Creates a new root context with the `loaded_image_device_path` which will be stored
|
/// Creates a new root context with the `loaded_image_device_path` which will be stored
|
||||||
/// in the context for easy access.
|
/// in the context for easy access. We also provide a `timer` which is used to measure elapsed
|
||||||
pub fn new(loaded_image_device_path: Box<DevicePath>, options: SproutOptions) -> Self {
|
/// time for the bootloader.
|
||||||
|
pub fn new(
|
||||||
|
loaded_image_device_path: Box<DevicePath>,
|
||||||
|
timer: PlatformTimer,
|
||||||
|
options: SproutOptions,
|
||||||
|
) -> Self {
|
||||||
Self {
|
Self {
|
||||||
actions: BTreeMap::new(),
|
actions: BTreeMap::new(),
|
||||||
|
timer,
|
||||||
loaded_image_path: Some(loaded_image_device_path),
|
loaded_image_path: Some(loaded_image_device_path),
|
||||||
options,
|
options,
|
||||||
}
|
}
|
||||||
@@ -43,6 +51,11 @@ impl RootContext {
|
|||||||
&mut self.actions
|
&mut self.actions
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Access the platform timer that is started at the beginning of the boot process.
|
||||||
|
pub fn timer(&self) -> &PlatformTimer {
|
||||||
|
&self.timer
|
||||||
|
}
|
||||||
|
|
||||||
/// Access the device path of the loaded Sprout image.
|
/// Access the device path of the loaded Sprout image.
|
||||||
pub fn loaded_image_path(&self) -> Result<&DevicePath> {
|
pub fn loaded_image_path(&self) -> Result<&DevicePath> {
|
||||||
self.loaded_image_path
|
self.loaded_image_path
|
||||||
|
|||||||
@@ -28,6 +28,7 @@ pub struct BootableEntry {
|
|||||||
context: Rc<SproutContext>,
|
context: Rc<SproutContext>,
|
||||||
declaration: EntryDeclaration,
|
declaration: EntryDeclaration,
|
||||||
default: bool,
|
default: bool,
|
||||||
|
pin_name: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl BootableEntry {
|
impl BootableEntry {
|
||||||
@@ -44,6 +45,7 @@ impl BootableEntry {
|
|||||||
context,
|
context,
|
||||||
declaration,
|
declaration,
|
||||||
default: false,
|
default: false,
|
||||||
|
pin_name: false,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -72,6 +74,11 @@ impl BootableEntry {
|
|||||||
self.default
|
self.default
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Fetch whether the entry is pinned, which prevents prefixing.
|
||||||
|
pub fn is_pin_name(&self) -> bool {
|
||||||
|
self.pin_name
|
||||||
|
}
|
||||||
|
|
||||||
/// Swap out the context of the entry.
|
/// Swap out the context of the entry.
|
||||||
pub fn swap_context(&mut self, context: Rc<SproutContext>) {
|
pub fn swap_context(&mut self, context: Rc<SproutContext>) {
|
||||||
self.context = context;
|
self.context = context;
|
||||||
@@ -87,6 +94,11 @@ impl BootableEntry {
|
|||||||
self.default = true;
|
self.default = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Mark this entry as being pinned, which prevents prefixing.
|
||||||
|
pub fn mark_pin_name(&mut self) {
|
||||||
|
self.pin_name = true;
|
||||||
|
}
|
||||||
|
|
||||||
/// Prepend the name of the entry with `prefix`.
|
/// Prepend the name of the entry with `prefix`.
|
||||||
pub fn prepend_name_prefix(&mut self, prefix: &str) {
|
pub fn prepend_name_prefix(&mut self, prefix: &str) {
|
||||||
self.name.insert_str(0, prefix);
|
self.name.insert_str(0, prefix);
|
||||||
|
|||||||
@@ -9,9 +9,7 @@ use uefi::fs::{FileSystem, Path};
|
|||||||
use uefi::proto::device_path::DevicePath;
|
use uefi::proto::device_path::DevicePath;
|
||||||
use uefi::proto::media::file::{File, FileSystemVolumeLabel};
|
use uefi::proto::media::file::{File, FileSystemVolumeLabel};
|
||||||
use uefi::proto::media::fs::SimpleFileSystem;
|
use uefi::proto::media::fs::SimpleFileSystem;
|
||||||
use uefi::proto::media::partition::PartitionInfo;
|
use uefi::{CString16, Guid};
|
||||||
use uefi::{CString16, Guid, Handle};
|
|
||||||
use uefi_raw::Status;
|
|
||||||
|
|
||||||
/// The filesystem device match extractor.
|
/// The filesystem device match extractor.
|
||||||
/// This extractor finds a filesystem using some search criteria and returns
|
/// This extractor finds a filesystem using some search criteria and returns
|
||||||
@@ -41,53 +39,20 @@ pub struct FilesystemDeviceMatchExtractor {
|
|||||||
pub fallback: Option<String>,
|
pub fallback: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Represents the partition UUIDs for a filesystem.
|
|
||||||
struct PartitionIds {
|
|
||||||
/// The UUID of the partition.
|
|
||||||
partition_uuid: Guid,
|
|
||||||
/// The type UUID of the partition.
|
|
||||||
type_uuid: Guid,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Fetches the partition UUIDs for the specified filesystem handle.
|
|
||||||
fn fetch_partition_uuids(handle: Handle) -> Result<Option<PartitionIds>> {
|
|
||||||
// Open the partition info protocol for this handle.
|
|
||||||
let partition_info = uefi::boot::open_protocol_exclusive::<PartitionInfo>(handle);
|
|
||||||
|
|
||||||
match partition_info {
|
|
||||||
Ok(partition_info) => {
|
|
||||||
// GPT partitions have a unique partition GUID.
|
|
||||||
// MBR does not.
|
|
||||||
if let Some(gpt) = partition_info.gpt_partition_entry() {
|
|
||||||
let uuid = gpt.unique_partition_guid;
|
|
||||||
let type_uuid = gpt.partition_type_guid;
|
|
||||||
Ok(Some(PartitionIds {
|
|
||||||
partition_uuid: uuid,
|
|
||||||
type_uuid: type_uuid.0,
|
|
||||||
}))
|
|
||||||
} else {
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Err(error) => {
|
|
||||||
// If the filesystem does not have a partition, that is okay.
|
|
||||||
if error.status() == Status::NOT_FOUND || error.status() == Status::UNSUPPORTED {
|
|
||||||
Ok(None)
|
|
||||||
} else {
|
|
||||||
// We should still handle other errors gracefully.
|
|
||||||
Err(error).context("unable to open filesystem partition info")?;
|
|
||||||
unreachable!()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Extract a filesystem device path using the specified `context` and `extractor` configuration.
|
/// Extract a filesystem device path using the specified `context` and `extractor` configuration.
|
||||||
pub fn extract(
|
pub fn extract(
|
||||||
context: Rc<SproutContext>,
|
context: Rc<SproutContext>,
|
||||||
extractor: &FilesystemDeviceMatchExtractor,
|
extractor: &FilesystemDeviceMatchExtractor,
|
||||||
) -> Result<String> {
|
) -> Result<String> {
|
||||||
|
// If no criteria are provided, bail with an error.
|
||||||
|
if extractor.has_label.is_none()
|
||||||
|
&& extractor.has_item.is_none()
|
||||||
|
&& extractor.has_partition_uuid.is_none()
|
||||||
|
&& extractor.has_partition_type_uuid.is_none()
|
||||||
|
{
|
||||||
|
bail!("at least one criteria is required for filesystem-device-match");
|
||||||
|
}
|
||||||
|
|
||||||
// Find all the filesystems inside the UEFI stack.
|
// Find all the filesystems inside the UEFI stack.
|
||||||
let handles = uefi::boot::find_handles::<SimpleFileSystem>()
|
let handles = uefi::boot::find_handles::<SimpleFileSystem>()
|
||||||
.context("unable to find filesystem handles")?;
|
.context("unable to find filesystem handles")?;
|
||||||
@@ -97,30 +62,49 @@ pub fn extract(
|
|||||||
// This defines whether a match has been found.
|
// This defines whether a match has been found.
|
||||||
let mut has_match = false;
|
let mut has_match = false;
|
||||||
|
|
||||||
// Extract the partition info for this filesystem.
|
|
||||||
// There is no guarantee that the filesystem has a partition.
|
|
||||||
let partition_info =
|
|
||||||
fetch_partition_uuids(handle).context("unable to fetch partition info")?;
|
|
||||||
|
|
||||||
// Check if the partition info matches partition uuid criteria.
|
// Check if the partition info matches partition uuid criteria.
|
||||||
if let Some(ref partition_info) = partition_info
|
if let Some(ref has_partition_uuid) = extractor.has_partition_uuid {
|
||||||
&& let Some(ref has_partition_uuid) = extractor.has_partition_uuid
|
// Parse the partition uuid from the extractor.
|
||||||
{
|
|
||||||
let parsed_uuid = Guid::from_str(has_partition_uuid)
|
let parsed_uuid = Guid::from_str(has_partition_uuid)
|
||||||
.map_err(|e| anyhow!("unable to parse has-partition-uuid: {}", e))?;
|
.map_err(|e| anyhow!("unable to parse has-partition-uuid: {}", e))?;
|
||||||
if partition_info.partition_uuid != parsed_uuid {
|
|
||||||
|
// Fetch the root of the device.
|
||||||
|
let root = uefi::boot::open_protocol_exclusive::<DevicePath>(handle)
|
||||||
|
.context("unable to fetch the device path of the filesystem")?
|
||||||
|
.deref()
|
||||||
|
.to_boxed();
|
||||||
|
|
||||||
|
// Fetch the partition uuid for this filesystem.
|
||||||
|
let partition_uuid = utils::partition_guid(&root, utils::PartitionGuidForm::Partition)
|
||||||
|
.context("unable to fetch the partition uuid of the filesystem")?;
|
||||||
|
|
||||||
|
// Compare the partition uuid to the parsed uuid.
|
||||||
|
// If it does not match, continue to the next filesystem.
|
||||||
|
if partition_uuid != Some(parsed_uuid) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
has_match = true;
|
has_match = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if the partition info matches partition type uuid criteria.
|
// Check if the partition info matches partition type uuid criteria.
|
||||||
if let Some(ref partition_info) = partition_info
|
if let Some(ref has_partition_type_uuid) = extractor.has_partition_type_uuid {
|
||||||
&& let Some(ref has_partition_type_uuid) = extractor.has_partition_type_uuid
|
// Parse the partition type uuid from the extractor.
|
||||||
{
|
|
||||||
let parsed_uuid = Guid::from_str(has_partition_type_uuid)
|
let parsed_uuid = Guid::from_str(has_partition_type_uuid)
|
||||||
.map_err(|e| anyhow!("unable to parse has-partition-type-uuid: {}", e))?;
|
.map_err(|e| anyhow!("unable to parse has-partition-type-uuid: {}", e))?;
|
||||||
if partition_info.type_uuid != parsed_uuid {
|
|
||||||
|
// Fetch the root of the device.
|
||||||
|
let root = uefi::boot::open_protocol_exclusive::<DevicePath>(handle)
|
||||||
|
.context("unable to fetch the device path of the filesystem")?
|
||||||
|
.deref()
|
||||||
|
.to_boxed();
|
||||||
|
|
||||||
|
// Fetch the partition uuid for this filesystem.
|
||||||
|
let partition_type_uuid =
|
||||||
|
utils::partition_guid(&root, utils::PartitionGuidForm::Partition)
|
||||||
|
.context("unable to fetch the partition uuid of the filesystem")?;
|
||||||
|
// Compare the partition type uuid to the parsed uuid.
|
||||||
|
// If it does not match, continue to the next filesystem.
|
||||||
|
if partition_type_uuid != Some(parsed_uuid) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
has_match = true;
|
has_match = true;
|
||||||
|
|||||||
@@ -83,13 +83,16 @@ pub fn generate(context: Rc<SproutContext>, bls: &BlsConfiguration) -> Result<Ve
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Get the file name of the filesystem item.
|
// Get the file name of the filesystem item.
|
||||||
let name = entry.file_name().to_string();
|
let mut name = entry.file_name().to_string();
|
||||||
|
|
||||||
// Ignore files that are not .conf files.
|
// Ignore files that are not .conf files.
|
||||||
if !name.to_lowercase().ends_with(".conf") {
|
if !name.to_lowercase().ends_with(".conf") {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Remove the .conf extension.
|
||||||
|
name.truncate(name.len() - 5);
|
||||||
|
|
||||||
// Create a mutable path so we can append the file name to produce the full path.
|
// Create a mutable path so we can append the file name to produce the full path.
|
||||||
let mut full_entry_path = entries_path.to_path_buf();
|
let mut full_entry_path = entries_path.to_path_buf();
|
||||||
full_entry_path.push(entry.file_name());
|
full_entry_path.push(entry.file_name());
|
||||||
@@ -125,13 +128,21 @@ pub fn generate(context: Rc<SproutContext>, bls: &BlsConfiguration) -> Result<Ve
|
|||||||
context.set("options", options);
|
context.set("options", options);
|
||||||
context.set("initrd", initrd);
|
context.set("initrd", initrd);
|
||||||
|
|
||||||
// Add the entry to the list with a frozen context.
|
// Produce a new bootable entry.
|
||||||
entries.push(BootableEntry::new(
|
let mut entry = BootableEntry::new(
|
||||||
name,
|
name,
|
||||||
bls.entry.title.clone(),
|
bls.entry.title.clone(),
|
||||||
context.freeze(),
|
context.freeze(),
|
||||||
bls.entry.clone(),
|
bls.entry.clone(),
|
||||||
));
|
);
|
||||||
|
|
||||||
|
// Pin the entry name to prevent prefixing.
|
||||||
|
// This is needed as the bootloader interface requires the name to be
|
||||||
|
// the same as the entry file name, minus the .conf extension.
|
||||||
|
entry.mark_pin_name();
|
||||||
|
|
||||||
|
// Add the entry to the list with a frozen context.
|
||||||
|
entries.push(entry);
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(entries)
|
Ok(entries)
|
||||||
|
|||||||
2
src/integrations.rs
Normal file
2
src/integrations.rs
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
/// Implements support for the bootloader interface specification.
|
||||||
|
pub mod bootloader_interface;
|
||||||
125
src/integrations/bootloader_interface.rs
Normal file
125
src/integrations/bootloader_interface.rs
Normal file
@@ -0,0 +1,125 @@
|
|||||||
|
use crate::platform::timer::PlatformTimer;
|
||||||
|
use crate::utils::device_path_subpath;
|
||||||
|
use anyhow::{Context, Result};
|
||||||
|
use uefi::proto::device_path::DevicePath;
|
||||||
|
use uefi::{CString16, Guid, guid};
|
||||||
|
use uefi_raw::table::runtime::{VariableAttributes, VariableVendor};
|
||||||
|
|
||||||
|
/// The name of the bootloader to tell the system.
|
||||||
|
const LOADER_NAME: &str = "Sprout";
|
||||||
|
|
||||||
|
/// Bootloader Interface support.
|
||||||
|
pub struct BootloaderInterface;
|
||||||
|
|
||||||
|
impl BootloaderInterface {
|
||||||
|
/// Bootloader Interface GUID from https://systemd.io/BOOT_LOADER_INTERFACE
|
||||||
|
const VENDOR: VariableVendor = VariableVendor(guid!("4a67b082-0a4c-41cf-b6c7-440b29bb8c4f"));
|
||||||
|
|
||||||
|
/// Tell the system that Sprout was initialized at the current time.
|
||||||
|
pub fn mark_init(timer: &PlatformTimer) -> Result<()> {
|
||||||
|
Self::mark_time("LoaderTimeInitUSec", timer)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system that Sprout is about to execute the boot entry.
|
||||||
|
pub fn mark_exec(timer: &PlatformTimer) -> Result<()> {
|
||||||
|
Self::mark_time("LoaderTimeExecUSec", timer)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system that Sprout is about to display the menu.
|
||||||
|
pub fn mark_menu(timer: &PlatformTimer) -> Result<()> {
|
||||||
|
Self::mark_time("LoaderTimeMenuUsec", timer)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system about the current time as measured by the platform timer.
|
||||||
|
/// Sets the variable specified by `key` to the number of microseconds.
|
||||||
|
fn mark_time(key: &str, timer: &PlatformTimer) -> Result<()> {
|
||||||
|
// Measure the elapsed time since the hardware timer was started.
|
||||||
|
let elapsed = timer.elapsed_since_lifetime();
|
||||||
|
Self::set_cstr16(key, &elapsed.as_micros().to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system what loader is being used.
|
||||||
|
pub fn set_loader_info() -> Result<()> {
|
||||||
|
Self::set_cstr16("LoaderInfo", LOADER_NAME)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system the relative path to the partition root of the current bootloader.
|
||||||
|
pub fn set_loader_path(path: &DevicePath) -> Result<()> {
|
||||||
|
let subpath = device_path_subpath(path).context("unable to get loader path subpath")?;
|
||||||
|
Self::set_cstr16("LoaderImageIdentifier", &subpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system what the partition GUID of the ESP Sprout was booted from is.
|
||||||
|
pub fn set_partition_guid(guid: &Guid) -> Result<()> {
|
||||||
|
Self::set_cstr16("LoaderDevicePartUUID", &guid.to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system what boot entries are available.
|
||||||
|
pub fn set_entries<N: AsRef<str>>(entries: impl Iterator<Item = N>) -> Result<()> {
|
||||||
|
// Entries are stored as a null-terminated list of CString16 strings back to back.
|
||||||
|
// Iterate over the entries and convert them to CString16 placing them into data.
|
||||||
|
let mut data = Vec::new();
|
||||||
|
for entry in entries {
|
||||||
|
// Convert the entry to CString16 little endian.
|
||||||
|
let encoded = entry
|
||||||
|
.as_ref()
|
||||||
|
.encode_utf16()
|
||||||
|
.flat_map(|c| c.to_le_bytes())
|
||||||
|
.collect::<Vec<u8>>();
|
||||||
|
// Write the bytes (including the null terminator) into the data buffer.
|
||||||
|
data.extend_from_slice(&encoded);
|
||||||
|
}
|
||||||
|
Self::set("LoaderEntries", &data)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system what the default boot entry is.
|
||||||
|
pub fn set_default_entry(entry: String) -> Result<()> {
|
||||||
|
Self::set_cstr16("LoaderEntryDefault", &entry)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system what the selected boot entry is.
|
||||||
|
pub fn set_selected_entry(entry: String) -> Result<()> {
|
||||||
|
Self::set_cstr16("LoaderEntrySelected", &entry)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tell the system about the UEFI firmware we are running on.
|
||||||
|
pub fn set_firmware_info() -> Result<()> {
|
||||||
|
// Format the firmware information string into something human-readable.
|
||||||
|
let firmware_info = format!(
|
||||||
|
"{} {}.{:02}",
|
||||||
|
uefi::system::firmware_vendor(),
|
||||||
|
uefi::system::firmware_revision() >> 16,
|
||||||
|
uefi::system::firmware_revision() & 0xFFFFF,
|
||||||
|
);
|
||||||
|
Self::set_cstr16("LoaderFirmwareInfo", &firmware_info)?;
|
||||||
|
|
||||||
|
// Format the firmware revision into something human-readable.
|
||||||
|
let firmware_type = format!("UEFI {:02}", uefi::system::firmware_revision());
|
||||||
|
Self::set_cstr16("LoaderFirmwareType", &firmware_type)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// The [VariableAttributes] for bootloader interface variables.
|
||||||
|
fn attributes() -> VariableAttributes {
|
||||||
|
VariableAttributes::BOOTSERVICE_ACCESS | VariableAttributes::RUNTIME_ACCESS
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Set a bootloader interface variable specified by `key` to `value`.
|
||||||
|
fn set(key: &str, value: &[u8]) -> Result<()> {
|
||||||
|
let name =
|
||||||
|
CString16::try_from(key).context("unable to convert variable name to CString16")?;
|
||||||
|
uefi::runtime::set_variable(&name, &Self::VENDOR, Self::attributes(), value)
|
||||||
|
.with_context(|| format!("unable to set efi variable {}", key))?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Set a bootloader interface variable specified by `key` to `value`, converting the value to
|
||||||
|
/// a [CString16].
|
||||||
|
fn set_cstr16(key: &str, value: &str) -> Result<()> {
|
||||||
|
// Encode the value as a CString16 little endian.
|
||||||
|
let encoded = value
|
||||||
|
.encode_utf16()
|
||||||
|
.flat_map(|c| c.to_le_bytes())
|
||||||
|
.collect::<Vec<u8>>();
|
||||||
|
Self::set(key, &encoded)
|
||||||
|
}
|
||||||
|
}
|
||||||
79
src/main.rs
79
src/main.rs
@@ -1,12 +1,16 @@
|
|||||||
#![doc = include_str!("../README.md")]
|
#![doc = include_str!("../README.md")]
|
||||||
#![feature(uefi_std)]
|
#![feature(uefi_std)]
|
||||||
|
extern crate core;
|
||||||
|
|
||||||
use crate::config::RootConfiguration;
|
use crate::config::RootConfiguration;
|
||||||
use crate::context::{RootContext, SproutContext};
|
use crate::context::{RootContext, SproutContext};
|
||||||
use crate::entries::BootableEntry;
|
use crate::entries::BootableEntry;
|
||||||
|
use crate::integrations::bootloader_interface::BootloaderInterface;
|
||||||
use crate::options::SproutOptions;
|
use crate::options::SproutOptions;
|
||||||
use crate::options::parser::OptionsRepresentable;
|
use crate::options::parser::OptionsRepresentable;
|
||||||
use crate::phases::phase;
|
use crate::phases::phase;
|
||||||
|
use crate::platform::timer::PlatformTimer;
|
||||||
|
use crate::utils::PartitionGuidForm;
|
||||||
use anyhow::{Context, Result, bail};
|
use anyhow::{Context, Result, bail};
|
||||||
use log::{error, info};
|
use log::{error, info};
|
||||||
use std::collections::BTreeMap;
|
use std::collections::BTreeMap;
|
||||||
@@ -38,9 +42,15 @@ pub mod extractors;
|
|||||||
/// generators: Runtime code that can generate entries with specific values.
|
/// generators: Runtime code that can generate entries with specific values.
|
||||||
pub mod generators;
|
pub mod generators;
|
||||||
|
|
||||||
|
/// platform: Integration or support code for specific hardware platforms.
|
||||||
|
pub mod platform;
|
||||||
|
|
||||||
/// menu: Display a boot menu to select an entry to boot.
|
/// menu: Display a boot menu to select an entry to boot.
|
||||||
pub mod menu;
|
pub mod menu;
|
||||||
|
|
||||||
|
/// integrations: Code that interacts with other systems.
|
||||||
|
pub mod integrations;
|
||||||
|
|
||||||
/// phases: Hooks into specific parts of the boot process.
|
/// phases: Hooks into specific parts of the boot process.
|
||||||
pub mod phases;
|
pub mod phases;
|
||||||
|
|
||||||
@@ -55,6 +65,21 @@ pub mod utils;
|
|||||||
|
|
||||||
/// Run Sprout, returning an error if one occurs.
|
/// Run Sprout, returning an error if one occurs.
|
||||||
fn run() -> Result<()> {
|
fn run() -> Result<()> {
|
||||||
|
// Start the platform timer.
|
||||||
|
let timer = PlatformTimer::start();
|
||||||
|
|
||||||
|
// Mark the initialization of Sprout in the bootloader interface.
|
||||||
|
BootloaderInterface::mark_init(&timer)
|
||||||
|
.context("unable to mark initialization in bootloader interface")?;
|
||||||
|
|
||||||
|
// Tell the bootloader interface what firmware we are running on.
|
||||||
|
BootloaderInterface::set_firmware_info()
|
||||||
|
.context("unable to set firmware info in bootloader interface")?;
|
||||||
|
|
||||||
|
// Tell the bootloader interface what loader is being used.
|
||||||
|
BootloaderInterface::set_loader_info()
|
||||||
|
.context("unable to set loader info in bootloader interface")?;
|
||||||
|
|
||||||
// Parse the options to the sprout executable.
|
// Parse the options to the sprout executable.
|
||||||
let options = SproutOptions::parse().context("unable to parse options")?;
|
let options = SproutOptions::parse().context("unable to parse options")?;
|
||||||
|
|
||||||
@@ -69,17 +94,35 @@ fn run() -> Result<()> {
|
|||||||
config::loader::load(&options)?
|
config::loader::load(&options)?
|
||||||
};
|
};
|
||||||
|
|
||||||
// Load the root context.
|
// Grab the sprout.efi loaded image path.
|
||||||
// This is done in a block to ensure the release of the LoadedImageDevicePath protocol.
|
// This is done in a block to ensure the release of the LoadedImageDevicePath protocol.
|
||||||
let mut root = {
|
let loaded_image_path = {
|
||||||
let current_image_device_path_protocol = uefi::boot::open_protocol_exclusive::<
|
let current_image_device_path_protocol = uefi::boot::open_protocol_exclusive::<
|
||||||
LoadedImageDevicePath,
|
LoadedImageDevicePath,
|
||||||
>(uefi::boot::image_handle())
|
>(uefi::boot::image_handle())
|
||||||
.context("unable to get loaded image device path")?;
|
.context("unable to get loaded image device path")?;
|
||||||
let loaded_image_path = current_image_device_path_protocol.deref().to_boxed();
|
current_image_device_path_protocol.deref().to_boxed()
|
||||||
RootContext::new(loaded_image_path, options)
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Grab the partition GUID of the ESP that sprout was loaded from.
|
||||||
|
let loaded_image_partition_guid =
|
||||||
|
utils::partition_guid(&loaded_image_path, PartitionGuidForm::Partition)
|
||||||
|
.context("unable to retrieve loaded image partition guid")?;
|
||||||
|
|
||||||
|
// Set the partition GUID of the ESP that sprout was loaded from in the bootloader interface.
|
||||||
|
if let Some(loaded_image_partition_guid) = loaded_image_partition_guid {
|
||||||
|
// Tell the system about the partition GUID.
|
||||||
|
BootloaderInterface::set_partition_guid(&loaded_image_partition_guid)
|
||||||
|
.context("unable to set partition guid in bootloader interface")?;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tell the bootloader interface what the loaded image path is.
|
||||||
|
BootloaderInterface::set_loader_path(&loaded_image_path)
|
||||||
|
.context("unable to set loader path in bootloader interface")?;
|
||||||
|
|
||||||
|
// Create the root context.
|
||||||
|
let mut root = RootContext::new(loaded_image_path, timer, options);
|
||||||
|
|
||||||
// Insert the configuration actions into the root context.
|
// Insert the configuration actions into the root context.
|
||||||
root.actions_mut().extend(config.actions.clone());
|
root.actions_mut().extend(config.actions.clone());
|
||||||
|
|
||||||
@@ -159,13 +202,17 @@ fn run() -> Result<()> {
|
|||||||
for (name, generator) in config.generators {
|
for (name, generator) in config.generators {
|
||||||
let context = context.fork().freeze();
|
let context = context.fork().freeze();
|
||||||
|
|
||||||
// We will prefix all entries with [name]-.
|
// We will prefix all entries with [name]-, provided the name is not pinned.
|
||||||
let prefix = format!("{}-", name);
|
let prefix = format!("{}-", name);
|
||||||
|
|
||||||
// Add all the entries generated by the generator to the entry list.
|
// Add all the entries generated by the generator to the entry list.
|
||||||
// The generator specifies the context associated with the entry.
|
// The generator specifies the context associated with the entry.
|
||||||
for mut entry in generators::generate(context.clone(), &generator)? {
|
for mut entry in generators::generate(context.clone(), &generator)? {
|
||||||
|
// If the entry name is not pinned, prepend the name prefix.
|
||||||
|
if !entry.is_pin_name() {
|
||||||
entry.prepend_name_prefix(&prefix);
|
entry.prepend_name_prefix(&prefix);
|
||||||
|
}
|
||||||
|
|
||||||
entries.push(entry);
|
entries.push(entry);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -200,6 +247,21 @@ fn run() -> Result<()> {
|
|||||||
entry.mark_default();
|
entry.mark_default();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Iterate over all the entries and tell the bootloader interface what the entries are.
|
||||||
|
for entry in &entries {
|
||||||
|
// If the entry is the default entry, tell the bootloader interface it is the default.
|
||||||
|
if entry.is_default() {
|
||||||
|
// Tell the bootloader interface what the default entry is.
|
||||||
|
BootloaderInterface::set_default_entry(entry.name().to_string())
|
||||||
|
.context("unable to set default entry in bootloader interface")?;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tell the bootloader interface what entries are available.
|
||||||
|
BootloaderInterface::set_entries(entries.iter().map(|entry| entry.name()))
|
||||||
|
.context("unable to set entries in bootloader interface")?;
|
||||||
|
|
||||||
// Execute the late phase.
|
// Execute the late phase.
|
||||||
phase(context.clone(), &config.phases.late).context("unable to execute late phase")?;
|
phase(context.clone(), &config.phases.late).context("unable to execute late phase")?;
|
||||||
|
|
||||||
@@ -223,9 +285,14 @@ fn run() -> Result<()> {
|
|||||||
.context(format!("unable to find entry: {force_boot_entry}"))?
|
.context(format!("unable to find entry: {force_boot_entry}"))?
|
||||||
} else {
|
} else {
|
||||||
// Delegate to the menu to select an entry to boot.
|
// Delegate to the menu to select an entry to boot.
|
||||||
menu::select(menu_timeout, &entries).context("unable to select entry via boot menu")?
|
menu::select(&timer, menu_timeout, &entries)
|
||||||
|
.context("unable to select entry via boot menu")?
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Tell the bootloader interface what the selected entry is.
|
||||||
|
BootloaderInterface::set_selected_entry(entry.name().to_string())
|
||||||
|
.context("unable to set selected entry in bootloader interface")?;
|
||||||
|
|
||||||
// Execute all the actions for the selected entry.
|
// Execute all the actions for the selected entry.
|
||||||
for action in &entry.declaration().actions {
|
for action in &entry.declaration().actions {
|
||||||
let action = entry.context().stamp(action);
|
let action = entry.context().stamp(action);
|
||||||
|
|||||||
12
src/menu.rs
12
src/menu.rs
@@ -1,4 +1,6 @@
|
|||||||
use crate::entries::BootableEntry;
|
use crate::entries::BootableEntry;
|
||||||
|
use crate::integrations::bootloader_interface::BootloaderInterface;
|
||||||
|
use crate::platform::timer::PlatformTimer;
|
||||||
use anyhow::{Context, Result, bail};
|
use anyhow::{Context, Result, bail};
|
||||||
use log::info;
|
use log::info;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
@@ -162,7 +164,15 @@ fn select_with_input<'a>(
|
|||||||
/// Shows a boot menu to select a bootable entry to boot.
|
/// Shows a boot menu to select a bootable entry to boot.
|
||||||
/// The actual work is done internally in [select_with_input] which is called
|
/// The actual work is done internally in [select_with_input] which is called
|
||||||
/// within the context of the standard input device.
|
/// within the context of the standard input device.
|
||||||
pub fn select(timeout: Duration, entries: &[BootableEntry]) -> Result<&BootableEntry> {
|
pub fn select<'live>(
|
||||||
|
timer: &'live PlatformTimer,
|
||||||
|
timeout: Duration,
|
||||||
|
entries: &'live [BootableEntry],
|
||||||
|
) -> Result<&'live BootableEntry> {
|
||||||
|
// Notify the bootloader interface that we are about to display the menu.
|
||||||
|
BootloaderInterface::mark_menu(timer)
|
||||||
|
.context("unable to mark menu display in bootloader interface")?;
|
||||||
|
|
||||||
// Acquire the standard input device and run the boot menu.
|
// Acquire the standard input device and run the boot menu.
|
||||||
uefi::system::with_stdin(move |input| select_with_input(input, timeout, entries))
|
uefi::system::with_stdin(move |input| select_with_input(input, timeout, entries))
|
||||||
}
|
}
|
||||||
|
|||||||
2
src/platform.rs
Normal file
2
src/platform.rs
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
/// timer: Platform timer support.
|
||||||
|
pub mod timer;
|
||||||
89
src/platform/timer.rs
Normal file
89
src/platform/timer.rs
Normal file
@@ -0,0 +1,89 @@
|
|||||||
|
// Referenced https://github.com/sheroz/tick_counter (MIT license) as a baseline.
|
||||||
|
// Architecturally modified to support UEFI and remove x86 (32-bit) support.
|
||||||
|
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
|
/// Support for aarch64 timers.
|
||||||
|
#[cfg(target_arch = "aarch64")]
|
||||||
|
pub mod aarch64;
|
||||||
|
|
||||||
|
/// Support for x86_64 timers.
|
||||||
|
#[cfg(target_arch = "x86_64")]
|
||||||
|
pub mod x86_64;
|
||||||
|
|
||||||
|
/// The tick frequency of the platform.
|
||||||
|
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
|
||||||
|
pub enum TickFrequency {
|
||||||
|
/// The platform provides the tick frequency.
|
||||||
|
Hardware(u64),
|
||||||
|
/// The tick frequency is measured internally.
|
||||||
|
Measured(u64, Duration),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TickFrequency {
|
||||||
|
/// Acquire the tick frequency reported by the platform.
|
||||||
|
fn ticks(&self) -> u64 {
|
||||||
|
match self {
|
||||||
|
TickFrequency::Hardware(frequency) => *frequency,
|
||||||
|
TickFrequency::Measured(frequency, _) => *frequency,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Calculate the nanoseconds represented by a tick.
|
||||||
|
fn nanos(&self) -> f64 {
|
||||||
|
1.0e9_f64 / (self.ticks() as f64)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Produce a duration from the provided elapsed `ticks` value.
|
||||||
|
fn duration(&self, ticks: u64) -> Duration {
|
||||||
|
let accuracy = self.nanos();
|
||||||
|
let nanos = ticks as f64 * accuracy;
|
||||||
|
Duration::from_nanos(nanos as u64)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Acquire the tick value reported by the platform.
|
||||||
|
fn arch_ticks() -> u64 {
|
||||||
|
#[cfg(target_arch = "aarch64")]
|
||||||
|
return aarch64::ticks();
|
||||||
|
#[cfg(target_arch = "x86_64")]
|
||||||
|
return x86_64::ticks();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Acquire the tick frequency reported by the platform.
|
||||||
|
fn arch_frequency() -> TickFrequency {
|
||||||
|
#[cfg(target_arch = "aarch64")]
|
||||||
|
return aarch64::frequency();
|
||||||
|
#[cfg(target_arch = "x86_64")]
|
||||||
|
return x86_64::frequency();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Platform timer that allows measurement of the elapsed time.
|
||||||
|
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
|
||||||
|
pub struct PlatformTimer {
|
||||||
|
/// The start tick value.
|
||||||
|
start: u64,
|
||||||
|
/// The tick frequency of the platform.
|
||||||
|
frequency: TickFrequency,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PlatformTimer {
|
||||||
|
/// Start a platform timer at the current instant.
|
||||||
|
pub fn start() -> Self {
|
||||||
|
Self {
|
||||||
|
start: arch_ticks(),
|
||||||
|
frequency: arch_frequency(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Measure the elapsed duration since the hardware started ticking upwards.
|
||||||
|
pub fn elapsed_since_lifetime(&self) -> Duration {
|
||||||
|
self.frequency.duration(arch_ticks())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Measure the elapsed duration since the timer was started.
|
||||||
|
pub fn elapsed_since_start(&self) -> Duration {
|
||||||
|
let duration = arch_ticks() - self.start;
|
||||||
|
self.frequency.duration(duration)
|
||||||
|
}
|
||||||
|
}
|
||||||
33
src/platform/timer/aarch64.rs
Normal file
33
src/platform/timer/aarch64.rs
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
use crate::platform::timer::TickFrequency;
|
||||||
|
use std::arch::asm;
|
||||||
|
|
||||||
|
/// Reads the cntvct_el0 counter and returns the value.
|
||||||
|
pub fn ticks() -> u64 {
|
||||||
|
let counter: u64;
|
||||||
|
unsafe {
|
||||||
|
asm!("mrs x0, cntvct_el0", out("x0") counter);
|
||||||
|
}
|
||||||
|
counter
|
||||||
|
}
|
||||||
|
|
||||||
|
/// We can use the actual ticks value as our start value.
|
||||||
|
pub fn start() -> u64 {
|
||||||
|
ticks()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// We can use the actual ticks value as our stop value.
|
||||||
|
pub fn stop() -> u64 {
|
||||||
|
ticks()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Our frequency is provided by cntfrq_el0 on the platform.
|
||||||
|
pub fn frequency() -> TickFrequency {
|
||||||
|
let frequency: u64;
|
||||||
|
unsafe {
|
||||||
|
asm!(
|
||||||
|
"mrs x0, cntfrq_el0",
|
||||||
|
out("x0") frequency
|
||||||
|
);
|
||||||
|
}
|
||||||
|
TickFrequency::Hardware(frequency)
|
||||||
|
}
|
||||||
66
src/platform/timer/x86_64.rs
Normal file
66
src/platform/timer/x86_64.rs
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
use crate::platform::timer::TickFrequency;
|
||||||
|
use core::arch::asm;
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
|
/// We will measure the frequency of the timer based on 1000 microseconds.
|
||||||
|
/// This will result in a call to BS->Stall(1000) in the end.
|
||||||
|
const MEASURE_FREQUENCY_DURATION: Duration = Duration::from_micros(1000);
|
||||||
|
|
||||||
|
/// Read the number of ticks from the platform timer.
|
||||||
|
pub fn ticks() -> u64 {
|
||||||
|
let mut eax: u32;
|
||||||
|
let mut edx: u32;
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
asm!("rdtsc", out("eax") eax, out("edx") edx);
|
||||||
|
}
|
||||||
|
|
||||||
|
(edx as u64) << 32 | eax as u64
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Read the starting number of ticks from the platform timer.
|
||||||
|
pub fn start() -> u64 {
|
||||||
|
let rax: u64;
|
||||||
|
unsafe {
|
||||||
|
asm!(
|
||||||
|
"mfence",
|
||||||
|
"lfence",
|
||||||
|
"rdtsc",
|
||||||
|
"shl rdx, 32",
|
||||||
|
"or rax, rdx",
|
||||||
|
out("rax") rax
|
||||||
|
);
|
||||||
|
}
|
||||||
|
rax
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Read the ending number of ticks from the platform timer.
|
||||||
|
pub fn stop() -> u64 {
|
||||||
|
let rax: u64;
|
||||||
|
unsafe {
|
||||||
|
asm!(
|
||||||
|
"rdtsc",
|
||||||
|
"lfence",
|
||||||
|
"shl rdx, 32",
|
||||||
|
"or rax, rdx",
|
||||||
|
out("rax") rax
|
||||||
|
);
|
||||||
|
}
|
||||||
|
rax
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Measure the frequency of the platform timer.
|
||||||
|
fn measure_frequency(duration: &Duration) -> u64 {
|
||||||
|
let start = start();
|
||||||
|
uefi::boot::stall(*duration);
|
||||||
|
let stop = stop();
|
||||||
|
let elapsed = (stop - start) as f64;
|
||||||
|
(elapsed / duration.as_secs_f64()) as u64
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Acquire the platform timer frequency.
|
||||||
|
/// On x86_64, this is slightly expensive, so it should be done once.
|
||||||
|
pub fn frequency() -> TickFrequency {
|
||||||
|
let frequency = measure_frequency(&MEASURE_FREQUENCY_DURATION);
|
||||||
|
TickFrequency::Measured(frequency, MEASURE_FREQUENCY_DURATION)
|
||||||
|
}
|
||||||
50
src/utils.rs
50
src/utils.rs
@@ -4,7 +4,9 @@ use uefi::fs::{FileSystem, Path};
|
|||||||
use uefi::proto::device_path::text::{AllowShortcuts, DevicePathFromText, DisplayOnly};
|
use uefi::proto::device_path::text::{AllowShortcuts, DevicePathFromText, DisplayOnly};
|
||||||
use uefi::proto::device_path::{DevicePath, PoolDevicePath};
|
use uefi::proto::device_path::{DevicePath, PoolDevicePath};
|
||||||
use uefi::proto::media::fs::SimpleFileSystem;
|
use uefi::proto::media::fs::SimpleFileSystem;
|
||||||
use uefi::{CString16, Handle};
|
use uefi::proto::media::partition::PartitionInfo;
|
||||||
|
use uefi::{CString16, Guid, Handle};
|
||||||
|
use uefi_raw::Status;
|
||||||
|
|
||||||
/// Support code for the EFI framebuffer.
|
/// Support code for the EFI framebuffer.
|
||||||
pub mod framebuffer;
|
pub mod framebuffer;
|
||||||
@@ -181,3 +183,49 @@ pub fn combine_options<T: AsRef<str>>(options: impl Iterator<Item = T>) -> Strin
|
|||||||
pub fn unique_hash(input: &str) -> String {
|
pub fn unique_hash(input: &str) -> String {
|
||||||
sha256::digest(input.as_bytes())
|
sha256::digest(input.as_bytes())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Represents the type of partition GUID that can be retrieved.
|
||||||
|
#[derive(PartialEq, Eq)]
|
||||||
|
pub enum PartitionGuidForm {
|
||||||
|
Partition,
|
||||||
|
PartitionType,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Retrieve the partition / partition type GUID of the device root `path`.
|
||||||
|
/// This only works on GPT partitions. If the root is not a GPT partition, None is returned.
|
||||||
|
pub fn partition_guid(path: &DevicePath, form: PartitionGuidForm) -> Result<Option<Guid>> {
|
||||||
|
// Clone the path so we can pass it to the UEFI stack.
|
||||||
|
let path = path.to_boxed();
|
||||||
|
let result = uefi::boot::locate_device_path::<PartitionInfo>(&mut &*path);
|
||||||
|
let handle = match result {
|
||||||
|
Ok(handle) => Ok(Some(handle)),
|
||||||
|
Err(error) => {
|
||||||
|
// If the error is NOT_FOUND or UNSUPPORTED, we can return None.
|
||||||
|
// These are non-fatal errors.
|
||||||
|
if error.status() == Status::NOT_FOUND || error.status() == Status::UNSUPPORTED {
|
||||||
|
Ok(None)
|
||||||
|
} else {
|
||||||
|
Err(error)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.context("unable to locate device path")?;
|
||||||
|
|
||||||
|
// If we have the handle, we can try to open the partition info protocol.
|
||||||
|
if let Some(handle) = handle {
|
||||||
|
// Open the partition info protocol.
|
||||||
|
let partition_info = uefi::boot::open_protocol_exclusive::<PartitionInfo>(handle)
|
||||||
|
.context("unable to open partition info protocol")?;
|
||||||
|
// Find the unique partition GUID.
|
||||||
|
// If this is not a GPT partition, this will produce None.
|
||||||
|
Ok(partition_info
|
||||||
|
.gpt_partition_entry()
|
||||||
|
.map(|entry| match form {
|
||||||
|
// Match the form of the partition GUID.
|
||||||
|
PartitionGuidForm::Partition => entry.unique_partition_guid,
|
||||||
|
PartitionGuidForm::PartitionType => entry.partition_type_guid.0,
|
||||||
|
}))
|
||||||
|
} else {
|
||||||
|
Ok(None)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user