Add sources for lessons 45/46/47

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

1 files changed, 111 insertions, 0 deletions

diff --git a/Lessons/Lesson_47/HIIStringsC/HIIStringsC.c b/Lessons/Lesson_47/HIIStringsC/HIIStringsC.c
new file mode 100644
index 0000000..307351f
--- /dev/null
+++ b/Lessons/Lesson_47/HIIStringsC/HIIStringsC.c
@@ -0,0 +1,111 @@
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/UefiLib.h>
+
+#include <Library/MemoryAllocationLib.h>
+#include <Library/UefiHiiServicesLib.h>
+
+
+EFI_STATUS PrintStringFromHiiHandle(EFI_HII_HANDLE* Handle, CHAR8* Language, UINTN StringId)
+{
+  EFI_STRING String = NULL;
+  UINTN StringSize = 0;
+  EFI_STATUS Status = gHiiString->GetString(gHiiString, Language, *Handle, StringId, String, &StringSize, NULL);
+
+  if (Status != EFI_BUFFER_TOO_SMALL) {
+    return Status;
+  }
+
+  String = AllocateZeroPool (StringSize);
+  if (String == NULL) {
+    return Status;
+  }
+
+  Status = gHiiString->GetString(gHiiString, Language, *Handle, StringId, String, &StringSize, NULL);
+
+  Print(L"Status = %r, %s\n", Status, String);
+
+  FreePool(String);
+
+  return EFI_SUCCESS;
+}
+
+
+UINT32 InitStringPackage(CHAR8* Ptr, CHAR8* Language, CHAR16** Strings, UINTN StringsLen)
+{
+  UINT32 Size = 0;
+  EFI_HII_STRING_PACKAGE_HDR* HIIStringPackageHdr = (EFI_HII_STRING_PACKAGE_HDR*)&Ptr[0];
+  HIIStringPackageHdr->Header.Type = EFI_HII_PACKAGE_STRINGS;
+  UINT32 HeaderSize = (UINT32) (AsciiStrSize(Language) - 1 + sizeof (EFI_HII_STRING_PACKAGE_HDR));
+  HIIStringPackageHdr->HdrSize = HeaderSize;
+  HIIStringPackageHdr->StringInfoOffset = HeaderSize;
+  HIIStringPackageHdr->LanguageName = 1;
+  AsciiStrCpyS (HIIStringPackageHdr->Language,
+                (HeaderSize - OFFSET_OF(EFI_HII_STRING_PACKAGE_HDR,Language)) / sizeof (CHAR8),
+                (CHAR8 *) Language);
+  Size += HeaderSize;
+
+  for (UINTN i=0; i<StringsLen; i++)
+  {
+    EFI_HII_SIBT_STRING_UCS2_BLOCK* StrBlock = (EFI_HII_SIBT_STRING_UCS2_BLOCK*)&Ptr[Size];
+    StrBlock->Header.BlockType = EFI_HII_SIBT_STRING_UCS2;
+    StrCpyS(StrBlock->StringText, StrLen(Strings[i])+1, Strings[i]);
+    Size += sizeof(EFI_HII_SIBT_STRING_UCS2_BLOCK) + StrLen(Strings[i])*2;  
+  }
+
+  EFI_HII_SIBT_END_BLOCK* EndBlock = (EFI_HII_SIBT_END_BLOCK*)&Ptr[Size];
+  EndBlock->Header.BlockType = EFI_HII_SIBT_END;
+  Size += sizeof(EFI_HII_SIBT_END_BLOCK);
+
+  HIIStringPackageHdr->Header.Length = Size;
+
+  return Size;
+}
+
+
+EFI_STATUS
+EFIAPI
+UefiMain (
+  IN EFI_HANDLE        ImageHandle,
+  IN EFI_SYSTEM_TABLE  *SystemTable
+  )
+{
+  CHAR8* Data = (CHAR8*) AllocateZeroPool(200);          // CHEAT! NEEDS CORRECTION FOR YOUR OWN PACKAGES!
+  UINT32 offset = 0;
+  EFI_HII_PACKAGE_LIST_HEADER* PackageListHdr = (EFI_HII_PACKAGE_LIST_HEADER*)&Data[offset];
+  PackageListHdr->PackageListGuid = gHIIStringsCGuid;
+  offset += sizeof(EFI_HII_PACKAGE_LIST_HEADER);
+
+  CHAR16* EnStrings[] = {
+    L"English",
+    L"Hello",
+  };
+  offset += InitStringPackage(&Data[offset], "en-US", EnStrings, sizeof(EnStrings)/sizeof(EnStrings[0]));
+
+  CHAR16* FrStrings[] = {
+    L"French",
+    L"Bonjour",
+  };
+  offset += InitStringPackage(&Data[offset], "fr-FR", FrStrings, sizeof(FrStrings)/sizeof(FrStrings[0]));
+
+  EFI_HII_PACKAGE_HEADER* HIIEndPackageHdr = (EFI_HII_PACKAGE_HEADER*)&Data[offset];
+  HIIEndPackageHdr->Type = EFI_HII_PACKAGE_END;
+  HIIEndPackageHdr->Length = sizeof(EFI_HII_PACKAGE_HEADER);
+  offset += sizeof(EFI_HII_PACKAGE_HEADER);
+
+  PackageListHdr->PackageLength = offset;
+
+  EFI_HII_HANDLE Handle;
+  EFI_STATUS Status = gHiiDatabase->NewPackageList(gHiiDatabase, PackageListHdr, NULL, &Handle);
+  if (EFI_ERROR(Status))
+  {
+    Print(L"Can't register HII Package list %g, status = %r\n", gHIIStringsCGuid, Status);
+  }
+  FreePool(Data);  
+
+  PrintStringFromHiiHandle(&Handle, "en-US", 1);
+  PrintStringFromHiiHandle(&Handle, "en-US", 2);
+  PrintStringFromHiiHandle(&Handle, "fr-FR", 1);
+  PrintStringFromHiiHandle(&Handle, "fr-FR", 2);
+
+  return EFI_SUCCESS;
+}