]> git.proxmox.com Git - mirror_edk2.git/commitdiff
MdeModulePkg: add generic SataController driver.
authorFeng Tian <feng.tian@intel.com>
Tue, 19 Jul 2016 05:57:49 +0000 (13:57 +0800)
committerFeng Tian <feng.tian@intel.com>
Wed, 3 Aug 2016 01:52:18 +0000 (09:52 +0800)
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Feng Tian <feng.tian@intel.com>
Reviewed-by: Giri P Mudusuru <giri.p.mudusuru@intel.com>
Reviewed-by: Amy Chan <amy.chan@intel.com>
MdeModulePkg/Bus/Pci/SataControllerDxe/ComponentName.c [new file with mode: 0644]
MdeModulePkg/Bus/Pci/SataControllerDxe/SataController.c [new file with mode: 0644]
MdeModulePkg/Bus/Pci/SataControllerDxe/SataController.h [new file with mode: 0644]
MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxe.inf [new file with mode: 0644]
MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxe.uni [new file with mode: 0644]
MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxeExtra.uni [new file with mode: 0644]
MdeModulePkg/MdeModulePkg.dsc

diff --git a/MdeModulePkg/Bus/Pci/SataControllerDxe/ComponentName.c b/MdeModulePkg/Bus/Pci/SataControllerDxe/ComponentName.c
new file mode 100644 (file)
index 0000000..467ad11
--- /dev/null
@@ -0,0 +1,177 @@
+/** @file\r
+  UEFI Component Name(2) protocol implementation for Sata Controller driver.\r
+\r
+  Copyright (c) 2011 - 2016, 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
+  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
+#include "SataController.h"\r
+\r
+//\r
+/// EFI Component Name Protocol\r
+///\r
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_COMPONENT_NAME_PROTOCOL  gSataControllerComponentName = {\r
+  SataControllerComponentNameGetDriverName,\r
+  SataControllerComponentNameGetControllerName,\r
+  "eng"\r
+};\r
+\r
+//\r
+/// EFI Component Name 2 Protocol\r
+///\r
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_COMPONENT_NAME2_PROTOCOL gSataControllerComponentName2 = {\r
+  (EFI_COMPONENT_NAME2_GET_DRIVER_NAME) SataControllerComponentNameGetDriverName,\r
+  (EFI_COMPONENT_NAME2_GET_CONTROLLER_NAME) SataControllerComponentNameGetControllerName,\r
+  "en"\r
+};\r
+\r
+//\r
+/// Driver Name Strings\r
+///\r
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_UNICODE_STRING_TABLE mSataControllerDriverNameTable[] = {\r
+  {\r
+    "eng;en",\r
+    (CHAR16 *)L"Sata Controller Init Driver"\r
+  },\r
+  {\r
+    NULL,\r
+    NULL\r
+  }\r
+};\r
+\r
+///\r
+/// Controller Name Strings\r
+///\r
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_UNICODE_STRING_TABLE mSataControllerControllerNameTable[] = {\r
+  {\r
+    "eng;en",\r
+    (CHAR16 *)L"Sata Controller"\r
+  },\r
+  {\r
+    NULL,\r
+    NULL\r
+  }\r
+};\r
+\r
+/**\r
+  Retrieves a Unicode string that is the user readable name of the UEFI Driver.\r
+\r
+  @param This                   A pointer to the EFI_COMPONENT_NAME_PROTOCOL instance.\r
+  @param Language               A pointer to a three character ISO 639-2 language identifier.\r
+                                This is the language of the driver name that that the caller\r
+                                is requesting, and it must match one of the languages specified\r
+                                in SupportedLanguages.  The number of languages supported by a\r
+                                driver is up to the driver writer.\r
+  @param DriverName             A pointer to the Unicode string to return.  This Unicode string\r
+                                is the name of the driver specified by This in the language\r
+                                specified by Language.\r
+  \r
+  @retval EFI_SUCCESS           The Unicode string for the Driver specified by This\r
+                                and the language specified by Language was returned\r
+                                in DriverName.\r
+  @retval EFI_INVALID_PARAMETER Language is NULL.\r
+  @retval EFI_INVALID_PARAMETER DriverName is NULL.\r
+  @retval EFI_UNSUPPORTED       The driver specified by This does not support the\r
+                                language specified by Language.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerComponentNameGetDriverName (\r
+  IN EFI_COMPONENT_NAME_PROTOCOL    *This,\r
+  IN CHAR8                          *Language,\r
+  OUT CHAR16                        **DriverName\r
+  )\r
+{\r
+  return LookupUnicodeString2 (\r
+           Language,\r
+           This->SupportedLanguages,\r
+           mSataControllerDriverNameTable,\r
+           DriverName,\r
+           (BOOLEAN)(This == &gSataControllerComponentName)\r
+           );\r
+}\r
+\r
+/**\r
+  Retrieves a Unicode string that is the user readable name of the controller\r
+  that is being managed by an UEFI Driver.  \r
+\r
+  @param This                   A pointer to the EFI_COMPONENT_NAME_PROTOCOL instance.\r
+  @param ControllerHandle       The handle of a controller that the driver specified by\r
+                                This is managing.  This handle specifies the controller\r
+                                whose name is to be returned.\r
+  @param ChildHandle OPTIONAL   The handle of the child controller to retrieve the name\r
+                                of.  This is an optional parameter that may be NULL.  It\r
+                                will be NULL for device drivers.  It will also be NULL\r
+                                for a bus drivers that wish to retrieve the name of the\r
+                                bus controller.  It will not be NULL for a bus driver\r
+                                that wishes to retrieve the name of a child controller.\r
+  @param Language               A pointer to a three character ISO 639-2 language\r
+                                identifier.  This is the language of the controller name\r
+                                that that the caller is requesting, and it must match one\r
+                                of the languages specified in SupportedLanguages.  The\r
+                                number of languages supported by a driver is up to the\r
+                                driver writer.\r
+  @param ControllerName         A pointer to the Unicode string to return.  This Unicode\r
+                                string is the name of the controller specified by\r
+                                ControllerHandle and ChildHandle in the language\r
+                                specified by Language from the point of view of the\r
+                                driver specified by This.\r
+  \r
+  @retval EFI_SUCCESS           The Unicode string for the user readable name in the\r
+                                language specified by Language for the driver\r
+                                specified by This was returned in DriverName.\r
+  @retval EFI_INVALID_PARAMETER ControllerHandle is not a valid EFI_HANDLE.\r
+  @retval EFI_INVALID_PARAMETER ChildHandle is not NULL and it is not a valid\r
+                                EFI_HANDLE.\r
+  @retval EFI_INVALID_PARAMETER Language is NULL.\r
+  @retval EFI_INVALID_PARAMETER ControllerName is NULL.\r
+  @retval EFI_UNSUPPORTED       The driver specified by This is not currently\r
+                                managing the controller specified by\r
+                                ControllerHandle and ChildHandle.\r
+  @retval EFI_UNSUPPORTED       The driver specified by This does not support the\r
+                                language specified by Language.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerComponentNameGetControllerName (\r
+  IN EFI_COMPONENT_NAME_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     ControllerHandle,\r
+  IN EFI_HANDLE                     ChildHandle OPTIONAL,\r
+  IN CHAR8                          *Language,\r
+  OUT CHAR16                        **ControllerName\r
+  )\r
+{\r
+  EFI_STATUS    Status;\r
+\r
+  //\r
+  // Make sure this driver is currently managing ControllHandle\r
+  //\r
+  Status = EfiTestManagedDevice (\r
+             ControllerHandle,\r
+             gSataControllerDriverBinding.DriverBindingHandle,\r
+             &gEfiPciIoProtocolGuid\r
+             );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  if (ChildHandle != NULL) {\r
+    return EFI_UNSUPPORTED;\r
+  }\r
+\r
+  return LookupUnicodeString2 (\r
+          Language,\r
+          This->SupportedLanguages,\r
+          mSataControllerControllerNameTable,\r
+          ControllerName,\r
+          (BOOLEAN)(This == &gSataControllerComponentName)\r
+          );\r
+}\r
+\r
diff --git a/MdeModulePkg/Bus/Pci/SataControllerDxe/SataController.c b/MdeModulePkg/Bus/Pci/SataControllerDxe/SataController.c
new file mode 100644 (file)
index 0000000..a6d55c1
--- /dev/null
@@ -0,0 +1,1021 @@
+/** @file\r
+  This driver module produces IDE_CONTROLLER_INIT protocol for Sata Controllers.\r
+\r
+  Copyright (c) 2011 - 2016, 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
+  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
+#include "SataController.h"\r
+\r
+///\r
+/// EFI_DRIVER_BINDING_PROTOCOL instance\r
+///\r
+EFI_DRIVER_BINDING_PROTOCOL gSataControllerDriverBinding = {\r
+  SataControllerSupported,\r
+  SataControllerStart,\r
+  SataControllerStop,\r
+  0xa,\r
+  NULL,\r
+  NULL\r
+};\r
+\r
+/**\r
+  Read AHCI Operation register.\r
+\r
+  @param PciIo      The PCI IO protocol instance.\r
+  @param Offset     The operation register offset.\r
+\r
+  @return The register content read.\r
+\r
+**/\r
+UINT32\r
+EFIAPI\r
+AhciReadReg (\r
+  IN EFI_PCI_IO_PROTOCOL    *PciIo,\r
+  IN UINT32                 Offset\r
+  )\r
+{\r
+  UINT32    Data;\r
+\r
+  ASSERT (PciIo != NULL);\r
+  \r
+  Data = 0;\r
+\r
+  PciIo->Mem.Read (\r
+               PciIo,\r
+               EfiPciIoWidthUint32,\r
+               AHCI_BAR_INDEX,\r
+               (UINT64) Offset,\r
+               1,\r
+               &Data\r
+               );\r
+\r
+  return Data;\r
+}\r
+\r
+/**\r
+  This function is used to calculate the best PIO mode supported by specific IDE device\r
+\r
+  @param IdentifyData   The identify data of specific IDE device.\r
+  @param DisPioMode     Disqualified PIO modes collection.\r
+  @param SelectedMode   Available PIO modes collection.\r
+\r
+  @retval EFI_SUCCESS       Best PIO modes are returned.\r
+  @retval EFI_UNSUPPORTED   The device doesn't support PIO mode,\r
+                            or all supported modes have been disqualified.\r
+**/\r
+EFI_STATUS\r
+CalculateBestPioMode (\r
+  IN EFI_IDENTIFY_DATA  *IdentifyData,\r
+  IN UINT16             *DisPioMode OPTIONAL,\r
+  OUT UINT16            *SelectedMode\r
+  )\r
+{\r
+  UINT16    PioMode;\r
+  UINT16    AdvancedPioMode;\r
+  UINT16    Temp;\r
+  UINT16    Index;\r
+  UINT16    MinimumPioCycleTime;\r
+\r
+  Temp = 0xff;\r
+\r
+  PioMode = (UINT8) (((ATA5_IDENTIFY_DATA *) (&(IdentifyData->AtaData)))->pio_cycle_timing >> 8);\r
+\r
+  //\r
+  // See whether Identify Data word 64 - 70 are valid\r
+  //\r
+  if ((IdentifyData->AtaData.field_validity & 0x02) == 0x02) {\r
+\r
+    AdvancedPioMode = IdentifyData->AtaData.advanced_pio_modes;\r
+    DEBUG ((EFI_D_INFO, "CalculateBestPioMode: AdvancedPioMode = %x\n", AdvancedPioMode));\r
+\r
+    for (Index = 0; Index < 8; Index++) {\r
+      if ((AdvancedPioMode & 0x01) != 0) {\r
+        Temp = Index;\r
+      }\r
+\r
+      AdvancedPioMode >>= 1;\r
+    }\r
+\r
+    //\r
+    // If Temp is modified, mean the advanced_pio_modes is not zero;\r
+    // if Temp is not modified, mean there is no advanced PIO mode supported,\r
+    // the best PIO Mode is the value in pio_cycle_timing.\r
+    //\r
+    if (Temp != 0xff) {\r
+      AdvancedPioMode = (UINT16) (Temp + 3);\r
+    } else {\r
+      AdvancedPioMode = PioMode;\r
+    }\r
+\r
+    //\r
+    // Limit the PIO mode to at most PIO4.\r
+    //\r
+    PioMode = (UINT16) MIN (AdvancedPioMode, 4);\r
+\r
+    MinimumPioCycleTime = IdentifyData->AtaData.min_pio_cycle_time_with_flow_control;\r
+\r
+    if (MinimumPioCycleTime <= 120) {\r
+      PioMode = (UINT16) MIN (4, PioMode);\r
+    } else if (MinimumPioCycleTime <= 180) {\r
+      PioMode = (UINT16) MIN (3, PioMode);\r
+    } else if (MinimumPioCycleTime <= 240) {\r
+      PioMode = (UINT16) MIN (2, PioMode);\r
+    } else {\r
+      PioMode = 0;\r
+    }\r
+\r
+    //\r
+    // Degrade the PIO mode if the mode has been disqualified\r
+    //\r
+    if (DisPioMode != NULL) {\r
+      if (*DisPioMode < 2) {\r
+        return EFI_UNSUPPORTED; // no mode below ATA_PIO_MODE_BELOW_2\r
+      }\r
+\r
+      if (PioMode >= *DisPioMode) {\r
+        PioMode = (UINT16) (*DisPioMode - 1);\r
+      }\r
+    }\r
+\r
+    if (PioMode < 2) {\r
+      *SelectedMode = 1;        // ATA_PIO_MODE_BELOW_2;\r
+    } else {\r
+      *SelectedMode = PioMode;  // ATA_PIO_MODE_2 to ATA_PIO_MODE_4;\r
+    }\r
+\r
+  } else {\r
+    //\r
+    // Identify Data word 64 - 70 are not valid\r
+    // Degrade the PIO mode if the mode has been disqualified\r
+    //\r
+    if (DisPioMode != NULL) {\r
+      if (*DisPioMode < 2) {\r
+        return EFI_UNSUPPORTED; // no mode below ATA_PIO_MODE_BELOW_2\r
+      }\r
+\r
+      if (PioMode == *DisPioMode) {\r
+        PioMode--;\r
+      }\r
+    }\r
+\r
+    if (PioMode < 2) {\r
+      *SelectedMode = 1;        // ATA_PIO_MODE_BELOW_2;\r
+    } else {\r
+      *SelectedMode = 2;        // ATA_PIO_MODE_2;\r
+    }\r
+\r
+  }\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  This function is used to calculate the best UDMA mode supported by specific IDE device\r
+\r
+  @param IdentifyData   The identify data of specific IDE device.\r
+  @param DisUDmaMode     Disqualified UDMA modes collection.\r
+  @param SelectedMode   Available UDMA modes collection.\r
+\r
+  @retval EFI_SUCCESS       Best UDMA modes are returned.\r
+  @retval EFI_UNSUPPORTED   The device doesn't support UDMA mode,\r
+                            or all supported modes have been disqualified.\r
+**/\r
+EFI_STATUS\r
+CalculateBestUdmaMode (\r
+  IN EFI_IDENTIFY_DATA  *IdentifyData,\r
+  IN UINT16             *DisUDmaMode OPTIONAL,\r
+  OUT UINT16            *SelectedMode\r
+  )\r
+{\r
+  UINT16    TempMode;\r
+  UINT16    DeviceUDmaMode;\r
+\r
+  DeviceUDmaMode = 0;\r
+\r
+  //\r
+  // Check whether the WORD 88 (supported UltraDMA by drive) is valid\r
+  //\r
+  if ((IdentifyData->AtaData.field_validity & 0x04) == 0x00) {\r
+    return EFI_UNSUPPORTED;\r
+  }\r
+\r
+  DeviceUDmaMode = IdentifyData->AtaData.ultra_dma_mode;\r
+  DEBUG ((EFI_D_INFO, "CalculateBestUdmaMode: DeviceUDmaMode = %x\n", DeviceUDmaMode));\r
+  DeviceUDmaMode &= 0x3f;\r
+  TempMode = 0;                 // initialize it to UDMA-0\r
+\r
+  while ((DeviceUDmaMode >>= 1) != 0) {\r
+    TempMode++;\r
+  }\r
+\r
+  //\r
+  // Degrade the UDMA mode if the mode has been disqualified\r
+  //\r
+  if (DisUDmaMode != NULL) {\r
+    if (*DisUDmaMode == 0) {\r
+      *SelectedMode = 0;\r
+      return EFI_UNSUPPORTED;   // no mode below ATA_UDMA_MODE_0\r
+    }\r
+\r
+    if (TempMode >= *DisUDmaMode) {\r
+      TempMode = (UINT16) (*DisUDmaMode - 1);\r
+    }\r
+  }\r
+\r
+  //\r
+  // Possible returned mode is between ATA_UDMA_MODE_0 and ATA_UDMA_MODE_5\r
+  //\r
+  *SelectedMode = TempMode;\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  The Entry Point of module. It follows the standard UEFI driver model.\r
+\r
+  @param[in] ImageHandle    The firmware allocated handle for the EFI image.  \r
+  @param[in] SystemTable    A pointer to the EFI System Table.\r
+  \r
+  @retval EFI_SUCCESS   The entry point is executed successfully.\r
+  @retval other         Some error occurs when executing this entry point.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+InitializeSataControllerDriver (\r
+  IN EFI_HANDLE         ImageHandle,\r
+  IN EFI_SYSTEM_TABLE   *SystemTable\r
+  )\r
+{\r
+  EFI_STATUS    Status;\r
+\r
+  //\r
+  // Install driver model protocol(s).\r
+  //\r
+  Status = EfiLibInstallDriverBindingComponentName2 (\r
+             ImageHandle,\r
+             SystemTable,\r
+             &gSataControllerDriverBinding,\r
+             ImageHandle,\r
+             &gSataControllerComponentName,\r
+             &gSataControllerComponentName2\r
+             );\r
+  ASSERT_EFI_ERROR (Status);\r
+\r
+  return Status;\r
+}\r
+\r
+/**\r
+  Supported function of Driver Binding protocol for this driver.\r
+  Test to see if this driver supports ControllerHandle.\r
+\r
+  @param This                   Protocol instance pointer.\r
+  @param Controller             Handle of device to test.\r
+  @param RemainingDevicePath    A pointer to the device path.\r
+                                it should be ignored by device driver.\r
+\r
+  @retval EFI_SUCCESS           This driver supports this device.\r
+  @retval EFI_ALREADY_STARTED   This driver is already running on this device.\r
+  @retval other                 This driver does not support this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerSupported (\r
+  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     Controller,\r
+  IN EFI_DEVICE_PATH_PROTOCOL       *RemainingDevicePath\r
+  )\r
+{\r
+  EFI_STATUS            Status;\r
+  EFI_PCI_IO_PROTOCOL   *PciIo;\r
+  PCI_TYPE00            PciData;\r
+\r
+  //\r
+  // Attempt to open PCI I/O Protocol\r
+  //\r
+  Status = gBS->OpenProtocol (\r
+                  Controller,\r
+                  &gEfiPciIoProtocolGuid,\r
+                  (VOID **) &PciIo,\r
+                  This->DriverBindingHandle,\r
+                  Controller,\r
+                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  //\r
+  // Now further check the PCI header: Base Class (offset 0x0B) and\r
+  // Sub Class (offset 0x0A). This controller should be an SATA controller\r
+  //\r
+  Status = PciIo->Pci.Read (\r
+                        PciIo,\r
+                        EfiPciIoWidthUint8,\r
+                        PCI_CLASSCODE_OFFSET,\r
+                        sizeof (PciData.Hdr.ClassCode),\r
+                        PciData.Hdr.ClassCode\r
+                        );\r
+  if (EFI_ERROR (Status)) {\r
+    return EFI_UNSUPPORTED;\r
+  }\r
+\r
+  if (IS_PCI_IDE (&PciData) || IS_PCI_SATADPA (&PciData)) {\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  return EFI_UNSUPPORTED;\r
+}\r
+\r
+/**\r
+  This routine is called right after the .Supported() called and \r
+  Start this driver on ControllerHandle.\r
+\r
+  @param This                   Protocol instance pointer.\r
+  @param Controller             Handle of device to bind driver to.\r
+  @param RemainingDevicePath    A pointer to the device path.\r
+                                it should be ignored by device driver.\r
+\r
+  @retval EFI_SUCCESS           This driver is added to this device.\r
+  @retval EFI_ALREADY_STARTED   This driver is already running on this device.\r
+  @retval other                 Some error occurs when binding this driver to this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerStart (\r
+  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     Controller,\r
+  IN EFI_DEVICE_PATH_PROTOCOL       *RemainingDevicePath\r
+  )\r
+{\r
+  EFI_STATUS                        Status;\r
+  EFI_PCI_IO_PROTOCOL               *PciIo;\r
+  PCI_TYPE00                        PciData;\r
+  EFI_SATA_CONTROLLER_PRIVATE_DATA  *Private;\r
+  UINT32                            Data32;\r
+  UINTN                             TotalCount;\r
+\r
+  DEBUG ((EFI_D_INFO, "SataControllerStart start\n"));\r
+\r
+  Private = NULL;\r
+\r
+  //\r
+  // Now test and open PCI I/O Protocol\r
+  //\r
+  Status = gBS->OpenProtocol (\r
+                  Controller,\r
+                  &gEfiPciIoProtocolGuid,\r
+                  (VOID **) &PciIo,\r
+                  This->DriverBindingHandle,\r
+                  Controller,\r
+                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    DEBUG ((EFI_D_ERROR, "SataControllerStart error. return status = %r\n", Status));\r
+    return Status;\r
+  }\r
+\r
+  //\r
+  // Allocate Sata Private Data structure\r
+  //\r
+  Private = AllocateZeroPool (sizeof (EFI_SATA_CONTROLLER_PRIVATE_DATA));\r
+  if (Private == NULL) {\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto Done;\r
+  }\r
+\r
+  //\r
+  // Initialize Sata Private Data\r
+  //\r
+  Private->Signature = SATA_CONTROLLER_SIGNATURE;\r
+  Private->PciIo     = PciIo;\r
+  Private->IdeInit.GetChannelInfo = IdeInitGetChannelInfo;\r
+  Private->IdeInit.NotifyPhase    = IdeInitNotifyPhase;\r
+  Private->IdeInit.SubmitData     = IdeInitSubmitData;\r
+  Private->IdeInit.DisqualifyMode = IdeInitDisqualifyMode;\r
+  Private->IdeInit.CalculateMode  = IdeInitCalculateMode;\r
+  Private->IdeInit.SetTiming      = IdeInitSetTiming;\r
+  Private->IdeInit.EnumAll        = SATA_ENUMER_ALL;\r
+\r
+  Status = PciIo->Pci.Read (\r
+                        PciIo,\r
+                        EfiPciIoWidthUint8,\r
+                        PCI_CLASSCODE_OFFSET,\r
+                        sizeof (PciData.Hdr.ClassCode),\r
+                        PciData.Hdr.ClassCode\r
+                        );\r
+  ASSERT_EFI_ERROR (Status);\r
+\r
+  if (IS_PCI_IDE (&PciData)) {\r
+    Private->IdeInit.ChannelCount = IDE_MAX_CHANNEL;\r
+    Private->DeviceCount          = IDE_MAX_DEVICES;\r
+  } else if (IS_PCI_SATADPA (&PciData)) {\r
+    //\r
+    // Read Host Capability Register(CAP) to get Number of Ports(NPS) and Supports Port Multiplier(SPM)\r
+    //   NPS is 0's based value indicating the maximum number of ports supported by the HBA silicon.\r
+    //   A maximum of 32 ports can be supported. A value of '0h', indicating one port, is the minimum requirement.\r
+    //\r
+    Data32 = AhciReadReg (PciIo, R_AHCI_CAP);\r
+    Private->IdeInit.ChannelCount = (UINT8) ((Data32 & B_AHCI_CAP_NPS) + 1);\r
+    Private->DeviceCount          = AHCI_MAX_DEVICES;\r
+    if ((Data32 & B_AHCI_CAP_SPM) == B_AHCI_CAP_SPM) {\r
+      Private->DeviceCount = AHCI_MULTI_MAX_DEVICES;\r
+    }\r
+  }\r
+\r
+  TotalCount = (UINTN) (Private->IdeInit.ChannelCount) * (UINTN) (Private->DeviceCount);\r
+  Private->DisqualifiedModes = AllocateZeroPool ((sizeof (EFI_ATA_COLLECTIVE_MODE)) * TotalCount);\r
+  if (Private->DisqualifiedModes == NULL) {\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto Done;\r
+  }\r
+\r
+  Private->IdentifyData = AllocateZeroPool ((sizeof (EFI_IDENTIFY_DATA)) * TotalCount);\r
+  if (Private->IdentifyData == NULL) {\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto Done;\r
+  }\r
+\r
+  Private->IdentifyValid = AllocateZeroPool ((sizeof (BOOLEAN)) * TotalCount);\r
+  if (Private->IdentifyValid == NULL) {\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto Done;\r
+  }\r
+\r
+  //\r
+  // Install IDE Controller Init Protocol to this instance\r
+  //\r
+  Status = gBS->InstallMultipleProtocolInterfaces (\r
+                  &Controller,\r
+                  &gEfiIdeControllerInitProtocolGuid,\r
+                  &(Private->IdeInit),\r
+                  NULL\r
+                  );\r
+\r
+Done:\r
+  if (EFI_ERROR (Status)) {\r
+\r
+    gBS->CloseProtocol (\r
+          Controller,\r
+          &gEfiPciIoProtocolGuid,\r
+          This->DriverBindingHandle,\r
+          Controller\r
+          );\r
+    if (Private != NULL) {\r
+      if (Private->DisqualifiedModes != NULL) {\r
+        FreePool (Private->DisqualifiedModes);\r
+      }\r
+      if (Private->IdentifyData != NULL) {\r
+        FreePool (Private->IdentifyData);\r
+      }\r
+      if (Private->IdentifyValid != NULL) {\r
+        FreePool (Private->IdentifyValid);\r
+      }\r
+      FreePool (Private);\r
+    }\r
+  }\r
+\r
+  DEBUG ((EFI_D_INFO, "SataControllerStart end with %r\n", Status));\r
+\r
+  return Status;\r
+}\r
+\r
+/**\r
+  Stop this driver on ControllerHandle.\r
+\r
+  @param This               A pointer to the EFI_DRIVER_BINDING_PROTOCOL instance.\r
+  @param Controller         A handle to the device being stopped.\r
+  @param NumberOfChildren   The number of child device handles in ChildHandleBuffer.\r
+  @param ChildHandleBuffer  An array of child handles to be freed. \r
+\r
+  @retval EFI_SUCCESS       This driver is removed from this device.\r
+  @retval other             Some error occurs when removing this driver from this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerStop (\r
+  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     Controller,\r
+  IN UINTN                          NumberOfChildren,\r
+  IN EFI_HANDLE                     *ChildHandleBuffer\r
+  )\r
+{\r
+  EFI_STATUS                        Status;\r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL  *IdeInit;\r
+  EFI_SATA_CONTROLLER_PRIVATE_DATA  *Private;\r
+\r
+  //\r
+  // Open the produced protocol\r
+  //\r
+  Status = gBS->OpenProtocol (\r
+                  Controller,\r
+                  &gEfiIdeControllerInitProtocolGuid,\r
+                  (VOID **) &IdeInit,\r
+                  This->DriverBindingHandle,\r
+                  Controller,\r
+                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    return EFI_UNSUPPORTED;\r
+  }\r
+\r
+  Private = SATA_CONTROLLER_PRIVATE_DATA_FROM_THIS (IdeInit);\r
+  ASSERT (Private != NULL);\r
+\r
+  //\r
+  // Uninstall the IDE Controller Init Protocol from this instance\r
+  //\r
+  Status = gBS->UninstallMultipleProtocolInterfaces (\r
+                  Controller,\r
+                  &gEfiIdeControllerInitProtocolGuid,\r
+                  &(Private->IdeInit),\r
+                  NULL\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  if (Private != NULL) {\r
+    if (Private->DisqualifiedModes != NULL) {\r
+      FreePool (Private->DisqualifiedModes);\r
+    }\r
+    if (Private->IdentifyData != NULL) {\r
+      FreePool (Private->IdentifyData);\r
+    }\r
+    if (Private->IdentifyValid != NULL) {\r
+      FreePool (Private->IdentifyValid);\r
+    }\r
+    FreePool (Private);\r
+  }\r
+\r
+  //\r
+  // Close protocols opened by Sata Controller driver\r
+  //\r
+  return gBS->CloseProtocol (\r
+                Controller,\r
+                &gEfiPciIoProtocolGuid,\r
+                This->DriverBindingHandle,\r
+                Controller\r
+                );\r
+}\r
+\r
+/**\r
+  Calculate the flat array subscript of a (Channel, Device) pair.\r
+\r
+  @param[in] Private  The private data structure corresponding to the\r
+                      SATA controller that attaches the device for\r
+                      which the flat array subscript is being\r
+                      calculated.\r
+\r
+  @param[in] Channel  The channel (ie. port) number on the SATA\r
+                      controller that the device is attached to.\r
+\r
+  @param[in] Device   The device number on the channel.\r
+\r
+  @return  The flat array subscript suitable for indexing DisqualifiedModes,\r
+           IdentifyData, and IdentifyValid.\r
+**/\r
+STATIC\r
+UINTN\r
+FlatDeviceIndex (\r
+  IN CONST EFI_SATA_CONTROLLER_PRIVATE_DATA  *Private,\r
+  IN UINTN                                   Channel,\r
+  IN UINTN                                   Device\r
+  )\r
+{\r
+  ASSERT (Private != NULL);\r
+  ASSERT (Channel < Private->IdeInit.ChannelCount);\r
+  ASSERT (Device < Private->DeviceCount);\r
+\r
+  return Channel * Private->DeviceCount + Device;\r
+}\r
+\r
+//\r
+// Interface functions of IDE_CONTROLLER_INIT protocol\r
+//\r
+/**\r
+  Returns the information about the specified IDE channel.\r
+  \r
+  This function can be used to obtain information about a particular IDE channel.\r
+  The driver entity uses this information during the enumeration process. \r
+  \r
+  If Enabled is set to FALSE, the driver entity will not scan the channel. Note \r
+  that it will not prevent an operating system driver from scanning the channel.\r
+  \r
+  For most of today's controllers, MaxDevices will either be 1 or 2. For SATA \r
+  controllers, this value will always be 1. SATA configurations can contain SATA \r
+  port multipliers. SATA port multipliers behave like SATA bridges and can support\r
+  up to 16 devices on the other side. If a SATA port out of the IDE controller \r
+  is connected to a port multiplier, MaxDevices will be set to the number of SATA \r
+  devices that the port multiplier supports. Because today's port multipliers \r
+  support up to fifteen SATA devices, this number can be as large as fifteen. The IDE  \r
+  bus driver is required to scan for the presence of port multipliers behind an SATA \r
+  controller and enumerate up to MaxDevices number of devices behind the port \r
+  multiplier.    \r
+  \r
+  In this context, the devices behind a port multiplier constitute a channel.  \r
+  \r
+  @param[in]  This                The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in]  Channel             Zero-based channel number.\r
+  @param[out] Enabled             TRUE if this channel is enabled.  Disabled channels \r
+                                  are not scanned to see if any devices are present.\r
+  @param[out] MaxDevices          The maximum number of IDE devices that the bus driver\r
+                                  can expect on this channel.  For the ATA/ATAPI \r
+                                  specification, version 6, this number will either be \r
+                                  one or two. For Serial ATA (SATA) configurations with a \r
+                                  port multiplier, this number can be as large as fifteen.\r
+\r
+  @retval EFI_SUCCESS             Information was returned without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitGetChannelInfo (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  OUT BOOLEAN                           *Enabled,\r
+  OUT UINT8                             *MaxDevices\r
+  )\r
+{\r
+  EFI_SATA_CONTROLLER_PRIVATE_DATA  *Private;\r
+  Private = SATA_CONTROLLER_PRIVATE_DATA_FROM_THIS (This);\r
+  ASSERT (Private != NULL);\r
+\r
+  if (Channel < This->ChannelCount) {\r
+    *Enabled = TRUE;\r
+    *MaxDevices = Private->DeviceCount;\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  *Enabled = FALSE;\r
+  return EFI_INVALID_PARAMETER;\r
+}\r
+\r
+/**\r
+  The notifications from the driver entity that it is about to enter a certain\r
+  phase of the IDE channel enumeration process.\r
+  \r
+  This function can be used to notify the IDE controller driver to perform \r
+  specific actions, including any chipset-specific initialization, so that the \r
+  chipset is ready to enter the next phase. Seven notification points are defined \r
+  at this time. \r
+  \r
+  More synchronization points may be added as required in the future.  \r
+\r
+  @param[in] This                 The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL\r
+                                  instance.\r
+  @param[in] Phase                The phase during enumeration.\r
+  @param[in] Channel              Zero-based channel number.\r
+\r
+  @retval EFI_SUCCESS             The notification was accepted without any errors.\r
+  @retval EFI_UNSUPPORTED         Phase is not supported.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_NOT_READY           This phase cannot be entered at this time; for \r
+                                  example, an attempt was made to enter a Phase \r
+                                  without having entered one or more previous \r
+                                  Phase.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitNotifyPhase (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN EFI_IDE_CONTROLLER_ENUM_PHASE      Phase,\r
+  IN UINT8                              Channel\r
+  )\r
+{\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  Submits the device information to the IDE controller driver.\r
+\r
+  This function is used by the driver entity to pass detailed information about \r
+  a particular device to the IDE controller driver. The driver entity obtains \r
+  this information by issuing an ATA or ATAPI IDENTIFY_DEVICE command. IdentifyData\r
+  is the pointer to the response data buffer. The IdentifyData buffer is owned \r
+  by the driver entity, and the IDE controller driver must make a local copy \r
+  of the entire buffer or parts of the buffer as needed. The original IdentifyData \r
+  buffer pointer may not be valid when\r
+  \r
+    - EFI_IDE_CONTROLLER_INIT_PROTOCOL.CalculateMode() or\r
+    - EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode() is called at a later point.\r
+    \r
+  The IDE controller driver may consult various fields of EFI_IDENTIFY_DATA to \r
+  compute the optimum mode for the device. These fields are not limited to the \r
+  timing information. For example, an implementation of the IDE controller driver \r
+  may examine the vendor and type/mode field to match known bad drives.  \r
+  \r
+  The driver entity may submit drive information in any order, as long as it \r
+  submits information for all the devices belonging to the enumeration group \r
+  before EFI_IDE_CONTROLLER_INIT_PROTOCOL.CalculateMode() is called for any device\r
+  in that enumeration group. If a device is absent, EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData()\r
+  should be called with IdentifyData set to NULL.  The IDE controller driver may \r
+  not have any other mechanism to know whether a device is present or not. Therefore, \r
+  setting IdentifyData to NULL does not constitute an error condition. \r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData() can be called only once for a \r
+  given (Channel, Device) pair.  \r
+    \r
+  @param[in] This                 A pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in] Channel              Zero-based channel number.\r
+  @param[in] Device               Zero-based device number on the Channel.\r
+  @param[in] IdentifyData         The device's response to the ATA IDENTIFY_DEVICE command.\r
+\r
+  @retval EFI_SUCCESS             The information was accepted without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitSubmitData (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  IN EFI_IDENTIFY_DATA                  *IdentifyData\r
+  )\r
+{\r
+  EFI_SATA_CONTROLLER_PRIVATE_DATA  *Private;\r
+  UINTN                             DeviceIndex;\r
+\r
+  Private = SATA_CONTROLLER_PRIVATE_DATA_FROM_THIS (This);\r
+  ASSERT (Private != NULL);\r
+\r
+  if ((Channel >= This->ChannelCount) || (Device >= Private->DeviceCount)) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  DeviceIndex = FlatDeviceIndex (Private, Channel, Device);\r
+\r
+  //\r
+  // Make a local copy of device's IdentifyData and mark the valid flag\r
+  //\r
+  if (IdentifyData != NULL) {\r
+    CopyMem (\r
+      &(Private->IdentifyData[DeviceIndex]),\r
+      IdentifyData,\r
+      sizeof (EFI_IDENTIFY_DATA)\r
+      );\r
+\r
+    Private->IdentifyValid[DeviceIndex] = TRUE;\r
+  } else {\r
+    Private->IdentifyValid[DeviceIndex] = FALSE;\r
+  }\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  Disqualifies specific modes for an IDE device.\r
+\r
+  This function allows the driver entity or other drivers (such as platform \r
+  drivers) to reject certain timing modes and request the IDE controller driver\r
+  to recalculate modes. This function allows the driver entity and the IDE \r
+  controller driver to negotiate the timings on a per-device basis. This function \r
+  is useful in the case of drives that lie about their capabilities. An example \r
+  is when the IDE device fails to accept the timing modes that are calculated \r
+  by the IDE controller driver based on the response to the Identify Drive command.\r
+\r
+  If the driver entity does not want to limit the ATA timing modes and leave that \r
+  decision to the IDE controller driver, it can either not call this function for \r
+  the given device or call this function and set the Valid flag to FALSE for all \r
+  modes that are listed in EFI_ATA_COLLECTIVE_MODE.\r
+  \r
+  The driver entity may disqualify modes for a device in any order and any number \r
+  of times.\r
+  \r
+  This function can be called multiple times to invalidate multiple modes of the \r
+  same type (e.g., Programmed Input/Output [PIO] modes 3 and 4). See the ATA/ATAPI \r
+  specification for more information on PIO modes.  \r
+  \r
+  For Serial ATA (SATA) controllers, this member function can be used to disqualify\r
+  a higher transfer rate mode on a given channel. For example, a platform driver\r
+  may inform the IDE controller driver to not use second-generation (Gen2) speeds \r
+  for a certain SATA drive.\r
+  \r
+  @param[in] This                 The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in] Channel              The zero-based channel number.\r
+  @param[in] Device               The zero-based device number on the Channel.\r
+  @param[in] BadModes             The modes that the device does not support and that\r
+                                  should be disqualified.\r
+\r
+  @retval EFI_SUCCESS             The modes were accepted without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid.\r
+  @retval EFI_INVALID_PARAMETER   IdentifyData is NULL.\r
+                                \r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitDisqualifyMode (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  IN EFI_ATA_COLLECTIVE_MODE            *BadModes\r
+  )\r
+{\r
+  EFI_SATA_CONTROLLER_PRIVATE_DATA  *Private;\r
+  UINTN                             DeviceIndex;\r
+\r
+  Private = SATA_CONTROLLER_PRIVATE_DATA_FROM_THIS (This);\r
+  ASSERT (Private != NULL);\r
+\r
+  if ((Channel >= This->ChannelCount) || (BadModes == NULL) || (Device >= Private->DeviceCount)) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  DeviceIndex = FlatDeviceIndex (Private, Channel, Device);\r
+\r
+  //\r
+  // Record the disqualified modes per channel per device. From ATA/ATAPI spec,\r
+  // if a mode is not supported, the modes higher than it is also not supported.\r
+  //\r
+  CopyMem (\r
+    &(Private->DisqualifiedModes[DeviceIndex]),\r
+    BadModes,\r
+    sizeof (EFI_ATA_COLLECTIVE_MODE)\r
+    );\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  Returns the information about the optimum modes for the specified IDE device.\r
+\r
+  This function is used by the driver entity to obtain the optimum ATA modes for\r
+  a specific device.  The IDE controller driver takes into account the following \r
+  while calculating the mode:\r
+    - The IdentifyData inputs to EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData()\r
+    - The BadModes inputs to EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode()\r
+\r
+  The driver entity is required to call EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData() \r
+  for all the devices that belong to an enumeration group before calling \r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL.CalculateMode() for any device in the same group.  \r
+  \r
+  The IDE controller driver will use controller- and possibly platform-specific \r
+  algorithms to arrive at SupportedModes.  The IDE controller may base its \r
+  decision on user preferences and other considerations as well. This function \r
+  may be called multiple times because the driver entity may renegotiate the mode \r
+  with the IDE controller driver using EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode().\r
+    \r
+  The driver entity may collect timing information for various devices in any \r
+  order. The driver entity is responsible for making sure that all the dependencies\r
+  are satisfied. For example, the SupportedModes information for device A that \r
+  was previously returned may become stale after a call to \r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode() for device B.\r
+  \r
+  The buffer SupportedModes is allocated by the callee because the caller does \r
+  not necessarily know the size of the buffer. The type EFI_ATA_COLLECTIVE_MODE \r
+  is defined in a way that allows for future extensibility and can be of variable \r
+  length. This memory pool should be deallocated by the caller when it is no \r
+  longer necessary.  \r
+  \r
+  The IDE controller driver for a Serial ATA (SATA) controller can use this \r
+  member function to force a lower speed (first-generation [Gen1] speeds on a \r
+  second-generation [Gen2]-capable hardware).  The IDE controller driver can \r
+  also allow the driver entity to stay with the speed that has been negotiated \r
+  by the physical layer.\r
+  \r
+  @param[in]  This                The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in]  Channel             A zero-based channel number.\r
+  @param[in]  Device              A zero-based device number on the Channel.\r
+  @param[out] SupportedModes      The optimum modes for the device.\r
+\r
+  @retval EFI_SUCCESS             SupportedModes was returned.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid. \r
+  @retval EFI_INVALID_PARAMETER   SupportedModes is NULL.\r
+  @retval EFI_NOT_READY           Modes cannot be calculated due to a lack of \r
+                                  data.  This error may happen if \r
+                                  EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData() \r
+                                  and EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyData() \r
+                                  were not called for at least one drive in the \r
+                                  same enumeration group.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitCalculateMode (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  OUT EFI_ATA_COLLECTIVE_MODE           **SupportedModes\r
+  )\r
+{\r
+  EFI_SATA_CONTROLLER_PRIVATE_DATA  *Private;\r
+  EFI_IDENTIFY_DATA                 *IdentifyData;\r
+  BOOLEAN                           IdentifyValid;\r
+  EFI_ATA_COLLECTIVE_MODE           *DisqualifiedModes;\r
+  UINT16                            SelectedMode;\r
+  EFI_STATUS                        Status;\r
+  UINTN                             DeviceIndex;\r
+\r
+  Private = SATA_CONTROLLER_PRIVATE_DATA_FROM_THIS (This);\r
+  ASSERT (Private != NULL);\r
+\r
+  if ((Channel >= This->ChannelCount) || (SupportedModes == NULL) || (Device >= Private->DeviceCount)) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  *SupportedModes = AllocateZeroPool (sizeof (EFI_ATA_COLLECTIVE_MODE));\r
+  if (*SupportedModes == NULL) {\r
+    ASSERT (*SupportedModes != NULL);\r
+    return EFI_OUT_OF_RESOURCES;\r
+  }\r
+\r
+  DeviceIndex = FlatDeviceIndex (Private, Channel, Device);\r
+\r
+  IdentifyData = &(Private->IdentifyData[DeviceIndex]);\r
+  IdentifyValid = Private->IdentifyValid[DeviceIndex];\r
+  DisqualifiedModes = &(Private->DisqualifiedModes[DeviceIndex]);\r
+\r
+  //\r
+  // Make sure we've got the valid identify data of the device from SubmitData()\r
+  //\r
+  if (!IdentifyValid) {\r
+    FreePool (*SupportedModes);\r
+    return EFI_NOT_READY;\r
+  }\r
+\r
+  Status = CalculateBestPioMode (\r
+            IdentifyData,\r
+            (DisqualifiedModes->PioMode.Valid ? ((UINT16 *) &(DisqualifiedModes->PioMode.Mode)) : NULL),\r
+            &SelectedMode\r
+            );\r
+  if (!EFI_ERROR (Status)) {\r
+    (*SupportedModes)->PioMode.Valid = TRUE;\r
+    (*SupportedModes)->PioMode.Mode = SelectedMode;\r
+\r
+  } else {\r
+    (*SupportedModes)->PioMode.Valid = FALSE;\r
+  }\r
+  DEBUG ((EFI_D_INFO, "IdeInitCalculateMode: PioMode = %x\n", (*SupportedModes)->PioMode.Mode));\r
+\r
+  Status = CalculateBestUdmaMode (\r
+            IdentifyData,\r
+            (DisqualifiedModes->UdmaMode.Valid ? ((UINT16 *) &(DisqualifiedModes->UdmaMode.Mode)) : NULL),\r
+            &SelectedMode\r
+            );\r
+\r
+  if (!EFI_ERROR (Status)) {\r
+    (*SupportedModes)->UdmaMode.Valid = TRUE;\r
+    (*SupportedModes)->UdmaMode.Mode  = SelectedMode;\r
+\r
+  } else {\r
+    (*SupportedModes)->UdmaMode.Valid = FALSE;\r
+  }\r
+  DEBUG ((EFI_D_INFO, "IdeInitCalculateMode: UdmaMode = %x\n", (*SupportedModes)->UdmaMode.Mode));\r
+\r
+  //\r
+  // The modes other than PIO and UDMA are not supported\r
+  //\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  Commands the IDE controller driver to program the IDE controller hardware\r
+  so that the specified device can operate at the specified mode.\r
+\r
+  This function is used by the driver entity to instruct the IDE controller \r
+  driver to program the IDE controller hardware to the specified modes. This \r
+  function can be called only once for a particular device. For a Serial ATA \r
+  (SATA) Advanced Host Controller Interface (AHCI) controller, no controller-\r
+  specific programming may be required.\r
+\r
+  @param[in] This                 Pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in] Channel              Zero-based channel number.\r
+  @param[in] Device               Zero-based device number on the Channel.\r
+  @param[in] Modes                The modes to set.\r
+\r
+  @retval EFI_SUCCESS             The command was accepted without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid.\r
+  @retval EFI_NOT_READY           Modes cannot be set at this time due to lack of data.\r
+  @retval EFI_DEVICE_ERROR        Modes cannot be set due to hardware failure.\r
+                                  The driver entity should not use this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitSetTiming (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  IN EFI_ATA_COLLECTIVE_MODE            *Modes\r
+  )\r
+{\r
+  return EFI_SUCCESS;\r
+}\r
+\r
diff --git a/MdeModulePkg/Bus/Pci/SataControllerDxe/SataController.h b/MdeModulePkg/Bus/Pci/SataControllerDxe/SataController.h
new file mode 100644 (file)
index 0000000..f7db3b8
--- /dev/null
@@ -0,0 +1,536 @@
+/** @file\r
+  Header file for Sata Controller driver.\r
+\r
+  Copyright (c) 2011 - 2016, 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
+  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
+#ifndef _SATA_CONTROLLER_H_\r
+#define _SATA_CONTROLLER_H_\r
+\r
+#include <Uefi.h>\r
+\r
+#include <IndustryStandard/Pci.h>\r
+\r
+#include <Protocol/ComponentName.h>\r
+#include <Protocol/DriverBinding.h>\r
+#include <Protocol/PciIo.h>\r
+#include <Protocol/IdeControllerInit.h>\r
+\r
+#include <Library/BaseLib.h>\r
+#include <Library/BaseMemoryLib.h>\r
+#include <Library/DebugLib.h>\r
+#include <Library/UefiLib.h>\r
+#include <Library/UefiDriverEntryPoint.h>\r
+#include <Library/MemoryAllocationLib.h>\r
+#include <Library/UefiBootServicesTableLib.h>\r
+\r
+//\r
+// Global Variables definitions\r
+//\r
+extern EFI_DRIVER_BINDING_PROTOCOL  gSataControllerDriverBinding;\r
+extern EFI_COMPONENT_NAME_PROTOCOL  gSataControllerComponentName;\r
+extern EFI_COMPONENT_NAME2_PROTOCOL gSataControllerComponentName2;\r
+\r
+#define AHCI_BAR_INDEX 0x05\r
+#define R_AHCI_CAP 0x0\r
+#define   B_AHCI_CAP_NPS (BIT4 | BIT3 | BIT2 | BIT1 | BIT0) // Number of Ports\r
+#define   B_AHCI_CAP_SPM BIT17                              // Supports Port Multiplier\r
+\r
+///\r
+/// AHCI each channel can have up to 1 device\r
+///\r
+#define AHCI_MAX_DEVICES       0x01\r
+\r
+///\r
+/// AHCI each channel can have 15 devices in the presence of a multiplier\r
+///\r
+#define AHCI_MULTI_MAX_DEVICES 0x0F\r
+\r
+///\r
+/// IDE supports 2 channel max\r
+///\r
+#define IDE_MAX_CHANNEL 0x02\r
+\r
+///\r
+/// IDE supports 2 devices max\r
+///\r
+#define IDE_MAX_DEVICES 0x02\r
+\r
+#define SATA_ENUMER_ALL FALSE\r
+\r
+//\r
+// Sata Controller driver private data structure\r
+//\r
+#define SATA_CONTROLLER_SIGNATURE SIGNATURE_32('S','A','T','A')\r
+\r
+typedef struct _EFI_SATA_CONTROLLER_PRIVATE_DATA {\r
+  //\r
+  // Standard signature used to identify Sata Controller private data\r
+  //\r
+  UINT32                            Signature;\r
+\r
+  //\r
+  // Protocol instance of IDE_CONTROLLER_INIT produced by this driver\r
+  //\r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL  IdeInit;\r
+\r
+  //\r
+  // Copy of protocol pointers used by this driver\r
+  //\r
+  EFI_PCI_IO_PROTOCOL               *PciIo;\r
+\r
+  //\r
+  // The number of devices that are supported by this channel\r
+  //\r
+  UINT8                             DeviceCount;\r
+\r
+  //\r
+  // The highest disqulified mode for each attached device,\r
+  // From ATA/ATAPI spec, if a mode is not supported,\r
+  // the modes higher than it is also not supported\r
+  //\r
+  EFI_ATA_COLLECTIVE_MODE           *DisqualifiedModes;\r
+\r
+  //\r
+  // A copy of EFI_IDENTIFY_DATA data for each attached SATA device and its flag\r
+  //\r
+  EFI_IDENTIFY_DATA                 *IdentifyData;\r
+  BOOLEAN                           *IdentifyValid;\r
+} EFI_SATA_CONTROLLER_PRIVATE_DATA;\r
+\r
+#define SATA_CONTROLLER_PRIVATE_DATA_FROM_THIS(a) CR(a, EFI_SATA_CONTROLLER_PRIVATE_DATA, IdeInit, SATA_CONTROLLER_SIGNATURE)\r
+\r
+//\r
+// Driver binding functions declaration\r
+//\r
+/**\r
+  Supported function of Driver Binding protocol for this driver.\r
+  Test to see if this driver supports ControllerHandle.\r
+\r
+  @param This                   Protocol instance pointer.\r
+  @param Controller             Handle of device to test.\r
+  @param RemainingDevicePath    A pointer to the device path. Should be ignored by\r
+                                device driver.\r
+\r
+  @retval EFI_SUCCESS           This driver supports this device.\r
+  @retval EFI_ALREADY_STARTED   This driver is already running on this device.\r
+  @retval other                 This driver does not support this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerSupported (\r
+  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     Controller,\r
+  IN EFI_DEVICE_PATH_PROTOCOL       *RemainingDevicePath\r
+  );\r
+\r
+/**\r
+  This routine is called right after the .Supported() called and \r
+  Start this driver on ControllerHandle.\r
+\r
+  @param This                   Protocol instance pointer.\r
+  @param Controller             Handle of device to bind driver to.\r
+  @param RemainingDevicePath    A pointer to the device path. Should be ignored by\r
+                                device driver.\r
+\r
+  @retval EFI_SUCCESS           This driver is added to this device.\r
+  @retval EFI_ALREADY_STARTED   This driver is already running on this device.\r
+  @retval other                 Some error occurs when binding this driver to this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerStart (\r
+  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     Controller,\r
+  IN EFI_DEVICE_PATH_PROTOCOL       *RemainingDevicePath\r
+  );\r
+\r
+/**\r
+  Stop this driver on ControllerHandle.\r
+\r
+  @param This               A pointer to the EFI_DRIVER_BINDING_PROTOCOL instance.\r
+  @param Controller         A handle to the device being stopped.\r
+  @param NumberOfChildren   The number of child device handles in ChildHandleBuffer.\r
+  @param ChildHandleBuffer  An array of child handles to be freed. \r
+\r
+  @retval EFI_SUCCESS       This driver is removed from this device.\r
+  @retval other             Some error occurs when removing this driver from this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerStop (\r
+  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     Controller,\r
+  IN UINTN                          NumberOfChildren,\r
+  IN EFI_HANDLE                     *ChildHandleBuffer\r
+  );\r
+\r
+//\r
+// IDE controller init functions declaration\r
+//\r
+/**\r
+  Returns the information about the specified IDE channel.\r
+  \r
+  This function can be used to obtain information about a particular IDE channel.\r
+  The driver entity uses this information during the enumeration process. \r
+  \r
+  If Enabled is set to FALSE, the driver entity will not scan the channel. Note \r
+  that it will not prevent an operating system driver from scanning the channel.\r
+  \r
+  For most of today's controllers, MaxDevices will either be 1 or 2. For SATA \r
+  controllers, this value will always be 1. SATA configurations can contain SATA \r
+  port multipliers. SATA port multipliers behave like SATA bridges and can support\r
+  up to 16 devices on the other side. If a SATA port out of the IDE controller \r
+  is connected to a port multiplier, MaxDevices will be set to the number of SATA \r
+  devices that the port multiplier supports. Because today's port multipliers \r
+  support up to fifteen SATA devices, this number can be as large as fifteen. The IDE  \r
+  bus driver is required to scan for the presence of port multipliers behind an SATA \r
+  controller and enumerate up to MaxDevices number of devices behind the port \r
+  multiplier.    \r
+  \r
+  In this context, the devices behind a port multiplier constitute a channel.  \r
+  \r
+  @param[in]  This                The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in]  Channel             Zero-based channel number.\r
+  @param[out] Enabled             TRUE if this channel is enabled.  Disabled channels \r
+                                  are not scanned to see if any devices are present.\r
+  @param[out] MaxDevices          The maximum number of IDE devices that the bus driver\r
+                                  can expect on this channel.  For the ATA/ATAPI \r
+                                  specification, version 6, this number will either be \r
+                                  one or two. For Serial ATA (SATA) configurations with a \r
+                                  port multiplier, this number can be as large as fifteen.\r
+\r
+\r
+  @retval EFI_SUCCESS             Information was returned without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitGetChannelInfo (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  OUT BOOLEAN                           *Enabled,\r
+  OUT UINT8                             *MaxDevices\r
+  );\r
+\r
+/**\r
+  The notifications from the driver entity that it is about to enter a certain\r
+  phase of the IDE channel enumeration process.\r
+  \r
+  This function can be used to notify the IDE controller driver to perform \r
+  specific actions, including any chipset-specific initialization, so that the \r
+  chipset is ready to enter the next phase. Seven notification points are defined \r
+  at this time. \r
+  \r
+  More synchronization points may be added as required in the future.  \r
+\r
+  @param[in] This                 The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL\r
+                                  instance.\r
+  @param[in] Phase                The phase during enumeration.\r
+  @param[in] Channel              Zero-based channel number.\r
+\r
+  @retval EFI_SUCCESS             The notification was accepted without any errors.\r
+  @retval EFI_UNSUPPORTED         Phase is not supported.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_NOT_READY           This phase cannot be entered at this time; for \r
+                                  example, an attempt was made to enter a Phase \r
+                                  without having entered one or more previous \r
+                                  Phase.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitNotifyPhase (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN EFI_IDE_CONTROLLER_ENUM_PHASE      Phase,\r
+  IN UINT8                              Channel\r
+  );\r
+\r
+/**\r
+  Submits the device information to the IDE controller driver.\r
+\r
+  This function is used by the driver entity to pass detailed information about \r
+  a particular device to the IDE controller driver. The driver entity obtains \r
+  this information by issuing an ATA or ATAPI IDENTIFY_DEVICE command. IdentifyData\r
+  is the pointer to the response data buffer. The IdentifyData buffer is owned \r
+  by the driver entity, and the IDE controller driver must make a local copy \r
+  of the entire buffer or parts of the buffer as needed. The original IdentifyData \r
+  buffer pointer may not be valid when\r
+  \r
+    - EFI_IDE_CONTROLLER_INIT_PROTOCOL.CalculateMode() or\r
+    - EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode() is called at a later point.\r
+    \r
+  The IDE controller driver may consult various fields of EFI_IDENTIFY_DATA to \r
+  compute the optimum mode for the device. These fields are not limited to the \r
+  timing information. For example, an implementation of the IDE controller driver \r
+  may examine the vendor and type/mode field to match known bad drives.  \r
+  \r
+  The driver entity may submit drive information in any order, as long as it \r
+  submits information for all the devices belonging to the enumeration group \r
+  before EFI_IDE_CONTROLLER_INIT_PROTOCOL.CalculateMode() is called for any device\r
+  in that enumeration group. If a device is absent, EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData()\r
+  should be called with IdentifyData set to NULL.  The IDE controller driver may \r
+  not have any other mechanism to know whether a device is present or not. Therefore, \r
+  setting IdentifyData to NULL does not constitute an error condition. \r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData() can be called only once for a \r
+  given (Channel, Device) pair.  \r
+    \r
+  @param[in] This                 A pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in] Channel              Zero-based channel number.\r
+  @param[in] Device               Zero-based device number on the Channel.\r
+  @param[in] IdentifyData         The device's response to the ATA IDENTIFY_DEVICE command.\r
+\r
+  @retval EFI_SUCCESS             The information was accepted without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitSubmitData (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  IN EFI_IDENTIFY_DATA                  *IdentifyData\r
+  );\r
+\r
+/**\r
+  Disqualifies specific modes for an IDE device.\r
+\r
+  This function allows the driver entity or other drivers (such as platform \r
+  drivers) to reject certain timing modes and request the IDE controller driver\r
+  to recalculate modes. This function allows the driver entity and the IDE \r
+  controller driver to negotiate the timings on a per-device basis. This function \r
+  is useful in the case of drives that lie about their capabilities. An example \r
+  is when the IDE device fails to accept the timing modes that are calculated \r
+  by the IDE controller driver based on the response to the Identify Drive command.\r
+\r
+  If the driver entity does not want to limit the ATA timing modes and leave that \r
+  decision to the IDE controller driver, it can either not call this function for \r
+  the given device or call this function and set the Valid flag to FALSE for all \r
+  modes that are listed in EFI_ATA_COLLECTIVE_MODE.\r
+  \r
+  The driver entity may disqualify modes for a device in any order and any number \r
+  of times.\r
+  \r
+  This function can be called multiple times to invalidate multiple modes of the \r
+  same type (e.g., Programmed Input/Output [PIO] modes 3 and 4). See the ATA/ATAPI \r
+  specification for more information on PIO modes.  \r
+  \r
+  For Serial ATA (SATA) controllers, this member function can be used to disqualify\r
+  a higher transfer rate mode on a given channel. For example, a platform driver\r
+  may inform the IDE controller driver to not use second-generation (Gen2) speeds \r
+  for a certain SATA drive.\r
+  \r
+  @param[in] This                 The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in] Channel              The zero-based channel number.\r
+  @param[in] Device               The zero-based device number on the Channel.\r
+  @param[in] BadModes             The modes that the device does not support and that\r
+                                  should be disqualified.\r
+\r
+  @retval EFI_SUCCESS             The modes were accepted without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid.\r
+  @retval EFI_INVALID_PARAMETER   IdentifyData is NULL.\r
+                                \r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitDisqualifyMode (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  IN EFI_ATA_COLLECTIVE_MODE            *BadModes\r
+  );\r
+\r
+/**\r
+  Returns the information about the optimum modes for the specified IDE device.\r
+\r
+  This function is used by the driver entity to obtain the optimum ATA modes for\r
+  a specific device.  The IDE controller driver takes into account the following \r
+  while calculating the mode:\r
+    - The IdentifyData inputs to EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData()\r
+    - The BadModes inputs to EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode()\r
+\r
+  The driver entity is required to call EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData() \r
+  for all the devices that belong to an enumeration group before calling \r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL.CalculateMode() for any device in the same group.  \r
+  \r
+  The IDE controller driver will use controller- and possibly platform-specific \r
+  algorithms to arrive at SupportedModes.  The IDE controller may base its \r
+  decision on user preferences and other considerations as well. This function \r
+  may be called multiple times because the driver entity may renegotiate the mode \r
+  with the IDE controller driver using EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode().\r
+    \r
+  The driver entity may collect timing information for various devices in any \r
+  order. The driver entity is responsible for making sure that all the dependencies\r
+  are satisfied. For example, the SupportedModes information for device A that \r
+  was previously returned may become stale after a call to \r
+  EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyMode() for device B.\r
+  \r
+  The buffer SupportedModes is allocated by the callee because the caller does \r
+  not necessarily know the size of the buffer. The type EFI_ATA_COLLECTIVE_MODE \r
+  is defined in a way that allows for future extensibility and can be of variable \r
+  length. This memory pool should be deallocated by the caller when it is no \r
+  longer necessary.  \r
+  \r
+  The IDE controller driver for a Serial ATA (SATA) controller can use this \r
+  member function to force a lower speed (first-generation [Gen1] speeds on a \r
+  second-generation [Gen2]-capable hardware).  The IDE controller driver can \r
+  also allow the driver entity to stay with the speed that has been negotiated \r
+  by the physical layer.\r
+  \r
+  @param[in]  This                The pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in]  Channel             A zero-based channel number.\r
+  @param[in]  Device              A zero-based device number on the Channel.\r
+  @param[out] SupportedModes      The optimum modes for the device.\r
+\r
+  @retval EFI_SUCCESS             SupportedModes was returned.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid. \r
+  @retval EFI_INVALID_PARAMETER   SupportedModes is NULL.\r
+  @retval EFI_NOT_READY           Modes cannot be calculated due to a lack of \r
+                                  data.  This error may happen if \r
+                                  EFI_IDE_CONTROLLER_INIT_PROTOCOL.SubmitData() \r
+                                  and EFI_IDE_CONTROLLER_INIT_PROTOCOL.DisqualifyData() \r
+                                  were not called for at least one drive in the \r
+                                  same enumeration group.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitCalculateMode (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  OUT EFI_ATA_COLLECTIVE_MODE           **SupportedModes\r
+  );\r
+\r
+/**\r
+  Commands the IDE controller driver to program the IDE controller hardware\r
+  so that the specified device can operate at the specified mode.\r
+\r
+  This function is used by the driver entity to instruct the IDE controller \r
+  driver to program the IDE controller hardware to the specified modes. This \r
+  function can be called only once for a particular device. For a Serial ATA \r
+  (SATA) Advanced Host Controller Interface (AHCI) controller, no controller-\r
+  specific programming may be required.\r
+\r
+  @param[in] This                 Pointer to the EFI_IDE_CONTROLLER_INIT_PROTOCOL instance.\r
+  @param[in] Channel              Zero-based channel number.\r
+  @param[in] Device               Zero-based device number on the Channel.\r
+  @param[in] Modes                The modes to set.\r
+\r
+  @retval EFI_SUCCESS             The command was accepted without any errors.\r
+  @retval EFI_INVALID_PARAMETER   Channel is invalid (Channel >= ChannelCount).\r
+  @retval EFI_INVALID_PARAMETER   Device is invalid.\r
+  @retval EFI_NOT_READY           Modes cannot be set at this time due to lack of data.\r
+  @retval EFI_DEVICE_ERROR        Modes cannot be set due to hardware failure.\r
+                                  The driver entity should not use this device.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+IdeInitSetTiming (\r
+  IN EFI_IDE_CONTROLLER_INIT_PROTOCOL   *This,\r
+  IN UINT8                              Channel,\r
+  IN UINT8                              Device,\r
+  IN EFI_ATA_COLLECTIVE_MODE            *Modes\r
+  );\r
+\r
+//\r
+// Forward reference declaration\r
+//\r
+/**\r
+  Retrieves a Unicode string that is the user readable name of the UEFI Driver.\r
+\r
+  @param This           A pointer to the EFI_COMPONENT_NAME_PROTOCOL instance.\r
+  @param Language       A pointer to a three character ISO 639-2 language identifier.\r
+                        This is the language of the driver name that that the caller\r
+                        is requesting, and it must match one of the languages specified\r
+                        in SupportedLanguages.  The number of languages supported by a\r
+                        driver is up to the driver writer.\r
+  @param DriverName     A pointer to the Unicode string to return.  This Unicode string\r
+                        is the name of the driver specified by This in the language\r
+                        specified by Language.\r
+\r
+  @retval EFI_SUCCESS           The Unicode string for the Driver specified by This\r
+                                and the language specified by Language was returned\r
+                                in DriverName.\r
+  @retval EFI_INVALID_PARAMETER Language is NULL.\r
+  @retval EFI_INVALID_PARAMETER DriverName is NULL.\r
+  @retval EFI_UNSUPPORTED       The driver specified by This does not support the\r
+                                language specified by Language.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerComponentNameGetDriverName (\r
+  IN EFI_COMPONENT_NAME_PROTOCOL    *This,\r
+  IN CHAR8                          *Language,\r
+  OUT CHAR16                        **DriverName\r
+  );\r
+\r
+/**\r
+  Retrieves a Unicode string that is the user readable name of the controller\r
+  that is being managed by an UEFI Driver.\r
+\r
+  @param This                   A pointer to the EFI_COMPONENT_NAME_PROTOCOL instance.\r
+  @param ControllerHandle       The handle of a controller that the driver specified by\r
+                                This is managing.  This handle specifies the controller\r
+                                whose name is to be returned.\r
+  @param OPTIONAL   ChildHandle The handle of the child controller to retrieve the name\r
+                                of.  This is an optional parameter that may be NULL.  It\r
+                                will be NULL for device drivers.  It will also be NULL\r
+                                for a bus drivers that wish to retrieve the name of the\r
+                                bus controller.  It will not be NULL for a bus driver\r
+                                that wishes to retrieve the name of a child controller.\r
+  @param Language               A pointer to a three character ISO 639-2 language\r
+                                identifier.  This is the language of the controller name\r
+                                that that the caller is requesting, and it must match one\r
+                                of the languages specified in SupportedLanguages.  The\r
+                                number of languages supported by a driver is up to the\r
+                                driver writer.\r
+  @param ControllerName         A pointer to the Unicode string to return.  This Unicode\r
+                                string is the name of the controller specified by\r
+                                ControllerHandle and ChildHandle in the language\r
+                                specified by Language from the point of view of the\r
+                                driver specified by This.\r
+\r
+  @retval EFI_SUCCESS           The Unicode string for the user readable name in the\r
+                                language specified by Language for the driver\r
+                                specified by This was returned in DriverName.\r
+  @retval EFI_INVALID_PARAMETER ControllerHandle is not a valid EFI_HANDLE.\r
+  @retval EFI_INVALID_PARAMETER ChildHandle is not NULL and it is not a valid\r
+                                EFI_HANDLE.\r
+  @retval EFI_INVALID_PARAMETER Language is NULL.\r
+  @retval EFI_INVALID_PARAMETER ControllerName is NULL.\r
+  @retval EFI_UNSUPPORTED       The driver specified by This is not currently\r
+                                managing the controller specified by\r
+                                ControllerHandle and ChildHandle.\r
+  @retval EFI_UNSUPPORTED       The driver specified by This does not support the\r
+                                language specified by Language.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+SataControllerComponentNameGetControllerName (\r
+  IN EFI_COMPONENT_NAME_PROTOCOL    *This,\r
+  IN EFI_HANDLE                     ControllerHandle,\r
+  IN EFI_HANDLE                     ChildHandle OPTIONAL,\r
+  IN CHAR8                          *Language,\r
+  OUT CHAR16                        **ControllerName\r
+  );\r
+\r
+#endif\r
+\r
diff --git a/MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxe.inf b/MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxe.inf
new file mode 100644 (file)
index 0000000..e24b2a0
--- /dev/null
@@ -0,0 +1,58 @@
+## @file\r
+#  \r
+#    Component description file for the Sata Controller driver.\r
+#\r
+#  Copyright (c) 2011 - 2016, 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
+#  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
+[Defines]\r
+  INF_VERSION                    = 0x00010005\r
+  BASE_NAME                      = SataController\r
+  MODULE_UNI_FILE                = SataControllerDxe.uni\r
+  FILE_GUID                      = 820C59BB-274C-43B2-83EA-DAC673035A59\r
+  MODULE_TYPE                    = UEFI_DRIVER\r
+  VERSION_STRING                 = 1.0\r
+  ENTRY_POINT                    = InitializeSataControllerDriver\r
+\r
+#\r
+# The following information is for reference only and not required by the build tools.\r
+#\r
+#  VALID_ARCHITECTURES           = IA32 X64 IPF EBC ARM AARCH64\r
+#\r
+#  DRIVER_BINDING                =  gSataControllerDriverBinding                        \r
+#  COMPONENT_NAME                =  gSataControllerComponentName\r
+#  COMPONENT_NAME2               =  gSataControllerComponentName2\r
+#\r
+\r
+[Sources]\r
+  ComponentName.c\r
+  SataController.c\r
+  SataController.h\r
+\r
+[Packages]\r
+  MdePkg/MdePkg.dec\r
+\r
+[LibraryClasses]\r
+  UefiDriverEntryPoint\r
+  DebugLib\r
+  UefiLib\r
+  BaseLib\r
+  BaseMemoryLib\r
+  MemoryAllocationLib\r
+  UefiBootServicesTableLib\r
+\r
+[Protocols]\r
+  gEfiPciIoProtocolGuid                     ## TO_START\r
+  gEfiIdeControllerInitProtocolGuid         ## BY_START\r
+\r
+[UserExtensions.TianoCore."ExtraFiles"]\r
+  SataControllerDxeExtra.uni\r
+\r
diff --git a/MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxe.uni b/MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxe.uni
new file mode 100644 (file)
index 0000000..a956327
--- /dev/null
@@ -0,0 +1,22 @@
+// /** @file\r
+// The SataControllerDxe driver is responsible for managing the standard SATA controller.\r
+//\r
+// It consumes PciIo protocol and produces IdeControllerInit protocol for upper layer use.\r
+//\r
+// Copyright (c) 2016, Intel Corporation. All rights reserved.<BR>\r
+//\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
+// 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
+#string STR_MODULE_ABSTRACT             #language en-US "Responsible for managing the standard SATA controller"\r
+\r
+#string STR_MODULE_DESCRIPTION          #language en-US "Implements the IdeControllerInit protocol interface for upper layer use\n"\r
\ No newline at end of file
diff --git a/MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxeExtra.uni b/MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxeExtra.uni
new file mode 100644 (file)
index 0000000..ed25611
--- /dev/null
@@ -0,0 +1,20 @@
+// /** @file\r
+// SataControllerDxe Localized Strings and Content\r
+//\r
+// Copyright (c) 2016, Intel Corporation. All rights reserved.<BR>\r
+//\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
+// 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
+#string STR_PROPERTIES_MODULE_NAME \r
+#language en-US \r
+"SATA Controller DXE Driver"\r
+\r
+\r
index 8c0875be69b8c1e5450bb36010d2518ca241a16f..ce29eb96d13feea7aa4c099bda9c3da8d1ef02a2 100644 (file)
   MdeModulePkg/Bus/Pci/IdeBusPei/IdeBusPei.inf\r
   MdeModulePkg/Bus/Usb/UsbBusPei/UsbBusPei.inf\r
   MdeModulePkg/Bus/Usb/UsbBotPei/UsbBotPei.inf\r
+  MdeModulePkg/Bus/Pci/SataControllerDxe/SataControllerDxe.inf\r
   MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBusDxe.inf\r
   MdeModulePkg/Bus/Ata/AtaAtapiPassThru/AtaAtapiPassThru.inf\r
   MdeModulePkg/Bus/Scsi/ScsiBusDxe/ScsiBusDxe.inf\r