/**

 * WinPR: Windows Portable Runtime

 * Data Alignment

 *

 * Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <winpr/crt.h>

/* Data Alignment: http://msdn.microsoft.com/en-us/library/fs9stz4e/ */

#ifndef _WIN32

#include <stdint.h>

#include <limits.h>

#define WINPR_ALIGNED_MEM_SIGNATURE 0x0BA0BAB

#define WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(_memptr) \

	(WINPR_ALIGNED_MEM*)(((size_t)(((BYTE*)_memptr) - sizeof(WINPR_ALIGNED_MEM))));

#include <stdlib.h>

#ifdef __APPLE__

#include <malloc/malloc.h>

#elif __FreeBSD__ || __OpenBSD__

#include <stdlib.h>

#else

#include <malloc.h>

#endif

#include "../log.h"

#define TAG WINPR_TAG("crt")

struct winpr_aligned_mem

{

	UINT32 sig;

	size_t size;

	void* base_addr;

};

typedef struct winpr_aligned_mem WINPR_ALIGNED_MEM;

void* _aligned_malloc(size_t size, size_t alignment)

{

	return _aligned_offset_malloc(size, alignment, 0);

}

void* _aligned_realloc(void* memblock, size_t size, size_t alignment)

{

	return _aligned_offset_realloc(memblock, size, alignment, 0);

}

void* _aligned_recalloc(void* memblock, size_t num, size_t size, size_t alignment)

{

	return _aligned_offset_recalloc(memblock, num, size, alignment, 0);

}

void* _aligned_offset_malloc(size_t size, size_t alignment, size_t offset)

{

	size_t header, alignsize;

	uintptr_t basesize;

	void* base;

	void* memblock;

	WINPR_ALIGNED_MEM* pMem;

	/* alignment must be a power of 2 */

	if (alignment % 2 == 1)

		return NULL;

	/* offset must be less than size */

	if (offset >= size)

		return NULL;

	/* minimum alignment is pointer size */

	if (alignment < sizeof(void*))

		alignment = sizeof(void*);

	if (alignment > SIZE_MAX - sizeof(WINPR_ALIGNED_MEM))

		return NULL;

	header = sizeof(WINPR_ALIGNED_MEM) + alignment;

	if (size > SIZE_MAX - header)

		return NULL;

	alignsize = size + header;

	/* malloc size + alignment to make sure we can align afterwards */

	base = malloc(alignsize);

	if (!base)

		return NULL;

	basesize = (uintptr_t)base;

	if ((offset > UINTPTR_MAX) || (header > UINTPTR_MAX - offset) ||

	    (basesize > UINTPTR_MAX - header - offset))

	{

		free(base);

		return NULL;

	}

	memblock = (void*)(((basesize + header + offset) & ~(alignment - 1)) - offset);

	pMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(memblock);

	pMem->sig = WINPR_ALIGNED_MEM_SIGNATURE;

	pMem->base_addr = base;

	pMem->size = size;

	return memblock;

}

void* _aligned_offset_realloc(void* memblock, size_t size, size_t alignment, size_t offset)

{

	size_t copySize;

	void* newMemblock;

	WINPR_ALIGNED_MEM* pMem;

	WINPR_ALIGNED_MEM* pNewMem;

	if (!memblock)

		return _aligned_offset_malloc(size, alignment, offset);

	pMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(memblock);

	if (pMem->sig != WINPR_ALIGNED_MEM_SIGNATURE)

	{

		WLog_ERR(TAG,

		         "_aligned_offset_realloc: memory block was not allocated by _aligned_malloc!");

		return NULL;

	}

	if (size == 0)

	{

		_aligned_free(memblock);

		return NULL;

	}

	newMemblock = _aligned_offset_malloc(size, alignment, offset);

	if (!newMemblock)

		return NULL;

	pNewMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(newMemblock);

	copySize = (pNewMem->size < pMem->size) ? pNewMem->size : pMem->size;

	CopyMemory(newMemblock, memblock, copySize);

	_aligned_free(memblock);

	return newMemblock;

}

void* _aligned_offset_recalloc(void* memblock, size_t num, size_t size, size_t alignment,

                               size_t offset)

{

	void* newMemblock;

	WINPR_ALIGNED_MEM* pMem;

	WINPR_ALIGNED_MEM* pNewMem;

	if (!memblock)

	{

		newMemblock = _aligned_offset_malloc(size * num, alignment, offset);

		if (newMemblock)

		{

			pNewMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(newMemblock);

			ZeroMemory(newMemblock, pNewMem->size);

		}

		return memblock;

	}

	pMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(memblock);

	if (pMem->sig != WINPR_ALIGNED_MEM_SIGNATURE)

	{

		WLog_ERR(TAG,

		         "_aligned_offset_recalloc: memory block was not allocated by _aligned_malloc!");

		return NULL;

	}

	if (size == 0)

	{

		_aligned_free(memblock);

		return NULL;

	}

	newMemblock = _aligned_offset_malloc(size * num, alignment, offset);

	if (!newMemblock)

		return NULL;

	pNewMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(newMemblock);

	ZeroMemory(newMemblock, pNewMem->size);

	_aligned_free(memblock);

	return newMemblock;

}

size_t _aligned_msize(void* memblock, size_t alignment, size_t offset)

{

	WINPR_ALIGNED_MEM* pMem;

	if (!memblock)

		return 0;

	pMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(memblock);

	if (pMem->sig != WINPR_ALIGNED_MEM_SIGNATURE)

	{

		WLog_ERR(TAG, "_aligned_msize: memory block was not allocated by _aligned_malloc!");

		return 0;

	}

	return pMem->size;

}

void _aligned_free(void* memblock)

{

	WINPR_ALIGNED_MEM* pMem;

	if (!memblock)

		return;

	pMem = WINPR_ALIGNED_MEM_STRUCT_FROM_PTR(memblock);

	if (pMem->sig != WINPR_ALIGNED_MEM_SIGNATURE)

	{

		WLog_ERR(TAG, "_aligned_free: memory block was not allocated by _aligned_malloc!");

		return;

	}

	free(pMem->base_addr);

}

#endif