tyler.durden/audk
1
0
Fork 0
mirror of https://github.com/acidanthera/audk.git synced 2025-07-26 15:14:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

DxeRing3: Added FixInterface().

Browse Source
This commit is contained in:
Mikhail Krichanov 2024-02-29 12:25:40 +03:00
parent 173d9edad4
commit 09ab943f86
1 changed files with 64 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

122
MdeModulePkg/Core/Dxe/DxeRing3/Ring3UefiBootServices.c
View File

@ -17,6 +17,67 @@
BOOLEAN mOnGuarding = FALSE; BOOLEAN mOnGuarding = FALSE;
STATIC
EFI_STATUS
EFIAPI
FixInterface (
IN EFI_GUID *Protocol,
IN OUT VOID **Interface
)
{
EFI_LOADED_IMAGE_PROTOCOL *LoadedImage;
EFI_BLOCK_IO_PROTOCOL *BlockIo;
EFI_DISK_IO_PROTOCOL *DiskIo;
EFI_DEVICE_PATH_UTILITIES_PROTOCOL *DevicePath;
ASSERT (Protocol != NULL);
ASSERT (Interface != NULL);
if (CompareGuid (Protocol, &gEfiDevicePathProtocolGuid)) {
} else if (CompareGuid (Protocol, &gEfiLoadedImageProtocolGuid)) {
LoadedImage = (EFI_LOADED_IMAGE_PROTOCOL *)*Interface;
// TODO: Copy User changes to Core? Resembles InstallMultipleProtocolInterfaces().
LoadedImage->Unload = NULL;
} else if (CompareGuid (Protocol, &gEfiBlockIoProtocolGuid)) {
BlockIo = (EFI_BLOCK_IO_PROTOCOL *)*Interface;
BlockIo->Reset = Ring3BlockIoReset;
BlockIo->ReadBlocks = Ring3BlockIoRead;
BlockIo->WriteBlocks = Ring3BlockIoWrite;
BlockIo->FlushBlocks = Ring3BlockIoFlush;
} else if (CompareGuid (Protocol, &gEfiDiskIoProtocolGuid)) {
DiskIo = (EFI_DISK_IO_PROTOCOL *)*Interface;
DiskIo->ReadDisk = Ring3DiskIoRead;
DiskIo->WriteDisk = Ring3DiskIoWrite;
} else if (CompareGuid (Protocol, &gEfiDevicePathUtilitiesProtocolGuid)) {
DevicePath = (EFI_DEVICE_PATH_UTILITIES_PROTOCOL *)*Interface;
DevicePath->GetDevicePathSize = NULL;
DevicePath->DuplicateDevicePath = NULL;
DevicePath->AppendDevicePath = NULL;
DevicePath->AppendDeviceNode = NULL;
DevicePath->AppendDevicePathInstance = NULL;
DevicePath->GetNextDevicePathInstance = NULL;
DevicePath->IsDevicePathMultiInstance = NULL;
DevicePath->CreateDeviceNode = NULL;
} else {
return EFI_UNSUPPORTED;
}
return EFI_SUCCESS;
}
EFI_TPL EFI_TPL
EFIAPI EFIAPI
Ring3RaiseTpl ( Ring3RaiseTpl (
@ -218,13 +279,7 @@ Ring3HandleProtocol (
return Status; return Status;
} }
if (CompareGuid (Protocol, &gEfiDevicePathProtocolGuid)) { return FixInterface (Protocol, Interface);
Status = EFI_SUCCESS;
} else {
Status = EFI_UNSUPPORTED;
}
return Status;
} }
EFI_STATUS EFI_STATUS
@ -394,10 +449,6 @@ Ring3OpenProtocol (
{ {
EFI_STATUS Status; EFI_STATUS Status;
EFI_LOADED_IMAGE_PROTOCOL *LoadedImage;
EFI_BLOCK_IO_PROTOCOL *BlockIo;
EFI_DISK_IO_PROTOCOL *DiskIo;
Status = SysCall ( Status = SysCall (
SysCallOpenProtocol, SysCallOpenProtocol,
CoreUserHandle, CoreUserHandle,
@ -411,35 +462,7 @@ Ring3OpenProtocol (
return Status; return Status;
} }
if (CompareGuid (Protocol, &gEfiLoadedImageProtocolGuid)) { return FixInterface (Protocol, Interface);
LoadedImage = (EFI_LOADED_IMAGE_PROTOCOL *)*Interface;
// TODO: Copy User changes to Core? Resembles InstallMultipleProtocolInterfaces().
LoadedImage->Unload = NULL;
} else if (CompareGuid (Protocol, &gEfiBlockIoProtocolGuid)) {
BlockIo = (EFI_BLOCK_IO_PROTOCOL *)*Interface;
BlockIo->Reset = Ring3BlockIoReset;
BlockIo->ReadBlocks = Ring3BlockIoRead;
BlockIo->WriteBlocks = Ring3BlockIoWrite;
BlockIo->FlushBlocks = Ring3BlockIoFlush;
} else if (CompareGuid (Protocol, &gEfiDiskIoProtocolGuid)) {
DiskIo = (EFI_DISK_IO_PROTOCOL *)*Interface;
DiskIo->ReadDisk = Ring3DiskIoRead;
DiskIo->WriteDisk = Ring3DiskIoWrite;
} else {
Status = EFI_UNSUPPORTED;
}
return Status;
} }
EFI_STATUS EFI_STATUS
@ -506,8 +529,6 @@ Ring3LocateProtocol (
{ {
EFI_STATUS Status; EFI_STATUS Status;
EFI_DEVICE_PATH_UTILITIES_PROTOCOL *DevicePath;
Status = SysCall ( Status = SysCall (
SysCallLocateProtocol, SysCallLocateProtocol,
Protocol, Protocol,
@ -519,22 +540,7 @@ Ring3LocateProtocol (
return Status; return Status;
} }
if (CompareGuid (Protocol, &gEfiDevicePathUtilitiesProtocolGuid)) { return FixInterface (Protocol, Interface);
DevicePath = (EFI_DEVICE_PATH_UTILITIES_PROTOCOL *)*Interface;
DevicePath->GetDevicePathSize = NULL;
DevicePath->DuplicateDevicePath = NULL;
DevicePath->AppendDevicePath = NULL;
DevicePath->AppendDeviceNode = NULL;
DevicePath->AppendDevicePathInstance = NULL;
DevicePath->GetNextDevicePathInstance = NULL;
DevicePath->IsDevicePathMultiInstance = NULL;
DevicePath->CreateDeviceNode = NULL;
return Status;
}
return EFI_UNSUPPORTED;
} }
EFI_STATUS EFI_STATUS