atapiextpassthru.c

来自「EFI BIOS是Intel提出的下一代的BIOS标准。这里上传的Edk源代码是」· C语言 代码 · 共 2,448 行 · 第 1/5 页

/*++

Copyright (c) 2004 - 2005, Intel Corporation                                                         
All rights reserved. This program and the accompanying materials                          
are licensed and made available under the terms and conditions of the BSD License         
which accompanies this distribution.  The full text of the license may be found at        
http://opensource.org/licenses/bsd-license.php                                            
                                                                                          
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             

Module Name:

    AtapiPassThru.c
    
Abstract: 
    

Revision History
--*/

#include "AtapiExtPassThru.h"

EFI_STATUS
EFIAPI
AtapiExtScsiPassThruDriverBindingSupported (
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,
  IN EFI_HANDLE                   Controller,
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath
  );

EFI_STATUS
EFIAPI
AtapiExtScsiPassThruDriverBindingStart (
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,
  IN EFI_HANDLE                   Controller,
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath
  );

EFI_STATUS
EFIAPI
AtapiExtScsiPassThruDriverBindingStop (
  IN  EFI_DRIVER_BINDING_PROTOCOL     *This,
  IN  EFI_HANDLE                      Controller,
  IN  UINTN                           NumberOfChildren,
  IN  EFI_HANDLE                      *ChildHandleBuffer
  );

STATIC  
EFI_STATUS
CheckSCSIRequestPacket (
  EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET      *Packet
  );

STATIC  
BOOLEAN
IsCommandValid (
  EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET   *Packet
  );

STATIC  
EFI_STATUS
SubmitBlockingIoCommand (
  ATAPI_EXT_SCSI_PASS_THRU_DEV                  *AtapiScsiPrivate,
  UINT32                                        Target,
  EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET    *Packet
  );

STATIC  
EFI_STATUS
RequestSenseCommand (
  ATAPI_EXT_SCSI_PASS_THRU_DEV    *AtapiScsiPrivate,
  UINT32                          Target,
  UINT64                          Timeout,
  VOID                            *SenseData,
  UINT8                           *SenseDataLength
  );

STATIC  
EFI_STATUS
AtapiPacketCommand (
  ATAPI_EXT_SCSI_PASS_THRU_DEV    *AtapiScsiPrivate,
  UINT32                          Target,
  UINT8                           *PacketCommand,
  VOID                            *Buffer,
  UINT32                          *ByteCount,
  DATA_DIRECTION                  Direction,
  UINT64                          TimeoutInMicroSeconds
  );

STATIC  
EFI_STATUS
AtapiPassThruPioReadWriteData (
  ATAPI_EXT_SCSI_PASS_THRU_DEV  *AtapiScsiPrivate,
  UINT16                        *Buffer,
  UINT32                        *ByteCount,
  DATA_DIRECTION                Direction,
  UINT64                        TimeoutInMicroSeconds
  );

STATIC
UINT8
ReadPortB (
  IN  EFI_PCI_IO_PROTOCOL   *PciIo,
  IN  UINT16                Port
  );

STATIC
UINT16
ReadPortW (
  IN  EFI_PCI_IO_PROTOCOL   *PciIo,
  IN  UINT16                Port
  );

STATIC
VOID
WritePortB (
  IN  EFI_PCI_IO_PROTOCOL   *PciIo,
  IN  UINT16                Port,
  IN  UINT8                 Data
  );

STATIC
VOID
WritePortW (
  IN  EFI_PCI_IO_PROTOCOL   *PciIo,
  IN  UINT16                Port,
  IN  UINT16                Data
  );

STATIC
EFI_STATUS
StatusDRQClear (
  ATAPI_EXT_SCSI_PASS_THRU_DEV        *AtapiScsiPrivate,
  UINT64                              TimeoutInMicroSeconds
  );

STATIC  
EFI_STATUS
AltStatusDRQClear (
  ATAPI_EXT_SCSI_PASS_THRU_DEV        *AtapiScsiPrivate,
  UINT64                              TimeoutInMicroSeconds
  );

STATIC  
EFI_STATUS
StatusDRQReady (
  ATAPI_EXT_SCSI_PASS_THRU_DEV        *AtapiScsiPrivate,
  UINT64                              TimeoutInMicroSeconds
  );

STATIC  
EFI_STATUS
StatusWaitForBSYClear (
  ATAPI_EXT_SCSI_PASS_THRU_DEV    *AtapiScsiPrivate,
  UINT64                          TimeoutInMicroSeconds
  );


STATIC  
EFI_STATUS
AltStatusWaitForBSYClear (
  ATAPI_EXT_SCSI_PASS_THRU_DEV    *AtapiScsiPrivate,
  UINT64                          TimeoutInMicroSeconds
  );

STATIC  
EFI_STATUS
StatusDRDYReady (
  ATAPI_EXT_SCSI_PASS_THRU_DEV     *AtapiScsiPrivate,
  UINT64                           TimeoutInMicroSeconds
  );

