drop host (wip)

Author: Sangho Lee

litebox_platform/src/host/hv_impl.rs

@@ -0,0 +1,59 @@
+//! An implementation of [`HostInterface`] for (Virtual) Machine
+
+use crate::host::per_cpu_variables::with_per_cpu_variables_mut;
+
+pub type Machine = crate::LinuxKernel;
+
+#[cfg(not(test))]
+mod alloc {
+    const HEAP_ORDER: usize = 25;
+
+    #[global_allocator]
+    static MACHINE_ALLOCATOR: litebox::mm::allocator::SafeZoneAllocator<
+        'static,
+        HEAP_ORDER,
+        crate::host::Machine,
+    > = litebox::mm::allocator::SafeZoneAllocator::new();
+
+    impl litebox::mm::allocator::MemoryProvider for super::Machine {
+        fn alloc(_layout: &core::alloc::Layout) -> Option<(usize, usize)> {
+            // dynamic memory allocation here implies memory ballooning or hotplugging
+            unimplemented!()
+        }
+
+        unsafe fn free(_addr: usize) {
+            unimplemented!()
+        }
+    }
+
+    impl crate::mm::MemoryProvider for super::Machine {
+        const GVA_OFFSET: x86_64::VirtAddr = x86_64::VirtAddr::new(0x18000000000);
+        const PRIVATE_PTE_MASK: u64 = 0;
+
+        fn mem_allocate_pages(order: u32) -> Option<*mut u8> {
+            MACHINE_ALLOCATOR.allocate_pages(order)
+        }
+
+        unsafe fn mem_free_pages(ptr: *mut u8, order: u32) {
+            unsafe {
+                MACHINE_ALLOCATOR.free_pages(ptr, order);
+            }
+        }
+
+        unsafe fn mem_fill_pages(start: usize, size: usize) {
+            unsafe { MACHINE_ALLOCATOR.fill_pages(start, size) };
+        }
+    }
+}
+
+unsafe impl litebox::platform::ThreadLocalStorageProvider for Machine {
+    fn get_thread_local_storage() -> *mut () {
+        let tls = with_per_cpu_variables_mut(|pcv| pcv.tls);
+        tls.as_mut_ptr::<()>()
+    }
+
+    unsafe fn replace_thread_local_storage(value: *mut ()) -> *mut () {
+        let tls = with_per_cpu_variables_mut(|pcv| pcv.tls);
+        core::mem::replace(&mut tls.as_mut_ptr::<()>(), value.cast()).cast()
+    }
+}

litebox_platform/src/host/machine_impl.rs

@@ -1,9 +1,9 @@
 //! Different host implementations of [`super::HostInterface`]
-pub mod hv_impl;
 pub mod linux;
+pub mod machine_impl;
 pub mod per_cpu_variables;
 
-pub use hv_impl::Hypervisor;
+pub use machine_impl::Machine;
 
 #[cfg(test)]
 pub mod mock;