1 files changed, 13 insertions, 10 deletions

diff --git a/src/memory/physalloc.c b/src/memory/physalloc.c
index b8eada6..7414d2e 100644
--- a/src/memory/physalloc.c
+++ b/src/memory/physalloc.c
@@ -7,8 +7,9 @@
 #include <memory/physalloc.h>
 #include <memory/virtalloc.h>
 
-extern uintptr_t kernel_start;
-extern uintptr_t kernel_end;
+extern char kernel_start;
+extern char kernel_end;
+#define kaddr(addr) ((uintptr_t)(&addr))
 
 // between memory_start and kernel_start will be the bitmap
 static uintptr_t memory_start = 0;
@@ -113,9 +114,9 @@ void free_phys_pages(void *ptr, int pages) {
 
 static bool segment_invalid(const struct memory_segment *segment) {
 	if (segment->type != 1) return false;
-	if (segment->addr < kernel_start) return false;
+	if (segment->addr < kaddr(kernel_start)) return false;
 	if (segment->addr + segment->len < memory_start) return false;
-	if (segment->addr + segment->len < kernel_start) return false;
+	if (segment->addr + segment->len < kaddr(kernel_start)) return false;
 	return true;
 }
 
@@ -127,7 +128,7 @@ static struct memory_area segment_to_area(const struct memory_segment *segment)
 	if (memory_start)
 		start = memory_start;
 	else
-		start = kernel_end;
+		start = kaddr(kernel_end);
 
 	if (segment->addr < start) {
 		addr = start;
@@ -144,7 +145,7 @@ static struct memory_area segment_to_area(const struct memory_segment *segment)
 }
 
 static uintptr_t page_align(uintptr_t ptr) {
-	return ptr + PAGE_SIZE - 1 / PAGE_SIZE;
+	return (ptr + PAGE_SIZE - 1) / PAGE_SIZE * PAGE_SIZE;
 }
 
 void memory_init(struct memory_map *map) {
@@ -165,7 +166,7 @@ void memory_init(struct memory_map *map) {
 
 	struct memory_segment *segment = &map->entries[0];
 	int segment_count = 0;
-
+	
 	for(; (uintptr_t) segment < end; segment++) {
 		if (segment_invalid(segment))
 			continue;
@@ -180,15 +181,17 @@ void memory_init(struct memory_map *map) {
 
 	//HACK: terrible hack bad bad bad bad
 	long bitmap_size = bitmap_pages * PAGE_SIZE;
-	bitmap = (uint64_t *) page_at(page_count - bitmap_pages);
+	bitmap = (uint64_t *) page_align(kaddr(kernel_end));
+	char buf[20];
 
 	long page_area_size = segment_count * sizeof(struct memory_area);
 	char *page_area_addr = (char *)bitmap + bitmap_size;
-	
 	bitmap = mmap(bitmap, bitmap_size);
+	ltoa((long)bitmap, buf, 16);
+	memset(bitmap, 0, 12);
 	memset(bitmap, 0, bitmap_size);
 	
-	memory_start = page_align(kernel_end + bitmap_size + page_area_size);
+	memory_start = page_align(kaddr(kernel_end) + bitmap_size + page_area_size);
 
 	page_area_addr = mmap(page_area_addr, page_area_size);
 	page_start = (struct memory_area *) page_area_addr;