From feea45c45c76cfdc820e8a8439b30994c82c168c Mon Sep 17 00:00:00 2001
From: OuyangHang33 <hank.ouyang@intel.com>
Date: Mon, 17 Jun 2024 21:00:26 +0800
Subject: [PATCH] Extend TdInfo struct to add vcpu_index field

Signed-off-by: OuyangHang33 <hank.ouyang@intel.com>
---
 tdx-tdcall/src/tdx.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tdx-tdcall/src/tdx.rs b/tdx-tdcall/src/tdx.rs
index 420fa8db..a150d056 100644
--- a/tdx-tdcall/src/tdx.rs
+++ b/tdx-tdcall/src/tdx.rs
@@ -37,7 +37,8 @@ pub struct TdInfo {
     pub attributes: u64,
     pub max_vcpus: u32,
     pub num_vcpus: u32,
-    pub rsvd: [u64; 3],
+    pub vcpu_index: u32,
+    pub rsvd: [u32; 5],
 }
 
 /// Virtualization exception information returned from TDG.VP.VEINFO.GET leaf
@@ -506,6 +507,7 @@ pub fn tdcall_get_td_info() -> Result<TdInfo, TdCallError> {
         attributes: args.rdx,
         max_vcpus: (args.r8 >> 32) as u32,
         num_vcpus: args.r8 as u32,
+        vcpu_index: args.r9 as u32,
         ..Default::default()
     };