#include "plat.h"
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>

void *malloc_aligned(size_t size)
{
	void *mem = malloc(size + 8 + sizeof(void *) * 2);
	void **aligned_ptr = (void **)((uintptr_t)(mem + 8 + sizeof(void *)) & ~7);
	aligned_ptr[-1] = mem;
	aligned_ptr[-2] = (void *)size;
	return aligned_ptr;
}

void *realloc_aligned(void *addr, size_t size)
{
	void *mem = malloc(size + 8 + sizeof(void *) * 2);
	void **aligned_ptr = (void **)((uintptr_t)(mem + 8 + sizeof(void *)) & ~7);
	aligned_ptr[-1] = mem;

	memcpy(aligned_ptr, addr, (uintptr_t)aligned_ptr[-2]);

	return aligned_ptr;
}

void free_aligned(void *addr)
{
	void **ptr = (void **)addr;
	free(ptr[-1]);
}

void *link(dasm_State **Dst)
{
	size_t size;
	void *buf;

	dasm_link(Dst, &size);
	buf = mmap(0, size, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, -1,
	           0);
	dasm_encode(Dst, buf);

	mprotect(buf, size, PROT_READ | PROT_EXEC);
	return buf;
}