fix(mm): make virtualmem and physicalmem functions unsafe as appropriate

Author: mkroening

src/mm/physicalmem.rs

@@ -22,7 +22,9 @@ pub struct FrameAlloc;
 
 impl PageRangeAllocator for FrameAlloc {
 	unsafe fn init() {
-		init();
+		unsafe {
+			init();
+		}
 	}
 
 	fn allocate(layout: PageLayout) -> Result<PageRange, AllocError> {
@@ -99,7 +101,7 @@ pub unsafe fn map_frame_range(frame_range: PageRange) {
 	}
 }
 
-fn detect_from_fdt() -> Result<(), ()> {
+unsafe fn detect_from_fdt() -> Result<(), ()> {
 	let fdt = env::fdt().ok_or(())?;
 
 	let all_regions = fdt
@@ -198,7 +200,7 @@ impl PageRangeExt for PageRange {
 }
 
 #[cfg(any(target_arch = "aarch64", target_arch = "riscv64"))]
-fn detect_from_limits() -> Result<(), ()> {
+unsafe fn detect_from_limits() -> Result<(), ()> {
 	let limit = crate::arch::kernel::get_limit();
 	if limit == 0 {
 		return Err(());
@@ -220,19 +222,19 @@ fn detect_from_limits() -> Result<(), ()> {
 	Ok(())
 }
 
-fn init() {
+unsafe fn init() {
 	if env::is_uefi() && DeviceAlloc.phys_offset() != VirtAddr::zero() {
 		let start = DeviceAlloc.phys_offset();
 		let count = DeviceAlloc.phys_offset().as_u64() / HugePageSize::SIZE;
 		let count = usize::try_from(count).unwrap();
 		paging::unmap::<HugePageSize>(start, count);
 	}
 
-	if let Err(_err) = detect_from_fdt() {
+	if let Err(_err) = unsafe { detect_from_fdt() } {
 		cfg_if::cfg_if! {
 			if #[cfg(any(target_arch = "aarch64", target_arch = "riscv64"))] {
 				error!("Could not detect physical memory from FDT");
-				detect_from_limits().unwrap();
+				unsafe { detect_from_limits().unwrap(); }
 			} else {
 				panic!("Could not detect physical memory from FDT");
 			}

src/mm/virtualmem.rs

@@ -14,7 +14,9 @@ pub struct PageAlloc;
 
 impl PageRangeAllocator for PageAlloc {
 	unsafe fn init() {
-		init();
+		unsafe {
+			init();
+		}
 	}
 
 	fn allocate(layout: PageLayout) -> Result<PageRange, AllocError> {
@@ -47,7 +49,7 @@ impl fmt::Display for PageAlloc {
 
 pub type PageBox = PageRangeBox<PageAlloc>;
 
-fn init() {
+unsafe fn init() {
 	let range = PageRange::new(
 		kernel_heap_end().as_usize().div_ceil(2),
 		kernel_heap_end().as_usize() + 1,