IntelSiliconPkg/IntelVTd: update PlatformVtdPolicy

[mirror_edk2.git] / IntelSiliconPkg / IntelVTdDxe / TranslationTableEx.c

diff --git a/IntelSiliconPkg/IntelVTdDxe/TranslationTableEx.c b/IntelSiliconPkg/IntelVTdDxe/TranslationTableEx.c
index 9d4e6eae2c1d619ba98563f178454fa9e6ff0560..68b25a79bf390c9ee2d2e7b55a7bec6e43756298 100644 (file)
--- a/IntelSiliconPkg/IntelVTdDxe/TranslationTableEx.c
+++ b/IntelSiliconPkg/IntelVTdDxe/TranslationTableEx.c
@@ -33,16 +33,16 @@ CreateExtContextEntry (
   VTD_EXT_ROOT_ENTRY     *ExtRootEntry;\r
   VTD_EXT_CONTEXT_ENTRY  *ExtContextEntryTable;\r
   VTD_EXT_CONTEXT_ENTRY  *ExtContextEntry;\r
-  VTD_SOURCE_ID          *PciDescriptor;\r
+  VTD_SOURCE_ID          *PciSourceId;\r
   VTD_SOURCE_ID          SourceId;\r
   UINTN                  MaxBusNumber;\r
   UINTN                  EntryTablePages;\r
 \r
   MaxBusNumber = 0;\r
-  for (Index = 0; Index < mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptorNumber; Index++) {\r
-    PciDescriptor = &mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptors[Index];\r
-    if (PciDescriptor->Bits.Bus > MaxBusNumber) {\r
-      MaxBusNumber = PciDescriptor->Bits.Bus;\r
+  for (Index = 0; Index < mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDeviceDataNumber; Index++) {\r
+    PciSourceId = &mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDeviceData[Index].PciSourceId;\r
+    if (PciSourceId->Bits.Bus > MaxBusNumber) {\r
+      MaxBusNumber = PciSourceId->Bits.Bus;\r
     }\r
   }\r
   DEBUG ((DEBUG_INFO,"  MaxBusNumber - 0x%x\n", MaxBusNumber));\r
@@ -58,12 +58,12 @@ CreateExtContextEntry (
   mVtdUnitInformation[VtdIndex].ExtRootEntryTable = (VTD_EXT_ROOT_ENTRY *)Buffer;\r
   Buffer = (UINT8 *)Buffer + EFI_PAGES_TO_SIZE (RootPages);\r
 \r
-  for (Index = 0; Index < mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptorNumber; Index++) {\r
-    PciDescriptor = &mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptors[Index];\r
+  for (Index = 0; Index < mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDeviceDataNumber; Index++) {\r
+    PciSourceId = &mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDeviceData[Index].PciSourceId;\r
 \r
-    SourceId.Bits.Bus = PciDescriptor->Bits.Bus;\r
-    SourceId.Bits.Device = PciDescriptor->Bits.Device;\r
-    SourceId.Bits.Function = PciDescriptor->Bits.Function;\r
+    SourceId.Bits.Bus = PciSourceId->Bits.Bus;\r
+    SourceId.Bits.Device = PciSourceId->Bits.Device;\r
+    SourceId.Bits.Function = PciSourceId->Bits.Function;\r
 \r
     ExtRootEntry = &mVtdUnitInformation[VtdIndex].ExtRootEntryTable[SourceId.Index.RootIndex];\r
     if (ExtRootEntry->Bits.LowerPresent == 0) {\r