blob: 7a04ffd8c31aa38443034fa300942bb8adf2259e [file] [log] [blame]
Jens Axboe91aea6e2011-04-25 20:46:58 +02001#include <stdlib.h>
2#include <assert.h>
Bruce Crane43606c2012-02-20 09:34:24 +01003#include <inttypes.h>
Jens Axboe91aea6e2011-04-25 20:46:58 +02004
5#include "memalign.h"
6
7struct align_footer {
8 unsigned int offset;
9};
10
11#define PTR_ALIGN(ptr, mask) \
Bruce Crane43606c2012-02-20 09:34:24 +010012 (char *) (((uintptr_t) ((ptr) + (mask)) & ~(mask)))
Jens Axboe91aea6e2011-04-25 20:46:58 +020013
14void *fio_memalign(size_t alignment, size_t size)
15{
16 struct align_footer *f;
17 void *ptr, *ret = NULL;
18
19 assert(!(alignment & (alignment - 1)));
20
21 ptr = malloc(size + alignment + size + sizeof(*f) - 1);
22 if (ptr) {
23 ret = PTR_ALIGN(ptr, alignment);
24 f = ret + size;
Bruce Crane43606c2012-02-20 09:34:24 +010025 f->offset = (uintptr_t) ret - (uintptr_t) ptr;
Jens Axboe91aea6e2011-04-25 20:46:58 +020026 }
27
28 return ret;
29}
30
31void fio_memfree(void *ptr, size_t size)
32{
33 struct align_footer *f = ptr + size;
34
35 free(ptr - f->offset);
36}