Kernel: Fix small out of memory issues

1 files changed, 3 insertions, 1 deletions

diff --git a/kernel/sched/scheduler.c b/kernel/sched/scheduler.c
index a2b6091..b249e3d 100644
--- a/kernel/sched/scheduler.c
+++ b/kernel/sched/scheduler.c
@@ -608,12 +608,14 @@ void *mmap(void *addr, size_t length, int prot, int flags, int fd,
     return (void *)-1;
   }
   *ptr = kmalloc(sizeof(MemoryMap));
+  if (!*ptr) {
+    return (void *)-ENOMEM;
+  }
   MemoryMap *free_map = *ptr;
 
   if (-1 == fd) {
     void *rc = allocate_virtual_user_memory(length, prot, flags);
     if (!rc) {
-      kprintf("ENOMEM\n");
       return (void *)-ENOMEM;
     }
     free_map->u_address = rc;