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 6df3d2c8d190618674b690bb92c799aeb0dd4e99..31b072c118785b78481401091c156e30e0afcf53 100644 (file)
--- a/MdeModulePkg/Bus/Ata/AhciPei/AhciPei.c
+++ b/MdeModulePkg/Bus/Ata/AhciPei/AhciPei.c
@@ -4,14 +4,7 @@
 \r
   Copyright (c) 2019, Intel Corporation. All rights reserved.<BR>\r
 \r
-  This program and the accompanying materials\r
-  are licensed and made available under the terms and conditions\r
-  of the BSD License which accompanies this distribution.  The\r
-  full text of the license may be found at\r
-  http://opensource.org/licenses/bsd-license.php\r
-\r
-  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+  SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
@@ -23,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
@@ -272,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