MdeModulePkg/PlatformDriOverrideDxe: fix HiiOpCodeHandle leak

Signed-off-by: Mike Maslenkin <mike.maslenkin@gmail.com>
This commit is contained in:
Mike Maslenkin 2024-11-07 02:21:28 +03:00 committed by mergify[bot]
parent 1b283cf437
commit f6422011e5

View File

@ -421,6 +421,8 @@ UpdateDeviceSelectPage (
&mDevicePathHandleBuffer
);
if (EFI_ERROR (Status) || (DevicePathHandleCount == 0)) {
HiiFreeOpCodeHandle (StartOpCodeHandle);
HiiFreeOpCodeHandle (EndOpCodeHandle);
return EFI_SUCCESS;
}
@ -708,6 +710,8 @@ UpdateBindingDriverSelectPage (
&mDriverImageHandleBuffer
);
if (EFI_ERROR (Status) || (DriverImageHandleCount == 0)) {
HiiFreeOpCodeHandle (StartOpCodeHandle);
HiiFreeOpCodeHandle (EndOpCodeHandle);
return EFI_NOT_FOUND;
}