c14e6c0819
- Show number/bytes of extents of each size that are dirty, muzzy, retained.
2221 lines
66 KiB
C
2221 lines
66 KiB
C
#define JEMALLOC_EXTENT_C_
|
|
#include "jemalloc/internal/jemalloc_preamble.h"
|
|
#include "jemalloc/internal/jemalloc_internal_includes.h"
|
|
|
|
#include "jemalloc/internal/assert.h"
|
|
#include "jemalloc/internal/extent_dss.h"
|
|
#include "jemalloc/internal/extent_mmap.h"
|
|
#include "jemalloc/internal/ph.h"
|
|
#include "jemalloc/internal/rtree.h"
|
|
#include "jemalloc/internal/mutex.h"
|
|
#include "jemalloc/internal/mutex_pool.h"
|
|
|
|
/******************************************************************************/
|
|
/* Data. */
|
|
|
|
rtree_t extents_rtree;
|
|
/* Keyed by the address of the extent_t being protected. */
|
|
mutex_pool_t extent_mutex_pool;
|
|
|
|
size_t opt_lg_extent_max_active_fit = LG_EXTENT_MAX_ACTIVE_FIT_DEFAULT;
|
|
|
|
static const bitmap_info_t extents_bitmap_info =
|
|
BITMAP_INFO_INITIALIZER(SC_NPSIZES+1);
|
|
|
|
static void *extent_alloc_default(extent_hooks_t *extent_hooks, void *new_addr,
|
|
size_t size, size_t alignment, bool *zero, bool *commit,
|
|
unsigned arena_ind);
|
|
static bool extent_dalloc_default(extent_hooks_t *extent_hooks, void *addr,
|
|
size_t size, bool committed, unsigned arena_ind);
|
|
static void extent_destroy_default(extent_hooks_t *extent_hooks, void *addr,
|
|
size_t size, bool committed, unsigned arena_ind);
|
|
static bool extent_commit_default(extent_hooks_t *extent_hooks, void *addr,
|
|
size_t size, size_t offset, size_t length, unsigned arena_ind);
|
|
static bool extent_commit_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length, bool growing_retained);
|
|
static bool extent_decommit_default(extent_hooks_t *extent_hooks,
|
|
void *addr, size_t size, size_t offset, size_t length, unsigned arena_ind);
|
|
#ifdef PAGES_CAN_PURGE_LAZY
|
|
static bool extent_purge_lazy_default(extent_hooks_t *extent_hooks, void *addr,
|
|
size_t size, size_t offset, size_t length, unsigned arena_ind);
|
|
#endif
|
|
static bool extent_purge_lazy_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length, bool growing_retained);
|
|
#ifdef PAGES_CAN_PURGE_FORCED
|
|
static bool extent_purge_forced_default(extent_hooks_t *extent_hooks,
|
|
void *addr, size_t size, size_t offset, size_t length, unsigned arena_ind);
|
|
#endif
|
|
static bool extent_purge_forced_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length, bool growing_retained);
|
|
#ifdef JEMALLOC_MAPS_COALESCE
|
|
static bool extent_split_default(extent_hooks_t *extent_hooks, void *addr,
|
|
size_t size, size_t size_a, size_t size_b, bool committed,
|
|
unsigned arena_ind);
|
|
#endif
|
|
static extent_t *extent_split_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t size_a,
|
|
szind_t szind_a, bool slab_a, size_t size_b, szind_t szind_b, bool slab_b,
|
|
bool growing_retained);
|
|
#ifdef JEMALLOC_MAPS_COALESCE
|
|
static bool extent_merge_default(extent_hooks_t *extent_hooks, void *addr_a,
|
|
size_t size_a, void *addr_b, size_t size_b, bool committed,
|
|
unsigned arena_ind);
|
|
#endif
|
|
static bool extent_merge_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *a, extent_t *b,
|
|
bool growing_retained);
|
|
|
|
const extent_hooks_t extent_hooks_default = {
|
|
extent_alloc_default,
|
|
extent_dalloc_default,
|
|
extent_destroy_default,
|
|
extent_commit_default,
|
|
extent_decommit_default
|
|
#ifdef PAGES_CAN_PURGE_LAZY
|
|
,
|
|
extent_purge_lazy_default
|
|
#else
|
|
,
|
|
NULL
|
|
#endif
|
|
#ifdef PAGES_CAN_PURGE_FORCED
|
|
,
|
|
extent_purge_forced_default
|
|
#else
|
|
,
|
|
NULL
|
|
#endif
|
|
#ifdef JEMALLOC_MAPS_COALESCE
|
|
,
|
|
extent_split_default,
|
|
extent_merge_default
|
|
#endif
|
|
};
|
|
|
|
/* Used exclusively for gdump triggering. */
|
|
static atomic_zu_t curpages;
|
|
static atomic_zu_t highpages;
|
|
|
|
/******************************************************************************/
|
|
/*
|
|
* Function prototypes for static functions that are referenced prior to
|
|
* definition.
|
|
*/
|
|
|
|
static void extent_deregister(tsdn_t *tsdn, extent_t *extent);
|
|
static extent_t *extent_recycle(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extents_t *extents, void *new_addr,
|
|
size_t usize, size_t pad, size_t alignment, bool slab, szind_t szind,
|
|
bool *zero, bool *commit, bool growing_retained);
|
|
static extent_t *extent_try_coalesce(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, rtree_ctx_t *rtree_ctx, extents_t *extents,
|
|
extent_t *extent, bool *coalesced, bool growing_retained);
|
|
static void extent_record(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extents_t *extents, extent_t *extent,
|
|
bool growing_retained);
|
|
|
|
/******************************************************************************/
|
|
|
|
#define ATTR_NONE /* does nothing */
|
|
|
|
ph_gen(ATTR_NONE, extent_avail_, extent_tree_t, extent_t, ph_link,
|
|
extent_esnead_comp)
|
|
|
|
#undef ATTR_NONE
|
|
|
|
typedef enum {
|
|
lock_result_success,
|
|
lock_result_failure,
|
|
lock_result_no_extent
|
|
} lock_result_t;
|
|
|
|
static lock_result_t
|
|
extent_rtree_leaf_elm_try_lock(tsdn_t *tsdn, rtree_leaf_elm_t *elm,
|
|
extent_t **result) {
|
|
extent_t *extent1 = rtree_leaf_elm_extent_read(tsdn, &extents_rtree,
|
|
elm, true);
|
|
|
|
if (extent1 == NULL) {
|
|
return lock_result_no_extent;
|
|
}
|
|
/*
|
|
* It's possible that the extent changed out from under us, and with it
|
|
* the leaf->extent mapping. We have to recheck while holding the lock.
|
|
*/
|
|
extent_lock(tsdn, extent1);
|
|
extent_t *extent2 = rtree_leaf_elm_extent_read(tsdn,
|
|
&extents_rtree, elm, true);
|
|
|
|
if (extent1 == extent2) {
|
|
*result = extent1;
|
|
return lock_result_success;
|
|
} else {
|
|
extent_unlock(tsdn, extent1);
|
|
return lock_result_failure;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Returns a pool-locked extent_t * if there's one associated with the given
|
|
* address, and NULL otherwise.
|
|
*/
|
|
static extent_t *
|
|
extent_lock_from_addr(tsdn_t *tsdn, rtree_ctx_t *rtree_ctx, void *addr) {
|
|
extent_t *ret = NULL;
|
|
rtree_leaf_elm_t *elm = rtree_leaf_elm_lookup(tsdn, &extents_rtree,
|
|
rtree_ctx, (uintptr_t)addr, false, false);
|
|
if (elm == NULL) {
|
|
return NULL;
|
|
}
|
|
lock_result_t lock_result;
|
|
do {
|
|
lock_result = extent_rtree_leaf_elm_try_lock(tsdn, elm, &ret);
|
|
} while (lock_result == lock_result_failure);
|
|
return ret;
|
|
}
|
|
|
|
extent_t *
|
|
extent_alloc(tsdn_t *tsdn, arena_t *arena) {
|
|
malloc_mutex_lock(tsdn, &arena->extent_avail_mtx);
|
|
extent_t *extent = extent_avail_first(&arena->extent_avail);
|
|
if (extent == NULL) {
|
|
malloc_mutex_unlock(tsdn, &arena->extent_avail_mtx);
|
|
return base_alloc_extent(tsdn, arena->base);
|
|
}
|
|
extent_avail_remove(&arena->extent_avail, extent);
|
|
malloc_mutex_unlock(tsdn, &arena->extent_avail_mtx);
|
|
return extent;
|
|
}
|
|
|
|
void
|
|
extent_dalloc(tsdn_t *tsdn, arena_t *arena, extent_t *extent) {
|
|
malloc_mutex_lock(tsdn, &arena->extent_avail_mtx);
|
|
extent_avail_insert(&arena->extent_avail, extent);
|
|
malloc_mutex_unlock(tsdn, &arena->extent_avail_mtx);
|
|
}
|
|
|
|
extent_hooks_t *
|
|
extent_hooks_get(arena_t *arena) {
|
|
return base_extent_hooks_get(arena->base);
|
|
}
|
|
|
|
extent_hooks_t *
|
|
extent_hooks_set(tsd_t *tsd, arena_t *arena, extent_hooks_t *extent_hooks) {
|
|
background_thread_info_t *info;
|
|
if (have_background_thread) {
|
|
info = arena_background_thread_info_get(arena);
|
|
malloc_mutex_lock(tsd_tsdn(tsd), &info->mtx);
|
|
}
|
|
extent_hooks_t *ret = base_extent_hooks_set(arena->base, extent_hooks);
|
|
if (have_background_thread) {
|
|
malloc_mutex_unlock(tsd_tsdn(tsd), &info->mtx);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void
|
|
extent_hooks_assure_initialized(arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks) {
|
|
if (*r_extent_hooks == EXTENT_HOOKS_INITIALIZER) {
|
|
*r_extent_hooks = extent_hooks_get(arena);
|
|
}
|
|
}
|
|
|
|
#ifndef JEMALLOC_JET
|
|
static
|
|
#endif
|
|
size_t
|
|
extent_size_quantize_floor(size_t size) {
|
|
size_t ret;
|
|
pszind_t pind;
|
|
|
|
assert(size > 0);
|
|
assert((size & PAGE_MASK) == 0);
|
|
|
|
pind = sz_psz2ind(size - sz_large_pad + 1);
|
|
if (pind == 0) {
|
|
/*
|
|
* Avoid underflow. This short-circuit would also do the right
|
|
* thing for all sizes in the range for which there are
|
|
* PAGE-spaced size classes, but it's simplest to just handle
|
|
* the one case that would cause erroneous results.
|
|
*/
|
|
return size;
|
|
}
|
|
ret = sz_pind2sz(pind - 1) + sz_large_pad;
|
|
assert(ret <= size);
|
|
return ret;
|
|
}
|
|
|
|
#ifndef JEMALLOC_JET
|
|
static
|
|
#endif
|
|
size_t
|
|
extent_size_quantize_ceil(size_t size) {
|
|
size_t ret;
|
|
|
|
assert(size > 0);
|
|
assert(size - sz_large_pad <= SC_LARGE_MAXCLASS);
|
|
assert((size & PAGE_MASK) == 0);
|
|
|
|
ret = extent_size_quantize_floor(size);
|
|
if (ret < size) {
|
|
/*
|
|
* Skip a quantization that may have an adequately large extent,
|
|
* because under-sized extents may be mixed in. This only
|
|
* happens when an unusual size is requested, i.e. for aligned
|
|
* allocation, and is just one of several places where linear
|
|
* search would potentially find sufficiently aligned available
|
|
* memory somewhere lower.
|
|
*/
|
|
ret = sz_pind2sz(sz_psz2ind(ret - sz_large_pad + 1)) +
|
|
sz_large_pad;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/* Generate pairing heap functions. */
|
|
ph_gen(, extent_heap_, extent_heap_t, extent_t, ph_link, extent_snad_comp)
|
|
|
|
bool
|
|
extents_init(tsdn_t *tsdn, extents_t *extents, extent_state_t state,
|
|
bool delay_coalesce) {
|
|
if (malloc_mutex_init(&extents->mtx, "extents", WITNESS_RANK_EXTENTS,
|
|
malloc_mutex_rank_exclusive)) {
|
|
return true;
|
|
}
|
|
for (unsigned i = 0; i < SC_NPSIZES + 1; i++) {
|
|
extent_heap_new(&extents->heaps[i]);
|
|
}
|
|
bitmap_init(extents->bitmap, &extents_bitmap_info, true);
|
|
extent_list_init(&extents->lru);
|
|
atomic_store_zu(&extents->npages, 0, ATOMIC_RELAXED);
|
|
extents->state = state;
|
|
extents->delay_coalesce = delay_coalesce;
|
|
return false;
|
|
}
|
|
|
|
extent_state_t
|
|
extents_state_get(const extents_t *extents) {
|
|
return extents->state;
|
|
}
|
|
|
|
size_t
|
|
extents_npages_get(extents_t *extents) {
|
|
return atomic_load_zu(&extents->npages, ATOMIC_RELAXED);
|
|
}
|
|
|
|
size_t
|
|
extents_nextents_get(extents_t *extents, pszind_t pind) {
|
|
return atomic_load_zu(&extents->nextents[pind], ATOMIC_RELAXED);
|
|
}
|
|
|
|
size_t
|
|
extents_nbytes_get(extents_t *extents, pszind_t pind) {
|
|
return atomic_load_zu(&extents->nbytes[pind], ATOMIC_RELAXED);
|
|
}
|
|
|
|
static void
|
|
extents_stats_add(extents_t *extent, pszind_t pind, size_t sz) {
|
|
size_t cur = atomic_load_zu(&extent->nextents[pind], ATOMIC_RELAXED);
|
|
atomic_store_zu(&extent->nextents[pind], cur + 1, ATOMIC_RELAXED);
|
|
cur = atomic_load_zu(&extent->nbytes[pind], ATOMIC_RELAXED);
|
|
atomic_store_zu(&extent->nbytes[pind], cur + sz, ATOMIC_RELAXED);
|
|
}
|
|
|
|
static void
|
|
extents_stats_sub(extents_t *extent, pszind_t pind, size_t sz) {
|
|
size_t cur = atomic_load_zu(&extent->nextents[pind], ATOMIC_RELAXED);
|
|
atomic_store_zu(&extent->nextents[pind], cur - 1, ATOMIC_RELAXED);
|
|
cur = atomic_load_zu(&extent->nbytes[pind], ATOMIC_RELAXED);
|
|
atomic_store_zu(&extent->nbytes[pind], cur - sz, ATOMIC_RELAXED);
|
|
}
|
|
|
|
static void
|
|
extents_insert_locked(tsdn_t *tsdn, extents_t *extents, extent_t *extent) {
|
|
malloc_mutex_assert_owner(tsdn, &extents->mtx);
|
|
assert(extent_state_get(extent) == extents->state);
|
|
|
|
size_t size = extent_size_get(extent);
|
|
size_t psz = extent_size_quantize_floor(size);
|
|
pszind_t pind = sz_psz2ind(psz);
|
|
if (extent_heap_empty(&extents->heaps[pind])) {
|
|
bitmap_unset(extents->bitmap, &extents_bitmap_info,
|
|
(size_t)pind);
|
|
}
|
|
extent_heap_insert(&extents->heaps[pind], extent);
|
|
|
|
if (config_stats) {
|
|
extents_stats_add(extents, pind, size);
|
|
}
|
|
|
|
extent_list_append(&extents->lru, extent);
|
|
size_t npages = size >> LG_PAGE;
|
|
/*
|
|
* All modifications to npages hold the mutex (as asserted above), so we
|
|
* don't need an atomic fetch-add; we can get by with a load followed by
|
|
* a store.
|
|
*/
|
|
size_t cur_extents_npages =
|
|
atomic_load_zu(&extents->npages, ATOMIC_RELAXED);
|
|
atomic_store_zu(&extents->npages, cur_extents_npages + npages,
|
|
ATOMIC_RELAXED);
|
|
}
|
|
|
|
static void
|
|
extents_remove_locked(tsdn_t *tsdn, extents_t *extents, extent_t *extent) {
|
|
malloc_mutex_assert_owner(tsdn, &extents->mtx);
|
|
assert(extent_state_get(extent) == extents->state);
|
|
|
|
size_t size = extent_size_get(extent);
|
|
size_t psz = extent_size_quantize_floor(size);
|
|
pszind_t pind = sz_psz2ind(psz);
|
|
extent_heap_remove(&extents->heaps[pind], extent);
|
|
|
|
if (config_stats) {
|
|
extents_stats_sub(extents, pind, size);
|
|
}
|
|
|
|
if (extent_heap_empty(&extents->heaps[pind])) {
|
|
bitmap_set(extents->bitmap, &extents_bitmap_info,
|
|
(size_t)pind);
|
|
}
|
|
extent_list_remove(&extents->lru, extent);
|
|
size_t npages = size >> LG_PAGE;
|
|
/*
|
|
* As in extents_insert_locked, we hold extents->mtx and so don't need
|
|
* atomic operations for updating extents->npages.
|
|
*/
|
|
size_t cur_extents_npages =
|
|
atomic_load_zu(&extents->npages, ATOMIC_RELAXED);
|
|
assert(cur_extents_npages >= npages);
|
|
atomic_store_zu(&extents->npages,
|
|
cur_extents_npages - (size >> LG_PAGE), ATOMIC_RELAXED);
|
|
}
|
|
|
|
/*
|
|
* Find an extent with size [min_size, max_size) to satisfy the alignment
|
|
* requirement. For each size, try only the first extent in the heap.
|
|
*/
|
|
static extent_t *
|
|
extents_fit_alignment(extents_t *extents, size_t min_size, size_t max_size,
|
|
size_t alignment) {
|
|
pszind_t pind = sz_psz2ind(extent_size_quantize_ceil(min_size));
|
|
pszind_t pind_max = sz_psz2ind(extent_size_quantize_ceil(max_size));
|
|
|
|
for (pszind_t i = (pszind_t)bitmap_ffu(extents->bitmap,
|
|
&extents_bitmap_info, (size_t)pind); i < pind_max; i =
|
|
(pszind_t)bitmap_ffu(extents->bitmap, &extents_bitmap_info,
|
|
(size_t)i+1)) {
|
|
assert(i < SC_NPSIZES);
|
|
assert(!extent_heap_empty(&extents->heaps[i]));
|
|
extent_t *extent = extent_heap_first(&extents->heaps[i]);
|
|
uintptr_t base = (uintptr_t)extent_base_get(extent);
|
|
size_t candidate_size = extent_size_get(extent);
|
|
assert(candidate_size >= min_size);
|
|
|
|
uintptr_t next_align = ALIGNMENT_CEILING((uintptr_t)base,
|
|
PAGE_CEILING(alignment));
|
|
if (base > next_align || base + candidate_size <= next_align) {
|
|
/* Overflow or not crossing the next alignment. */
|
|
continue;
|
|
}
|
|
|
|
size_t leadsize = next_align - base;
|
|
if (candidate_size - leadsize >= min_size) {
|
|
return extent;
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/* Do any-best-fit extent selection, i.e. select any extent that best fits. */
|
|
static extent_t *
|
|
extents_best_fit_locked(tsdn_t *tsdn, arena_t *arena, extents_t *extents,
|
|
size_t size) {
|
|
pszind_t pind = sz_psz2ind(extent_size_quantize_ceil(size));
|
|
pszind_t i = (pszind_t)bitmap_ffu(extents->bitmap, &extents_bitmap_info,
|
|
(size_t)pind);
|
|
if (i < SC_NPSIZES + 1) {
|
|
/*
|
|
* In order to reduce fragmentation, avoid reusing and splitting
|
|
* large extents for much smaller sizes.
|
|
*/
|
|
if ((sz_pind2sz(i) >> opt_lg_extent_max_active_fit) > size) {
|
|
return NULL;
|
|
}
|
|
assert(!extent_heap_empty(&extents->heaps[i]));
|
|
extent_t *extent = extent_heap_first(&extents->heaps[i]);
|
|
assert(extent_size_get(extent) >= size);
|
|
return extent;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/*
|
|
* Do first-fit extent selection, i.e. select the oldest/lowest extent that is
|
|
* large enough.
|
|
*/
|
|
static extent_t *
|
|
extents_first_fit_locked(tsdn_t *tsdn, arena_t *arena, extents_t *extents,
|
|
size_t size) {
|
|
extent_t *ret = NULL;
|
|
|
|
pszind_t pind = sz_psz2ind(extent_size_quantize_ceil(size));
|
|
for (pszind_t i = (pszind_t)bitmap_ffu(extents->bitmap,
|
|
&extents_bitmap_info, (size_t)pind);
|
|
i < SC_NPSIZES + 1;
|
|
i = (pszind_t)bitmap_ffu(extents->bitmap, &extents_bitmap_info,
|
|
(size_t)i+1)) {
|
|
assert(!extent_heap_empty(&extents->heaps[i]));
|
|
extent_t *extent = extent_heap_first(&extents->heaps[i]);
|
|
assert(extent_size_get(extent) >= size);
|
|
if (ret == NULL || extent_snad_comp(extent, ret) < 0) {
|
|
ret = extent;
|
|
}
|
|
if (i == SC_NPSIZES) {
|
|
break;
|
|
}
|
|
assert(i < SC_NPSIZES);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/*
|
|
* Do {best,first}-fit extent selection, where the selection policy choice is
|
|
* based on extents->delay_coalesce. Best-fit selection requires less
|
|
* searching, but its layout policy is less stable and may cause higher virtual
|
|
* memory fragmentation as a side effect.
|
|
*/
|
|
static extent_t *
|
|
extents_fit_locked(tsdn_t *tsdn, arena_t *arena, extents_t *extents,
|
|
size_t esize, size_t alignment) {
|
|
malloc_mutex_assert_owner(tsdn, &extents->mtx);
|
|
|
|
size_t max_size = esize + PAGE_CEILING(alignment) - PAGE;
|
|
/* Beware size_t wrap-around. */
|
|
if (max_size < esize) {
|
|
return NULL;
|
|
}
|
|
|
|
extent_t *extent = extents->delay_coalesce ?
|
|
extents_best_fit_locked(tsdn, arena, extents, max_size) :
|
|
extents_first_fit_locked(tsdn, arena, extents, max_size);
|
|
|
|
if (alignment > PAGE && extent == NULL) {
|
|
/*
|
|
* max_size guarantees the alignment requirement but is rather
|
|
* pessimistic. Next we try to satisfy the aligned allocation
|
|
* with sizes in [esize, max_size).
|
|
*/
|
|
extent = extents_fit_alignment(extents, esize, max_size,
|
|
alignment);
|
|
}
|
|
|
|
return extent;
|
|
}
|
|
|
|
static bool
|
|
extent_try_delayed_coalesce(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, rtree_ctx_t *rtree_ctx, extents_t *extents,
|
|
extent_t *extent) {
|
|
extent_state_set(extent, extent_state_active);
|
|
bool coalesced;
|
|
extent = extent_try_coalesce(tsdn, arena, r_extent_hooks, rtree_ctx,
|
|
extents, extent, &coalesced, false);
|
|
extent_state_set(extent, extents_state_get(extents));
|
|
|
|
if (!coalesced) {
|
|
return true;
|
|
}
|
|
extents_insert_locked(tsdn, extents, extent);
|
|
return false;
|
|
}
|
|
|
|
extent_t *
|
|
extents_alloc(tsdn_t *tsdn, arena_t *arena, extent_hooks_t **r_extent_hooks,
|
|
extents_t *extents, void *new_addr, size_t size, size_t pad,
|
|
size_t alignment, bool slab, szind_t szind, bool *zero, bool *commit) {
|
|
assert(size + pad != 0);
|
|
assert(alignment != 0);
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
extent_t *extent = extent_recycle(tsdn, arena, r_extent_hooks, extents,
|
|
new_addr, size, pad, alignment, slab, szind, zero, commit, false);
|
|
assert(extent == NULL || extent_dumpable_get(extent));
|
|
return extent;
|
|
}
|
|
|
|
void
|
|
extents_dalloc(tsdn_t *tsdn, arena_t *arena, extent_hooks_t **r_extent_hooks,
|
|
extents_t *extents, extent_t *extent) {
|
|
assert(extent_base_get(extent) != NULL);
|
|
assert(extent_size_get(extent) != 0);
|
|
assert(extent_dumpable_get(extent));
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
extent_addr_set(extent, extent_base_get(extent));
|
|
extent_zeroed_set(extent, false);
|
|
|
|
extent_record(tsdn, arena, r_extent_hooks, extents, extent, false);
|
|
}
|
|
|
|
extent_t *
|
|
extents_evict(tsdn_t *tsdn, arena_t *arena, extent_hooks_t **r_extent_hooks,
|
|
extents_t *extents, size_t npages_min) {
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
|
|
malloc_mutex_lock(tsdn, &extents->mtx);
|
|
|
|
/*
|
|
* Get the LRU coalesced extent, if any. If coalescing was delayed,
|
|
* the loop will iterate until the LRU extent is fully coalesced.
|
|
*/
|
|
extent_t *extent;
|
|
while (true) {
|
|
/* Get the LRU extent, if any. */
|
|
extent = extent_list_first(&extents->lru);
|
|
if (extent == NULL) {
|
|
goto label_return;
|
|
}
|
|
/* Check the eviction limit. */
|
|
size_t extents_npages = atomic_load_zu(&extents->npages,
|
|
ATOMIC_RELAXED);
|
|
if (extents_npages <= npages_min) {
|
|
extent = NULL;
|
|
goto label_return;
|
|
}
|
|
extents_remove_locked(tsdn, extents, extent);
|
|
if (!extents->delay_coalesce) {
|
|
break;
|
|
}
|
|
/* Try to coalesce. */
|
|
if (extent_try_delayed_coalesce(tsdn, arena, r_extent_hooks,
|
|
rtree_ctx, extents, extent)) {
|
|
break;
|
|
}
|
|
/*
|
|
* The LRU extent was just coalesced and the result placed in
|
|
* the LRU at its neighbor's position. Start over.
|
|
*/
|
|
}
|
|
|
|
/*
|
|
* Either mark the extent active or deregister it to protect against
|
|
* concurrent operations.
|
|
*/
|
|
switch (extents_state_get(extents)) {
|
|
case extent_state_active:
|
|
not_reached();
|
|
case extent_state_dirty:
|
|
case extent_state_muzzy:
|
|
extent_state_set(extent, extent_state_active);
|
|
break;
|
|
case extent_state_retained:
|
|
extent_deregister(tsdn, extent);
|
|
break;
|
|
default:
|
|
not_reached();
|
|
}
|
|
|
|
label_return:
|
|
malloc_mutex_unlock(tsdn, &extents->mtx);
|
|
return extent;
|
|
}
|
|
|
|
static void
|
|
extents_leak(tsdn_t *tsdn, arena_t *arena, extent_hooks_t **r_extent_hooks,
|
|
extents_t *extents, extent_t *extent, bool growing_retained) {
|
|
/*
|
|
* Leak extent after making sure its pages have already been purged, so
|
|
* that this is only a virtual memory leak.
|
|
*/
|
|
if (extents_state_get(extents) == extent_state_dirty) {
|
|
if (extent_purge_lazy_impl(tsdn, arena, r_extent_hooks,
|
|
extent, 0, extent_size_get(extent), growing_retained)) {
|
|
extent_purge_forced_impl(tsdn, arena, r_extent_hooks,
|
|
extent, 0, extent_size_get(extent),
|
|
growing_retained);
|
|
}
|
|
}
|
|
extent_dalloc(tsdn, arena, extent);
|
|
}
|
|
|
|
void
|
|
extents_prefork(tsdn_t *tsdn, extents_t *extents) {
|
|
malloc_mutex_prefork(tsdn, &extents->mtx);
|
|
}
|
|
|
|
void
|
|
extents_postfork_parent(tsdn_t *tsdn, extents_t *extents) {
|
|
malloc_mutex_postfork_parent(tsdn, &extents->mtx);
|
|
}
|
|
|
|
void
|
|
extents_postfork_child(tsdn_t *tsdn, extents_t *extents) {
|
|
malloc_mutex_postfork_child(tsdn, &extents->mtx);
|
|
}
|
|
|
|
static void
|
|
extent_deactivate_locked(tsdn_t *tsdn, arena_t *arena, extents_t *extents,
|
|
extent_t *extent) {
|
|
assert(extent_arena_get(extent) == arena);
|
|
assert(extent_state_get(extent) == extent_state_active);
|
|
|
|
extent_state_set(extent, extents_state_get(extents));
|
|
extents_insert_locked(tsdn, extents, extent);
|
|
}
|
|
|
|
static void
|
|
extent_deactivate(tsdn_t *tsdn, arena_t *arena, extents_t *extents,
|
|
extent_t *extent) {
|
|
malloc_mutex_lock(tsdn, &extents->mtx);
|
|
extent_deactivate_locked(tsdn, arena, extents, extent);
|
|
malloc_mutex_unlock(tsdn, &extents->mtx);
|
|
}
|
|
|
|
static void
|
|
extent_activate_locked(tsdn_t *tsdn, arena_t *arena, extents_t *extents,
|
|
extent_t *extent) {
|
|
assert(extent_arena_get(extent) == arena);
|
|
assert(extent_state_get(extent) == extents_state_get(extents));
|
|
|
|
extents_remove_locked(tsdn, extents, extent);
|
|
extent_state_set(extent, extent_state_active);
|
|
}
|
|
|
|
static bool
|
|
extent_rtree_leaf_elms_lookup(tsdn_t *tsdn, rtree_ctx_t *rtree_ctx,
|
|
const extent_t *extent, bool dependent, bool init_missing,
|
|
rtree_leaf_elm_t **r_elm_a, rtree_leaf_elm_t **r_elm_b) {
|
|
*r_elm_a = rtree_leaf_elm_lookup(tsdn, &extents_rtree, rtree_ctx,
|
|
(uintptr_t)extent_base_get(extent), dependent, init_missing);
|
|
if (!dependent && *r_elm_a == NULL) {
|
|
return true;
|
|
}
|
|
assert(*r_elm_a != NULL);
|
|
|
|
*r_elm_b = rtree_leaf_elm_lookup(tsdn, &extents_rtree, rtree_ctx,
|
|
(uintptr_t)extent_last_get(extent), dependent, init_missing);
|
|
if (!dependent && *r_elm_b == NULL) {
|
|
return true;
|
|
}
|
|
assert(*r_elm_b != NULL);
|
|
|
|
return false;
|
|
}
|
|
|
|
static void
|
|
extent_rtree_write_acquired(tsdn_t *tsdn, rtree_leaf_elm_t *elm_a,
|
|
rtree_leaf_elm_t *elm_b, extent_t *extent, szind_t szind, bool slab) {
|
|
rtree_leaf_elm_write(tsdn, &extents_rtree, elm_a, extent, szind, slab);
|
|
if (elm_b != NULL) {
|
|
rtree_leaf_elm_write(tsdn, &extents_rtree, elm_b, extent, szind,
|
|
slab);
|
|
}
|
|
}
|
|
|
|
static void
|
|
extent_interior_register(tsdn_t *tsdn, rtree_ctx_t *rtree_ctx, extent_t *extent,
|
|
szind_t szind) {
|
|
assert(extent_slab_get(extent));
|
|
|
|
/* Register interior. */
|
|
for (size_t i = 1; i < (extent_size_get(extent) >> LG_PAGE) - 1; i++) {
|
|
rtree_write(tsdn, &extents_rtree, rtree_ctx,
|
|
(uintptr_t)extent_base_get(extent) + (uintptr_t)(i <<
|
|
LG_PAGE), extent, szind, true);
|
|
}
|
|
}
|
|
|
|
static void
|
|
extent_gdump_add(tsdn_t *tsdn, const extent_t *extent) {
|
|
cassert(config_prof);
|
|
/* prof_gdump() requirement. */
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
if (opt_prof && extent_state_get(extent) == extent_state_active) {
|
|
size_t nadd = extent_size_get(extent) >> LG_PAGE;
|
|
size_t cur = atomic_fetch_add_zu(&curpages, nadd,
|
|
ATOMIC_RELAXED) + nadd;
|
|
size_t high = atomic_load_zu(&highpages, ATOMIC_RELAXED);
|
|
while (cur > high && !atomic_compare_exchange_weak_zu(
|
|
&highpages, &high, cur, ATOMIC_RELAXED, ATOMIC_RELAXED)) {
|
|
/*
|
|
* Don't refresh cur, because it may have decreased
|
|
* since this thread lost the highpages update race.
|
|
* Note that high is updated in case of CAS failure.
|
|
*/
|
|
}
|
|
if (cur > high && prof_gdump_get_unlocked()) {
|
|
prof_gdump(tsdn);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void
|
|
extent_gdump_sub(tsdn_t *tsdn, const extent_t *extent) {
|
|
cassert(config_prof);
|
|
|
|
if (opt_prof && extent_state_get(extent) == extent_state_active) {
|
|
size_t nsub = extent_size_get(extent) >> LG_PAGE;
|
|
assert(atomic_load_zu(&curpages, ATOMIC_RELAXED) >= nsub);
|
|
atomic_fetch_sub_zu(&curpages, nsub, ATOMIC_RELAXED);
|
|
}
|
|
}
|
|
|
|
static bool
|
|
extent_register_impl(tsdn_t *tsdn, extent_t *extent, bool gdump_add) {
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
rtree_leaf_elm_t *elm_a, *elm_b;
|
|
|
|
/*
|
|
* We need to hold the lock to protect against a concurrent coalesce
|
|
* operation that sees us in a partial state.
|
|
*/
|
|
extent_lock(tsdn, extent);
|
|
|
|
if (extent_rtree_leaf_elms_lookup(tsdn, rtree_ctx, extent, false, true,
|
|
&elm_a, &elm_b)) {
|
|
return true;
|
|
}
|
|
|
|
szind_t szind = extent_szind_get_maybe_invalid(extent);
|
|
bool slab = extent_slab_get(extent);
|
|
extent_rtree_write_acquired(tsdn, elm_a, elm_b, extent, szind, slab);
|
|
if (slab) {
|
|
extent_interior_register(tsdn, rtree_ctx, extent, szind);
|
|
}
|
|
|
|
extent_unlock(tsdn, extent);
|
|
|
|
if (config_prof && gdump_add) {
|
|
extent_gdump_add(tsdn, extent);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
static bool
|
|
extent_register(tsdn_t *tsdn, extent_t *extent) {
|
|
return extent_register_impl(tsdn, extent, true);
|
|
}
|
|
|
|
static bool
|
|
extent_register_no_gdump_add(tsdn_t *tsdn, extent_t *extent) {
|
|
return extent_register_impl(tsdn, extent, false);
|
|
}
|
|
|
|
static void
|
|
extent_reregister(tsdn_t *tsdn, extent_t *extent) {
|
|
bool err = extent_register(tsdn, extent);
|
|
assert(!err);
|
|
}
|
|
|
|
/*
|
|
* Removes all pointers to the given extent from the global rtree indices for
|
|
* its interior. This is relevant for slab extents, for which we need to do
|
|
* metadata lookups at places other than the head of the extent. We deregister
|
|
* on the interior, then, when an extent moves from being an active slab to an
|
|
* inactive state.
|
|
*/
|
|
static void
|
|
extent_interior_deregister(tsdn_t *tsdn, rtree_ctx_t *rtree_ctx,
|
|
extent_t *extent) {
|
|
size_t i;
|
|
|
|
assert(extent_slab_get(extent));
|
|
|
|
for (i = 1; i < (extent_size_get(extent) >> LG_PAGE) - 1; i++) {
|
|
rtree_clear(tsdn, &extents_rtree, rtree_ctx,
|
|
(uintptr_t)extent_base_get(extent) + (uintptr_t)(i <<
|
|
LG_PAGE));
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Removes all pointers to the given extent from the global rtree.
|
|
*/
|
|
static void
|
|
extent_deregister_impl(tsdn_t *tsdn, extent_t *extent, bool gdump) {
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
rtree_leaf_elm_t *elm_a, *elm_b;
|
|
extent_rtree_leaf_elms_lookup(tsdn, rtree_ctx, extent, true, false,
|
|
&elm_a, &elm_b);
|
|
|
|
extent_lock(tsdn, extent);
|
|
|
|
extent_rtree_write_acquired(tsdn, elm_a, elm_b, NULL, SC_NSIZES, false);
|
|
if (extent_slab_get(extent)) {
|
|
extent_interior_deregister(tsdn, rtree_ctx, extent);
|
|
extent_slab_set(extent, false);
|
|
}
|
|
|
|
extent_unlock(tsdn, extent);
|
|
|
|
if (config_prof && gdump) {
|
|
extent_gdump_sub(tsdn, extent);
|
|
}
|
|
}
|
|
|
|
static void
|
|
extent_deregister(tsdn_t *tsdn, extent_t *extent) {
|
|
extent_deregister_impl(tsdn, extent, true);
|
|
}
|
|
|
|
static void
|
|
extent_deregister_no_gdump_sub(tsdn_t *tsdn, extent_t *extent) {
|
|
extent_deregister_impl(tsdn, extent, false);
|
|
}
|
|
|
|
/*
|
|
* Tries to find and remove an extent from extents that can be used for the
|
|
* given allocation request.
|
|
*/
|
|
static extent_t *
|
|
extent_recycle_extract(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, rtree_ctx_t *rtree_ctx, extents_t *extents,
|
|
void *new_addr, size_t size, size_t pad, size_t alignment, bool slab,
|
|
bool growing_retained) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, growing_retained ? 1 : 0);
|
|
assert(alignment > 0);
|
|
if (config_debug && new_addr != NULL) {
|
|
/*
|
|
* Non-NULL new_addr has two use cases:
|
|
*
|
|
* 1) Recycle a known-extant extent, e.g. during purging.
|
|
* 2) Perform in-place expanding reallocation.
|
|
*
|
|
* Regardless of use case, new_addr must either refer to a
|
|
* non-existing extent, or to the base of an extant extent,
|
|
* since only active slabs support interior lookups (which of
|
|
* course cannot be recycled).
|
|
*/
|
|
assert(PAGE_ADDR2BASE(new_addr) == new_addr);
|
|
assert(pad == 0);
|
|
assert(alignment <= PAGE);
|
|
}
|
|
|
|
size_t esize = size + pad;
|
|
malloc_mutex_lock(tsdn, &extents->mtx);
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
extent_t *extent;
|
|
if (new_addr != NULL) {
|
|
extent = extent_lock_from_addr(tsdn, rtree_ctx, new_addr);
|
|
if (extent != NULL) {
|
|
/*
|
|
* We might null-out extent to report an error, but we
|
|
* still need to unlock the associated mutex after.
|
|
*/
|
|
extent_t *unlock_extent = extent;
|
|
assert(extent_base_get(extent) == new_addr);
|
|
if (extent_arena_get(extent) != arena ||
|
|
extent_size_get(extent) < esize ||
|
|
extent_state_get(extent) !=
|
|
extents_state_get(extents)) {
|
|
extent = NULL;
|
|
}
|
|
extent_unlock(tsdn, unlock_extent);
|
|
}
|
|
} else {
|
|
extent = extents_fit_locked(tsdn, arena, extents, esize,
|
|
alignment);
|
|
}
|
|
if (extent == NULL) {
|
|
malloc_mutex_unlock(tsdn, &extents->mtx);
|
|
return NULL;
|
|
}
|
|
|
|
extent_activate_locked(tsdn, arena, extents, extent);
|
|
malloc_mutex_unlock(tsdn, &extents->mtx);
|
|
|
|
return extent;
|
|
}
|
|
|
|
/*
|
|
* Given an allocation request and an extent guaranteed to be able to satisfy
|
|
* it, this splits off lead and trail extents, leaving extent pointing to an
|
|
* extent satisfying the allocation.
|
|
* This function doesn't put lead or trail into any extents_t; it's the caller's
|
|
* job to ensure that they can be reused.
|
|
*/
|
|
typedef enum {
|
|
/*
|
|
* Split successfully. lead, extent, and trail, are modified to extents
|
|
* describing the ranges before, in, and after the given allocation.
|
|
*/
|
|
extent_split_interior_ok,
|
|
/*
|
|
* The extent can't satisfy the given allocation request. None of the
|
|
* input extent_t *s are touched.
|
|
*/
|
|
extent_split_interior_cant_alloc,
|
|
/*
|
|
* In a potentially invalid state. Must leak (if *to_leak is non-NULL),
|
|
* and salvage what's still salvageable (if *to_salvage is non-NULL).
|
|
* None of lead, extent, or trail are valid.
|
|
*/
|
|
extent_split_interior_error
|
|
} extent_split_interior_result_t;
|
|
|
|
static extent_split_interior_result_t
|
|
extent_split_interior(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, rtree_ctx_t *rtree_ctx,
|
|
/* The result of splitting, in case of success. */
|
|
extent_t **extent, extent_t **lead, extent_t **trail,
|
|
/* The mess to clean up, in case of error. */
|
|
extent_t **to_leak, extent_t **to_salvage,
|
|
void *new_addr, size_t size, size_t pad, size_t alignment, bool slab,
|
|
szind_t szind, bool growing_retained) {
|
|
size_t esize = size + pad;
|
|
size_t leadsize = ALIGNMENT_CEILING((uintptr_t)extent_base_get(*extent),
|
|
PAGE_CEILING(alignment)) - (uintptr_t)extent_base_get(*extent);
|
|
assert(new_addr == NULL || leadsize == 0);
|
|
if (extent_size_get(*extent) < leadsize + esize) {
|
|
return extent_split_interior_cant_alloc;
|
|
}
|
|
size_t trailsize = extent_size_get(*extent) - leadsize - esize;
|
|
|
|
*lead = NULL;
|
|
*trail = NULL;
|
|
*to_leak = NULL;
|
|
*to_salvage = NULL;
|
|
|
|
/* Split the lead. */
|
|
if (leadsize != 0) {
|
|
*lead = *extent;
|
|
*extent = extent_split_impl(tsdn, arena, r_extent_hooks,
|
|
*lead, leadsize, SC_NSIZES, false, esize + trailsize, szind,
|
|
slab, growing_retained);
|
|
if (*extent == NULL) {
|
|
*to_leak = *lead;
|
|
*lead = NULL;
|
|
return extent_split_interior_error;
|
|
}
|
|
}
|
|
|
|
/* Split the trail. */
|
|
if (trailsize != 0) {
|
|
*trail = extent_split_impl(tsdn, arena, r_extent_hooks, *extent,
|
|
esize, szind, slab, trailsize, SC_NSIZES, false,
|
|
growing_retained);
|
|
if (*trail == NULL) {
|
|
*to_leak = *extent;
|
|
*to_salvage = *lead;
|
|
*lead = NULL;
|
|
*extent = NULL;
|
|
return extent_split_interior_error;
|
|
}
|
|
}
|
|
|
|
if (leadsize == 0 && trailsize == 0) {
|
|
/*
|
|
* Splitting causes szind to be set as a side effect, but no
|
|
* splitting occurred.
|
|
*/
|
|
extent_szind_set(*extent, szind);
|
|
if (szind != SC_NSIZES) {
|
|
rtree_szind_slab_update(tsdn, &extents_rtree, rtree_ctx,
|
|
(uintptr_t)extent_addr_get(*extent), szind, slab);
|
|
if (slab && extent_size_get(*extent) > PAGE) {
|
|
rtree_szind_slab_update(tsdn, &extents_rtree,
|
|
rtree_ctx,
|
|
(uintptr_t)extent_past_get(*extent) -
|
|
(uintptr_t)PAGE, szind, slab);
|
|
}
|
|
}
|
|
}
|
|
|
|
return extent_split_interior_ok;
|
|
}
|
|
|
|
/*
|
|
* This fulfills the indicated allocation request out of the given extent (which
|
|
* the caller should have ensured was big enough). If there's any unused space
|
|
* before or after the resulting allocation, that space is given its own extent
|
|
* and put back into extents.
|
|
*/
|
|
static extent_t *
|
|
extent_recycle_split(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, rtree_ctx_t *rtree_ctx, extents_t *extents,
|
|
void *new_addr, size_t size, size_t pad, size_t alignment, bool slab,
|
|
szind_t szind, extent_t *extent, bool growing_retained) {
|
|
extent_t *lead;
|
|
extent_t *trail;
|
|
extent_t *to_leak;
|
|
extent_t *to_salvage;
|
|
|
|
extent_split_interior_result_t result = extent_split_interior(
|
|
tsdn, arena, r_extent_hooks, rtree_ctx, &extent, &lead, &trail,
|
|
&to_leak, &to_salvage, new_addr, size, pad, alignment, slab, szind,
|
|
growing_retained);
|
|
|
|
if (result == extent_split_interior_ok) {
|
|
if (lead != NULL) {
|
|
extent_deactivate(tsdn, arena, extents, lead);
|
|
}
|
|
if (trail != NULL) {
|
|
extent_deactivate(tsdn, arena, extents, trail);
|
|
}
|
|
return extent;
|
|
} else {
|
|
/*
|
|
* We should have picked an extent that was large enough to
|
|
* fulfill our allocation request.
|
|
*/
|
|
assert(result == extent_split_interior_error);
|
|
if (to_salvage != NULL) {
|
|
extent_deregister(tsdn, to_salvage);
|
|
}
|
|
if (to_leak != NULL) {
|
|
void *leak = extent_base_get(to_leak);
|
|
extent_deregister_no_gdump_sub(tsdn, to_leak);
|
|
extents_leak(tsdn, arena, r_extent_hooks, extents,
|
|
to_leak, growing_retained);
|
|
assert(extent_lock_from_addr(tsdn, rtree_ctx, leak)
|
|
== NULL);
|
|
}
|
|
return NULL;
|
|
}
|
|
unreachable();
|
|
}
|
|
|
|
/*
|
|
* Tries to satisfy the given allocation request by reusing one of the extents
|
|
* in the given extents_t.
|
|
*/
|
|
static extent_t *
|
|
extent_recycle(tsdn_t *tsdn, arena_t *arena, extent_hooks_t **r_extent_hooks,
|
|
extents_t *extents, void *new_addr, size_t size, size_t pad,
|
|
size_t alignment, bool slab, szind_t szind, bool *zero, bool *commit,
|
|
bool growing_retained) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, growing_retained ? 1 : 0);
|
|
assert(new_addr == NULL || !slab);
|
|
assert(pad == 0 || !slab);
|
|
assert(!*zero || !slab);
|
|
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
|
|
extent_t *extent = extent_recycle_extract(tsdn, arena, r_extent_hooks,
|
|
rtree_ctx, extents, new_addr, size, pad, alignment, slab,
|
|
growing_retained);
|
|
if (extent == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
extent = extent_recycle_split(tsdn, arena, r_extent_hooks, rtree_ctx,
|
|
extents, new_addr, size, pad, alignment, slab, szind, extent,
|
|
growing_retained);
|
|
if (extent == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (*commit && !extent_committed_get(extent)) {
|
|
if (extent_commit_impl(tsdn, arena, r_extent_hooks, extent,
|
|
0, extent_size_get(extent), growing_retained)) {
|
|
extent_record(tsdn, arena, r_extent_hooks, extents,
|
|
extent, growing_retained);
|
|
return NULL;
|
|
}
|
|
extent_zeroed_set(extent, true);
|
|
}
|
|
|
|
if (extent_committed_get(extent)) {
|
|
*commit = true;
|
|
}
|
|
if (extent_zeroed_get(extent)) {
|
|
*zero = true;
|
|
}
|
|
|
|
if (pad != 0) {
|
|
extent_addr_randomize(tsdn, extent, alignment);
|
|
}
|
|
assert(extent_state_get(extent) == extent_state_active);
|
|
if (slab) {
|
|
extent_slab_set(extent, slab);
|
|
extent_interior_register(tsdn, rtree_ctx, extent, szind);
|
|
}
|
|
|
|
if (*zero) {
|
|
void *addr = extent_base_get(extent);
|
|
size_t size = extent_size_get(extent);
|
|
if (!extent_zeroed_get(extent)) {
|
|
if (pages_purge_forced(addr, size)) {
|
|
memset(addr, 0, size);
|
|
}
|
|
} else if (config_debug) {
|
|
size_t *p = (size_t *)(uintptr_t)addr;
|
|
for (size_t i = 0; i < size / sizeof(size_t); i++) {
|
|
assert(p[i] == 0);
|
|
}
|
|
}
|
|
}
|
|
return extent;
|
|
}
|
|
|
|
/*
|
|
* If the caller specifies (!*zero), it is still possible to receive zeroed
|
|
* memory, in which case *zero is toggled to true. arena_extent_alloc() takes
|
|
* advantage of this to avoid demanding zeroed extents, but taking advantage of
|
|
* them if they are returned.
|
|
*/
|
|
static void *
|
|
extent_alloc_core(tsdn_t *tsdn, arena_t *arena, void *new_addr, size_t size,
|
|
size_t alignment, bool *zero, bool *commit, dss_prec_t dss_prec) {
|
|
void *ret;
|
|
|
|
assert(size != 0);
|
|
assert(alignment != 0);
|
|
|
|
/* "primary" dss. */
|
|
if (have_dss && dss_prec == dss_prec_primary && (ret =
|
|
extent_alloc_dss(tsdn, arena, new_addr, size, alignment, zero,
|
|
commit)) != NULL) {
|
|
return ret;
|
|
}
|
|
/* mmap. */
|
|
if ((ret = extent_alloc_mmap(new_addr, size, alignment, zero, commit))
|
|
!= NULL) {
|
|
return ret;
|
|
}
|
|
/* "secondary" dss. */
|
|
if (have_dss && dss_prec == dss_prec_secondary && (ret =
|
|
extent_alloc_dss(tsdn, arena, new_addr, size, alignment, zero,
|
|
commit)) != NULL) {
|
|
return ret;
|
|
}
|
|
|
|
/* All strategies for allocation failed. */
|
|
return NULL;
|
|
}
|
|
|
|
static void *
|
|
extent_alloc_default_impl(tsdn_t *tsdn, arena_t *arena, void *new_addr,
|
|
size_t size, size_t alignment, bool *zero, bool *commit) {
|
|
void *ret = extent_alloc_core(tsdn, arena, new_addr, size, alignment, zero,
|
|
commit, (dss_prec_t)atomic_load_u(&arena->dss_prec,
|
|
ATOMIC_RELAXED));
|
|
if (have_madvise_huge && ret) {
|
|
pages_set_thp_state(ret, size);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static void *
|
|
extent_alloc_default(extent_hooks_t *extent_hooks, void *new_addr, size_t size,
|
|
size_t alignment, bool *zero, bool *commit, unsigned arena_ind) {
|
|
tsdn_t *tsdn;
|
|
arena_t *arena;
|
|
|
|
tsdn = tsdn_fetch();
|
|
arena = arena_get(tsdn, arena_ind, false);
|
|
/*
|
|
* The arena we're allocating on behalf of must have been initialized
|
|
* already.
|
|
*/
|
|
assert(arena != NULL);
|
|
|
|
return extent_alloc_default_impl(tsdn, arena, new_addr, size,
|
|
alignment, zero, commit);
|
|
}
|
|
|
|
static void
|
|
extent_hook_pre_reentrancy(tsdn_t *tsdn, arena_t *arena) {
|
|
tsd_t *tsd = tsdn_null(tsdn) ? tsd_fetch() : tsdn_tsd(tsdn);
|
|
if (arena == arena_get(tsd_tsdn(tsd), 0, false)) {
|
|
/*
|
|
* The only legitimate case of customized extent hooks for a0 is
|
|
* hooks with no allocation activities. One such example is to
|
|
* place metadata on pre-allocated resources such as huge pages.
|
|
* In that case, rely on reentrancy_level checks to catch
|
|
* infinite recursions.
|
|
*/
|
|
pre_reentrancy(tsd, NULL);
|
|
} else {
|
|
pre_reentrancy(tsd, arena);
|
|
}
|
|
}
|
|
|
|
static void
|
|
extent_hook_post_reentrancy(tsdn_t *tsdn) {
|
|
tsd_t *tsd = tsdn_null(tsdn) ? tsd_fetch() : tsdn_tsd(tsdn);
|
|
post_reentrancy(tsd);
|
|
}
|
|
|
|
/*
|
|
* If virtual memory is retained, create increasingly larger extents from which
|
|
* to split requested extents in order to limit the total number of disjoint
|
|
* virtual memory ranges retained by each arena.
|
|
*/
|
|
static extent_t *
|
|
extent_grow_retained(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, size_t size, size_t pad, size_t alignment,
|
|
bool slab, szind_t szind, bool *zero, bool *commit) {
|
|
malloc_mutex_assert_owner(tsdn, &arena->extent_grow_mtx);
|
|
assert(pad == 0 || !slab);
|
|
assert(!*zero || !slab);
|
|
|
|
size_t esize = size + pad;
|
|
size_t alloc_size_min = esize + PAGE_CEILING(alignment) - PAGE;
|
|
/* Beware size_t wrap-around. */
|
|
if (alloc_size_min < esize) {
|
|
goto label_err;
|
|
}
|
|
/*
|
|
* Find the next extent size in the series that would be large enough to
|
|
* satisfy this request.
|
|
*/
|
|
pszind_t egn_skip = 0;
|
|
size_t alloc_size = sz_pind2sz(arena->extent_grow_next + egn_skip);
|
|
while (alloc_size < alloc_size_min) {
|
|
egn_skip++;
|
|
if (arena->extent_grow_next + egn_skip >=
|
|
sz_psz2ind(SC_LARGE_MAXCLASS)) {
|
|
/* Outside legal range. */
|
|
goto label_err;
|
|
}
|
|
alloc_size = sz_pind2sz(arena->extent_grow_next + egn_skip);
|
|
}
|
|
|
|
extent_t *extent = extent_alloc(tsdn, arena);
|
|
if (extent == NULL) {
|
|
goto label_err;
|
|
}
|
|
bool zeroed = false;
|
|
bool committed = false;
|
|
|
|
void *ptr;
|
|
if (*r_extent_hooks == &extent_hooks_default) {
|
|
ptr = extent_alloc_default_impl(tsdn, arena, NULL,
|
|
alloc_size, PAGE, &zeroed, &committed);
|
|
} else {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
ptr = (*r_extent_hooks)->alloc(*r_extent_hooks, NULL,
|
|
alloc_size, PAGE, &zeroed, &committed,
|
|
arena_ind_get(arena));
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
|
|
extent_init(extent, arena, ptr, alloc_size, false, SC_NSIZES,
|
|
arena_extent_sn_next(arena), extent_state_active, zeroed,
|
|
committed, true);
|
|
if (ptr == NULL) {
|
|
extent_dalloc(tsdn, arena, extent);
|
|
goto label_err;
|
|
}
|
|
|
|
if (extent_register_no_gdump_add(tsdn, extent)) {
|
|
extents_leak(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, extent, true);
|
|
goto label_err;
|
|
}
|
|
|
|
if (extent_zeroed_get(extent) && extent_committed_get(extent)) {
|
|
*zero = true;
|
|
}
|
|
if (extent_committed_get(extent)) {
|
|
*commit = true;
|
|
}
|
|
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
|
|
extent_t *lead;
|
|
extent_t *trail;
|
|
extent_t *to_leak;
|
|
extent_t *to_salvage;
|
|
extent_split_interior_result_t result = extent_split_interior(
|
|
tsdn, arena, r_extent_hooks, rtree_ctx, &extent, &lead, &trail,
|
|
&to_leak, &to_salvage, NULL, size, pad, alignment, slab, szind,
|
|
true);
|
|
|
|
if (result == extent_split_interior_ok) {
|
|
if (lead != NULL) {
|
|
extent_record(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, lead, true);
|
|
}
|
|
if (trail != NULL) {
|
|
extent_record(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, trail, true);
|
|
}
|
|
} else {
|
|
/*
|
|
* We should have allocated a sufficiently large extent; the
|
|
* cant_alloc case should not occur.
|
|
*/
|
|
assert(result == extent_split_interior_error);
|
|
if (to_salvage != NULL) {
|
|
if (config_prof) {
|
|
extent_gdump_add(tsdn, to_salvage);
|
|
}
|
|
extent_record(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, to_salvage, true);
|
|
}
|
|
if (to_leak != NULL) {
|
|
extent_deregister_no_gdump_sub(tsdn, to_leak);
|
|
extents_leak(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, to_leak, true);
|
|
}
|
|
goto label_err;
|
|
}
|
|
|
|
if (*commit && !extent_committed_get(extent)) {
|
|
if (extent_commit_impl(tsdn, arena, r_extent_hooks, extent, 0,
|
|
extent_size_get(extent), true)) {
|
|
extent_record(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, extent, true);
|
|
goto label_err;
|
|
}
|
|
extent_zeroed_set(extent, true);
|
|
}
|
|
|
|
/*
|
|
* Increment extent_grow_next if doing so wouldn't exceed the allowed
|
|
* range.
|
|
*/
|
|
if (arena->extent_grow_next + egn_skip + 1 <=
|
|
arena->retain_grow_limit) {
|
|
arena->extent_grow_next += egn_skip + 1;
|
|
} else {
|
|
arena->extent_grow_next = arena->retain_grow_limit;
|
|
}
|
|
/* All opportunities for failure are past. */
|
|
malloc_mutex_unlock(tsdn, &arena->extent_grow_mtx);
|
|
|
|
if (config_prof) {
|
|
/* Adjust gdump stats now that extent is final size. */
|
|
extent_gdump_add(tsdn, extent);
|
|
}
|
|
if (pad != 0) {
|
|
extent_addr_randomize(tsdn, extent, alignment);
|
|
}
|
|
if (slab) {
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn,
|
|
&rtree_ctx_fallback);
|
|
|
|
extent_slab_set(extent, true);
|
|
extent_interior_register(tsdn, rtree_ctx, extent, szind);
|
|
}
|
|
if (*zero && !extent_zeroed_get(extent)) {
|
|
void *addr = extent_base_get(extent);
|
|
size_t size = extent_size_get(extent);
|
|
if (pages_purge_forced(addr, size)) {
|
|
memset(addr, 0, size);
|
|
}
|
|
}
|
|
|
|
return extent;
|
|
label_err:
|
|
malloc_mutex_unlock(tsdn, &arena->extent_grow_mtx);
|
|
return NULL;
|
|
}
|
|
|
|
static extent_t *
|
|
extent_alloc_retained(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, void *new_addr, size_t size, size_t pad,
|
|
size_t alignment, bool slab, szind_t szind, bool *zero, bool *commit) {
|
|
assert(size != 0);
|
|
assert(alignment != 0);
|
|
|
|
malloc_mutex_lock(tsdn, &arena->extent_grow_mtx);
|
|
|
|
extent_t *extent = extent_recycle(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, new_addr, size, pad, alignment, slab,
|
|
szind, zero, commit, true);
|
|
if (extent != NULL) {
|
|
malloc_mutex_unlock(tsdn, &arena->extent_grow_mtx);
|
|
if (config_prof) {
|
|
extent_gdump_add(tsdn, extent);
|
|
}
|
|
} else if (opt_retain && new_addr == NULL) {
|
|
extent = extent_grow_retained(tsdn, arena, r_extent_hooks, size,
|
|
pad, alignment, slab, szind, zero, commit);
|
|
/* extent_grow_retained() always releases extent_grow_mtx. */
|
|
} else {
|
|
malloc_mutex_unlock(tsdn, &arena->extent_grow_mtx);
|
|
}
|
|
malloc_mutex_assert_not_owner(tsdn, &arena->extent_grow_mtx);
|
|
|
|
return extent;
|
|
}
|
|
|
|
static extent_t *
|
|
extent_alloc_wrapper_hard(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, void *new_addr, size_t size, size_t pad,
|
|
size_t alignment, bool slab, szind_t szind, bool *zero, bool *commit) {
|
|
size_t esize = size + pad;
|
|
extent_t *extent = extent_alloc(tsdn, arena);
|
|
if (extent == NULL) {
|
|
return NULL;
|
|
}
|
|
void *addr;
|
|
if (*r_extent_hooks == &extent_hooks_default) {
|
|
/* Call directly to propagate tsdn. */
|
|
addr = extent_alloc_default_impl(tsdn, arena, new_addr, esize,
|
|
alignment, zero, commit);
|
|
} else {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
addr = (*r_extent_hooks)->alloc(*r_extent_hooks, new_addr,
|
|
esize, alignment, zero, commit, arena_ind_get(arena));
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
if (addr == NULL) {
|
|
extent_dalloc(tsdn, arena, extent);
|
|
return NULL;
|
|
}
|
|
extent_init(extent, arena, addr, esize, slab, szind,
|
|
arena_extent_sn_next(arena), extent_state_active, *zero, *commit,
|
|
true);
|
|
if (pad != 0) {
|
|
extent_addr_randomize(tsdn, extent, alignment);
|
|
}
|
|
if (extent_register(tsdn, extent)) {
|
|
extents_leak(tsdn, arena, r_extent_hooks,
|
|
&arena->extents_retained, extent, false);
|
|
return NULL;
|
|
}
|
|
|
|
return extent;
|
|
}
|
|
|
|
extent_t *
|
|
extent_alloc_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, void *new_addr, size_t size, size_t pad,
|
|
size_t alignment, bool slab, szind_t szind, bool *zero, bool *commit) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
|
|
extent_t *extent = extent_alloc_retained(tsdn, arena, r_extent_hooks,
|
|
new_addr, size, pad, alignment, slab, szind, zero, commit);
|
|
if (extent == NULL) {
|
|
if (opt_retain && new_addr != NULL) {
|
|
/*
|
|
* When retain is enabled and new_addr is set, we do not
|
|
* attempt extent_alloc_wrapper_hard which does mmap
|
|
* that is very unlikely to succeed (unless it happens
|
|
* to be at the end).
|
|
*/
|
|
return NULL;
|
|
}
|
|
extent = extent_alloc_wrapper_hard(tsdn, arena, r_extent_hooks,
|
|
new_addr, size, pad, alignment, slab, szind, zero, commit);
|
|
}
|
|
|
|
assert(extent == NULL || extent_dumpable_get(extent));
|
|
return extent;
|
|
}
|
|
|
|
static bool
|
|
extent_can_coalesce(arena_t *arena, extents_t *extents, const extent_t *inner,
|
|
const extent_t *outer) {
|
|
assert(extent_arena_get(inner) == arena);
|
|
if (extent_arena_get(outer) != arena) {
|
|
return false;
|
|
}
|
|
|
|
assert(extent_state_get(inner) == extent_state_active);
|
|
if (extent_state_get(outer) != extents->state) {
|
|
return false;
|
|
}
|
|
|
|
if (extent_committed_get(inner) != extent_committed_get(outer)) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool
|
|
extent_coalesce(tsdn_t *tsdn, arena_t *arena, extent_hooks_t **r_extent_hooks,
|
|
extents_t *extents, extent_t *inner, extent_t *outer, bool forward,
|
|
bool growing_retained) {
|
|
assert(extent_can_coalesce(arena, extents, inner, outer));
|
|
|
|
extent_activate_locked(tsdn, arena, extents, outer);
|
|
|
|
malloc_mutex_unlock(tsdn, &extents->mtx);
|
|
bool err = extent_merge_impl(tsdn, arena, r_extent_hooks,
|
|
forward ? inner : outer, forward ? outer : inner, growing_retained);
|
|
malloc_mutex_lock(tsdn, &extents->mtx);
|
|
|
|
if (err) {
|
|
extent_deactivate_locked(tsdn, arena, extents, outer);
|
|
}
|
|
|
|
return err;
|
|
}
|
|
|
|
static extent_t *
|
|
extent_try_coalesce(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, rtree_ctx_t *rtree_ctx, extents_t *extents,
|
|
extent_t *extent, bool *coalesced, bool growing_retained) {
|
|
/*
|
|
* Continue attempting to coalesce until failure, to protect against
|
|
* races with other threads that are thwarted by this one.
|
|
*/
|
|
bool again;
|
|
do {
|
|
again = false;
|
|
|
|
/* Try to coalesce forward. */
|
|
extent_t *next = extent_lock_from_addr(tsdn, rtree_ctx,
|
|
extent_past_get(extent));
|
|
if (next != NULL) {
|
|
/*
|
|
* extents->mtx only protects against races for
|
|
* like-state extents, so call extent_can_coalesce()
|
|
* before releasing next's pool lock.
|
|
*/
|
|
bool can_coalesce = extent_can_coalesce(arena, extents,
|
|
extent, next);
|
|
|
|
extent_unlock(tsdn, next);
|
|
|
|
if (can_coalesce && !extent_coalesce(tsdn, arena,
|
|
r_extent_hooks, extents, extent, next, true,
|
|
growing_retained)) {
|
|
if (extents->delay_coalesce) {
|
|
/* Do minimal coalescing. */
|
|
*coalesced = true;
|
|
return extent;
|
|
}
|
|
again = true;
|
|
}
|
|
}
|
|
|
|
/* Try to coalesce backward. */
|
|
extent_t *prev = extent_lock_from_addr(tsdn, rtree_ctx,
|
|
extent_before_get(extent));
|
|
if (prev != NULL) {
|
|
bool can_coalesce = extent_can_coalesce(arena, extents,
|
|
extent, prev);
|
|
extent_unlock(tsdn, prev);
|
|
|
|
if (can_coalesce && !extent_coalesce(tsdn, arena,
|
|
r_extent_hooks, extents, extent, prev, false,
|
|
growing_retained)) {
|
|
extent = prev;
|
|
if (extents->delay_coalesce) {
|
|
/* Do minimal coalescing. */
|
|
*coalesced = true;
|
|
return extent;
|
|
}
|
|
again = true;
|
|
}
|
|
}
|
|
} while (again);
|
|
|
|
if (extents->delay_coalesce) {
|
|
*coalesced = false;
|
|
}
|
|
return extent;
|
|
}
|
|
|
|
/*
|
|
* Does the metadata management portions of putting an unused extent into the
|
|
* given extents_t (coalesces, deregisters slab interiors, the heap operations).
|
|
*/
|
|
static void
|
|
extent_record(tsdn_t *tsdn, arena_t *arena, extent_hooks_t **r_extent_hooks,
|
|
extents_t *extents, extent_t *extent, bool growing_retained) {
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
|
|
assert((extents_state_get(extents) != extent_state_dirty &&
|
|
extents_state_get(extents) != extent_state_muzzy) ||
|
|
!extent_zeroed_get(extent));
|
|
|
|
malloc_mutex_lock(tsdn, &extents->mtx);
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
|
|
extent_szind_set(extent, SC_NSIZES);
|
|
if (extent_slab_get(extent)) {
|
|
extent_interior_deregister(tsdn, rtree_ctx, extent);
|
|
extent_slab_set(extent, false);
|
|
}
|
|
|
|
assert(rtree_extent_read(tsdn, &extents_rtree, rtree_ctx,
|
|
(uintptr_t)extent_base_get(extent), true) == extent);
|
|
|
|
if (!extents->delay_coalesce) {
|
|
extent = extent_try_coalesce(tsdn, arena, r_extent_hooks,
|
|
rtree_ctx, extents, extent, NULL, growing_retained);
|
|
} else if (extent_size_get(extent) >= SC_LARGE_MINCLASS) {
|
|
/* Always coalesce large extents eagerly. */
|
|
bool coalesced;
|
|
size_t prev_size;
|
|
do {
|
|
prev_size = extent_size_get(extent);
|
|
assert(extent_state_get(extent) == extent_state_active);
|
|
extent = extent_try_coalesce(tsdn, arena,
|
|
r_extent_hooks, rtree_ctx, extents, extent,
|
|
&coalesced, growing_retained);
|
|
} while (coalesced &&
|
|
extent_size_get(extent)
|
|
>= prev_size + SC_LARGE_MINCLASS);
|
|
}
|
|
extent_deactivate_locked(tsdn, arena, extents, extent);
|
|
|
|
malloc_mutex_unlock(tsdn, &extents->mtx);
|
|
}
|
|
|
|
void
|
|
extent_dalloc_gap(tsdn_t *tsdn, arena_t *arena, extent_t *extent) {
|
|
extent_hooks_t *extent_hooks = EXTENT_HOOKS_INITIALIZER;
|
|
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
if (extent_register(tsdn, extent)) {
|
|
extents_leak(tsdn, arena, &extent_hooks,
|
|
&arena->extents_retained, extent, false);
|
|
return;
|
|
}
|
|
extent_dalloc_wrapper(tsdn, arena, &extent_hooks, extent);
|
|
}
|
|
|
|
static bool
|
|
extent_dalloc_default_impl(void *addr, size_t size) {
|
|
if (!have_dss || !extent_in_dss(addr)) {
|
|
return extent_dalloc_mmap(addr, size);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool
|
|
extent_dalloc_default(extent_hooks_t *extent_hooks, void *addr, size_t size,
|
|
bool committed, unsigned arena_ind) {
|
|
return extent_dalloc_default_impl(addr, size);
|
|
}
|
|
|
|
static bool
|
|
extent_dalloc_wrapper_try(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent) {
|
|
bool err;
|
|
|
|
assert(extent_base_get(extent) != NULL);
|
|
assert(extent_size_get(extent) != 0);
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
extent_addr_set(extent, extent_base_get(extent));
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
/* Try to deallocate. */
|
|
if (*r_extent_hooks == &extent_hooks_default) {
|
|
/* Call directly to propagate tsdn. */
|
|
err = extent_dalloc_default_impl(extent_base_get(extent),
|
|
extent_size_get(extent));
|
|
} else {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
err = ((*r_extent_hooks)->dalloc == NULL ||
|
|
(*r_extent_hooks)->dalloc(*r_extent_hooks,
|
|
extent_base_get(extent), extent_size_get(extent),
|
|
extent_committed_get(extent), arena_ind_get(arena)));
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
|
|
if (!err) {
|
|
extent_dalloc(tsdn, arena, extent);
|
|
}
|
|
|
|
return err;
|
|
}
|
|
|
|
void
|
|
extent_dalloc_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent) {
|
|
assert(extent_dumpable_get(extent));
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
/*
|
|
* Deregister first to avoid a race with other allocating threads, and
|
|
* reregister if deallocation fails.
|
|
*/
|
|
extent_deregister(tsdn, extent);
|
|
if (!extent_dalloc_wrapper_try(tsdn, arena, r_extent_hooks, extent)) {
|
|
return;
|
|
}
|
|
|
|
extent_reregister(tsdn, extent);
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
}
|
|
/* Try to decommit; purge if that fails. */
|
|
bool zeroed;
|
|
if (!extent_committed_get(extent)) {
|
|
zeroed = true;
|
|
} else if (!extent_decommit_wrapper(tsdn, arena, r_extent_hooks, extent,
|
|
0, extent_size_get(extent))) {
|
|
zeroed = true;
|
|
} else if ((*r_extent_hooks)->purge_forced != NULL &&
|
|
!(*r_extent_hooks)->purge_forced(*r_extent_hooks,
|
|
extent_base_get(extent), extent_size_get(extent), 0,
|
|
extent_size_get(extent), arena_ind_get(arena))) {
|
|
zeroed = true;
|
|
} else if (extent_state_get(extent) == extent_state_muzzy ||
|
|
((*r_extent_hooks)->purge_lazy != NULL &&
|
|
!(*r_extent_hooks)->purge_lazy(*r_extent_hooks,
|
|
extent_base_get(extent), extent_size_get(extent), 0,
|
|
extent_size_get(extent), arena_ind_get(arena)))) {
|
|
zeroed = false;
|
|
} else {
|
|
zeroed = false;
|
|
}
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
extent_zeroed_set(extent, zeroed);
|
|
|
|
if (config_prof) {
|
|
extent_gdump_sub(tsdn, extent);
|
|
}
|
|
|
|
extent_record(tsdn, arena, r_extent_hooks, &arena->extents_retained,
|
|
extent, false);
|
|
}
|
|
|
|
static void
|
|
extent_destroy_default_impl(void *addr, size_t size) {
|
|
if (!have_dss || !extent_in_dss(addr)) {
|
|
pages_unmap(addr, size);
|
|
}
|
|
}
|
|
|
|
static void
|
|
extent_destroy_default(extent_hooks_t *extent_hooks, void *addr, size_t size,
|
|
bool committed, unsigned arena_ind) {
|
|
extent_destroy_default_impl(addr, size);
|
|
}
|
|
|
|
void
|
|
extent_destroy_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent) {
|
|
assert(extent_base_get(extent) != NULL);
|
|
assert(extent_size_get(extent) != 0);
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
/* Deregister first to avoid a race with other allocating threads. */
|
|
extent_deregister(tsdn, extent);
|
|
|
|
extent_addr_set(extent, extent_base_get(extent));
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
/* Try to destroy; silently fail otherwise. */
|
|
if (*r_extent_hooks == &extent_hooks_default) {
|
|
/* Call directly to propagate tsdn. */
|
|
extent_destroy_default_impl(extent_base_get(extent),
|
|
extent_size_get(extent));
|
|
} else if ((*r_extent_hooks)->destroy != NULL) {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
(*r_extent_hooks)->destroy(*r_extent_hooks,
|
|
extent_base_get(extent), extent_size_get(extent),
|
|
extent_committed_get(extent), arena_ind_get(arena));
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
|
|
extent_dalloc(tsdn, arena, extent);
|
|
}
|
|
|
|
static bool
|
|
extent_commit_default(extent_hooks_t *extent_hooks, void *addr, size_t size,
|
|
size_t offset, size_t length, unsigned arena_ind) {
|
|
return pages_commit((void *)((uintptr_t)addr + (uintptr_t)offset),
|
|
length);
|
|
}
|
|
|
|
static bool
|
|
extent_commit_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length, bool growing_retained) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, growing_retained ? 1 : 0);
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
}
|
|
bool err = ((*r_extent_hooks)->commit == NULL ||
|
|
(*r_extent_hooks)->commit(*r_extent_hooks, extent_base_get(extent),
|
|
extent_size_get(extent), offset, length, arena_ind_get(arena)));
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
extent_committed_set(extent, extent_committed_get(extent) || !err);
|
|
return err;
|
|
}
|
|
|
|
bool
|
|
extent_commit_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length) {
|
|
return extent_commit_impl(tsdn, arena, r_extent_hooks, extent, offset,
|
|
length, false);
|
|
}
|
|
|
|
static bool
|
|
extent_decommit_default(extent_hooks_t *extent_hooks, void *addr, size_t size,
|
|
size_t offset, size_t length, unsigned arena_ind) {
|
|
return pages_decommit((void *)((uintptr_t)addr + (uintptr_t)offset),
|
|
length);
|
|
}
|
|
|
|
bool
|
|
extent_decommit_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, 0);
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
}
|
|
bool err = ((*r_extent_hooks)->decommit == NULL ||
|
|
(*r_extent_hooks)->decommit(*r_extent_hooks,
|
|
extent_base_get(extent), extent_size_get(extent), offset, length,
|
|
arena_ind_get(arena)));
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
extent_committed_set(extent, extent_committed_get(extent) && err);
|
|
return err;
|
|
}
|
|
|
|
#ifdef PAGES_CAN_PURGE_LAZY
|
|
static bool
|
|
extent_purge_lazy_default(extent_hooks_t *extent_hooks, void *addr, size_t size,
|
|
size_t offset, size_t length, unsigned arena_ind) {
|
|
assert(addr != NULL);
|
|
assert((offset & PAGE_MASK) == 0);
|
|
assert(length != 0);
|
|
assert((length & PAGE_MASK) == 0);
|
|
|
|
return pages_purge_lazy((void *)((uintptr_t)addr + (uintptr_t)offset),
|
|
length);
|
|
}
|
|
#endif
|
|
|
|
static bool
|
|
extent_purge_lazy_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length, bool growing_retained) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, growing_retained ? 1 : 0);
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
|
|
if ((*r_extent_hooks)->purge_lazy == NULL) {
|
|
return true;
|
|
}
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
}
|
|
bool err = (*r_extent_hooks)->purge_lazy(*r_extent_hooks,
|
|
extent_base_get(extent), extent_size_get(extent), offset, length,
|
|
arena_ind_get(arena));
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
|
|
return err;
|
|
}
|
|
|
|
bool
|
|
extent_purge_lazy_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length) {
|
|
return extent_purge_lazy_impl(tsdn, arena, r_extent_hooks, extent,
|
|
offset, length, false);
|
|
}
|
|
|
|
#ifdef PAGES_CAN_PURGE_FORCED
|
|
static bool
|
|
extent_purge_forced_default(extent_hooks_t *extent_hooks, void *addr,
|
|
size_t size, size_t offset, size_t length, unsigned arena_ind) {
|
|
assert(addr != NULL);
|
|
assert((offset & PAGE_MASK) == 0);
|
|
assert(length != 0);
|
|
assert((length & PAGE_MASK) == 0);
|
|
|
|
return pages_purge_forced((void *)((uintptr_t)addr +
|
|
(uintptr_t)offset), length);
|
|
}
|
|
#endif
|
|
|
|
static bool
|
|
extent_purge_forced_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length, bool growing_retained) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, growing_retained ? 1 : 0);
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
|
|
if ((*r_extent_hooks)->purge_forced == NULL) {
|
|
return true;
|
|
}
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
}
|
|
bool err = (*r_extent_hooks)->purge_forced(*r_extent_hooks,
|
|
extent_base_get(extent), extent_size_get(extent), offset, length,
|
|
arena_ind_get(arena));
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
return err;
|
|
}
|
|
|
|
bool
|
|
extent_purge_forced_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t offset,
|
|
size_t length) {
|
|
return extent_purge_forced_impl(tsdn, arena, r_extent_hooks, extent,
|
|
offset, length, false);
|
|
}
|
|
|
|
#ifdef JEMALLOC_MAPS_COALESCE
|
|
static bool
|
|
extent_split_default(extent_hooks_t *extent_hooks, void *addr, size_t size,
|
|
size_t size_a, size_t size_b, bool committed, unsigned arena_ind) {
|
|
return !maps_coalesce;
|
|
}
|
|
#endif
|
|
|
|
/*
|
|
* Accepts the extent to split, and the characteristics of each side of the
|
|
* split. The 'a' parameters go with the 'lead' of the resulting pair of
|
|
* extents (the lower addressed portion of the split), and the 'b' parameters go
|
|
* with the trail (the higher addressed portion). This makes 'extent' the lead,
|
|
* and returns the trail (except in case of error).
|
|
*/
|
|
static extent_t *
|
|
extent_split_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t size_a,
|
|
szind_t szind_a, bool slab_a, size_t size_b, szind_t szind_b, bool slab_b,
|
|
bool growing_retained) {
|
|
assert(extent_size_get(extent) == size_a + size_b);
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, growing_retained ? 1 : 0);
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
|
|
if ((*r_extent_hooks)->split == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
extent_t *trail = extent_alloc(tsdn, arena);
|
|
if (trail == NULL) {
|
|
goto label_error_a;
|
|
}
|
|
|
|
extent_init(trail, arena, (void *)((uintptr_t)extent_base_get(extent) +
|
|
size_a), size_b, slab_b, szind_b, extent_sn_get(extent),
|
|
extent_state_get(extent), extent_zeroed_get(extent),
|
|
extent_committed_get(extent), extent_dumpable_get(extent));
|
|
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
rtree_leaf_elm_t *lead_elm_a, *lead_elm_b;
|
|
{
|
|
extent_t lead;
|
|
|
|
extent_init(&lead, arena, extent_addr_get(extent), size_a,
|
|
slab_a, szind_a, extent_sn_get(extent),
|
|
extent_state_get(extent), extent_zeroed_get(extent),
|
|
extent_committed_get(extent), extent_dumpable_get(extent));
|
|
|
|
extent_rtree_leaf_elms_lookup(tsdn, rtree_ctx, &lead, false,
|
|
true, &lead_elm_a, &lead_elm_b);
|
|
}
|
|
rtree_leaf_elm_t *trail_elm_a, *trail_elm_b;
|
|
extent_rtree_leaf_elms_lookup(tsdn, rtree_ctx, trail, false, true,
|
|
&trail_elm_a, &trail_elm_b);
|
|
|
|
if (lead_elm_a == NULL || lead_elm_b == NULL || trail_elm_a == NULL
|
|
|| trail_elm_b == NULL) {
|
|
goto label_error_b;
|
|
}
|
|
|
|
extent_lock2(tsdn, extent, trail);
|
|
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
}
|
|
bool err = (*r_extent_hooks)->split(*r_extent_hooks, extent_base_get(extent),
|
|
size_a + size_b, size_a, size_b, extent_committed_get(extent),
|
|
arena_ind_get(arena));
|
|
if (*r_extent_hooks != &extent_hooks_default) {
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
if (err) {
|
|
goto label_error_c;
|
|
}
|
|
|
|
extent_size_set(extent, size_a);
|
|
extent_szind_set(extent, szind_a);
|
|
|
|
extent_rtree_write_acquired(tsdn, lead_elm_a, lead_elm_b, extent,
|
|
szind_a, slab_a);
|
|
extent_rtree_write_acquired(tsdn, trail_elm_a, trail_elm_b, trail,
|
|
szind_b, slab_b);
|
|
|
|
extent_unlock2(tsdn, extent, trail);
|
|
|
|
return trail;
|
|
label_error_c:
|
|
extent_unlock2(tsdn, extent, trail);
|
|
label_error_b:
|
|
extent_dalloc(tsdn, arena, trail);
|
|
label_error_a:
|
|
return NULL;
|
|
}
|
|
|
|
extent_t *
|
|
extent_split_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *extent, size_t size_a,
|
|
szind_t szind_a, bool slab_a, size_t size_b, szind_t szind_b, bool slab_b) {
|
|
return extent_split_impl(tsdn, arena, r_extent_hooks, extent, size_a,
|
|
szind_a, slab_a, size_b, szind_b, slab_b, false);
|
|
}
|
|
|
|
static bool
|
|
extent_merge_default_impl(void *addr_a, void *addr_b) {
|
|
if (!maps_coalesce) {
|
|
return true;
|
|
}
|
|
if (have_dss && !extent_dss_mergeable(addr_a, addr_b)) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
#ifdef JEMALLOC_MAPS_COALESCE
|
|
static bool
|
|
extent_merge_default(extent_hooks_t *extent_hooks, void *addr_a, size_t size_a,
|
|
void *addr_b, size_t size_b, bool committed, unsigned arena_ind) {
|
|
return extent_merge_default_impl(addr_a, addr_b);
|
|
}
|
|
#endif
|
|
|
|
static bool
|
|
extent_merge_impl(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *a, extent_t *b,
|
|
bool growing_retained) {
|
|
witness_assert_depth_to_rank(tsdn_witness_tsdp_get(tsdn),
|
|
WITNESS_RANK_CORE, growing_retained ? 1 : 0);
|
|
|
|
extent_hooks_assure_initialized(arena, r_extent_hooks);
|
|
|
|
if ((*r_extent_hooks)->merge == NULL) {
|
|
return true;
|
|
}
|
|
|
|
bool err;
|
|
if (*r_extent_hooks == &extent_hooks_default) {
|
|
/* Call directly to propagate tsdn. */
|
|
err = extent_merge_default_impl(extent_base_get(a),
|
|
extent_base_get(b));
|
|
} else {
|
|
extent_hook_pre_reentrancy(tsdn, arena);
|
|
err = (*r_extent_hooks)->merge(*r_extent_hooks,
|
|
extent_base_get(a), extent_size_get(a), extent_base_get(b),
|
|
extent_size_get(b), extent_committed_get(a),
|
|
arena_ind_get(arena));
|
|
extent_hook_post_reentrancy(tsdn);
|
|
}
|
|
|
|
if (err) {
|
|
return true;
|
|
}
|
|
|
|
/*
|
|
* The rtree writes must happen while all the relevant elements are
|
|
* owned, so the following code uses decomposed helper functions rather
|
|
* than extent_{,de}register() to do things in the right order.
|
|
*/
|
|
rtree_ctx_t rtree_ctx_fallback;
|
|
rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
|
|
rtree_leaf_elm_t *a_elm_a, *a_elm_b, *b_elm_a, *b_elm_b;
|
|
extent_rtree_leaf_elms_lookup(tsdn, rtree_ctx, a, true, false, &a_elm_a,
|
|
&a_elm_b);
|
|
extent_rtree_leaf_elms_lookup(tsdn, rtree_ctx, b, true, false, &b_elm_a,
|
|
&b_elm_b);
|
|
|
|
extent_lock2(tsdn, a, b);
|
|
|
|
if (a_elm_b != NULL) {
|
|
rtree_leaf_elm_write(tsdn, &extents_rtree, a_elm_b, NULL,
|
|
SC_NSIZES, false);
|
|
}
|
|
if (b_elm_b != NULL) {
|
|
rtree_leaf_elm_write(tsdn, &extents_rtree, b_elm_a, NULL,
|
|
SC_NSIZES, false);
|
|
} else {
|
|
b_elm_b = b_elm_a;
|
|
}
|
|
|
|
extent_size_set(a, extent_size_get(a) + extent_size_get(b));
|
|
extent_szind_set(a, SC_NSIZES);
|
|
extent_sn_set(a, (extent_sn_get(a) < extent_sn_get(b)) ?
|
|
extent_sn_get(a) : extent_sn_get(b));
|
|
extent_zeroed_set(a, extent_zeroed_get(a) && extent_zeroed_get(b));
|
|
|
|
extent_rtree_write_acquired(tsdn, a_elm_a, b_elm_b, a, SC_NSIZES,
|
|
false);
|
|
|
|
extent_unlock2(tsdn, a, b);
|
|
|
|
extent_dalloc(tsdn, extent_arena_get(b), b);
|
|
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
extent_merge_wrapper(tsdn_t *tsdn, arena_t *arena,
|
|
extent_hooks_t **r_extent_hooks, extent_t *a, extent_t *b) {
|
|
return extent_merge_impl(tsdn, arena, r_extent_hooks, a, b, false);
|
|
}
|
|
|
|
bool
|
|
extent_boot(void) {
|
|
if (rtree_new(&extents_rtree, true)) {
|
|
return true;
|
|
}
|
|
|
|
if (mutex_pool_init(&extent_mutex_pool, "extent_mutex_pool",
|
|
WITNESS_RANK_EXTENT_POOL)) {
|
|
return true;
|
|
}
|
|
|
|
if (have_dss) {
|
|
extent_dss_boot();
|
|
}
|
|
|
|
return false;
|
|
}
|