TdInfo.c

Go to the documentation of this file.

 
#include <Library/BaseLib.h>
#include <Library/DebugLib.h>
#include <IndustryStandard/Tdx.h>
#include <Uefi/UefiBaseType.h>
#include <Library/TdxLib.h>
#include <Library/BaseMemoryLib.h>
 
UINT64   mTdSharedPageMask = 0;
UINT32   mTdMaxVCpuNum     = 0;
UINT32   mTdVCpuNum        = 0;
BOOLEAN  mTdDataReturned   = FALSE;
 
BOOLEAN
GetTdInfo (
  VOID
  )
{
  UINT64          Status;
  TD_RETURN_DATA  TdReturnData;
  UINT8           Gpaw;
 
  Status = TdCall (TDCALL_TDINFO, 0, 0, 0, &TdReturnData);
  if (Status == TDX_EXIT_REASON_SUCCESS) {
    Gpaw              = (UINT8)(TdReturnData.TdInfo.Gpaw & 0x3f);
    mTdSharedPageMask = 1ULL << (Gpaw - 1);
    mTdMaxVCpuNum     = TdReturnData.TdInfo.MaxVcpus;
    mTdVCpuNum        = TdReturnData.TdInfo.NumVcpus;
    mTdDataReturned   = TRUE;
  } else {
    DEBUG ((DEBUG_ERROR, "Failed call TDCALL_TDINFO. %llx\n", Status));
    mTdDataReturned = FALSE;
  }
 
  return mTdDataReturned;
}
 
UINT64
EFIAPI
TdSharedPageMask (
  VOID
  )
{
  if (mTdDataReturned) {
    return mTdSharedPageMask;
  }
 
  return GetTdInfo () ? mTdSharedPageMask : 0;
}
 
UINT32
EFIAPI
TdMaxVCpuNum (
  VOID
  )
{
  if (mTdDataReturned) {
    return mTdMaxVCpuNum;
  }
 
  return GetTdInfo () ? mTdMaxVCpuNum : 0;
}
 
UINT32
EFIAPI
TdVCpuNum (
  VOID
  )
{
  if (mTdDataReturned) {
    return mTdVCpuNum;
  }
 
  return GetTdInfo () ? mTdVCpuNum : 0;
}