2010-02-03 12:11:58 +00:00
|
|
|
/**************************************************************************
|
2022-03-18 05:14:55 +00:00
|
|
|
*
|
2010-02-03 12:11:58 +00:00
|
|
|
* Copyright 2008-2010 VMware, Inc.
|
|
|
|
* All Rights Reserved.
|
2022-03-18 05:14:55 +00:00
|
|
|
*
|
2010-02-03 12:11:58 +00:00
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
|
|
* copy of this software and associated documentation files (the
|
|
|
|
* "Software"), to deal in the Software without restriction, including
|
|
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
|
|
* distribute, sub license, and/or sell copies of the Software, and to
|
|
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
|
|
* the following conditions:
|
2022-03-18 05:14:55 +00:00
|
|
|
*
|
2010-02-03 12:11:58 +00:00
|
|
|
* The above copyright notice and this permission notice (including the
|
|
|
|
* next paragraph) shall be included in all copies or substantial portions
|
|
|
|
* of the Software.
|
2022-03-18 05:14:55 +00:00
|
|
|
*
|
2010-02-03 12:11:58 +00:00
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
|
|
|
|
* IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
|
|
|
|
* ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
|
|
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
|
|
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
2022-03-18 05:14:55 +00:00
|
|
|
*
|
2010-02-03 12:11:58 +00:00
|
|
|
**************************************************************************/
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Memory alignment wrappers.
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
#ifndef _OS_MEMORY_H_
|
|
|
|
#error "Must not be included directly. Include os_memory.h instead"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2015-04-22 20:23:59 +01:00
|
|
|
/**
|
|
|
|
* Add two size_t values with integer overflow check.
|
|
|
|
* TODO: leverage __builtin_add_overflow where available
|
|
|
|
*/
|
|
|
|
static inline bool
|
|
|
|
add_overflow_size_t(size_t a, size_t b, size_t *res)
|
|
|
|
{
|
|
|
|
*res = a + b;
|
|
|
|
return *res < a || *res < b;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2018-09-13 00:26:38 +01:00
|
|
|
#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
|
|
|
|
|
2010-02-03 12:11:58 +00:00
|
|
|
/**
|
|
|
|
* Return memory on given byte alignment
|
|
|
|
*/
|
2015-07-21 00:58:43 +01:00
|
|
|
static inline void *
|
2010-02-04 18:06:55 +00:00
|
|
|
os_malloc_aligned(size_t size, size_t alignment)
|
2010-02-03 12:11:58 +00:00
|
|
|
{
|
|
|
|
char *ptr, *buf;
|
2015-04-22 20:23:59 +01:00
|
|
|
size_t alloc_size;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Calculate
|
|
|
|
*
|
|
|
|
* alloc_size = size + alignment + sizeof(void *)
|
|
|
|
*
|
|
|
|
* while checking for overflow.
|
|
|
|
*/
|
|
|
|
if (add_overflow_size_t(size, alignment, &alloc_size) ||
|
|
|
|
add_overflow_size_t(alloc_size, sizeof(void *), &alloc_size)) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-02-03 12:11:58 +00:00
|
|
|
|
2015-04-22 20:23:59 +01:00
|
|
|
ptr = (char *) os_malloc(alloc_size);
|
2010-02-03 12:11:58 +00:00
|
|
|
if (!ptr)
|
|
|
|
return NULL;
|
|
|
|
|
2010-02-04 18:06:55 +00:00
|
|
|
buf = (char *)(((uintptr_t)ptr + sizeof(void *) + alignment - 1) & ~((uintptr_t)(alignment - 1)));
|
2010-02-03 12:11:58 +00:00
|
|
|
*(char **)(buf - sizeof(void *)) = ptr;
|
|
|
|
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
2017-10-27 01:30:51 +01:00
|
|
|
* Free memory returned by os_malloc_aligned().
|
2010-02-03 12:11:58 +00:00
|
|
|
*/
|
2015-07-21 00:58:43 +01:00
|
|
|
static inline void
|
2010-02-03 12:11:58 +00:00
|
|
|
os_free_aligned(void *ptr)
|
|
|
|
{
|
|
|
|
if (ptr) {
|
|
|
|
void **cubbyHole = (void **) ((char *) ptr - sizeof(void *));
|
|
|
|
void *realAddr = *cubbyHole;
|
|
|
|
os_free(realAddr);
|
|
|
|
}
|
|
|
|
}
|
2018-09-13 00:26:38 +01:00
|
|
|
|
|
|
|
#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;
|
|
|
|
}
|