2 files changed, 89 insertions, 1 deletions

diff --git a/UefiLessonsPkg/AcpiInfo/AcpiInfo.c b/UefiLessonsPkg/AcpiInfo/AcpiInfo.c
index 88c78b6..9b3c866 100644
--- a/UefiLessonsPkg/AcpiInfo/AcpiInfo.c
+++ b/UefiLessonsPkg/AcpiInfo/AcpiInfo.c
@@ -2,6 +2,75 @@
 #include <Library/UefiLib.h>
 
 #include <Library/BaseMemoryLib.h>
+#include <Protocol/Shell.h>
+
+EFI_SHELL_PROTOCOL* ShellProtocol;
+
+EFI_STATUS SaveACPITable(UINT32 Signature, VOID* addr, UINTN size) {
+  CHAR16 TableName[5];
+  TableName[0] = (CHAR16)((Signature>> 0)&0xFF);
+  TableName[1] = (CHAR16)((Signature>> 8)&0xFF);
+  TableName[2] = (CHAR16)((Signature>>16)&0xFF);
+  TableName[3] = (CHAR16)((Signature>>24)&0xFF);
+  TableName[4] = 0;
+
+  CHAR16 FileName[9] = {0};
+  StrCpyS(FileName, 9, TableName);
+  StrCatS(FileName, 9, L".aml");
+  SHELL_FILE_HANDLE FileHandle;
+  EFI_STATUS Status = ShellProtocol->OpenFileByName(FileName,
+                                                    &FileHandle,
+                                                    EFI_FILE_MODE_CREATE |
+                                                    EFI_FILE_MODE_WRITE |
+                                                    EFI_FILE_MODE_READ);
+  if (!EFI_ERROR(Status)) {
+    Status = ShellProtocol->WriteFile(FileHandle, &size, addr);
+    if (EFI_ERROR(Status)) {
+      Print(L"Error in WriteFile: %r\n", Status);
+    }
+    Status = ShellProtocol->CloseFile(FileHandle);
+    if (EFI_ERROR(Status)) {
+      Print(L"Error in CloseFile: %r\n", Status);
+    }
+  } else {
+    Print(L"Error in OpenFileByName: %r\n", Status);
+  }
+  return Status;
+}
+
+
+VOID CheckSubtables(EFI_ACPI_6_3_COMMON_HEADER* table)
+{
+  if (((CHAR8)((table->Signature >>  0) & 0xFF) == 'F') &&
+      ((CHAR8)((table->Signature >>  8) & 0xFF) == 'A') &&
+      ((CHAR8)((table->Signature >> 16) & 0xFF) == 'C') &&
+      ((CHAR8)((table->Signature >> 24) & 0xFF) == 'P')) {
+    EFI_ACPI_6_3_FIXED_ACPI_DESCRIPTION_TABLE* FADT = (EFI_ACPI_6_3_FIXED_ACPI_DESCRIPTION_TABLE*)table;
+
+    EFI_ACPI_6_3_COMMON_HEADER* DSDT = (EFI_ACPI_6_3_COMMON_HEADER*)(UINT64)(FADT->Dsdt);
+    if (((CHAR8)((DSDT->Signature >>  0) & 0xFF) == 'D') &&
+        ((CHAR8)((DSDT->Signature >>  8) & 0xFF) == 'S') &&
+        ((CHAR8)((DSDT->Signature >> 16) & 0xFF) == 'D') &&
+        ((CHAR8)((DSDT->Signature >> 24) & 0xFF) == 'T')) {
+      Print(L"\tDSDT table is placed at address %p with length 0x%x\n", DSDT, DSDT->Length);
+      SaveACPITable(DSDT->Signature, DSDT, DSDT->Length);
+    } else {
+      Print(L"\tError! DSDT signature is not valid!\n");
+    }
+
+    EFI_ACPI_6_3_COMMON_HEADER* FACS = (EFI_ACPI_6_3_COMMON_HEADER*)(UINT64)(FADT->FirmwareCtrl);
+    if (((CHAR8)((FACS->Signature >>  0) & 0xFF) == 'F') &&
+        ((CHAR8)((FACS->Signature >>  8) & 0xFF) == 'A') &&
+        ((CHAR8)((FACS->Signature >> 16) & 0xFF) == 'C') &&
+        ((CHAR8)((FACS->Signature >> 24) & 0xFF) == 'S')) {
+      Print(L"\tFACS table is placed at address %p with length 0x%x\n", FACS, FACS->Length);
+      SaveACPITable(FACS->Signature, FACS, FACS->Length);
+    } else {
+      Print(L"\tError! FACS signature is not valid!\n");
+    }
+  }
+}
+
 
 EFI_STATUS
 EFIAPI
@@ -10,6 +79,17 @@ UefiMain (
   IN EFI_SYSTEM_TABLE  *SystemTable
   )
 {
+  EFI_STATUS Status = gBS->LocateProtocol(
+    &gEfiShellProtocolGuid,
+    NULL,
+    (VOID **)&ShellProtocol
+  );
+
+  if (EFI_ERROR(Status)) {
+    Print(L"Can't open EFI_SHELL_PROTOCOL: %r\n", Status);
+    return EFI_SUCCESS;
+  }
+
   EFI_ACPI_6_3_ROOT_SYSTEM_DESCRIPTION_POINTER* RSDP = NULL;
 
   for (UINTN i=0; i<SystemTable->NumberOfTableEntries; i++) {
@@ -55,13 +135,18 @@ UefiMain (
   while (offset < XSDT->Length) {
     UINT64* table_address = (UINT64*)((UINT8*)XSDT + offset);
     EFI_ACPI_6_3_COMMON_HEADER* table = (EFI_ACPI_6_3_COMMON_HEADER*)(*table_address);
-    Print(L"\t%c%c%c%c table is placed at address %p with length 0x%x\n", 
+    Print(L"\t%c%c%c%c table is placed at address %p with length 0x%x\n",
                                              (CHAR8)((table->Signature>> 0)&0xFF),
                                              (CHAR8)((table->Signature>> 8)&0xFF),
                                              (CHAR8)((table->Signature>>16)&0xFF),
                                              (CHAR8)((table->Signature>>24)&0xFF),
                                              table,
                                              table->Length);
+
+    SaveACPITable(table->Signature, table, table->Length);
+
+    CheckSubtables(table);
+
     offset += sizeof(UINT64);
   }
 
diff --git a/UefiLessonsPkg/AcpiInfo/AcpiInfo.inf b/UefiLessonsPkg/AcpiInfo/AcpiInfo.inf
index 3884e76..53d0356 100644
--- a/UefiLessonsPkg/AcpiInfo/AcpiInfo.inf
+++ b/UefiLessonsPkg/AcpiInfo/AcpiInfo.inf
@@ -19,3 +19,6 @@
 [Guids]
   gEfiAcpi20TableGuid
 
+[Protocols]
+  gEfiShellProtocolGuid
+