feat: UEFI MVP

Co-authored-by: Sarah Simon <sarah.simon@rwth-aachen.de>
Signed-off-by: Martin Kröning <martin.kroening@eonerc.rwth-aachen.de>

Cargo.toml

@@ -36,7 +36,6 @@ goblin = { version = "0.8", default-features = false, features = ["elf64"] }
 
 [target.'cfg(target_os = "uefi")'.dependencies]
 uefi = { version = "0.32", features = ["alloc", "panic_handler", "qemu"] }
-qemu-exit = "3"
 
 [target.'cfg(target_arch = "riscv64")'.dependencies]
 fdt = "0.1"

src/arch/x86_64/mod.rs

@@ -18,8 +18,7 @@ pub use console::Console;
 
 #[cfg(target_os = "none")]
 const KERNEL_STACK_SIZE: u64 = 32_768;
-#[cfg(target_os = "none")]
-const SERIAL_IO_PORT: u16 = 0x3F8;
+pub const SERIAL_IO_PORT: u16 = 0x3F8;
 
 #[cfg(target_os = "none")]
 unsafe fn map_memory(address: usize, memory_size: usize) -> usize {
@@ -45,8 +44,7 @@ pub unsafe fn get_memory(memory_size: u64) -> u64 {
 	unsafe { map_memory(address, memory_size as usize) as u64 }
 }
 
-#[cfg(target_os = "none")]
-unsafe fn enter_kernel(
+pub unsafe fn enter_kernel(
 	stack: *mut u8,
 	entry: *const (),
 	raw_boot_info: &'static hermit_entry::boot_info::RawBootInfo,

src/main.rs

@@ -5,9 +5,7 @@
 #![allow(unstable_name_collisions)]
 #![allow(clippy::missing_safety_doc)]
 
-#[cfg(not(target_os = "uefi"))]
 use ::log::info;
-#[cfg(not(target_os = "uefi"))]
 use hermit_entry::boot_info::{BootInfo, RawBootInfo};
 
 #[macro_use]
@@ -24,12 +22,10 @@ mod os;
 ))]
 extern crate alloc;
 
-#[cfg(not(target_os = "uefi"))]
 trait BootInfoExt {
 	fn write(self) -> &'static RawBootInfo;
 }
 
