lib/memalign.c

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