Implement userspace malloc

1 files changed, 26 insertions, 0 deletions

diff --git a/src/apps/init/main.c b/src/apps/init/main.c
index 0defd13..25f08e8 100644
--- a/src/apps/init/main.c
+++ b/src/apps/init/main.c
@@ -1,11 +1,37 @@
 #include <string.h>
 
+#include <malloc.h>
+
 #include <syscall.h>
 #include <debug.h>
+#include <user_region.h>
 
 int main(int argc, char **argv) {
 	dbg_print("Hello, world! from user process.\n");
 
+	dbg_print_region_info();
+
+	dbg_printf("Doing malloc test...\n");
+	const int m = 200;
+	uint16_t** ptr = malloc(m * sizeof(uint32_t));
+	for (int i = 0; i < m; i++) {
+		size_t s = 1 << ((i * 7) % 11 + 2);
+		ptr[i] = (uint16_t*)malloc(s);
+		ASSERT((size_t)ptr[i] >= 0x40000000 && (size_t)ptr[i] < 0xB0000000);
+		*ptr[i] = ((i * 211) % 1024);
+	}
+	dbg_printf("Fully allocated.\n");
+	dbg_print_region_info();
+	for (int i = 0; i < m; i++) {
+		for (int j = i; j < m; j++) {
+			ASSERT(*ptr[j] == (j * 211) % 1024);
+		}
+		free(ptr[i]);
+	}
+	free(ptr);
+	dbg_printf("malloc test OK.\n");
+	dbg_print_region_info();
+
 	return 0;
 }