feat: implement decompression of Hermit images

Author: fogti

Cargo.lock

@@ -11,7 +11,8 @@ align-address = "0.3"
 allocator-api2 = { version = "0.3", default-features = false }
 anstyle = { version = "1", default-features = false }
 cfg-if = "1"
-hermit-entry = { version = "0.10", features = ["loader"] }
+hermit-entry = { version = "0.10", git = "https://github.com/fogti/hermit-entry.git", branch = "image-reader", features = ["allocator-api2", "loader"] }
+lazy_static = { version = "1.5.0", features = ["spin_no_std"] }
 log = "0.4"
 one-shot-mutex = "0.2"
 sptr = "0.3"

Cargo.toml

@@ -20,7 +20,7 @@ use sptr::Strict;
 
 use crate::BootInfoExt;
 use crate::arch::paging::*;
-use crate::os::CONSOLE;
+use crate::os::{CONSOLE, ExtraBootInfo};
 
 unsafe extern "C" {
 	static mut loader_end: u8;
@@ -110,12 +110,14 @@ pub fn find_kernel() -> &'static [u8] {
 	}
 }
 
-pub unsafe fn boot_kernel(kernel_info: LoadedKernel) -> ! {
+pub unsafe fn boot_kernel(kernel_info: LoadedKernel, extra_info: ExtraBootInfo) -> ! {
 	let LoadedKernel {
 		load_info,
 		entry_point,
 	} = kernel_info;
 
+	assert!(extra_info.image.is_none());
+
 	let dtb = unsafe {
 		Fdt::from_ptr(sptr::from_exposed_addr(get_dtb_addr() as usize))
 			.expect(".dtb file has invalid header")

src/arch/aarch64/mod.rs

@@ -5,7 +5,7 @@ cfg_if::cfg_if! {
 	} else if #[cfg(target_arch = "riscv64")] {
 		mod riscv64;
 		pub use self::riscv64::*;
-	} else if #[cfg(all(target_arch = "x86_64"))] {
+	} else if #[cfg(target_arch = "x86_64")] {
 		mod x86_64;
 		pub use self::x86_64::*;
 	}

src/arch/mod.rs

@@ -17,6 +17,7 @@ use log::info;
 use sptr::Strict;
 
 use crate::BootInfoExt;
+use crate::os::ExtraBootInfo;
 
 fn find_kernel_linux(chosen: &FdtNode<'_, '_>) -> Option<&'static [u8]> {
 	let initrd_start = chosen.property("linux,initrd-start")?.as_usize()?;
@@ -90,7 +91,7 @@ pub unsafe fn get_memory(memory_size: u64) -> u64 {
 	u64::try_from(start_address).unwrap()
 }
 
-pub unsafe fn boot_kernel(kernel_info: LoadedKernel) -> ! {
+pub unsafe fn boot_kernel(kernel_info: LoadedKernel, extra_info: ExtraBootInfo) -> ! {
 	let LoadedKernel {
 		load_info,
 		entry_point,
@@ -118,6 +119,8 @@ pub unsafe fn boot_kernel(kernel_info: LoadedKernel) -> ! {
 		DeviceTreeAddress::new(fdt_addr.try_into().unwrap())
 	};
 
+	assert!(extra_info.image.is_none());
+
 	let boot_info = BootInfo {
 		hardware_info: HardwareInfo {
 			phys_addr_range,

src/arch/riscv64/mod.rs

@@ -20,6 +20,7 @@ use super::physicalmem::PhysAlloc;
 use super::{KERNEL_STACK_SIZE, SERIAL_IO_PORT, paging};
 use crate::BootInfoExt;
 use crate::fdt::Fdt;
+use crate::os::ExtraBootInfo;
 
 unsafe extern "C" {
 	static mut loader_end: u8;
@@ -118,7 +119,7 @@ pub fn find_kernel() -> &'static [u8] {
 	unsafe { slice::from_raw_parts(sptr::from_exposed_addr(elf_start), elf_len) }
 }
 
-pub unsafe fn boot_kernel(kernel_info: LoadedKernel) -> ! {
+pub unsafe fn boot_kernel(kernel_info: LoadedKernel, extra_info: ExtraBootInfo) -> ! {
 	let LoadedKernel {
 		load_info,
 		entry_point,
@@ -172,7 +173,12 @@ pub unsafe fn boot_kernel(kernel_info: LoadedKernel) -> ! {
 		write_bytes(stack, 0, KERNEL_STACK_SIZE.try_into().unwrap());
 	}
 
-	let mut fdt = Fdt::new("firecracker").unwrap();
+	let maybe_image = match extra_info.image {
+		None => &[],
+		Some(tar_image) => tar_image,
+	};
+
+	let mut fdt = Fdt::new("firecracker", maybe_image).unwrap();
 
 	// Load the boot_param memory-map information
 	let linux_e820_entries =