Impl PMEM experimental code

Author: hikalium

.clang-format

@@ -0,0 +1 @@
+BasedOnStyle: Chromium

Makefile

@@ -42,3 +42,6 @@ run_pmem : src/BOOTX64.EFI pmem.img
 
 clean :
 	make -C src clean
+
+format :
+	make -C src format

src/Makefile

@@ -1,5 +1,5 @@
-OBJS=efimain.o
-HEADERS=efi.h
+OBJS=efi.o guid.o liumos.o static.o
+HEADERS=efi.h acpi.h liumos.h
 OVMF=ovmf/bios64.bin
 QEMU=qemu-system-x86_64
 
@@ -26,7 +26,7 @@ default: BOOTX64.EFI
 
 BOOTX64.EFI : $(OBJS) $(HEADERS) Makefile
 	$(LD) \
-		-subsystem:efi_application -nodefaultlib -dll \
+		-subsystem:efi_application -nodefaultlib \
 		-entry:efi_main $(OBJS) -out:$@
 
 run:
@@ -36,3 +36,6 @@ clean :
 	-rm *.EFI
 	-rm *.lib
 	-rm *.o
+
+format :
+	clang-format -i *.c *.h

src/acpi.h

@@ -0,0 +1,75 @@
+#pragma once
+
+#include "generic.h"
+#include "guid.h"
+
+#define packed_struct struct __attribute__((__packed__))
+#define ACPI_DESCRIPTION_HEADER_SIZE 36
+
+typedef struct ACPI_ROOT_SYSTEM_DESCRIPTION_POINTER ACPI_RSDP;
+typedef struct ACPI_EXTENDED_SYSTEM_DESCRIPTION_TABLE ACPI_XSDT;
+typedef struct ACPI_NVDIMM_FIRMWARE_INTERFACE_TABLE ACPI_NFIT;
+typedef struct ACPI_NFIT_SYSTEM_PHYSICAL_ADDRESS_RANGE_STRUCTURE
+    ACPI_NFIT_SPARange;
+
+packed_struct ACPI_ROOT_SYSTEM_DESCRIPTION_POINTER {
+  char signature[8];
+  uint8_t checksum;
+  uint8_t oem_id[6];
+  uint8_t revision;
+  uint32_t rsdt_address;
+  uint32_t length;
+  ACPI_XSDT* xsdt;
+  uint8_t extended_checksum;
+  uint8_t reserved;
+};
+
+packed_struct ACPI_EXTENDED_SYSTEM_DESCRIPTION_TABLE {
+  char signature[4];
+  uint32_t length;
+  uint8_t revision;
+  uint8_t checksum;
+  uint8_t oem_id[6];
+  uint64_t oem_table_id;
+  uint32_t oem_revision;
+  uint32_t creator_id;
+  uint32_t creator_revision;
+  void* entry[];
+};
+
+enum ACPI_NFITStructureType {
+  kSystemPhysicalAddressRangeStructure,
+  kMemoryDeviceToSystemAddressRangeMapStructure,
+  kInterleaveStructure,
+  kSMBIOSManagementInformationStructure,
+  kNVDIMMControlRegionStructure,
+  kNVDIMMBlockDataWindowRegionStructure,
+  kFlushHintAddressStructure,
+};
+
+packed_struct ACPI_NVDIMM_FIRMWARE_INTERFACE_TABLE {
+  char signature[4];
+  uint32_t length;
+  uint8_t revision;
+  uint8_t checksum;
+  uint8_t oem_id[6];
+  uint64_t oem_table_id;
+  uint32_t oem_revision;
+  uint32_t creator_id;
+  uint32_t creator_revision;
+  uint32_t reserved;
+  uint16_t entry[];
+};
+
+packed_struct ACPI_NFIT_SYSTEM_PHYSICAL_ADDRESS_RANGE_STRUCTURE {
+  uint16_t type;
+  uint16_t length;
+  uint16_t spa_range_structure_index;
+  uint16_t flags;
+  uint32_t reserved;
+  uint32_t proximity_domain;
+  uint64_t address_range_type_guid[2];
+  uint64_t system_physical_address_range_base;
+  uint64_t system_physical_address_range_length;
+  uint64_t address_range_memory_mapping_attribute;
+};

src/efi.c

