[mirror_edk2.git] / OvmfPkg / PvScsiDxe / PvScsi.c

/** @file\r
\r
  This driver produces Extended SCSI Pass Thru Protocol instances for\r
  pvscsi devices.\r
\r
  Copyright (C) 2020, Oracle and/or its affiliates.\r
\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <IndustryStandard/Pci.h>\r
#include <IndustryStandard/PvScsi.h>\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/UefiLib.h>\r
#include <Protocol/PciIo.h>\r
#include <Protocol/PciRootBridgeIo.h>\r
#include <Uefi/UefiSpec.h>\r
\r
#include "PvScsi.h"\r
\r
//\r
// Higher versions will be used before lower, 0x10-0xffffffef is the version\r
// range for IHV (Indie Hardware Vendors)\r
//\r
#define PVSCSI_BINDING_VERSION      0x10\r
\r
//\r
// Ext SCSI Pass Thru utilities\r
//\r
\r
/**\r
  Writes a 32-bit value into BAR0 using MMIO\r
**/\r
STATIC\r
EFI_STATUS\r
PvScsiMmioWrite32 (\r
  IN CONST PVSCSI_DEV   *Dev,\r
  IN UINT64             Offset,\r
  IN UINT32             Value\r
  )\r
{\r
  return Dev->PciIo->Mem.Write (\r
                           Dev->PciIo,\r
                           EfiPciIoWidthUint32,\r
                           PCI_BAR_IDX0,\r
                           Offset,\r
                           1,   // Count\r
                           &Value\r
                           );\r
}\r
\r
/**\r
  Writes multiple words of data into BAR0 using MMIO\r
**/\r
STATIC\r
EFI_STATUS\r
PvScsiMmioWrite32Multiple (\r
  IN CONST PVSCSI_DEV   *Dev,\r
  IN UINT64             Offset,\r
  IN UINTN              Count,\r
  IN UINT32             *Words\r
  )\r
{\r
  return Dev->PciIo->Mem.Write (\r
                           Dev->PciIo,\r
                           EfiPciIoWidthFifoUint32,\r
                           PCI_BAR_IDX0,\r
                           Offset,\r
                           Count,\r
                           Words\r
                           );\r
}\r
\r
/**\r
  Send a PVSCSI command to device.\r
\r
  @param[in] Dev                    The pvscsi host device.\r
  @param[in] Cmd                    The command to send to device.\r
  @param[in] OPTIONAL DescWords     An optional command descriptor (If command\r
                                    have a descriptor). The descriptor is\r
                                    provided as an array of UINT32 words and\r
                                    is must be 32-bit aligned.\r
  @param[in] DescWordsCount         The number of words in command descriptor.\r
                                    Caller must specify here 0 if DescWords\r
                                    is not supplied (It is optional). In that\r
                                    case, DescWords is ignored.\r
\r
  @return   Status codes returned by Dev->PciIo->Mem.Write().\r
\r
**/\r
STATIC\r
EFI_STATUS\r
PvScsiWriteCmdDesc (\r
  IN CONST PVSCSI_DEV   *Dev,\r
  IN UINT32             Cmd,\r
  IN UINT32             *DescWords      OPTIONAL,\r
  IN UINTN              DescWordsCount\r
  )\r
{\r
  EFI_STATUS Status;\r
\r
  if (DescWordsCount > PVSCSI_MAX_CMD_DATA_WORDS) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = PvScsiMmioWrite32 (Dev, PvScsiRegOffsetCommand, Cmd);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (DescWordsCount > 0) {\r
    return PvScsiMmioWrite32Multiple (\r
             Dev,\r
             PvScsiRegOffsetCommandData,\r
             DescWordsCount,\r
             DescWords\r
             );\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
EFI_STATUS\r
PvScsiResetAdapter (\r
  IN CONST PVSCSI_DEV   *Dev\r
  )\r
{\r
  return PvScsiWriteCmdDesc (Dev, PvScsiCmdAdapterReset, NULL, 0);\r
}\r
\r
/**\r
  Check if Target argument to EXT_SCSI_PASS_THRU.GetNextTarget() and\r
  EXT_SCSI_PASS_THRU.GetNextTargetLun() is initialized\r
**/\r
STATIC\r
BOOLEAN\r
IsTargetInitialized (\r
  IN UINT8                                          *Target\r
  )\r
{\r
  UINTN Idx;\r
\r
  for (Idx = 0; Idx < TARGET_MAX_BYTES; ++Idx) {\r
    if (Target[Idx] != 0xFF) {\r
      return TRUE;\r
    }\r
  }\r
  return FALSE;\r
}\r
\r
//\r
// Ext SCSI Pass Thru\r
//\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiPassThru (\r
  IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL                *This,\r
  IN UINT8                                          *Target,\r
  IN UINT64                                         Lun,\r
  IN OUT EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET *Packet,\r
  IN EFI_EVENT                                      Event    OPTIONAL\r
  )\r
{\r
  return EFI_UNSUPPORTED;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiGetNextTargetLun (\r
  IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL                *This,\r
  IN OUT UINT8                                      **Target,\r
  IN OUT UINT64                                     *Lun\r
  )\r
{\r
  UINT8      *TargetPtr;\r
  UINT8      LastTarget;\r
  PVSCSI_DEV *Dev;\r
\r
  if (Target == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // The Target input parameter is unnecessarily a pointer-to-pointer\r
  //\r
  TargetPtr = *Target;\r
\r
  //\r
  // If target not initialized, return first target & LUN\r
  //\r
  if (!IsTargetInitialized (TargetPtr)) {\r
    ZeroMem (TargetPtr, TARGET_MAX_BYTES);\r
    *Lun = 0;\r
    return EFI_SUCCESS;\r
  }\r
\r
  //\r
  // We only use first byte of target identifer\r
  //\r
  LastTarget = *TargetPtr;\r
\r
  //\r
  // Increment (target, LUN) pair if valid on input\r
  //\r
  Dev = PVSCSI_FROM_PASS_THRU (This);\r
  if (LastTarget > Dev->MaxTarget || *Lun > Dev->MaxLun) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (*Lun < Dev->MaxLun) {\r
    ++*Lun;\r
    return EFI_SUCCESS;\r
  }\r
\r
  if (LastTarget < Dev->MaxTarget) {\r
    *Lun = 0;\r
    ++LastTarget;\r
    *TargetPtr = LastTarget;\r
    return EFI_SUCCESS;\r
  }\r
\r
  return EFI_NOT_FOUND;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiBuildDevicePath (\r
  IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL                *This,\r
  IN UINT8                                          *Target,\r
  IN UINT64                                         Lun,\r
  IN OUT EFI_DEVICE_PATH_PROTOCOL                   **DevicePath\r
  )\r
{\r
  UINT8             TargetValue;\r
  PVSCSI_DEV        *Dev;\r
  SCSI_DEVICE_PATH  *ScsiDevicePath;\r
\r
  if (DevicePath == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // We only use first byte of target identifer\r
  //\r
  TargetValue = *Target;\r
\r
  Dev = PVSCSI_FROM_PASS_THRU (This);\r
  if (TargetValue > Dev->MaxTarget || Lun > Dev->MaxLun) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  ScsiDevicePath = AllocatePool (sizeof (*ScsiDevicePath));\r
  if (ScsiDevicePath == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  ScsiDevicePath->Header.Type      = MESSAGING_DEVICE_PATH;\r
  ScsiDevicePath->Header.SubType   = MSG_SCSI_DP;\r
  ScsiDevicePath->Header.Length[0] = (UINT8)sizeof (*ScsiDevicePath);\r
  ScsiDevicePath->Header.Length[1] = (UINT8)(sizeof (*ScsiDevicePath) >> 8);\r
  ScsiDevicePath->Pun              = TargetValue;\r
  ScsiDevicePath->Lun              = (UINT16)Lun;\r
\r
  *DevicePath = &ScsiDevicePath->Header;\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiGetTargetLun (\r
  IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL                *This,\r
  IN EFI_DEVICE_PATH_PROTOCOL                       *DevicePath,\r
  OUT UINT8                                         **Target,\r
  OUT UINT64                                        *Lun\r
  )\r
{\r
  SCSI_DEVICE_PATH *ScsiDevicePath;\r
  PVSCSI_DEV       *Dev;\r
\r
  if (DevicePath == NULL || Target == NULL || *Target == NULL || Lun == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (DevicePath->Type    != MESSAGING_DEVICE_PATH ||\r
      DevicePath->SubType != MSG_SCSI_DP) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  ScsiDevicePath = (SCSI_DEVICE_PATH *)DevicePath;\r
  Dev = PVSCSI_FROM_PASS_THRU (This);\r
  if (ScsiDevicePath->Pun > Dev->MaxTarget ||\r
      ScsiDevicePath->Lun > Dev->MaxLun) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  //\r
  // We only use first byte of target identifer\r
  //\r
  **Target = (UINT8)ScsiDevicePath->Pun;\r
  ZeroMem (*Target + 1, TARGET_MAX_BYTES - 1);\r
  *Lun = ScsiDevicePath->Lun;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiResetChannel (\r
  IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL                *This\r
  )\r
{\r
  return EFI_UNSUPPORTED;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiResetTargetLun (\r
  IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL                *This,\r
  IN UINT8                                          *Target,\r
  IN UINT64                                         Lun\r
  )\r
{\r
  return EFI_UNSUPPORTED;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiGetNextTarget (\r
  IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL                *This,\r
  IN OUT UINT8                                      **Target\r
  )\r
{\r
  UINT8      *TargetPtr;\r
  UINT8      LastTarget;\r
  PVSCSI_DEV *Dev;\r
\r
  if (Target == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // The Target input parameter is unnecessarily a pointer-to-pointer\r
  //\r
  TargetPtr = *Target;\r
\r
  //\r
  // If target not initialized, return first target\r
  //\r
  if (!IsTargetInitialized (TargetPtr)) {\r
    ZeroMem (TargetPtr, TARGET_MAX_BYTES);\r
    return EFI_SUCCESS;\r
  }\r
\r
  //\r
  // We only use first byte of target identifer\r
  //\r
  LastTarget = *TargetPtr;\r
\r
  //\r
  // Increment target if valid on input\r
  //\r
  Dev = PVSCSI_FROM_PASS_THRU (This);\r
  if (LastTarget > Dev->MaxTarget) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (LastTarget < Dev->MaxTarget) {\r
    ++LastTarget;\r
    *TargetPtr = LastTarget;\r
    return EFI_SUCCESS;\r
  }\r
\r
  return EFI_NOT_FOUND;\r
}\r
\r
STATIC\r
EFI_STATUS\r
PvScsiSetPciAttributes (\r
  IN OUT PVSCSI_DEV *Dev\r
  )\r
{\r
  EFI_STATUS Status;\r
\r
  //\r
  // Backup original PCI Attributes\r
  //\r
  Status = Dev->PciIo->Attributes (\r
                         Dev->PciIo,\r
                         EfiPciIoAttributeOperationGet,\r
                         0,\r
                         &Dev->OriginalPciAttributes\r
                         );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Enable MMIO-Space & Bus-Mastering\r
  //\r
  Status = Dev->PciIo->Attributes (\r
                         Dev->PciIo,\r
                         EfiPciIoAttributeOperationEnable,\r
                         (EFI_PCI_IO_ATTRIBUTE_MEMORY |\r
                          EFI_PCI_IO_ATTRIBUTE_BUS_MASTER),\r
                         NULL\r
                         );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
VOID\r
PvScsiRestorePciAttributes (\r
  IN PVSCSI_DEV *Dev\r
  )\r
{\r
  Dev->PciIo->Attributes (\r
                Dev->PciIo,\r
                EfiPciIoAttributeOperationSet,\r
                Dev->OriginalPciAttributes,\r
                NULL\r
                );\r
}\r
\r
STATIC\r
EFI_STATUS\r
PvScsiAllocateSharedPages (\r
  IN PVSCSI_DEV                     *Dev,\r
  IN UINTN                          Pages,\r
  OUT VOID                          **HostAddress,\r
  OUT PVSCSI_DMA_DESC               *DmaDesc\r
  )\r
{\r
  EFI_STATUS Status;\r
  UINTN      NumberOfBytes;\r
\r
  Status = Dev->PciIo->AllocateBuffer (\r
                         Dev->PciIo,\r
                         AllocateAnyPages,\r
                         EfiBootServicesData,\r
                         Pages,\r
                         HostAddress,\r
                         EFI_PCI_ATTRIBUTE_MEMORY_CACHED\r
                         );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  NumberOfBytes = EFI_PAGES_TO_SIZE (Pages);\r
  Status = Dev->PciIo->Map (\r
                         Dev->PciIo,\r
                         EfiPciIoOperationBusMasterCommonBuffer,\r
                         *HostAddress,\r
                         &NumberOfBytes,\r
                         &DmaDesc->DeviceAddress,\r
                         &DmaDesc->Mapping\r
                         );\r
  if (EFI_ERROR (Status)) {\r
    goto FreeBuffer;\r
  }\r
\r
  if (NumberOfBytes != EFI_PAGES_TO_SIZE (Pages)) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Unmap;\r
  }\r
\r
  return EFI_SUCCESS;\r
\r
Unmap:\r
  Dev->PciIo->Unmap (Dev->PciIo, DmaDesc->Mapping);\r
\r
FreeBuffer:\r
  Dev->PciIo->FreeBuffer (Dev->PciIo, Pages, *HostAddress);\r
\r
  return Status;\r
}\r
\r
STATIC\r
VOID\r
PvScsiFreeSharedPages (\r
  IN PVSCSI_DEV                     *Dev,\r
  IN UINTN                          Pages,\r
  IN VOID                           *HostAddress,\r
  IN PVSCSI_DMA_DESC                *DmaDesc\r
  )\r
{\r
  Dev->PciIo->Unmap (Dev->PciIo, DmaDesc->Mapping);\r
  Dev->PciIo->FreeBuffer (Dev->PciIo, Pages, HostAddress);\r
}\r
\r
STATIC\r
EFI_STATUS\r
PvScsiInitRings (\r
  IN OUT PVSCSI_DEV *Dev\r
  )\r
{\r
  EFI_STATUS Status;\r
  union {\r
    PVSCSI_CMD_DESC_SETUP_RINGS Cmd;\r
    UINT32                      Uint32;\r
  } AlignedCmd;\r
  PVSCSI_CMD_DESC_SETUP_RINGS *Cmd;\r
\r
  Cmd = &AlignedCmd.Cmd;\r
\r
  Status = PvScsiAllocateSharedPages (\r
             Dev,\r
             1,\r
             (VOID **)&Dev->RingDesc.RingState,\r
             &Dev->RingDesc.RingStateDmaDesc\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  ZeroMem (Dev->RingDesc.RingState, EFI_PAGE_SIZE);\r
\r
  Status = PvScsiAllocateSharedPages (\r
             Dev,\r
             1,\r
             (VOID **)&Dev->RingDesc.RingReqs,\r
             &Dev->RingDesc.RingReqsDmaDesc\r
             );\r
  if (EFI_ERROR (Status)) {\r
    goto FreeRingState;\r
  }\r
  ZeroMem (Dev->RingDesc.RingReqs, EFI_PAGE_SIZE);\r
\r
  Status = PvScsiAllocateSharedPages (\r
             Dev,\r
             1,\r
             (VOID **)&Dev->RingDesc.RingCmps,\r
             &Dev->RingDesc.RingCmpsDmaDesc\r
             );\r
  if (EFI_ERROR (Status)) {\r
    goto FreeRingReqs;\r
  }\r
  ZeroMem (Dev->RingDesc.RingCmps, EFI_PAGE_SIZE);\r
\r
  ZeroMem (Cmd, sizeof (*Cmd));\r
  Cmd->ReqRingNumPages = 1;\r
  Cmd->CmpRingNumPages = 1;\r
  Cmd->RingsStatePPN = RShiftU64 (\r
                         Dev->RingDesc.RingStateDmaDesc.DeviceAddress,\r
                         EFI_PAGE_SHIFT\r
                         );\r
  Cmd->ReqRingPPNs[0] = RShiftU64 (\r
                          Dev->RingDesc.RingReqsDmaDesc.DeviceAddress,\r
                          EFI_PAGE_SHIFT\r
                          );\r
  Cmd->CmpRingPPNs[0] = RShiftU64 (\r
                          Dev->RingDesc.RingCmpsDmaDesc.DeviceAddress,\r
                          EFI_PAGE_SHIFT\r
                          );\r
\r
  STATIC_ASSERT (\r
    sizeof (*Cmd) % sizeof (UINT32) == 0,\r
    "Cmd must be multiple of 32-bit words"\r
    );\r
  Status = PvScsiWriteCmdDesc (\r
             Dev,\r
             PvScsiCmdSetupRings,\r
             (UINT32 *)Cmd,\r
             sizeof (*Cmd) / sizeof (UINT32)\r
             );\r
  if (EFI_ERROR (Status)) {\r
    goto FreeRingCmps;\r
  }\r
\r
  return EFI_SUCCESS;\r
\r
FreeRingCmps:\r
  PvScsiFreeSharedPages (\r
    Dev,\r
    1,\r
    Dev->RingDesc.RingCmps,\r
    &Dev->RingDesc.RingCmpsDmaDesc\r
    );\r
\r
FreeRingReqs:\r
  PvScsiFreeSharedPages (\r
    Dev,\r
    1,\r
    Dev->RingDesc.RingReqs,\r
    &Dev->RingDesc.RingReqsDmaDesc\r
    );\r
\r
FreeRingState:\r
  PvScsiFreeSharedPages (\r
    Dev,\r
    1,\r
    Dev->RingDesc.RingState,\r
    &Dev->RingDesc.RingStateDmaDesc\r
    );\r
\r
  return Status;\r
}\r
\r
STATIC\r
VOID\r
PvScsiFreeRings (\r
  IN OUT PVSCSI_DEV *Dev\r
  )\r
{\r
  PvScsiFreeSharedPages (\r
    Dev,\r
    1,\r
    Dev->RingDesc.RingCmps,\r
    &Dev->RingDesc.RingCmpsDmaDesc\r
    );\r
\r
  PvScsiFreeSharedPages (\r
    Dev,\r
    1,\r
    Dev->RingDesc.RingReqs,\r
    &Dev->RingDesc.RingReqsDmaDesc\r
    );\r
\r
  PvScsiFreeSharedPages (\r
    Dev,\r
    1,\r
    Dev->RingDesc.RingState,\r
    &Dev->RingDesc.RingStateDmaDesc\r
    );\r
}\r
\r
STATIC\r
EFI_STATUS\r
PvScsiInit (\r
  IN OUT PVSCSI_DEV *Dev\r
  )\r
{\r
  EFI_STATUS Status;\r
\r
  //\r
  // Init configuration\r
  //\r
  Dev->MaxTarget = PcdGet8 (PcdPvScsiMaxTargetLimit);\r
  Dev->MaxLun = PcdGet8 (PcdPvScsiMaxLunLimit);\r
\r
  //\r
  // Set PCI Attributes\r
  //\r
  Status = PvScsiSetPciAttributes (Dev);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Reset adapter\r
  //\r
  Status = PvScsiResetAdapter (Dev);\r
  if (EFI_ERROR (Status)) {\r
    goto RestorePciAttributes;\r
  }\r
\r
  //\r
  // Init PVSCSI rings\r
  //\r
  Status = PvScsiInitRings (Dev);\r
  if (EFI_ERROR (Status)) {\r
    goto RestorePciAttributes;\r
  }\r
\r
  //\r
  // Populate the exported interface's attributes\r
  //\r
  Dev->PassThru.Mode             = &Dev->PassThruMode;\r
  Dev->PassThru.PassThru         = &PvScsiPassThru;\r
  Dev->PassThru.GetNextTargetLun = &PvScsiGetNextTargetLun;\r
  Dev->PassThru.BuildDevicePath  = &PvScsiBuildDevicePath;\r
  Dev->PassThru.GetTargetLun     = &PvScsiGetTargetLun;\r
  Dev->PassThru.ResetChannel     = &PvScsiResetChannel;\r
  Dev->PassThru.ResetTargetLun   = &PvScsiResetTargetLun;\r
  Dev->PassThru.GetNextTarget    = &PvScsiGetNextTarget;\r
\r
  //\r
  // AdapterId is a target for which no handle will be created during bus scan.\r
  // Prevent any conflict with real devices.\r
  //\r
  Dev->PassThruMode.AdapterId = MAX_UINT32;\r
\r
  //\r
  // Set both physical and logical attributes for non-RAID SCSI channel\r
  //\r
  Dev->PassThruMode.Attributes = EFI_EXT_SCSI_PASS_THRU_ATTRIBUTES_PHYSICAL |\r
                                 EFI_EXT_SCSI_PASS_THRU_ATTRIBUTES_LOGICAL;\r
\r
  //\r
  // No restriction on transfer buffer alignment\r
  //\r
  Dev->PassThruMode.IoAlign = 0;\r
\r
  return EFI_SUCCESS;\r
\r
RestorePciAttributes:\r
  PvScsiRestorePciAttributes (Dev);\r
\r
  return Status;\r
}\r
\r
STATIC\r
VOID\r
PvScsiUninit (\r
  IN OUT PVSCSI_DEV *Dev\r
  )\r
{\r
  //\r
  // Reset device to stop device usage of the rings.\r
  // This is required to safely free the rings.\r
  //\r
  PvScsiResetAdapter (Dev);\r
\r
  PvScsiFreeRings (Dev);\r
\r
  PvScsiRestorePciAttributes (Dev);\r
}\r
\r
//\r
// Driver Binding\r
//\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiDriverBindingSupported (\r
  IN EFI_DRIVER_BINDING_PROTOCOL *This,\r
  IN EFI_HANDLE                  ControllerHandle,\r
  IN EFI_DEVICE_PATH_PROTOCOL    *RemainingDevicePath OPTIONAL\r
  )\r
{\r
  EFI_STATUS          Status;\r
  EFI_PCI_IO_PROTOCOL *PciIo;\r
  PCI_TYPE00          Pci;\r
\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiPciIoProtocolGuid,\r
                  (VOID **)&PciIo,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = PciIo->Pci.Read (\r
                        PciIo,\r
                        EfiPciIoWidthUint32,\r
                        0,\r
                        sizeof (Pci) / sizeof (UINT32),\r
                        &Pci\r
                        );\r
  if (EFI_ERROR (Status)) {\r
    goto Done;\r
  }\r
\r
  if ((Pci.Hdr.VendorId != PCI_VENDOR_ID_VMWARE) ||\r
      (Pci.Hdr.DeviceId != PCI_DEVICE_ID_VMWARE_PVSCSI)) {\r
    Status = EFI_UNSUPPORTED;\r
    goto Done;\r
  }\r
\r
  Status = EFI_SUCCESS;\r
\r
Done:\r
  gBS->CloseProtocol (\r
         ControllerHandle,\r
         &gEfiPciIoProtocolGuid,\r
         This->DriverBindingHandle,\r
         ControllerHandle\r
         );\r
\r
  return Status;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiDriverBindingStart (\r
  IN EFI_DRIVER_BINDING_PROTOCOL *This,\r
  IN EFI_HANDLE                  ControllerHandle,\r
  IN EFI_DEVICE_PATH_PROTOCOL    *RemainingDevicePath OPTIONAL\r
  )\r
{\r
  PVSCSI_DEV *Dev;\r
  EFI_STATUS Status;\r
\r
  Dev = (PVSCSI_DEV *) AllocateZeroPool (sizeof (*Dev));\r
  if (Dev == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiPciIoProtocolGuid,\r
                  (VOID **)&Dev->PciIo,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto FreePvScsi;\r
  }\r
\r
  Status = PvScsiInit (Dev);\r
  if (EFI_ERROR (Status)) {\r
    goto ClosePciIo;\r
  }\r
\r
  //\r
  // Setup complete, attempt to export the driver instance's PassThru interface\r
  //\r
  Dev->Signature = PVSCSI_SIG;\r
  Status = gBS->InstallProtocolInterface (\r
                  &ControllerHandle,\r
                  &gEfiExtScsiPassThruProtocolGuid,\r
                  EFI_NATIVE_INTERFACE,\r
                  &Dev->PassThru\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto UninitDev;\r
  }\r
\r
  return EFI_SUCCESS;\r
\r
UninitDev:\r
  PvScsiUninit (Dev);\r
\r
ClosePciIo:\r
  gBS->CloseProtocol (\r
         ControllerHandle,\r
         &gEfiPciIoProtocolGuid,\r
         This->DriverBindingHandle,\r
         ControllerHandle\r
         );\r
\r
FreePvScsi:\r
  FreePool (Dev);\r
\r
  return Status;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiDriverBindingStop (\r
  IN EFI_DRIVER_BINDING_PROTOCOL *This,\r
  IN EFI_HANDLE                  ControllerHandle,\r
  IN UINTN                       NumberOfChildren,\r
  IN EFI_HANDLE                  *ChildHandleBuffer\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  EFI_EXT_SCSI_PASS_THRU_PROTOCOL *PassThru;\r
  PVSCSI_DEV                      *Dev;\r
\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiExtScsiPassThruProtocolGuid,\r
                  (VOID **)&PassThru,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL // Lookup only\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Dev = PVSCSI_FROM_PASS_THRU (PassThru);\r
\r
  Status = gBS->UninstallProtocolInterface (\r
                  ControllerHandle,\r
                  &gEfiExtScsiPassThruProtocolGuid,\r
                  &Dev->PassThru\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  PvScsiUninit (Dev);\r
\r
  gBS->CloseProtocol (\r
         ControllerHandle,\r
         &gEfiPciIoProtocolGuid,\r
         This->DriverBindingHandle,\r
         ControllerHandle\r
         );\r
\r
  FreePool (Dev);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC EFI_DRIVER_BINDING_PROTOCOL mPvScsiDriverBinding = {\r
  &PvScsiDriverBindingSupported,\r
  &PvScsiDriverBindingStart,\r
  &PvScsiDriverBindingStop,\r
  PVSCSI_BINDING_VERSION,\r
  NULL, // ImageHandle, filled by EfiLibInstallDriverBindingComponentName2()\r
  NULL  // DriverBindingHandle, filled as well\r
};\r
\r
//\r
// Component Name\r
//\r
\r
STATIC EFI_UNICODE_STRING_TABLE mDriverNameTable[] = {\r
  { "eng;en", L"PVSCSI Host Driver" },\r
  { NULL,     NULL                  }\r
};\r
\r
STATIC EFI_COMPONENT_NAME_PROTOCOL mComponentName;\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiGetDriverName (\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
           mDriverNameTable,\r
           DriverName,\r
           (BOOLEAN)(This == &mComponentName) // Iso639Language\r
           );\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
PvScsiGetDeviceName (\r
  IN  EFI_COMPONENT_NAME_PROTOCOL *This,\r
  IN  EFI_HANDLE                  DeviceHandle,\r
  IN  EFI_HANDLE                  ChildHandle,\r
  IN  CHAR8                       *Language,\r
  OUT CHAR16                      **ControllerName\r
  )\r
{\r
  return EFI_UNSUPPORTED;\r
}\r
\r
STATIC EFI_COMPONENT_NAME_PROTOCOL mComponentName = {\r
  &PvScsiGetDriverName,\r
  &PvScsiGetDeviceName,\r
  "eng" // SupportedLanguages, ISO 639-2 language codes\r
};\r
\r
STATIC EFI_COMPONENT_NAME2_PROTOCOL mComponentName2 = {\r
  (EFI_COMPONENT_NAME2_GET_DRIVER_NAME)     &PvScsiGetDriverName,\r
  (EFI_COMPONENT_NAME2_GET_CONTROLLER_NAME) &PvScsiGetDeviceName,\r
  "en" // SupportedLanguages, RFC 4646 language codes\r
};\r
\r
//\r
// Entry Point\r
//\r
\r
EFI_STATUS\r
EFIAPI\r
PvScsiEntryPoint (\r
  IN EFI_HANDLE       ImageHandle,\r
  IN EFI_SYSTEM_TABLE *SystemTable\r
  )\r
{\r
  return EfiLibInstallDriverBindingComponentName2 (\r
           ImageHandle,\r
           SystemTable,\r
           &mPvScsiDriverBinding,\r
           ImageHandle,\r
           &mComponentName,\r
           &mComponentName2\r
           );\r
}\r