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]>

Author: gaojiaqi7

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