mirror of https://github.com/acidanthera/audk.git
Add DiskIo.c
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2841 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
vanjeff
parent
61076c5b77
commit
1e95872165
|
|
@ -0,0 +1,737 @@
|
||||||
|
/** @file
|
||||||
|
DiskIo driver that layers it's self on every Block IO protocol in the system.
|
||||||
|
DiskIo converts a block oriented device to a byte oriented device.
|
||||||
|
|
||||||
|
ReadDisk may have to do reads that are not aligned on sector boundaries.
|
||||||
|
There are three cases:
|
||||||
|
UnderRun - The first byte is not on a sector boundary or the read request is
|
||||||
|
less than a sector in length.
|
||||||
|
Aligned - A read of N contiguous sectors.
|
||||||
|
OverRun - The last byte is not on a sector boundary.
|
||||||
|
|
||||||
|
Copyright (c) 2006 - 2007, 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.
|
||||||
|
|
||||||
|
**/
|
||||||
|
|
||||||
|
#include "DiskIo.h"
|
||||||
|
|
||||||
|
EFI_DRIVER_BINDING_PROTOCOL gDiskIoDriverBinding = {
|
||||||
|
DiskIoDriverBindingSupported,
|
||||||
|
DiskIoDriverBindingStart,
|
||||||
|
DiskIoDriverBindingStop,
|
||||||
|
0xa,
|
||||||
|
NULL,
|
||||||
|
NULL
|
||||||
|
};
|
||||||
|
|
||||||
|
DISK_IO_PRIVATE_DATA gDiskIoPrivateDataTemplate = {
|
||||||
|
DISK_IO_PRIVATE_DATA_SIGNATURE,
|
||||||
|
{
|
||||||
|
EFI_DISK_IO_PROTOCOL_REVISION,
|
||||||
|
DiskIoReadDisk,
|
||||||
|
DiskIoWriteDisk
|
||||||
|
},
|
||||||
|
NULL
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
Test to see if this driver supports ControllerHandle.
|
||||||
|
|
||||||
|
@param This Protocol instance pointer.
|
||||||
|
@param ControllerHandle Handle of device to test
|
||||||
|
@param RemainingDevicePath Optional parameter use to pick a specific child
|
||||||
|
device to start.
|
||||||
|
|
||||||
|
@retval EFI_SUCCESS This driver supports this device
|
||||||
|
@retval EFI_ALREADY_STARTED This driver is already running on this device
|
||||||
|
@retval other This driver does not support this device
|
||||||
|
|
||||||
|
**/
|
||||||
|
EFI_STATUS
|
||||||
|
EFIAPI
|
||||||
|
DiskIoDriverBindingSupported (
|
||||||
|
IN EFI_DRIVER_BINDING_PROTOCOL *This,
|
||||||
|
IN EFI_HANDLE ControllerHandle,
|
||||||
|
IN EFI_DEVICE_PATH_PROTOCOL *RemainingDevicePath OPTIONAL
|
||||||
|
)
|
||||||
|
{
|
||||||
|
EFI_STATUS Status;
|
||||||
|
EFI_BLOCK_IO_PROTOCOL *BlockIo;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Open the IO Abstraction(s) needed to perform the supported test.
|
||||||
|
//
|
||||||
|
Status = gBS->OpenProtocol (
|
||||||
|
ControllerHandle,
|
||||||
|
&gEfiBlockIoProtocolGuid,
|
||||||
|
(VOID **) &BlockIo,
|
||||||
|
This->DriverBindingHandle,
|
||||||
|
ControllerHandle,
|
||||||
|
EFI_OPEN_PROTOCOL_BY_DRIVER
|
||||||
|
);
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
return Status;
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Close the I/O Abstraction(s) used to perform the supported test.
|
||||||
|
//
|
||||||
|
gBS->CloseProtocol (
|
||||||
|
ControllerHandle,
|
||||||
|
&gEfiBlockIoProtocolGuid,
|
||||||
|
This->DriverBindingHandle,
|
||||||
|
ControllerHandle
|
||||||
|
);
|
||||||
|
return EFI_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
Start this driver on ControllerHandle by opening a Block IO protocol and
|
||||||
|
installing a Disk IO protocol on ControllerHandle.
|
||||||
|
|
||||||
|
@param This Protocol instance pointer.
|
||||||
|
@param ControllerHandle Handle of device to bind driver to
|
||||||
|
@param RemainingDevicePath Optional parameter use to pick a specific child
|
||||||
|
device to start.
|
||||||
|
|
||||||
|
@retval EFI_SUCCESS This driver is added to ControllerHandle
|
||||||
|
@retval EFI_ALREADY_STARTED This driver is already running on ControllerHandle
|
||||||
|
@retval other This driver does not support this device
|
||||||
|
|
||||||
|
**/
|
||||||
|
EFI_STATUS
|
||||||
|
EFIAPI
|
||||||
|
DiskIoDriverBindingStart (
|
||||||
|
IN EFI_DRIVER_BINDING_PROTOCOL *This,
|
||||||
|
IN EFI_HANDLE ControllerHandle,
|
||||||
|
IN EFI_DEVICE_PATH_PROTOCOL *RemainingDevicePath OPTIONAL
|
||||||
|
)
|
||||||
|
{
|
||||||
|
EFI_STATUS Status;
|
||||||
|
DISK_IO_PRIVATE_DATA *Private;
|
||||||
|
|
||||||
|
Private = NULL;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Connect to the Block IO interface on ControllerHandle.
|
||||||
|
//
|
||||||
|
Status = gBS->OpenProtocol (
|
||||||
|
ControllerHandle,
|
||||||
|
&gEfiBlockIoProtocolGuid,
|
||||||
|
(VOID **) &gDiskIoPrivateDataTemplate.BlockIo,
|
||||||
|
This->DriverBindingHandle,
|
||||||
|
ControllerHandle,
|
||||||
|
EFI_OPEN_PROTOCOL_BY_DRIVER
|
||||||
|
);
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
return Status;
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Initialize the Disk IO device instance.
|
||||||
|
//
|
||||||
|
Private = AllocateCopyPool (sizeof (DISK_IO_PRIVATE_DATA), &gDiskIoPrivateDataTemplate);
|
||||||
|
if (Private == NULL) {
|
||||||
|
Status = EFI_OUT_OF_RESOURCES;
|
||||||
|
goto ErrorExit;
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Install protocol interfaces for the Disk IO device.
|
||||||
|
//
|
||||||
|
Status = gBS->InstallProtocolInterface (
|
||||||
|
&ControllerHandle,
|
||||||
|
&gEfiDiskIoProtocolGuid,
|
||||||
|
EFI_NATIVE_INTERFACE,
|
||||||
|
&Private->DiskIo
|
||||||
|
);
|
||||||
|
|
||||||
|
ErrorExit:
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
|
||||||
|
if (Private != NULL) {
|
||||||
|
FreePool (Private);
|
||||||
|
}
|
||||||
|
|
||||||
|
gBS->CloseProtocol (
|
||||||
|
ControllerHandle,
|
||||||
|
&gEfiBlockIoProtocolGuid,
|
||||||
|
This->DriverBindingHandle,
|
||||||
|
ControllerHandle
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return Status;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
Stop this driver on ControllerHandle by removing Disk IO protocol and closing
|
||||||
|
the Block IO protocol on ControllerHandle.
|
||||||
|
|
||||||
|
@param This Protocol instance pointer.
|
||||||
|
@param ControllerHandle Handle of device to stop driver on
|
||||||
|
@param NumberOfChildren Number of Handles in ChildHandleBuffer. If number of
|
||||||
|
children is zero stop the entire bus driver.
|
||||||
|
@param ChildHandleBuffer List of Child Handles to Stop.
|
||||||
|
|
||||||
|
@retval EFI_SUCCESS This driver is removed ControllerHandle
|
||||||
|
@retval other This driver was not removed from this device
|
||||||
|
|
||||||
|
**/
|
||||||
|
EFI_STATUS
|
||||||
|
EFIAPI
|
||||||
|
DiskIoDriverBindingStop (
|
||||||
|
IN EFI_DRIVER_BINDING_PROTOCOL *This,
|
||||||
|
IN EFI_HANDLE ControllerHandle,
|
||||||
|
IN UINTN NumberOfChildren,
|
||||||
|
IN EFI_HANDLE *ChildHandleBuffer
|
||||||
|
)
|
||||||
|
{
|
||||||
|
EFI_STATUS Status;
|
||||||
|
EFI_DISK_IO_PROTOCOL *DiskIo;
|
||||||
|
DISK_IO_PRIVATE_DATA *Private;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Get our context back.
|
||||||
|
//
|
||||||
|
Status = gBS->OpenProtocol (
|
||||||
|
ControllerHandle,
|
||||||
|
&gEfiDiskIoProtocolGuid,
|
||||||
|
(VOID **) &DiskIo,
|
||||||
|
This->DriverBindingHandle,
|
||||||
|
ControllerHandle,
|
||||||
|
EFI_OPEN_PROTOCOL_GET_PROTOCOL
|
||||||
|
);
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
return EFI_UNSUPPORTED;
|
||||||
|
}
|
||||||
|
|
||||||
|
Private = DISK_IO_PRIVATE_DATA_FROM_THIS (DiskIo);
|
||||||
|
|
||||||
|
Status = gBS->UninstallProtocolInterface (
|
||||||
|
ControllerHandle,
|
||||||
|
&gEfiDiskIoProtocolGuid,
|
||||||
|
&Private->DiskIo
|
||||||
|
);
|
||||||
|
if (!EFI_ERROR (Status)) {
|
||||||
|
|
||||||
|
Status = gBS->CloseProtocol (
|
||||||
|
ControllerHandle,
|
||||||
|
&gEfiBlockIoProtocolGuid,
|
||||||
|
This->DriverBindingHandle,
|
||||||
|
ControllerHandle
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!EFI_ERROR (Status)) {
|
||||||
|
FreePool (Private);
|
||||||
|
}
|
||||||
|
|
||||||
|
return Status;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
Read BufferSize bytes from Offset into Buffer.
|
||||||
|
Reads may support reads that are not aligned on
|
||||||
|
sector boundaries. There are three cases:
|
||||||
|
UnderRun - The first byte is not on a sector boundary or the read request is
|
||||||
|
less than a sector in length.
|
||||||
|
Aligned - A read of N contiguous sectors.
|
||||||
|
OverRun - The last byte is not on a sector boundary.
|
||||||
|
|
||||||
|
@param This Protocol instance pointer.
|
||||||
|
@param MediaId Id of the media, changes every time the media is replaced.
|
||||||
|
@param Offset The starting byte offset to read from
|
||||||
|
@param BufferSize Size of Buffer
|
||||||
|
@param Buffer Buffer containing read data
|
||||||
|
|
||||||
|
@retval EFI_SUCCESS The data was read correctly from the device.
|
||||||
|
@retval EFI_DEVICE_ERROR The device reported an error while performing the read.
|
||||||
|
@retval EFI_NO_MEDIA There is no media in the device.
|
||||||
|
@retval EFI_MEDIA_CHNAGED The MediaId does not matched the current device.
|
||||||
|
@retval EFI_INVALID_PARAMETER The read request contains device addresses that are not
|
||||||
|
valid for the device.
|
||||||
|
|
||||||
|
**/
|
||||||
|
EFI_STATUS
|
||||||
|
EFIAPI
|
||||||
|
DiskIoReadDisk (
|
||||||
|
IN EFI_DISK_IO_PROTOCOL *This,
|
||||||
|
IN UINT32 MediaId,
|
||||||
|
IN UINT64 Offset,
|
||||||
|
IN UINTN BufferSize,
|
||||||
|
OUT VOID *Buffer
|
||||||
|
)
|
||||||
|
{
|
||||||
|
EFI_STATUS Status;
|
||||||
|
DISK_IO_PRIVATE_DATA *Private;
|
||||||
|
EFI_BLOCK_IO_PROTOCOL *BlockIo;
|
||||||
|
EFI_BLOCK_IO_MEDIA *Media;
|
||||||
|
UINT32 BlockSize;
|
||||||
|
UINT64 Lba;
|
||||||
|
UINT64 OverRunLba;
|
||||||
|
UINT32 UnderRun;
|
||||||
|
UINT32 OverRun;
|
||||||
|
BOOLEAN TransactionComplete;
|
||||||
|
UINTN WorkingBufferSize;
|
||||||
|
UINT8 *WorkingBuffer;
|
||||||
|
UINTN Length;
|
||||||
|
UINT8 *Data;
|
||||||
|
UINT8 *PreData;
|
||||||
|
UINTN IsBufferAligned;
|
||||||
|
UINTN DataBufferSize;
|
||||||
|
BOOLEAN LastRead;
|
||||||
|
|
||||||
|
Private = DISK_IO_PRIVATE_DATA_FROM_THIS (This);
|
||||||
|
|
||||||
|
BlockIo = Private->BlockIo;
|
||||||
|
Media = BlockIo->Media;
|
||||||
|
BlockSize = Media->BlockSize;
|
||||||
|
|
||||||
|
if (Media->MediaId != MediaId) {
|
||||||
|
return EFI_MEDIA_CHANGED;
|
||||||
|
}
|
||||||
|
|
||||||
|
WorkingBuffer = Buffer;
|
||||||
|
WorkingBufferSize = BufferSize;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Allocate a temporary buffer for operation
|
||||||
|
//
|
||||||
|
DataBufferSize = BlockSize * DATA_BUFFER_BLOCK_NUM;
|
||||||
|
|
||||||
|
if (Media->IoAlign > 1) {
|
||||||
|
PreData = AllocatePool (DataBufferSize + Media->IoAlign);
|
||||||
|
Data = PreData - ((UINTN) PreData & (Media->IoAlign - 1)) + Media->IoAlign;
|
||||||
|
} else {
|
||||||
|
PreData = AllocatePool (DataBufferSize);
|
||||||
|
Data = PreData;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (PreData == NULL) {
|
||||||
|
return EFI_OUT_OF_RESOURCES;
|
||||||
|
}
|
||||||
|
|
||||||
|
Lba = DivU64x32Remainder (Offset, BlockSize, &UnderRun);
|
||||||
|
|
||||||
|
Length = BlockSize - UnderRun;
|
||||||
|
TransactionComplete = FALSE;
|
||||||
|
|
||||||
|
Status = EFI_SUCCESS;
|
||||||
|
if (UnderRun != 0) {
|
||||||
|
//
|
||||||
|
// Offset starts in the middle of an Lba, so read the entire block.
|
||||||
|
//
|
||||||
|
Status = BlockIo->ReadBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
Lba,
|
||||||
|
BlockSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Length > BufferSize) {
|
||||||
|
Length = BufferSize;
|
||||||
|
TransactionComplete = TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
CopyMem (WorkingBuffer, Data + UnderRun, Length);
|
||||||
|
|
||||||
|
WorkingBuffer += Length;
|
||||||
|
|
||||||
|
WorkingBufferSize -= Length;
|
||||||
|
if (WorkingBufferSize == 0) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
Lba += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
OverRunLba = Lba + DivU64x32Remainder (WorkingBufferSize, BlockSize, &OverRun);
|
||||||
|
|
||||||
|
if (!TransactionComplete && WorkingBufferSize >= BlockSize) {
|
||||||
|
//
|
||||||
|
// If the DiskIo maps directly to a BlockIo device do the read.
|
||||||
|
//
|
||||||
|
if (OverRun != 0) {
|
||||||
|
WorkingBufferSize -= OverRun;
|
||||||
|
}
|
||||||
|
//
|
||||||
|
// Check buffer alignment
|
||||||
|
//
|
||||||
|
IsBufferAligned = (UINTN) WorkingBuffer & (UINTN) (Media->IoAlign - 1);
|
||||||
|
|
||||||
|
if (Media->IoAlign <= 1 || IsBufferAligned == 0) {
|
||||||
|
//
|
||||||
|
// Alignment is satisfied, so read them together
|
||||||
|
//
|
||||||
|
Status = BlockIo->ReadBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
Lba,
|
||||||
|
WorkingBufferSize,
|
||||||
|
WorkingBuffer
|
||||||
|
);
|
||||||
|
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
WorkingBuffer += WorkingBufferSize;
|
||||||
|
|
||||||
|
} else {
|
||||||
|
//
|
||||||
|
// Use the allocated buffer instead of the original buffer
|
||||||
|
// to avoid alignment issue.
|
||||||
|
// Here, the allocated buffer (8-byte align) can satisfy the alignment
|
||||||
|
//
|
||||||
|
LastRead = FALSE;
|
||||||
|
do {
|
||||||
|
if (WorkingBufferSize <= DataBufferSize) {
|
||||||
|
//
|
||||||
|
// It is the last calling to readblocks in this loop
|
||||||
|
//
|
||||||
|
DataBufferSize = WorkingBufferSize;
|
||||||
|
LastRead = TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
Status = BlockIo->ReadBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
Lba,
|
||||||
|
DataBufferSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
CopyMem (WorkingBuffer, Data, DataBufferSize);
|
||||||
|
WorkingBufferSize -= DataBufferSize;
|
||||||
|
WorkingBuffer += DataBufferSize;
|
||||||
|
Lba += DATA_BUFFER_BLOCK_NUM;
|
||||||
|
} while (!LastRead);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!TransactionComplete && OverRun != 0) {
|
||||||
|
//
|
||||||
|
// Last read is not a complete block.
|
||||||
|
//
|
||||||
|
Status = BlockIo->ReadBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
OverRunLba,
|
||||||
|
BlockSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
CopyMem (WorkingBuffer, Data, OverRun);
|
||||||
|
}
|
||||||
|
|
||||||
|
Done:
|
||||||
|
if (PreData != NULL) {
|
||||||
|
FreePool (PreData);
|
||||||
|
}
|
||||||
|
|
||||||
|
return Status;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
Read BufferSize bytes from Offset into Buffer.
|
||||||
|
Writes may require a read modify write to support writes that are not
|
||||||
|
aligned on sector boundaries. There are three cases:
|
||||||
|
UnderRun - The first byte is not on a sector boundary or the write request
|
||||||
|
is less than a sector in length. Read modify write is required.
|
||||||
|
Aligned - A write of N contiguous sectors.
|
||||||
|
OverRun - The last byte is not on a sector boundary. Read modified write
|
||||||
|
required.
|
||||||
|
|
||||||
|
@param This Protocol instance pointer.
|
||||||
|
@param MediaId Id of the media, changes every time the media is replaced.
|
||||||
|
@param Offset The starting byte offset to read from
|
||||||
|
@param BufferSize Size of Buffer
|
||||||
|
@param Buffer Buffer containing read data
|
||||||
|
|
||||||
|
@retval EFI_SUCCESS The data was written correctly to the device.
|
||||||
|
@retval EFI_WRITE_PROTECTED The device can not be written to.
|
||||||
|
@retval EFI_DEVICE_ERROR The device reported an error while performing the write.
|
||||||
|
@retval EFI_NO_MEDIA There is no media in the device.
|
||||||
|
@retval EFI_MEDIA_CHNAGED The MediaId does not matched the current device.
|
||||||
|
@retval EFI_INVALID_PARAMETER The write request contains device addresses that are not
|
||||||
|
valid for the device.
|
||||||
|
|
||||||
|
**/
|
||||||
|
EFI_STATUS
|
||||||
|
EFIAPI
|
||||||
|
DiskIoWriteDisk (
|
||||||
|
IN EFI_DISK_IO_PROTOCOL *This,
|
||||||
|
IN UINT32 MediaId,
|
||||||
|
IN UINT64 Offset,
|
||||||
|
IN UINTN BufferSize,
|
||||||
|
IN VOID *Buffer
|
||||||
|
)
|
||||||
|
{
|
||||||
|
EFI_STATUS Status;
|
||||||
|
DISK_IO_PRIVATE_DATA *Private;
|
||||||
|
EFI_BLOCK_IO_PROTOCOL *BlockIo;
|
||||||
|
EFI_BLOCK_IO_MEDIA *Media;
|
||||||
|
UINT32 BlockSize;
|
||||||
|
UINT64 Lba;
|
||||||
|
UINT64 OverRunLba;
|
||||||
|
UINT32 UnderRun;
|
||||||
|
UINT32 OverRun;
|
||||||
|
BOOLEAN TransactionComplete;
|
||||||
|
UINTN WorkingBufferSize;
|
||||||
|
UINT8 *WorkingBuffer;
|
||||||
|
UINTN Length;
|
||||||
|
UINT8 *Data;
|
||||||
|
UINT8 *PreData;
|
||||||
|
UINTN IsBufferAligned;
|
||||||
|
UINTN DataBufferSize;
|
||||||
|
BOOLEAN LastWrite;
|
||||||
|
|
||||||
|
Private = DISK_IO_PRIVATE_DATA_FROM_THIS (This);
|
||||||
|
|
||||||
|
BlockIo = Private->BlockIo;
|
||||||
|
Media = BlockIo->Media;
|
||||||
|
BlockSize = Media->BlockSize;
|
||||||
|
|
||||||
|
if (Media->ReadOnly) {
|
||||||
|
return EFI_WRITE_PROTECTED;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Media->MediaId != MediaId) {
|
||||||
|
return EFI_MEDIA_CHANGED;
|
||||||
|
}
|
||||||
|
|
||||||
|
DataBufferSize = BlockSize * DATA_BUFFER_BLOCK_NUM;
|
||||||
|
|
||||||
|
if (Media->IoAlign > 1) {
|
||||||
|
PreData = AllocatePool (DataBufferSize + Media->IoAlign);
|
||||||
|
Data = PreData - ((UINTN) PreData & (Media->IoAlign - 1)) + Media->IoAlign;
|
||||||
|
} else {
|
||||||
|
PreData = AllocatePool (DataBufferSize);
|
||||||
|
Data = PreData;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (PreData == NULL) {
|
||||||
|
return EFI_OUT_OF_RESOURCES;
|
||||||
|
}
|
||||||
|
|
||||||
|
WorkingBuffer = Buffer;
|
||||||
|
WorkingBufferSize = BufferSize;
|
||||||
|
|
||||||
|
Lba = DivU64x32Remainder (Offset, BlockSize, &UnderRun);
|
||||||
|
|
||||||
|
Length = BlockSize - UnderRun;
|
||||||
|
TransactionComplete = FALSE;
|
||||||
|
|
||||||
|
Status = EFI_SUCCESS;
|
||||||
|
if (UnderRun != 0) {
|
||||||
|
//
|
||||||
|
// Offset starts in the middle of an Lba, so do read modify write.
|
||||||
|
//
|
||||||
|
Status = BlockIo->ReadBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
Lba,
|
||||||
|
BlockSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Length > BufferSize) {
|
||||||
|
Length = BufferSize;
|
||||||
|
TransactionComplete = TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
CopyMem (Data + UnderRun, WorkingBuffer, Length);
|
||||||
|
|
||||||
|
Status = BlockIo->WriteBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
Lba,
|
||||||
|
BlockSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
WorkingBuffer += Length;
|
||||||
|
WorkingBufferSize -= Length;
|
||||||
|
if (WorkingBufferSize == 0) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
Lba += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
OverRunLba = Lba + DivU64x32Remainder (WorkingBufferSize, BlockSize, &OverRun);
|
||||||
|
|
||||||
|
if (!TransactionComplete && WorkingBufferSize >= BlockSize) {
|
||||||
|
//
|
||||||
|
// If the DiskIo maps directly to a BlockIo device do the write.
|
||||||
|
//
|
||||||
|
if (OverRun != 0) {
|
||||||
|
WorkingBufferSize -= OverRun;
|
||||||
|
}
|
||||||
|
//
|
||||||
|
// Check buffer alignment
|
||||||
|
//
|
||||||
|
IsBufferAligned = (UINTN) WorkingBuffer & (UINTN) (Media->IoAlign - 1);
|
||||||
|
|
||||||
|
if (Media->IoAlign <= 1 || IsBufferAligned == 0) {
|
||||||
|
//
|
||||||
|
// Alignment is satisfied, so write them together
|
||||||
|
//
|
||||||
|
Status = BlockIo->WriteBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
Lba,
|
||||||
|
WorkingBufferSize,
|
||||||
|
WorkingBuffer
|
||||||
|
);
|
||||||
|
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
WorkingBuffer += WorkingBufferSize;
|
||||||
|
|
||||||
|
} else {
|
||||||
|
//
|
||||||
|
// The buffer parameter is not aligned with the request
|
||||||
|
// So use the allocated instead.
|
||||||
|
// It can fit almost all the cases.
|
||||||
|
//
|
||||||
|
LastWrite = FALSE;
|
||||||
|
do {
|
||||||
|
if (WorkingBufferSize <= DataBufferSize) {
|
||||||
|
//
|
||||||
|
// It is the last calling to writeblocks in this loop
|
||||||
|
//
|
||||||
|
DataBufferSize = WorkingBufferSize;
|
||||||
|
LastWrite = TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
CopyMem (Data, WorkingBuffer, DataBufferSize);
|
||||||
|
Status = BlockIo->WriteBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
Lba,
|
||||||
|
DataBufferSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
WorkingBufferSize -= DataBufferSize;
|
||||||
|
WorkingBuffer += DataBufferSize;
|
||||||
|
Lba += DATA_BUFFER_BLOCK_NUM;
|
||||||
|
} while (!LastWrite);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!TransactionComplete && OverRun != 0) {
|
||||||
|
//
|
||||||
|
// Last bit is not a complete block, so do a read modify write.
|
||||||
|
//
|
||||||
|
Status = BlockIo->ReadBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
OverRunLba,
|
||||||
|
BlockSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
|
||||||
|
CopyMem (Data, WorkingBuffer, OverRun);
|
||||||
|
|
||||||
|
Status = BlockIo->WriteBlocks (
|
||||||
|
BlockIo,
|
||||||
|
MediaId,
|
||||||
|
OverRunLba,
|
||||||
|
BlockSize,
|
||||||
|
Data
|
||||||
|
);
|
||||||
|
if (EFI_ERROR (Status)) {
|
||||||
|
goto Done;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Done:
|
||||||
|
if (PreData != NULL) {
|
||||||
|
FreePool (PreData);
|
||||||
|
}
|
||||||
|
|
||||||
|
return Status;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
The user Entry Point for module DiskIo. The user code starts with this function.
|
||||||
|
|
||||||
|
@param[in] ImageHandle The firmware allocated handle for the EFI image.
|
||||||
|
@param[in] SystemTable A pointer to the EFI System Table.
|
||||||
|
|
||||||
|
@retval EFI_SUCCESS The entry point is executed successfully.
|
||||||
|
@retval other Some error occurs when executing this entry point.
|
||||||
|
|
||||||
|
**/
|
||||||
|
EFI_STATUS
|
||||||
|
EFIAPI
|
||||||
|
InitializeDiskIo (
|
||||||
|
IN EFI_HANDLE ImageHandle,
|
||||||
|
IN EFI_SYSTEM_TABLE *SystemTable
|
||||||
|
)
|
||||||
|
{
|
||||||
|
EFI_STATUS Status;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Install driver model protocol(s).
|
||||||
|
//
|
||||||
|
Status = EfiLibInstallAllDriverProtocols (
|
||||||
|
ImageHandle,
|
||||||
|
SystemTable,
|
||||||
|
&gDiskIoDriverBinding,
|
||||||
|
ImageHandle,
|
||||||
|
&gDiskIoComponentName,
|
||||||
|
NULL,
|
||||||
|
NULL
|
||||||
|
);
|
||||||
|
ASSERT_EFI_ERROR (Status);
|
||||||
|
|
||||||
|
|
||||||
|
return Status;
|
||||||
|
}
|
||||||
|
|
||||||
Loading…
Reference in New Issue