[mirror_edk2.git] / MdeModulePkg / Bus / Ata / AtaBusDxe / AtaPassThruExecute.c

/** @file\r
  This file implements ATA pass through transaction for ATA bus driver.\r
\r
  This file implements the low level execution of ATA pass through transaction.\r
  It transforms the high level identity, read/write, reset command to ATA pass\r
  through command and protocol. \r
    \r
  Copyright (c) 2009 Intel Corporation. <BR>\r
  All rights reserved. This program and the accompanying materials\r
  are licensed and made available under the terms and conditions of the BSD License\r
  which accompanies this distribution.  The 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
\r
#include "AtaBus.h"\r
\r
//\r
// Look up table (UdmaValid, IsWrite) for EFI_ATA_PASS_THRU_CMD_PROTOCOL\r
//\r
EFI_ATA_PASS_THRU_CMD_PROTOCOL mAtaPassThruCmdProtocols[][2] = {\r
  {\r
    EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_IN,\r
    EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_OUT\r
  },\r
  {\r
    EFI_ATA_PASS_THRU_PROTOCOL_UDMA_DATA_IN,\r
    EFI_ATA_PASS_THRU_PROTOCOL_UDMA_DATA_OUT,\r
  }\r
};\r
\r
//\r
// Look up table (UdmaValid, Lba48Bit, IsIsWrite) for ATA_CMD\r
//\r
UINT8 mAtaCommands[][2][2] = {\r
  {\r
    {\r
      ATA_CMD_READ_SECTORS,            // 28-bit LBA; PIO read\r
      ATA_CMD_WRITE_SECTORS            // 28-bit LBA; PIO write\r
    },\r
    {\r
      ATA_CMD_READ_SECTORS_EXT,        // 48-bit LBA; PIO read\r
      ATA_CMD_WRITE_SECTORS_EXT        // 48-bit LBA; PIO write\r
    }\r
  },\r
  {\r
    {\r
      ATA_CMD_READ_DMA,                // 28-bit LBA; DMA read\r
      ATA_CMD_WRITE_DMA                // 28-bit LBA; DMA write\r
    },\r
    {\r
      ATA_CMD_READ_DMA_EXT,            // 48-bit LBA; DMA read\r
      ATA_CMD_WRITE_DMA_EXT            // 48-bit LBA; DMA write\r
    }\r
  }\r
};\r
\r
//\r
// Look up table (Lba48Bit) for maximum transfer block number\r
//\r
UINTN mMaxTransferBlockNumber[] = {\r
  MAX_28BIT_TRANSFER_BLOCK_NUM,\r
  MAX_48BIT_TRANSFER_BLOCK_NUM\r
};\r
\r
\r
/**\r
  Wrapper for EFI_ATA_PASS_THRU_PROTOCOL.PassThru().\r
\r
  This function wraps the PassThru() invocation for ATA pass through function\r
  for an ATA device. It assembles the ATA pass through command packet for ATA\r
  transaction.\r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
\r
  @return The return status from EFI_ATA_PASS_THRU_PROTOCOL.PassThru().\r
\r
**/\r
EFI_STATUS\r
AtaDevicePassThru (\r
  IN OUT ATA_DEVICE                       *AtaDevice\r
  )\r
{\r
  EFI_STATUS                              Status;\r
  EFI_ATA_PASS_THRU_PROTOCOL              *AtaPassThru;\r
  EFI_ATA_PASS_THRU_COMMAND_PACKET        *Packet;\r
\r
  //\r
  // Assemble packet\r
  //\r
  Packet = &AtaDevice->Packet;\r
  Packet->Asb = AtaDevice->Asb;\r
  Packet->Acb = &AtaDevice->Acb;\r
  Packet->Timeout = ATA_TIMEOUT;\r
\r
  AtaPassThru = AtaDevice->AtaBusDriverData->AtaPassThru;\r
\r
  Status = AtaPassThru->PassThru (\r
                          AtaPassThru,\r
                          AtaDevice->Port,\r
                          AtaDevice->PortMultiplierPort,\r
                          Packet,\r
                          NULL\r
                          );\r
  //\r
  // Ensure ATA pass through caller and callee have the same\r
  // interpretation of ATA pass through protocol. \r
  //\r
  ASSERT (Status != EFI_INVALID_PARAMETER);\r
  ASSERT (Status != EFI_BAD_BUFFER_SIZE);\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  Wrapper for EFI_ATA_PASS_THRU_PROTOCOL.ResetDevice().\r
\r
  This function wraps the ResetDevice() invocation for ATA pass through function\r
  for an ATA device. \r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
\r
  @return The return status from EFI_ATA_PASS_THRU_PROTOCOL.PassThru().\r
\r
**/\r
EFI_STATUS\r
ResetAtaDevice (\r
  IN ATA_DEVICE                           *AtaDevice\r
  )\r
{\r
  EFI_ATA_PASS_THRU_PROTOCOL              *AtaPassThru;\r
  \r
  AtaPassThru = AtaDevice->AtaBusDriverData->AtaPassThru;\r
  \r
  return AtaPassThru->ResetDevice (\r
                        AtaPassThru,\r
                        AtaDevice->Port,\r
                        AtaDevice->PortMultiplierPort\r
                        );\r
}\r
\r
\r
/**\r
  Prints ATA model name to ATA device structure.\r
\r
  This function converts ATA device model name from ATA identify data \r
  to a string in ATA device structure. It needs to change the character\r
  order in the original model name string.\r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
\r
**/\r
VOID\r
PrintAtaModelName (\r
  IN OUT ATA_DEVICE  *AtaDevice\r
  )\r
{\r
  UINTN   Index;\r
  CHAR8   *Source;\r
  CHAR16  *Destination;\r
\r
  Source = AtaDevice->IdentifyData->AtaData.ModelName;\r
  Destination = AtaDevice->ModelName;\r
\r
  //\r
  // Swap the byte order in the original module name.\r
  //\r
  for (Index = 0; Index < MAX_MODEL_NAME_LEN; Index += 2) {\r
    Destination[Index]      = Source[Index + 1];\r
    Destination[Index + 1]  = Source[Index];\r
  }\r
  AtaDevice->ModelName[MAX_MODEL_NAME_LEN] = L'\0';\r
}\r
\r
\r
/**\r
  Gets ATA device Capacity according to ATA 6.\r
\r
  This function returns the capacity of the ATA device if it follows\r
  ATA 6 to support 48 bit addressing.\r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
\r
  @return The capacity of the ATA device or 0 if the device does not support\r
          48-bit addressing defined in ATA 6.\r
\r
**/\r
EFI_LBA\r
GetAtapi6Capacity (\r
  IN ATA_DEVICE                 *AtaDevice\r
  )\r
{\r
  EFI_LBA                       Capacity;\r
  EFI_LBA                       TmpLba;\r
  UINTN                         Index;\r
  ATAPI_IDENTIFY_DATA           *IdentifyData;\r
\r
  IdentifyData = (ATAPI_IDENTIFY_DATA *) AtaDevice->IdentifyData;\r
  if ((IdentifyData->cmd_set_support_83 & BIT10) == 0) {\r
    //\r
    // The device doesn't support 48 bit addressing\r
    //\r
    return 0;\r
  }\r
\r
  //\r
  // 48 bit address feature set is supported, get maximum capacity\r
  //\r
  Capacity = 0;\r
  for (Index = 0; Index < 4; Index++) {\r
    //\r
    // Lower byte goes first: word[100] is the lowest word, word[103] is highest\r
    //\r
    TmpLba = IdentifyData->max_user_lba_for_48bit_addr[Index];\r
    Capacity |= LShiftU64 (TmpLba, 16 * Index);\r
  }\r
\r
  return Capacity;\r
}\r
\r
\r
/**\r
  Identifies ATA device via the Identify data.\r
\r
  This function identifies the ATA device and initializes the Media information in \r
  Block IO protocol interface.\r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
\r
  @retval EFI_UNSUPPORTED   The device is not a valid ATA device (hard disk).\r
  @retval EFI_SUCCESS       The device is successfully identified and Media information\r
                            is correctly initialized.\r
\r
**/\r
EFI_STATUS\r
IdentifyAtaDevice (\r
  IN OUT ATA_DEVICE                 *AtaDevice\r
  )\r
{\r
  EFI_ATA_IDENTIFY_DATA             *IdentifyData;\r
  EFI_BLOCK_IO_MEDIA                *BlockMedia;\r
  EFI_LBA                           Capacity;\r
  UINT16                            PhyLogicSectorSupport;\r
  UINT16                            UdmaMode;\r
\r
  IdentifyData = &AtaDevice->IdentifyData->AtaData;\r
\r
  if ((IdentifyData->config & BIT15) != 0) {\r
    //\r
    // This is not an hard disk\r
    //\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  //\r
  // Check whether the WORD 88 (supported UltraDMA by drive) is valid\r
  //\r
  if ((IdentifyData->field_validity & BIT2) != 0) {\r
    UdmaMode = IdentifyData->ultra_dma_mode;\r
    if ((UdmaMode & (BIT0 | BIT1 | BIT2 | BIT3 | BIT4 | BIT5 | BIT6)) != 0) {\r
      //\r
      // If BIT0~BIT6 is selected, then UDMA is supported\r
      //\r
      AtaDevice->UdmaValid = TRUE;\r
    }\r
  }\r
\r
  Capacity = GetAtapi6Capacity (AtaDevice);\r
  if (Capacity > MAX_28BIT_ADDRESSING_CAPACITY) {\r
    //\r
    // Capacity exceeds 120GB. 48-bit addressing is really needed\r
    //\r
    AtaDevice->Lba48Bit = TRUE;\r
  } else {\r
    //\r
    // This is a hard disk <= 120GB capacity, treat it as normal hard disk\r
    //\r
    Capacity = ((UINT32)IdentifyData->user_addressable_sectors_hi << 16) | IdentifyData->user_addressable_sectors_lo;\r
    AtaDevice->Lba48Bit = FALSE;\r
  }\r
\r
  //\r
  // Block Media Information:\r
  //\r
  BlockMedia = &AtaDevice->BlockMedia;\r
  BlockMedia->LastBlock = Capacity - 1;\r
  //\r
  // Check whether Long Physical Sector Feature is supported\r
  //\r
  PhyLogicSectorSupport = IdentifyData->phy_logic_sector_support;\r
  if ((PhyLogicSectorSupport & (BIT14 | BIT15)) == BIT14) {\r
    //\r
    // Check whether one physical block contains multiple physical blocks\r
    //\r
    if ((PhyLogicSectorSupport & BIT13) != 0) {\r
      BlockMedia->LogicalBlocksPerPhysicalBlock = (UINT32) (1 << (PhyLogicSectorSupport & 0x000f));\r
      //\r
      // Check lowest alignment of logical blocks within physical block\r
      //\r
      if ((IdentifyData->alignment_logic_in_phy_blocks & (BIT14 | BIT15)) == BIT14) {\r
        BlockMedia->LowestAlignedLba = (EFI_LBA) (IdentifyData->alignment_logic_in_phy_blocks & 0x3fff);\r
      }\r
    }\r
    //\r
    // Check logical block size\r
    //\r
    if ((PhyLogicSectorSupport & BIT12) != 0) {\r
      BlockMedia->BlockSize = (UINT32) (((IdentifyData->logic_sector_size_hi << 16) | IdentifyData->logic_sector_size_lo) * sizeof (UINT16));\r
    }\r
    AtaDevice->BlockIo.Revision = EFI_BLOCK_IO_PROTOCOL_REVISION2;\r
  }\r
  //\r
  // Get ATA model name from identify data structure. \r
  //\r
  PrintAtaModelName (AtaDevice); \r
\r
  return EFI_SUCCESS;\r
}\r
\r
\r
/**\r
  Discovers whether it is a valid ATA device.\r
\r
  This function issues ATA_CMD_IDENTIFY_DRIVE command to the ATA device to identify it.\r
  If the command is executed successfully, it then identifies it and initializes\r
  the Media information in Block IO protocol interface.\r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
\r
  @retval EFI_SUCCESS       The device is successfully identified and Media information\r
                            is correctly initialized.\r
  @return others            Some error occurs when discovering the ATA device. \r
\r
**/\r
EFI_STATUS\r
DiscoverAtaDevice (\r
  IN OUT ATA_DEVICE                 *AtaDevice\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_ATA_COMMAND_BLOCK             *Acb;\r
  EFI_ATA_PASS_THRU_COMMAND_PACKET  *Packet;\r
  UINTN                             Retry;\r
\r
  //\r
  // Prepare for ATA command block.\r
  //\r
  Acb = ZeroMem (&AtaDevice->Acb, sizeof (*Acb));\r
  Acb->AtaCommand = ATA_CMD_IDENTIFY_DRIVE;\r
\r
  //\r
  // Prepare for ATA pass through packet.\r
  //\r
  Packet = ZeroMem (&AtaDevice->Packet, sizeof (*Packet));\r
  Packet->InDataBuffer = AtaDevice->IdentifyData;\r
  Packet->InTransferLength = sizeof (*AtaDevice->IdentifyData);\r
  Packet->Protocol = EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_IN;\r
  Packet->Length = EFI_ATA_PASS_THRU_LENGTH_BYTES | EFI_ATA_PASS_THRU_LENGTH_SECTOR_COUNT;\r
\r
  Retry = MAX_RETRY_TIMES;\r
  do {\r
    Status = AtaDevicePassThru (AtaDevice);\r
    if (!EFI_ERROR (Status)) {\r
      //\r
      // The command is issued successfully\r
      //\r
      Status = IdentifyAtaDevice (AtaDevice);\r
      if (!EFI_ERROR (Status)) {\r
        return Status;\r
      }\r
    }\r
  } while (Retry-- > 0);\r
\r
  return Status;\r
}\r
\r
/**\r
  Transfer data from ATA device.\r
\r
  This function performs one ATA pass through transaction to transfer data from/to\r
  ATA device. It chooses the appropriate ATA command and protocol to invoke PassThru\r
  interface of ATA pass through.\r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
  @param  Buffer            The pointer to the current transaction buffer.\r
  @param  StartLba          The starting logical block address to be accessed.\r
  @param  TransferLength    The block number or sector count of the transfer.\r
  @param  IsWrite           Indicates whether it is a write operation.\r
\r
  @retval EFI_SUCCESS       The data transfer is complete successfully.\r
  @return others            Some error occurs when transferring data. \r
\r
**/\r
EFI_STATUS\r
TransferAtaDevice (\r
  IN OUT ATA_DEVICE                 *AtaDevice,\r
  IN OUT VOID                       *Buffer,\r
  IN EFI_LBA                        StartLba,\r
  IN UINT32                         TransferLength,\r
  IN BOOLEAN                        IsWrite\r
  )\r
{\r
  EFI_ATA_COMMAND_BLOCK             *Acb;\r
  EFI_ATA_PASS_THRU_COMMAND_PACKET  *Packet;\r
\r
  //\r
  // Ensure AtaDevice->UdmaValid, AtaDevice->Lba48Bit and IsWrite are valid boolean values \r
  //\r
  ASSERT ((UINTN) AtaDevice->UdmaValid < 2);\r
  ASSERT ((UINTN) AtaDevice->Lba48Bit < 2);\r
  ASSERT ((UINTN) IsWrite < 2);\r
  //\r
  // Prepare for ATA command block.\r
  //\r
  Acb = ZeroMem (&AtaDevice->Acb, sizeof (*Acb));\r
  Acb->AtaCommand = mAtaCommands[AtaDevice->UdmaValid][AtaDevice->Lba48Bit][IsWrite];\r
  Acb->AtaSectorNumber = (UINT8) StartLba;\r
  Acb->AtaCylinderLow = (UINT8) RShiftU64 (StartLba, 8);\r
  Acb->AtaCylinderHigh = (UINT8) RShiftU64 (StartLba, 16);\r
  Acb->AtaDeviceHead = (UINT8) (BIT7 | BIT6 | BIT5 | (AtaDevice->PortMultiplierPort << 4)); \r
  Acb->AtaSectorCount = (UINT8) TransferLength;\r
  if (AtaDevice->Lba48Bit) {\r
    Acb->AtaSectorNumberExp = (UINT8) RShiftU64 (StartLba, 24);\r
    Acb->AtaCylinderLowExp = (UINT8) RShiftU64 (StartLba, 32);\r
    Acb->AtaCylinderHighExp = (UINT8) RShiftU64 (StartLba, 40);\r
    Acb->AtaSectorCountExp = (UINT8) (TransferLength >> 8);\r
  } else {\r
    Acb->AtaDeviceHead = (UINT8) (Acb->AtaDeviceHead | RShiftU64 (StartLba, 24));\r
  }\r
\r
  //\r
  // Prepare for ATA pass through packet.\r
  //\r
  Packet = ZeroMem (&AtaDevice->Packet, sizeof (*Packet));\r
  if (IsWrite) {\r
    Packet->OutDataBuffer = Buffer;\r
    Packet->OutTransferLength = TransferLength;\r
  } else {\r
    Packet->InDataBuffer = Buffer;\r
    Packet->InTransferLength = TransferLength;\r
  }\r
  Packet->Protocol = mAtaPassThruCmdProtocols[AtaDevice->UdmaValid][IsWrite];\r
  Packet->Length = EFI_ATA_PASS_THRU_LENGTH_SECTOR_COUNT;\r
\r
  return AtaDevicePassThru (AtaDevice); \r
}\r
\r
/**\r
  Read or write a number of blocks from ATA device.\r
\r
  This function performs ATA pass through transactions to read/write data from/to\r
  ATA device. It may separate the read/write request into several ATA pass through\r
  transactions.\r
\r
  @param  AtaDevice         The ATA child device involved for the operation.\r
  @param  Buffer            The pointer to the current transaction buffer.\r
  @param  StartLba          The starting logical block address to be accessed.\r
  @param  NumberOfBlocks    The block number or sector count of the transfer.\r
  @param  IsWrite           Indicates whether it is a write operation.\r
\r
  @retval EFI_SUCCESS       The data transfer is complete successfully.\r
  @return others            Some error occurs when transferring data. \r
\r
**/\r
EFI_STATUS \r
AccessAtaDevice(\r
  IN OUT ATA_DEVICE                 *AtaDevice,\r
  IN OUT UINT8                      *Buffer,\r
  IN EFI_LBA                        StartLba,\r
  IN UINTN                          NumberOfBlocks,\r
  IN BOOLEAN                        IsWrite\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  UINTN                             MaxTransferBlockNumber;\r
  UINTN                             TransferBlockNumber;\r
  UINTN                             BlockSize;\r
 \r
  //\r
  // Ensure AtaDevice->Lba48Bit is a valid boolean value \r
  //\r
  ASSERT ((UINTN) AtaDevice->Lba48Bit < 2);\r
  MaxTransferBlockNumber = mMaxTransferBlockNumber[AtaDevice->Lba48Bit];\r
  BlockSize = AtaDevice->BlockMedia.BlockSize;\r
  do {\r
    if (NumberOfBlocks > MaxTransferBlockNumber) {\r
      TransferBlockNumber = MaxTransferBlockNumber;\r
      NumberOfBlocks -= MaxTransferBlockNumber;\r
    } else  {\r
      TransferBlockNumber = NumberOfBlocks;\r
      NumberOfBlocks  = 0;\r
    }\r
\r
    Status = TransferAtaDevice (AtaDevice, Buffer, StartLba, (UINT32) TransferBlockNumber, IsWrite);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    StartLba += TransferBlockNumber;\r
    Buffer   += TransferBlockNumber * BlockSize;\r
  } while (NumberOfBlocks > 0);\r
\r
  return Status;\r
}\r
Commit	Line	Data
ad86a50a	1	/** @file\r
	2	This file implements ATA pass through transaction for ATA bus driver.\r
	3	\r
	4	This file implements the low level execution of ATA pass through transaction.\r
	5	It transforms the high level identity, read/write, reset command to ATA pass\r
	6	through command and protocol. \r
	7	\r
	8	Copyright (c) 2009 Intel Corporation. <BR>\r
	9	All rights reserved. This program and the accompanying materials\r
	10	are licensed and made available under the terms and conditions of the BSD License\r
	11	which accompanies this distribution. The full text of the license may be found at\r
	12	http://opensource.org/licenses/bsd-license.php\r
	13	\r
	14	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	15	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	16	\r
	17	\r
	18	**/\r
	19	\r
	20	#include "AtaBus.h"\r
	21	\r
	22	//\r
	23	// Look up table (UdmaValid, IsWrite) for EFI_ATA_PASS_THRU_CMD_PROTOCOL\r
	24	//\r
	25	EFI_ATA_PASS_THRU_CMD_PROTOCOL mAtaPassThruCmdProtocols[][2] = {\r
	26	{\r
	27	EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_IN,\r
	28	EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_OUT\r
	29	},\r
	30	{\r
	31	EFI_ATA_PASS_THRU_PROTOCOL_UDMA_DATA_IN,\r
	32	EFI_ATA_PASS_THRU_PROTOCOL_UDMA_DATA_OUT,\r
	33	}\r
	34	};\r
	35	\r
	36	//\r
	37	// Look up table (UdmaValid, Lba48Bit, IsIsWrite) for ATA_CMD\r
	38	//\r
	39	UINT8 mAtaCommands[][2][2] = {\r
	40	{\r
	41	{\r
	42	ATA_CMD_READ_SECTORS, // 28-bit LBA; PIO read\r
	43	ATA_CMD_WRITE_SECTORS // 28-bit LBA; PIO write\r
	44	},\r
	45	{\r
	46	ATA_CMD_READ_SECTORS_EXT, // 48-bit LBA; PIO read\r
	47	ATA_CMD_WRITE_SECTORS_EXT // 48-bit LBA; PIO write\r
	48	}\r
	49	},\r
	50	{\r
	51	{\r
	52	ATA_CMD_READ_DMA, // 28-bit LBA; DMA read\r
	53	ATA_CMD_WRITE_DMA // 28-bit LBA; DMA write\r
	54	},\r
	55	{\r
	56	ATA_CMD_READ_DMA_EXT, // 48-bit LBA; DMA read\r
	57	ATA_CMD_WRITE_DMA_EXT // 48-bit LBA; DMA write\r
	58	}\r
	59	}\r
	60	};\r
	61	\r
	62	//\r
	63	// Look up table (Lba48Bit) for maximum transfer block number\r
	64	//\r
65	UINTN mMaxTransferBlockNumber[] = {\r
66	MAX_28BIT_TRANSFER_BLOCK_NUM,\r
67	MAX_48BIT_TRANSFER_BLOCK_NUM\r
68	};\r
69	\r
70	\r
71	/**\r
72	Wrapper for EFI_ATA_PASS_THRU_PROTOCOL.PassThru().\r
73	\r
74	This function wraps the PassThru() invocation for ATA pass through function\r
75	for an ATA device. It assembles the ATA pass through command packet for ATA\r
76	transaction.\r
77	\r
78	@param AtaDevice The ATA child device involved for the operation.\r
79	\r
80	@return The return status from EFI_ATA_PASS_THRU_PROTOCOL.PassThru().\r
81	\r
82	**/\r
83	EFI_STATUS\r
84	AtaDevicePassThru (\r
85	IN OUT ATA_DEVICE *AtaDevice\r
86	)\r
87	{\r
88	EFI_STATUS Status;\r
89	EFI_ATA_PASS_THRU_PROTOCOL *AtaPassThru;\r
90	EFI_ATA_PASS_THRU_COMMAND_PACKET *Packet;\r
91	\r
92	//\r
93	// Assemble packet\r
94	//\r
95	Packet = &AtaDevice->Packet;\r
96	Packet->Asb = AtaDevice->Asb;\r
97	Packet->Acb = &AtaDevice->Acb;\r
98	Packet->Timeout = ATA_TIMEOUT;\r
99	\r
100	AtaPassThru = AtaDevice->AtaBusDriverData->AtaPassThru;\r
101	\r
102	Status = AtaPassThru->PassThru (\r
103	AtaPassThru,\r
104	AtaDevice->Port,\r
105	AtaDevice->PortMultiplierPort,\r
106	Packet,\r
107	NULL\r
108	);\r
109	//\r
110	// Ensure ATA pass through caller and callee have the same\r
111	// interpretation of ATA pass through protocol. \r
112	//\r
113	ASSERT (Status != EFI_INVALID_PARAMETER);\r
114	ASSERT (Status != EFI_BAD_BUFFER_SIZE);\r
115	\r
116	return Status;\r
117	}\r
118	\r
119	\r
120	/**\r
121	Wrapper for EFI_ATA_PASS_THRU_PROTOCOL.ResetDevice().\r
122	\r
123	This function wraps the ResetDevice() invocation for ATA pass through function\r
124	for an ATA device. \r
125	\r
126	@param AtaDevice The ATA child device involved for the operation.\r
127	\r
128	@return The return status from EFI_ATA_PASS_THRU_PROTOCOL.PassThru().\r
129	\r
130	**/\r
131	EFI_STATUS\r
132	ResetAtaDevice (\r
133	IN ATA_DEVICE *AtaDevice\r
134	)\r
135	{\r
136	EFI_ATA_PASS_THRU_PROTOCOL *AtaPassThru;\r
137	\r
138	AtaPassThru = AtaDevice->AtaBusDriverData->AtaPassThru;\r
139	\r
140	return AtaPassThru->ResetDevice (\r
141	AtaPassThru,\r
142	AtaDevice->Port,\r
143	AtaDevice->PortMultiplierPort\r
144	);\r
145	}\r
146	\r
147	\r
148	/**\r
149	Prints ATA model name to ATA device structure.\r
150	\r
151	This function converts ATA device model name from ATA identify data \r
152	to a string in ATA device structure. It needs to change the character\r
153	order in the original model name string.\r
154	\r
155	@param AtaDevice The ATA child device involved for the operation.\r
156	\r
157	**/\r
158	VOID\r
159	PrintAtaModelName (\r
160	IN OUT ATA_DEVICE *AtaDevice\r
161	)\r
162	{\r
163	UINTN Index;\r
164	CHAR8 *Source;\r
165	CHAR16 *Destination;\r
166	\r
167	Source = AtaDevice->IdentifyData->AtaData.ModelName;\r
168	Destination = AtaDevice->ModelName;\r
169	\r
170	//\r
171	// Swap the byte order in the original module name.\r
172	//\r
173	for (Index = 0; Index < MAX_MODEL_NAME_LEN; Index += 2) {\r
174	Destination[Index] = Source[Index + 1];\r
175	Destination[Index + 1] = Source[Index];\r
176	}\r
177	AtaDevice->ModelName[MAX_MODEL_NAME_LEN] = L'\0';\r
178	}\r
179	\r
180	\r
181	/**\r
182	Gets ATA device Capacity according to ATA 6.\r
183	\r
184	This function returns the capacity of the ATA device if it follows\r
185	ATA 6 to support 48 bit addressing.\r
186	\r
187	@param AtaDevice The ATA child device involved for the operation.\r
188	\r
189	@return The capacity of the ATA device or 0 if the device does not support\r
190	48-bit addressing defined in ATA 6.\r
191	\r
192	**/\r
193	EFI_LBA\r
194	GetAtapi6Capacity (\r
195	IN ATA_DEVICE *AtaDevice\r
196	)\r
197	{\r
198	EFI_LBA Capacity;\r
199	EFI_LBA TmpLba;\r
200	UINTN Index;\r
201	ATAPI_IDENTIFY_DATA *IdentifyData;\r
202	\r
203	IdentifyData = (ATAPI_IDENTIFY_DATA *) AtaDevice->IdentifyData;\r
204	if ((IdentifyData->cmd_set_support_83 & BIT10) == 0) {\r
205	//\r
206	// The device doesn't support 48 bit addressing\r
207	//\r
208	return 0;\r
209	}\r
210	\r
211	//\r
212	// 48 bit address feature set is supported, get maximum capacity\r
213	//\r
214	Capacity = 0;\r
215	for (Index = 0; Index < 4; Index++) {\r
216	//\r
217	// Lower byte goes first: word[100] is the lowest word, word[103] is highest\r
218	//\r
219	TmpLba = IdentifyData->max_user_lba_for_48bit_addr[Index];\r
220	Capacity \|= LShiftU64 (TmpLba, 16 * Index);\r
221	}\r
222	\r
223	return Capacity;\r
224	}\r
225	\r
226	\r
227	/**\r
228	Identifies ATA device via the Identify data.\r
229	\r
230	This function identifies the ATA device and initializes the Media information in \r
231	Block IO protocol interface.\r
232	\r
233	@param AtaDevice The ATA child device involved for the operation.\r
234	\r
235	@retval EFI_UNSUPPORTED The device is not a valid ATA device (hard disk).\r
236	@retval EFI_SUCCESS The device is successfully identified and Media information\r
237	is correctly initialized.\r
238	\r
239	**/\r
240	EFI_STATUS\r
241	IdentifyAtaDevice (\r
242	IN OUT ATA_DEVICE *AtaDevice\r
243	)\r
244	{\r
245	EFI_ATA_IDENTIFY_DATA *IdentifyData;\r
246	EFI_BLOCK_IO_MEDIA *BlockMedia;\r
247	EFI_LBA Capacity;\r
248	UINT16 PhyLogicSectorSupport;\r
249	UINT16 UdmaMode;\r
250	\r
251	IdentifyData = &AtaDevice->IdentifyData->AtaData;\r
252	\r
253	if ((IdentifyData->config & BIT15) != 0) {\r
254	//\r
255	// This is not an hard disk\r
256	//\r
257	return EFI_UNSUPPORTED;\r
258	}\r
259	\r
260	//\r
261	// Check whether the WORD 88 (supported UltraDMA by drive) is valid\r
262	//\r
263	if ((IdentifyData->field_validity & BIT2) != 0) {\r
264	UdmaMode = IdentifyData->ultra_dma_mode;\r
265	if ((UdmaMode & (BIT0 \| BIT1 \| BIT2 \| BIT3 \| BIT4 \| BIT5 \| BIT6)) != 0) {\r
266	//\r
267	// If BIT0~BIT6 is selected, then UDMA is supported\r
268	//\r
269	AtaDevice->UdmaValid = TRUE;\r
270	}\r
271	}\r
272	\r
273	Capacity = GetAtapi6Capacity (AtaDevice);\r
274	if (Capacity > MAX_28BIT_ADDRESSING_CAPACITY) {\r
275	//\r
276	// Capacity exceeds 120GB. 48-bit addressing is really needed\r
277	//\r
278	AtaDevice->Lba48Bit = TRUE;\r
279	} else {\r
280	//\r
281	// This is a hard disk <= 120GB capacity, treat it as normal hard disk\r
282	//\r
283	Capacity = ((UINT32)IdentifyData->user_addressable_sectors_hi << 16) \| IdentifyData->user_addressable_sectors_lo;\r
284	AtaDevice->Lba48Bit = FALSE;\r
285	}\r
286	\r
287	//\r
288	// Block Media Information:\r
289	//\r
290	BlockMedia = &AtaDevice->BlockMedia;\r
291	BlockMedia->LastBlock = Capacity - 1;\r
292	//\r
293	// Check whether Long Physical Sector Feature is supported\r
294	//\r
295	PhyLogicSectorSupport = IdentifyData->phy_logic_sector_support;\r
296	if ((PhyLogicSectorSupport & (BIT14 \| BIT15)) == BIT14) {\r
297	//\r
298	// Check whether one physical block contains multiple physical blocks\r
299	//\r
300	if ((PhyLogicSectorSupport & BIT13) != 0) {\r
301	BlockMedia->LogicalBlocksPerPhysicalBlock = (UINT32) (1 << (PhyLogicSectorSupport & 0x000f));\r
302	//\r
303	// Check lowest alignment of logical blocks within physical block\r
304	//\r
305	if ((IdentifyData->alignment_logic_in_phy_blocks & (BIT14 \| BIT15)) == BIT14) {\r
306	BlockMedia->LowestAlignedLba = (EFI_LBA) (IdentifyData->alignment_logic_in_phy_blocks & 0x3fff);\r
307	}\r
308	}\r
309	//\r
310	// Check logical block size\r
311	//\r
312	if ((PhyLogicSectorSupport & BIT12) != 0) {\r
313	BlockMedia->BlockSize = (UINT32) (((IdentifyData->logic_sector_size_hi << 16) \| IdentifyData->logic_sector_size_lo) * sizeof (UINT16));\r
314	}\r
315	AtaDevice->BlockIo.Revision = EFI_BLOCK_IO_PROTOCOL_REVISION2;\r
316	}\r
317	//\r
318	// Get ATA model name from identify data structure. \r
319	//\r
320	PrintAtaModelName (AtaDevice); \r
321	\r
322	return EFI_SUCCESS;\r
323	}\r
324	\r
325	\r
326	/**\r
327	Discovers whether it is a valid ATA device.\r
328	\r
329	This function issues ATA_CMD_IDENTIFY_DRIVE command to the ATA device to identify it.\r
330	If the command is executed successfully, it then identifies it and initializes\r
331	the Media information in Block IO protocol interface.\r
332	\r
333	@param AtaDevice The ATA child device involved for the operation.\r
334	\r
335	@retval EFI_SUCCESS The device is successfully identified and Media information\r
336	is correctly initialized.\r
337	@return others Some error occurs when discovering the ATA device. \r
338	\r
339	**/\r
340	EFI_STATUS\r
341	DiscoverAtaDevice (\r
342	IN OUT ATA_DEVICE *AtaDevice\r
343	)\r
344	{\r
345	EFI_STATUS Status;\r
346	EFI_ATA_COMMAND_BLOCK *Acb;\r
347	EFI_ATA_PASS_THRU_COMMAND_PACKET *Packet;\r
348	UINTN Retry;\r
349	\r
350	//\r
351	// Prepare for ATA command block.\r
352	//\r
353	Acb = ZeroMem (&AtaDevice->Acb, sizeof (*Acb));\r
354	Acb->AtaCommand = ATA_CMD_IDENTIFY_DRIVE;\r
355	\r
356	//\r
357	// Prepare for ATA pass through packet.\r
358	//\r
359	Packet = ZeroMem (&AtaDevice->Packet, sizeof (*Packet));\r
360	Packet->InDataBuffer = AtaDevice->IdentifyData;\r
361	Packet->InTransferLength = sizeof (*AtaDevice->IdentifyData);\r
362	Packet->Protocol = EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_IN;\r
363	Packet->Length = EFI_ATA_PASS_THRU_LENGTH_BYTES \| EFI_ATA_PASS_THRU_LENGTH_SECTOR_COUNT;\r
364	\r
365	Retry = MAX_RETRY_TIMES;\r
366	do {\r
367	Status = AtaDevicePassThru (AtaDevice);\r
368	if (!EFI_ERROR (Status)) {\r
369	//\r
370	// The command is issued successfully\r
371	//\r
372	Status = IdentifyAtaDevice (AtaDevice);\r
373	if (!EFI_ERROR (Status)) {\r
374	return Status;\r
375	}\r
376	}\r
377	} while (Retry-- > 0);\r
378	\r
379	return Status;\r
380	}\r
381	\r
382	/**\r
383	Transfer data from ATA device.\r
384	\r
385	This function performs one ATA pass through transaction to transfer data from/to\r
386	ATA device. It chooses the appropriate ATA command and protocol to invoke PassThru\r
387	interface of ATA pass through.\r
388	\r
389	@param AtaDevice The ATA child device involved for the operation.\r
390	@param Buffer The pointer to the current transaction buffer.\r
391	@param StartLba The starting logical block address to be accessed.\r
392	@param TransferLength The block number or sector count of the transfer.\r
393	@param IsWrite Indicates whether it is a write operation.\r
394	\r
395	@retval EFI_SUCCESS The data transfer is complete successfully.\r
396	@return others Some error occurs when transferring data. \r
397	\r
398	**/\r
399	EFI_STATUS\r
400	TransferAtaDevice (\r
401	IN OUT ATA_DEVICE *AtaDevice,\r
402	IN OUT VOID *Buffer,\r
403	IN EFI_LBA StartLba,\r
404	IN UINT32 TransferLength,\r
405	IN BOOLEAN IsWrite\r
406	)\r
407	{\r
408	EFI_ATA_COMMAND_BLOCK *Acb;\r
409	EFI_ATA_PASS_THRU_COMMAND_PACKET *Packet;\r
410	\r
423401f9	411	//\r
61d4f8f9	412	// Ensure AtaDevice->UdmaValid, AtaDevice->Lba48Bit and IsWrite are valid boolean values \r
423401f9	413	//\r
61d4f8f9	414	ASSERT ((UINTN) AtaDevice->UdmaValid < 2);\r
423401f9	415	ASSERT ((UINTN) AtaDevice->Lba48Bit < 2);\r
423401f9	416	ASSERT ((UINTN) IsWrite < 2);\r
ad86a50a	417	//\r
	418	// Prepare for ATA command block.\r
	419	//\r
	420	Acb = ZeroMem (&AtaDevice->Acb, sizeof (*Acb));\r
	421	Acb->AtaCommand = mAtaCommands[AtaDevice->UdmaValid][AtaDevice->Lba48Bit][IsWrite];\r
	422	Acb->AtaSectorNumber = (UINT8) StartLba;\r
	423	Acb->AtaCylinderLow = (UINT8) RShiftU64 (StartLba, 8);\r
	424	Acb->AtaCylinderHigh = (UINT8) RShiftU64 (StartLba, 16);\r
	425	Acb->AtaDeviceHead = (UINT8) (BIT7 \| BIT6 \| BIT5 \| (AtaDevice->PortMultiplierPort << 4)); \r
423401f9	426	Acb->AtaSectorCount = (UINT8) TransferLength;\r
ad86a50a	427	if (AtaDevice->Lba48Bit) {\r
ad86a50a	428	Acb->AtaSectorNumberExp = (UINT8) RShiftU64 (StartLba, 24);\r
423401f9	429	Acb->AtaCylinderLowExp = (UINT8) RShiftU64 (StartLba, 32);\r
	430	Acb->AtaCylinderHighExp = (UINT8) RShiftU64 (StartLba, 40);\r
	431	Acb->AtaSectorCountExp = (UINT8) (TransferLength >> 8);\r
ad86a50a	432	} else {\r
	433	Acb->AtaDeviceHead = (UINT8) (Acb->AtaDeviceHead \| RShiftU64 (StartLba, 24));\r
	434	}\r
ad86a50a	435	\r
	436	//\r
	437	// Prepare for ATA pass through packet.\r
	438	//\r
	439	Packet = ZeroMem (&AtaDevice->Packet, sizeof (*Packet));\r
	440	if (IsWrite) {\r
	441	Packet->OutDataBuffer = Buffer;\r
	442	Packet->OutTransferLength = TransferLength;\r
	443	} else {\r
	444	Packet->InDataBuffer = Buffer;\r
	445	Packet->InTransferLength = TransferLength;\r
	446	}\r
	447	Packet->Protocol = mAtaPassThruCmdProtocols[AtaDevice->UdmaValid][IsWrite];\r
	448	Packet->Length = EFI_ATA_PASS_THRU_LENGTH_SECTOR_COUNT;\r
	449	\r
	450	return AtaDevicePassThru (AtaDevice); \r
	451	}\r
	452	\r
	453	/**\r
	454	Read or write a number of blocks from ATA device.\r
	455	\r
	456	This function performs ATA pass through transactions to read/write data from/to\r
	457	ATA device. It may separate the read/write request into several ATA pass through\r
	458	transactions.\r
	459	\r
	460	@param AtaDevice The ATA child device involved for the operation.\r
	461	@param Buffer The pointer to the current transaction buffer.\r
	462	@param StartLba The starting logical block address to be accessed.\r
	463	@param NumberOfBlocks The block number or sector count of the transfer.\r
	464	@param IsWrite Indicates whether it is a write operation.\r
	465	\r
	466	@retval EFI_SUCCESS The data transfer is complete successfully.\r
	467	@return others Some error occurs when transferring data. \r
	468	\r
	469	**/\r
	470	EFI_STATUS \r
	471	AccessAtaDevice(\r
	472	IN OUT ATA_DEVICE *AtaDevice,\r
	473	IN OUT UINT8 *Buffer,\r
	474	IN EFI_LBA StartLba,\r
	475	IN UINTN NumberOfBlocks,\r
	476	IN BOOLEAN IsWrite\r
	477	)\r
	478	{\r
	479	EFI_STATUS Status;\r
	480	UINTN MaxTransferBlockNumber;\r
	481	UINTN TransferBlockNumber;\r
	482	UINTN BlockSize;\r
	483	\r
423401f9	484	//\r
	485	// Ensure AtaDevice->Lba48Bit is a valid boolean value \r
	486	//\r
	487	ASSERT ((UINTN) AtaDevice->Lba48Bit < 2);\r
ad86a50a	488	MaxTransferBlockNumber = mMaxTransferBlockNumber[AtaDevice->Lba48Bit];\r
	489	BlockSize = AtaDevice->BlockMedia.BlockSize;\r
	490	do {\r
	491	if (NumberOfBlocks > MaxTransferBlockNumber) {\r
	492	TransferBlockNumber = MaxTransferBlockNumber;\r
	493	NumberOfBlocks -= MaxTransferBlockNumber;\r
	494	} else {\r
	495	TransferBlockNumber = NumberOfBlocks;\r
	496	NumberOfBlocks = 0;\r
	497	}\r
	498	\r
	499	Status = TransferAtaDevice (AtaDevice, Buffer, StartLba, (UINT32) TransferBlockNumber, IsWrite);\r
	500	if (EFI_ERROR (Status)) {\r
	501	return Status;\r
	502	}\r
	503	StartLba += TransferBlockNumber;\r
	504	Buffer += TransferBlockNumber * BlockSize;\r
	505	} while (NumberOfBlocks > 0);\r
	506	\r
	507	return Status;\r
	508	}\r