-#[cfg(not(target_os = "uefi"))]
 impl BootInfoExt for BootInfo {
 	fn write(self) -> &'static RawBootInfo {
 		info!("boot_info = {self:#x?}");

src/os/uefi/fdt.rs

@@ -0,0 +1,118 @@
+use alloc::format;
+use alloc::vec::Vec;
+use core::fmt::{self, Write};
+
+use log::info;
+use uefi::mem::memory_map::{MemoryMap, MemoryMapMut};
+use uefi::table::boot::{MemoryDescriptor, MemoryType, PAGE_SIZE};
+use vm_fdt::{FdtWriter, FdtWriterNode, FdtWriterResult};
+
+pub struct Fdt {
+	writer: FdtWriter,
+	root_node: FdtWriterNode,
+}
+
+impl Fdt {
+	pub fn new() -> FdtWriterResult<Self> {
+		let mut writer = FdtWriter::new()?;
+
+		let root_node = writer.begin_node("")?;
+		writer.property_string("compatible", "hermit,uefi")?;
+		writer.property_u32("#address-cells", 0x2)?;
+		writer.property_u32("#size-cells", 0x2)?;
+
+		Ok(Self { writer, root_node })
+	}
+
+	pub fn finish(mut self) -> FdtWriterResult<Vec<u8>> {
+		self.writer.end_node(self.root_node)?;
+
+		self.writer.finish()
+	}
+
+	pub fn rsdp(mut self, rsdp: u64) -> FdtWriterResult<Self> {
+		let rsdp_node = self.writer.begin_node(&format!("hermit,rsdp@{rsdp:x}"))?;
+		self.writer.property_array_u64("reg", &[rsdp, 1])?;
+		self.writer.end_node(rsdp_node)?;
+
+		Ok(self)
+	}
+
+	pub fn memory_map(mut self, memory_map: &mut impl MemoryMapMut) -> FdtWriterResult<Self> {
+		memory_map.sort();
+		info!("Memory map:\n{}", memory_map.display());
+
+		let entries = memory_map
+			.entries()
+			.filter(|entry| entry.ty == MemoryType::CONVENTIONAL);
+
+		for entry in entries {
+			let memory_node = self
+				.writer
+				.begin_node(format!("memory@{:x}", entry.phys_start).as_str())?;
+			self.writer.property_string("device_type", "memory")?;
+			self.writer.property_array_u64(
+				"reg",
+				&[entry.phys_start, entry.page_count * PAGE_SIZE as u64],
+			)?;
+			self.writer.end_node(memory_node)?;
+		}
+
+		Ok(self)
+	}
+}
+
+trait MemoryMapExt: MemoryMap {
+	fn display(&self) -> MemoryMapDisplay<'_, Self> {
+		MemoryMapDisplay { inner: self }
+	}
+}
+
+impl<T> MemoryMapExt for T where T: MemoryMap {}
+
+struct MemoryMapDisplay<'a, T: ?Sized> {
+	inner: &'a T,
+}
+
+impl<'a, T> fmt::Display for MemoryMapDisplay<'a, T>
+where
+	T: MemoryMap,
+{
+	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+		let mut has_fields = false;
+
+		for desc in self.inner.entries() {
+			if has_fields {
+				f.write_char('\n')?;
+			}
+			write!(f, "{}", desc.display())?;
+
+			has_fields = true;
+		}
+		Ok(())
+	}
+}
+
+trait MemoryDescriptorExt {
+	fn display(&self) -> MemoryDescriptorDisplay<'_>;
+}
+
+impl MemoryDescriptorExt for MemoryDescriptor {
+	fn display(&self) -> MemoryDescriptorDisplay<'_> {
+		MemoryDescriptorDisplay { inner: self }
+	}
+}
+
+struct MemoryDescriptorDisplay<'a> {
+	inner: &'a MemoryDescriptor,
+}
+
+impl<'a> fmt::Display for MemoryDescriptorDisplay<'a> {
+	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+		write!(
+			f,
+			"start: {:#12x}, pages: {:#8x}, type: {:?}",
+			self.inner.phys_start, self.inner.page_count, self.inner.ty
+		)
+	}
+}

src/os/uefi/mod.rs

@@ -1,37 +1,57 @@
 mod allocator;
 mod console;
+mod fdt;
 
-use alloc::string::String;
 use alloc::vec::Vec;
+use core::ffi::c_void;
+use core::mem::MaybeUninit;
+use core::slice;
 
+use align_address::Align;
+use hermit_entry::boot_info::{
+	BootInfo, DeviceTreeAddress, HardwareInfo, PlatformInfo, SerialPortBase,
+};
+use hermit_entry::elf::{KernelObject, LoadedKernel};
 use log::info;
-use qemu_exit::QEMUExit;
+use sptr::Strict;
 use uefi::fs::{FileSystem, Path};
 use uefi::prelude::*;
-use uefi::table::boot::MemoryType;
+use uefi::table::boot::{AllocateType, MemoryType, PAGE_SIZE};
+use uefi::table::cfg;
 
 pub use self::console::CONSOLE;
