usbmassstorage.c

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

  //
  Status = gBS->UninstallProtocolInterface (
                  Controller,
                  &gEfiBlockIoProtocolGuid,
                  &UsbFloppyDevice->BlkIo
                  );
  if (EFI_ERROR (Status)) {
    return Status;
  }
  //
  // Stop using EFI_USB_ATAPI_PROTOCOL
  //
  gBS->CloseProtocol (
        Controller,
        &gEfiUsbAtapiProtocolGuid,
        This->DriverBindingHandle,
        Controller
        );

  if (UsbFloppyDevice->SenseData != NULL) {
    gBS->FreePool (UsbFloppyDevice->SenseData);
  }

  gBS->FreePool (UsbFloppyDevice);

  return EFI_SUCCESS;
}


STATIC
EFI_STATUS
EFIAPI
USBFloppyReset (
  IN  EFI_BLOCK_IO_PROTOCOL   *This,
  IN  BOOLEAN                 ExtendedVerification
  )
/*++

  Routine Description:
    Implements EFI_BLOCK_IO_PROTOCOL.Reset() function.
  
  Arguments:
    This     The EFI_BLOCK_IO_PROTOCOL instance.
    ExtendedVerification
              Indicates that the driver may perform a more exhaustive
              verification operation of the device during reset.
              (This parameter is ingored in this driver.)
    
  Returns:  
    EFI_SUCCESS - Success
--*/      
{
  USB_FLOPPY_DEV          *UsbFloppyDevice;
  EFI_USB_ATAPI_PROTOCOL  *UsbAtapiInterface;
  EFI_STATUS              Status;

  UsbFloppyDevice   = USB_FLOPPY_DEV_FROM_THIS (This);

  UsbAtapiInterface = UsbFloppyDevice->AtapiProtocol;

  //
  // directly calling EFI_USB_ATAPI_PROTOCOL.Reset() to implement reset.
  //
  Status = UsbAtapiInterface->UsbAtapiReset (UsbAtapiInterface, ExtendedVerification);

  return Status;
}

STATIC
EFI_STATUS
EFIAPI
USBFloppyReadBlocks (
  IN  EFI_BLOCK_IO_PROTOCOL   *This,
  IN  UINT32                  MediaId,
  IN  EFI_LBA                 LBA,
  IN  UINTN                   BufferSize,
  OUT VOID                    *Buffer
  )
/*++

  Routine Description:
    Implements EFI_BLOCK_IO_PROTOCOL.ReadBlocks() function.
  
  Arguments:
    This     The EFI_BLOCK_IO_PROTOCOL instance.
    MediaId  The media id that the read request is for.
    LBA      The starting logical block address to read from on the device.
    BufferSize
              The size of the Buffer in bytes. This must be a multiple of 
              the intrinsic block size of the device.
    Buffer    A pointer to the destination buffer for the data. The caller 
              is responsible for either having implicit or explicit ownership
              of the buffer.                               
  
  Returns:  
    EFI_INVALID_PARAMETER - Parameter is error
    EFI_SUCCESS           - Success  
    EFI_DEVICE_ERROR      - Hardware Error
    EFI_NO_MEDIA          - No media
    EFI_MEDIA_CHANGED     - Media Change
    EFI_BAD_BUFFER_SIZE   - Buffer size is bad
 --*/      
{
  USB_FLOPPY_DEV      *UsbFloppyDevice;
  EFI_STATUS          Status;
  EFI_BLOCK_IO_MEDIA  *Media;
  UINTN               BlockSize;
  UINTN               NumberOfBlocks;
  BOOLEAN             MediaChange;

  Status          = EFI_SUCCESS;
  MediaChange     = FALSE;
  UsbFloppyDevice = USB_FLOPPY_DEV_FROM_THIS (This);

  //
  // Check parameters
  //
  if (!Buffer) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  if (BufferSize == 0) {
    Status = EFI_SUCCESS;
    goto Done;
  }

  UsbFloppyTestUnitReady (UsbFloppyDevice);

  Status = UsbFloppyDetectMedia (UsbFloppyDevice, &MediaChange);
  if (EFI_ERROR (Status)) {

    Status = EFI_DEVICE_ERROR;
    goto Done;
  }

  if (MediaChange) {
    gBS->ReinstallProtocolInterface (
          UsbFloppyDevice->Handle,
          &gEfiBlockIoProtocolGuid,
          &UsbFloppyDevice->BlkIo,
          &UsbFloppyDevice->BlkIo
          );
  }

  Media           = UsbFloppyDevice->BlkIo.Media;
  BlockSize       = Media->BlockSize;
  NumberOfBlocks  = BufferSize / BlockSize;

  if (!(Media->MediaPresent)) {
    Status = EFI_NO_MEDIA;
    goto Done;
  }

  if (MediaId != Media->MediaId) {
    Status = EFI_MEDIA_CHANGED;
    goto Done;
  }

  if (BufferSize % BlockSize != 0) {
    Status = EFI_BAD_BUFFER_SIZE;
    goto Done;
  }

  if (LBA > Media->LastBlock) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  if ((LBA + NumberOfBlocks - 1) > Media->LastBlock) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  if ((Media->IoAlign > 1) && (((UINTN) Buffer & (Media->IoAlign - 1)) != 0)) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  while (NumberOfBlocks > 0) {
   
    if (NumberOfBlocks > BLOCK_UNIT) {
      Status = USBFloppyRead10 (UsbFloppyDevice, Buffer, LBA, BLOCK_UNIT);
    } else {
      Status = USBFloppyRead10 (UsbFloppyDevice, Buffer, LBA, NumberOfBlocks);
    }

    if (EFI_ERROR (Status)) {
      This->Reset (This, TRUE);
      Status = EFI_DEVICE_ERROR;
    }

    if (NumberOfBlocks > BLOCK_UNIT) {
       NumberOfBlocks -= BLOCK_UNIT;
       LBA += BLOCK_UNIT;
       Buffer = (UINT8 *) Buffer + This->Media->BlockSize * BLOCK_UNIT;
    } else {
       NumberOfBlocks -= NumberOfBlocks;
       LBA += NumberOfBlocks;
       Buffer = (UINT8 *) Buffer + This->Media->BlockSize * NumberOfBlocks;
    }
 }

 Done:
  return Status;
}

