2 This file implements ATA pass through transaction for ATA bus driver.
4 This file implements the low level execution of ATA pass through transaction.
5 It transforms the high level identity, read/write, reset command to ATA pass
6 through command and protocol.
8 NOTE: This file also implements the StorageSecurityCommandProtocol(SSP). For input
9 parameter SecurityProtocolSpecificData, ATA spec has no explicitly definition
10 for Security Protocol Specific layout. This implementation uses big endian for
13 Copyright (c) 2009 - 2012, Intel Corporation. All rights reserved.<BR>
14 This program and the accompanying materials
15 are licensed and made available under the terms and conditions of the BSD License
16 which accompanies this distribution. The full text of the license may be found at
17 http://opensource.org/licenses/bsd-license.php
19 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
20 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
27 #define ATA_CMD_TRUST_NON_DATA 0x5B
28 #define ATA_CMD_TRUST_RECEIVE 0x5C
29 #define ATA_CMD_TRUST_RECEIVE_DMA 0x5D
30 #define ATA_CMD_TRUST_SEND 0x5E
31 #define ATA_CMD_TRUST_SEND_DMA 0x5F
34 // Look up table (UdmaValid, IsWrite) for EFI_ATA_PASS_THRU_CMD_PROTOCOL
36 EFI_ATA_PASS_THRU_CMD_PROTOCOL mAtaPassThruCmdProtocols
[][2] = {
38 EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_IN
,
39 EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_OUT
42 EFI_ATA_PASS_THRU_PROTOCOL_UDMA_DATA_IN
,
43 EFI_ATA_PASS_THRU_PROTOCOL_UDMA_DATA_OUT
,
48 // Look up table (UdmaValid, Lba48Bit, IsIsWrite) for ATA_CMD
50 UINT8 mAtaCommands
[][2][2] = {
53 ATA_CMD_READ_SECTORS
, // 28-bit LBA; PIO read
54 ATA_CMD_WRITE_SECTORS
// 28-bit LBA; PIO write
57 ATA_CMD_READ_SECTORS_EXT
, // 48-bit LBA; PIO read
58 ATA_CMD_WRITE_SECTORS_EXT
// 48-bit LBA; PIO write
63 ATA_CMD_READ_DMA
, // 28-bit LBA; DMA read
64 ATA_CMD_WRITE_DMA
// 28-bit LBA; DMA write
67 ATA_CMD_READ_DMA_EXT
, // 48-bit LBA; DMA read
68 ATA_CMD_WRITE_DMA_EXT
// 48-bit LBA; DMA write
74 // Look up table (UdmaValid, IsTrustSend) for ATA_CMD
76 UINT8 mAtaTrustCommands
[2][2] = {
78 ATA_CMD_TRUST_RECEIVE
, // PIO read
79 ATA_CMD_TRUST_SEND
// PIO write
82 ATA_CMD_TRUST_RECEIVE_DMA
, // DMA read
83 ATA_CMD_TRUST_SEND_DMA
// DMA write
89 // Look up table (Lba48Bit) for maximum transfer block number
91 UINTN mMaxTransferBlockNumber
[] = {
92 MAX_28BIT_TRANSFER_BLOCK_NUM
,
93 MAX_48BIT_TRANSFER_BLOCK_NUM
98 Wrapper for EFI_ATA_PASS_THRU_PROTOCOL.PassThru().
100 This function wraps the PassThru() invocation for ATA pass through function
101 for an ATA device. It assembles the ATA pass through command packet for ATA
104 @param[in, out] AtaDevice The ATA child device involved for the operation.
105 @param[in, out] TaskPacket Pointer to a Pass Thru Command Packet. Optional,
106 if it is NULL, blocking mode, and use the packet
107 in AtaDevice. If it is not NULL, non blocking mode,
108 and pass down this Packet.
109 @param[in, out] Event If Event is NULL, then blocking I/O is performed.
110 If Event is not NULL and non-blocking I/O is
111 supported,then non-blocking I/O is performed,
112 and Event will be signaled when the write
113 request is completed.
115 @return The return status from EFI_ATA_PASS_THRU_PROTOCOL.PassThru().
120 IN OUT ATA_DEVICE
*AtaDevice
,
121 IN OUT EFI_ATA_PASS_THRU_COMMAND_PACKET
*TaskPacket
, OPTIONAL
122 IN OUT EFI_EVENT Event OPTIONAL
126 EFI_ATA_PASS_THRU_PROTOCOL
*AtaPassThru
;
127 EFI_ATA_PASS_THRU_COMMAND_PACKET
*Packet
;
130 // Assemble packet. If it is non blocking mode, the Ata driver should keep each
131 // subtask and clean them when the event is signaled.
133 if (TaskPacket
!= NULL
) {
135 Packet
->Asb
= AllocateAlignedBuffer (AtaDevice
, sizeof (EFI_ATA_STATUS_BLOCK
));
136 if (Packet
->Asb
== NULL
) {
137 return EFI_OUT_OF_RESOURCES
;
140 CopyMem (Packet
->Asb
, AtaDevice
->Asb
, sizeof (EFI_ATA_STATUS_BLOCK
));
141 Packet
->Acb
= AllocateCopyPool (sizeof (EFI_ATA_COMMAND_BLOCK
), &AtaDevice
->Acb
);
143 Packet
= &AtaDevice
->Packet
;
144 Packet
->Asb
= AtaDevice
->Asb
;
145 Packet
->Acb
= &AtaDevice
->Acb
;
148 AtaPassThru
= AtaDevice
->AtaBusDriverData
->AtaPassThru
;
150 Status
= AtaPassThru
->PassThru (
153 AtaDevice
->PortMultiplierPort
,
158 // Ensure ATA pass through caller and callee have the same
159 // interpretation of ATA pass through protocol.
161 ASSERT (Status
!= EFI_INVALID_PARAMETER
);
162 ASSERT (Status
!= EFI_BAD_BUFFER_SIZE
);
169 Wrapper for EFI_ATA_PASS_THRU_PROTOCOL.ResetDevice().
171 This function wraps the ResetDevice() invocation for ATA pass through function
174 @param AtaDevice The ATA child device involved for the operation.
176 @return The return status from EFI_ATA_PASS_THRU_PROTOCOL.PassThru().
181 IN ATA_DEVICE
*AtaDevice
184 EFI_ATA_PASS_THRU_PROTOCOL
*AtaPassThru
;
186 AtaPassThru
= AtaDevice
->AtaBusDriverData
->AtaPassThru
;
189 // Report Status Code to indicate reset happens
191 REPORT_STATUS_CODE_WITH_DEVICE_PATH (
193 (EFI_IO_BUS_ATA_ATAPI
| EFI_IOB_PC_RESET
),
194 AtaDevice
->AtaBusDriverData
->ParentDevicePath
197 return AtaPassThru
->ResetDevice (
200 AtaDevice
->PortMultiplierPort
206 Prints ATA model name to ATA device structure.
208 This function converts ATA device model name from ATA identify data
209 to a string in ATA device structure. It needs to change the character
210 order in the original model name string.
212 @param AtaDevice The ATA child device involved for the operation.
217 IN OUT ATA_DEVICE
*AtaDevice
224 Source
= AtaDevice
->IdentifyData
->ModelName
;
225 Destination
= AtaDevice
->ModelName
;
228 // Swap the byte order in the original module name.
230 for (Index
= 0; Index
< MAX_MODEL_NAME_LEN
; Index
+= 2) {
231 Destination
[Index
] = Source
[Index
+ 1];
232 Destination
[Index
+ 1] = Source
[Index
];
234 AtaDevice
->ModelName
[MAX_MODEL_NAME_LEN
] = L
'\0';
239 Gets ATA device Capacity according to ATA 6.
241 This function returns the capacity of the ATA device if it follows
242 ATA 6 to support 48 bit addressing.
244 @param AtaDevice The ATA child device involved for the operation.
246 @return The capacity of the ATA device or 0 if the device does not support
247 48-bit addressing defined in ATA 6.
252 IN ATA_DEVICE
*AtaDevice
258 ATA_IDENTIFY_DATA
*IdentifyData
;
260 IdentifyData
= AtaDevice
->IdentifyData
;
261 if ((IdentifyData
->command_set_supported_83
& BIT10
) == 0) {
263 // The device doesn't support 48 bit addressing
269 // 48 bit address feature set is supported, get maximum capacity
272 for (Index
= 0; Index
< 4; Index
++) {
274 // Lower byte goes first: word[100] is the lowest word, word[103] is highest
276 TmpLba
= IdentifyData
->maximum_lba_for_48bit_addressing
[Index
];
277 Capacity
|= LShiftU64 (TmpLba
, 16 * Index
);
285 Identifies ATA device via the Identify data.
287 This function identifies the ATA device and initializes the Media information in
288 Block IO protocol interface.
290 @param AtaDevice The ATA child device involved for the operation.
292 @retval EFI_UNSUPPORTED The device is not a valid ATA device (hard disk).
293 @retval EFI_SUCCESS The device is successfully identified and Media information
294 is correctly initialized.
299 IN OUT ATA_DEVICE
*AtaDevice
302 ATA_IDENTIFY_DATA
*IdentifyData
;
303 EFI_BLOCK_IO_MEDIA
*BlockMedia
;
305 UINT16 PhyLogicSectorSupport
;
308 IdentifyData
= AtaDevice
->IdentifyData
;
310 if ((IdentifyData
->config
& BIT15
) != 0) {
312 // This is not an hard disk
314 return EFI_UNSUPPORTED
;
317 DEBUG ((EFI_D_INFO
, "AtaBus - Identify Device: Port %x PortMultiplierPort %x\n", AtaDevice
->Port
, AtaDevice
->PortMultiplierPort
));
320 // Check whether the WORD 88 (supported UltraDMA by drive) is valid
322 if ((IdentifyData
->field_validity
& BIT2
) != 0) {
323 UdmaMode
= IdentifyData
->ultra_dma_mode
;
324 if ((UdmaMode
& (BIT0
| BIT1
| BIT2
| BIT3
| BIT4
| BIT5
| BIT6
)) != 0) {
326 // If BIT0~BIT6 is selected, then UDMA is supported
328 AtaDevice
->UdmaValid
= TRUE
;
332 Capacity
= GetAtapi6Capacity (AtaDevice
);
333 if (Capacity
> MAX_28BIT_ADDRESSING_CAPACITY
) {
335 // Capacity exceeds 120GB. 48-bit addressing is really needed
337 AtaDevice
->Lba48Bit
= TRUE
;
340 // This is a hard disk <= 120GB capacity, treat it as normal hard disk
342 Capacity
= ((UINT32
)IdentifyData
->user_addressable_sectors_hi
<< 16) | IdentifyData
->user_addressable_sectors_lo
;
343 AtaDevice
->Lba48Bit
= FALSE
;
347 // Block Media Information:
349 BlockMedia
= &AtaDevice
->BlockMedia
;
350 BlockMedia
->LastBlock
= Capacity
- 1;
351 BlockMedia
->IoAlign
= AtaDevice
->AtaBusDriverData
->AtaPassThru
->Mode
->IoAlign
;
353 // Check whether Long Physical Sector Feature is supported
355 PhyLogicSectorSupport
= IdentifyData
->phy_logic_sector_support
;
356 if ((PhyLogicSectorSupport
& (BIT14
| BIT15
)) == BIT14
) {
358 // Check whether one physical block contains multiple physical blocks
360 if ((PhyLogicSectorSupport
& BIT13
) != 0) {
361 BlockMedia
->LogicalBlocksPerPhysicalBlock
= (UINT32
) (1 << (PhyLogicSectorSupport
& 0x000f));
363 // Check lowest alignment of logical blocks within physical block
365 if ((IdentifyData
->alignment_logic_in_phy_blocks
& (BIT14
| BIT15
)) == BIT14
) {
366 BlockMedia
->LowestAlignedLba
= (EFI_LBA
) ((BlockMedia
->LogicalBlocksPerPhysicalBlock
- ((UINT32
)IdentifyData
->alignment_logic_in_phy_blocks
& 0x3fff)) %
367 BlockMedia
->LogicalBlocksPerPhysicalBlock
);
371 // Check logical block size
373 if ((PhyLogicSectorSupport
& BIT12
) != 0) {
374 BlockMedia
->BlockSize
= (UINT32
) (((IdentifyData
->logic_sector_size_hi
<< 16) | IdentifyData
->logic_sector_size_lo
) * sizeof (UINT16
));
376 AtaDevice
->BlockIo
.Revision
= EFI_BLOCK_IO_PROTOCOL_REVISION2
;
379 // Get ATA model name from identify data structure.
381 PrintAtaModelName (AtaDevice
);
388 Discovers whether it is a valid ATA device.
390 This function issues ATA_CMD_IDENTIFY_DRIVE command to the ATA device to identify it.
391 If the command is executed successfully, it then identifies it and initializes
392 the Media information in Block IO protocol interface.
394 @param AtaDevice The ATA child device involved for the operation.
396 @retval EFI_SUCCESS The device is successfully identified and Media information
397 is correctly initialized.
398 @return others Some error occurs when discovering the ATA device.
403 IN OUT ATA_DEVICE
*AtaDevice
407 EFI_ATA_COMMAND_BLOCK
*Acb
;
408 EFI_ATA_PASS_THRU_COMMAND_PACKET
*Packet
;
412 // Prepare for ATA command block.
414 Acb
= ZeroMem (&AtaDevice
->Acb
, sizeof (EFI_ATA_COMMAND_BLOCK
));
415 Acb
->AtaCommand
= ATA_CMD_IDENTIFY_DRIVE
;
416 Acb
->AtaDeviceHead
= (UINT8
) (BIT7
| BIT6
| BIT5
| (AtaDevice
->PortMultiplierPort
<< 4));
419 // Prepare for ATA pass through packet.
421 Packet
= ZeroMem (&AtaDevice
->Packet
, sizeof (EFI_ATA_PASS_THRU_COMMAND_PACKET
));
422 Packet
->InDataBuffer
= AtaDevice
->IdentifyData
;
423 Packet
->InTransferLength
= sizeof (ATA_IDENTIFY_DATA
);
424 Packet
->Protocol
= EFI_ATA_PASS_THRU_PROTOCOL_PIO_DATA_IN
;
425 Packet
->Length
= EFI_ATA_PASS_THRU_LENGTH_BYTES
| EFI_ATA_PASS_THRU_LENGTH_SECTOR_COUNT
;
426 Packet
->Timeout
= ATA_TIMEOUT
;
428 Retry
= MAX_RETRY_TIMES
;
430 Status
= AtaDevicePassThru (AtaDevice
, NULL
, NULL
);
431 if (!EFI_ERROR (Status
)) {
433 // The command is issued successfully
435 Status
= IdentifyAtaDevice (AtaDevice
);
438 } while (Retry
-- > 0);
444 Transfer data from ATA device.
446 This function performs one ATA pass through transaction to transfer data from/to
447 ATA device. It chooses the appropriate ATA command and protocol to invoke PassThru
448 interface of ATA pass through.
450 @param[in, out] AtaDevice The ATA child device involved for the operation.
451 @param[in, out] TaskPacket Pointer to a Pass Thru Command Packet. Optional,
452 if it is NULL, blocking mode, and use the packet
453 in AtaDevice. If it is not NULL, non blocking mode,
454 and pass down this Packet.
455 @param[in, out] Buffer The pointer to the current transaction buffer.
456 @param[in] StartLba The starting logical block address to be accessed.
457 @param[in] TransferLength The block number or sector count of the transfer.
458 @param[in] IsWrite Indicates whether it is a write operation.
459 @param[in] Event If Event is NULL, then blocking I/O is performed.
460 If Event is not NULL and non-blocking I/O is
461 supported,then non-blocking I/O is performed,
462 and Event will be signaled when the write
463 request is completed.
465 @retval EFI_SUCCESS The data transfer is complete successfully.
466 @return others Some error occurs when transferring data.
471 IN OUT ATA_DEVICE
*AtaDevice
,
472 IN OUT EFI_ATA_PASS_THRU_COMMAND_PACKET
*TaskPacket
, OPTIONAL
475 IN UINT32 TransferLength
,
477 IN EFI_EVENT Event OPTIONAL
480 EFI_ATA_COMMAND_BLOCK
*Acb
;
481 EFI_ATA_PASS_THRU_COMMAND_PACKET
*Packet
;
484 // Ensure AtaDevice->UdmaValid, AtaDevice->Lba48Bit and IsWrite are valid boolean values
486 ASSERT ((UINTN
) AtaDevice
->UdmaValid
< 2);
487 ASSERT ((UINTN
) AtaDevice
->Lba48Bit
< 2);
488 ASSERT ((UINTN
) IsWrite
< 2);
490 // Prepare for ATA command block.
492 Acb
= ZeroMem (&AtaDevice
->Acb
, sizeof (EFI_ATA_COMMAND_BLOCK
));
493 Acb
->AtaCommand
= mAtaCommands
[AtaDevice
->UdmaValid
][AtaDevice
->Lba48Bit
][IsWrite
];
494 Acb
->AtaSectorNumber
= (UINT8
) StartLba
;
495 Acb
->AtaCylinderLow
= (UINT8
) RShiftU64 (StartLba
, 8);
496 Acb
->AtaCylinderHigh
= (UINT8
) RShiftU64 (StartLba
, 16);
497 Acb
->AtaDeviceHead
= (UINT8
) (BIT7
| BIT6
| BIT5
| (AtaDevice
->PortMultiplierPort
<< 4));
498 Acb
->AtaSectorCount
= (UINT8
) TransferLength
;
499 if (AtaDevice
->Lba48Bit
) {
500 Acb
->AtaSectorNumberExp
= (UINT8
) RShiftU64 (StartLba
, 24);
501 Acb
->AtaCylinderLowExp
= (UINT8
) RShiftU64 (StartLba
, 32);
502 Acb
->AtaCylinderHighExp
= (UINT8
) RShiftU64 (StartLba
, 40);
503 Acb
->AtaSectorCountExp
= (UINT8
) (TransferLength
>> 8);
505 Acb
->AtaDeviceHead
= (UINT8
) (Acb
->AtaDeviceHead
| RShiftU64 (StartLba
, 24));
509 // Prepare for ATA pass through packet.
511 if (TaskPacket
!= NULL
) {
512 Packet
= ZeroMem (TaskPacket
, sizeof (EFI_ATA_PASS_THRU_COMMAND_PACKET
));
514 Packet
= ZeroMem (&AtaDevice
->Packet
, sizeof (EFI_ATA_PASS_THRU_COMMAND_PACKET
));
518 Packet
->OutDataBuffer
= Buffer
;
519 Packet
->OutTransferLength
= TransferLength
;
521 Packet
->InDataBuffer
= Buffer
;
522 Packet
->InTransferLength
= TransferLength
;
525 Packet
->Protocol
= mAtaPassThruCmdProtocols
[AtaDevice
->UdmaValid
][IsWrite
];
526 Packet
->Length
= EFI_ATA_PASS_THRU_LENGTH_SECTOR_COUNT
;
527 Packet
->Timeout
= ATA_TIMEOUT
;
529 return AtaDevicePassThru (AtaDevice
, TaskPacket
, Event
);
535 @param[in, out] Task Pointer to task to be freed.
541 IN OUT ATA_BUS_ASYN_SUB_TASK
*Task
544 if (Task
->Packet
.Asb
!= NULL
) {
545 FreeAlignedBuffer (Task
->Packet
.Asb
, sizeof (EFI_ATA_STATUS_BLOCK
));
547 if (Task
->Packet
.Acb
!= NULL
) {
548 FreePool (Task
->Packet
.Acb
);
555 Call back funtion when the event is signaled.
557 @param[in] Event The Event this notify function registered to.
558 @param[in] Context Pointer to the context data registered to the
564 AtaNonBlockingCallBack (
569 ATA_BUS_ASYN_SUB_TASK
*Task
;
570 ATA_BUS_ASYN_TASK
*AtaTask
;
571 ATA_DEVICE
*AtaDevice
;
575 Task
= (ATA_BUS_ASYN_SUB_TASK
*) Context
;
576 gBS
->CloseEvent (Event
);
578 AtaDevice
= Task
->AtaDevice
;
581 // Check the command status.
582 // If there is error during the sub task source allocation, the error status
583 // should be returned to the caller directly, so here the Task->Token may already
584 // be deleted by the caller and no need to update the status.
586 if ((!(*Task
->IsError
)) && ((Task
->Packet
.Asb
->AtaStatus
& 0x01) == 0x01)) {
587 Task
->Token
->TransactionStatus
= EFI_DEVICE_ERROR
;
591 "NON-BLOCKING EVENT FINISHED!- STATUS = %r\n",
592 Task
->Token
->TransactionStatus
596 // Reduce the SubEventCount, till it comes to zero.
598 (*Task
->UnsignalledEventCount
) --;
599 DEBUG ((EFI_D_BLKIO
, "UnsignalledEventCount = %d\n", *Task
->UnsignalledEventCount
));
602 // Remove the SubTask from the Task list.
604 RemoveEntryList (&Task
->TaskEntry
);
605 if ((*Task
->UnsignalledEventCount
) == 0) {
607 // All Sub tasks are done, then signal the upper layer event.
608 // Except there is error during the sub task source allocation.
610 if (!(*Task
->IsError
)) {
611 gBS
->SignalEvent (Task
->Token
->Event
);
612 DEBUG ((EFI_D_BLKIO
, "Signal the upper layer event!\n"));
615 FreePool (Task
->UnsignalledEventCount
);
616 FreePool (Task
->IsError
);
620 // Finish all subtasks and move to the next task in AtaTaskList.
622 if (!IsListEmpty (&AtaDevice
->AtaTaskList
)) {
623 Entry
= GetFirstNode (&AtaDevice
->AtaTaskList
);
624 AtaTask
= ATA_AYNS_TASK_FROM_ENTRY (Entry
);
625 DEBUG ((EFI_D_BLKIO
, "Start to embark a new Ata Task\n"));
626 DEBUG ((EFI_D_BLKIO
, "AtaTask->NumberOfBlocks = %x; AtaTask->Token=%x\n", AtaTask
->NumberOfBlocks
, AtaTask
->Token
));
627 Status
= AccessAtaDevice (
631 AtaTask
->NumberOfBlocks
,
635 if (EFI_ERROR (Status
)) {
636 AtaTask
->Token
->TransactionStatus
= Status
;
637 gBS
->SignalEvent (AtaTask
->Token
->Event
);
639 RemoveEntryList (Entry
);
646 "PACKET INFO: Write=%s, Length=%x, LowCylinder=%x, HighCylinder=%x, SectionNumber=%x\n",
647 Task
->Packet
.OutDataBuffer
!= NULL
? L
"YES" : L
"NO",
648 Task
->Packet
.OutDataBuffer
!= NULL
? Task
->Packet
.OutTransferLength
: Task
->Packet
.InTransferLength
,
649 Task
->Packet
.Acb
->AtaCylinderLow
,
650 Task
->Packet
.Acb
->AtaCylinderHigh
,
651 Task
->Packet
.Acb
->AtaSectorCount
655 // Free the buffer of SubTask.
657 FreeAtaSubTask (Task
);
661 Read or write a number of blocks from ATA device.
663 This function performs ATA pass through transactions to read/write data from/to
664 ATA device. It may separate the read/write request into several ATA pass through
667 @param[in, out] AtaDevice The ATA child device involved for the operation.
668 @param[in, out] Buffer The pointer to the current transaction buffer.
669 @param[in] StartLba The starting logical block address to be accessed.
670 @param[in] NumberOfBlocks The block number or sector count of the transfer.
671 @param[in] IsWrite Indicates whether it is a write operation.
672 @param[in, out] Token A pointer to the token associated with the transaction.
674 @retval EFI_SUCCESS The data transfer is complete successfully.
675 @return others Some error occurs when transferring data.
680 IN OUT ATA_DEVICE
*AtaDevice
,
681 IN OUT UINT8
*Buffer
,
683 IN UINTN NumberOfBlocks
,
685 IN OUT EFI_BLOCK_IO2_TOKEN
*Token
689 UINTN MaxTransferBlockNumber
;
690 UINTN TransferBlockNumber
;
692 ATA_BUS_ASYN_SUB_TASK
*SubTask
;
695 ATA_BUS_ASYN_TASK
*AtaTask
;
702 Status
= EFI_SUCCESS
;
711 // Ensure AtaDevice->Lba48Bit is a valid boolean value
713 ASSERT ((UINTN
) AtaDevice
->Lba48Bit
< 2);
714 MaxTransferBlockNumber
= mMaxTransferBlockNumber
[AtaDevice
->Lba48Bit
];
715 BlockSize
= AtaDevice
->BlockMedia
.BlockSize
;
718 // Initial the return status and shared account for Non Blocking.
720 if ((Token
!= NULL
) && (Token
->Event
!= NULL
)) {
721 OldTpl
= gBS
->RaiseTPL (TPL_NOTIFY
);
722 if (!IsListEmpty (&AtaDevice
->AtaSubTaskList
)) {
723 AtaTask
= AllocateZeroPool (sizeof (ATA_BUS_ASYN_TASK
));
724 if (AtaTask
== NULL
) {
725 gBS
->RestoreTPL (OldTpl
);
726 return EFI_OUT_OF_RESOURCES
;
728 AtaTask
->AtaDevice
= AtaDevice
;
729 AtaTask
->Buffer
= Buffer
;
730 AtaTask
->IsWrite
= IsWrite
;
731 AtaTask
->NumberOfBlocks
= NumberOfBlocks
;
732 AtaTask
->Signature
= ATA_TASK_SIGNATURE
;
733 AtaTask
->StartLba
= StartLba
;
734 AtaTask
->Token
= Token
;
736 InsertTailList (&AtaDevice
->AtaTaskList
, &AtaTask
->TaskEntry
);
737 gBS
->RestoreTPL (OldTpl
);
740 gBS
->RestoreTPL (OldTpl
);
742 Token
->TransactionStatus
= EFI_SUCCESS
;
743 EventCount
= AllocateZeroPool (sizeof (UINTN
));
744 if (EventCount
== NULL
) {
745 return EFI_OUT_OF_RESOURCES
;
748 IsError
= AllocateZeroPool (sizeof (BOOLEAN
));
749 if (IsError
== NULL
) {
750 FreePool (EventCount
);
751 return EFI_OUT_OF_RESOURCES
;
753 DEBUG ((EFI_D_BLKIO
, "Allocation IsError Addr=%x\n", IsError
));
755 TempCount
= (NumberOfBlocks
+ MaxTransferBlockNumber
- 1) / MaxTransferBlockNumber
;
756 *EventCount
= TempCount
;
757 DEBUG ((EFI_D_BLKIO
, "AccessAtaDevice, NumberOfBlocks=%x\n", NumberOfBlocks
));
758 DEBUG ((EFI_D_BLKIO
, "AccessAtaDevice, MaxTransferBlockNumber=%x\n", MaxTransferBlockNumber
));
759 DEBUG ((EFI_D_BLKIO
, "AccessAtaDevice, EventCount=%x\n", TempCount
));
761 while (!IsListEmpty (&AtaDevice
->AtaTaskList
) || !IsListEmpty (&AtaDevice
->AtaSubTaskList
)) {
765 MicroSecondDelay (100);
770 if (NumberOfBlocks
> MaxTransferBlockNumber
) {
771 TransferBlockNumber
= MaxTransferBlockNumber
;
772 NumberOfBlocks
-= MaxTransferBlockNumber
;
774 TransferBlockNumber
= NumberOfBlocks
;
779 // Create sub event for the sub ata task. Non-blocking mode.
781 if ((Token
!= NULL
) && (Token
->Event
!= NULL
)) {
785 SubTask
= AllocateZeroPool (sizeof (ATA_BUS_ASYN_SUB_TASK
));
786 if (SubTask
== NULL
) {
787 Status
= EFI_OUT_OF_RESOURCES
;
791 OldTpl
= gBS
->RaiseTPL (TPL_NOTIFY
);
792 SubTask
->UnsignalledEventCount
= EventCount
;
793 SubTask
->Signature
= ATA_SUB_TASK_SIGNATURE
;
794 SubTask
->AtaDevice
= AtaDevice
;
795 SubTask
->Token
= Token
;
796 SubTask
->IsError
= IsError
;
797 InsertTailList (&AtaDevice
->AtaSubTaskList
, &SubTask
->TaskEntry
);
798 gBS
->RestoreTPL (OldTpl
);
800 Status
= gBS
->CreateEvent (
803 AtaNonBlockingCallBack
,
808 // If resource allocation fail, the un-signalled event count should equal to
809 // the original one minus the unassigned subtasks number.
811 if (EFI_ERROR (Status
)) {
812 Status
= EFI_OUT_OF_RESOURCES
;
816 Status
= TransferAtaDevice (AtaDevice
, &SubTask
->Packet
, Buffer
, StartLba
, (UINT32
) TransferBlockNumber
, IsWrite
, SubEvent
);
821 DEBUG ((EFI_D_BLKIO
, "Blocking AccessAtaDevice, TransferBlockNumber=%x; StartLba = %x\n", TransferBlockNumber
, StartLba
));
822 Status
= TransferAtaDevice (AtaDevice
, NULL
, Buffer
, StartLba
, (UINT32
) TransferBlockNumber
, IsWrite
, NULL
);
825 if (EFI_ERROR (Status
)) {
830 StartLba
+= TransferBlockNumber
;
831 Buffer
+= TransferBlockNumber
* BlockSize
;
832 } while (NumberOfBlocks
> 0);
835 if ((Token
!= NULL
) && (Token
->Event
!= NULL
)) {
837 // Release resource at non-blocking mode.
839 if (EFI_ERROR (Status
)) {
840 OldTpl
= gBS
->RaiseTPL (TPL_NOTIFY
);
841 Token
->TransactionStatus
= Status
;
842 *EventCount
= (*EventCount
) - (TempCount
- Index
);
845 if (*EventCount
== 0) {
846 FreePool (EventCount
);
850 if (SubTask
!= NULL
) {
851 RemoveEntryList (&SubTask
->TaskEntry
);
852 FreeAtaSubTask (SubTask
);
855 if (SubEvent
!= NULL
) {
856 gBS
->CloseEvent (SubEvent
);
858 gBS
->RestoreTPL (OldTpl
);
866 Trust transfer data from/to ATA device.
868 This function performs one ATA pass through transaction to do a trust transfer from/to
869 ATA device. It chooses the appropriate ATA command and protocol to invoke PassThru
870 interface of ATA pass through.
872 @param AtaDevice The ATA child device involved for the operation.
873 @param Buffer The pointer to the current transaction buffer.
874 @param SecurityProtocolId The value of the "Security Protocol" parameter of
875 the security protocol command to be sent.
876 @param SecurityProtocolSpecificData The value of the "Security Protocol Specific" parameter
877 of the security protocol command to be sent.
878 @param TransferLength The block number or sector count of the transfer.
879 @param IsTrustSend Indicates whether it is a trust send operation or not.
880 @param Timeout The timeout, in 100ns units, to use for the execution
881 of the security protocol command. A Timeout value of 0
882 means that this function will wait indefinitely for the
883 security protocol command to execute. If Timeout is greater
884 than zero, then this function will return EFI_TIMEOUT
885 if the time required to execute the receive data command
886 is greater than Timeout.
887 @param TransferLengthOut A pointer to a buffer to store the size in bytes of the data
888 written to the buffer. Ignore it when IsTrustSend is TRUE.
890 @retval EFI_SUCCESS The data transfer is complete successfully.
891 @return others Some error occurs when transferring data.
896 TrustTransferAtaDevice (
897 IN OUT ATA_DEVICE
*AtaDevice
,
899 IN UINT8 SecurityProtocolId
,
900 IN UINT16 SecurityProtocolSpecificData
,
901 IN UINTN TransferLength
,
902 IN BOOLEAN IsTrustSend
,
904 OUT UINTN
*TransferLengthOut
907 EFI_ATA_COMMAND_BLOCK
*Acb
;
908 EFI_ATA_PASS_THRU_COMMAND_PACKET
*Packet
;
911 EFI_ATA_PASS_THRU_PROTOCOL
*AtaPassThru
;
914 // Ensure AtaDevice->UdmaValid and IsTrustSend are valid boolean values
916 ASSERT ((UINTN
) AtaDevice
->UdmaValid
< 2);
917 ASSERT ((UINTN
) IsTrustSend
< 2);
919 // Prepare for ATA command block.
921 Acb
= ZeroMem (&AtaDevice
->Acb
, sizeof (EFI_ATA_COMMAND_BLOCK
));
922 if (TransferLength
== 0) {
923 Acb
->AtaCommand
= ATA_CMD_TRUST_NON_DATA
;
925 Acb
->AtaCommand
= mAtaTrustCommands
[AtaDevice
->UdmaValid
][IsTrustSend
];
927 Acb
->AtaFeatures
= SecurityProtocolId
;
928 Acb
->AtaSectorCount
= (UINT8
) (TransferLength
/ 512);
929 Acb
->AtaSectorNumber
= (UINT8
) ((TransferLength
/ 512) >> 8);
931 // NOTE: ATA Spec has no explicitly definition for Security Protocol Specific layout.
932 // Here use big endian for Cylinder register.
934 Acb
->AtaCylinderHigh
= (UINT8
) SecurityProtocolSpecificData
;
935 Acb
->AtaCylinderLow
= (UINT8
) (SecurityProtocolSpecificData
>> 8);
936 Acb
->AtaDeviceHead
= (UINT8
) (BIT7
| BIT6
| BIT5
| (AtaDevice
->PortMultiplierPort
<< 4));
939 // Prepare for ATA pass through packet.
941 Packet
= ZeroMem (&AtaDevice
->Packet
, sizeof (EFI_ATA_PASS_THRU_COMMAND_PACKET
));
942 if (TransferLength
== 0) {
943 Packet
->InTransferLength
= 0;
944 Packet
->OutTransferLength
= 0;
945 Packet
->Protocol
= EFI_ATA_PASS_THRU_PROTOCOL_ATA_NON_DATA
;
946 } else if (IsTrustSend
) {
948 // Check the alignment of the incoming buffer prior to invoking underlying ATA PassThru
950 AtaPassThru
= AtaDevice
->AtaBusDriverData
->AtaPassThru
;
951 if ((AtaPassThru
->Mode
->IoAlign
> 1) && !IS_ALIGNED (Buffer
, AtaPassThru
->Mode
->IoAlign
)) {
952 NewBuffer
= AllocateAlignedBuffer (AtaDevice
, TransferLength
);
953 if (NewBuffer
== NULL
) {
954 return EFI_OUT_OF_RESOURCES
;
957 CopyMem (NewBuffer
, Buffer
, TransferLength
);
961 Packet
->OutDataBuffer
= Buffer
;
962 Packet
->OutTransferLength
= (UINT32
) TransferLength
;
963 Packet
->Protocol
= mAtaPassThruCmdProtocols
[AtaDevice
->UdmaValid
][IsTrustSend
];
965 Packet
->InDataBuffer
= Buffer
;
966 Packet
->InTransferLength
= (UINT32
) TransferLength
;
967 Packet
->Protocol
= mAtaPassThruCmdProtocols
[AtaDevice
->UdmaValid
][IsTrustSend
];
969 Packet
->Length
= EFI_ATA_PASS_THRU_LENGTH_BYTES
;
970 Packet
->Timeout
= Timeout
;
972 Status
= AtaDevicePassThru (AtaDevice
, NULL
, NULL
);
973 if (TransferLengthOut
!= NULL
) {
975 *TransferLengthOut
= Packet
->InTransferLength
;