system/kernel/ke/dumbmalloc.c

   1
   2 #include "mm/memory.h"
   3 #include "ke/debug.h"
   4 #include "ke/level.h"
   5 #include "ke/spinlock.h"
   6 #include <string.h>
   7
   8 void *malloc(size_t size)
   9 {
  10         uint32_t memsize = size + 4;
  11         KeExecLevel oldlevel = keSetExecutionLevel(KE_LEVEL_HIGH);
  12         keLockSpinlock(mmGetMemoryLock());
  13
  14         uintptr_t vaddr = mmFindFreeKernelPages(MM_MAX_KERNEL_PAGE,
  15                 MM_MIN_KERNEL_PAGE, 1, (memsize + 0xFFF) & ~0xFFF);
  16         if (!vaddr)
  17         {
  18             kePrint("Out of kernel space.\n");
  19             return 0;
  20         }
  21         uint32_t i;
  22         for (i = 0; i < (memsize + 0xFFF) / 0x1000; ++i)
  23         {
  24         uintptr_t paddr = mmAllocPhysicalMemory(0, 0, 0x1000);
  25         if (!paddr)
  26         {
  27             kePrint("Out of kernel space.\n");
  28             return 0;
  29         }
  30             mmMapKernelMemory(paddr, vaddr + i * 0x1000, MM_MAP_READ | MM_MAP_WRITE);
  31         }
  32
  33         keUnlockSpinlock(mmGetMemoryLock());
  34         keSetExecutionLevel(oldlevel);
  35
  36         *(uint32_t*)vaddr = size;
  37
  38         memset((void*)(vaddr + 4), 0xF0, size);
  39
  40         return (void*)(vaddr + 4);
  41 }
  42
  43 void *calloc(size_t nmemb, size_t size)
  44 {
  45         void *data = malloc(nmemb * size);
  46         memset(data, 0, nmemb * size);
  47         return data;
  48 }
  49
  50 void free(void *ptr)
  51 {
  52         if (!ptr) return;
  53         uintptr_t addr = (uintptr_t)ptr - 4;
  54         uint32_t memsize = *(uint32_t*)addr + 4;
  55
  56         KeExecLevel oldlevel = keSetExecutionLevel(KE_LEVEL_HIGH);
  57         keLockSpinlock(mmGetMemoryLock());
  58
  59         uint32_t i;
  60         for (i = 0; i < (memsize + 0xFFF) / 0x1000; ++i)
  61         {
  62             uintptr_t vaddr = (uintptr_t)addr + i * 0x1000;
  63             uintptr_t paddr = mmKernelGetPhysAddress(vaddr);
  64             if (paddr != 0)
  65             mmFreePhysicalMemory(paddr, 0x1000);
  66         mmMapKernelMemory(paddr, vaddr, MM_MAP_READ | MM_MAP_WRITE);
  67         }
  68
  69         keUnlockSpinlock(mmGetMemoryLock());
  70         keSetExecutionLevel(oldlevel);
  71 }
  72
  73 void *realloc(void *ptr, size_t size)
  74 {
  75         uint32_t srcsize = 0;
  76         if (ptr) srcsize = *(uint32_t*)((uintptr_t)ptr - 4);
  77         if (size > srcsize)
  78         {
  79                 void *newdata = malloc(size);
  80                 memcpy(newdata, ptr, srcsize);
  81                 free(ptr);
  82                 return newdata;
  83         }
  84         else
  85                 return ptr;
  86 }
  87