STATIC  
EFI_STATUS
AltStatusDRDYReady (
  ATAPI_EXT_SCSI_PASS_THRU_DEV     *AtapiScsiPrivate,
  UINT64                           TimeoutInMicroSeconds
  );

STATIC  
EFI_STATUS
AtapiPassThruCheckErrorStatus (
  ATAPI_EXT_SCSI_PASS_THRU_DEV        *AtapiScsiPrivate
  );


//
// IDE registers' fixed address
//
static IDE_BASE_REGISTERS   AtapiIoPortRegisters[2] = {
        {0x1f0, 0x1f1, 0x1f2, 0x1f3, 0x1f4, 0x1f5, 0x1f6, 0x1f7, 0x3f6, 0x3f7},
        {0x170, 0x171, 0x172, 0x173, 0x174, 0x175, 0x176, 0x177, 0x376, 0x377},
};

static SCSI_COMMAND_SET     EndTable = { 0xff, 0xff };

//
// This table contains all the supported ATAPI commands.
//
static SCSI_COMMAND_SET     SupportedATAPICommands[] = {
    OP_INQUIRY,                     DataIn,
    OP_LOAD_UNLOAD_CD,              NoData,
    OP_MECHANISM_STATUS,            DataIn,
    OP_MODE_SELECT_10,              DataOut,
    OP_MODE_SENSE_10,               DataIn,
    OP_PAUSE_RESUME,                NoData,
    OP_PLAY_AUDIO_10,               DataIn,
    OP_PLAY_AUDIO_MSF,              DataIn,
    OP_PLAY_CD,                     DataIn,
    OP_PLAY_CD_MSF,                 DataIn,
    OP_PREVENT_ALLOW_MEDIUM_REMOVAL,NoData,
    OP_READ_10,                     DataIn,
    OP_READ_12,                     DataIn,
    OP_READ_CAPACITY,               DataIn,
    OP_READ_CD,                     DataIn,
    OP_READ_CD_MSF,                 DataIn,
    OP_READ_HEADER,                 DataIn,
    OP_READ_SUB_CHANNEL,            DataIn,
    OP_READ_TOC,                    DataIn,
    OP_REQUEST_SENSE,               DataIn,
    OP_SCAN,                        NoData,
    OP_SEEK_10,                     NoData,
    OP_SET_CD_SPEED,                DataOut,
    OP_STOPPLAY_SCAN,               NoData,
    OP_START_STOP_UNIT,             NoData,
    OP_TEST_UNIT_READY,             NoData,
    OP_FORMAT_UNIT,                 DataOut,
    OP_READ_FORMAT_CAPACITIES,      DataIn,
    OP_VERIFY,                      DataOut,
    OP_WRITE_10,                    DataOut,
    OP_WRITE_12,                    DataOut,
    OP_WRITE_AND_VERIFY,            DataOut,
    0xff,                           0xff
};

static CHAR16               ControllerNameString[]  = L"ATAPI Controller";
static CHAR16               AtapiChannelString[]    = L"ATAPI Channel";

EFI_DRIVER_BINDING_PROTOCOL gAtapiExtScsiPassThruDriverBinding = {
  AtapiExtScsiPassThruDriverBindingSupported,
  AtapiExtScsiPassThruDriverBindingStart,
  AtapiExtScsiPassThruDriverBindingStop,
  0x10,
  NULL,
  NULL
};


EFI_DRIVER_ENTRY_POINT (AtapiExtScsiPassThruDriverEntryPoint)

EFI_STATUS
EFIAPI
AtapiExtScsiPassThruDriverEntryPoint (
  IN EFI_HANDLE         ImageHandle,
  IN EFI_SYSTEM_TABLE   *SystemTable
  )
/*++

Routine Description:

  Entry point for EFI drivers.

Arguments:

  ImageHandle - EFI_HANDLE
  SystemTable - EFI_SYSTEM_TABLE

Returns:

  EFI_SUCCESS
  Others 
  
--*/
{
  return EfiLibInstallAllDriverProtocols (
          ImageHandle,
          SystemTable,
          &gAtapiExtScsiPassThruDriverBinding,
          ImageHandle,
          &gAtapiScsiPassThruComponentName,
          NULL,
          NULL
          );
}

EFI_STATUS
EFIAPI
AtapiExtScsiPassThruDriverBindingSupported (
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,
  IN EFI_HANDLE                   Controller,
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath
  )
