util: Add an aligned realloc function

Mesa has one of these in imports.h, so u_memory needs one as well. This
is the version from mesa ported.

Reviewed-by: Marek Olšák <marek.olsak@amd.com>
Reviewed-by: Kristian H. Kristensen <hoegsberg@google.com>
Reviewed-by: Matt Turner <mattst88@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/3024>
This commit is contained in:
Dylan Baker 2018-09-12 16:26:38 -07:00
parent b85775900d
commit bb560f2d65
4 changed files with 40 additions and 17 deletions

View file

@ -60,6 +60,9 @@ os_malloc_aligned(size_t size, size_t alignment);
void
os_free_aligned(void *ptr);
void *
os_realloc_aligned(void *ptr, size_t oldsize, size_t newsize, size_t alignemnt);
#ifdef __cplusplus
}
#endif

View file

@ -36,7 +36,6 @@
#endif
/**
* Add two size_t values with integer overflow check.
* TODO: leverage __builtin_add_overflow where available
@ -49,6 +48,22 @@ add_overflow_size_t(size_t a, size_t b, size_t *res)
}
#if defined(HAVE_POSIX_MEMALIGN)
static inline void *
os_malloc_aligned(size_t size, size_t alignment)
{
void *ptr;
alignment = (alignment + sizeof(void*) - 1) & ~(sizeof(void*) - 1);
if(posix_memalign(&ptr, alignment, size) != 0)
return NULL;
return ptr;
}
#define os_free_aligned(_ptr) free(_ptr)
#else
/**
* Return memory on given byte alignment
*/
@ -93,3 +108,21 @@ os_free_aligned(void *ptr)
os_free(realAddr);
}
}
#endif
/**
* Reallocate memeory, with alignment
*/
static inline void *
os_realloc_aligned(void *ptr, size_t oldsize, size_t newsize, size_t alignment)
{
const size_t copySize = MIN2(oldsize, newsize);
void *newBuf = os_malloc_aligned(newsize, alignment);
if (newBuf && ptr && copySize > 0) {
memcpy(newBuf, ptr, copySize);
}
os_free_aligned(ptr);
return newBuf;
}

View file

@ -45,27 +45,13 @@
#define os_realloc( _old_ptr, _old_size, _new_size) \
realloc(_old_ptr, _new_size + 0*(_old_size))
#if defined(HAVE_POSIX_MEMALIGN)
static inline void *
os_malloc_aligned(size_t size, size_t alignment)
{
void *ptr;
alignment = (alignment + sizeof(void*) - 1) & ~(sizeof(void*) - 1);
if(posix_memalign(&ptr, alignment, size) != 0)
return NULL;
return ptr;
}
#define os_free_aligned(_ptr) free(_ptr)
#elif DETECT_OS_WINDOWS
#if DETECT_OS_WINDOWS
#include <malloc.h>
#define os_malloc_aligned(_size, _align) _aligned_malloc(_size, _align)
#define os_free_aligned(_ptr) _aligned_free(_ptr)
#define os_realloc_aligned(_ptr, _oldsize, _newsize, _alignment) _aligned_realloc(_ptr, _newsize, _alignment)
#else

View file

@ -60,6 +60,7 @@ extern "C" {
#define align_malloc(_size, _alignment) os_malloc_aligned(_size, _alignment)
#define align_free(_ptr) os_free_aligned(_ptr)
#define align_realloc(_ptr, _oldsize, _newsize, _alignment) os_realloc_aligned(_ptr, _oldsize, _newsize, _alignment)
static inline void *
align_calloc(size_t size, unsigned long alignment)