[fio.git] / lib / memalign.c

#include <assert.h>
#include <stdlib.h>

#include "memalign.h"

#define PTR_ALIGN(ptr, mask)   \
	(char *)((uintptr_t)((ptr) + (mask)) & ~(mask))

struct align_footer {
	unsigned int offset;
};

void *fio_memalign(size_t alignment, size_t size)
{
	struct align_footer *f;
	void *ptr, *ret = NULL;

	assert(!(alignment & (alignment - 1)));

	ptr = malloc(size + alignment + sizeof(*f) - 1);
	if (ptr) {
		ret = PTR_ALIGN(ptr, alignment - 1);
		f = ret + size;
		f->offset = (uintptr_t) ret - (uintptr_t) ptr;
	}

	return ret;
}

void fio_memfree(void *ptr, size_t size)
{
	struct align_footer *f = ptr + size;

	free(ptr - f->offset);
}
Commit	Line	Data
91aea6ec	1	#include <assert.h>
3d2d14bc	2	#include <stdlib.h>
91aea6ec JA	3
91aea6ec JA	4	#include "memalign.h"
087c0cba TK	5
	6	#define PTR_ALIGN(ptr, mask) \
	7	(char *)((uintptr_t)((ptr) + (mask)) & ~(mask))
91aea6ec JA	8
	9	struct align_footer {
	10	unsigned int offset;
	11	};
	12
91aea6ec JA	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
ba8b89a1	20	ptr = malloc(size + alignment + sizeof(*f) - 1);
91aea6ec	21	if (ptr) {
5fb55c77	22	ret = PTR_ALIGN(ptr, alignment - 1);
91aea6ec	23	f = ret + size;
e43606c2	24	f->offset = (uintptr_t) ret - (uintptr_t) ptr;
91aea6ec JA	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	}