diff options
| -rw-r--r-- | sys/include/vm/vm.h | 2 | ||||
| -rw-r--r-- | sys/vm/vm_dynalloc.c | 14 |
2 files changed, 14 insertions, 2 deletions
diff --git a/sys/include/vm/vm.h b/sys/include/vm/vm.h index 59cf3bc..2b4ccad 100644 --- a/sys/include/vm/vm.h +++ b/sys/include/vm/vm.h @@ -33,6 +33,7 @@ #include <sys/types.h> #include <sys/limine.h> #include <sys/cdefs.h> +#include <sys/spinlock.h> #include <vm/page.h> #include <vm/pmap.h> #include <vm/tlsf.h> @@ -53,6 +54,7 @@ struct vm_ctx { uintptr_t dynalloc_pool_phys; size_t dynalloc_pool_sz; /* In bytes */ tlsf_t tlsf_ctx; + struct spinlock dynalloc_lock; }; /* 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); } |