@@ -0,0 +1,122 @@
+#include "efi.h"
+
+#define MEMORY_MAP_ENTRIES 64
+
+EFISystemTable* _system_table;
+EFIMemoryDescriptor memory_map[MEMORY_MAP_ENTRIES];
+EFI_UINTN memory_map_used;
+
+bool IsEqualStringWithSize(const char* s1, const char* s2, int n) {
+  for (int i = 0; i < n; i++) {
+    if (s1[i] != s2[i])
+      return false;
+  }
+  return true;
+}
+
+void EFIInit(struct EFI_SYSTEM_TABLE* system_table) {
+  _system_table = system_table;
+}
+
+void EFIClearScreen() {
+  _system_table->con_out->clear_screen(_system_table->con_out);
+}
+
+void EFIPutString(wchar_t* s) {
+  _system_table->con_out->output_string(_system_table->con_out, s);
+}
+
+void EFIPutCString(char* s) {
+  wchar_t buf[2];
+  buf[1] = 0;
+  while (*s) {
+    buf[0] = *s;
+    _system_table->con_out->output_string(_system_table->con_out, buf);
+    s++;
+  }
+}
+
+void EFIPutnCString(char* s, int n) {
+  wchar_t buf[2];
+  buf[1] = 0;
+  for (int i = 0; i < n; i++) {
+    buf[0] = s[i];
+    _system_table->con_out->output_string(_system_table->con_out, buf);
+  }
+}
+
+void EFIGetMemoryMap() {
+  EFI_UINTN map_key;
+  EFI_UINTN desc_size;
+  uint32_t desc_version;
+  EFI_UINTN memory_map_byte_size =
+      sizeof(EFIMemoryDescriptor) * MEMORY_MAP_ENTRIES;
+  _system_table->boot_services->GetMemoryMap(
+      &memory_map_byte_size, memory_map, &map_key, &desc_size, &desc_version);
+  memory_map_used = memory_map_byte_size / desc_size;
+}
+
+void EFIPrintHex64(uint64_t value) {
+  int i;
+  wchar_t s[2];
+  s[1] = 0;
+  for (i = 15; i > 0; i--) {
+    if ((value >> (4 * i)) & 0xF)
+      break;
+  }
+  for (; i >= 0; i--) {
+    s[0] = (value >> (4 * i)) & 0xF;
+    if (s[0] < 10)
+      s[0] += '0';
+    else
+      s[0] += 'A' - 10;
+    EFIPutString(s);
+  }
+}
+
+void EFIPrintStringAndHex(wchar_t* s, uint64_t value) {
+  EFIPutString(s);
+  EFIPutString(L": 0x");
+  EFIPrintHex64(value);
+  EFIPutString(L"\r\n");
+}
+
+void EFIPrintMemoryDescriptor(EFIMemoryDescriptor* desc) {
+  EFIPutString(L" Type ");
+  EFIPrintHex64(desc->type);
+  EFIPutString(L" Phys ");
+  EFIPrintHex64(desc->physical_start);
+  EFIPutString(L" Virt ");
+  EFIPrintHex64(desc->virtual_start);
+  EFIPutString(L" NumOfPages ");
+  EFIPrintHex64(desc->number_of_pages);
+  EFIPutString(L" Attr ");
+  EFIPrintHex64(desc->attribute);
+}
+
+void EFIPrintMemoryMap() {
+  EFIPrintStringAndHex(L"Map entries: ", memory_map_used);
+  for (int i = 0; i < memory_map_used; i++) {
+    EFIPutString(L"#");
+    EFIPrintHex64(i);
+    EFIPrintMemoryDescriptor(&memory_map[i]);
+    EFIPutString(L"\r\n");
+  }
+  EFIPutString(L"NV Map entries: \r\n");
+  for (int i = 0; i < memory_map_used; i++) {
+    if (!(memory_map[i].attribute & EFI_MEMORY_NV))
+      continue;
+    EFIPutString(L"#");
+    EFIPrintHex64(i);
+    EFIPrintMemoryDescriptor(&memory_map[i]);
+    EFIPutString(L"\r\n");
+  }
+}
+
+void* EFIGetConfigurationTableByUUID(const GUID* guid) {
+  for (int i = 0; i < _system_table->number_of_table_entries; i++) {
+    if (IsEqualGUID(guid, &_system_table->configuration_table[i].vendor_guid))
+      return _system_table->configuration_table[i].vendor_table;
+  }
+  return NULL;
+}