Full UBSan support - many fixes!

1 files changed, 2 insertions, 0 deletions

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);