From 6064c1e48b622f53538f4df9bdd402c607a87d51 Mon Sep 17 00:00:00 2001
From: Konstantin Aladyshev <aladyshev22@gmail.com>
Date: Sat, 10 Jul 2021 00:04:40 +0300
Subject: Move lessons to separate folder

Signed-off-by: Konstantin Aladyshev <aladyshev22@gmail.com>
---
 .../UefiLessonsPkg/MemoryInfo/MemoryInfo.c         | 134 +++++++++++++++++++++
 .../UefiLessonsPkg/MemoryInfo/MemoryInfo.inf       |  18 +++
 2 files changed, 152 insertions(+)
 create mode 100644 Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.c
 create mode 100644 Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.inf

(limited to 'Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo')

diff --git a/Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.c b/Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.c
new file mode 100644
index 0000000..a8fcb83
--- /dev/null
+++ b/Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.c
@@ -0,0 +1,134 @@
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/UefiLib.h>
+
+// for SetMem
+#include <Library/BaseMemoryLib.h>
+
+const CHAR16 *memory_types[] = { 
+    L"EfiReservedMemoryType", 
+    L"EfiLoaderCode", 
+    L"EfiLoaderData", 
+    L"EfiBootServicesCode", 
+    L"EfiBootServicesData", 
+    L"EfiRuntimeServicesCode", 
+    L"EfiRuntimeServicesData", 
+    L"EfiConventionalMemory", 
+    L"EfiUnusableMemory", 
+    L"EfiACPIReclaimMemory", 
+    L"EfiACPIMemoryNVS", 
+    L"EfiMemoryMappedIO", 
+    L"EfiMemoryMappedIOPortSpace", 
+    L"EfiPalCode",
+    L"EfiPersistentMemory",
+    L"EfiMaxMemoryType"
+};
+
+const CHAR16 * 
+memory_type_to_str(UINT32 type) 
+{ 
+    if (type > sizeof(memory_types)/sizeof(CHAR16 *)) 
+        return L"Unknown"; 
+
+    return memory_types[type]; 
+}
+
+#define ATTRIBUTE_STR_SIZE 50
+
+#define CHECK_EFI_MEMORY_ATTRIBUTE(attr) if (attrs & EFI_MEMORY_##attr) { \
+                                           StrCpyS(&str[i], ATTRIBUTE_STR_SIZE, L" "#attr); \
+                                           i+=StrLen(L" "#attr); \
+                                         }
+
+const CHAR16 *
+memory_attrs_to_str(CHAR16* str, UINT64 attrs)
+{
+  int i=0;
+  SetMem((VOID *)str, sizeof(str), 0);
+
+  CHECK_EFI_MEMORY_ATTRIBUTE(UC)
+  CHECK_EFI_MEMORY_ATTRIBUTE(WC)
+  CHECK_EFI_MEMORY_ATTRIBUTE(WT)
+  CHECK_EFI_MEMORY_ATTRIBUTE(WB)
+  CHECK_EFI_MEMORY_ATTRIBUTE(UCE)
+  CHECK_EFI_MEMORY_ATTRIBUTE(WP)
+  CHECK_EFI_MEMORY_ATTRIBUTE(RP)
+  CHECK_EFI_MEMORY_ATTRIBUTE(XP)
+  CHECK_EFI_MEMORY_ATTRIBUTE(NV)
+  CHECK_EFI_MEMORY_ATTRIBUTE(MORE_RELIABLE)
+  CHECK_EFI_MEMORY_ATTRIBUTE(RO)
+  CHECK_EFI_MEMORY_ATTRIBUTE(SP)
+  CHECK_EFI_MEMORY_ATTRIBUTE(CPU_CRYPTO)
+  CHECK_EFI_MEMORY_ATTRIBUTE(RUNTIME)
+
+  return str;
+}
+
+EFI_STATUS
+EFIAPI
+UefiMain (
+  IN EFI_HANDLE        ImageHandle,
+  IN EFI_SYSTEM_TABLE  *SystemTable
+  )
+{
+  UINTN MemoryMapSize = 0;
+  EFI_MEMORY_DESCRIPTOR* MemoryMap = NULL;
+  UINTN MapKey;
+  UINTN DescriptorSize;
+  UINT32 DescriptorVersion;
+
+  EFI_STATUS Status;
+  Status = gBS->GetMemoryMap(
+        &MemoryMapSize,
+        MemoryMap,
+        &MapKey,
+        &DescriptorSize,
+        &DescriptorVersion
+  );
+
+  if (Status == EFI_BUFFER_TOO_SMALL) {
+    Status = gBS->AllocatePool(
+          EfiBootServicesData,
+          MemoryMapSize,
+          (void**)&MemoryMap
+    );
+
+    if (EFI_ERROR(Status)) {
+      Print(L"AllocatePool error: %r\n", Status);
+      return Status;
+    }
+
+    Status = gBS->GetMemoryMap(
+          &MemoryMapSize,
+          MemoryMap,
+          &MapKey,
+          &DescriptorSize,
+          &DescriptorVersion
+    );
+
+    if (!EFI_ERROR(Status))
+    {
+      EFI_MEMORY_DESCRIPTOR* desc = MemoryMap;
+      int i = 0;
+      while ((UINT8 *)desc <  (UINT8 *)MemoryMap + MemoryMapSize) {
+        UINTN PAGE_SIZE = 4096;
+        UINTN mapping_size =(UINTN) desc->NumberOfPages * PAGE_SIZE;
+
+        
+        CHAR16 str[ATTRIBUTE_STR_SIZE];
+        Print(L"[#%02d] Type: %s  Attr: %s\n", i++, memory_type_to_str(desc->Type), memory_attrs_to_str(str, desc->Attribute));
+        Print(L"      Phys: %016llx-%016llx\n", desc->PhysicalStart, desc->PhysicalStart + mapping_size - 1);
+
+        desc = (EFI_MEMORY_DESCRIPTOR *)((UINT8 *)desc + DescriptorSize);
+      }
+
+      gBS->FreePool(MemoryMap);
+    } else {
+      Print(L"GetMemoryMap with buffer error: %r\n", Status);
+    }
+  } else {
+    Print(L"GetMemoryMap without buffer error: %r\n", Status);
+  }
+
+  return Status;
+}
+
diff --git a/Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.inf b/Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.inf
new file mode 100644
index 0000000..d18c087
--- /dev/null
+++ b/Lessons/Lesson_11/UefiLessonsPkg/MemoryInfo/MemoryInfo.inf
@@ -0,0 +1,18 @@
+[Defines]
+  INF_VERSION                    = 1.25
+  BASE_NAME                      = MemoryInfo
+  FILE_GUID                      = d2ce1d65-603e-4b48-b3e1-fe65d5d0bca8
+  MODULE_TYPE                    = UEFI_APPLICATION
+  VERSION_STRING                 = 1.0
+  ENTRY_POINT                    = UefiMain
+
+[Sources]
+  MemoryInfo.c
+
+[Packages]
+  MdePkg/MdePkg.dec
+
+[LibraryClasses]
+  UefiApplicationEntryPoint
+  UefiLib
+
-- 
cgit v1.2.3-18-g5258