UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm

[mirror_edk2.git] / MdeModulePkg / Bus / Usb / UsbBotPei / PeiAtapi.c

/** @file\r
Pei USB ATAPI command implementations.\r
\r
Copyright (c) 1999 - 2018, Intel Corporation. All rights reserved.<BR>\r
Copyright (C) 2022 Advanced Micro Devices, Inc. All rights reserved.<BR>\r
\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "UsbBotPeim.h"\r
#include "BotPeim.h"\r
\r
#define MAXSENSEKEY  5\r
\r
/**\r
  Sends out ATAPI Inquiry Packet Command to the specified device. This command will\r
  return INQUIRY data of the device.\r
\r
  @param PeiServices    The pointer of EFI_PEI_SERVICES.\r
  @param PeiBotDevice   The pointer to PEI_BOT_DEVICE instance.\r
\r
  @retval EFI_SUCCESS       Inquiry command completes successfully.\r
  @retval EFI_DEVICE_ERROR  Inquiry command failed.\r
\r
**/\r
EFI_STATUS\r
PeiUsbInquiry (\r
  IN  EFI_PEI_SERVICES  **PeiServices,\r
  IN  PEI_BOT_DEVICE    *PeiBotDevice\r
  )\r
{\r
  ATAPI_PACKET_COMMAND  Packet;\r
  EFI_STATUS            Status;\r
  ATAPI_INQUIRY_DATA    Idata;\r
\r
  //\r
  // fill command packet\r
  //\r
  ZeroMem (&Packet, sizeof (ATAPI_PACKET_COMMAND));\r
  ZeroMem (&Idata, sizeof (ATAPI_INQUIRY_DATA));\r
\r
  Packet.Inquiry.opcode            = ATA_CMD_INQUIRY;\r
  Packet.Inquiry.page_code         = 0;\r
  Packet.Inquiry.allocation_length = 36;\r
\r
  //\r
  // Send scsi INQUIRY command packet.\r
  // According to SCSI Primary Commands-2 spec, host only needs to\r
  // retrieve the first 36 bytes for standard INQUIRY data.\r
  //\r
  Status = PeiAtapiCommand (\r
             PeiServices,\r
             PeiBotDevice,\r
             &Packet,\r
             (UINT8)sizeof (ATAPI_PACKET_COMMAND),\r
             &Idata,\r
             36,\r
             EfiUsbDataIn,\r
             2000\r
             );\r
\r
  if (EFI_ERROR (Status)) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  if ((Idata.peripheral_type & 0x1f) == 0x05) {\r
    PeiBotDevice->DeviceType            = USBCDROM;\r
    PeiBotDevice->Media.BlockSize       = 0x800;\r
    PeiBotDevice->Media2.ReadOnly       = TRUE;\r
    PeiBotDevice->Media2.RemovableMedia = TRUE;\r
    PeiBotDevice->Media2.BlockSize      = 0x800;\r
  } else {\r
    PeiBotDevice->DeviceType            = USBFLOPPY;\r
    PeiBotDevice->Media.BlockSize       = 0x200;\r
    PeiBotDevice->Media2.ReadOnly       = FALSE;\r
    PeiBotDevice->Media2.RemovableMedia = TRUE;\r
    PeiBotDevice->Media2.BlockSize      = 0x200;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Sends out ATAPI Test Unit Ready Packet Command to the specified device\r
  to find out whether device is accessible.\r
\r
  @param PeiServices    The pointer of EFI_PEI_SERVICES.\r
  @param PeiBotDevice   The pointer to PEI_BOT_DEVICE instance.\r
\r
  @retval EFI_SUCCESS        TestUnit command executed successfully.\r
  @retval EFI_DEVICE_ERROR   Device cannot be executed TestUnit command successfully.\r
\r
**/\r
EFI_STATUS\r
PeiUsbTestUnitReady (\r
  IN  EFI_PEI_SERVICES  **PeiServices,\r
  IN  PEI_BOT_DEVICE    *PeiBotDevice\r
  )\r
{\r
  ATAPI_PACKET_COMMAND  Packet;\r
  EFI_STATUS            Status;\r
\r
  //\r
  // fill command packet\r
  //\r
  ZeroMem (&Packet, sizeof (ATAPI_PACKET_COMMAND));\r
  Packet.TestUnitReady.opcode = ATA_CMD_TEST_UNIT_READY;\r
\r
  //\r
  // send command packet\r
  //\r
  Status = PeiAtapiCommand (\r
             PeiServices,\r
             PeiBotDevice,\r
             &Packet,\r
             (UINT8)sizeof (ATAPI_PACKET_COMMAND),\r
             NULL,\r
             0,\r
             EfiUsbNoData,\r
             2000\r
             );\r
\r
  if (EFI_ERROR (Status)) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Sends out ATAPI Request Sense Packet Command to the specified device.\r
\r
  @param PeiServices    The pointer of EFI_PEI_SERVICES.\r
  @param PeiBotDevice   The pointer to PEI_BOT_DEVICE instance.\r
  @param SenseCounts    Length of sense buffer.\r
  @param SenseKeyBuffer Pointer to sense buffer.\r
\r
  @retval EFI_SUCCESS           Command executed successfully.\r
  @retval EFI_DEVICE_ERROR      Some device errors happen.\r
\r
**/\r
EFI_STATUS\r
PeiUsbRequestSense (\r
  IN  EFI_PEI_SERVICES  **PeiServices,\r
  IN  PEI_BOT_DEVICE    *PeiBotDevice,\r
  OUT UINTN             *SenseCounts,\r
  IN  UINT8             *SenseKeyBuffer\r
  )\r
{\r
  EFI_STATUS                Status;\r
  ATAPI_PACKET_COMMAND      Packet;\r
  UINT8                     *Ptr;\r
  BOOLEAN                   SenseReq;\r
  ATAPI_REQUEST_SENSE_DATA  *Sense;\r
\r
  *SenseCounts = 0;\r
\r
  //\r
  // fill command packet for Request Sense Packet Command\r
  //\r
  ZeroMem (&Packet, sizeof (ATAPI_PACKET_COMMAND));\r
  Packet.RequestSence.opcode            = ATA_CMD_REQUEST_SENSE;\r
  Packet.RequestSence.allocation_length = (UINT8)sizeof (ATAPI_REQUEST_SENSE_DATA);\r
\r
  Ptr = SenseKeyBuffer;\r
\r
  SenseReq = TRUE;\r
\r
  //\r
  //  request sense data from device continuously\r
  //  until no sense data exists in the device.\r
  //\r
  while (SenseReq) {\r
    Sense = (ATAPI_REQUEST_SENSE_DATA *)Ptr;\r
\r
    //\r
    // send out Request Sense Packet Command and get one Sense\r
    // data form device.\r
    //\r
    Status = PeiAtapiCommand (\r
               PeiServices,\r
               PeiBotDevice,\r
               &Packet,\r
               (UINT8)sizeof (ATAPI_PACKET_COMMAND),\r
               (VOID *)Ptr,\r
               sizeof (ATAPI_REQUEST_SENSE_DATA),\r
               EfiUsbDataIn,\r
               2000\r
               );\r
\r
    //\r
    // failed to get Sense data\r
    //\r
    if (EFI_ERROR (Status)) {\r
      if (*SenseCounts == 0) {\r
        return EFI_DEVICE_ERROR;\r
      } else {\r
        return EFI_SUCCESS;\r
      }\r
    }\r
\r
    if (Sense->sense_key != ATA_SK_NO_SENSE) {\r
      Ptr += sizeof (ATAPI_REQUEST_SENSE_DATA);\r
      //\r
      // Ptr is byte based pointer\r
      //\r
      (*SenseCounts)++;\r
\r
      if (*SenseCounts == MAXSENSEKEY) {\r
        break;\r
      }\r
    } else {\r
      //\r
      // when no sense key, skip out the loop\r
      //\r
      SenseReq = FALSE;\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Sends out ATAPI Read Capacity Packet Command to the specified device.\r
  This command will return the information regarding the capacity of the\r
  media in the device.\r
\r
  @param PeiServices    The pointer of EFI_PEI_SERVICES.\r
  @param PeiBotDevice   The pointer to PEI_BOT_DEVICE instance.\r
\r
  @retval EFI_SUCCESS           Command executed successfully.\r
  @retval EFI_DEVICE_ERROR      Some device errors happen.\r
\r
**/\r
EFI_STATUS\r
PeiUsbReadCapacity (\r
  IN  EFI_PEI_SERVICES  **PeiServices,\r
  IN  PEI_BOT_DEVICE    *PeiBotDevice\r
  )\r
{\r
  EFI_STATUS                Status;\r
  ATAPI_PACKET_COMMAND      Packet;\r
  ATAPI_READ_CAPACITY_DATA  Data;\r
  UINT32                    LastBlock;\r
\r
  ZeroMem (&Data, sizeof (ATAPI_READ_CAPACITY_DATA));\r
  ZeroMem (&Packet, sizeof (ATAPI_PACKET_COMMAND));\r
\r
  Packet.Inquiry.opcode = ATA_CMD_READ_CAPACITY;\r
\r
  //\r
  // send command packet\r
  //\r
  Status = PeiAtapiCommand (\r
             PeiServices,\r
             PeiBotDevice,\r
             &Packet,\r
             (UINT8)sizeof (ATAPI_PACKET_COMMAND),\r
             (VOID *)&Data,\r
             sizeof (ATAPI_READ_CAPACITY_DATA),\r
             EfiUsbDataIn,\r
             2000\r
             );\r
\r
  if (EFI_ERROR (Status)) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  LastBlock = ((UINT32)Data.LastLba3 << 24) | (Data.LastLba2 << 16) | (Data.LastLba1 << 8) | Data.LastLba0;\r
\r
  if (LastBlock == 0xFFFFFFFF) {\r
    DEBUG ((DEBUG_INFO, "The usb device LBA count is larger than 0xFFFFFFFF!\n"));\r
  }\r
\r
  PeiBotDevice->Media.LastBlock    = LastBlock;\r
  PeiBotDevice->Media.MediaPresent = TRUE;\r
\r
  PeiBotDevice->Media2.LastBlock    = LastBlock;\r
  PeiBotDevice->Media2.MediaPresent = TRUE;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Sends out ATAPI Read Format Capacity Data Command to the specified device.\r
  This command will return the information regarding the capacity of the\r
  media in the device.\r
\r
  @param PeiServices    The pointer of EFI_PEI_SERVICES.\r
  @param PeiBotDevice   The pointer to PEI_BOT_DEVICE instance.\r
\r
  @retval EFI_SUCCESS           Command executed successfully.\r
  @retval EFI_DEVICE_ERROR      Some device errors happen.\r
\r
**/\r
EFI_STATUS\r
PeiUsbReadFormattedCapacity (\r
  IN  EFI_PEI_SERVICES  **PeiServices,\r
  IN  PEI_BOT_DEVICE    *PeiBotDevice\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  ATAPI_PACKET_COMMAND             Packet;\r
  ATAPI_READ_FORMAT_CAPACITY_DATA  FormatData;\r
  UINT32                           LastBlock;\r
\r
  ZeroMem (&FormatData, sizeof (ATAPI_READ_FORMAT_CAPACITY_DATA));\r
  ZeroMem (&Packet, sizeof (ATAPI_PACKET_COMMAND));\r
\r
  Packet.ReadFormatCapacity.opcode               = ATA_CMD_READ_FORMAT_CAPACITY;\r
  Packet.ReadFormatCapacity.allocation_length_lo = 12;\r
\r
  //\r
  // send command packet\r
  //\r
  Status = PeiAtapiCommand (\r
             PeiServices,\r
             PeiBotDevice,\r
             &Packet,\r
             (UINT8)sizeof (ATAPI_PACKET_COMMAND),\r
             (VOID *)&FormatData,\r
             sizeof (ATAPI_READ_FORMAT_CAPACITY_DATA),\r
             EfiUsbDataIn,\r
             2000\r
             );\r
\r
  if (EFI_ERROR (Status)) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  if (FormatData.DesCode == 3) {\r
    //\r
    // Media is not present\r
    //\r
    PeiBotDevice->Media.MediaPresent  = FALSE;\r
    PeiBotDevice->Media.LastBlock     = 0;\r
    PeiBotDevice->Media2.MediaPresent = FALSE;\r
    PeiBotDevice->Media2.LastBlock    = 0;\r
  } else {\r
    LastBlock = ((UINT32)FormatData.LastLba3 << 24) | (FormatData.LastLba2 << 16) | (FormatData.LastLba1 << 8) | FormatData.LastLba0;\r
    if (LastBlock == 0xFFFFFFFF) {\r
      DEBUG ((DEBUG_INFO, "The usb device LBA count is larger than 0xFFFFFFFF!\n"));\r
    }\r
\r
    PeiBotDevice->Media.LastBlock = LastBlock;\r
\r
    PeiBotDevice->Media.LastBlock--;\r
\r
    PeiBotDevice->Media.MediaPresent = TRUE;\r
\r
    PeiBotDevice->Media2.MediaPresent = TRUE;\r
    PeiBotDevice->Media2.LastBlock    = PeiBotDevice->Media.LastBlock;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Execute Read(10) ATAPI command on a specific SCSI target.\r
\r
  Executes the ATAPI Read(10) command on the ATAPI target specified by PeiBotDevice.\r
\r
  @param PeiServices       The pointer of EFI_PEI_SERVICES.\r
  @param PeiBotDevice      The pointer to PEI_BOT_DEVICE instance.\r
  @param Buffer            The pointer to data buffer.\r
  @param Lba               The start logic block address of reading.\r
  @param NumberOfBlocks    The block number of reading.\r
\r
  @retval EFI_SUCCESS           Command executed successfully.\r
  @retval EFI_DEVICE_ERROR      Some device errors happen.\r
\r
**/\r
EFI_STATUS\r
PeiUsbRead10 (\r
  IN  EFI_PEI_SERVICES  **PeiServices,\r
  IN  PEI_BOT_DEVICE    *PeiBotDevice,\r
  IN  VOID              *Buffer,\r
  IN  EFI_PEI_LBA       Lba,\r
  IN  UINTN             NumberOfBlocks\r
  )\r
{\r
  ATAPI_PACKET_COMMAND  Packet;\r
  ATAPI_READ10_CMD      *Read10Packet;\r
  UINT16                MaxBlock;\r
  UINT32                BlocksRemaining;\r
  UINT32                SectorCount;\r
  UINT32                Lba32;\r
  UINT32                BlockSize;\r
  UINT32                ByteCount;\r
  VOID                  *PtrBuffer;\r
  EFI_STATUS            Status;\r
  UINT32                TimeOut;\r
\r
  //\r
  // prepare command packet for the Inquiry Packet Command.\r
  //\r
  ZeroMem (&Packet, sizeof (ATAPI_PACKET_COMMAND));\r
  Read10Packet = &Packet.Read10;\r
  Lba32        = (UINT32)Lba;\r
  PtrBuffer    = Buffer;\r
\r
  BlockSize = (UINT32)PeiBotDevice->Media.BlockSize;\r
\r
  MaxBlock = (UINT16)(MAX_UINT16 / BlockSize);\r
  ASSERT (NumberOfBlocks < MAX_UINT32);\r
  BlocksRemaining = (UINT32)NumberOfBlocks;\r
\r
  Status = EFI_SUCCESS;\r
  while (BlocksRemaining > 0) {\r
    SectorCount = MIN (BlocksRemaining, MaxBlock);\r
\r
    //\r
    // fill the Packet data structure\r
    //\r
    Read10Packet->opcode = ATA_CMD_READ_10;\r
\r
    //\r
    // Lba0 ~ Lba3 specify the start logical block address of the data transfer.\r
    // Lba0 is MSB, Lba3 is LSB\r
    //\r
    Read10Packet->Lba3 = (UINT8)(Lba32 & 0xff);\r
    Read10Packet->Lba2 = (UINT8)(Lba32 >> 8);\r
    Read10Packet->Lba1 = (UINT8)(Lba32 >> 16);\r
    Read10Packet->Lba0 = (UINT8)(Lba32 >> 24);\r
\r
    //\r
    // TranLen0 ~ TranLen1 specify the transfer length in block unit.\r
    // TranLen0 is MSB, TranLen is LSB\r
    //\r
    Read10Packet->TranLen1 = (UINT8)(SectorCount & 0xff);\r
    Read10Packet->TranLen0 = (UINT8)(SectorCount >> 8);\r
\r
    ByteCount = SectorCount * BlockSize;\r
\r
    TimeOut = SectorCount * 2000;\r
\r
    //\r
    // send command packet\r
    //\r
    Status = PeiAtapiCommand (\r
               PeiServices,\r
               PeiBotDevice,\r
               &Packet,\r
               (UINT8)sizeof (ATAPI_PACKET_COMMAND),\r
               (VOID *)PtrBuffer,\r
               ByteCount,\r
               EfiUsbDataIn,\r
               (UINT16)MIN (TimeOut, MAX_UINT16)\r
               );\r
\r
    if (Status != EFI_SUCCESS) {\r
      return Status;\r
    }\r
\r
    ASSERT (Lba32 <= (MAX_UINT32-SectorCount));\r
    Lba32          += SectorCount;\r
    PtrBuffer       = (UINT8 *)PtrBuffer + SectorCount * BlockSize;\r
    BlocksRemaining = BlocksRemaining - SectorCount;\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  Check if there is media according to sense data.\r
\r
  @param  SenseData   Pointer to sense data.\r
  @param  SenseCounts Count of sense data.\r
\r
  @retval TRUE    No media\r
  @retval FALSE   Media exists\r
\r
**/\r
BOOLEAN\r
IsNoMedia (\r
  IN  ATAPI_REQUEST_SENSE_DATA  *SenseData,\r
  IN  UINTN                     SenseCounts\r
  )\r
{\r
  ATAPI_REQUEST_SENSE_DATA  *SensePtr;\r
  UINTN                     Index;\r
  BOOLEAN                   NoMedia;\r
\r
  NoMedia  = FALSE;\r
  SensePtr = SenseData;\r
\r
  for (Index = 0; Index < SenseCounts; Index++) {\r
    switch (SensePtr->sense_key) {\r
      case ATA_SK_NOT_READY:\r
        switch (SensePtr->addnl_sense_code) {\r
          //\r
          // if no media, fill IdeDev parameter with specific info.\r
          //\r
          case ATA_ASC_NO_MEDIA:\r
            NoMedia = TRUE;\r
            break;\r
\r
          default:\r
            break;\r
        }\r
\r
        break;\r
\r
      default:\r
        break;\r
    }\r
\r
    SensePtr++;\r
  }\r
\r
  return NoMedia;\r
}\r
\r
/**\r
  Check if there is media error according to sense data.\r
\r
  @param  SenseData   Pointer to sense data.\r
  @param  SenseCounts Count of sense data.\r
\r
  @retval TRUE    Media error\r
  @retval FALSE   No media error\r
\r
**/\r
BOOLEAN\r
IsMediaError (\r
  IN  ATAPI_REQUEST_SENSE_DATA  *SenseData,\r
  IN  UINTN                     SenseCounts\r
  )\r
{\r
  ATAPI_REQUEST_SENSE_DATA  *SensePtr;\r
  UINTN                     Index;\r
  BOOLEAN                   Error;\r
\r
  SensePtr = SenseData;\r
  Error    = FALSE;\r
\r
  for (Index = 0; Index < SenseCounts; Index++) {\r
    switch (SensePtr->sense_key) {\r
      //\r
      // Medium error case\r
      //\r
      case ATA_SK_MEDIUM_ERROR:\r
        switch (SensePtr->addnl_sense_code) {\r
          case ATA_ASC_MEDIA_ERR1:\r
          //\r
          // fall through\r
          //\r
          case ATA_ASC_MEDIA_ERR2:\r
          //\r
          // fall through\r
          //\r
          case ATA_ASC_MEDIA_ERR3:\r
          //\r
          // fall through\r
          //\r
          case ATA_ASC_MEDIA_ERR4:\r
            Error = TRUE;\r
            break;\r
\r
          default:\r
            break;\r
        }\r
\r
        break;\r
\r
      //\r
      // Medium upside-down case\r
      //\r
      case ATA_SK_NOT_READY:\r
        switch (SensePtr->addnl_sense_code) {\r
          case ATA_ASC_MEDIA_UPSIDE_DOWN:\r
            Error = TRUE;\r
            break;\r
\r
          default:\r
            break;\r
        }\r
\r
        break;\r
\r
      default:\r
        break;\r
    }\r
\r
    SensePtr++;\r
  }\r
\r
  return Error;\r
}\r
\r
/**\r
  Check if media is changed according to sense data.\r
\r
  @param  SenseData   Pointer to sense data.\r
  @param  SenseCounts Count of sense data.\r
\r
  @retval TRUE    There is media change event.\r
  @retval FALSE   media is NOT changed.\r
\r
**/\r
BOOLEAN\r
IsMediaChange (\r
  IN  ATAPI_REQUEST_SENSE_DATA  *SenseData,\r
  IN  UINTN                     SenseCounts\r
  )\r
{\r
  ATAPI_REQUEST_SENSE_DATA  *SensePtr;\r
  UINTN                     Index;\r
  BOOLEAN                   MediaChange;\r
\r
  MediaChange = FALSE;\r
\r
  SensePtr = SenseData;\r
\r
  for (Index = 0; Index < SenseCounts; Index++) {\r
    //\r
    // catch media change sense key and addition sense data\r
    //\r
    switch (SensePtr->sense_key) {\r
      case ATA_SK_UNIT_ATTENTION:\r
        switch (SensePtr->addnl_sense_code) {\r
          case ATA_ASC_MEDIA_CHANGE:\r
            MediaChange = TRUE;\r
            break;\r
\r
          default:\r
            break;\r
        }\r
\r
        break;\r
\r
      default:\r
        break;\r
    }\r
\r
    SensePtr++;\r
  }\r
\r
  return MediaChange;\r
}\r