\r
if (mVtdUnitInformation[VtdIndex].ExtRootEntryTable != 0) {\r
ExtRootEntry = &mVtdUnitInformation[VtdIndex].ExtRootEntryTable[SourceId.Index.RootIndex];\r
- ExtContextEntryTable = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)LShiftU64 (ExtRootEntry->Bits.LowerContextTablePointer, 12) ;\r
+ ExtContextEntryTable = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry->Bits.LowerContextTablePointerLo, ExtRootEntry->Bits.LowerContextTablePointerHi) ;\r
ThisExtContextEntry = &ExtContextEntryTable[SourceId.Index.ContextIndex];\r
if (ThisExtContextEntry->Bits.AddressWidth == 0) {\r
continue;\r
*ContextEntry = NULL;\r
} else {\r
RootEntry = &mVtdUnitInformation[VtdIndex].RootEntryTable[SourceId.Index.RootIndex];\r
- ContextEntryTable = (VTD_CONTEXT_ENTRY *)(UINTN)LShiftU64 (RootEntry->Bits.ContextTablePointer, 12) ;\r
+ ContextEntryTable = (VTD_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(RootEntry->Bits.ContextTablePointerLo, RootEntry->Bits.ContextTablePointerHi) ;\r
ThisContextEntry = &ContextEntryTable[SourceId.Index.ContextIndex];\r
if (ThisContextEntry->Bits.AddressWidth == 0) {\r
continue;\r