IntelSiliconPkg/VtdPmrPei: Add premem support.
[mirror_edk2.git] / IntelSiliconPkg / Feature / VTd / IntelVTdPmrPei / IntelVTdPmr.c
index 6179dfe..000a81b 100644 (file)
 \r
 #include "IntelVTdPmrPei.h"\r
 \r
-extern VTD_INFO                *mVTdInfo;\r
-\r
 /**\r
   Get protected low memory alignment.\r
 \r
+  @param HostAddressWidth   The host address width.\r
   @param VtdUnitBaseAddress The base address of the VTd engine.\r
 \r
   @return protected low memory alignment.\r
 **/\r
 UINT32\r
 GetPlmrAlignment (\r
+  IN UINT8         HostAddressWidth,\r
   IN UINTN         VtdUnitBaseAddress\r
   )\r
 {\r
@@ -48,19 +48,18 @@ GetPlmrAlignment (
 /**\r
   Get protected high memory alignment.\r
 \r
+  @param HostAddressWidth   The host address width.\r
   @param VtdUnitBaseAddress The base address of the VTd engine.\r
 \r
   @return protected high memory alignment.\r
 **/\r
 UINT64\r
 GetPhmrAlignment (\r
+  IN UINT8         HostAddressWidth,\r
   IN UINTN         VtdUnitBaseAddress\r
   )\r
 {\r
   UINT64        Data64;\r
-  UINT8         HostAddressWidth;\r
-\r
-  HostAddressWidth = mVTdInfo->HostAddressWidth;\r
 \r
   MmioWrite64 (VtdUnitBaseAddress + R_PMEN_HIGH_BASE_REG, 0xFFFFFFFFFFFFFFFF);\r
   Data64 = MmioRead64 (VtdUnitBaseAddress + R_PMEN_HIGH_BASE_REG);\r
@@ -73,12 +72,14 @@ GetPhmrAlignment (
 /**\r
   Get protected low memory alignment.\r
 \r
+  @param VTdInfo            The VTd engine context information.\r
   @param EngineMask         The mask of the VTd engine to be accessed.\r
 \r
   @return protected low memory alignment.\r
 **/\r
 UINT32\r
 GetLowMemoryAlignment (\r
+  IN VTD_INFO      *VTdInfo,\r
   IN UINT64        EngineMask\r
   )\r
 {\r
@@ -87,11 +88,11 @@ GetLowMemoryAlignment (
   UINT32        FinalAlignment;\r
 \r
   FinalAlignment = 0;\r
-  for (Index = 0; Index < mVTdInfo->VTdEngineCount; Index++) {\r
+  for (Index = 0; Index < VTdInfo->VTdEngineCount; Index++) {\r
     if ((EngineMask & LShiftU64(1, Index)) == 0) {\r
       continue;\r
     }\r
-    Alignment = GetPlmrAlignment ((UINTN)mVTdInfo->VTdEngineAddress[Index]);\r
+    Alignment = GetPlmrAlignment (VTdInfo->HostAddressWidth, (UINTN)VTdInfo->VTdEngineAddress[Index]);\r
     if (FinalAlignment < Alignment) {\r
       FinalAlignment = Alignment;\r
     }\r
@@ -102,12 +103,14 @@ GetLowMemoryAlignment (
 /**\r
   Get protected high memory alignment.\r
 \r
+  @param VTdInfo            The VTd engine context information.\r
   @param EngineMask         The mask of the VTd engine to be accessed.\r
 \r
   @return protected high memory alignment.\r
 **/\r
 UINT64\r
 GetHighMemoryAlignment (\r
+  IN VTD_INFO      *VTdInfo,\r
   IN UINT64        EngineMask\r
   )\r
 {\r
@@ -116,11 +119,11 @@ GetHighMemoryAlignment (
   UINT64        FinalAlignment;\r
 \r
   FinalAlignment = 0;\r
-  for (Index = 0; Index < mVTdInfo->VTdEngineCount; Index++) {\r
+  for (Index = 0; Index < VTdInfo->VTdEngineCount; Index++) {\r
     if ((EngineMask & LShiftU64(1, Index)) == 0) {\r
       continue;\r
     }\r
-    Alignment = GetPhmrAlignment ((UINTN)mVTdInfo->VTdEngineAddress[Index]);\r
+    Alignment = GetPhmrAlignment (VTdInfo->HostAddressWidth, (UINTN)VTdInfo->VTdEngineAddress[Index]);\r
     if (FinalAlignment < Alignment) {\r
       FinalAlignment = Alignment;\r
     }\r
@@ -144,12 +147,19 @@ EnablePmr (
   UINT32        Reg32;\r
   VTD_CAP_REG   CapReg;\r
 \r
+  DEBUG ((DEBUG_INFO, "EnablePmr - %x\n", VtdUnitBaseAddress));\r
+\r
   CapReg.Uint64 = MmioRead64 (VtdUnitBaseAddress + R_CAP_REG);\r
   if (CapReg.Bits.PLMR == 0 || CapReg.Bits.PHMR == 0) {\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
   Reg32 = MmioRead32 (VtdUnitBaseAddress + R_PMEN_ENABLE_REG);\r
+  if (Reg32 == 0xFFFFFFFF) {\r
+    DEBUG ((DEBUG_ERROR, "R_PMEN_ENABLE_REG - 0x%x\n", Reg32));\r
+    ASSERT(FALSE);\r
+  }\r
+\r
   if ((Reg32 & BIT0) == 0) {\r
     MmioWrite32 (VtdUnitBaseAddress + R_PMEN_ENABLE_REG, BIT31);\r
     do {\r
@@ -157,6 +167,8 @@ EnablePmr (
     } while((Reg32 & BIT0) == 0);\r
   }\r
 \r
+  DEBUG ((DEBUG_INFO, "EnablePmr - Done\n"));\r
+\r
   return EFI_SUCCESS;\r
 }\r
 \r
@@ -182,6 +194,11 @@ DisablePmr (
   }\r
 \r
   Reg32 = MmioRead32 (VtdUnitBaseAddress + R_PMEN_ENABLE_REG);\r
+  if (Reg32 == 0xFFFFFFFF) {\r
+    DEBUG ((DEBUG_ERROR, "R_PMEN_ENABLE_REG - 0x%x\n", Reg32));\r
+    ASSERT(FALSE);\r
+  }\r
+\r
   if ((Reg32 & BIT0) != 0) {\r
     MmioWrite32 (VtdUnitBaseAddress + R_PMEN_ENABLE_REG, 0x0);\r
     do {\r
@@ -195,6 +212,7 @@ DisablePmr (
 /**\r
   Set PMR region in the VTd engine.\r
 \r
+  @param HostAddressWidth   The host address width.\r
   @param VtdUnitBaseAddress The base address of the VTd engine.\r
   @param LowMemoryBase      The protected low memory region base.\r
   @param LowMemoryLength    The protected low memory region length.\r
@@ -206,6 +224,7 @@ DisablePmr (
 **/\r
 EFI_STATUS\r
 SetPmrRegion (\r
+  IN UINT8         HostAddressWidth,\r
   IN UINTN         VtdUnitBaseAddress,\r
   IN UINT32        LowMemoryBase,\r
   IN UINT32        LowMemoryLength,\r
@@ -225,9 +244,9 @@ SetPmrRegion (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  PlmrAlignment = GetPlmrAlignment (VtdUnitBaseAddress);\r
+  PlmrAlignment = GetPlmrAlignment (HostAddressWidth, VtdUnitBaseAddress);\r
   DEBUG ((DEBUG_INFO, "PlmrAlignment - 0x%x\n", PlmrAlignment));\r
-  PhmrAlignment = GetPhmrAlignment (VtdUnitBaseAddress);\r
+  PhmrAlignment = GetPhmrAlignment (HostAddressWidth, VtdUnitBaseAddress);\r
   DEBUG ((DEBUG_INFO, "PhmrAlignment - 0x%lx\n", PhmrAlignment));\r
 \r
   if ((LowMemoryBase    != ALIGN_VALUE(LowMemoryBase, PlmrAlignment)) ||\r
@@ -247,8 +266,10 @@ SetPmrRegion (
 \r
   MmioWrite32 (VtdUnitBaseAddress + R_PMEN_LOW_BASE_REG,    LowMemoryBase);\r
   MmioWrite32 (VtdUnitBaseAddress + R_PMEN_LOW_LIMITE_REG,  LowMemoryBase + LowMemoryLength - 1);\r
+  DEBUG ((DEBUG_INFO, "PLMR set done\n"));\r
   MmioWrite64 (VtdUnitBaseAddress + R_PMEN_HIGH_BASE_REG,   HighMemoryBase);\r
   MmioWrite64 (VtdUnitBaseAddress + R_PMEN_HIGH_LIMITE_REG, HighMemoryBase + HighMemoryLength - 1);\r
+  DEBUG ((DEBUG_INFO, "PHMR set done\n"));\r
 \r
   return EFI_SUCCESS;\r
 }\r
@@ -256,6 +277,7 @@ SetPmrRegion (
 /**\r
   Set DMA protected region.\r
 \r
+  @param VTdInfo            The VTd engine context information.\r
   @param EngineMask         The mask of the VTd engine to be accessed.\r
   @param LowMemoryBase      The protected low memory region base.\r
   @param LowMemoryLength    The protected low memory region length.\r
@@ -267,6 +289,7 @@ SetPmrRegion (
 **/\r
 EFI_STATUS\r
 SetDmaProtectedRange (\r
+  IN VTD_INFO      *VTdInfo,\r
   IN UINT64        EngineMask,\r
   IN UINT32        LowMemoryBase,\r
   IN UINT32        LowMemoryLength,\r
@@ -279,13 +302,14 @@ SetDmaProtectedRange (
 \r
   DEBUG ((DEBUG_INFO, "SetDmaProtectedRange(0x%lx) - [0x%x, 0x%x] [0x%lx, 0x%lx]\n", EngineMask, LowMemoryBase, LowMemoryLength, HighMemoryBase, HighMemoryLength));\r
 \r
-  for (Index = 0; Index < mVTdInfo->VTdEngineCount; Index++) {\r
+  for (Index = 0; Index < VTdInfo->VTdEngineCount; Index++) {\r
     if ((EngineMask & LShiftU64(1, Index)) == 0) {\r
       continue;\r
     }\r
-    DisablePmr ((UINTN)mVTdInfo->VTdEngineAddress[Index]);\r
+    DisablePmr ((UINTN)VTdInfo->VTdEngineAddress[Index]);\r
     Status = SetPmrRegion (\r
-               (UINTN)mVTdInfo->VTdEngineAddress[Index],\r
+               VTdInfo->HostAddressWidth,\r
+               (UINTN)VTdInfo->VTdEngineAddress[Index],\r
                LowMemoryBase,\r
                LowMemoryLength,\r
                HighMemoryBase,\r
@@ -294,7 +318,7 @@ SetDmaProtectedRange (
     if (EFI_ERROR(Status)) {\r
       return Status;\r
     }\r
-    Status = EnablePmr ((UINTN)mVTdInfo->VTdEngineAddress[Index]);\r
+    Status = EnablePmr ((UINTN)VTdInfo->VTdEngineAddress[Index]);\r
     if (EFI_ERROR(Status)) {\r
       return Status;\r
     }\r
@@ -306,25 +330,29 @@ SetDmaProtectedRange (
 /**\r
   Diable DMA protection.\r
 \r
+  @param VTdInfo            The VTd engine context information.\r
   @param EngineMask         The mask of the VTd engine to be accessed.\r
 \r
-  @retval DMA protection is disabled.\r
+  @retval EFI_SUCCESS DMA protection is disabled.\r
 **/\r
 EFI_STATUS\r
 DisableDmaProtection (\r
+  IN VTD_INFO      *VTdInfo,\r
   IN UINT64        EngineMask\r
   )\r
 {\r
   UINTN       Index;\r
   EFI_STATUS  Status;\r
 \r
-  DEBUG ((DEBUG_INFO, "DisableDmaProtection\n"));\r
+  DEBUG ((DEBUG_INFO, "DisableDmaProtection - 0x%lx\n", EngineMask));\r
+\r
+  for (Index = 0; Index < VTdInfo->VTdEngineCount; Index++) {\r
+    DEBUG ((DEBUG_INFO, "Disabling...%d\n", Index));\r
 \r
-  for (Index = 0; Index < mVTdInfo->VTdEngineCount; Index++) {\r
     if ((EngineMask & LShiftU64(1, Index)) == 0) {\r
       continue;\r
     }\r
-    Status = DisablePmr ((UINTN)mVTdInfo->VTdEngineAddress[Index]);\r
+    Status = DisablePmr ((UINTN)VTdInfo->VTdEngineAddress[Index]);\r
     if (EFI_ERROR(Status)) {\r
       return Status;\r
     }\r
@@ -332,3 +360,67 @@ DisableDmaProtection (
 \r
   return EFI_SUCCESS;\r
 }\r
+\r
+/**\r
+  Return if the PMR is enabled.\r
+\r
+  @param VtdUnitBaseAddress The base address of the VTd engine.\r
+\r
+  @retval TRUE  PMR is enabled.\r
+  @retval FALSE PMR is disabled or unsupported.\r
+**/\r
+BOOLEAN\r
+IsPmrEnabled (\r
+  IN UINTN         VtdUnitBaseAddress\r
+  )\r
+{\r
+  UINT32        Reg32;\r
+  VTD_CAP_REG   CapReg;\r
+\r
+  CapReg.Uint64 = MmioRead64 (VtdUnitBaseAddress + R_CAP_REG);\r
+  if (CapReg.Bits.PLMR == 0 || CapReg.Bits.PHMR == 0) {\r
+    return FALSE;\r
+  }\r
+\r
+  Reg32 = MmioRead32 (VtdUnitBaseAddress + R_PMEN_ENABLE_REG);\r
+  if ((Reg32 & BIT0) == 0) {\r
+    return FALSE;\r
+  }\r
+\r
+  return TRUE;\r
+}\r
+\r
+/**\r
+  Return the mask of the VTd engine which is enabled.\r
+\r
+  @param VTdInfo            The VTd engine context information.\r
+  @param EngineMask         The mask of the VTd engine to be accessed.\r
+\r
+  @return the mask of the VTd engine which is enabled.\r
+**/\r
+UINT64\r
+GetDmaProtectionEnabledEngineMask (\r
+  IN VTD_INFO      *VTdInfo,\r
+  IN UINT64        EngineMask\r
+  )\r
+{\r
+  UINTN       Index;\r
+  BOOLEAN     Result;\r
+  UINT64      EnabledEngineMask;\r
+\r
+  DEBUG ((DEBUG_INFO, "GetDmaProtectionEnabledEngineMask - 0x%lx\n", EngineMask));\r
+\r
+  EnabledEngineMask = 0;\r
+  for (Index = 0; Index < VTdInfo->VTdEngineCount; Index++) {\r
+    if ((EngineMask & LShiftU64(1, Index)) == 0) {\r
+      continue;\r
+    }\r
+    Result = IsPmrEnabled ((UINTN)VTdInfo->VTdEngineAddress[Index]);\r
+    if (Result) {\r
+      EnabledEngineMask |= LShiftU64(1, Index);\r
+    }\r
+  }\r
+\r
+  DEBUG ((DEBUG_INFO, "EnabledEngineMask - 0x%lx\n", EnabledEngineMask));\r
+  return EnabledEngineMask;\r
+}\r