/*++
  
Routine Description:
  Test to see if this driver supports ControllerHandle. Any ControllerHandle
  that has gEfiPciIoProtocolGuid installed and is IDE Controller it will be supported.
    
Arguments:

  This                - Protocol instance pointer.
  Controller          - Handle of device to test
  RemainingDevicePath - Not used
    
Returns:
    EFI_STATUS
  
--*/
{
  EFI_STATUS          Status;
  EFI_PCI_IO_PROTOCOL *PciIo;
  PCI_TYPE00          Pci;

  //
  // Open the IO Abstraction(s) needed to perform the supported test
  //
  Status = gBS->OpenProtocol (
                  Controller,
                  &gEfiPciIoProtocolGuid,
                  (VOID **) &PciIo,
                  This->DriverBindingHandle,
                  Controller,
                  EFI_OPEN_PROTOCOL_BY_DRIVER
                  );
  if (EFI_ERROR (Status)) {
    return Status;
  }
  //
  // Use the PCI I/O Protocol to see if Controller is a IDE Controller that
  // can be managed by this driver.  Read the PCI Configuration Header
  // for this device.
  //
  Status = PciIo->Pci.Read (
                    PciIo,
                    EfiPciIoWidthUint32,
                    0,
                    sizeof (Pci) / sizeof (UINT32),
                    &Pci
                    );
  if (EFI_ERROR (Status)) {
    gBS->CloseProtocol (
           Controller,
           &gEfiPciIoProtocolGuid,
           This->DriverBindingHandle,
           Controller
           );
    return EFI_UNSUPPORTED;
  }

  if (Pci.Hdr.ClassCode[2] != PCI_CLASS_MASS_STORAGE || Pci.Hdr.ClassCode[1] != PCI_CLASS_IDE) {

    Status = EFI_UNSUPPORTED;
  }

  gBS->CloseProtocol (
         Controller,
         &gEfiPciIoProtocolGuid,
         This->DriverBindingHandle,
         Controller
         );

  return Status;
}

EFI_STATUS
EFIAPI
AtapiExtScsiPassThruDriverBindingStart (
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,
  IN EFI_HANDLE                   Controller,
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath
  )
/*++
  
Routine Description:
  Create handles for IDE channels specified by RemainingDevicePath.
  Install Extended SCSI Pass Thru Protocol onto each created handle.
  
Arguments:

  This                - Protocol instance pointer.
  Controller          - Handle of device to test
  RemainingDevicePath - Not used
    
Returns:
  EFI_STATUS
    
--*/
{
  EFI_STATUS          Status;
  EFI_PCI_IO_PROTOCOL *PciIo;

  PciIo = NULL;
  Status = gBS->OpenProtocol (
                  Controller,
                  &gEfiPciIoProtocolGuid,
                  (VOID **) &PciIo,
                  This->DriverBindingHandle,
                  Controller,
                  EFI_OPEN_PROTOCOL_BY_DRIVER
                  );
  if (EFI_ERROR (Status)) {
    return Status;
  }

  Status = PciIo->Attributes (
                    PciIo,
                    EfiPciIoAttributeOperationEnable,
                    EFI_PCI_IO_ATTRIBUTE_IDE_PRIMARY_IO | EFI_PCI_IO_ATTRIBUTE_IDE_SECONDARY_IO | EFI_PCI_DEVICE_ENABLE,
                    NULL
                    );
  if (EFI_ERROR (Status)) {
    goto Done;
  }

  //
  // Create Extended SCSI Pass Thru instance for the IDE channel.
  //
  Status = RegisterAtapiExtScsiPassThru (This, Controller, PciIo);

Done:
  if (EFI_ERROR (Status)) {
    if (PciIo) {
      PciIo->Attributes (
               PciIo,
               EfiPciIoAttributeOperationDisable,
               EFI_PCI_IO_ATTRIBUTE_IDE_PRIMARY_IO | EFI_PCI_IO_ATTRIBUTE_IDE_SECONDARY_IO | EFI_PCI_DEVICE_ENABLE,
               NULL
               );
    }

    gBS->CloseProtocol (
           Controller,
           &gEfiPciIoProtocolGuid,
           This->DriverBindingHandle,
           Controller
           );
  }

  return Status;
}

EFI_STATUS
EFIAPI
AtapiExtScsiPassThruDriverBindingStop (
  IN  EFI_DRIVER_BINDING_PROTOCOL     *This,
  IN  EFI_HANDLE                      Controller,
  IN  UINTN                           NumberOfChildren,
  IN  EFI_HANDLE                      *ChildHandleBuffer
  )
/*++
  
Routine Description:

  Stop this driver on ControllerHandle. Support stoping any child handles
  created by this driver.

Arguments:

  This              - Protocol instance pointer.
  Controller        - Handle of device to stop driver on
  NumberOfChildren  - Number of Children in the ChildHandleBuffer
  ChildHandleBuffer - List of handles for the children we need to stop.
  
Returns:

  EFI_STATUS
  
--*/
{
  EFI_STATUS                       Status;
  EFI_EXT_SCSI_PASS_THRU_PROTOCOL  *ScsiPassThru;
  ATAPI_EXT_SCSI_PASS_THRU_DEV     *AtapiScsiPrivate;

  Status = gBS->OpenProtocol (
                  Controller,
                  &gEfiExtScsiPassThruProtocolGuid,
                  (VOID **) &ScsiPassThru,
                  This->DriverBindingHandle,
                  Controller,
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL
                  );
  if (EFI_ERROR (Status)) {
    return Status;
  }

  AtapiScsiPrivate = ATAPI_EXT_SCSI_PASS_THRU_DEV_FROM_THIS (ScsiPassThru);

  Status = gBS->UninstallProtocolInterface (
                  Controller,