kernel: vm: Ensure calls are serialized

Signed-off-by: Ian Moffett <ian@osmora.org>

1 files changed, 12 insertions, 2 deletions

diff --git a/sys/vm/vm_dynalloc.c b/sys/vm/vm_dynalloc.c
index e7acd19..a656a00 100644
--- a/sys/vm/vm_dynalloc.c
+++ b/sys/vm/vm_dynalloc.c
@@ -40,8 +40,12 @@ dynalloc(size_t sz)
 {
     /* TODO: Per CPU */
     struct vm_ctx vm_ctx = vm_get_bsp_ctx();
+    void *tmp;
 
-    return tlsf_malloc(vm_ctx.tlsf_ctx, sz);
+    spinlock_acquire(&vm_ctx.dynalloc_lock);
+    tmp = tlsf_malloc(vm_ctx.tlsf_ctx, sz);
+    spinlock_release(&vm_ctx.dynalloc_lock);
+    return tmp;
 }
 
 /*
@@ -55,8 +59,12 @@ dynrealloc(void *old_ptr, size_t newsize)
 {
     /* TODO: Per CPU */
     struct vm_ctx vm_ctx = vm_get_bsp_ctx();
+    void *tmp;
 
-    return tlsf_realloc(vm_ctx.tlsf_ctx, old_ptr, newsize);
+    spinlock_acquire(&vm_ctx.dynalloc_lock);
+    tmp = tlsf_realloc(vm_ctx.tlsf_ctx, old_ptr, newsize);
+    spinlock_release(&vm_ctx.dynalloc_lock);
+    return tmp;
 }
 
 /*
@@ -70,5 +78,7 @@ dynfree(void *ptr)
     /* TODO: Per CPU */
     struct vm_ctx vm_ctx = vm_get_bsp_ctx();
 
+    spinlock_acquire(&vm_ctx.dynalloc_lock);
     tlsf_free(vm_ctx.tlsf_ctx, ptr);
+    spinlock_release(&vm_ctx.dynalloc_lock);
 }