+use self::fdt::Fdt;
+use crate::{arch, BootInfoExt};
 
 // Entry Point of the Uefi Loader
 #[entry]
 fn main() -> Status {
 	uefi::helpers::init().unwrap();
 	crate::log::init();
 
-	let app = read_app();
+	let kernel_image = read_app();
+	let kernel = KernelObject::parse(&kernel_image).unwrap();
 
-	let string = String::from_utf8(app).unwrap();
-	println!("{string}");
+	let kernel_memory = alloc_page_slice(kernel.mem_size()).unwrap();
+	let kernel_memory = &mut kernel_memory[..kernel.mem_size()];
+
+	let kernel_info = kernel.load_kernel(kernel_memory, kernel_memory.as_ptr() as u64);
+
+	let rsdp = rsdp();
+
+	drop(kernel_image);
+
+	let fdt = Fdt::new()
+		.unwrap()
+		.rsdp(u64::try_from(rsdp.expose_addr()).unwrap())
+		.unwrap();
 
 	allocator::exit_boot_services();
-	let _memory_map = unsafe { boot::exit_boot_services(MemoryType::LOADER_DATA) };
+	let mut memory_map = unsafe { boot::exit_boot_services(MemoryType::LOADER_DATA) };
 
-	println!("Exited boot services!");
-	println!("Allocations still {}!", String::from("work"));
+	let fdt = fdt.memory_map(&mut memory_map).unwrap().finish().unwrap();
 
-	let custom_exit_success = 3;
-	let qemu_exit_handle = qemu_exit::X86::new(0xf4, custom_exit_success);
-	qemu_exit_handle.exit_success()
+	unsafe { boot_kernel(kernel_info, fdt) }
 }
 
 fn read_app() -> Vec<u8> {
@@ -49,3 +69,64 @@ fn read_app() -> Vec<u8> {
 
 	data
 }
+
+pub unsafe fn boot_kernel(kernel_info: LoadedKernel, fdt: Vec<u8>) -> ! {
+	let LoadedKernel {
+		load_info,
+		entry_point,
+	} = kernel_info;
+
+	let device_tree = DeviceTreeAddress::new(u64::try_from(fdt.leak().as_ptr().addr()).unwrap());
+
+	let boot_info = BootInfo {
+		hardware_info: HardwareInfo {
+			phys_addr_range: 0..0,
+			serial_port_base: SerialPortBase::new(arch::SERIAL_IO_PORT),
+			device_tree,
+		},
+		load_info,
+		platform_info: PlatformInfo::Fdt,
+	};
+
+	let stack = usize::try_from(boot_info.load_info.kernel_image_addr_range.end)
+		.unwrap()
+		.align_down(PAGE_SIZE);
+	let entry = sptr::from_exposed_addr(entry_point.try_into().unwrap());
+	let stack = sptr::from_exposed_addr_mut(stack);
+	let raw_boot_info = boot_info.write();
+
+	unsafe { arch::enter_kernel(stack, entry, raw_boot_info) }
+}
+
+fn alloc_page_slice(size: usize) -> uefi::Result<&'static mut [MaybeUninit<u8>]> {
+	let size = size.align_up(PAGE_SIZE);
+	let ptr = boot::allocate_pages(
+		AllocateType::AnyPages,
+		MemoryType::LOADER_DATA,
+		size / PAGE_SIZE,
+	)?;
+	Ok(unsafe { slice::from_raw_parts_mut(ptr.cast().as_ptr(), size) })
+}
+
+/// Returns the RSDP.
+///
+/// This must be called before exiting boot services.
+/// See [5.2.5.2. Finding the RSDP on UEFI Enabled Systems — ACPI Specification 6.5 documentation](https://uefi.org/specs/ACPI/6.5/05_ACPI_Software_Programming_Model.html#finding-the-rsdp-on-uefi-enabled-systems) for details.
+fn rsdp() -> *const c_void {
+	system::with_config_table(|config_table| {
+		let (rsdp, version) = if let Some(entry) = config_table
+			.iter()
+			.find(|entry| entry.guid == cfg::ACPI2_GUID)
+		{
+			(entry.address, 2)
+		} else {
+			let entry = config_table
+				.iter()
+				.find(|entry| entry.guid == cfg::ACPI_GUID)
+				.unwrap();
+			(entry.address, 1)
+		};
+		info!("Found ACPI {version} RSDP at {rsdp:p}");
+		rsdp
+	})
+}

xtask/src/ci/qemu.rs

@@ -44,7 +44,10 @@ impl Qemu {
 		if self.build.target() == Target::X86_64Uefi {
 			sh.create_dir("target/esp/efi/boot")?;
 			sh.copy_file(self.build.dist_object(), "target/esp/efi/boot/bootx64.efi")?;
-			sh.write_file("target/esp/efi/boot/hermit-app", "Hello, UEFI!\n")?;
+			sh.copy_file(
+				self.build.ci_image(&self.image),
+				"target/esp/efi/boot/hermit-app",
+			)?;
 		}
 
 		let target = self.build.target();
@@ -198,13 +201,10 @@ impl Qemu {
 	}
 
 	fn memory(&self) -> usize {
-		let mut memory = 32usize;
-		if self.image == "hello_c" {
-			memory = memory.max(64);
-		}
+		let mut memory = 64usize;
 		match self.build.target() {
 			Target::X86_64Uefi => {
-				memory = memory.max(64);
+				memory = memory.max(512);
 			}
 			Target::Aarch64 => {
 				memory = memory.max(256);