Buckets:
arudradey/ml-cpu-storage / emsdk /upstream /emscripten /cache /sysroot /include /mimalloc /internal.h
| /* ---------------------------------------------------------------------------- | |
| Copyright (c) 2018-2025, Microsoft Research, Daan Leijen | |
| This is free software; you can redistribute it and/or modify it under the | |
| terms of the MIT license. A copy of the license can be found in the file | |
| "LICENSE" at the root of this distribution. | |
| -----------------------------------------------------------------------------*/ | |
| // -------------------------------------------------------------------------- | |
| // This file contains the internal API's of mimalloc and various utility | |
| // functions and macros. | |
| // -------------------------------------------------------------------------- | |
| // -------------------------------------------------------------------------- | |
| // Compiler defines | |
| // -------------------------------------------------------------------------- | |
| // -------------------------------------------------------------------------- | |
| // Internal functions | |
| // -------------------------------------------------------------------------- | |
| // "libc.c" | |
| int _mi_vsnprintf(char* buf, size_t bufsize, const char* fmt, va_list args); | |
| int _mi_snprintf(char* buf, size_t buflen, const char* fmt, ...); | |
| char _mi_toupper(char c); | |
| int _mi_strnicmp(const char* s, const char* t, size_t n); | |
| void _mi_strlcpy(char* dest, const char* src, size_t dest_size); | |
| void _mi_strlcat(char* dest, const char* src, size_t dest_size); | |
| size_t _mi_strlen(const char* s); | |
| size_t _mi_strnlen(const char* s, size_t max_len); | |
| char* _mi_strnstr(char* s, size_t max_len, const char* pat); | |
| bool _mi_streq(const char* s, const char* t); | |
| bool _mi_getenv(const char* name, char* result, size_t result_size); | |
| // "options.c" | |
| void _mi_fputs(mi_output_fun* out, void* arg, const char* prefix, const char* message); | |
| void _mi_fprintf(mi_output_fun* out, void* arg, const char* fmt, ...); | |
| void _mi_raw_message(const char* fmt, ...); | |
| void _mi_message(const char* fmt, ...); | |
| void _mi_warning_message(const char* fmt, ...); | |
| void _mi_verbose_message(const char* fmt, ...); | |
| void _mi_trace_message(const char* fmt, ...); | |
| void _mi_options_init(void); | |
| void _mi_options_post_init(void); | |
| long _mi_option_get_fast(mi_option_t option); | |
| void _mi_error_message(int err, const char* fmt, ...); | |
| // random.c | |
| void _mi_random_init(mi_random_ctx_t* ctx); | |
| void _mi_random_init_weak(mi_random_ctx_t* ctx); | |
| void _mi_random_reinit_if_weak(mi_random_ctx_t * ctx); | |
| void _mi_random_split(mi_random_ctx_t* ctx, mi_random_ctx_t* new_ctx); | |
| uintptr_t _mi_random_next(mi_random_ctx_t* ctx); | |
| uintptr_t _mi_theap_random_next(mi_theap_t* theap); | |
| uintptr_t _mi_os_random_weak(uintptr_t extra_seed); | |
| static inline uintptr_t _mi_random_shuffle(uintptr_t x); | |
| // init.c | |
| extern mi_decl_hidden mi_decl_cache_align const mi_page_t _mi_page_empty; | |
| void _mi_auto_process_init(void); | |
| void mi_cdecl _mi_auto_process_done(void) mi_attr_noexcept; | |
| bool _mi_is_redirected(void); | |
| bool _mi_allocator_init(const char** message); | |
| void _mi_allocator_done(void); | |
| bool _mi_is_main_thread(void); | |
| bool _mi_preloading(void); // true while the C runtime is not initialized yet | |
| void _mi_thread_done(mi_theap_t* theap); | |
| mi_subproc_t* _mi_subproc(void); | |
| mi_subproc_t* _mi_subproc_main(void); | |
| mi_heap_t* _mi_subproc_heap_main(mi_subproc_t* subproc); | |
| mi_subproc_t* _mi_subproc_from_id(mi_subproc_id_t subproc_id); | |
| mi_threadid_t _mi_thread_id(void) mi_attr_noexcept; | |
| size_t _mi_thread_seq_id(void) mi_attr_noexcept; | |
| bool _mi_is_heap_main(const mi_heap_t* heap); | |
| bool _mi_is_theap_main(const mi_theap_t* theap); | |
| void _mi_theap_guarded_init(mi_theap_t* theap); | |
| void _mi_theap_options_init(mi_theap_t* theap); | |
| mi_theap_t* _mi_theap_default_safe(void); // ensure the returned theap is initialized | |
| mi_theap_t* _mi_theap_main_safe(void); | |
| // os.c | |
| void _mi_os_init(void); // called from process init | |
| void* _mi_os_alloc(size_t size, mi_memid_t* memid); | |
| void* _mi_os_zalloc(size_t size, mi_memid_t* memid); | |
| void _mi_os_free(void* p, size_t size, mi_memid_t memid); | |
| void _mi_os_free_ex(void* p, size_t size, bool still_committed, mi_memid_t memid, mi_subproc_t* subproc ); | |
| size_t _mi_os_page_size(void); | |
| size_t _mi_os_guard_page_size(void); | |
| size_t _mi_os_good_alloc_size(size_t size); | |
| bool _mi_os_has_overcommit(void); | |
| bool _mi_os_has_virtual_reserve(void); | |
| size_t _mi_os_virtual_address_bits(void); | |
| size_t _mi_os_minimal_purge_size(void); | |
| bool _mi_os_reset(void* addr, size_t size); | |
| bool _mi_os_decommit(void* addr, size_t size); | |
| void _mi_os_reuse(void* p, size_t size); | |
| mi_decl_nodiscard bool _mi_os_commit(void* p, size_t size, bool* is_zero); | |
| mi_decl_nodiscard bool _mi_os_commit_ex(void* addr, size_t size, bool* is_zero, size_t stat_size); | |
| mi_decl_nodiscard bool _mi_os_protect(void* addr, size_t size); | |
| bool _mi_os_unprotect(void* addr, size_t size); | |
| bool _mi_os_purge(void* p, size_t size); | |
| bool _mi_os_purge_ex(void* p, size_t size, bool allow_reset, size_t stats_size, mi_commit_fun_t* commit_fun, void* commit_fun_arg); | |
| size_t _mi_os_secure_guard_page_size(void); | |
| bool _mi_os_secure_guard_page_set_at(void* addr, mi_memid_t memid); | |
| bool _mi_os_secure_guard_page_set_before(void* addr, mi_memid_t memid); | |
| bool _mi_os_secure_guard_page_reset_at(void* addr, mi_memid_t memid); | |
| bool _mi_os_secure_guard_page_reset_before(void* addr, mi_memid_t memid); | |
| int _mi_os_numa_node(void); | |
| int _mi_os_numa_node_count(void); | |
| void* _mi_os_alloc_aligned(size_t size, size_t alignment, bool commit, bool allow_large, mi_memid_t* memid); | |
| void* _mi_os_alloc_aligned_at_offset(size_t size, size_t alignment, size_t align_offset, bool commit, bool allow_large, mi_memid_t* memid); | |
| void* _mi_os_get_aligned_hint(size_t try_alignment, size_t size); | |
| bool _mi_os_canuse_large_page(size_t size, size_t alignment); | |
| size_t _mi_os_large_page_size(void); | |
| void* _mi_os_alloc_huge_os_pages(size_t pages, int numa_node, mi_msecs_t max_secs, size_t* pages_reserved, size_t* psize, mi_memid_t* memid); | |
| // threadlocal.c | |
| mi_thread_local_t _mi_thread_local_create(void); | |
| void _mi_thread_local_free( mi_thread_local_t key ); | |
| bool _mi_thread_local_set( mi_thread_local_t key, void* val ); | |
| void* _mi_thread_local_get( mi_thread_local_t key ); | |
| void _mi_thread_locals_init(void); | |
| void _mi_thread_locals_done(void); | |
| void _mi_thread_locals_thread_done(void); | |
| // arena.c | |
| mi_arena_id_t _mi_arena_id_none(void); | |
| mi_arena_t* _mi_arena_from_id(mi_arena_id_t id); | |
| bool _mi_arena_memid_is_suitable(mi_memid_t memid, mi_arena_t* request_arena); | |
| void* _mi_arenas_alloc(mi_heap_t* heap, size_t size, bool commit, bool allow_pinned, mi_arena_t* req_arena, size_t tseq, int numa_node, mi_memid_t* memid); | |
| void* _mi_arenas_alloc_aligned(mi_heap_t* heap, size_t size, size_t alignment, size_t align_offset, bool commit, bool allow_pinned, mi_arena_t* req_arena, size_t tseq, int numa_node, mi_memid_t* memid); | |
| void _mi_arenas_free(void* p, size_t size, mi_memid_t memid); | |
| bool _mi_arenas_contain(const void* p); | |
| void _mi_arenas_collect(bool force_purge, bool visit_all, mi_tld_t* tld); | |
| void _mi_arenas_unsafe_destroy_all(mi_subproc_t* subproc); | |
| mi_page_t* _mi_arenas_page_alloc(mi_theap_t* theap, size_t block_size, size_t page_alignment); | |
| void _mi_arenas_page_free(mi_page_t* page, mi_theap_t* current_theapx /* can be NULL */); | |
| void _mi_arenas_page_abandon(mi_page_t* page, mi_theap_t* current_theap); | |
| void _mi_arenas_page_unabandon(mi_page_t* page, mi_theap_t* current_theapx /* can be NULL */); | |
| bool _mi_arenas_page_try_reabandon_to_mapped(mi_page_t* page); | |
| // arena-meta.c | |
| void* _mi_meta_zalloc( size_t size, mi_memid_t* memid ); | |
| void _mi_meta_free(void* p, size_t size, mi_memid_t memid); | |
| bool _mi_meta_is_meta_page(void* p); | |
| // "page-map.c" | |
| bool _mi_page_map_init(void); | |
| mi_decl_nodiscard bool _mi_page_map_register(mi_page_t* page); | |
| void _mi_page_map_unregister(mi_page_t* page); | |
| void _mi_page_map_unregister_range(void* start, size_t size); | |
| mi_page_t* _mi_safe_ptr_page(const void* p); | |
| void _mi_page_map_unsafe_destroy(mi_subproc_t* subproc); | |
| // "page.c" | |
| void* _mi_malloc_generic(mi_theap_t* theap, size_t size, size_t zero_huge_alignment, size_t* usable) mi_attr_noexcept mi_attr_malloc; | |
| void _mi_page_retire(mi_page_t* page) mi_attr_noexcept; // free the page if there are no other pages with many free blocks | |
| void _mi_page_unfull(mi_page_t* page); | |
| void _mi_page_free(mi_page_t* page, mi_page_queue_t* pq); // free the page | |
| void _mi_page_abandon(mi_page_t* page, mi_page_queue_t* pq); // abandon the page, to be picked up by another thread... | |
| size_t _mi_page_queue_append(mi_theap_t* theap, mi_page_queue_t* pq, mi_page_queue_t* append); | |
| void _mi_deferred_free(mi_theap_t* theap, bool force); | |
| void _mi_page_free_collect(mi_page_t* page, bool force); | |
| void _mi_page_free_collect_partly(mi_page_t* page, mi_block_t* head); | |
| mi_decl_nodiscard bool _mi_page_init(mi_theap_t* theap, mi_page_t* page); | |
| bool _mi_page_queue_is_valid(mi_theap_t* theap, const mi_page_queue_t* pq); | |
| size_t _mi_page_stats_bin(const mi_page_t* page); // for stats | |
| size_t _mi_bin_size(size_t bin); // for stats | |
| size_t _mi_bin(size_t size); // for stats | |
| // "theap.c" | |
| mi_theap_t* _mi_theap_create(mi_heap_t* heap, mi_tld_t* tld); | |
| void _mi_theap_delete(mi_theap_t* theap, bool acquire_tld_theaps_lock); | |
| void _mi_theap_default_set(mi_theap_t* theap); | |
| void _mi_theap_cached_set(mi_theap_t* theap); | |
| void _mi_theap_collect_retired(mi_theap_t* theap, bool force); | |
| void _mi_theap_collect_abandon(mi_theap_t* theap); | |
| bool _mi_theap_area_visit_blocks(const mi_heap_area_t* area, mi_page_t* page, mi_block_visit_fun* visitor, void* arg); | |
| void _mi_theap_page_reclaim(mi_theap_t* theap, mi_page_t* page); | |
| bool _mi_theap_free(mi_theap_t* theap, bool acquire_heap_theaps_lock, bool acquire_tld_theaps_lock); | |
| void _mi_theap_incref(mi_theap_t* theap); | |
| void _mi_theap_decref(mi_theap_t* theap); | |
| // "heap.c" | |
| void _mi_heap_area_init(mi_heap_area_t* area, mi_page_t* page); | |
| mi_decl_cold mi_theap_t* _mi_heap_theap_get_or_init(const mi_heap_t* heap); // get (and possible create) the theap belonging to a heap | |
| mi_decl_cold mi_theap_t* _mi_heap_theap_get_peek(const mi_heap_t* heap); // get the theap for a heap without initializing (and return NULL in that case) | |
| void _mi_heap_move_pages(mi_heap_t* heap_from, mi_heap_t* heap_to); // in "arena.c" | |
| void _mi_heap_destroy_pages(mi_heap_t* heap_from); // in "arena.c" | |
| void _mi_heap_force_destroy(mi_heap_t* heap); // allow destroying the main heap | |
| // "stats.c" | |
| void _mi_stats_init(void); | |
| void _mi_stats_merge_into(mi_stats_t* to, mi_stats_t* from); | |
| mi_msecs_t _mi_clock_now(void); | |
| mi_msecs_t _mi_clock_end(mi_msecs_t start); | |
| mi_msecs_t _mi_clock_start(void); | |
| // "alloc.c" | |
| void* _mi_page_malloc_zero(mi_theap_t* theap, mi_page_t* page, size_t size, bool zero) mi_attr_noexcept; // called from `_mi_theap_malloc_aligned` | |
| void* _mi_theap_malloc_zero(mi_theap_t* theap, size_t size, bool zero, size_t* usable) mi_attr_noexcept; | |
| void* _mi_theap_malloc_zero_ex(mi_theap_t* theap, size_t size, bool zero, size_t huge_alignment, size_t* usable) mi_attr_noexcept; // called from `_mi_theap_malloc_aligned` | |
| void* _mi_theap_realloc_zero(mi_theap_t* theap, void* p, size_t newsize, bool zero, size_t* usable_pre, size_t* usable_post) mi_attr_noexcept; | |
| mi_block_t* _mi_page_ptr_unalign(const mi_page_t* page, const void* p); | |
| void _mi_padding_shrink(const mi_page_t* page, const mi_block_t* block, const size_t min_size); | |
| bool _mi_page_is_valid(mi_page_t* page); | |
| // ------------------------------------------------------ | |
| // Assertions | |
| // ------------------------------------------------------ | |
| // use our own assertion to print without memory allocation | |
| mi_decl_noreturn mi_decl_cold void _mi_assert_fail(const char* assertion, const char* fname, unsigned int line, const char* func) mi_attr_noexcept; | |
| /* ----------------------------------------------------------- | |
| Statistics (in `stats.c`) | |
| ----------------------------------------------------------- */ | |
| // add to stat keeping track of the peak | |
| void __mi_stat_increase(mi_stat_count_t* stat, size_t amount); | |
| void __mi_stat_decrease(mi_stat_count_t* stat, size_t amount); | |
| void __mi_stat_increase_mt(mi_stat_count_t* stat, size_t amount); | |
| void __mi_stat_decrease_mt(mi_stat_count_t* stat, size_t amount); | |
| // adjust stat in special cases to compensate for double counting (and does not adjust peak values and can decrease the total) | |
| void __mi_stat_adjust_increase(mi_stat_count_t* stat, size_t amount); | |
| void __mi_stat_adjust_decrease(mi_stat_count_t* stat, size_t amount); | |
| void __mi_stat_adjust_increase_mt(mi_stat_count_t* stat, size_t amount); | |
| void __mi_stat_adjust_decrease_mt(mi_stat_count_t* stat, size_t amount); | |
| // counters can just be increased | |
| void __mi_stat_counter_increase(mi_stat_counter_t* stat, size_t amount); | |
| void __mi_stat_counter_increase_mt(mi_stat_counter_t* stat, size_t amount); | |
| /* ----------------------------------------------------------- | |
| Options (exposed for the debugger) | |
| ----------------------------------------------------------- */ | |
| typedef enum mi_option_init_e { | |
| MI_OPTION_UNINIT, // not yet initialized | |
| MI_OPTION_DEFAULTED, // not found in the environment, use default value | |
| MI_OPTION_INITIALIZED // found in environment or set explicitly | |
| } mi_option_init_t; | |
| typedef struct mi_option_desc_s { | |
| long value; // the value | |
| mi_option_init_t init; // is it initialized yet? (from the environment) | |
| mi_option_t option; // for debugging: the option index should match the option | |
| const char* name; // option name without `mimalloc_` prefix | |
| const char* legacy_name; // potential legacy option name | |
| } mi_option_desc_t; | |
| /* ----------------------------------------------------------- | |
| Inlined definitions | |
| ----------------------------------------------------------- */ | |
| // initialize a local variable to zero; use memset as compilers optimize constant sized memset's | |
| // Is `x` a power of two? (0 is considered a power of two) | |
| static inline bool _mi_is_power_of_two(uintptr_t x) { | |
| return ((x & (x - 1)) == 0); | |
| } | |
| // Is a pointer aligned? | |
| static inline bool _mi_is_aligned(const void* p, size_t alignment) { | |
| mi_assert_internal(alignment != 0); | |
| return (((uintptr_t)p % alignment) == 0); | |
| } | |
| // Align upwards | |
| static inline uintptr_t _mi_align_up(uintptr_t sz, size_t alignment) { | |
| mi_assert_internal(alignment != 0); | |
| uintptr_t mask = alignment - 1; | |
| if ((alignment & mask) == 0) { // power of two? | |
| return ((sz + mask) & ~mask); | |
| } | |
| else { | |
| return (((sz + mask)/alignment)*alignment); | |
| } | |
| } | |
| // Align a pointer upwards | |
| static inline void* _mi_align_up_ptr(const void* p, size_t alignment) { | |
| return (void*)_mi_align_up((uintptr_t)p, alignment); | |
| } | |
| static inline uintptr_t _mi_align_down(uintptr_t sz, size_t alignment) { | |
| mi_assert_internal(alignment != 0); | |
| uintptr_t mask = alignment - 1; | |
| if ((alignment & mask) == 0) { // power of two? | |
| return (sz & ~mask); | |
| } | |
| else { | |
| return ((sz / alignment) * alignment); | |
| } | |
| } | |
| // align a pointer downwards | |
| static inline void* _mi_align_down_ptr(const void* p, size_t alignment) { | |
| return (void*)_mi_align_down((uintptr_t)p, alignment); | |
| } | |
| // Divide upwards: `s <= _mi_divide_up(s,d)*d < s+d`. | |
| static inline uintptr_t _mi_divide_up(uintptr_t size, size_t divider) { | |
| mi_assert_internal(divider != 0); | |
| return (divider == 0 ? size : ((size + divider - 1) / divider)); | |
| } | |
| // clamp an integer | |
| static inline size_t _mi_clamp(size_t sz, size_t min, size_t max) { | |
| if (sz < min) return min; | |
| else if (sz > max) return max; | |
| else return sz; | |
| } | |
| // Is memory zero initialized? | |
| static inline bool mi_mem_is_zero(const void* p, size_t size) { | |
| for (size_t i = 0; i < size; i++) { | |
| if (((uint8_t*)p)[i] != 0) return false; | |
| } | |
| return true; | |
| } | |
| // Align a byte size to a size in _machine words_, | |
| // i.e. byte size == `wsize*sizeof(void*)`. | |
| static inline size_t _mi_wsize_from_size(size_t size) { | |
| mi_assert_internal(size <= SIZE_MAX - sizeof(uintptr_t)); | |
| return (size + sizeof(uintptr_t) - 1) / sizeof(uintptr_t); | |
| } | |
| // Overflow detecting multiply | |
| static inline bool mi_mul_overflow(size_t count, size_t size, size_t* total) { | |
| return __builtin_umull_overflow(count, size, (unsigned long *)total); | |
| return __builtin_umul_overflow(count, size, (unsigned int *)total); | |
| return __builtin_umulll_overflow(count, size, (unsigned long long *)total); | |
| } | |
| static inline bool mi_mul_overflow(size_t count, size_t size, size_t* total) { | |
| *total = count*size; | |
| if mi_likely(((size|count)>>(4*MI_SIZE_SIZE))==0) { // did size and count fit both in the lower half bits of a size_t? | |
| return false; | |
| } | |
| else { | |
| return (size!=0 && (SIZE_MAX / size) < count); | |
| } | |
| } | |
| // Safe multiply `count*size` into `total`; return `true` on overflow. | |
| static inline bool mi_count_size_overflow(size_t count, size_t size, size_t* total) { | |
| if (count==1) { // quick check for the case where count is one (common for C++ allocators) | |
| *total = size; | |
| return false; | |
| } | |
| else if mi_likely(!mi_mul_overflow(count, size, total)) { | |
| return false; | |
| } | |
| else { | |
| _mi_error_message(EOVERFLOW, "allocation request is too large (%zu * %zu bytes)\n", count, size); | |
| *total = SIZE_MAX; | |
| return true; | |
| } | |
| } | |
| /*---------------------------------------------------------------------------------------- | |
| Heap functions | |
| ------------------------------------------------------------------------------------------- */ | |
| extern mi_decl_hidden const mi_theap_t _mi_theap_empty; // read-only empty theap, initial value of the thread local default theap (in the MI_TLS_MODEL_THREAD_LOCAL) | |
| extern mi_decl_hidden const mi_theap_t _mi_theap_empty_wrong; // read-only empty theap used to signal that a theap for a heap could not be allocated | |
| static inline mi_heap_t* _mi_theap_heap(const mi_theap_t* theap) { | |
| return mi_atomic_load_ptr_acquire(mi_heap_t,&theap->heap); | |
| } | |
| static inline bool mi_theap_is_initialized(const mi_theap_t* theap) { | |
| return (theap != NULL && _mi_theap_heap(theap) != NULL); | |
| } | |
| static inline mi_page_t* _mi_theap_get_free_small_page(mi_theap_t* theap, size_t size) { | |
| mi_assert_internal(size <= (MI_SMALL_SIZE_MAX + MI_PADDING_SIZE)); | |
| const size_t idx = _mi_wsize_from_size(size); | |
| mi_assert_internal(idx < MI_PAGES_DIRECT); | |
| return theap->pages_free_direct[idx]; | |
| } | |
| //static inline uintptr_t _mi_ptr_cookie(const void* p) { | |
| // extern mi_theap_t _mi_theap_main; | |
| // mi_assert_internal(_mi_theap_main.cookie != 0); | |
| // return ((uintptr_t)p ^ _mi_theap_main.cookie); | |
| //} | |
| /* ----------------------------------------------------------- | |
| The page map maps addresses to `mi_page_t` pointers | |
| ----------------------------------------------------------- */ | |
| // flat page-map committed on demand, using one byte per slice (64 KiB). | |
| // single indirection and low commit, but large initial virtual reserve (4 GiB with 48 bit virtual addresses) | |
| // used by default on <= 40 bit virtual address spaces. | |
| extern mi_decl_hidden uint8_t* _mi_page_map; | |
| static inline size_t _mi_page_map_index(const void* p) { | |
| return (size_t)((uintptr_t)p >> MI_ARENA_SLICE_SHIFT); | |
| } | |
| static inline mi_page_t* _mi_ptr_page_ex(const void* p, bool* valid) { | |
| const size_t idx = _mi_page_map_index(p); | |
| const size_t ofs = _mi_page_map[idx]; | |
| if (valid != NULL) { *valid = (ofs != 0); } | |
| return (mi_page_t*)((((uintptr_t)p >> MI_ARENA_SLICE_SHIFT) + 1 - ofs) << MI_ARENA_SLICE_SHIFT); | |
| } | |
| static inline mi_page_t* _mi_checked_ptr_page(const void* p) { | |
| bool valid; | |
| mi_page_t* const page = _mi_ptr_page_ex(p, &valid); | |
| return (valid ? page : NULL); | |
| } | |
| static inline mi_page_t* _mi_unchecked_ptr_page(const void* p) { | |
| return _mi_ptr_page_ex(p, NULL); | |
| } | |
| // 2-level page map: | |
| // double indirection, but low commit and low virtual reserve. | |
| // | |
| // the page-map is usually 4 MiB (for 48 bit virtual addresses) and points to sub maps of 64 KiB. | |
| // the page-map is committed on-demand (in 64 KiB parts) (and sub-maps are committed on-demand as well) | |
| // one sub page-map = 64 KiB => covers 2^(16-3) * 2^16 = 2^29 = 512 MiB address space | |
| // the page-map needs 48-(16+13) = 19 bits => 2^19 sub map pointers = 2^22 bytes = 4 MiB reserved size. | |
| typedef mi_page_t** mi_submap_t; | |
| extern mi_decl_hidden _Atomic(mi_submap_t)* _mi_page_map; | |
| static inline size_t _mi_page_map_index(const void* p, size_t* sub_idx) { | |
| const size_t u = (size_t)((uintptr_t)p / MI_ARENA_SLICE_SIZE); | |
| if (sub_idx != NULL) { *sub_idx = u % MI_PAGE_MAP_SUB_COUNT; } | |
| return (u / MI_PAGE_MAP_SUB_COUNT); | |
| } | |
| static inline mi_submap_t _mi_page_map_at(size_t idx) { | |
| return mi_atomic_load_ptr_relaxed(mi_page_t*, &_mi_page_map[idx]); | |
| } | |
| static inline mi_page_t* _mi_unchecked_ptr_page(const void* p) { | |
| size_t sub_idx; | |
| const size_t idx = _mi_page_map_index(p, &sub_idx); | |
| return (_mi_page_map_at(idx))[sub_idx]; // NULL if p==NULL | |
| } | |
| static inline mi_page_t* _mi_checked_ptr_page(const void* p) { | |
| size_t sub_idx; | |
| const size_t idx = _mi_page_map_index(p, &sub_idx); | |
| mi_submap_t const sub = _mi_page_map_at(idx); | |
| if mi_unlikely(sub == NULL) return NULL; | |
| return sub[sub_idx]; | |
| } | |
| static inline mi_page_t* _mi_ptr_page(const void* p) { | |
| mi_assert_internal(p==NULL || mi_is_in_heap_region(p)); | |
| return _mi_checked_ptr_page(p); | |
| return _mi_unchecked_ptr_page(p); | |
| } | |
| // Get the block size of a page | |
| static inline size_t mi_page_block_size(const mi_page_t* page) { | |
| mi_assert_internal(page->block_size > 0); | |
| return page->block_size; | |
| } | |
| // Page start | |
| static inline uint8_t* mi_page_start(const mi_page_t* page) { | |
| return page->page_start; | |
| } | |
| static inline size_t mi_page_size(const mi_page_t* page) { | |
| return mi_page_block_size(page) * page->reserved; | |
| } | |
| static inline uint8_t* mi_page_area(const mi_page_t* page, size_t* size) { | |
| if (size) { *size = mi_page_size(page); } | |
| return mi_page_start(page); | |
| } | |
| static inline size_t mi_page_info_size(void) { | |
| return _mi_align_up(sizeof(mi_page_t), MI_MAX_ALIGN_SIZE); | |
| } | |
| static inline bool mi_page_contains_address(const mi_page_t* page, const void* p) { | |
| size_t psize; | |
| uint8_t* start = mi_page_area(page, &psize); | |
| return (start <= (uint8_t*)p && (uint8_t*)p < start + psize); | |
| } | |
| static inline bool mi_page_is_in_arena(const mi_page_t* page) { | |
| return (page->memid.memkind == MI_MEM_ARENA); | |
| } | |
| static inline bool mi_page_is_singleton(const mi_page_t* page) { | |
| return (page->reserved == 1); | |
| } | |
| // Get the usable block size of a page without fixed padding. | |
| // This may still include internal padding due to alignment and rounding up size classes. | |
| static inline size_t mi_page_usable_block_size(const mi_page_t* page) { | |
| return mi_page_block_size(page) - MI_PADDING_SIZE; | |
| } | |
| static inline bool mi_page_meta_is_separated(const mi_page_t* page) { | |
| // usually separated but can still be in front for direct OS allocations (due to size or alignment) or due to MI_PAGE_META_ALIGNED_FREE_SMALL | |
| return (page->memid.memkind == MI_MEM_ARENA && page != _mi_align_down_ptr(page->page_start, MI_ARENA_SLICE_ALIGN)); | |
| MI_UNUSED(page); | |
| return false; | |
| } | |
| static inline uint8_t* mi_page_slice_start(const mi_page_t* page) { | |
| if (mi_page_meta_is_separated(page)) { | |
| // page meta info is at a separate location (at `arena->pages`) | |
| return (uint8_t*)_mi_align_down_ptr(page->page_start, MI_ARENA_SLICE_ALIGN); | |
| } | |
| else { | |
| // page meta info is at the start of the page slices | |
| return (uint8_t*)page; | |
| } | |
| } | |
| // This gives the offset relative to the start slice of a page. | |
| static inline size_t mi_page_slice_offset_of(const mi_page_t* page, size_t offset_relative_to_page_start) { | |
| return (page->page_start - mi_page_slice_start(page)) + offset_relative_to_page_start; | |
| } | |
| // Currently committed part of a page | |
| static inline size_t mi_page_committed(const mi_page_t* page) { | |
| return (page->slice_committed == 0 ? mi_page_size(page) : page->slice_committed - mi_page_slice_offset_of(page,0)); | |
| } | |
| // are all blocks in a page freed? | |
| // note: needs up-to-date used count, (as the `xthread_free` list may not be empty). see `_mi_page_collect_free`. | |
| static inline bool mi_page_all_free(const mi_page_t* page) { | |
| mi_assert_internal(page != NULL); | |
| return (page->used == 0); | |
| } | |
| // are there immediately available blocks, i.e. blocks available on the free list. | |
| static inline bool mi_page_immediate_available(const mi_page_t* page) { | |
| mi_assert_internal(page != NULL); | |
| return (page->free != NULL); | |
| } | |
| // is the page not yet used up to its reserved space? | |
| static inline bool mi_page_is_expandable(const mi_page_t* page) { | |
| mi_assert_internal(page != NULL); | |
| mi_assert_internal(page->capacity <= page->reserved); | |
| return (page->capacity < page->reserved); | |
| } | |
| static inline bool mi_page_is_full(const mi_page_t* page) { | |
| const bool full = (page->reserved == page->used); | |
| mi_assert_internal(!full || page->free == NULL); | |
| return full; | |
| } | |
| // is more than 7/8th of a page in use? | |
| static inline bool mi_page_is_mostly_used(const mi_page_t* page) { | |
| if (page==NULL) return true; | |
| uint16_t frac = page->reserved / 8U; | |
| return (page->reserved - page->used <= frac); | |
| } | |
| // is more than (n-1)/n'th of a page in use? | |
| static inline bool mi_page_is_used_at_frac(const mi_page_t* page, uint16_t n) { | |
| if (page==NULL) return true; | |
| uint16_t frac = page->reserved / n; | |
| return (page->reserved - page->used <= frac); | |
| } | |
| static inline bool mi_page_is_huge(const mi_page_t* page) { | |
| return (mi_page_is_singleton(page) && | |
| (page->block_size > MI_LARGE_MAX_OBJ_SIZE || | |
| (mi_memkind_is_os(page->memid.memkind) && page->memid.mem.os.base < (void*)page))); | |
| } | |
| static inline mi_page_queue_t* mi_page_queue(const mi_theap_t* theap, size_t size) { | |
| mi_page_queue_t* const pq = &((mi_theap_t*)theap)->pages[_mi_bin(size)]; | |
| if (size <= MI_LARGE_MAX_OBJ_SIZE) { mi_assert_internal(pq->block_size <= MI_LARGE_MAX_OBJ_SIZE); } | |
| return pq; | |
| } | |
| //----------------------------------------------------------- | |
| // Page thread id and flags | |
| //----------------------------------------------------------- | |
| // Thread id of thread that owns this page (with flags in the bottom 2 bits) | |
| static inline mi_threadid_t mi_page_xthread_id(const mi_page_t* page) { | |
| return mi_atomic_load_relaxed(&((mi_page_t*)page)->xthread_id); | |
| } | |
| // Plain thread id of the thread that owns this page | |
| static inline mi_threadid_t mi_page_thread_id(const mi_page_t* page) { | |
| return (mi_page_xthread_id(page) & ~MI_PAGE_FLAG_MASK); | |
| } | |
| static inline mi_page_flags_t mi_page_flags(const mi_page_t* page) { | |
| return (mi_page_xthread_id(page) & MI_PAGE_FLAG_MASK); | |
| } | |
| static inline bool mi_page_flags_set(mi_page_t* page, bool set, mi_page_flags_t newflag) { | |
| mi_page_flags_t old; | |
| if (set) { old = mi_atomic_or_relaxed(&page->xthread_id, newflag); } | |
| else { old = mi_atomic_and_relaxed(&page->xthread_id, ~newflag); } | |
| return ((old & newflag) == newflag); | |
| } | |
| static inline bool mi_page_is_in_full(const mi_page_t* page) { | |
| return ((mi_page_flags(page) & MI_PAGE_IN_FULL_QUEUE) != 0); | |
| } | |
| static inline void mi_page_set_in_full(mi_page_t* page, bool in_full) { | |
| const bool was_in_full = mi_page_flags_set(page, in_full, MI_PAGE_IN_FULL_QUEUE); | |
| if (was_in_full != in_full) { | |
| // optimize: maintain pages_full_size to avoid visiting the full queue (issue #1220) | |
| mi_theap_t* const theap = page->theap; | |
| mi_assert_internal(theap!=NULL); | |
| if (theap != NULL) { | |
| const size_t size = page->capacity * mi_page_block_size(page); | |
| if (in_full) { theap->pages_full_size += size; } | |
| else { mi_assert_internal(size <= theap->pages_full_size); theap->pages_full_size -= size; } | |
| } | |
| } | |
| } | |
| static inline bool mi_page_has_interior_pointers(const mi_page_t* page) { | |
| return ((mi_page_flags(page) & MI_PAGE_HAS_INTERIOR_POINTERS) != 0); | |
| } | |
| static inline void mi_page_set_has_interior_pointers(mi_page_t* page, bool has_aligned) { | |
| mi_page_flags_set(page, has_aligned, MI_PAGE_HAS_INTERIOR_POINTERS); | |
| } | |
| static inline void mi_page_set_theap(mi_page_t* page, mi_theap_t* theap) { | |
| // mi_assert_internal(!mi_page_is_in_full(page)); // can happen when destroying pages on theap_destroy | |
| page->theap = theap; | |
| const mi_threadid_t tid = (theap == NULL ? MI_THREADID_ABANDONED : theap->tld->thread_id); | |
| mi_assert_internal((tid & MI_PAGE_FLAG_MASK) == 0); | |
| // we need to use an atomic cas since a concurrent thread may still set the MI_PAGE_HAS_INTERIOR_POINTERS flag (see `alloc_aligned.c`). | |
| mi_threadid_t xtid_old = mi_page_xthread_id(page); | |
| mi_threadid_t xtid; | |
| do { | |
| xtid = tid | (xtid_old & MI_PAGE_FLAG_MASK); | |
| } while (!mi_atomic_cas_weak_release(&page->xthread_id, &xtid_old, xtid)); | |
| } | |
| static inline bool mi_page_is_abandoned(const mi_page_t* page) { | |
| // note: the xtheap field of an abandoned theap is set to the subproc (for fast reclaim-on-free) | |
| return (mi_page_thread_id(page) <= MI_THREADID_ABANDONED_MAPPED); | |
| } | |
| static inline bool mi_page_is_abandoned_mapped(const mi_page_t* page) { | |
| return (mi_page_thread_id(page) == MI_THREADID_ABANDONED_MAPPED); | |
| } | |
| static inline void mi_page_set_abandoned_mapped(mi_page_t* page) { | |
| mi_assert_internal(mi_page_is_abandoned(page)); | |
| mi_atomic_or_relaxed(&page->xthread_id, (mi_threadid_t)MI_THREADID_ABANDONED_MAPPED); | |
| } | |
| static inline void mi_page_clear_abandoned_mapped(mi_page_t* page) { | |
| mi_assert_internal(mi_page_is_abandoned_mapped(page)); | |
| mi_atomic_and_relaxed(&page->xthread_id, (mi_threadid_t)MI_PAGE_FLAG_MASK); | |
| } | |
| static inline mi_theap_t* mi_page_theap(const mi_page_t* page) { | |
| mi_assert_internal(!mi_page_is_abandoned(page)); | |
| mi_assert_internal(page->theap != NULL); | |
| return page->theap; | |
| } | |
| static inline mi_tld_t* mi_page_tld(const mi_page_t* page) { | |
| mi_assert_internal(!mi_page_is_abandoned(page)); | |
| mi_assert_internal(page->theap != NULL); | |
| return page->theap->tld; | |
| } | |
| static inline mi_heap_t* mi_page_heap(const mi_page_t* page) { | |
| mi_heap_t* heap = page->heap; | |
| // we use NULL for the main heap to make `_mi_page_get_associated_theap` fast in `free.c:mi_abandoned_page_try_reclaim`. | |
| if mi_likely(heap==NULL) heap = mi_heap_main(); | |
| mi_assert_internal(heap != NULL); | |
| return heap; | |
| } | |
| //----------------------------------------------------------- | |
| // Thread free list and ownership | |
| //----------------------------------------------------------- | |
| // Thread free flag helpers | |
| static inline mi_block_t* mi_tf_block(mi_thread_free_t tf) { | |
| return (mi_block_t*)(tf & ~1); | |
| } | |
| static inline bool mi_tf_is_owned(mi_thread_free_t tf) { | |
| return ((tf & 1) == 1); | |
| } | |
| static inline mi_thread_free_t mi_tf_create(mi_block_t* block, bool owned) { | |
| return (mi_thread_free_t)((uintptr_t)block | (owned ? 1 : 0)); | |
| } | |
| // Thread free access | |
| static inline mi_block_t* mi_page_thread_free(const mi_page_t* page) { | |
| return mi_tf_block(mi_atomic_load_relaxed(&((mi_page_t*)page)->xthread_free)); | |
| } | |
| // are there any available blocks? | |
| static inline bool mi_page_has_any_available(const mi_page_t* page) { | |
| mi_assert_internal(page != NULL && page->reserved > 0); | |
| return (page->used < page->reserved || (mi_page_thread_free(page) != NULL)); | |
| } | |
| // Owned? | |
| static inline bool mi_page_is_owned(const mi_page_t* page) { | |
| return mi_tf_is_owned(mi_atomic_load_relaxed(&((mi_page_t*)page)->xthread_free)); | |
| } | |
| // get ownership; returns true if the page was not owned before. | |
| static inline bool mi_page_claim_ownership(mi_page_t* page) { | |
| const uintptr_t old = mi_atomic_or_acq_rel(&page->xthread_free, (uintptr_t)1); | |
| return ((old&1)==0); | |
| } | |
| /* ------------------------------------------------------------------- | |
| Guarded objects | |
| ------------------------------------------------------------------- */ | |
| // we always align guarded pointers in a block at an offset | |
| // the block `next` field is then used as a tag to distinguish regular offset aligned blocks from guarded ones | |
| static inline bool mi_block_ptr_is_guarded(const mi_block_t* block, const void* p) { | |
| const ptrdiff_t offset = (uint8_t*)p - (uint8_t*)block; | |
| return (offset >= (ptrdiff_t)(sizeof(mi_block_t)) && block->next == MI_BLOCK_TAG_GUARDED); | |
| MI_UNUSED(block); MI_UNUSED(p); | |
| return false; | |
| } | |
| static inline bool mi_theap_malloc_use_guarded(mi_theap_t* theap, size_t size) { | |
| // this code is written to result in fast assembly as it is on the hot path for allocation | |
| const size_t count = theap->guarded_sample_count - 1; // if the rate was 0, this will underflow and count for a long time.. | |
| if mi_likely(count != 0) { | |
| // no sample | |
| theap->guarded_sample_count = count; | |
| return false; | |
| } | |
| else if (size >= theap->guarded_size_min && size <= theap->guarded_size_max) { | |
| // use guarded allocation | |
| theap->guarded_sample_count = theap->guarded_sample_rate; // reset | |
| return (theap->guarded_sample_rate != 0); | |
| } | |
| else { | |
| // failed size criteria, rewind count (but don't write to an empty theap) | |
| if (theap->guarded_sample_rate != 0) { theap->guarded_sample_count = 1; } | |
| return false; | |
| } | |
| } | |
| mi_decl_restrict void* _mi_theap_malloc_guarded(mi_theap_t* theap, size_t size, bool zero) mi_attr_noexcept; | |
| /* ------------------------------------------------------------------- | |
| Encoding/Decoding the free list next pointers | |
| This is to protect against buffer overflow exploits where the | |
| free list is mutated. Many hardened allocators xor the next pointer `p` | |
| with a secret key `k1`, as `p^k1`. This prevents overwriting with known | |
| values but might be still too weak: if the attacker can guess | |
| the pointer `p` this can reveal `k1` (since `p^k1^p == k1`). | |
| Moreover, if multiple blocks can be read as well, the attacker can | |
| xor both as `(p1^k1) ^ (p2^k1) == p1^p2` which may reveal a lot | |
| about the pointers (and subsequently `k1`). | |
| Instead mimalloc uses an extra key `k2` and encodes as `((p^k2)<<<k1)+k1`. | |
| Since these operations are not associative, the above approaches do not | |
| work so well any more even if the `p` can be guesstimated. For example, | |
| for the read case we can subtract two entries to discard the `+k1` term, | |
| but that leads to `((p1^k2)<<<k1) - ((p2^k2)<<<k1)` at best. | |
| We include the left-rotation since xor and addition are otherwise linear | |
| in the lowest bit. Finally, both keys are unique per page which reduces | |
| the re-use of keys by a large factor. | |
| We also pass a separate `null` value to be used as `NULL` or otherwise | |
| `(k2<<<k1)+k1` would appear (too) often as a sentinel value. | |
| ------------------------------------------------------------------- */ | |
| static inline bool mi_is_in_same_page(const void* p, const void* q) { | |
| mi_page_t* page = _mi_ptr_page(p); | |
| return mi_page_contains_address(page,q); | |
| // return (_mi_ptr_page(p) == _mi_ptr_page(q)); | |
| } | |
| static inline void* mi_ptr_decode(const void* null, const mi_encoded_t x, const uintptr_t* keys) { | |
| void* p = (void*)(mi_rotr(x - keys[0], keys[0]) ^ keys[1]); | |
| return (p==null ? NULL : p); | |
| } | |
| static inline mi_encoded_t mi_ptr_encode(const void* null, const void* p, const uintptr_t* keys) { | |
| uintptr_t x = (uintptr_t)(p==NULL ? null : p); | |
| return mi_rotl(x ^ keys[1], keys[0]) + keys[0]; | |
| } | |
| static inline uint32_t mi_ptr_encode_canary(const void* null, const void* p, const uintptr_t* keys) { | |
| const uint32_t x = (uint32_t)(mi_ptr_encode(null,p,keys)); | |
| // make the lowest byte 0 to prevent spurious read overflows which could be a security issue (issue #951) | |
| return (x & 0x00FFFFFF); | |
| return (x & 0xFFFFFF00); | |
| } | |
| static inline mi_block_t* mi_block_nextx( const void* null, const mi_block_t* block, const uintptr_t* keys ) { | |
| mi_track_mem_defined(block,sizeof(mi_block_t)); | |
| mi_block_t* next; | |
| next = (mi_block_t*)mi_ptr_decode(null, block->next, keys); | |
| MI_UNUSED(keys); MI_UNUSED(null); | |
| next = (mi_block_t*)block->next; | |
| mi_track_mem_noaccess(block,sizeof(mi_block_t)); | |
| return next; | |
| } | |
| static inline void mi_block_set_nextx(const void* null, mi_block_t* block, const mi_block_t* next, const uintptr_t* keys) { | |
| mi_track_mem_undefined(block,sizeof(mi_block_t)); | |
| block->next = mi_ptr_encode(null, next, keys); | |
| MI_UNUSED(keys); MI_UNUSED(null); | |
| block->next = (mi_encoded_t)next; | |
| mi_track_mem_noaccess(block,sizeof(mi_block_t)); | |
| } | |
| static inline mi_block_t* mi_block_next(const mi_page_t* page, const mi_block_t* block) { | |
| mi_block_t* next = mi_block_nextx(page,block,page->keys); | |
| // check for free list corruption: is `next` at least in the same page? | |
| // TODO: check if `next` is `page->block_size` aligned? | |
| if mi_unlikely(next!=NULL && !mi_is_in_same_page(block, next)) { | |
| _mi_error_message(EFAULT, "corrupted free list entry of size %zub at %p: value 0x%zx\n", mi_page_block_size(page), block, (uintptr_t)next); | |
| next = NULL; | |
| } | |
| return next; | |
| MI_UNUSED(page); | |
| return mi_block_nextx(page,block,NULL); | |
| } | |
| static inline void mi_block_set_next(const mi_page_t* page, mi_block_t* block, const mi_block_t* next) { | |
| mi_block_set_nextx(page,block,next, page->keys); | |
| MI_UNUSED(page); | |
| mi_block_set_nextx(page,block,next,NULL); | |
| } | |
| /* ----------------------------------------------------------- | |
| arena blocks | |
| ----------------------------------------------------------- */ | |
| // Blocks needed for a given byte size | |
| static inline size_t mi_slice_count_of_size(size_t size) { | |
| return _mi_divide_up(size, MI_ARENA_SLICE_SIZE); | |
| } | |
| // Byte size of a number of blocks | |
| static inline size_t mi_size_of_slices(size_t bcount) { | |
| return (bcount * MI_ARENA_SLICE_SIZE); | |
| } | |
| /* ----------------------------------------------------------- | |
| memory id's | |
| ----------------------------------------------------------- */ | |
| static inline mi_memid_t _mi_memid_create(mi_memkind_t memkind) { | |
| mi_memid_t memid; | |
| _mi_memzero_var(memid); | |
| memid.memkind = memkind; | |
| return memid; | |
| } | |
| static inline mi_memid_t _mi_memid_none(void) { | |
| return _mi_memid_create(MI_MEM_NONE); | |
| } | |
| static inline mi_memid_t _mi_memid_create_os(void* base, size_t size, bool committed, bool is_zero, bool is_large) { | |
| mi_memid_t memid = _mi_memid_create(MI_MEM_OS); | |
| memid.mem.os.base = base; | |
| memid.mem.os.size = size; | |
| memid.initially_committed = committed; | |
| memid.initially_zero = is_zero; | |
| memid.is_pinned = is_large; | |
| return memid; | |
| } | |
| static inline mi_memid_t _mi_memid_create_meta(void* mpage, size_t block_idx, size_t block_count) { | |
| mi_memid_t memid = _mi_memid_create(MI_MEM_META); | |
| memid.mem.meta.meta_page = mpage; | |
| memid.mem.meta.block_index = (uint32_t)block_idx; | |
| memid.mem.meta.block_count = (uint32_t)block_count; | |
| memid.initially_committed = true; | |
| memid.initially_zero = true; | |
| memid.is_pinned = true; | |
| return memid; | |
| } | |
| // ------------------------------------------------------------------- | |
| // Fast "random" shuffle | |
| // ------------------------------------------------------------------- | |
| static inline uintptr_t _mi_random_shuffle(uintptr_t x) { | |
| if (x==0) { x = 17; } // ensure we don't get stuck in generating zeros | |
| // by Sebastiano Vigna, see: <http://xoshiro.di.unimi.it/splitmix64.c> | |
| x ^= x >> 30; | |
| x *= 0xbf58476d1ce4e5b9UL; | |
| x ^= x >> 27; | |
| x *= 0x94d049bb133111ebUL; | |
| x ^= x >> 31; | |
| // by Chris Wellons, see: <https://nullprogram.com/blog/2018/07/31/> | |
| x ^= x >> 16; | |
| x *= 0x7feb352dUL; | |
| x ^= x >> 15; | |
| x *= 0x846ca68bUL; | |
| x ^= x >> 16; | |
| return x; | |
| } | |
| // --------------------------------------------------------------------------------- | |
| // Provide our own `_mi_memcpy/set` for potential performance optimizations. | |
| // | |
| // For now, only on x64/x86 we optimize to `rep movsb/stosb`. | |
| // Generally, we check for "fast short rep movsb/stosb" (FSRM/FSRS) or "fast enhanced rep movsb" (ERMS) support | |
| // (AMD Zen3+ (~2020) or Intel Ice Lake+ (~2017). See also issue #201 and pr #253. | |
| // Todo: we see improvements on win32 but less with glibc; we might want to only enable this on windows. | |
| // --------------------------------------------------------------------------------- | |
| extern mi_decl_hidden size_t _mi_cpu_movsb_max; // in init.c | |
| extern mi_decl_hidden size_t _mi_cpu_stosb_max; | |
| static inline void mi_rep_movsb(void* dst, const void* src, size_t n) { | |
| __asm volatile("rep movsb" : "+D"(dst), "+c"(n), "+S"(src) : : "memory"); | |
| __movsb((unsigned char*)dst, (const unsigned char*)src, n); | |
| } | |
| static inline void mi_rep_stosb(void* dst, uint8_t val, size_t n) { | |
| __asm volatile("rep stosb" : "+D"(dst), "+c"(n) : "a"(val) : "memory"); | |
| __stosb((unsigned char*)dst, val, n); | |
| } | |
| static inline void _mi_memcpy(void* dst, const void* src, size_t n) { | |
| if mi_likely(n <= _mi_cpu_movsb_max) { // has fsrm && n <= 127 (todo: and maybe has erms?) | |
| mi_rep_movsb(dst, src, n); | |
| } | |
| else { | |
| memcpy(dst, src, n); | |
| } | |
| } | |
| static inline void _mi_memset(void* dst, int val, size_t n) { | |
| if mi_likely(n <= _mi_cpu_stosb_max) { // has fsrs && n <= 127 | |
| mi_rep_stosb(dst, (uint8_t)val, n); | |
| } | |
| else { | |
| memset(dst, val, n); | |
| } | |
| } | |
| static inline void _mi_memcpy(void* dst, const void* src, size_t n) { | |
| memcpy(dst, src, n); | |
| } | |
| static inline void _mi_memset(void* dst, int val, size_t n) { | |
| memset(dst, val, n); | |
| } | |
| // ------------------------------------------------------------------------------- | |
| // The `_mi_memcpy_aligned` can be used if the pointers are machine-word aligned | |
| // This is used for example in `mi_realloc`. | |
| // ------------------------------------------------------------------------------- | |
| // On GCC/CLang we provide a hint that the pointers are word aligned. | |
| static inline void _mi_memcpy_aligned(void* dst, const void* src, size_t n) { | |
| mi_assert_internal(((uintptr_t)dst % MI_INTPTR_SIZE == 0) && ((uintptr_t)src % MI_INTPTR_SIZE == 0)); | |
| void* adst = __builtin_assume_aligned(dst, MI_INTPTR_SIZE); | |
| const void* asrc = __builtin_assume_aligned(src, MI_INTPTR_SIZE); | |
| _mi_memcpy(adst, asrc, n); | |
| } | |
| static inline void _mi_memset_aligned(void* dst, int val, size_t n) { | |
| mi_assert_internal((uintptr_t)dst % MI_INTPTR_SIZE == 0); | |
| void* adst = __builtin_assume_aligned(dst, MI_INTPTR_SIZE); | |
| _mi_memset(adst, val, n); | |
| } | |
| // Default fallback on `_mi_memcpy` | |
| static inline void _mi_memcpy_aligned(void* dst, const void* src, size_t n) { | |
| mi_assert_internal(((uintptr_t)dst % MI_INTPTR_SIZE == 0) && ((uintptr_t)src % MI_INTPTR_SIZE == 0)); | |
| _mi_memcpy(dst, src, n); | |
| } | |
| static inline void _mi_memset_aligned(void* dst, int val, size_t n) { | |
| mi_assert_internal((uintptr_t)dst % MI_INTPTR_SIZE == 0); | |
| _mi_memset(dst, val, n); | |
| } | |
| static inline void _mi_memzero(void* dst, size_t n) { | |
| _mi_memset(dst, 0, n); | |
| } | |
| static inline void _mi_memzero_aligned(void* dst, size_t n) { | |
| _mi_memset_aligned(dst, 0, n); | |
| } | |
Xet Storage Details
- Size:
- 51.1 kB
- Xet hash:
- 3b750161c94fb71bacedf5b9c6d315d32a3b8152604345ee3949153a9b192d4a
·
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.