vpdf/system/memory.c

   1
   2 /*
   3         Simple memory allocation routines
   4 */
   5
   6 #include <exec/memory.h>
   7 #include <proto/exec.h>
   8 #include <string.h>
   9 #include <stdlib.h>
  10
  11 typedef struct memheader_t
  12 {
  13         unsigned int    magic;
  14         int                             size;
  15         int                             pad[2];                 /* to make header 16 bytes alligned */
  16 } MemHeader;
  17
  18 static APTR memoryPool = NULL;
  19
  20 static void MemoryExit(void);
  21 void kprintf(char *fmt,...);
  22
  23 void MemoryInit(void)
  24 {
  25         // initialize memory pool:
  26         // puddle size: 1MB
  27         // threshold : 256kB ( FIXME: find out better value )
  28
  29         memoryPool = CreatePool(MEMF_ANY | MEMF_SEM_PROTECTED, 1024*1024, 64*1024);
  30
  31         atexit(MemoryExit);
  32 }
  33
  34 static void MemoryExit(void)
  35 {
  36         if (memoryPool != NULL)
  37                 DeletePool(memoryPool);
  38 }
  39
  40
  41 void mfree(void *mem)
  42 {
  43         if (memoryPool != NULL)
  44         {
  45                 if (mem != NULL)
  46                 {
  47                         MemHeader *mh = (MemHeader*)( (unsigned char*)mem - sizeof(MemHeader));
  48                         unsigned int *endofmem = (unsigned int*)((unsigned char*)mem + mh->size);
  49
  50                         if (mh->magic != 0xDEADBEEF || *endofmem != 0xDEADBEEF)
  51                         {
  52                                 kprintf("[Memory]: Invalid memory header! ptr = 0x%x size = %d head = 0x%x, tail = 0x%x\n", mem, mh->size, mh->magic, *endofmem);
  53                                 DumpTaskState(FindTask(NULL));
  54                                 return;
  55                         }
  56                         else
  57                         {
  58                                 mh->magic = 0xABADCAFE;
  59                                 FreePooled(memoryPool, mh, mh->size + sizeof(MemHeader) + 4);
  60                         }
  61                 }
  62         }
  63 }
  64
  65 void *mmalloc(int size)
  66 {
  67         if (size == 0)
  68                 return NULL;
  69
  70         if (memoryPool != NULL)
  71         {
  72                 MemHeader *mh;
  73                 unsigned char *mem;
  74
  75                 mh = AllocPooledAligned(memoryPool, size + sizeof(MemHeader) + 4, 16, 0);
  76                 mem = (unsigned char*)mh + sizeof(MemHeader);
  77
  78                 if (mh != NULL)
  79                 {
  80             mh->size = size;
  81                         mh->magic = 0xDEADBEEF;
  82                         *(unsigned int*)(mem + size) = 0xDEADBEEF;
  83
  84                         return mem;
  85                 }
  86         }
  87
  88         return NULL;
  89
  90 }
  91
  92 void *mcalloc(int a, int b)
  93 {
  94         void *mem = mmalloc(a * b);
  95
  96         if (mem != NULL)
  97         {
  98                 memset(mem, 0, a * b);
  99                 return mem;
 100         }
 101
 102         return NULL;
 103 }
 104
 105