Add lesson 43

Signed-off-by: Konstantin Aladyshev <aladyshev22@gmail.com>

3 files changed, 141 insertions, 0 deletions

diff --git a/UefiLessonsPkg/ShowHII/ShowHII.c b/UefiLessonsPkg/ShowHII/ShowHII.c
new file mode 100644
index 0000000..878a270
--- /dev/null
+++ b/UefiLessonsPkg/ShowHII/ShowHII.c
@@ -0,0 +1,120 @@
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/UefiLib.h>
+
+#include <Protocol/HiiDatabase.h>
+#include <Library/MemoryAllocationLib.h>
+
+
+CHAR16* PackageType(UINTN Type)
+{
+  switch(Type) {
+    case EFI_HII_PACKAGE_TYPE_ALL:
+      return L"ALL";
+    case EFI_HII_PACKAGE_TYPE_GUID:
+      return L"GUID";
+    case EFI_HII_PACKAGE_FORMS:
+      return L"FORMS";
+    case EFI_HII_PACKAGE_STRINGS:
+      return L"STRINGS";
+    case EFI_HII_PACKAGE_FONTS:
+      return L"FONTS";
+    case EFI_HII_PACKAGE_IMAGES:
+      return L"IMAGES";
+    case EFI_HII_PACKAGE_SIMPLE_FONTS:
+      return L"SIMPLE_FONTS";
+    case EFI_HII_PACKAGE_DEVICE_PATH:
+      return L"DEVICE_PATH";
+    case EFI_HII_PACKAGE_KEYBOARD_LAYOUT:
+      return L"KEYBOARD_LAYOUT";
+    case EFI_HII_PACKAGE_ANIMATIONS:
+      return L"ANIMATIONS";
+    case EFI_HII_PACKAGE_END:
+      return L"END";
+    case EFI_HII_PACKAGE_TYPE_SYSTEM_BEGIN:
+      return L"SYSTEM_BEGIN";
+    case EFI_HII_PACKAGE_TYPE_SYSTEM_END:
+      return L"SYSTEM_END";
+  }
+  return L"UNKNOWN";
+}
+
+VOID ParseHiiPackageLists(EFI_HII_PACKAGE_LIST_HEADER* HiiDatabase, UINTN HiiDatabaseSize)
+{
+  EFI_HII_PACKAGE_LIST_HEADER* HiiPackageListHeader;
+  HiiPackageListHeader = (EFI_HII_PACKAGE_LIST_HEADER*) HiiDatabase;
+
+  UINTN i=0;
+  while ((UINTN) HiiPackageListHeader < ((UINTN) HiiDatabase + HiiDatabaseSize)) {
+    UINTN HiiPackageListSize = HiiPackageListHeader->PackageLength;
+    if (HiiPackageListSize == 0)
+      break;
+    Print(L"PackageList[%d]: GUID=%g; size=0x%X\n", i++, HiiPackageListHeader->PackageListGuid, HiiPackageListHeader->PackageLength);
+
+    EFI_HII_PACKAGE_HEADER* HiiPackageHeader = (EFI_HII_PACKAGE_HEADER*)((UINTN) HiiPackageListHeader + sizeof(EFI_HII_PACKAGE_LIST_HEADER));
+    UINTN j=0;
+    while ((UINTN) HiiPackageHeader < ((UINTN) HiiPackageListHeader + HiiPackageListSize)) {
+      Print(L"\tPackage[%d]: type=%s; size=0x%X\n", j++, PackageType(HiiPackageHeader->Type), HiiPackageHeader->Length);
+
+      // Go to next Package
+      HiiPackageHeader = (EFI_HII_PACKAGE_HEADER*)((UINTN) HiiPackageHeader + HiiPackageHeader->Length);
+    }
+
+    // Go to next PackageList
+    HiiPackageListHeader = (EFI_HII_PACKAGE_LIST_HEADER*)((UINTN) HiiPackageListHeader + HiiPackageListSize);
+  }
+}
+
+EFI_STATUS
+EFIAPI
+UefiMain (
+  IN EFI_HANDLE        ImageHandle,
+  IN EFI_SYSTEM_TABLE  *SystemTable
+  )
+{
+  EFI_STATUS Status;
+  EFI_HII_DATABASE_PROTOCOL* HiiDbProtocol;
+  Status = gBS->LocateProtocol(&gEfiHiiDatabaseProtocolGuid,
+                               NULL, 
+                               (VOID**)&HiiDbProtocol);
+  if (EFI_ERROR(Status)) {
+    Print(L"ERROR: Could not find HII Database protocol: %r\n", Status);
+    return Status;
+  }
+
+  UINTN PackageListSize = 0;
+  EFI_HII_PACKAGE_LIST_HEADER* PackageList = NULL;
+
+  Status = HiiDbProtocol->ExportPackageLists(HiiDbProtocol,
+                                             NULL,		// All package lists
+                                             &PackageListSize,
+                                             PackageList);
+  if (Status != EFI_BUFFER_TOO_SMALL) {
+    Print(L"ERROR: Could not obtain package list size\n");
+    return Status;
+  }
+
+  Status = gBS->AllocatePool(EfiBootServicesData, 
+                             PackageListSize, 
+                             (VOID**)&PackageList);
+  if (EFI_ERROR(Status)) {
+    Print(L"ERROR: Could not allocate sufficient memory for package list: %r\n", Status);
+    return Status;
+  }
+
+
+  Status = HiiDbProtocol->ExportPackageLists(HiiDbProtocol,
+                                             NULL,
+                                             &PackageListSize,
+                                             PackageList);
+  if (EFI_ERROR(Status)) {
+    Print(L"ERROR: Could not retrieve the package list: %r\n", Status);
+    FreePool(PackageList);
+    return Status;
+  }
+
+  ParseHiiPackageLists(PackageList, PackageListSize);
+
+  FreePool(PackageList);
+
+  return EFI_SUCCESS;
+}
diff --git a/UefiLessonsPkg/ShowHII/ShowHII.inf b/UefiLessonsPkg/ShowHII/ShowHII.inf
new file mode 100644
index 0000000..5585876
--- /dev/null
+++ b/UefiLessonsPkg/ShowHII/ShowHII.inf
@@ -0,0 +1,20 @@
+[Defines]
+  INF_VERSION                    = 1.25
+  BASE_NAME                      = ShowHII
+  FILE_GUID                      = 44c09516-1a1b-48cb-844d-3858887e57cc
+  MODULE_TYPE                    = UEFI_APPLICATION
+  VERSION_STRING                 = 1.0
+  ENTRY_POINT                    = UefiMain
+
+[Sources]
+  ShowHII.c
+
+[Packages]
+  MdePkg/MdePkg.dec
+
+[LibraryClasses]
+  UefiApplicationEntryPoint
+  UefiLib
+
+[Protocols]
+  gEfiHiiDatabaseProtocolGuid
diff --git a/UefiLessonsPkg/UefiLessonsPkg.dsc b/UefiLessonsPkg/UefiLessonsPkg.dsc
index 88d299f..7912ad7 100644
--- a/UefiLessonsPkg/UefiLessonsPkg.dsc
+++ b/UefiLessonsPkg/UefiLessonsPkg.dsc
@@ -58,6 +58,7 @@
   UefiLessonsPkg/SimpleClassProtocol/SimpleClassProtocol.inf
   UefiLessonsPkg/SimpleClassUser/SimpleClassUser.inf
   UefiLessonsPkg/HotKeyDriver/HotKeyDriver.inf
+  UefiLessonsPkg/ShowHII/ShowHII.inf
 
 [PcdsFixedAtBuild]
   gUefiLessonsPkgTokenSpaceGuid.PcdMyVar32_2|44