IntelSiliconPkg/IntelVTdDxe/TranslationTableEx.c

   1 /** @file
   2
   3   Copyright (c) 2017, Intel Corporation. All rights reserved.<BR>
   4   This program and the accompanying materials
   5   are licensed and made available under the terms and conditions of the BSD License
   6   which accompanies this distribution.  The full text of the license may be found at
   7   http://opensource.org/licenses/bsd-license.php.
   8
   9   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  10   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  11
  12 **/
  13
  14 #include "DmaProtection.h"
  15
  16 /**
  17   Create extended context entry.
  18
  19   @param[in]  VtdIndex  The index of the VTd engine.
  20
  21   @retval EFI_SUCCESS          The extended context entry is created.
  22   @retval EFI_OUT_OF_RESOURCE  No enough resource to create extended context entry.
  23 **/
  24 EFI_STATUS
  25 CreateExtContextEntry (
  26   IN UINTN  VtdIndex
  27   )
  28 {
  29   UINTN                  Index;
  30   VOID                   *Buffer;
  31   UINTN                  RootPages;
  32   UINTN                  ContextPages;
  33   VTD_EXT_ROOT_ENTRY     *ExtRootEntry;
  34   VTD_EXT_CONTEXT_ENTRY  *ExtContextEntryTable;
  35   VTD_EXT_CONTEXT_ENTRY  *ExtContextEntry;
  36   VTD_SOURCE_ID          *PciDescriptor;
  37   VTD_SOURCE_ID          SourceId;
  38   UINTN                  MaxBusNumber;
  39   UINTN                  EntryTablePages;
  40
  41   MaxBusNumber = 0;
  42   for (Index = 0; Index < mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptorNumber; Index++) {
  43     PciDescriptor = &mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptors[Index];
  44     if (PciDescriptor->Bits.Bus > MaxBusNumber) {
  45       MaxBusNumber = PciDescriptor->Bits.Bus;
  46     }
  47   }
  48   DEBUG ((DEBUG_INFO,"  MaxBusNumber - 0x%x\n", MaxBusNumber));
  49
  50   RootPages = EFI_SIZE_TO_PAGES (sizeof (VTD_EXT_ROOT_ENTRY) * VTD_ROOT_ENTRY_NUMBER);
  51   ContextPages = EFI_SIZE_TO_PAGES (sizeof (VTD_EXT_CONTEXT_ENTRY) * VTD_CONTEXT_ENTRY_NUMBER);
  52   EntryTablePages = RootPages + ContextPages * (MaxBusNumber + 1);
  53   Buffer = AllocateZeroPages (EntryTablePages);
  54   if (Buffer == NULL) {
  55     DEBUG ((DEBUG_INFO,"Could not Alloc Root Entry Table.. \n"));
  56     return EFI_OUT_OF_RESOURCES;
  57   }
  58   mVtdUnitInformation[VtdIndex].ExtRootEntryTable = (VTD_EXT_ROOT_ENTRY *)Buffer;
  59   Buffer = (UINT8 *)Buffer + EFI_PAGES_TO_SIZE (RootPages);
  60
  61   for (Index = 0; Index < mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptorNumber; Index++) {
  62     PciDescriptor = &mVtdUnitInformation[VtdIndex].PciDeviceInfo.PciDescriptors[Index];
  63
  64     SourceId.Bits.Bus = PciDescriptor->Bits.Bus;
  65     SourceId.Bits.Device = PciDescriptor->Bits.Device;
  66     SourceId.Bits.Function = PciDescriptor->Bits.Function;
  67
  68     ExtRootEntry = &mVtdUnitInformation[VtdIndex].ExtRootEntryTable[SourceId.Index.RootIndex];
  69     if (ExtRootEntry->Bits.LowerPresent == 0) {
  70       ExtRootEntry->Bits.LowerContextTablePointerLo  = (UINT32) RShiftU64 ((UINT64)(UINTN)Buffer, 12);
  71       ExtRootEntry->Bits.LowerContextTablePointerHi  = (UINT32) RShiftU64 ((UINT64)(UINTN)Buffer, 32);
  72       ExtRootEntry->Bits.LowerPresent = 1;
  73       ExtRootEntry->Bits.UpperContextTablePointerLo  = (UINT32) RShiftU64 ((UINT64)(UINTN)Buffer, 12) + 1;
  74       ExtRootEntry->Bits.UpperContextTablePointerHi  = (UINT32) RShiftU64 (RShiftU64 ((UINT64)(UINTN)Buffer, 12) + 1, 20);
  75       ExtRootEntry->Bits.UpperPresent = 1;
  76       Buffer = (UINT8 *)Buffer + EFI_PAGES_TO_SIZE (ContextPages);
  77     }
  78
  79     ExtContextEntryTable = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry->Bits.LowerContextTablePointerLo, ExtRootEntry->Bits.LowerContextTablePointerHi) ;
  80     ExtContextEntry = &ExtContextEntryTable[SourceId.Index.ContextIndex];
  81     ExtContextEntry->Bits.TranslationType = 0;
  82     ExtContextEntry->Bits.FaultProcessingDisable = 0;
  83     ExtContextEntry->Bits.Present = 0;
  84
  85     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));
  86
  87     switch (mVtdUnitInformation[VtdIndex].CapReg.Bits.SAGAW) {
  88     case BIT1:
  89       ExtContextEntry->Bits.AddressWidth = 0x1;
  90       break;
  91     case BIT2:
  92       ExtContextEntry->Bits.AddressWidth = 0x2;
  93       break;
  94     }
  95   }
  96
  97   return EFI_SUCCESS;
  98 }
  99
 100 /**
 101   Dump DMAR extended context entry table.
 102
 103   @param[in]  ExtRootEntry DMAR extended root entry.
 104 **/
 105 VOID
 106 DumpDmarExtContextEntryTable (
 107   IN VTD_EXT_ROOT_ENTRY *ExtRootEntry
 108   )
 109 {
 110   UINTN                 Index;
 111   UINTN                 Index2;
 112   VTD_EXT_CONTEXT_ENTRY *ExtContextEntry;
 113
 114   DEBUG ((DEBUG_INFO,"=========================\n"));
 115   DEBUG ((DEBUG_INFO,"DMAR ExtContext Entry Table:\n"));
 116
 117   DEBUG ((DEBUG_INFO,"ExtRootEntry Address - 0x%x\n", ExtRootEntry));
 118
 119   for (Index = 0; Index < VTD_ROOT_ENTRY_NUMBER; Index++) {
 120     if ((ExtRootEntry[Index].Uint128.Uint64Lo != 0) || (ExtRootEntry[Index].Uint128.Uint64Hi != 0)) {
 121       DEBUG ((DEBUG_INFO,"  ExtRootEntry(0x%02x) B%02x - 0x%016lx %016lx\n",
 122         Index, Index, ExtRootEntry[Index].Uint128.Uint64Hi, ExtRootEntry[Index].Uint128.Uint64Lo));
 123     }
 124     if (ExtRootEntry[Index].Bits.LowerPresent == 0) {
 125       continue;
 126     }
 127     ExtContextEntry = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry[Index].Bits.LowerContextTablePointerLo, ExtRootEntry[Index].Bits.LowerContextTablePointerHi);
 128     for (Index2 = 0; Index2 < VTD_CONTEXT_ENTRY_NUMBER/2; Index2++) {
 129       if ((ExtContextEntry[Index2].Uint256.Uint64_1 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_2 != 0) ||
 130           (ExtContextEntry[Index2].Uint256.Uint64_3 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_4 != 0)) {
 131         DEBUG ((DEBUG_INFO,"    ExtContextEntryLower(0x%02x) D%02xF%02x - 0x%016lx %016lx %016lx %016lx\n",
 132           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));
 133       }
 134       if (ExtContextEntry[Index2].Bits.Present == 0) {
 135         continue;
 136       }
 137     }
 138
 139     if (ExtRootEntry[Index].Bits.UpperPresent == 0) {
 140       continue;
 141     }
 142     ExtContextEntry = (VTD_EXT_CONTEXT_ENTRY *)(UINTN)VTD_64BITS_ADDRESS(ExtRootEntry[Index].Bits.UpperContextTablePointerLo, ExtRootEntry[Index].Bits.UpperContextTablePointerHi);
 143     for (Index2 = 0; Index2 < VTD_CONTEXT_ENTRY_NUMBER/2; Index2++) {
 144       if ((ExtContextEntry[Index2].Uint256.Uint64_1 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_2 != 0) ||
 145           (ExtContextEntry[Index2].Uint256.Uint64_3 != 0) || (ExtContextEntry[Index2].Uint256.Uint64_4 != 0)) {
 146         DEBUG ((DEBUG_INFO,"    ExtContextEntryUpper(0x%02x) D%02xF%02x - 0x%016lx %016lx %016lx %016lx\n",
 147           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));
 148       }
 149       if (ExtContextEntry[Index2].Bits.Present == 0) {
 150         continue;
 151       }
 152     }
 153   }
 154   DEBUG ((DEBUG_INFO,"=========================\n"));
 155 }