X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=IntelSiliconPkg%2FIntelVTdDxe%2FPciInfo.c;fp=IntelSiliconPkg%2FIntelVTdDxe%2FPciInfo.c;h=d5f096fadd5dca5e514eeb9610bd9465a6e54c92;hp=ea84317c9ce439e761f1ae5e376c441fbc3c7b0f;hb=76c6f69ccadc7835c9616b077d9ff1b8e46fe49e;hpb=9169c6e81854c7479fcc99ce91704f3f3947e28f

diff --git a/IntelSiliconPkg/IntelVTdDxe/PciInfo.c b/IntelSiliconPkg/IntelVTdDxe/PciInfo.c
index ea84317c9c..d5f096fadd 100644
--- a/IntelSiliconPkg/IntelVTdDxe/PciInfo.c
+++ b/IntelSiliconPkg/IntelVTdDxe/PciInfo.c
@@ -289,7 +289,7 @@ FindVtdIndexByPciDevice (
 
     if (mVtdUnitInformation[VtdIndex].ExtRootEntryTable != 0) {
       ExtRootEntry = &mVtdUnitInformation[VtdIndex].ExtRootEntryTable[SourceId.Index.RootIndex];
-      ExtContextEntryTable = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)LShiftU64 (ExtRootEntry->Bits.LowerContextTablePointer, 12) ;
+      ExtContextEntryTable = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry->Bits.LowerContextTablePointerLo, ExtRootEntry->Bits.LowerContextTablePointerHi) ;
       ThisExtContextEntry  = &ExtContextEntryTable[SourceId.Index.ContextIndex];
       if (ThisExtContextEntry->Bits.AddressWidth == 0) {
         continue;
@@ -298,7 +298,7 @@ FindVtdIndexByPciDevice (
       *ContextEntry    = NULL;
     } else {
       RootEntry = &mVtdUnitInformation[VtdIndex].RootEntryTable[SourceId.Index.RootIndex];
-      ContextEntryTable = (VTD_CONTEXT_ENTRY *)(UINTN)LShiftU64 (RootEntry->Bits.ContextTablePointer, 12) ;
+      ContextEntryTable = (VTD_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(RootEntry->Bits.ContextTablePointerLo, RootEntry->Bits.ContextTablePointerHi) ;
       ThisContextEntry  = &ContextEntryTable[SourceId.Index.ContextIndex];
       if (ThisContextEntry->Bits.AddressWidth == 0) {
         continue;