/* radare - LGPL - Copyright 2019-2020 - pancake */

#include <r_util.h>
#include <r_util/r_alloc.h>

R_API void r_alloc_init(void) {
#if R_MALLOC_WRAPPER
	r_alloc_hooks (malloc, calloc, realloc, free);
#endif
}

#if R_MALLOC_WRAPPER

#if R_MALLOC_GLOBAL
R_API RMalloc *r_malloc = malloc;
R_API RCalloc *r_calloc = calloc;
R_API RRealloc *r_realloc = realloc;
R_API RFree *r_free = free;

R_API void r_alloc_hooks(RMalloc m, RCalloc c, RRealloc r, RFree f) {
	R_RETURN_IF_FAIL (m && c && r && f);
	r_malloc = m;
	r_calloc = c;
	r_realloc = r;
	r_free = f;
}

#else

static RMalloc *_r_malloc = malloc;
static RCalloc *_r_calloc = calloc;
static RRealloc *_r_realloc = realloc;
static RFree *_r_free = free;

R_API void r_alloc_hooks(RMalloc m, RCalloc c, RRealloc r, RFree f) {
	R_RETURN_IF_FAIL (m && c && r && f);
	_r_malloc = m;
	_r_calloc = c;
	_r_realloc = r;
	_r_free = f;
}

R_API void *r_malloc(size_t sz) {
	return _r_malloc (sz);
}

R_API void *r_calloc(size_t count, size_t sz) {
	return _r_calloc (count, sz);
}

R_API void *r_realloc(void *p, size_t sz) {
	return _r_realloc (p, sz);
}

R_API void r_free(void *p) {
	return _r_free (p);
}
#endif
#endif

R_API void* r_malloc_aligned(size_t size, size_t alignment) {
	int offset = alignment - 1 + sizeof (void*);
	void* p1 = _r_malloc (size + offset);
	if (!p1) {
		return NULL;
	}
	void **p2 = (void**)(((size_t)(p1) + offset) & ~(alignment - 1));
	p2[-1] = p1;
	return p2;
}

R_API void r_free_aligned(void *p) {
	_r_free (((void**)p)[-1]);
}