From 5cd84ad93f7d67cdaa134707fcce1b1ef85183ad Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Fri, 26 Feb 2021 18:40:37 +0100
Subject: Full UBSan support - many fixes!

---
 libc/mem.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'libc/mem.c')

diff --git a/libc/mem.c b/libc/mem.c
index 7e9590a..971315a 100644
--- a/libc/mem.c
+++ b/libc/mem.c
@@ -291,8 +291,10 @@ void heap_init(u32 start)
 	heap.end = (u32)start + HEAP_INIT_SIZE;
 }
 
+#define ALIGN sizeof(long)
 static void *_malloc(u32 size)
 {
+	size = ((size + ALIGN - 1) / ALIGN) * ALIGN; // Alignment
 	u32 index = bin_index(size);
 	struct h_bin *temp = (struct h_bin *)&heap.bins[index];
 	struct h_node *found = node_best_fit(temp, size);
-- 
cgit v1.2.3