tee-info-hash-tool: update the metadata check when type is TD_INFO

Signed-off-by: Lu, Xiaoyu1 <[email protected]>
confidential-containers · Dec 13, 2024 · b4ee544 · b4ee544
1 parent 51f8831
commit b4ee544
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/td-shim-tools/src/tee_info_hash.rs b/td-shim-tools/src/tee_info_hash.rs
@@ -290,7 +290,11 @@ impl TdInfoStruct {
                 panic!("Memory address must be 4K aligned!\n");
             }
 
-            if sec.memory_data_size < sec.raw_data_size as u64 {
+            // MemoryAddress and MemoryDataSize shall be zero when the section type is TD_INFO
+            if (sec.r#type != TDX_METADATA_SECTION_TYPE_TD_INFO)
+                && (sec.memory_address != 0 || sec.memory_data_size != 0)
+                && sec.memory_data_size < sec.raw_data_size as u64
+            {
                 panic!("Memory data size must exceed or equal the raw data size!\n");
             }