AhciPeiStorageSecurity.c

Go to the documentation of this file.

 
#include "AhciPei.h"
 
PEI_AHCI_ATA_DEVICE_DATA *
SearchTrustComputingDeviceByIndex (
  IN PEI_AHCI_CONTROLLER_PRIVATE_DATA  *Private,
  IN UINTN                             TrustComputingDeviceIndex
  )
{
  PEI_AHCI_ATA_DEVICE_DATA  *DeviceData;
  LIST_ENTRY                *Node;
 
  Node = GetFirstNode (&Private->DeviceList);
  while (!IsNull (&Private->DeviceList, Node)) {
    DeviceData = AHCI_PEI_ATA_DEVICE_INFO_FROM_THIS (Node);
 
    if (DeviceData->TrustComputingDeviceIndex == TrustComputingDeviceIndex) {
      return DeviceData;
    }
 
    Node = GetNextNode (&Private->DeviceList, Node);
  }
 
  return NULL;
}
 
EFI_STATUS
EFIAPI
AhciStorageSecurityGetDeviceNo (
  IN  EDKII_PEI_STORAGE_SECURITY_CMD_PPI  *This,
  OUT UINTN                               *NumberofDevices
  )
{
  PEI_AHCI_CONTROLLER_PRIVATE_DATA  *Private;
 
  if ((This == NULL) || (NumberofDevices == NULL)) {
    return EFI_INVALID_PARAMETER;
  }
 
  Private          = GET_AHCI_PEIM_HC_PRIVATE_DATA_FROM_THIS_STROAGE_SECURITY (This);
  *NumberofDevices = Private->TrustComputingDevices;
 
  return EFI_SUCCESS;
}
 
EFI_STATUS
EFIAPI
AhciStorageSecurityGetDevicePath (
  IN  EDKII_PEI_STORAGE_SECURITY_CMD_PPI  *This,
  IN  UINTN                               DeviceIndex,
  OUT UINTN                               *DevicePathLength,
  OUT EFI_DEVICE_PATH_PROTOCOL            **DevicePath
  )
{
  PEI_AHCI_CONTROLLER_PRIVATE_DATA  *Private;
  PEI_AHCI_ATA_DEVICE_DATA          *DeviceData;
  EFI_STATUS                        Status;
 
  if ((This == NULL) || (DevicePathLength == NULL) || (DevicePath == NULL)) {
    return EFI_INVALID_PARAMETER;
  }
 
  Private = GET_AHCI_PEIM_HC_PRIVATE_DATA_FROM_THIS_STROAGE_SECURITY (This);
  if ((DeviceIndex == 0) || (DeviceIndex > Private->TrustComputingDevices)) {
    return EFI_INVALID_PARAMETER;
  }
 
  DeviceData = SearchTrustComputingDeviceByIndex (Private, DeviceIndex);
  if (DeviceData == NULL) {
    return EFI_NOT_FOUND;
  }
 
  Status = AhciBuildDevicePath (
             Private,
             DeviceData->Port,
             DeviceData->PortMultiplier,
             DevicePathLength,
             DevicePath
             );
  if (EFI_ERROR (Status)) {
    return Status;
  }
 
  return EFI_SUCCESS;
}
 
EFI_STATUS
EFIAPI
AhciStorageSecurityReceiveData (
  IN  EDKII_PEI_STORAGE_SECURITY_CMD_PPI  *This,
  IN  UINTN                               DeviceIndex,
  IN  UINT64                              Timeout,
  IN  UINT8                               SecurityProtocolId,
  IN  UINT16                              SecurityProtocolSpecificData,
  IN  UINTN                               PayloadBufferSize,
  OUT VOID                                *PayloadBuffer,
  OUT UINTN                               *PayloadTransferSize
  )
{
  PEI_AHCI_CONTROLLER_PRIVATE_DATA  *Private;
  PEI_AHCI_ATA_DEVICE_DATA          *DeviceData;
 
  if ((PayloadBuffer == NULL) || (PayloadTransferSize == NULL) || (PayloadBufferSize == 0)) {
    return EFI_INVALID_PARAMETER;
  }
 
  Private = GET_AHCI_PEIM_HC_PRIVATE_DATA_FROM_THIS_STROAGE_SECURITY (This);
  if ((DeviceIndex == 0) || (DeviceIndex > Private->TrustComputingDevices)) {
    return EFI_INVALID_PARAMETER;
  }
 
  DeviceData = SearchTrustComputingDeviceByIndex (Private, DeviceIndex);
  if (DeviceData == NULL) {
    return EFI_NOT_FOUND;
  }
 
  ASSERT ((DeviceData->IdentifyData->trusted_computing_support & BIT0) != 0);
  if ((DeviceData->IdentifyData->trusted_computing_support & BIT0) == 0) {
    return EFI_UNSUPPORTED;
  }
 
  return TrustTransferAtaDevice (
           DeviceData,
           PayloadBuffer,
           SecurityProtocolId,
           SecurityProtocolSpecificData,
           PayloadBufferSize,
           FALSE,
           Timeout,
           PayloadTransferSize
           );
}
 
EFI_STATUS
EFIAPI
AhciStorageSecuritySendData (
  IN EDKII_PEI_STORAGE_SECURITY_CMD_PPI  *This,
  IN UINTN                               DeviceIndex,
  IN UINT64                              Timeout,
  IN UINT8                               SecurityProtocolId,
  IN UINT16                              SecurityProtocolSpecificData,
  IN UINTN                               PayloadBufferSize,
  IN VOID                                *PayloadBuffer
  )
{
  PEI_AHCI_CONTROLLER_PRIVATE_DATA  *Private;
  PEI_AHCI_ATA_DEVICE_DATA          *DeviceData;
 
  if ((PayloadBuffer == NULL) && (PayloadBufferSize != 0)) {
    return EFI_INVALID_PARAMETER;
  }
 
  Private = GET_AHCI_PEIM_HC_PRIVATE_DATA_FROM_THIS_STROAGE_SECURITY (This);
  if ((DeviceIndex == 0) || (DeviceIndex > Private->TrustComputingDevices)) {
    return EFI_INVALID_PARAMETER;
  }
 
  DeviceData = SearchTrustComputingDeviceByIndex (Private, DeviceIndex);
  if (DeviceData == NULL) {
    return EFI_NOT_FOUND;
  }
 
  ASSERT ((DeviceData->IdentifyData->trusted_computing_support & BIT0) != 0);
  if ((DeviceData->IdentifyData->trusted_computing_support & BIT0) == 0) {
    return EFI_UNSUPPORTED;
  }
 
  return TrustTransferAtaDevice (
           DeviceData,
           PayloadBuffer,
           SecurityProtocolId,
           SecurityProtocolSpecificData,
           PayloadBufferSize,
           TRUE,
           Timeout,
           NULL
           );
}