lib/memalign.c

   1 #include <assert.h>
   2 #include <stdlib.h>
   3
   4 #include "memalign.h"
   5 #include "smalloc.h"
   6
   7 #define PTR_ALIGN(ptr, mask)   \
   8         (char *)((uintptr_t)((ptr) + (mask)) & ~(mask))
   9
  10 struct align_footer {
  11         unsigned int offset;
  12 };
  13
  14 void *fio_memalign(size_t alignment, size_t size, bool shared)
  15 {
  16         struct align_footer *f;
  17         void *ptr, *ret = NULL;
  18
  19         assert(!(alignment & (alignment - 1)));
  20
  21         if (shared)
  22                 ptr = smalloc(size + alignment + sizeof(*f) - 1);
  23         else
  24                 ptr = malloc(size + alignment + sizeof(*f) - 1);
  25
  26         if (ptr) {
  27                 ret = PTR_ALIGN(ptr, alignment - 1);
  28                 f = ret + size;
  29                 f->offset = (uintptr_t) ret - (uintptr_t) ptr;
  30         }
  31
  32         return ret;
  33 }
  34
  35 void fio_memfree(void *ptr, size_t size, bool shared)
  36 {
  37         struct align_footer *f = ptr + size;
  38
  39         if (shared)
  40                 sfree(ptr - f->offset);
  41         else
  42                 free(ptr - f->offset);
  43 }