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