|
|
-
- #include "config.h"
-
- #include "almalloc.h"
-
- #include <cstdlib>
- #include <cstring>
- #ifdef HAVE_MALLOC_H
- #include <malloc.h>
- #endif
- #ifdef HAVE_WINDOWS_H
- #include <windows.h>
- #else
- #include <unistd.h>
- #endif
-
-
- #ifdef __GNUC__
- #define LIKELY(x) __builtin_expect(!!(x), !0)
- #define UNLIKELY(x) __builtin_expect(!!(x), 0)
- #else
- #define LIKELY(x) (!!(x))
- #define UNLIKELY(x) (!!(x))
- #endif
-
-
- void *al_malloc(size_t alignment, size_t size)
- {
- #if defined(HAVE_ALIGNED_ALLOC)
- size = (size+(alignment-1))&~(alignment-1);
- return aligned_alloc(alignment, size);
- #elif defined(HAVE_POSIX_MEMALIGN)
- void *ret;
- if(posix_memalign(&ret, alignment, size) == 0)
- return ret;
- return nullptr;
- #elif defined(HAVE__ALIGNED_MALLOC)
- return _aligned_malloc(size, alignment);
- #else
- char *ret = static_cast<char*>(malloc(size+alignment));
- if(ret != nullptr)
- {
- *(ret++) = 0x00;
- while(((ptrdiff_t)ret&(alignment-1)) != 0)
- *(ret++) = 0x55;
- }
- return ret;
- #endif
- }
-
- void *al_calloc(size_t alignment, size_t size)
- {
- void *ret = al_malloc(alignment, size);
- if(ret) memset(ret, 0, size);
- return ret;
- }
-
- void al_free(void *ptr) noexcept
- {
- #if defined(HAVE_ALIGNED_ALLOC) || defined(HAVE_POSIX_MEMALIGN)
- free(ptr);
- #elif defined(HAVE__ALIGNED_MALLOC)
- _aligned_free(ptr);
- #else
- if(ptr != nullptr)
- {
- char *finder = static_cast<char*>(ptr);
- do {
- --finder;
- } while(*finder == 0x55);
- free(finder);
- }
- #endif
- }
-
- size_t al_get_page_size() noexcept
- {
- static size_t psize = 0;
- if(UNLIKELY(!psize))
- {
- #ifdef HAVE_SYSCONF
- #if defined(_SC_PAGESIZE)
- if(!psize) psize = sysconf(_SC_PAGESIZE);
- #elif defined(_SC_PAGE_SIZE)
- if(!psize) psize = sysconf(_SC_PAGE_SIZE);
- #endif
- #endif /* HAVE_SYSCONF */
- #ifdef _WIN32
- if(!psize)
- {
- SYSTEM_INFO sysinfo{};
- GetSystemInfo(&sysinfo);
- psize = sysinfo.dwPageSize;
- }
- #endif
- if(!psize) psize = DEF_ALIGN;
- }
- return psize;
- }
-
- int al_is_sane_alignment_allocator() noexcept
- {
- #if defined(HAVE_ALIGNED_ALLOC) || defined(HAVE_POSIX_MEMALIGN) || defined(HAVE__ALIGNED_MALLOC)
- return 1;
- #else
- return 0;
- #endif
- }
|