From f6422011e5e7070f5622a359b06264f1a3d046af Mon Sep 17 00:00:00 2001
From: Mike Maslenkin <mike.maslenkin@gmail.com>
Date: Thu, 7 Nov 2024 02:21:28 +0300
Subject: [PATCH] MdeModulePkg/PlatformDriOverrideDxe: fix HiiOpCodeHandle leak

Signed-off-by: Mike Maslenkin <mike.maslenkin@gmail.com>
---
 .../Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c     | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c b/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c
index 3aa7660324..24da4a40cb 100644
--- a/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c
+++ b/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c
@@ -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;
   }