STATIC
EFI_STATUS
EFIAPI
USBFloppyWriteBlocks (
  IN  EFI_BLOCK_IO_PROTOCOL   *This,
  IN  UINT32                  MediaId,
  IN  EFI_LBA                 LBA,
  IN  UINTN                   BufferSize,
  IN  VOID                    *Buffer
  )
/*++

  Routine Description:
    Implements EFI_BLOCK_IO_PROTOCOL.WriteBlocks() function.
  
  Arguments:
    This     The EFI_BLOCK_IO_PROTOCOL instance.
    MediaId  The media id that the write request is for.
    LBA      The starting logical block address to be written.
             The caller is responsible for writing to only 
             legitimate locations.
    BufferSize
              The size of the Buffer in bytes. This must be a multiple of 
              the intrinsic block size of the device.
    Buffer    A pointer to the source buffer for the data. The caller 
              is responsible for either having implicit or explicit ownership
              of the buffer.                               
  
  Returns:  
    EFI_INVALID_PARAMETER - Parameter is error
    EFI_SUCCESS           - Success  
    EFI_DEVICE_ERROR      - Hardware Error
    EFI_NO_MEDIA          - No media
    EFI_MEDIA_CHANGED     - Media Change
    EFI_BAD_BUFFER_SIZE   - Buffer size is bad

--*/        
{
  USB_FLOPPY_DEV      *UsbFloppyDevice;
  EFI_STATUS          Status;
  EFI_BLOCK_IO_MEDIA  *Media;
  UINTN               BlockSize;
  UINTN               NumberOfBlocks;
  BOOLEAN             MediaChange;

  Status          = EFI_SUCCESS;
  MediaChange     = FALSE;

  UsbFloppyDevice = USB_FLOPPY_DEV_FROM_THIS (This);

  //
  // Check parameters
  //
  if (!Buffer) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  if (BufferSize == 0) {
    Status = EFI_SUCCESS;
    goto Done;
  }

  UsbFloppyTestUnitReady (UsbFloppyDevice);

  Status = UsbFloppyDetectMedia (UsbFloppyDevice, &MediaChange);
  if (EFI_ERROR (Status)) {

    Status = EFI_DEVICE_ERROR;
    goto Done;
  }

  if (MediaChange) {
    gBS->ReinstallProtocolInterface (
          UsbFloppyDevice->Handle,
          &gEfiBlockIoProtocolGuid,
          &UsbFloppyDevice->BlkIo,
          &UsbFloppyDevice->BlkIo
          );
  }

  Media           = UsbFloppyDevice->BlkIo.Media;
  BlockSize       = Media->BlockSize;
  NumberOfBlocks  = BufferSize / BlockSize;

  if (!(Media->MediaPresent)) {
    Status = EFI_NO_MEDIA;
    goto Done;
  }

  if (MediaId != Media->MediaId) {
    Status = EFI_MEDIA_CHANGED;
    goto Done;
  }

  if (BufferSize % BlockSize != 0) {
    Status = EFI_BAD_BUFFER_SIZE;
    goto Done;
  }

  if (LBA > Media->LastBlock) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  if ((LBA + NumberOfBlocks - 1) > Media->LastBlock) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  if ((Media->IoAlign > 1) && (((UINTN) Buffer & (Media->IoAlign - 1)) != 0)) {
    Status = EFI_INVALID_PARAMETER;
    goto Done;
  }

  if (UsbFloppyDevice->BlkMedia.ReadOnly) {
    Status = EFI_WRITE_PROTECTED;
    goto Done;
  }

  while (NumberOfBlocks > 0) {
   
    if (NumberOfBlocks > BLOCK_UNIT) {
      Status = USBFloppyWrite10 (UsbFloppyDevice, Buffer, LBA, BLOCK_UNIT);
    } else {
      Status = USBFloppyWrite10 (UsbFloppyDevice, Buffer, LBA, NumberOfBlocks);
    }

    if (EFI_ERROR (Status)) {
      This->Reset (This, TRUE);
      Status = EFI_DEVICE_ERROR;
    }

    if (NumberOfBlocks > BLOCK_UNIT) {
       NumberOfBlocks -= BLOCK_UNIT;
       LBA += BLOCK_UNIT;
       Buffer = (UINT8 *) Buffer + This->Media->BlockSize * BLOCK_UNIT;
    } else {
       NumberOfBlocks -= NumberOfBlocks;
       LBA += NumberOfBlocks;
       Buffer = (UINT8 *) Buffer + This->Media->BlockSize * NumberOfBlocks;
    }
 }

Done:

  return Status;
}

STATIC
EFI_STATUS
EFIAPI
USBFloppyFlushBlocks (
  IN  EFI_BLOCK_IO_PROTOCOL   *This
  )
/*++

  Routine Description:
    Implements EFI_BLOCK_IO_PROTOCOL.FlushBlocks() function.
    (In this driver, this function just returns EFI_SUCCESS.)
  
  Arguments:
    This     The EFI_BLOCK_IO_PROTOCOL instance.
  
  Returns:  
    EFI_SUCCESS - Success
--*/    
{
  return EFI_SUCCESS;
}