Blame memkind-1.10.0/jemalloc/src/tcache.c

Packit Service 724aca
#define JEMALLOC_TCACHE_C_
Packit Service 724aca
#include "jemalloc/internal/jemalloc_preamble.h"
Packit Service 724aca
#include "jemalloc/internal/jemalloc_internal_includes.h"
Packit Service 724aca
Packit Service 724aca
#include "jemalloc/internal/assert.h"
Packit Service 724aca
#include "jemalloc/internal/mutex.h"
Packit Service 724aca
#include "jemalloc/internal/safety_check.h"
Packit Service 724aca
#include "jemalloc/internal/sc.h"
Packit Service 724aca
Packit Service 724aca
/******************************************************************************/
Packit Service 724aca
/* Data. */
Packit Service 724aca
Packit Service 724aca
bool	opt_tcache = true;
Packit Service 724aca
ssize_t	opt_lg_tcache_max = LG_TCACHE_MAXCLASS_DEFAULT;
Packit Service 724aca
Packit Service 724aca
cache_bin_info_t	*tcache_bin_info;
Packit Service 724aca
static unsigned		stack_nelms; /* Total stack elms per tcache. */
Packit Service 724aca
Packit Service 724aca
unsigned		nhbins;
Packit Service 724aca
size_t			tcache_maxclass;
Packit Service 724aca
Packit Service 724aca
tcaches_t		*tcaches;
Packit Service 724aca
Packit Service 724aca
/* Index of first element within tcaches that has never been used. */
Packit Service 724aca
static unsigned		tcaches_past;
Packit Service 724aca
Packit Service 724aca
/* Head of singly linked list tracking available tcaches elements. */
Packit Service 724aca
static tcaches_t	*tcaches_avail;
Packit Service 724aca
Packit Service 724aca
/* Protects tcaches{,_past,_avail}. */
Packit Service 724aca
static malloc_mutex_t	tcaches_mtx;
Packit Service 724aca
Packit Service 724aca
/******************************************************************************/
Packit Service 724aca
Packit Service 724aca
size_t
Packit Service 724aca
tcache_salloc(tsdn_t *tsdn, const void *ptr) {
Packit Service 724aca
	return arena_salloc(tsdn, ptr);
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_event_hard(tsd_t *tsd, tcache_t *tcache) {
Packit Service 724aca
	szind_t binind = tcache->next_gc_bin;
Packit Service 724aca
Packit Service 724aca
	cache_bin_t *tbin;
Packit Service 724aca
	if (binind < SC_NBINS) {
Packit Service 724aca
		tbin = tcache_small_bin_get(tcache, binind);
Packit Service 724aca
	} else {
Packit Service 724aca
		tbin = tcache_large_bin_get(tcache, binind);
Packit Service 724aca
	}
Packit Service 724aca
	if (tbin->low_water > 0) {
Packit Service 724aca
		/*
Packit Service 724aca
		 * Flush (ceiling) 3/4 of the objects below the low water mark.
Packit Service 724aca
		 */
Packit Service 724aca
		if (binind < SC_NBINS) {
Packit Service 724aca
			tcache_bin_flush_small(tsd, tcache, tbin, binind,
Packit Service 724aca
			    tbin->ncached - tbin->low_water + (tbin->low_water
Packit Service 724aca
			    >> 2));
Packit Service 724aca
			/*
Packit Service 724aca
			 * Reduce fill count by 2X.  Limit lg_fill_div such that
Packit Service 724aca
			 * the fill count is always at least 1.
Packit Service 724aca
			 */
Packit Service 724aca
			cache_bin_info_t *tbin_info = &tcache_bin_info[binind];
Packit Service 724aca
			if ((tbin_info->ncached_max >>
Packit Service 724aca
			     (tcache->lg_fill_div[binind] + 1)) >= 1) {
Packit Service 724aca
				tcache->lg_fill_div[binind]++;
Packit Service 724aca
			}
Packit Service 724aca
		} else {
Packit Service 724aca
			tcache_bin_flush_large(tsd, tbin, binind, tbin->ncached
Packit Service 724aca
			    - tbin->low_water + (tbin->low_water >> 2), tcache);
Packit Service 724aca
		}
Packit Service 724aca
	} else if (tbin->low_water < 0) {
Packit Service 724aca
		/*
Packit Service 724aca
		 * Increase fill count by 2X for small bins.  Make sure
Packit Service 724aca
		 * lg_fill_div stays greater than 0.
Packit Service 724aca
		 */
Packit Service 724aca
		if (binind < SC_NBINS && tcache->lg_fill_div[binind] > 1) {
Packit Service 724aca
			tcache->lg_fill_div[binind]--;
Packit Service 724aca
		}
Packit Service 724aca
	}
Packit Service 724aca
	tbin->low_water = tbin->ncached;
Packit Service 724aca
Packit Service 724aca
	tcache->next_gc_bin++;
Packit Service 724aca
	if (tcache->next_gc_bin == nhbins) {
Packit Service 724aca
		tcache->next_gc_bin = 0;
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void *
Packit Service 724aca
tcache_alloc_small_hard(tsdn_t *tsdn, arena_t *arena, tcache_t *tcache,
Packit Service 724aca
    cache_bin_t *tbin, szind_t binind, bool *tcache_success) {
Packit Service 724aca
	void *ret;
Packit Service 724aca
Packit Service 724aca
	assert(tcache->arena != NULL);
Packit Service 724aca
	arena_tcache_fill_small(tsdn, arena, tcache, tbin, binind,
Packit Service 724aca
	    config_prof ? tcache->prof_accumbytes : 0);
Packit Service 724aca
	if (config_prof) {
Packit Service 724aca
		tcache->prof_accumbytes = 0;
Packit Service 724aca
	}
Packit Service 724aca
	ret = cache_bin_alloc_easy(tbin, tcache_success);
Packit Service 724aca
Packit Service 724aca
	return ret;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
/* Enabled with --enable-extra-size-check. */
Packit Service 724aca
static void
Packit Service 724aca
tbin_extents_lookup_size_check(tsdn_t *tsdn, cache_bin_t *tbin, szind_t binind,
Packit Service 724aca
    size_t nflush, extent_t **extents){
Packit Service 724aca
	rtree_ctx_t rtree_ctx_fallback;
Packit Service 724aca
	rtree_ctx_t *rtree_ctx = tsdn_rtree_ctx(tsdn, &rtree_ctx_fallback);
Packit Service 724aca
Packit Service 724aca
	/*
Packit Service 724aca
	 * Verify that the items in the tcache all have the correct size; this
Packit Service 724aca
	 * is useful for catching sized deallocation bugs, also to fail early
Packit Service 724aca
	 * instead of corrupting metadata.  Since this can be turned on for opt
Packit Service 724aca
	 * builds, avoid the branch in the loop.
Packit Service 724aca
	 */
Packit Service 724aca
	szind_t szind;
Packit Service 724aca
	size_t sz_sum = binind * nflush;
Packit Service 724aca
	for (unsigned i = 0 ; i < nflush; i++) {
Packit Service 724aca
		rtree_extent_szind_read(tsdn, &extents_rtree,
Packit Service 724aca
		    rtree_ctx, (uintptr_t)*(tbin->avail - 1 - i), true,
Packit Service 724aca
		    &extents[i], &szind);
Packit Service 724aca
		sz_sum -= szind;
Packit Service 724aca
	}
Packit Service 724aca
	if (sz_sum != 0) {
Packit Service 724aca
		safety_check_fail("<jemalloc>: size mismatch in thread cache "
Packit Service 724aca
		    "detected, likely caused by sized deallocation bugs by "
Packit Service 724aca
		    "application. Abort.\n");
Packit Service 724aca
		abort();
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_bin_flush_small(tsd_t *tsd, tcache_t *tcache, cache_bin_t *tbin,
Packit Service 724aca
    szind_t binind, unsigned rem) {
Packit Service 724aca
	bool merged_stats = false;
Packit Service 724aca
Packit Service 724aca
	assert(binind < SC_NBINS);
Packit Service 724aca
	assert((cache_bin_sz_t)rem <= tbin->ncached);
Packit Service 724aca
Packit Service 724aca
	arena_t *arena = tcache->arena;
Packit Service 724aca
	assert(arena != NULL);
Packit Service 724aca
	unsigned nflush = tbin->ncached - rem;
Packit Service 724aca
	VARIABLE_ARRAY(extent_t *, item_extent, nflush);
Packit Service 724aca
Packit Service 724aca
	/* Look up extent once per item. */
Packit Service 724aca
	if (config_opt_safety_checks) {
Packit Service 724aca
		tbin_extents_lookup_size_check(tsd_tsdn(tsd), tbin, binind,
Packit Service 724aca
		    nflush, item_extent);
Packit Service 724aca
	} else {
Packit Service 724aca
		for (unsigned i = 0 ; i < nflush; i++) {
Packit Service 724aca
			item_extent[i] = iealloc(tsd_tsdn(tsd),
Packit Service 724aca
			    *(tbin->avail - 1 - i));
Packit Service 724aca
		}
Packit Service 724aca
	}
Packit Service 724aca
	while (nflush > 0) {
Packit Service 724aca
		/* Lock the arena bin associated with the first object. */
Packit Service 724aca
		extent_t *extent = item_extent[0];
Packit Service 724aca
		unsigned bin_arena_ind = extent_arena_ind_get(extent);
Packit Service 724aca
		arena_t *bin_arena = arena_get(tsd_tsdn(tsd), bin_arena_ind,
Packit Service 724aca
		    false);
Packit Service 724aca
		unsigned binshard = extent_binshard_get(extent);
Packit Service 724aca
		assert(binshard < bin_infos[binind].n_shards);
Packit Service 724aca
		bin_t *bin = &bin_arena->bins[binind].bin_shards[binshard];
Packit Service 724aca
Packit Service 724aca
		if (config_prof && bin_arena == arena) {
Packit Service 724aca
			if (arena_prof_accum(tsd_tsdn(tsd), arena,
Packit Service 724aca
			    tcache->prof_accumbytes)) {
Packit Service 724aca
				prof_idump(tsd_tsdn(tsd));
Packit Service 724aca
			}
Packit Service 724aca
			tcache->prof_accumbytes = 0;
Packit Service 724aca
		}
Packit Service 724aca
Packit Service 724aca
		malloc_mutex_lock(tsd_tsdn(tsd), &bin->lock);
Packit Service 724aca
		if (config_stats && bin_arena == arena && !merged_stats) {
Packit Service 724aca
			merged_stats = true;
Packit Service 724aca
			bin->stats.nflushes++;
Packit Service 724aca
			bin->stats.nrequests += tbin->tstats.nrequests;
Packit Service 724aca
			tbin->tstats.nrequests = 0;
Packit Service 724aca
		}
Packit Service 724aca
		unsigned ndeferred = 0;
Packit Service 724aca
		for (unsigned i = 0; i < nflush; i++) {
Packit Service 724aca
			void *ptr = *(tbin->avail - 1 - i);
Packit Service 724aca
			extent = item_extent[i];
Packit Service 724aca
			assert(ptr != NULL && extent != NULL);
Packit Service 724aca
Packit Service 724aca
			if (extent_arena_ind_get(extent) == bin_arena_ind
Packit Service 724aca
			    && extent_binshard_get(extent) == binshard) {
Packit Service 724aca
				arena_dalloc_bin_junked_locked(tsd_tsdn(tsd),
Packit Service 724aca
				    bin_arena, bin, binind, extent, ptr);
Packit Service 724aca
			} else {
Packit Service 724aca
				/*
Packit Service 724aca
				 * This object was allocated via a different
Packit Service 724aca
				 * arena bin than the one that is currently
Packit Service 724aca
				 * locked.  Stash the object, so that it can be
Packit Service 724aca
				 * handled in a future pass.
Packit Service 724aca
				 */
Packit Service 724aca
				*(tbin->avail - 1 - ndeferred) = ptr;
Packit Service 724aca
				item_extent[ndeferred] = extent;
Packit Service 724aca
				ndeferred++;
Packit Service 724aca
			}
Packit Service 724aca
		}
Packit Service 724aca
		malloc_mutex_unlock(tsd_tsdn(tsd), &bin->lock);
Packit Service 724aca
		arena_decay_ticks(tsd_tsdn(tsd), bin_arena, nflush - ndeferred);
Packit Service 724aca
		nflush = ndeferred;
Packit Service 724aca
	}
Packit Service 724aca
	if (config_stats && !merged_stats) {
Packit Service 724aca
		/*
Packit Service 724aca
		 * The flush loop didn't happen to flush to this thread's
Packit Service 724aca
		 * arena, so the stats didn't get merged.  Manually do so now.
Packit Service 724aca
		 */
Packit Service 724aca
		unsigned binshard;
Packit Service 724aca
		bin_t *bin = arena_bin_choose_lock(tsd_tsdn(tsd), arena, binind,
Packit Service 724aca
		    &binshard);
Packit Service 724aca
		bin->stats.nflushes++;
Packit Service 724aca
		bin->stats.nrequests += tbin->tstats.nrequests;
Packit Service 724aca
		tbin->tstats.nrequests = 0;
Packit Service 724aca
		malloc_mutex_unlock(tsd_tsdn(tsd), &bin->lock);
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	memmove(tbin->avail - rem, tbin->avail - tbin->ncached, rem *
Packit Service 724aca
	    sizeof(void *));
Packit Service 724aca
	tbin->ncached = rem;
Packit Service 724aca
	if (tbin->ncached < tbin->low_water) {
Packit Service 724aca
		tbin->low_water = tbin->ncached;
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_bin_flush_large(tsd_t *tsd, cache_bin_t *tbin, szind_t binind,
Packit Service 724aca
    unsigned rem, tcache_t *tcache) {
Packit Service 724aca
	bool merged_stats = false;
Packit Service 724aca
Packit Service 724aca
	assert(binind < nhbins);
Packit Service 724aca
	assert((cache_bin_sz_t)rem <= tbin->ncached);
Packit Service 724aca
Packit Service 724aca
	arena_t *tcache_arena = tcache->arena;
Packit Service 724aca
	assert(tcache_arena != NULL);
Packit Service 724aca
	unsigned nflush = tbin->ncached - rem;
Packit Service 724aca
	VARIABLE_ARRAY(extent_t *, item_extent, nflush);
Packit Service 724aca
Packit Service 724aca
#ifndef JEMALLOC_EXTRA_SIZE_CHECK
Packit Service 724aca
	/* Look up extent once per item. */
Packit Service 724aca
	for (unsigned i = 0 ; i < nflush; i++) {
Packit Service 724aca
		item_extent[i] = iealloc(tsd_tsdn(tsd), *(tbin->avail - 1 - i));
Packit Service 724aca
	}
Packit Service 724aca
#else
Packit Service 724aca
	tbin_extents_lookup_size_check(tsd_tsdn(tsd), tbin, binind, nflush,
Packit Service 724aca
	    item_extent);
Packit Service 724aca
#endif
Packit Service 724aca
	while (nflush > 0) {
Packit Service 724aca
		/* Lock the arena associated with the first object. */
Packit Service 724aca
		extent_t *extent = item_extent[0];
Packit Service 724aca
		unsigned locked_arena_ind = extent_arena_ind_get(extent);
Packit Service 724aca
		arena_t *locked_arena = arena_get(tsd_tsdn(tsd),
Packit Service 724aca
		    locked_arena_ind, false);
Packit Service 724aca
		bool idump;
Packit Service 724aca
Packit Service 724aca
		if (config_prof) {
Packit Service 724aca
			idump = false;
Packit Service 724aca
		}
Packit Service 724aca
Packit Service 724aca
		bool lock_large = !arena_is_auto(locked_arena);
Packit Service 724aca
		if (lock_large) {
Packit Service 724aca
			malloc_mutex_lock(tsd_tsdn(tsd), &locked_arena->large_mtx);
Packit Service 724aca
		}
Packit Service 724aca
		for (unsigned i = 0; i < nflush; i++) {
Packit Service 724aca
			void *ptr = *(tbin->avail - 1 - i);
Packit Service 724aca
			assert(ptr != NULL);
Packit Service 724aca
			extent = item_extent[i];
Packit Service 724aca
			if (extent_arena_ind_get(extent) == locked_arena_ind) {
Packit Service 724aca
				large_dalloc_prep_junked_locked(tsd_tsdn(tsd),
Packit Service 724aca
				    extent);
Packit Service 724aca
			}
Packit Service 724aca
		}
Packit Service 724aca
		if ((config_prof || config_stats) &&
Packit Service 724aca
		    (locked_arena == tcache_arena)) {
Packit Service 724aca
			if (config_prof) {
Packit Service 724aca
				idump = arena_prof_accum(tsd_tsdn(tsd),
Packit Service 724aca
				    tcache_arena, tcache->prof_accumbytes);
Packit Service 724aca
				tcache->prof_accumbytes = 0;
Packit Service 724aca
			}
Packit Service 724aca
			if (config_stats) {
Packit Service 724aca
				merged_stats = true;
Packit Service 724aca
				arena_stats_large_flush_nrequests_add(
Packit Service 724aca
				    tsd_tsdn(tsd), &tcache_arena->stats, binind,
Packit Service 724aca
				    tbin->tstats.nrequests);
Packit Service 724aca
				tbin->tstats.nrequests = 0;
Packit Service 724aca
			}
Packit Service 724aca
		}
Packit Service 724aca
		if (lock_large) {
Packit Service 724aca
			malloc_mutex_unlock(tsd_tsdn(tsd), &locked_arena->large_mtx);
Packit Service 724aca
		}
Packit Service 724aca
Packit Service 724aca
		unsigned ndeferred = 0;
Packit Service 724aca
		for (unsigned i = 0; i < nflush; i++) {
Packit Service 724aca
			void *ptr = *(tbin->avail - 1 - i);
Packit Service 724aca
			extent = item_extent[i];
Packit Service 724aca
			assert(ptr != NULL && extent != NULL);
Packit Service 724aca
Packit Service 724aca
			if (extent_arena_ind_get(extent) == locked_arena_ind) {
Packit Service 724aca
				large_dalloc_finish(tsd_tsdn(tsd), extent);
Packit Service 724aca
			} else {
Packit Service 724aca
				/*
Packit Service 724aca
				 * This object was allocated via a different
Packit Service 724aca
				 * arena than the one that is currently locked.
Packit Service 724aca
				 * Stash the object, so that it can be handled
Packit Service 724aca
				 * in a future pass.
Packit Service 724aca
				 */
Packit Service 724aca
				*(tbin->avail - 1 - ndeferred) = ptr;
Packit Service 724aca
				item_extent[ndeferred] = extent;
Packit Service 724aca
				ndeferred++;
Packit Service 724aca
			}
Packit Service 724aca
		}
Packit Service 724aca
		if (config_prof && idump) {
Packit Service 724aca
			prof_idump(tsd_tsdn(tsd));
Packit Service 724aca
		}
Packit Service 724aca
		arena_decay_ticks(tsd_tsdn(tsd), locked_arena, nflush -
Packit Service 724aca
		    ndeferred);
Packit Service 724aca
		nflush = ndeferred;
Packit Service 724aca
	}
Packit Service 724aca
	if (config_stats && !merged_stats) {
Packit Service 724aca
		/*
Packit Service 724aca
		 * The flush loop didn't happen to flush to this thread's
Packit Service 724aca
		 * arena, so the stats didn't get merged.  Manually do so now.
Packit Service 724aca
		 */
Packit Service 724aca
		arena_stats_large_flush_nrequests_add(tsd_tsdn(tsd),
Packit Service 724aca
		    &tcache_arena->stats, binind, tbin->tstats.nrequests);
Packit Service 724aca
		tbin->tstats.nrequests = 0;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	memmove(tbin->avail - rem, tbin->avail - tbin->ncached, rem *
Packit Service 724aca
	    sizeof(void *));
Packit Service 724aca
	tbin->ncached = rem;
Packit Service 724aca
	if (tbin->ncached < tbin->low_water) {
Packit Service 724aca
		tbin->low_water = tbin->ncached;
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_arena_associate(tsdn_t *tsdn, tcache_t *tcache, arena_t *arena) {
Packit Service 724aca
	assert(tcache->arena == NULL);
Packit Service 724aca
	tcache->arena = arena;
Packit Service 724aca
Packit Service 724aca
	if (config_stats) {
Packit Service 724aca
		/* Link into list of extant tcaches. */
Packit Service 724aca
		malloc_mutex_lock(tsdn, &arena->tcache_ql_mtx);
Packit Service 724aca
Packit Service 724aca
		ql_elm_new(tcache, link);
Packit Service 724aca
		ql_tail_insert(&arena->tcache_ql, tcache, link);
Packit Service 724aca
		cache_bin_array_descriptor_init(
Packit Service 724aca
		    &tcache->cache_bin_array_descriptor, tcache->bins_small,
Packit Service 724aca
		    tcache->bins_large);
Packit Service 724aca
		ql_tail_insert(&arena->cache_bin_array_descriptor_ql,
Packit Service 724aca
		    &tcache->cache_bin_array_descriptor, link);
Packit Service 724aca
Packit Service 724aca
		malloc_mutex_unlock(tsdn, &arena->tcache_ql_mtx);
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
static void
Packit Service 724aca
tcache_arena_dissociate(tsdn_t *tsdn, tcache_t *tcache) {
Packit Service 724aca
	arena_t *arena = tcache->arena;
Packit Service 724aca
	assert(arena != NULL);
Packit Service 724aca
	if (config_stats) {
Packit Service 724aca
		/* Unlink from list of extant tcaches. */
Packit Service 724aca
		malloc_mutex_lock(tsdn, &arena->tcache_ql_mtx);
Packit Service 724aca
		if (config_debug) {
Packit Service 724aca
			bool in_ql = false;
Packit Service 724aca
			tcache_t *iter;
Packit Service 724aca
			ql_foreach(iter, &arena->tcache_ql, link) {
Packit Service 724aca
				if (iter == tcache) {
Packit Service 724aca
					in_ql = true;
Packit Service 724aca
					break;
Packit Service 724aca
				}
Packit Service 724aca
			}
Packit Service 724aca
			assert(in_ql);
Packit Service 724aca
		}
Packit Service 724aca
		ql_remove(&arena->tcache_ql, tcache, link);
Packit Service 724aca
		ql_remove(&arena->cache_bin_array_descriptor_ql,
Packit Service 724aca
		    &tcache->cache_bin_array_descriptor, link);
Packit Service 724aca
		tcache_stats_merge(tsdn, tcache, arena);
Packit Service 724aca
		malloc_mutex_unlock(tsdn, &arena->tcache_ql_mtx);
Packit Service 724aca
	}
Packit Service 724aca
	tcache->arena = NULL;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_arena_reassociate(tsdn_t *tsdn, tcache_t *tcache, arena_t *arena) {
Packit Service 724aca
	tcache_arena_dissociate(tsdn, tcache);
Packit Service 724aca
	tcache_arena_associate(tsdn, tcache, arena);
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
bool
Packit Service 724aca
tsd_tcache_enabled_data_init(tsd_t *tsd) {
Packit Service 724aca
	/* Called upon tsd initialization. */
Packit Service 724aca
	tsd_tcache_enabled_set(tsd, opt_tcache);
Packit Service 724aca
	tsd_slow_update(tsd);
Packit Service 724aca
Packit Service 724aca
	if (opt_tcache) {
Packit Service 724aca
		/* Trigger tcache init. */
Packit Service 724aca
		tsd_tcache_data_init(tsd);
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	return false;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
/* Initialize auto tcache (embedded in TSD). */
Packit Service 724aca
static void
Packit Service 724aca
tcache_init(tsd_t *tsd, tcache_t *tcache, void *avail_stack) {
Packit Service 724aca
	memset(&tcache->link, 0, sizeof(ql_elm(tcache_t)));
Packit Service 724aca
	tcache->prof_accumbytes = 0;
Packit Service 724aca
	tcache->next_gc_bin = 0;
Packit Service 724aca
	tcache->arena = NULL;
Packit Service 724aca
Packit Service 724aca
	ticker_init(&tcache->gc_ticker, TCACHE_GC_INCR);
Packit Service 724aca
Packit Service 724aca
	size_t stack_offset = 0;
Packit Service 724aca
	assert((TCACHE_NSLOTS_SMALL_MAX & 1U) == 0);
Packit Service 724aca
	memset(tcache->bins_small, 0, sizeof(cache_bin_t) * SC_NBINS);
Packit Service 724aca
	memset(tcache->bins_large, 0, sizeof(cache_bin_t) * (nhbins - SC_NBINS));
Packit Service 724aca
	unsigned i = 0;
Packit Service 724aca
	for (; i < SC_NBINS; i++) {
Packit Service 724aca
		tcache->lg_fill_div[i] = 1;
Packit Service 724aca
		stack_offset += tcache_bin_info[i].ncached_max * sizeof(void *);
Packit Service 724aca
		/*
Packit Service 724aca
		 * avail points past the available space.  Allocations will
Packit Service 724aca
		 * access the slots toward higher addresses (for the benefit of
Packit Service 724aca
		 * prefetch).
Packit Service 724aca
		 */
Packit Service 724aca
		tcache_small_bin_get(tcache, i)->avail =
Packit Service 724aca
		    (void **)((uintptr_t)avail_stack + (uintptr_t)stack_offset);
Packit Service 724aca
	}
Packit Service 724aca
	for (; i < nhbins; i++) {
Packit Service 724aca
		stack_offset += tcache_bin_info[i].ncached_max * sizeof(void *);
Packit Service 724aca
		tcache_large_bin_get(tcache, i)->avail =
Packit Service 724aca
		    (void **)((uintptr_t)avail_stack + (uintptr_t)stack_offset);
Packit Service 724aca
	}
Packit Service 724aca
	assert(stack_offset == stack_nelms * sizeof(void *));
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
/* Initialize auto tcache (embedded in TSD). */
Packit Service 724aca
bool
Packit Service 724aca
tsd_tcache_data_init(tsd_t *tsd) {
Packit Service 724aca
	tcache_t *tcache = tsd_tcachep_get_unsafe(tsd);
Packit Service 724aca
	assert(tcache_small_bin_get(tcache, 0)->avail == NULL);
Packit Service 724aca
	size_t size = stack_nelms * sizeof(void *);
Packit Service 724aca
	/* Avoid false cacheline sharing. */
Packit Service 724aca
	size = sz_sa2u(size, CACHELINE);
Packit Service 724aca
Packit Service 724aca
	void *avail_array = ipallocztm(tsd_tsdn(tsd), size, CACHELINE, true,
Packit Service 724aca
	    NULL, true, arena_get(TSDN_NULL, 0, true));
Packit Service 724aca
	if (avail_array == NULL) {
Packit Service 724aca
		return true;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	tcache_init(tsd, tcache, avail_array);
Packit Service 724aca
	/*
Packit Service 724aca
	 * Initialization is a bit tricky here.  After malloc init is done, all
Packit Service 724aca
	 * threads can rely on arena_choose and associate tcache accordingly.
Packit Service 724aca
	 * However, the thread that does actual malloc bootstrapping relies on
Packit Service 724aca
	 * functional tsd, and it can only rely on a0.  In that case, we
Packit Service 724aca
	 * associate its tcache to a0 temporarily, and later on
Packit Service 724aca
	 * arena_choose_hard() will re-associate properly.
Packit Service 724aca
	 */
Packit Service 724aca
	tcache->arena = NULL;
Packit Service 724aca
	arena_t *arena;
Packit Service 724aca
	if (!malloc_initialized()) {
Packit Service 724aca
		/* If in initialization, assign to a0. */
Packit Service 724aca
		arena = arena_get(tsd_tsdn(tsd), 0, false);
Packit Service 724aca
		tcache_arena_associate(tsd_tsdn(tsd), tcache, arena);
Packit Service 724aca
	} else {
Packit Service 724aca
		arena = arena_choose(tsd, NULL);
Packit Service 724aca
		/* This may happen if thread.tcache.enabled is used. */
Packit Service 724aca
		if (tcache->arena == NULL) {
Packit Service 724aca
			tcache_arena_associate(tsd_tsdn(tsd), tcache, arena);
Packit Service 724aca
		}
Packit Service 724aca
	}
Packit Service 724aca
	assert(arena == tcache->arena);
Packit Service 724aca
Packit Service 724aca
	return false;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
/* Created manual tcache for tcache.create mallctl. */
Packit Service 724aca
tcache_t *
Packit Service 724aca
tcache_create_explicit(tsd_t *tsd) {
Packit Service 724aca
	tcache_t *tcache;
Packit Service 724aca
	size_t size, stack_offset;
Packit Service 724aca
Packit Service 724aca
	size = sizeof(tcache_t);
Packit Service 724aca
	/* Naturally align the pointer stacks. */
Packit Service 724aca
	size = PTR_CEILING(size);
Packit Service 724aca
	stack_offset = size;
Packit Service 724aca
	size += stack_nelms * sizeof(void *);
Packit Service 724aca
	/* Avoid false cacheline sharing. */
Packit Service 724aca
	size = sz_sa2u(size, CACHELINE);
Packit Service 724aca
Packit Service 724aca
	tcache = ipallocztm(tsd_tsdn(tsd), size, CACHELINE, true, NULL, true,
Packit Service 724aca
	    arena_get(TSDN_NULL, 0, true));
Packit Service 724aca
	if (tcache == NULL) {
Packit Service 724aca
		return NULL;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	tcache_init(tsd, tcache,
Packit Service 724aca
	    (void *)((uintptr_t)tcache + (uintptr_t)stack_offset));
Packit Service 724aca
	tcache_arena_associate(tsd_tsdn(tsd), tcache, arena_ichoose(tsd, NULL));
Packit Service 724aca
Packit Service 724aca
	return tcache;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
static void
Packit Service 724aca
tcache_flush_cache(tsd_t *tsd, tcache_t *tcache) {
Packit Service 724aca
	assert(tcache->arena != NULL);
Packit Service 724aca
Packit Service 724aca
	for (unsigned i = 0; i < SC_NBINS; i++) {
Packit Service 724aca
		cache_bin_t *tbin = tcache_small_bin_get(tcache, i);
Packit Service 724aca
		tcache_bin_flush_small(tsd, tcache, tbin, i, 0);
Packit Service 724aca
Packit Service 724aca
		if (config_stats) {
Packit Service 724aca
			assert(tbin->tstats.nrequests == 0);
Packit Service 724aca
		}
Packit Service 724aca
	}
Packit Service 724aca
	for (unsigned i = SC_NBINS; i < nhbins; i++) {
Packit Service 724aca
		cache_bin_t *tbin = tcache_large_bin_get(tcache, i);
Packit Service 724aca
		tcache_bin_flush_large(tsd, tbin, i, 0, tcache);
Packit Service 724aca
Packit Service 724aca
		if (config_stats) {
Packit Service 724aca
			assert(tbin->tstats.nrequests == 0);
Packit Service 724aca
		}
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	if (config_prof && tcache->prof_accumbytes > 0 &&
Packit Service 724aca
	    arena_prof_accum(tsd_tsdn(tsd), tcache->arena,
Packit Service 724aca
	    tcache->prof_accumbytes)) {
Packit Service 724aca
		prof_idump(tsd_tsdn(tsd));
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_flush(tsd_t *tsd) {
Packit Service 724aca
	assert(tcache_available(tsd));
Packit Service 724aca
	tcache_flush_cache(tsd, tsd_tcachep_get(tsd));
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
static void
Packit Service 724aca
tcache_destroy(tsd_t *tsd, tcache_t *tcache, bool tsd_tcache) {
Packit Service 724aca
	tcache_flush_cache(tsd, tcache);
Packit Service 724aca
	arena_t *arena = tcache->arena;
Packit Service 724aca
	tcache_arena_dissociate(tsd_tsdn(tsd), tcache);
Packit Service 724aca
Packit Service 724aca
	if (tsd_tcache) {
Packit Service 724aca
		/* Release the avail array for the TSD embedded auto tcache. */
Packit Service 724aca
		void *avail_array =
Packit Service 724aca
		    (void *)((uintptr_t)tcache_small_bin_get(tcache, 0)->avail -
Packit Service 724aca
		    (uintptr_t)tcache_bin_info[0].ncached_max * sizeof(void *));
Packit Service 724aca
		idalloctm(tsd_tsdn(tsd), avail_array, NULL, NULL, true, true);
Packit Service 724aca
	} else {
Packit Service 724aca
		/* Release both the tcache struct and avail array. */
Packit Service 724aca
		idalloctm(tsd_tsdn(tsd), tcache, NULL, NULL, true, true);
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	/*
Packit Service 724aca
	 * The deallocation and tcache flush above may not trigger decay since
Packit Service 724aca
	 * we are on the tcache shutdown path (potentially with non-nominal
Packit Service 724aca
	 * tsd).  Manually trigger decay to avoid pathological cases.  Also
Packit Service 724aca
	 * include arena 0 because the tcache array is allocated from it.
Packit Service 724aca
	 */
Packit Service 724aca
	arena_decay(tsd_tsdn(tsd), arena_get(tsd_tsdn(tsd), 0, false),
Packit Service 724aca
	    false, false);
Packit Service 724aca
Packit Service 724aca
	if (arena_nthreads_get(arena, false) == 0 &&
Packit Service 724aca
	    !background_thread_enabled()) {
Packit Service 724aca
		/* Force purging when no threads assigned to the arena anymore. */
Packit Service 724aca
		arena_decay(tsd_tsdn(tsd), arena, false, true);
Packit Service 724aca
	} else {
Packit Service 724aca
		arena_decay(tsd_tsdn(tsd), arena, false, false);
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
/* For auto tcache (embedded in TSD) only. */
Packit Service 724aca
void
Packit Service 724aca
tcache_cleanup(tsd_t *tsd) {
Packit Service 724aca
	tcache_t *tcache = tsd_tcachep_get(tsd);
Packit Service 724aca
	if (!tcache_available(tsd)) {
Packit Service 724aca
		assert(tsd_tcache_enabled_get(tsd) == false);
Packit Service 724aca
		if (config_debug) {
Packit Service 724aca
			assert(tcache_small_bin_get(tcache, 0)->avail == NULL);
Packit Service 724aca
		}
Packit Service 724aca
		return;
Packit Service 724aca
	}
Packit Service 724aca
	assert(tsd_tcache_enabled_get(tsd));
Packit Service 724aca
	assert(tcache_small_bin_get(tcache, 0)->avail != NULL);
Packit Service 724aca
Packit Service 724aca
	tcache_destroy(tsd, tcache, true);
Packit Service 724aca
	if (config_debug) {
Packit Service 724aca
		tcache_small_bin_get(tcache, 0)->avail = NULL;
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_stats_merge(tsdn_t *tsdn, tcache_t *tcache, arena_t *arena) {
Packit Service 724aca
	unsigned i;
Packit Service 724aca
Packit Service 724aca
	cassert(config_stats);
Packit Service 724aca
Packit Service 724aca
	/* Merge and reset tcache stats. */
Packit Service 724aca
	for (i = 0; i < SC_NBINS; i++) {
Packit Service 724aca
		cache_bin_t *tbin = tcache_small_bin_get(tcache, i);
Packit Service 724aca
		unsigned binshard;
Packit Service 724aca
		bin_t *bin = arena_bin_choose_lock(tsdn, arena, i, &binshard);
Packit Service 724aca
		bin->stats.nrequests += tbin->tstats.nrequests;
Packit Service 724aca
		malloc_mutex_unlock(tsdn, &bin->lock);
Packit Service 724aca
		tbin->tstats.nrequests = 0;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	for (; i < nhbins; i++) {
Packit Service 724aca
		cache_bin_t *tbin = tcache_large_bin_get(tcache, i);
Packit Service 724aca
		arena_stats_large_flush_nrequests_add(tsdn, &arena->stats, i,
Packit Service 724aca
		    tbin->tstats.nrequests);
Packit Service 724aca
		tbin->tstats.nrequests = 0;
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
static bool
Packit Service 724aca
tcaches_create_prep(tsd_t *tsd) {
Packit Service 724aca
	bool err;
Packit Service 724aca
Packit Service 724aca
	malloc_mutex_lock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
Packit Service 724aca
	if (tcaches == NULL) {
Packit Service 724aca
		tcaches = base_alloc(tsd_tsdn(tsd), b0get(), sizeof(tcache_t *)
Packit Service 724aca
		    * (MALLOCX_TCACHE_MAX+1), CACHELINE);
Packit Service 724aca
		if (tcaches == NULL) {
Packit Service 724aca
			err = true;
Packit Service 724aca
			goto label_return;
Packit Service 724aca
		}
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	if (tcaches_avail == NULL && tcaches_past > MALLOCX_TCACHE_MAX) {
Packit Service 724aca
		err = true;
Packit Service 724aca
		goto label_return;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	err = false;
Packit Service 724aca
label_return:
Packit Service 724aca
	malloc_mutex_unlock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
	return err;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
bool
Packit Service 724aca
tcaches_create(tsd_t *tsd, unsigned *r_ind) {
Packit Service 724aca
	witness_assert_depth(tsdn_witness_tsdp_get(tsd_tsdn(tsd)), 0);
Packit Service 724aca
Packit Service 724aca
	bool err;
Packit Service 724aca
Packit Service 724aca
	if (tcaches_create_prep(tsd)) {
Packit Service 724aca
		err = true;
Packit Service 724aca
		goto label_return;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	tcache_t *tcache = tcache_create_explicit(tsd);
Packit Service 724aca
	if (tcache == NULL) {
Packit Service 724aca
		err = true;
Packit Service 724aca
		goto label_return;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	tcaches_t *elm;
Packit Service 724aca
	malloc_mutex_lock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
	if (tcaches_avail != NULL) {
Packit Service 724aca
		elm = tcaches_avail;
Packit Service 724aca
		tcaches_avail = tcaches_avail->next;
Packit Service 724aca
		elm->tcache = tcache;
Packit Service 724aca
		*r_ind = (unsigned)(elm - tcaches);
Packit Service 724aca
	} else {
Packit Service 724aca
		elm = &tcaches[tcaches_past];
Packit Service 724aca
		elm->tcache = tcache;
Packit Service 724aca
		*r_ind = tcaches_past;
Packit Service 724aca
		tcaches_past++;
Packit Service 724aca
	}
Packit Service 724aca
	malloc_mutex_unlock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
Packit Service 724aca
	err = false;
Packit Service 724aca
label_return:
Packit Service 724aca
	witness_assert_depth(tsdn_witness_tsdp_get(tsd_tsdn(tsd)), 0);
Packit Service 724aca
	return err;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
static tcache_t *
Packit Service 724aca
tcaches_elm_remove(tsd_t *tsd, tcaches_t *elm, bool allow_reinit) {
Packit Service 724aca
	malloc_mutex_assert_owner(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
Packit Service 724aca
	if (elm->tcache == NULL) {
Packit Service 724aca
		return NULL;
Packit Service 724aca
	}
Packit Service 724aca
	tcache_t *tcache = elm->tcache;
Packit Service 724aca
	if (allow_reinit) {
Packit Service 724aca
		elm->tcache = TCACHES_ELM_NEED_REINIT;
Packit Service 724aca
	} else {
Packit Service 724aca
		elm->tcache = NULL;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	if (tcache == TCACHES_ELM_NEED_REINIT) {
Packit Service 724aca
		return NULL;
Packit Service 724aca
	}
Packit Service 724aca
	return tcache;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcaches_flush(tsd_t *tsd, unsigned ind) {
Packit Service 724aca
	malloc_mutex_lock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
	tcache_t *tcache = tcaches_elm_remove(tsd, &tcaches[ind], true);
Packit Service 724aca
	malloc_mutex_unlock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
	if (tcache != NULL) {
Packit Service 724aca
		/* Destroy the tcache; recreate in tcaches_get() if needed. */
Packit Service 724aca
		tcache_destroy(tsd, tcache, false);
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcaches_destroy(tsd_t *tsd, unsigned ind) {
Packit Service 724aca
	malloc_mutex_lock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
	tcaches_t *elm = &tcaches[ind];
Packit Service 724aca
	tcache_t *tcache = tcaches_elm_remove(tsd, elm, false);
Packit Service 724aca
	elm->next = tcaches_avail;
Packit Service 724aca
	tcaches_avail = elm;
Packit Service 724aca
	malloc_mutex_unlock(tsd_tsdn(tsd), &tcaches_mtx);
Packit Service 724aca
	if (tcache != NULL) {
Packit Service 724aca
		tcache_destroy(tsd, tcache, false);
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
bool
Packit Service 724aca
tcache_boot(tsdn_t *tsdn) {
Packit Service 724aca
	/* If necessary, clamp opt_lg_tcache_max. */
Packit Service 724aca
	if (opt_lg_tcache_max < 0 || (ZU(1) << opt_lg_tcache_max) <
Packit Service 724aca
	    SC_SMALL_MAXCLASS) {
Packit Service 724aca
		tcache_maxclass = SC_SMALL_MAXCLASS;
Packit Service 724aca
	} else {
Packit Service 724aca
		tcache_maxclass = (ZU(1) << opt_lg_tcache_max);
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	if (malloc_mutex_init(&tcaches_mtx, "tcaches", WITNESS_RANK_TCACHES,
Packit Service 724aca
	    malloc_mutex_rank_exclusive)) {
Packit Service 724aca
		return true;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	nhbins = sz_size2index(tcache_maxclass) + 1;
Packit Service 724aca
Packit Service 724aca
	/* Initialize tcache_bin_info. */
Packit Service 724aca
	tcache_bin_info = (cache_bin_info_t *)base_alloc(tsdn, b0get(), nhbins
Packit Service 724aca
	    * sizeof(cache_bin_info_t), CACHELINE);
Packit Service 724aca
	if (tcache_bin_info == NULL) {
Packit Service 724aca
		return true;
Packit Service 724aca
	}
Packit Service 724aca
	stack_nelms = 0;
Packit Service 724aca
	unsigned i;
Packit Service 724aca
	for (i = 0; i < SC_NBINS; i++) {
Packit Service 724aca
		if ((bin_infos[i].nregs << 1) <= TCACHE_NSLOTS_SMALL_MIN) {
Packit Service 724aca
			tcache_bin_info[i].ncached_max =
Packit Service 724aca
			    TCACHE_NSLOTS_SMALL_MIN;
Packit Service 724aca
		} else if ((bin_infos[i].nregs << 1) <=
Packit Service 724aca
		    TCACHE_NSLOTS_SMALL_MAX) {
Packit Service 724aca
			tcache_bin_info[i].ncached_max =
Packit Service 724aca
			    (bin_infos[i].nregs << 1);
Packit Service 724aca
		} else {
Packit Service 724aca
			tcache_bin_info[i].ncached_max =
Packit Service 724aca
			    TCACHE_NSLOTS_SMALL_MAX;
Packit Service 724aca
		}
Packit Service 724aca
		stack_nelms += tcache_bin_info[i].ncached_max;
Packit Service 724aca
	}
Packit Service 724aca
	for (; i < nhbins; i++) {
Packit Service 724aca
		tcache_bin_info[i].ncached_max = TCACHE_NSLOTS_LARGE;
Packit Service 724aca
		stack_nelms += tcache_bin_info[i].ncached_max;
Packit Service 724aca
	}
Packit Service 724aca
Packit Service 724aca
	return false;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_prefork(tsdn_t *tsdn) {
Packit Service 724aca
	if (!config_prof && opt_tcache) {
Packit Service 724aca
		malloc_mutex_prefork(tsdn, &tcaches_mtx);
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_postfork_parent(tsdn_t *tsdn) {
Packit Service 724aca
	if (!config_prof && opt_tcache) {
Packit Service 724aca
		malloc_mutex_postfork_parent(tsdn, &tcaches_mtx);
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
void
Packit Service 724aca
tcache_postfork_child(tsdn_t *tsdn) {
Packit Service 724aca
	if (!config_prof && opt_tcache) {
Packit Service 724aca
		malloc_mutex_postfork_child(tsdn, &tcaches_mtx);
Packit Service 724aca
	}
Packit Service 724aca
}