| |
| |
| |
| |
| |
| |
|
|
| #include "mimalloc.h" |
| #include "mimalloc/internal.h" |
| #include "mimalloc/prim.h" |
|
|
| #include <string.h> |
|
|
| |
| |
| |
|
|
| static bool mi_malloc_is_naturally_aligned( size_t size, size_t alignment ) { |
| |
| |
| mi_assert_internal(_mi_is_power_of_two(alignment) && (alignment > 0)); |
| if (alignment > size) return false; |
| const size_t bsize = mi_good_size(size); |
| const bool ok = (bsize <= MI_PAGE_MAX_START_BLOCK_ALIGN2 && _mi_is_power_of_two(bsize)) || |
| (alignment==MI_PAGE_OSPAGE_BLOCK_ALIGN2 && (bsize % MI_PAGE_OSPAGE_BLOCK_ALIGN2)==0); |
| if (ok) { mi_assert_internal((bsize & (alignment-1)) == 0); } |
| return ok; |
| } |
|
|
| #if MI_GUARDED |
| static mi_decl_restrict void* mi_theap_malloc_guarded_aligned(mi_theap_t* theap, size_t size, size_t alignment, bool zero) mi_attr_noexcept { |
| |
| #if MI_THEAP_INITASNULL |
| if mi_unlikely(theap==NULL) { theap = _mi_theap_empty_get(); } |
| #endif |
| mi_assert_internal(alignment > 0 && alignment < MI_PAGE_MAX_OVERALLOC_ALIGN); |
| if mi_unlikely(alignment >= MI_PAGE_MAX_OVERALLOC_ALIGN || size > (MI_MAX_ALLOC_SIZE - MI_PADDING_SIZE - alignment)) { |
| _mi_error_message(EOVERFLOW, "(guarded) aligned allocation request is too large (size %zu, alignment %zu)\n", size, alignment); |
| return NULL; |
| } |
| const size_t oversize = size + alignment - 1; |
| void* base = _mi_theap_malloc_guarded(theap, oversize, zero); |
| void* p = _mi_align_up_ptr(base, alignment); |
| mi_track_align(base, p, (uint8_t*)p - (uint8_t*)base, size); |
| mi_assert_internal(mi_usable_size(p) >= size); |
| mi_assert_internal(_mi_is_aligned(p, alignment)); |
| return p; |
| } |
|
|
| static void* mi_theap_malloc_zero_no_guarded(mi_theap_t* theap, size_t size, bool zero, size_t* usable) { |
| #if MI_THEAP_INITASNULL |
| if mi_unlikely(theap==NULL) { theap = _mi_theap_empty_get(); } |
| #endif |
| const size_t rate = theap->guarded_sample_rate; |
| |
| if (rate != 0) { theap->guarded_sample_rate = 0; } |
| void* p = _mi_theap_malloc_zero(theap, size, zero, usable); |
| if (rate != 0) { theap->guarded_sample_rate = rate; } |
| return p; |
| } |
| #else |
| static void* mi_theap_malloc_zero_no_guarded(mi_theap_t* theap, size_t size, bool zero, size_t* usable) { |
| return _mi_theap_malloc_zero(theap, size, zero, usable); |
| } |
| #endif |
|
|
| |
| static mi_decl_noinline void* mi_theap_malloc_zero_aligned_at_overalloc(mi_theap_t* const theap, const size_t size, const size_t alignment, const size_t offset, const bool zero, size_t* usable) mi_attr_noexcept |
| { |
| mi_assert_internal(size <= (MI_MAX_ALLOC_SIZE - MI_PADDING_SIZE)); |
| mi_assert_internal(alignment != 0 && _mi_is_power_of_two(alignment)); |
|
|
| void* p; |
| size_t oversize; |
| if mi_unlikely(alignment > MI_PAGE_MAX_OVERALLOC_ALIGN) { |
| |
| |
| |
| if mi_unlikely(offset != 0) { |
| |
| _mi_error_message(EOVERFLOW, "aligned allocation with a large alignment cannot be used with an alignment offset (size %zu, alignment %zu, offset %zu)\n", size, alignment, offset); |
| return NULL; |
| } |
| oversize = (size <= MI_SMALL_SIZE_MAX ? MI_SMALL_SIZE_MAX + 1 : size); |
| |
| p = _mi_theap_malloc_zero_ex(theap, oversize, zero, alignment, usable); |
| if (p == NULL) return NULL; |
| } |
| else { |
| |
| mi_assert_internal(size <= (MI_MAX_ALLOC_SIZE - MI_PADDING_SIZE) && alignment <= MI_PAGE_MAX_OVERALLOC_ALIGN); |
| mi_assert_internal(size < SIZE_MAX - alignment); |
| oversize = (size < MI_MAX_ALIGN_SIZE ? MI_MAX_ALIGN_SIZE : size) + alignment - 1; |
| p = mi_theap_malloc_zero_no_guarded(theap, oversize, zero, usable); |
| if (p == NULL) return NULL; |
| } |
|
|
| |
| const uintptr_t align_mask = alignment - 1; |
| const uintptr_t poffset = ((uintptr_t)p + offset) & align_mask; |
| const uintptr_t adjust = (poffset == 0 ? 0 : alignment - poffset); |
| mi_assert_internal(adjust < alignment); |
| void* aligned_p = (void*)((uintptr_t)p + adjust); |
|
|
| |
| |
| |
| |
| mi_page_t* page = _mi_ptr_page(p); |
| if (aligned_p != p) { |
| mi_page_set_has_interior_pointers(page, true); |
| #if MI_GUARDED |
| |
| if (adjust >= sizeof(mi_block_t)) { |
| mi_block_t* const block = (mi_block_t*)p; |
| block->next = MI_BLOCK_TAG_ALIGNED; |
| } |
| #endif |
| _mi_padding_shrink(page, (mi_block_t*)p, adjust + size); |
| } |
| |
|
|
| mi_assert_internal(mi_page_usable_block_size(page) >= adjust + size); |
| mi_assert_internal(((uintptr_t)aligned_p + offset) % alignment == 0); |
| mi_assert_internal(mi_usable_size(aligned_p)>=size); |
| mi_assert_internal(mi_usable_size(p) == mi_usable_size(aligned_p)+adjust); |
| #if MI_DEBUG > 1 |
| mi_page_t* const apage = _mi_ptr_page(aligned_p); |
| void* unalign_p = _mi_page_ptr_unalign(apage, aligned_p); |
| mi_assert_internal(p == unalign_p); |
| #endif |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
|
|
| if (p != aligned_p) { |
| mi_track_align(p,aligned_p,adjust,mi_usable_size(aligned_p)); |
| #if MI_GUARDED |
| mi_track_mem_defined(p, sizeof(mi_block_t)); |
| #endif |
| } |
| return aligned_p; |
| } |
|
|
| |
| static mi_decl_noinline void* mi_theap_malloc_zero_aligned_at_generic(mi_theap_t* const theap, const size_t size, const size_t alignment, const size_t offset, const bool zero, size_t* usable) mi_attr_noexcept |
| { |
| mi_assert_internal(alignment != 0 && _mi_is_power_of_two(alignment)); |
| |
| if mi_unlikely(size > (MI_MAX_ALLOC_SIZE - MI_PADDING_SIZE)) { |
| _mi_error_message(EOVERFLOW, "aligned allocation request is too large (size %zu, alignment %zu)\n", size, alignment); |
| return NULL; |
| } |
|
|
| |
| |
| |
| if (offset == 0 && mi_malloc_is_naturally_aligned(size,alignment)) { |
| void* p = mi_theap_malloc_zero_no_guarded(theap, size, zero, usable); |
| mi_assert_internal(p == NULL || ((uintptr_t)p % alignment) == 0); |
| const bool is_aligned_or_null = (((uintptr_t)p) & (alignment-1))==0; |
| if mi_likely(is_aligned_or_null) { |
| return p; |
| } |
| else { |
| |
| mi_assert(false); |
| mi_free(p); |
| } |
| } |
|
|
| |
| return mi_theap_malloc_zero_aligned_at_overalloc(theap,size,alignment,offset,zero,usable); |
| } |
|
|
|
|
| |
| static inline void* mi_theap_malloc_zero_aligned_at(mi_theap_t* const theap, const size_t size, const size_t alignment, const size_t offset, const bool zero, size_t* usable) mi_attr_noexcept |
| { |
| |
| if mi_unlikely(alignment == 0 || !_mi_is_power_of_two(alignment)) { |
| #if MI_DEBUG > 0 |
| _mi_error_message(EOVERFLOW, "aligned allocation requires the alignment to be a power-of-two (size %zu, alignment %zu)\n", size, alignment); |
| #endif |
| return NULL; |
| } |
|
|
| #if MI_GUARDED |
| #if MI_THEAP_INITASNULL |
| if mi_likely(theap!=NULL) |
| #endif |
| if (offset==0 && alignment < MI_PAGE_MAX_OVERALLOC_ALIGN && mi_theap_malloc_use_guarded(theap,size)) { |
| return mi_theap_malloc_guarded_aligned(theap, size, alignment, zero); |
| } |
| #endif |
|
|
| |
| |
| |
| #if MI_THEAP_INITASNULL |
| if mi_likely(theap!=NULL) |
| #endif |
| { |
| if mi_likely(size <= MI_SMALL_SIZE_MAX && alignment <= size) { |
| const uintptr_t align_mask = alignment-1; |
| const size_t padsize = size + MI_PADDING_SIZE; |
| mi_page_t* page = _mi_theap_get_free_small_page(theap, padsize); |
| if mi_likely(page->free != NULL) { |
| const bool is_aligned = (((uintptr_t)page->free + offset) & align_mask)==0; |
| if mi_likely(is_aligned) |
| { |
| if (usable!=NULL) { *usable = mi_page_usable_block_size(page); } |
| void* p = _mi_page_malloc_zero(theap, page, padsize, zero); |
| mi_assert_internal(p != NULL); |
| mi_assert_internal(((uintptr_t)p + offset) % alignment == 0); |
| mi_track_malloc(p, size, zero); |
| return p; |
| } |
| } |
| } |
| } |
|
|
| |
| return mi_theap_malloc_zero_aligned_at_generic(theap, size, alignment, offset, zero, usable); |
| } |
|
|
|
|
| |
| |
| |
|
|
| static mi_decl_restrict void* mi_theap_malloc_aligned_at(mi_theap_t* theap, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_malloc_zero_aligned_at(theap, size, alignment, offset, false, NULL); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_theap_malloc_aligned(mi_theap_t* theap, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_malloc_aligned_at(theap, size, alignment, 0); |
| } |
|
|
| static mi_decl_restrict void* mi_theap_zalloc_aligned_at(mi_theap_t* theap, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_malloc_zero_aligned_at(theap, size, alignment, offset, true, NULL); |
| } |
|
|
| static mi_decl_restrict void* mi_theap_zalloc_aligned(mi_theap_t* theap, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_zalloc_aligned_at(theap, size, alignment, 0); |
| } |
|
|
| static mi_decl_restrict void* mi_theap_calloc_aligned_at(mi_theap_t* theap, size_t count, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| size_t total; |
| if (mi_count_size_overflow(count, size, &total)) return NULL; |
| return mi_theap_zalloc_aligned_at(theap, total, alignment, offset); |
| } |
|
|
| static mi_decl_restrict void* mi_theap_calloc_aligned(mi_theap_t* theap, size_t count, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_calloc_aligned_at(theap, count, size, alignment, 0); |
| } |
|
|
|
|
| |
| |
| |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_malloc_aligned_at(size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_malloc_aligned_at(_mi_theap_default(), size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_malloc_aligned(size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_malloc_aligned(_mi_theap_default(), size, alignment); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_umalloc_aligned(size_t size, size_t alignment, size_t* block_size) mi_attr_noexcept { |
| return mi_theap_malloc_zero_aligned_at(_mi_theap_default(), size, alignment, 0, false, block_size); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_zalloc_aligned_at(size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_zalloc_aligned_at(_mi_theap_default(), size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_zalloc_aligned(size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_zalloc_aligned(_mi_theap_default(), size, alignment); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_uzalloc_aligned(size_t size, size_t alignment, size_t* block_size) mi_attr_noexcept { |
| return mi_theap_malloc_zero_aligned_at(_mi_theap_default(), size, alignment, 0, true, block_size); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_calloc_aligned_at(size_t count, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_calloc_aligned_at(_mi_theap_default(), count, size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_calloc_aligned(size_t count, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_calloc_aligned(_mi_theap_default(), count, size, alignment); |
| } |
|
|
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_heap_malloc_aligned_at(mi_heap_t* heap, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_malloc_aligned_at(_mi_heap_theap(heap), size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_heap_malloc_aligned(mi_heap_t* heap, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_malloc_aligned(_mi_heap_theap(heap), size, alignment); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_heap_zalloc_aligned_at(mi_heap_t* heap, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_zalloc_aligned_at(_mi_heap_theap(heap), size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_heap_zalloc_aligned(mi_heap_t* heap, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_zalloc_aligned(_mi_heap_theap(heap), size, alignment); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_heap_calloc_aligned_at(mi_heap_t* heap, size_t count, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_calloc_aligned_at(_mi_heap_theap(heap), count, size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard mi_decl_restrict void* mi_heap_calloc_aligned(mi_heap_t* heap, size_t count, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_calloc_aligned(_mi_heap_theap(heap), count, size, alignment); |
| } |
|
|
|
|
| |
| |
| |
|
|
| static void* mi_theap_realloc_zero_aligned_at(mi_theap_t* theap, void* p, size_t newsize, size_t alignment, size_t offset, bool zero) mi_attr_noexcept { |
| mi_assert(alignment > 0); |
| if (alignment <= sizeof(uintptr_t) && offset==0) return _mi_theap_realloc_zero(theap,p,newsize,zero,NULL,NULL); |
| if (p == NULL) return mi_theap_malloc_zero_aligned_at(theap,newsize,alignment,offset,zero,NULL); |
| size_t size = mi_usable_size(p); |
| if (newsize <= size && newsize >= (size - (size / 2)) |
| && (((uintptr_t)p + offset) % alignment) == 0) { |
| return p; |
| } |
| else { |
| |
| void* newp = mi_theap_malloc_aligned_at(theap,newsize,alignment,offset); |
| if (newp != NULL) { |
| if (zero && newsize > size) { |
| |
| size_t start = (size >= sizeof(intptr_t) ? size - sizeof(intptr_t) : 0); |
| _mi_memzero((uint8_t*)newp + start, newsize - start); |
| } |
| _mi_memcpy_aligned(newp, p, (newsize > size ? size : newsize)); |
| mi_free(p); |
| } |
| return newp; |
| } |
| } |
|
|
| static void* mi_theap_realloc_zero_aligned(mi_theap_t* theap, void* p, size_t newsize, size_t alignment, bool zero) mi_attr_noexcept { |
| mi_assert(alignment > 0); |
| if (alignment <= sizeof(uintptr_t)) return _mi_theap_realloc_zero(theap,p,newsize,zero,NULL,NULL); |
| return mi_theap_realloc_zero_aligned_at(theap,p,newsize,alignment,0,zero); |
| } |
|
|
| static void* mi_theap_realloc_aligned_at(mi_theap_t* theap, void* p, size_t newsize, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_realloc_zero_aligned_at(theap,p,newsize,alignment,offset,false); |
| } |
|
|
| static void* mi_theap_realloc_aligned(mi_theap_t* theap, void* p, size_t newsize, size_t alignment) mi_attr_noexcept { |
| return mi_theap_realloc_zero_aligned(theap,p,newsize,alignment,false); |
| } |
|
|
| static void* mi_theap_rezalloc_aligned_at(mi_theap_t* theap, void* p, size_t newsize, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_realloc_zero_aligned_at(theap, p, newsize, alignment, offset, true); |
| } |
|
|
| static void* mi_theap_rezalloc_aligned(mi_theap_t* theap, void* p, size_t newsize, size_t alignment) mi_attr_noexcept { |
| return mi_theap_realloc_zero_aligned(theap, p, newsize, alignment, true); |
| } |
|
|
| static void* mi_theap_recalloc_aligned_at(mi_theap_t* theap, void* p, size_t newcount, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| size_t total; |
| if (mi_count_size_overflow(newcount, size, &total)) return NULL; |
| return mi_theap_rezalloc_aligned_at(theap, p, total, alignment, offset); |
| } |
|
|
| static void* mi_theap_recalloc_aligned(mi_theap_t* theap, void* p, size_t newcount, size_t size, size_t alignment) mi_attr_noexcept { |
| size_t total; |
| if (mi_count_size_overflow(newcount, size, &total)) return NULL; |
| return mi_theap_rezalloc_aligned(theap, p, total, alignment); |
| } |
|
|
|
|
| mi_decl_nodiscard void* mi_realloc_aligned_at(void* p, size_t newsize, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_realloc_aligned_at(_mi_theap_default(), p, newsize, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard void* mi_realloc_aligned(void* p, size_t newsize, size_t alignment) mi_attr_noexcept { |
| return mi_theap_realloc_aligned(_mi_theap_default(), p, newsize, alignment); |
| } |
|
|
| mi_decl_nodiscard void* mi_rezalloc_aligned_at(void* p, size_t newsize, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_rezalloc_aligned_at(_mi_theap_default(), p, newsize, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard void* mi_rezalloc_aligned(void* p, size_t newsize, size_t alignment) mi_attr_noexcept { |
| return mi_theap_rezalloc_aligned(_mi_theap_default(), p, newsize, alignment); |
| } |
|
|
| mi_decl_nodiscard void* mi_recalloc_aligned_at(void* p, size_t newcount, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_recalloc_aligned_at(_mi_theap_default(), p, newcount, size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard void* mi_recalloc_aligned(void* p, size_t newcount, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_recalloc_aligned(_mi_theap_default(), p, newcount, size, alignment); |
| } |
|
|
|
|
| mi_decl_nodiscard void* mi_heap_realloc_aligned_at(mi_heap_t* heap, void* p, size_t newsize, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_realloc_aligned_at(_mi_heap_theap(heap), p, newsize, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard void* mi_heap_realloc_aligned(mi_heap_t* heap, void* p, size_t newsize, size_t alignment) mi_attr_noexcept { |
| return mi_theap_realloc_aligned(_mi_heap_theap(heap), p, newsize, alignment); |
| } |
|
|
| mi_decl_nodiscard void* mi_heap_rezalloc_aligned_at(mi_heap_t* heap, void* p, size_t newsize, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_rezalloc_aligned_at(_mi_heap_theap(heap), p, newsize, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard void* mi_heap_rezalloc_aligned(mi_heap_t* heap, void* p, size_t newsize, size_t alignment) mi_attr_noexcept { |
| return mi_theap_rezalloc_aligned(_mi_heap_theap(heap), p, newsize, alignment); |
| } |
|
|
| mi_decl_nodiscard void* mi_heap_recalloc_aligned_at(mi_heap_t* heap, void* p, size_t newcount, size_t size, size_t alignment, size_t offset) mi_attr_noexcept { |
| return mi_theap_recalloc_aligned_at(_mi_heap_theap(heap), p, newcount, size, alignment, offset); |
| } |
|
|
| mi_decl_nodiscard void* mi_heap_recalloc_aligned(mi_heap_t* heap, void* p, size_t newcount, size_t size, size_t alignment) mi_attr_noexcept { |
| return mi_theap_recalloc_aligned(_mi_heap_theap(heap), p, newcount, size, alignment); |
| } |
|
|
|
|
|
|