From df423fd45d6c46a227b018c58e9e5b8b237078f5 Mon Sep 17 00:00:00 2001
From: Joursoir <chat@joursoir.net>
Date: Wed, 27 Apr 2022 13:02:48 +0300
Subject: main: remove HII string packages

---
 UefiMonitorTest/UefiMonitorTest.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'UefiMonitorTest')

diff --git a/UefiMonitorTest/UefiMonitorTest.c b/UefiMonitorTest/UefiMonitorTest.c
index bfe75ba..047d14b 100644
--- a/UefiMonitorTest/UefiMonitorTest.c
+++ b/UefiMonitorTest/UefiMonitorTest.c
@@ -50,6 +50,22 @@ RegisterHiiPackage (
   return Status;
 }
 
+STATIC
+EFI_STATUS
+UnregisterHiiPackage (
+  IN  EFI_HII_HANDLE  HiiHandle
+  )
+{
+  EFI_STATUS  Status;
+
+  Status = gHiiDatabase->RemovePackageList(gHiiDatabase, HiiHandle);
+  if (EFI_ERROR(Status)) {
+    DEBUG ((DEBUG_WARN, "Failed to remove HII Package list from HII database: %r\n", Status));
+  }
+
+  return Status;
+}
+
 STATIC
 EFI_GRAPHICS_OUTPUT_PROTOCOL *
 GetGraphicsOutputProtocol (
@@ -150,5 +166,7 @@ UefiMain (
 
   ForgetGraphicsInfo (&Graphics);
 
+  UnregisterHiiPackage (gUmtHiiHandle);
+
   return Status;
 }
-- 
cgit v1.2.3-18-g5258