1 files changed, 115 insertions, 0 deletions
diff --git a/core/mem/tests/meminit.c b/core/mem/tests/meminit.c
new file mode 100644
index 00000000..e6c25c71
--- /dev/null
+++ b/core/mem/tests/meminit.c
@@ -0,0 +1,115 @@
+#include "unittest/unittest.h"
+#include "unittest/memmap.h"
+
+/*
+ * Fake data objects.
+ *
+ * These are the dependencies required by mem_init().
+ */
+struct com32_sys_args {
+     unsigned long cs_memsize;
+} __com32 = {
+     .cs_memsize = 4
+};
+char __lowmem_heap[32];
+char free_high_memory[32];
+
+#include "../init.c"
+
+void __inject_free_block(struct free_arena_header *ah)
+{
+}
+
+static unsigned long free_start = (unsigned long)free_high_memory;
+
+static inline bool free_list_empty(void)
+{
+    if (__com32.cs_memsize != free_start)
+	return false;
+
+    return true;
+}
+
+static struct test_memmap_entry *__test_entries;
+static size_t __test_nr_entries;
+
+int syslinux_scan_memory(scan_memory_callback_t callback, void *data)
+{
+    struct test_memmap_entry *e;
+    int i;
+
+    for (i = 0; i < __test_nr_entries; i++) {
+	e = &__test_entries[i];
+	callback(data, e->start, e->size, e->type);
+    }
+
+    return 0;
+}
+
+void __setup(struct test_memmap_entry *entries, size_t nr_entries)
+{
+    uint16_t __fake_free_mem = 64;
+
+    bios_free_mem = &__fake_free_mem;
+
+    __test_entries = entries;
+    __test_nr_entries = nr_entries;
+}
+
+/*
+ * scan_highmem_area() will prepend a free arena header if the size of
+ * the region is larger than the following expression. Using this small
+ * size allows us to test the interface safely without worrying about
+ * scan_highmem_area() writing data to random parts of our address
+ * space.
+ */
+#define safe_entry_sz	((2 * sizeof(struct arena_header)) - 1)
+
+/*
+ * Can we add SMT_RESERVED regions to the free list?
+ */
+static int test_mem_init_reserved(void)
+{
+    struct test_memmap_entry entries[] = {
+	0x2000,   safe_entry_sz, SMT_RESERVED,
+	0x100000, safe_entry_sz, SMT_RESERVED,
+	0x2fffff, safe_entry_sz, SMT_RESERVED,
+	0x400000, safe_entry_sz, SMT_RESERVED,
+    };
+
+    __setup(entries, array_sz(entries));
+
+    mem_init();
+    syslinux_assert_str(free_list_empty(),
+			"Added SMT_RESERVED regions to free list");
+    return 0;
+}
+
+/*
+ * Can we add regions outside of the valid address range?
+ */
+static int test_mem_limits(void)
+{
+    struct test_memmap_entry entries[] = {
+	0x00000000,   safe_entry_sz, SMT_FREE,
+	0x000fffff,   safe_entry_sz, SMT_FREE,
+	E820_MEM_MAX + 1, safe_entry_sz, SMT_FREE,
+    };
+
+    __setup(entries, array_sz(entries));
+
+    mem_init();
+    syslinux_assert_str(free_list_empty(),
+			"Added regions outside of valid range to free list");
+
+    return 0;
+}
+
+
+int main(int argc, char **argv)
+{
+    test_mem_init_reserved();
+    test_mem_limits();
+
+    return 0;
+}