MdeModulePkg/SdMmcPciHcDxe: Add V3 64b DMA Support

[mirror_edk2.git] / MdeModulePkg / Bus / Pci / NvmExpressPei / NvmExpressPeiS3.c

/** @file\r
  The NvmExpressPei driver is used to manage non-volatile memory subsystem\r
  which follows NVM Express specification at PEI phase.\r
\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
\r
**/\r
\r
#include "NvmExpressPei.h"\r
\r
#include <Guid/S3StorageDeviceInitList.h>\r
\r
#include <Library/LockBoxLib.h>\r
\r
/**\r
  Determine if a specific NVM Express controller can be skipped for S3 phase.\r
\r
  @param[in]  HcDevicePath          Device path of the controller.\r
  @param[in]  HcDevicePathLength    Length of the device path specified by\r
                                    HcDevicePath.\r
\r
  @retval    The number of ports that need to be enumerated.\r
\r
**/\r
BOOLEAN\r
NvmeS3SkipThisController (\r
  IN  EFI_DEVICE_PATH_PROTOCOL    *HcDevicePath,\r
  IN  UINTN                       HcDevicePathLength\r
  )\r
{\r
  EFI_STATUS                  Status;\r
  UINT8                       DummyData;\r
  UINTN                       S3InitDevicesLength;\r
  EFI_DEVICE_PATH_PROTOCOL    *S3InitDevices;\r
  EFI_DEVICE_PATH_PROTOCOL    *DevicePathInst;\r
  UINTN                       DevicePathInstLength;\r
  BOOLEAN                     EntireEnd;\r
  BOOLEAN                     Skip;\r
\r
  //\r
  // From the LockBox, get the list of device paths for devices need to be\r
  // initialized in S3.\r
  //\r
  S3InitDevices       = NULL;\r
  S3InitDevicesLength = sizeof (DummyData);\r
  EntireEnd           = FALSE;\r
  Skip                = TRUE;\r
  Status = RestoreLockBox (&gS3StorageDeviceInitListGuid, &DummyData, &S3InitDevicesLength);\r
  if (Status != EFI_BUFFER_TOO_SMALL) {\r
    return Skip;\r
  } else {\r
    S3InitDevices = AllocatePool (S3InitDevicesLength);\r
    if (S3InitDevices == NULL) {\r
      return Skip;\r
    }\r
\r
    Status = RestoreLockBox (&gS3StorageDeviceInitListGuid, S3InitDevices, &S3InitDevicesLength);\r
    if (EFI_ERROR (Status)) {\r
      return Skip;\r
    }\r
  }\r
\r
  if (S3InitDevices == NULL) {\r
    return Skip;\r
  }\r
\r
  //\r
  // Only need to initialize the controllers that exist in the device list.\r
  //\r
  do {\r
    //\r
    // Fetch the size of current device path instance.\r
    //\r
    Status = GetDevicePathInstanceSize (\r
               S3InitDevices,\r
               &DevicePathInstLength,\r
               &EntireEnd\r
               );\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    DevicePathInst = S3InitDevices;\r
    S3InitDevices  = (EFI_DEVICE_PATH_PROTOCOL *)((UINTN) S3InitDevices + DevicePathInstLength);\r
\r
    if (HcDevicePathLength >= DevicePathInstLength) {\r
      continue;\r
    }\r
\r
    //\r
    // Compare the device paths to determine if the device is managed by this\r
    // controller.\r
    //\r
    if (CompareMem (\r
          DevicePathInst,\r
          HcDevicePath,\r
          HcDevicePathLength - sizeof (EFI_DEVICE_PATH_PROTOCOL)\r
          ) == 0) {\r
      Skip = FALSE;\r
      break;\r
    }\r
  } while (!EntireEnd);\r
\r
  return Skip;\r
}\r