td-payload: check the size for init_shared_memory

The size needs to be aligned with 4KiB, panic if not. Signed-off-by: Jiaqi Gao <[email protected]>
confidential-containers · Mar 4, 2024 · 36a9200 · 36a9200
1 parent dec6f6b
commit 36a9200
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/td-payload/src/mm/shared.rs b/td-payload/src/mm/shared.rs
@@ -11,6 +11,10 @@ use crate::arch::shared::decrypt;
 static SHARED_MEMORY_ALLOCATOR: LockedHeap = LockedHeap::empty();
 
 pub fn init_shared_memory(start: u64, size: usize) {
+    if size % SIZE_4K != 0 {
+        panic!("Failed to initialize shared memory: size needs to be aligned with 0x1000");
+    }
+
     // Set the shared memory region to be shared
     decrypt(start, size);
     // Initialize the shared memory allocator