+++ /dev/null
-/** @file\r
-\r
- Copyright (c) 2017, Intel Corporation. All rights reserved.<BR>\r
- SPDX-License-Identifier: BSD-2-Clause-Patent\r
-\r
-**/\r
-\r
-#include "DmaProtection.h"\r
-\r
-/**\r
- Create extended context entry.\r
-\r
- @param[in] VtdIndex The index of the VTd engine.\r
-\r
- @retval EFI_SUCCESS The extended context entry is created.\r
- @retval EFI_OUT_OF_RESOURCE No enough resource to create extended context entry.\r
-**/\r
-EFI_STATUS\r
-CreateExtContextEntry (\r
- IN UINTN VtdIndex\r
- )\r
-{\r
- UINTN Index;\r
- VOID *Buffer;\r
- UINTN RootPages;\r
- UINTN ContextPages;\r
- VTD_EXT_ROOT_ENTRY *ExtRootEntry;\r
- VTD_EXT_CONTEXT_ENTRY *ExtContextEntryTable;\r
- VTD_EXT_CONTEXT_ENTRY *ExtContextEntry;\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.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
-\r
- RootPages = EFI_SIZE_TO_PAGES (sizeof (VTD_EXT_ROOT_ENTRY) * VTD_ROOT_ENTRY_NUMBER);\r
- ContextPages = EFI_SIZE_TO_PAGES (sizeof (VTD_EXT_CONTEXT_ENTRY) * VTD_CONTEXT_ENTRY_NUMBER);\r
- EntryTablePages = RootPages + ContextPages * (MaxBusNumber + 1);\r
- Buffer = AllocateZeroPages (EntryTablePages);\r
- if (Buffer == NULL) {\r
- DEBUG ((DEBUG_INFO,"Could not Alloc Root Entry Table.. \n"));\r
- return EFI_OUT_OF_RESOURCES;\r
- }\r
- 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.PciDeviceDataNumber; Index++) {\r
- PciSourceId = &mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDeviceData[Index].PciSourceId;\r
-\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
- ExtRootEntry->Bits.LowerContextTablePointerLo = (UINT32) RShiftU64 ((UINT64)(UINTN)Buffer, 12);\r
- ExtRootEntry->Bits.LowerContextTablePointerHi = (UINT32) RShiftU64 ((UINT64)(UINTN)Buffer, 32);\r
- ExtRootEntry->Bits.LowerPresent = 1;\r
- ExtRootEntry->Bits.UpperContextTablePointerLo = (UINT32) RShiftU64 ((UINT64)(UINTN)Buffer, 12) + 1;\r
- ExtRootEntry->Bits.UpperContextTablePointerHi = (UINT32) RShiftU64 (RShiftU64 ((UINT64)(UINTN)Buffer, 12) + 1, 20);\r
- ExtRootEntry->Bits.UpperPresent = 1;\r
- Buffer = (UINT8 *)Buffer + EFI_PAGES_TO_SIZE (ContextPages);\r
- }\r
-\r
- ExtContextEntryTable = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry->Bits.LowerContextTablePointerLo, ExtRootEntry->Bits.LowerContextTablePointerHi) ;\r
- ExtContextEntry = &ExtContextEntryTable[SourceId.Index.ContextIndex];\r
- ExtContextEntry->Bits.TranslationType = 0;\r
- ExtContextEntry->Bits.FaultProcessingDisable = 0;\r
- ExtContextEntry->Bits.Present = 0;\r
-\r
- DEBUG ((DEBUG_INFO,"DOMAIN: S%04x, B%02x D%02x F%02x\n", mVtdUnitInformation[VtdIndex].Segment, SourceId.Bits.Bus, SourceId.Bits.Device, SourceId.Bits.Function));\r
-\r
- switch (mVtdUnitInformation[VtdIndex].CapReg.Bits.SAGAW) {\r
- case BIT1:\r
- ExtContextEntry->Bits.AddressWidth = 0x1;\r
- break;\r
- case BIT2:\r
- ExtContextEntry->Bits.AddressWidth = 0x2;\r
- break;\r
- }\r
- }\r
-\r
- FlushPageTableMemory (VtdIndex, (UINTN)mVtdUnitInformation[VtdIndex].ExtRootEntryTable, EFI_PAGES_TO_SIZE(EntryTablePages));\r
-\r
- return EFI_SUCCESS;\r
-}\r
-\r
-/**\r
- Dump DMAR extended context entry table.\r
-\r
- @param[in] ExtRootEntry DMAR extended root entry.\r
-**/\r
-VOID\r
-DumpDmarExtContextEntryTable (\r
- IN VTD_EXT_ROOT_ENTRY *ExtRootEntry\r
- )\r
-{\r
- UINTN Index;\r
- UINTN Index2;\r
- VTD_EXT_CONTEXT_ENTRY *ExtContextEntry;\r
-\r
- DEBUG ((DEBUG_INFO,"=========================\n"));\r
- DEBUG ((DEBUG_INFO,"DMAR ExtContext Entry Table:\n"));\r
-\r
- DEBUG ((DEBUG_INFO,"ExtRootEntry Address - 0x%x\n", ExtRootEntry));\r
-\r
- for (Index = 0; Index < VTD_ROOT_ENTRY_NUMBER; Index++) {\r
- if ((ExtRootEntry[Index].Uint128.Uint64Lo != 0) || (ExtRootEntry[Index].Uint128.Uint64Hi != 0)) {\r
- DEBUG ((DEBUG_INFO," ExtRootEntry(0x%02x) B%02x - 0x%016lx %016lx\n",\r
- Index, Index, ExtRootEntry[Index].Uint128.Uint64Hi, ExtRootEntry[Index].Uint128.Uint64Lo));\r
- }\r
- if (ExtRootEntry[Index].Bits.LowerPresent == 0) {\r
- continue;\r
- }\r
- ExtContextEntry = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry[Index].Bits.LowerContextTablePointerLo, ExtRootEntry[Index].Bits.LowerContextTablePointerHi);\r
- for (Index2 = 0; Index2 < VTD_CONTEXT_ENTRY_NUMBER/2; Index2++) {\r
- if ((ExtContextEntry[Index2].Uint256.Uint64_1 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_2 != 0) ||\r
- (ExtContextEntry[Index2].Uint256.Uint64_3 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_4 != 0)) {\r
- DEBUG ((DEBUG_INFO," ExtContextEntryLower(0x%02x) D%02xF%02x - 0x%016lx %016lx %016lx %016lx\n",\r
- Index2, Index2 >> 3, Index2 & 0x7, ExtContextEntry[Index2].Uint256.Uint64_4, ExtContextEntry[Index2].Uint256.Uint64_3, ExtContextEntry[Index2].Uint256.Uint64_2, ExtContextEntry[Index2].Uint256.Uint64_1));\r
- }\r
- if (ExtContextEntry[Index2].Bits.Present == 0) {\r
- continue;\r
- }\r
- DumpSecondLevelPagingEntry ((VOID *)(UINTN)VTD_64BITS_ADDRESS(ExtContextEntry[Index2].Bits.SecondLevelPageTranslationPointerLo, ExtContextEntry[Index2].Bits.SecondLevelPageTranslationPointerHi));\r
- }\r
-\r
- if (ExtRootEntry[Index].Bits.UpperPresent == 0) {\r
- continue;\r
- }\r
- ExtContextEntry = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry[Index].Bits.UpperContextTablePointerLo, ExtRootEntry[Index].Bits.UpperContextTablePointerHi);\r
- for (Index2 = 0; Index2 < VTD_CONTEXT_ENTRY_NUMBER/2; Index2++) {\r
- if ((ExtContextEntry[Index2].Uint256.Uint64_1 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_2 != 0) ||\r
- (ExtContextEntry[Index2].Uint256.Uint64_3 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_4 != 0)) {\r
- DEBUG ((DEBUG_INFO," ExtContextEntryUpper(0x%02x) D%02xF%02x - 0x%016lx %016lx %016lx %016lx\n",\r
- Index2, (Index2 + 128) >> 3, (Index2 + 128) & 0x7, ExtContextEntry[Index2].Uint256.Uint64_4, ExtContextEntry[Index2].Uint256.Uint64_3, ExtContextEntry[Index2].Uint256.Uint64_2, ExtContextEntry[Index2].Uint256.Uint64_1));\r
- }\r
- if (ExtContextEntry[Index2].Bits.Present == 0) {\r
- continue;\r
- }\r
- }\r
- }\r
- DEBUG ((DEBUG_INFO,"=========================\n"));\r
-}\r