osb/source/core/StarMemory.cpp
Kai Blaschke 9029f897da
Support prefixed and non-prefixed JeMalloc functions
Note that linking a JeMalloc library without prefixed functions will replace all memory allocations, including any call to "new", not just the ones specifically called via Star::malloc etc.
2024-02-21 18:08:56 +01:00

113 lines
2.1 KiB
C++

#include "StarMemory.hpp"
#ifdef STAR_USE_JEMALLOC
#include "jemalloc/jemalloc.h"
#endif
namespace Star {
#ifdef STAR_USE_JEMALLOC
#ifdef STAR_JEMALLOC_IS_PREFIXED
void* malloc(size_t size) {
return je_malloc(size);
}
void* realloc(void* ptr, size_t size) {
return je_realloc(ptr, size);
}
void free(void* ptr) {
je_free(ptr);
}
void free(void* ptr, size_t size) {
if (ptr)
je_sdallocx(ptr, size, 0);
}
#else
void* malloc(size_t size) {
return ::malloc(size);
}
void* realloc(void* ptr, size_t size) {
return ::realloc(ptr, size);
}
void free(void* ptr) {
::free(ptr);
}
void free(void* ptr, size_t size) {
if (ptr)
::sdallocx(ptr, size, 0);
}
#endif
#else
void* malloc(size_t size) {
return ::malloc(size);
}
void* realloc(void* ptr, size_t size) {
return ::realloc(ptr, size);
}
void free(void* ptr) {
return ::free(ptr);
}
void free(void* ptr, size_t) {
return ::free(ptr);
}
#endif
}
void* operator new(std::size_t size) {
auto ptr = Star::malloc(size);
if (!ptr)
throw std::bad_alloc();
return ptr;
}
void* operator new[](std::size_t size) {
auto ptr = Star::malloc(size);
if (!ptr)
throw std::bad_alloc();
return ptr;
}
// Globally override new and delete. As the per standard, new and delete must
// be defined in global scope, and must not be inline.
void* operator new(std::size_t size, std::nothrow_t const&) noexcept {
return Star::malloc(size);
}
void* operator new[](std::size_t size, std::nothrow_t const&) noexcept {
return Star::malloc(size);
}
void operator delete(void* ptr) noexcept {
Star::free(ptr);
}
void operator delete[](void* ptr) noexcept {
Star::free(ptr);
}
void operator delete(void* ptr, std::nothrow_t const&) noexcept {
Star::free(ptr);
}
void operator delete[](void* ptr, std::nothrow_t const&) noexcept {
Star::free(ptr);
}
void operator delete(void* ptr, std::size_t size) noexcept {
Star::free(ptr, size);
}
void operator delete[](void* ptr, std::size_t size) noexcept {
Star::free(ptr, size);
}