-/** @file\r
- Platform VTd Info Sample PEI driver.\r
-\r
- Copyright (c) 2017, Intel Corporation. All rights reserved.<BR>\r
- SPDX-License-Identifier: BSD-2-Clause-Patent\r
-\r
-**/\r
-\r
-#include <PiPei.h>\r
-\r
-#include <Ppi/VtdInfo.h>\r
-\r
-#include <Library/PeiServicesLib.h>\r
-#include <Library/DebugLib.h>\r
-#include <Library/PciLib.h>\r
-#include <Library/IoLib.h>\r
-#include <Library/MemoryAllocationLib.h>\r
-\r
-#define R_SA_MCHBAR (0x48)\r
-#define R_SA_GGC (0x50)\r
-#define N_SKL_SA_GGC_GGMS_OFFSET (0x6)\r
-#define B_SKL_SA_GGC_GGMS_MASK (0xc0)\r
-#define N_SKL_SA_GGC_GMS_OFFSET (0x8)\r
-#define B_SKL_SA_GGC_GMS_MASK (0xff00)\r
-#define V_SKL_SA_GGC_GGMS_8MB 3\r
-#define R_SA_TOLUD (0xbc)\r
-\r
-#define R_SA_MCHBAR_VTD1_OFFSET 0x5400 ///< HW UNIT for IGD\r
-#define R_SA_MCHBAR_VTD2_OFFSET 0x5410 ///< HW UNIT for all other - PEG, USB, SATA etc\r
-\r
-EFI_GUID gEdkiiSiliconInitializedPpiGuid = {0x82a72dc8, 0x61ec, 0x403e, {0xb1, 0x5a, 0x8d, 0x7a, 0x3a, 0x71, 0x84, 0x98}};\r
-\r
-typedef struct {\r
- EFI_ACPI_DMAR_HEADER DmarHeader;\r
- //\r
- // VTd engine 1 - integrated graphic\r
- //\r
- EFI_ACPI_DMAR_DRHD_HEADER Drhd1;\r
- EFI_ACPI_DMAR_DEVICE_SCOPE_STRUCTURE_HEADER Drhd11;\r
- EFI_ACPI_DMAR_PCI_PATH Drhd111;\r
- //\r
- // VTd engine 2 - all rest\r
- //\r
- EFI_ACPI_DMAR_DRHD_HEADER Drhd2;\r
- //\r
- // RMRR 1 - integrated graphic\r
- //\r
- EFI_ACPI_DMAR_RMRR_HEADER Rmrr1;\r
- EFI_ACPI_DMAR_DEVICE_SCOPE_STRUCTURE_HEADER Rmrr11;\r
- EFI_ACPI_DMAR_PCI_PATH Rmrr111;\r
-} MY_VTD_INFO_PPI;\r
-\r
-MY_VTD_INFO_PPI mPlatformVTdSample = {\r
- { // DmarHeader\r
- { // Header\r
- EFI_ACPI_4_0_DMA_REMAPPING_TABLE_SIGNATURE,\r
- sizeof(MY_VTD_INFO_PPI),\r
- EFI_ACPI_DMAR_REVISION,\r
- },\r
- 0x26, // HostAddressWidth\r
- },\r
-\r
- { // Drhd1\r
- { // Header\r
- EFI_ACPI_DMAR_TYPE_DRHD,\r
- sizeof(EFI_ACPI_DMAR_DRHD_HEADER) +\r
- sizeof(EFI_ACPI_DMAR_DEVICE_SCOPE_STRUCTURE_HEADER) +\r
- sizeof(EFI_ACPI_DMAR_PCI_PATH)\r
- },\r
- 0, // Flags\r
- 0, // Reserved\r
- 0, // SegmentNumber\r
- 0xFED90000 // RegisterBaseAddress -- TO BE PATCHED\r
- },\r
- { // Drhd11\r
- EFI_ACPI_DEVICE_SCOPE_ENTRY_TYPE_PCI_ENDPOINT,\r
- sizeof(EFI_ACPI_DMAR_DEVICE_SCOPE_STRUCTURE_HEADER) +\r
- sizeof(EFI_ACPI_DMAR_PCI_PATH),\r
- 0, // Reserved2\r
- 0, // EnumerationId\r
- 0 // StartBusNumber\r
- },\r
- { // Drhd111\r
- 2, // Device\r
- 0 // Function\r
- },\r
-\r
- { // Drhd2\r
- { // Header\r
- EFI_ACPI_DMAR_TYPE_DRHD,\r
- sizeof(EFI_ACPI_DMAR_DRHD_HEADER)\r
- },\r
- EFI_ACPI_DMAR_DRHD_FLAGS_INCLUDE_PCI_ALL, // Flags\r
- 0, // Reserved\r
- 0, // SegmentNumber\r
- 0xFED91000 // RegisterBaseAddress -- TO BE PATCHED\r
- },\r
-\r
- { // Rmrr1\r
- { // Header\r
- EFI_ACPI_DMAR_TYPE_RMRR,\r
- sizeof(EFI_ACPI_DMAR_RMRR_HEADER) +\r
- sizeof(EFI_ACPI_DMAR_DEVICE_SCOPE_STRUCTURE_HEADER) +\r
- sizeof(EFI_ACPI_DMAR_PCI_PATH)\r
- },\r
- {0}, // Reserved\r
- 0, // SegmentNumber\r
- 0x0, // ReservedMemoryRegionBaseAddress -- TO BE PATCHED\r
- 0x0 // ReservedMemoryRegionLimitAddress -- TO BE PATCHED\r
- },\r
- { // Rmrr11\r
- EFI_ACPI_DEVICE_SCOPE_ENTRY_TYPE_PCI_ENDPOINT,\r
- sizeof(EFI_ACPI_DMAR_DEVICE_SCOPE_STRUCTURE_HEADER) +\r
- sizeof(EFI_ACPI_DMAR_PCI_PATH),\r
- 0, // Reserved2\r
- 0, // EnumerationId\r
- 0 // StartBusNumber\r
- },\r
- { // Rmrr111\r
- 2, // Device\r
- 0 // Function\r
- },\r
-};\r
-\r
-EFI_PEI_PPI_DESCRIPTOR mPlatformVTdInfoSampleDesc = {\r
- (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
- &gEdkiiVTdInfoPpiGuid,\r
- &mPlatformVTdSample\r
-};\r
-\r
-typedef struct {\r
- EFI_ACPI_DMAR_HEADER DmarHeader;\r
- //\r
- // VTd engine 2 - all rest\r
- //\r
- EFI_ACPI_DMAR_DRHD_HEADER Drhd2;\r
-} MY_VTD_INFO_NO_IGD_PPI;\r
-\r
-MY_VTD_INFO_NO_IGD_PPI mPlatformVTdNoIgdSample = {\r
- { // DmarHeader\r
- { // Header\r
- EFI_ACPI_4_0_DMA_REMAPPING_TABLE_SIGNATURE,\r
- sizeof(MY_VTD_INFO_NO_IGD_PPI),\r
- EFI_ACPI_DMAR_REVISION,\r
- },\r
- 0x26, // HostAddressWidth\r
- },\r
-\r
- { // Drhd2\r
- { // Header\r
- EFI_ACPI_DMAR_TYPE_DRHD,\r
- sizeof(EFI_ACPI_DMAR_DRHD_HEADER)\r
- },\r
- EFI_ACPI_DMAR_DRHD_FLAGS_INCLUDE_PCI_ALL, // Flags\r
- 0, // Reserved\r
- 0, // SegmentNumber\r
- 0xFED91000 // RegisterBaseAddress -- TO BE PATCHED\r
- },\r
-};\r
-\r
-EFI_PEI_PPI_DESCRIPTOR mPlatformVTdNoIgdInfoSampleDesc = {\r
- (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
- &gEdkiiVTdInfoPpiGuid,\r
- &mPlatformVTdNoIgdSample\r
-};\r
-\r
-/**\r
- Initialize VTd register.\r
-**/\r
-VOID\r
-InitDmar (\r
- VOID\r
- )\r
-{\r
- UINT32 MchBar;\r
-\r
- DEBUG ((DEBUG_INFO, "InitDmar\n"));\r
-\r
- MchBar = PciRead32 (PCI_LIB_ADDRESS(0, 0, 0, R_SA_MCHBAR)) & ~BIT0;\r
- PciWrite32 (PCI_LIB_ADDRESS(0, 0, 0, R_SA_MCHBAR), 0xFED10000 | BIT0);\r
- DEBUG ((DEBUG_INFO, "MchBar - %x\n", MchBar));\r
-\r
- MmioWrite32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET, (UINT32)mPlatformVTdSample.Drhd2.RegisterBaseAddress | 1);\r
- DEBUG ((DEBUG_INFO, "VTd2 - %x\n", (MmioRead32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET))));\r
-}\r
-\r
-/**\r
- Patch Graphic UMA address in RMRR and base address.\r
-**/\r
-EFI_PEI_PPI_DESCRIPTOR *\r
-PatchDmar (\r
- VOID\r
- )\r
-{\r
- UINT32 MchBar;\r
- UINT16 IgdMode;\r
- UINT16 GttMode;\r
- UINT32 IgdMemSize;\r
- UINT32 GttMemSize;\r
- MY_VTD_INFO_PPI *PlatformVTdSample;\r
- EFI_PEI_PPI_DESCRIPTOR *PlatformVTdInfoSampleDesc;\r
- MY_VTD_INFO_NO_IGD_PPI *PlatformVTdNoIgdSample;\r
- EFI_PEI_PPI_DESCRIPTOR *PlatformVTdNoIgdInfoSampleDesc;\r
-\r
- DEBUG ((DEBUG_INFO, "PatchDmar\n"));\r
-\r
- if (PciRead16 (PCI_LIB_ADDRESS(0, 2, 0, 0)) != 0xFFFF) {\r
- PlatformVTdSample = AllocateCopyPool (sizeof(MY_VTD_INFO_PPI), &mPlatformVTdSample);\r
- ASSERT(PlatformVTdSample != NULL);\r
- PlatformVTdInfoSampleDesc = AllocateCopyPool (sizeof(EFI_PEI_PPI_DESCRIPTOR), &mPlatformVTdInfoSampleDesc);\r
- ASSERT(PlatformVTdInfoSampleDesc != NULL);\r
- PlatformVTdInfoSampleDesc->Ppi = PlatformVTdSample;\r
-\r
- ///\r
- /// Calculate IGD memsize\r
- ///\r
- IgdMode = ((PciRead16 (PCI_LIB_ADDRESS(0, 0, 0, R_SA_GGC)) & B_SKL_SA_GGC_GMS_MASK) >> N_SKL_SA_GGC_GMS_OFFSET) & 0xFF;\r
- if (IgdMode < 0xF0) {\r
- IgdMemSize = IgdMode * 32 * (1024) * (1024);\r
- } else {\r
- IgdMemSize = 4 * (IgdMode - 0xF0 + 1) * (1024) * (1024);\r
- }\r
-\r
- ///\r
- /// Calculate GTT mem size\r
- ///\r
- GttMemSize = 0;\r
- GttMode = (PciRead16 (PCI_LIB_ADDRESS(0, 0, 0, R_SA_GGC)) & B_SKL_SA_GGC_GGMS_MASK) >> N_SKL_SA_GGC_GGMS_OFFSET;\r
- if (GttMode <= V_SKL_SA_GGC_GGMS_8MB) {\r
- GttMemSize = (1 << GttMode) * (1024) * (1024);\r
- }\r
-\r
- PlatformVTdSample->Rmrr1.ReservedMemoryRegionBaseAddress = (PciRead32 (PCI_LIB_ADDRESS(0, 0, 0, R_SA_TOLUD)) & ~(0x01)) - IgdMemSize - GttMemSize;\r
- PlatformVTdSample->Rmrr1.ReservedMemoryRegionLimitAddress = PlatformVTdSample->Rmrr1.ReservedMemoryRegionBaseAddress + IgdMemSize + GttMemSize - 1;\r
-\r
- ///\r
- /// Update DRHD structures of DmarTable\r
- ///\r
- MchBar = PciRead32 (PCI_LIB_ADDRESS(0, 0, 0, R_SA_MCHBAR)) & ~BIT0;\r
-\r
- if ((MmioRead32 (MchBar + R_SA_MCHBAR_VTD1_OFFSET) &~1) != 0) {\r
- PlatformVTdSample->Drhd1.RegisterBaseAddress = (MmioRead32 (MchBar + R_SA_MCHBAR_VTD1_OFFSET) &~1);\r
- } else {\r
- MmioWrite32 (MchBar + R_SA_MCHBAR_VTD1_OFFSET, (UINT32)PlatformVTdSample->Drhd1.RegisterBaseAddress | 1);\r
- }\r
- DEBUG ((DEBUG_INFO, "VTd1 - %x\n", (MmioRead32 (MchBar + R_SA_MCHBAR_VTD1_OFFSET))));\r
-\r
- if ((MmioRead32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET) &~1) != 0) {\r
- PlatformVTdSample->Drhd2.RegisterBaseAddress = (MmioRead32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET) &~1);\r
- } else {\r
- MmioWrite32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET, (UINT32)PlatformVTdSample->Drhd2.RegisterBaseAddress | 1);\r
- }\r
- DEBUG ((DEBUG_INFO, "VTd2 - %x\n", (MmioRead32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET))));\r
-\r
- return PlatformVTdInfoSampleDesc;\r
- } else {\r
- PlatformVTdNoIgdSample = AllocateCopyPool (sizeof(MY_VTD_INFO_NO_IGD_PPI), &mPlatformVTdNoIgdSample);\r
- ASSERT(PlatformVTdNoIgdSample != NULL);\r
- PlatformVTdNoIgdInfoSampleDesc = AllocateCopyPool (sizeof(EFI_PEI_PPI_DESCRIPTOR), &mPlatformVTdNoIgdInfoSampleDesc);\r
- ASSERT(PlatformVTdNoIgdInfoSampleDesc != NULL);\r
- PlatformVTdNoIgdInfoSampleDesc->Ppi = PlatformVTdNoIgdSample;\r
-\r
- ///\r
- /// Update DRHD structures of DmarTable\r
- ///\r
- MchBar = PciRead32 (PCI_LIB_ADDRESS(0, 0, 0, R_SA_MCHBAR)) & ~BIT0;\r
-\r
- if ((MmioRead32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET) &~1) != 0) {\r
- PlatformVTdNoIgdSample->Drhd2.RegisterBaseAddress = (MmioRead32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET) &~1);\r
- } else {\r
- MmioWrite32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET, (UINT32)PlatformVTdNoIgdSample->Drhd2.RegisterBaseAddress | 1);\r
- }\r
- DEBUG ((DEBUG_INFO, "VTd2 - %x\n", (MmioRead32 (MchBar + R_SA_MCHBAR_VTD2_OFFSET))));\r
-\r
- return PlatformVTdNoIgdInfoSampleDesc;\r
- }\r
-}\r
-\r
-/**\r
- The callback function for SiliconInitializedPpi.\r
- It reinstalls VTD_INFO_PPI.\r
-\r
- @param[in] PeiServices General purpose services available to every PEIM.\r
- @param[in] NotifyDescriptor Notify that this module published.\r
- @param[in] Ppi PPI that was installed.\r
-\r
- @retval EFI_SUCCESS The function completed successfully.\r
-**/\r
-EFI_STATUS\r
-EFIAPI\r
-SiliconInitializedPpiNotifyCallback (\r
- IN CONST EFI_PEI_SERVICES **PeiServices,\r
- IN EFI_PEI_NOTIFY_DESCRIPTOR *NotifyDescriptor,\r
- IN VOID *Ppi\r
- )\r
-{\r
- EFI_STATUS Status;\r
- EFI_PEI_PPI_DESCRIPTOR *PpiDesc;\r
-\r
- PpiDesc = PatchDmar ();\r
-\r
- Status = PeiServicesReInstallPpi (&mPlatformVTdNoIgdInfoSampleDesc, PpiDesc);\r
- ASSERT_EFI_ERROR (Status);\r
- return EFI_SUCCESS;\r
-}\r
-\r
-EFI_PEI_NOTIFY_DESCRIPTOR mSiliconInitializedNotifyList = {\r
- (EFI_PEI_PPI_DESCRIPTOR_NOTIFY_CALLBACK | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
- &gEdkiiSiliconInitializedPpiGuid,\r
- (EFI_PEIM_NOTIFY_ENTRY_POINT) SiliconInitializedPpiNotifyCallback\r
-};\r
-\r
-/**\r
- Platform VTd Info sample driver.\r
-\r
- @param[in] FileHandle Handle of the file being invoked.\r
- @param[in] PeiServices Describes the list of possible PEI Services.\r
-\r
- @retval EFI_SUCCESS if it completed successfully.\r
-**/\r
-EFI_STATUS\r
-EFIAPI\r
-PlatformVTdInfoSampleInitialize (\r
- IN EFI_PEI_FILE_HANDLE FileHandle,\r
- IN CONST EFI_PEI_SERVICES **PeiServices\r
- )\r
-{\r
- EFI_STATUS Status;\r
- BOOLEAN SiliconInitialized;\r
- VOID *SiliconInitializedPpi;\r
- EFI_PEI_PPI_DESCRIPTOR *PpiDesc;\r
-\r
- SiliconInitialized = FALSE;\r
- //\r
- // Check if silicon is initialized.\r
- //\r
- Status = PeiServicesLocatePpi (\r
- &gEdkiiSiliconInitializedPpiGuid,\r
- 0,\r
- NULL,\r
- &SiliconInitializedPpi\r
- );\r
- if (!EFI_ERROR(Status)) {\r
- SiliconInitialized = TRUE;\r
- }\r
- DEBUG ((DEBUG_INFO, "SiliconInitialized - %x\n", SiliconInitialized));\r
- if (!SiliconInitialized) {\r
- Status = PeiServicesNotifyPpi (&mSiliconInitializedNotifyList);\r
- InitDmar ();\r
-\r
- Status = PeiServicesInstallPpi (&mPlatformVTdNoIgdInfoSampleDesc);\r
- ASSERT_EFI_ERROR (Status);\r
- } else {\r
- PpiDesc = PatchDmar ();\r
-\r
- Status = PeiServicesInstallPpi (PpiDesc);\r
- ASSERT_EFI_ERROR (Status);\r
- }\r
-\r
- return Status;\r
-}\r