load multiboot memory map, heap is done!!!

3 files changed, 181 insertions, 0 deletions

diff --git a/kernel/mboot/mboot.c b/kernel/mboot/mboot.c
new file mode 100644
index 0000000..08a0f37
--- /dev/null
+++ b/kernel/mboot/mboot.c
@@ -0,0 +1,26 @@
+
+#include "mboot.h"
+
+void *locate_mboot_table(volatile void *mboot, uint32_t type)
+{
+	struct mboot_info *info = (struct mboot_info *) mboot;
+	const char *mboot_end = ((char *) info) + info->total_size;
+
+	char *tag_ptr = info->tags;
+
+	while (tag_ptr < mboot_end) {
+		struct mboot_tag *tag = (struct mboot_tag *) tag_ptr;
+
+		if (tag->type == type)
+			return tag;
+
+		// goto next
+		int size = tag->size;
+		if (size % 8 != 0) {
+			size += 8 - (size % 8);
+		}
+		tag_ptr += size;
+	}
+
+	return NULL;
+}
diff --git a/kernel/mboot/mboot.h b/kernel/mboot/mboot.h
new file mode 100644
index 0000000..39a961d
--- /dev/null
+++ b/kernel/mboot/mboot.h
@@ -0,0 +1,106 @@
+/**
+ * @file mboot.h
+ *
+ * @author Freya Murphy <freya@freyacat.org>
+ *
+ * Internal multiboot structures
+ */
+
+#ifndef __MBOOT_H_
+#define __MBOOT_H_
+
+#include <lib.h>
+
+#define MBOOT_HEADER_MAGIC  0x36D76289
+
+#define MBOOT_CMDLINE       1
+#define MBOOT_MEMORY_MAP    6
+#define MBOOT_FRAMEBUFFER   8
+#define MBOOT_ELF_SYMBOLS   9
+#define MBOOT_OLD_RSDP     14
+#define MBOOT_NEW_RSDP     15
+
+struct mboot_info {
+	uint32_t total_size;
+	uint32_t reserved;
+	char tags[];
+};
+
+struct mboot_tag {
+	uint32_t type;
+	uint32_t size;
+	char data[];
+};
+
+struct mboot_tag_elf_sections {
+	uint32_t type;
+	uint32_t size;
+	uint16_t num;
+	uint16_t entsize;
+	uint16_t shndx;
+	uint16_t reserved;
+	char sections[];
+};
+
+struct mboot_tag_elf_sections_entry {
+	uint32_t sh_name;
+	uint32_t sh_type;
+	uint64_t sh_flags;
+	uint64_t sh_addr;
+	uint64_t sh_offset;
+	uint64_t sh_size;
+	uint32_t sh_link;
+	uint32_t sh_info;
+	uint64_t sh_addralign;
+	uint64_t sh_entsize;
+};
+
+struct mboot_mmap_entry {
+	uint64_t addr;
+	uint64_t len;
+	uint32_t type;
+	uint32_t zero;
+};
+
+struct mboot_tag_mmap {
+	uint32_t type;
+	uint32_t size;
+	uint32_t entry_size;
+	uint32_t entry_version;
+	struct mboot_mmap_entry entries[];
+};
+
+
+struct mboot_tag_old_rsdp {
+	uint32_t type;
+	uint32_t size;
+	char rsdp[];
+};
+
+struct mboot_tag_new_rsdp {
+	uint32_t type;
+	uint32_t size;
+	char rsdp[];
+};
+
+struct mboot_tag_cmdline {
+	uint32_t type;
+	uint32_t size;
+	uint8_t cmdline[];
+};
+
+struct mboot_tag_framebuffer {
+	uint32_t type;
+	uint32_t size;
+	uint64_t framebuffer_addr;
+	uint32_t framebuffer_pitch;
+	uint32_t framebuffer_width;
+	uint32_t framebuffer_height;
+	uint8_t framebuffer_bpp;
+	uint8_t framebuffer_type;
+	uint16_t reserved;
+};
+
+void *locate_mboot_table(volatile void *mboot, uint32_t type);
+
+#endif /* mboot.h */
diff --git a/kernel/mboot/mmap.c b/kernel/mboot/mmap.c
new file mode 100644
index 0000000..c17d510
--- /dev/null
+++ b/kernel/mboot/mmap.c
@@ -0,0 +1,49 @@
+#include <lib.h>
+#include <comus/mboot.h>
+
+#include "mboot.h"
+#include <stdint.h>
+#include <stdio.h>
+
+static const char *segment_type[] = {
+	"Reserved",
+	"Free",
+	"Reserved",
+	"ACPI Reserved",
+	"Hibernation",
+	"Defective",
+	"Unknown"
+};
+
+void mboot_load_mmap(volatile void *mboot, struct memory_map *res)
+{
+	void *tag = locate_mboot_table(mboot, MBOOT_MEMORY_MAP);
+	struct mboot_tag_mmap *mmap = (struct mboot_tag_mmap *) tag;
+
+	int idx = 0;
+	uintptr_t i = (uintptr_t)mmap->entries;
+	printf("MEMORY MAP\n");
+	char buf[20];
+	for (;
+			i < (uintptr_t)mmap->entries + mmap->size;
+			i += mmap->entry_size, idx++
+	) {
+		struct mboot_mmap_entry *seg = (struct mboot_mmap_entry *) i;
+		const char *type = NULL;
+		if (seg->type > 6)
+			type = segment_type[6];
+		else
+			type = segment_type[seg->type];
+		printf("ADDR: %16p  LEN: %4s  TYPE: %s (%d)\n",
+				(void *)seg->addr,
+				btoa(seg->len, buf),
+				type,
+				seg->type
+		);
+		if (seg->type != 1 || seg->len < 1)
+			continue;
+		res->entries[idx].addr = seg->addr;
+		res->entries[idx].len = seg->len;
+	}
+	res->entry_count = idx;
+}