Add lesson 20

1 files changed, 46 insertions, 0 deletions

diff --git a/Lesson_20/UefiLessonsPkg/ListVariables/ListVariables.c b/Lesson_20/UefiLessonsPkg/ListVariables/ListVariables.c
new file mode 100644
index 0000000..1f3ef4a
--- /dev/null
+++ b/Lesson_20/UefiLessonsPkg/ListVariables/ListVariables.c
@@ -0,0 +1,46 @@
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/UefiLib.h>
+
+
+#include <Library/MemoryAllocationLib.h>
+#include <Library/UefiRuntimeServicesTableLib.h>
+
+INTN EFIAPI ShellAppMain(IN UINTN Argc, IN CHAR16 **Argv)
+{
+  EFI_GUID VendorGuid;
+  UINTN VariableNameSize = sizeof (CHAR16);
+  CHAR16* VariableName = AllocateZeroPool(sizeof(CHAR16));
+  if (VariableName == NULL) {
+    Print(L"Error on AllocateZeroPool call\n");
+    return EFI_OUT_OF_RESOURCES;
+  }
+
+  while (TRUE)
+  {
+    UINTN VariableNameSizeOld = VariableNameSize;
+    EFI_STATUS Status = gRT->GetNextVariableName(&VariableNameSize, VariableName, &VendorGuid);
+    if (Status == EFI_SUCCESS) {
+      Print(L"%g: %s\n", VendorGuid, VariableName);
+    } else if (Status == EFI_BUFFER_TOO_SMALL) {
+      VariableName = ReallocatePool(VariableNameSizeOld, VariableNameSize, VariableName);
+      if (VariableName == NULL) {
+        Print(L"Error on ReallocatePool call\n");
+        return EFI_OUT_OF_RESOURCES;
+      }
+      Status = gRT->GetNextVariableName(&VariableNameSize, VariableName, &VendorGuid);
+      if (Status == EFI_SUCCESS) {
+        Print(L"%g: %s\n", VendorGuid, VariableName);
+      } else {
+        Print(L"Error on 'gRT->GetNextVariableName' call: %s\n", Status);
+        return Status;
+      }
+    } else if (Status == EFI_NOT_FOUND) {
+      FreePool(VariableName);
+      return EFI_SUCCESS;
+    } else {
+      Print(L"Error on 'gRT->GetNextVariableName' call: %s\n", Status);
+      return Status;
+    }
+  }
+  return EFI_SUCCESS;
+}