should panic when sec.memory_data_size < sec.raw_data_size

Signed-off-by: OuyangHang33 <[email protected]>
confidential-containers · May 7, 2024 · 48aef42 · 48aef42
1 parent de5a923
commit 48aef42
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/td-shim-tools/src/tee_info_hash.rs b/td-shim-tools/src/tee_info_hash.rs
@@ -290,6 +290,10 @@ impl TdInfoStruct {
                 panic!("Memory address must be 4K aligned!\n");
             }
 
+            if sec.memory_data_size < sec.raw_data_size as u64 {
+                panic!("Memory data size must exceed or equal the raw data size!\n");
+            }
+
             if sec.memory_data_size % PAGE_SIZE != 0 {
                 panic!("Memory data size must be 4K aligned!\n");
             }