UINT8 *Remaining;\r
UINTN PrdtNumber;\r
\r
- if ((BufferSize & (BIT0 | BIT1)) != 0) {\r
- BufferSize &= ~(BIT0 | BIT1);\r
- DEBUG ((DEBUG_WARN, "UfsInitUtpPrdt: The BufferSize [%d] is not dword-aligned!\n", BufferSize));\r
- }\r
+ ASSERT (((UINTN)Buffer & (BIT0 | BIT1)) == 0);\r
+ ASSERT ((BufferSize & (BIT1 | BIT0)) == 0);\r
\r
if (BufferSize == 0) {\r
return EFI_SUCCESS;\r
}\r
\r
- ASSERT (((UINTN)Buffer & (BIT0 | BIT1)) == 0);\r
-\r
RemainingLen = BufferSize;\r
Remaining = Buffer;\r
PrdtNumber = (UINTN)DivU64x32 ((UINT64)BufferSize + UFS_MAX_DATA_LEN_PER_PRD - 1, UFS_MAX_DATA_LEN_PER_PRD);\r
return Status;\r
}\r
\r
+/**\r
+ Cleanup data buffers after data transfer. This function\r
+ also takes care to copy all data to user memory pool for\r
+ unaligned data transfers.\r
+\r
+ @param[in] Private Pointer to the UFS_PASS_THRU_PRIVATE_DATA\r
+ @param[in] TransReq Pointer to the transfer request\r
+**/\r
+VOID\r
+UfsReconcileDataTransferBuffer (\r
+ IN UFS_PASS_THRU_PRIVATE_DATA *Private,\r
+ IN UFS_PASS_THRU_TRANS_REQ *TransReq\r
+ )\r
+{\r
+ if (TransReq->DataBufMapping != NULL) {\r
+ Private->UfsHostController->Unmap (\r
+ Private->UfsHostController,\r
+ TransReq->DataBufMapping\r
+ );\r
+ }\r
+\r
+ //\r
+ // Check if unaligned transfer was performed. If it was and we read\r
+ // data from device copy memory to user data buffers before cleanup.\r
+ // The assumption is if auxiliary aligned data buffer is not NULL then\r
+ // unaligned transfer has been performed.\r
+ //\r
+ if (TransReq->AlignedDataBuf != NULL) {\r
+ if (TransReq->Packet->DataDirection == EFI_EXT_SCSI_DATA_DIRECTION_READ) {\r
+ CopyMem (TransReq->Packet->InDataBuffer, TransReq->AlignedDataBuf, TransReq->Packet->InTransferLength);\r
+ }\r
+ //\r
+ // Wipe out the transfer buffer in case it contains sensitive data.\r
+ //\r
+ ZeroMem (TransReq->AlignedDataBuf, TransReq->AlignedDataBufSize);\r
+ FreeAlignedPages (TransReq->AlignedDataBuf, EFI_SIZE_TO_PAGES (TransReq->AlignedDataBufSize));\r
+ TransReq->AlignedDataBuf = NULL;\r
+ }\r
+}\r
+\r
+/**\r
+ Prepare data buffer for transfer.\r
+\r
+ @param[in] Private Pointer to the UFS_PASS_THRU_PRIVATE_DATA\r
+ @param[in, out] TransReq Pointer to the transfer request\r
+\r
+ @retval EFI_DEVICE_ERROR Failed to prepare buffer for transfer\r
+ @retval EFI_SUCCESS Buffer ready for transfer\r
+**/\r
+EFI_STATUS\r
+UfsPrepareDataTransferBuffer (\r
+ IN UFS_PASS_THRU_PRIVATE_DATA *Private,\r
+ IN OUT UFS_PASS_THRU_TRANS_REQ *TransReq\r
+ )\r
+{\r
+ EFI_STATUS Status;\r
+ VOID *DataBuf;\r
+ UINT32 DataLen;\r
+ UINTN MapLength;\r
+ EFI_PHYSICAL_ADDRESS DataBufPhyAddr;\r
+ EDKII_UFS_HOST_CONTROLLER_OPERATION Flag;\r
+ UTP_TR_PRD *PrdtBase;\r
+\r
+ DataBufPhyAddr = 0;\r
+ DataBuf = NULL;\r
+\r
+ //\r
+ // For unaligned data transfers we allocate auxiliary DWORD aligned memory pool.\r
+ // When command is finished auxiliary memory pool is copied into actual user memory.\r
+ // This is requiered to assure data transfer safety(DWORD alignment required by UFS spec.)\r
+ //\r
+ if (TransReq->Packet->DataDirection == EFI_EXT_SCSI_DATA_DIRECTION_READ) {\r
+ if (((UINTN)TransReq->Packet->InDataBuffer % 4 != 0) || (TransReq->Packet->InTransferLength % 4 != 0)) {\r
+ DataLen = TransReq->Packet->InTransferLength + (4 - (TransReq->Packet->InTransferLength % 4));\r
+ DataBuf = AllocateAlignedPages (EFI_SIZE_TO_PAGES (DataLen), 4);\r
+ if (DataBuf == NULL) {\r
+ return EFI_DEVICE_ERROR;\r
+ }\r
+ ZeroMem (DataBuf, DataLen);\r
+ TransReq->AlignedDataBuf = DataBuf;\r
+ TransReq->AlignedDataBufSize = DataLen;\r
+ } else {\r
+ DataLen = TransReq->Packet->InTransferLength;\r
+ DataBuf = TransReq->Packet->InDataBuffer;\r
+ }\r
+ Flag = EdkiiUfsHcOperationBusMasterWrite;\r
+ } else {\r
+ if (((UINTN)TransReq->Packet->OutDataBuffer % 4 != 0) || (TransReq->Packet->OutTransferLength % 4 != 0)) {\r
+ DataLen = TransReq->Packet->OutTransferLength + (4 - (TransReq->Packet->OutTransferLength % 4));\r
+ DataBuf = AllocateAlignedPages (EFI_SIZE_TO_PAGES (DataLen), 4);\r
+ if (DataBuf == NULL) {\r
+ return EFI_DEVICE_ERROR;\r
+ }\r
+ CopyMem (DataBuf, TransReq->Packet->OutDataBuffer, TransReq->Packet->OutTransferLength);\r
+ TransReq->AlignedDataBuf = DataBuf;\r
+ TransReq->AlignedDataBufSize = DataLen;\r
+ } else {\r
+ DataLen = TransReq->Packet->OutTransferLength;\r
+ DataBuf = TransReq->Packet->OutDataBuffer;\r
+ }\r
+ Flag = EdkiiUfsHcOperationBusMasterRead;\r
+ }\r
+\r
+ if (DataLen != 0) {\r
+ MapLength = DataLen;\r
+ Status = Private->UfsHostController->Map (\r
+ Private->UfsHostController,\r
+ Flag,\r
+ DataBuf,\r
+ &MapLength,\r
+ &DataBufPhyAddr,\r
+ &TransReq->DataBufMapping\r
+ );\r
+\r
+ if (EFI_ERROR (Status) || (DataLen != MapLength)) {\r
+ if (TransReq->AlignedDataBuf != NULL) {\r
+ //\r
+ // Wipe out the transfer buffer in case it contains sensitive data.\r
+ //\r
+ ZeroMem (TransReq->AlignedDataBuf, TransReq->AlignedDataBufSize);\r
+ FreeAlignedPages (TransReq->AlignedDataBuf, EFI_SIZE_TO_PAGES (TransReq->AlignedDataBufSize));\r
+ TransReq->AlignedDataBuf = NULL;\r
+ }\r
+ return EFI_DEVICE_ERROR;\r
+ }\r
+ }\r
+\r
+ //\r
+ // Fill PRDT table of Command UPIU for executed SCSI cmd.\r
+ //\r
+ PrdtBase = (UTP_TR_PRD*)((UINT8*)TransReq->CmdDescHost + ROUNDUP8 (sizeof (UTP_COMMAND_UPIU)) + ROUNDUP8 (sizeof (UTP_RESPONSE_UPIU)));\r
+ ASSERT (PrdtBase != NULL);\r
+ UfsInitUtpPrdt (PrdtBase, (VOID*)(UINTN)DataBufPhyAddr, DataLen);\r
+\r
+ return EFI_SUCCESS;\r
+}\r
+\r
/**\r
Sends a UFS-supported SCSI Request Packet to a UFS device that is attached to the UFS host controller.\r
\r
UTP_RESPONSE_UPIU *Response;\r
UINT16 SenseDataLen;\r
UINT32 ResTranCount;\r
- VOID *DataBuf;\r
- EFI_PHYSICAL_ADDRESS DataBufPhyAddr;\r
- UINT32 DataLen;\r
- UINTN MapLength;\r
- EDKII_UFS_HOST_CONTROLLER_PROTOCOL *UfsHc;\r
- EDKII_UFS_HOST_CONTROLLER_OPERATION Flag;\r
- UTP_TR_PRD *PrdtBase;\r
EFI_TPL OldTpl;\r
UFS_PASS_THRU_TRANS_REQ *TransReq;\r
+ EDKII_UFS_HOST_CONTROLLER_PROTOCOL *UfsHc;\r
\r
- TransReq = AllocateZeroPool (sizeof (UFS_PASS_THRU_TRANS_REQ));\r
+ TransReq = AllocateZeroPool (sizeof (UFS_PASS_THRU_TRANS_REQ));\r
if (TransReq == NULL) {\r
return EFI_OUT_OF_RESOURCES;\r
}\r
\r
TransReq->Signature = UFS_PASS_THRU_TRANS_REQ_SIG;\r
TransReq->TimeoutRemain = Packet->Timeout;\r
- DataBufPhyAddr = 0;\r
+ TransReq->Packet = Packet;\r
+\r
UfsHc = Private->UfsHostController;\r
//\r
// Find out which slot of transfer request list is available.\r
\r
TransReq->CmdDescSize = TransReq->Trd->PrdtO * sizeof (UINT32) + TransReq->Trd->PrdtL * sizeof (UTP_TR_PRD);\r
\r
- if (Packet->DataDirection == EFI_EXT_SCSI_DATA_DIRECTION_READ) {\r
- DataBuf = Packet->InDataBuffer;\r
- DataLen = Packet->InTransferLength;\r
- Flag = EdkiiUfsHcOperationBusMasterWrite;\r
- } else {\r
- DataBuf = Packet->OutDataBuffer;\r
- DataLen = Packet->OutTransferLength;\r
- Flag = EdkiiUfsHcOperationBusMasterRead;\r
- }\r
-\r
- if (DataLen != 0) {\r
- MapLength = DataLen;\r
- Status = UfsHc->Map (\r
- UfsHc,\r
- Flag,\r
- DataBuf,\r
- &MapLength,\r
- &DataBufPhyAddr,\r
- &TransReq->DataBufMapping\r
- );\r
-\r
- if (EFI_ERROR (Status) || (DataLen != MapLength)) {\r
- goto Exit1;\r
- }\r
+ Status = UfsPrepareDataTransferBuffer (Private, TransReq);\r
+ if (EFI_ERROR (Status)) {\r
+ goto Exit1;\r
}\r
- //\r
- // Fill PRDT table of Command UPIU for executed SCSI cmd.\r
- //\r
- PrdtBase = (UTP_TR_PRD*)((UINT8*)TransReq->CmdDescHost + ROUNDUP8 (sizeof (UTP_COMMAND_UPIU)) + ROUNDUP8 (sizeof (UTP_RESPONSE_UPIU)));\r
- ASSERT (PrdtBase != NULL);\r
- UfsInitUtpPrdt (PrdtBase, (VOID*)(UINTN)DataBufPhyAddr, DataLen);\r
\r
//\r
// Insert the async SCSI cmd to the Async I/O list\r
//\r
if (Event != NULL) {\r
OldTpl = gBS->RaiseTPL (TPL_NOTIFY);\r
- TransReq->Packet = Packet;\r
TransReq->CallerEvent = Event;\r
InsertTailList (&Private->Queue, &TransReq->TransferList);\r
gBS->RestoreTPL (OldTpl);\r
\r
UfsStopExecCmd (Private, TransReq->Slot);\r
\r
- if (TransReq->DataBufMapping != NULL) {\r
- UfsHc->Unmap (UfsHc, TransReq->DataBufMapping);\r
- }\r
+ UfsReconcileDataTransferBuffer (Private, TransReq);\r
\r
Exit1:\r
if (TransReq->CmdDescMapping != NULL) {\r
return Status;\r
}\r
\r
-\r
/**\r
Send UIC command.\r
\r
return EFI_SUCCESS;\r
}\r
\r
-\r
/**\r
Internal helper function which will signal the caller event and clean up\r
resources.\r
\r
UfsStopExecCmd (Private, TransReq->Slot);\r
\r
- if (TransReq->DataBufMapping != NULL) {\r
- UfsHc->Unmap (UfsHc, TransReq->DataBufMapping);\r
- }\r
+ UfsReconcileDataTransferBuffer (Private, TransReq);\r
\r
if (TransReq->CmdDescMapping != NULL) {\r
UfsHc->Unmap (UfsHc, TransReq->CmdDescMapping);\r