MdeModulePkg/AtaAtapiPassThruDxe: Support 4K bytes block size HDDs

[mirror_edk2.git] / MdeModulePkg / Bus / Ata / AtaAtapiPassThru / AtaAtapiPassThru.c

diff --git a/MdeModulePkg/Bus/Ata/AtaAtapiPassThru/AtaAtapiPassThru.c b/MdeModulePkg/Bus/Ata/AtaAtapiPassThru/AtaAtapiPassThru.c
index 6ccf0a3f014995545d08875f8f87247b5c1a93f4..870900f1060db9310d4008e4bf1e1efae89dd341 100644 (file)
--- a/MdeModulePkg/Bus/Ata/AtaAtapiPassThru/AtaAtapiPassThru.c
+++ b/MdeModulePkg/Bus/Ata/AtaAtapiPassThru/AtaAtapiPassThru.c
@@ -2,7 +2,7 @@
   This file implements ATA_PASSTHRU_PROCTOCOL and EXT_SCSI_PASSTHRU_PROTOCOL interfaces\r
   for managed ATA controllers.\r
 \r
-  Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2010 - 2015, Intel Corporation. All rights reserved.<BR>\r
   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
@@ -1255,6 +1255,7 @@ AtaPassThruPassThru (
   UINT32                          MaxSectorCount;\r
   ATA_NONBLOCK_TASK               *Task;\r
   EFI_TPL                         OldTpl;\r
+  UINT32                          BlockSize;\r
 \r
   Instance = ATA_PASS_THRU_PRIVATE_DATA_FROM_THIS (This);\r
 \r
@@ -1279,22 +1280,6 @@ AtaPassThruPassThru (
     }\r
   }\r
 \r
-  //\r
-  // convert the transfer length from sector count to byte.\r
-  //\r
-  if (((Packet->Length & EFI_ATA_PASS_THRU_LENGTH_BYTES) == 0) &&\r
-       (Packet->InTransferLength != 0)) {\r
-    Packet->InTransferLength = Packet->InTransferLength * 0x200;\r
-  }\r
-\r
-  //\r
-  // convert the transfer length from sector count to byte.\r
-  //\r
-  if (((Packet->Length & EFI_ATA_PASS_THRU_LENGTH_BYTES) == 0) &&\r
-       (Packet->OutTransferLength != 0)) {\r
-    Packet->OutTransferLength = Packet->OutTransferLength * 0x200;\r
-  }\r
-\r
   //\r
   // Check whether this device needs 48-bit addressing (ATAPI-6 ata device).\r
   // Per ATA-6 spec, word83: bit15 is zero and bit14 is one.\r
@@ -1314,13 +1299,39 @@ AtaPassThruPassThru (
     }\r
   }\r
 \r
+  BlockSize = 0x200;\r
+  if ((IdentifyData->AtaData.phy_logic_sector_support & (BIT14 | BIT15)) == BIT14) {\r
+    //\r
+    // Check logical block size\r
+    //\r
+    if ((IdentifyData->AtaData.phy_logic_sector_support & BIT12) != 0) {\r
+      BlockSize = (UINT32) (((IdentifyData->AtaData.logic_sector_size_hi << 16) | IdentifyData->AtaData.logic_sector_size_lo) * sizeof (UINT16));\r
+    }\r
+  }\r
+\r
+  //\r
+  // convert the transfer length from sector count to byte.\r
+  //\r
+  if (((Packet->Length & EFI_ATA_PASS_THRU_LENGTH_BYTES) == 0) &&\r
+       (Packet->InTransferLength != 0)) {\r
+    Packet->InTransferLength = Packet->InTransferLength * BlockSize;\r
+  }\r
+\r
+  //\r
+  // convert the transfer length from sector count to byte.\r
+  //\r
+  if (((Packet->Length & EFI_ATA_PASS_THRU_LENGTH_BYTES) == 0) &&\r
+       (Packet->OutTransferLength != 0)) {\r
+    Packet->OutTransferLength = Packet->OutTransferLength * BlockSize;\r
+  }\r
+\r
   //\r
   // If the data buffer described by InDataBuffer/OutDataBuffer and InTransferLength/OutTransferLength\r
   // is too big to be transferred in a single command, then no data is transferred and EFI_BAD_BUFFER_SIZE\r
   // is returned.\r
   //\r
-  if (((Packet->InTransferLength != 0) && (Packet->InTransferLength > MaxSectorCount * 0x200)) ||\r
-      ((Packet->OutTransferLength != 0) && (Packet->OutTransferLength > MaxSectorCount * 0x200))) {\r
+  if (((Packet->InTransferLength != 0) && (Packet->InTransferLength > MaxSectorCount * BlockSize)) ||\r
+      ((Packet->OutTransferLength != 0) && (Packet->OutTransferLength > MaxSectorCount * BlockSize))) {\r
     return EFI_BAD_BUFFER_SIZE;\r
   }\r
 \r
@@ -2074,7 +2085,7 @@ ExtScsiPassThruPassThru (
       // no more sense key or number of sense keys exceeds predefined,\r
       // skip the loop.\r
       //\r
-      if ((PtrSenseData->Sense_Key == EFI_SCSI_SK_NO_SENSE) || \r
+      if ((PtrSenseData->Sense_Key == EFI_SCSI_SK_NO_SENSE) ||\r
           (SenseDataLen + sizeof (EFI_SCSI_SENSE_DATA) > Packet->SenseDataLength)) {\r
         SenseReq = FALSE;\r
       }\r