MdeModulePkg/AhciPei: Add PEI BlockIO support

[mirror_edk2.git] / MdeModulePkg / Bus / Ata / AhciPei / AhciPei.c

diff --git a/MdeModulePkg/Bus/Ata/AhciPei/AhciPei.c b/MdeModulePkg/Bus/Ata/AhciPei/AhciPei.c
index 29e0aa7d65ad8d849459f0c2a2563986ad22d319..31b072c118785b78481401091c156e30e0afcf53 100644 (file)
--- a/MdeModulePkg/Bus/Ata/AhciPei/AhciPei.c
+++ b/MdeModulePkg/Bus/Ata/AhciPei/AhciPei.c
@@ -16,6 +16,18 @@ EFI_PEI_PPI_DESCRIPTOR  mAhciAtaPassThruPpiListTemplate = {
   NULL\r
 };\r
 \r
+EFI_PEI_PPI_DESCRIPTOR  mAhciBlkIoPpiListTemplate = {\r
+  (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
+  &gEfiPeiVirtualBlockIoPpiGuid,\r
+  NULL\r
+};\r
+\r
+EFI_PEI_PPI_DESCRIPTOR  mAhciBlkIo2PpiListTemplate = {\r
+  (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
+  &gEfiPeiVirtualBlockIo2PpiGuid,\r
+  NULL\r
+};\r
+\r
 EFI_PEI_PPI_DESCRIPTOR  mAhciStorageSecurityPpiListTemplate = {\r
   (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
   &gEdkiiPeiStorageSecurityCommandPpiGuid,\r
@@ -265,6 +277,29 @@ AtaAhciPeimEntry (
     Private->AtaPassThruPpiList.Ppi       = &Private->AtaPassThruPpi;\r
     PeiServicesInstallPpi (&Private->AtaPassThruPpiList);\r
 \r
+    Private->BlkIoPpi.GetNumberOfBlockDevices  = AhciBlockIoGetDeviceNo;\r
+    Private->BlkIoPpi.GetBlockDeviceMediaInfo  = AhciBlockIoGetMediaInfo;\r
+    Private->BlkIoPpi.ReadBlocks               = AhciBlockIoReadBlocks;\r
+    CopyMem (\r
+      &Private->BlkIoPpiList,\r
+      &mAhciBlkIoPpiListTemplate,\r
+      sizeof (EFI_PEI_PPI_DESCRIPTOR)\r
+      );\r
+    Private->BlkIoPpiList.Ppi                  = &Private->BlkIoPpi;\r
+    PeiServicesInstallPpi (&Private->BlkIoPpiList);\r
+\r
+    Private->BlkIo2Ppi.Revision                = EFI_PEI_RECOVERY_BLOCK_IO2_PPI_REVISION;\r
+    Private->BlkIo2Ppi.GetNumberOfBlockDevices = AhciBlockIoGetDeviceNo2;\r
+    Private->BlkIo2Ppi.GetBlockDeviceMediaInfo = AhciBlockIoGetMediaInfo2;\r
+    Private->BlkIo2Ppi.ReadBlocks              = AhciBlockIoReadBlocks2;\r
+    CopyMem (\r
+      &Private->BlkIo2PpiList,\r
+      &mAhciBlkIo2PpiListTemplate,\r
+      sizeof (EFI_PEI_PPI_DESCRIPTOR)\r
+      );\r
+    Private->BlkIo2PpiList.Ppi                 = &Private->BlkIo2Ppi;\r
+    PeiServicesInstallPpi (&Private->BlkIo2PpiList);\r
+\r
     if (Private->TrustComputingDevices != 0) {\r
       DEBUG ((\r
         DEBUG_INFO,\r