#include "jemalloc/internal/jemalloc_preamble.h"
#include "jemalloc/internal/jemalloc_internal_includes.h"

/*
 * This file is logically part of the PA module.  While pa.c contains the core
 * allocator functionality, this file contains boring integration functionality;
 * things like the pre- and post- fork handlers, and stats merging for CTL
 * refreshes.
 */

void
pa_shard_prefork0(tsdn_t *tsdn, pa_shard_t *shard) {
	malloc_mutex_prefork(tsdn, &shard->decay_dirty.mtx);
	malloc_mutex_prefork(tsdn, &shard->decay_muzzy.mtx);
}

void
pa_shard_prefork2(tsdn_t *tsdn, pa_shard_t *shard) {
	ecache_grow_prefork(tsdn, &shard->ecache_grow);
}

void
pa_shard_prefork3(tsdn_t *tsdn, pa_shard_t *shard) {
	ecache_prefork(tsdn, &shard->ecache_dirty);
	ecache_prefork(tsdn, &shard->ecache_muzzy);
	ecache_prefork(tsdn, &shard->ecache_retained);
}


void
pa_shard_prefork4(tsdn_t *tsdn, pa_shard_t *shard) {
	edata_cache_prefork(tsdn, &shard->edata_cache);
}

void
pa_shard_postfork_parent(tsdn_t *tsdn, pa_shard_t *shard) {
	edata_cache_postfork_parent(tsdn, &shard->edata_cache);
	ecache_postfork_parent(tsdn, &shard->ecache_dirty);
	ecache_postfork_parent(tsdn, &shard->ecache_muzzy);
	ecache_postfork_parent(tsdn, &shard->ecache_retained);
	ecache_grow_postfork_parent(tsdn, &shard->ecache_grow);
	malloc_mutex_postfork_parent(tsdn, &shard->decay_dirty.mtx);
	malloc_mutex_postfork_parent(tsdn, &shard->decay_muzzy.mtx);
}

void
pa_shard_postfork_child(tsdn_t *tsdn, pa_shard_t *shard) {
	edata_cache_postfork_child(tsdn, &shard->edata_cache);
	ecache_postfork_child(tsdn, &shard->ecache_dirty);
	ecache_postfork_child(tsdn, &shard->ecache_muzzy);
	ecache_postfork_child(tsdn, &shard->ecache_retained);
	ecache_grow_postfork_child(tsdn, &shard->ecache_grow);
	malloc_mutex_postfork_child(tsdn, &shard->decay_dirty.mtx);
	malloc_mutex_postfork_child(tsdn, &shard->decay_muzzy.mtx);
}

void
pa_shard_basic_stats_merge(pa_shard_t *shard, size_t *nactive, size_t *ndirty,
    size_t *nmuzzy) {
	*nactive += atomic_load_zu(&shard->nactive, ATOMIC_RELAXED);
	*ndirty += ecache_npages_get(&shard->ecache_dirty);
	*nmuzzy += ecache_npages_get(&shard->ecache_muzzy);
}

void
pa_shard_stats_merge(tsdn_t *tsdn, pa_shard_t *shard,
    pa_shard_stats_t *shard_stats_out, pa_extent_stats_t *extent_stats_out,
    size_t *resident) {
	cassert(config_stats);

	shard_stats_out->retained +=
	    ecache_npages_get(&shard->ecache_retained) << LG_PAGE;
	shard_stats_out->edata_avail += atomic_load_zu(
	    &shard->edata_cache.count, ATOMIC_RELAXED);

	size_t resident_pgs = 0;
	resident_pgs += atomic_load_zu(&shard->nactive, ATOMIC_RELAXED);
	resident_pgs += ecache_npages_get(&shard->ecache_dirty);
	*resident += (resident_pgs << LG_PAGE);

	/* Dirty decay stats */
	locked_inc_u64_unsynchronized(
	    &shard_stats_out->decay_dirty.npurge,
	    locked_read_u64(tsdn, LOCKEDINT_MTX(*shard->stats_mtx),
	    &shard->stats->decay_dirty.npurge));
	locked_inc_u64_unsynchronized(
	    &shard_stats_out->decay_dirty.nmadvise,
	    locked_read_u64(tsdn, LOCKEDINT_MTX(*shard->stats_mtx),
	    &shard->stats->decay_dirty.nmadvise));
	locked_inc_u64_unsynchronized(
	    &shard_stats_out->decay_dirty.purged,
	    locked_read_u64(tsdn, LOCKEDINT_MTX(*shard->stats_mtx),
	    &shard->stats->decay_dirty.purged));

	/* Muzzy decay stats */
	locked_inc_u64_unsynchronized(
	    &shard_stats_out->decay_muzzy.npurge,
	    locked_read_u64(tsdn, LOCKEDINT_MTX(*shard->stats_mtx),
	    &shard->stats->decay_muzzy.npurge));
	locked_inc_u64_unsynchronized(
	    &shard_stats_out->decay_muzzy.nmadvise,
	    locked_read_u64(tsdn, LOCKEDINT_MTX(*shard->stats_mtx),
	    &shard->stats->decay_muzzy.nmadvise));
	locked_inc_u64_unsynchronized(
	    &shard_stats_out->decay_muzzy.purged,
	    locked_read_u64(tsdn, LOCKEDINT_MTX(*shard->stats_mtx),
	    &shard->stats->decay_muzzy.purged));

	atomic_load_add_store_zu(&shard_stats_out->abandoned_vm,
	    atomic_load_zu(&shard->stats->abandoned_vm, ATOMIC_RELAXED));

	for (pszind_t i = 0; i < SC_NPSIZES; i++) {
		size_t dirty, muzzy, retained, dirty_bytes, muzzy_bytes,
		    retained_bytes;
		dirty = ecache_nextents_get(&shard->ecache_dirty, i);
		muzzy = ecache_nextents_get(&shard->ecache_muzzy, i);
		retained = ecache_nextents_get(&shard->ecache_retained, i);
		dirty_bytes = ecache_nbytes_get(&shard->ecache_dirty, i);
		muzzy_bytes = ecache_nbytes_get(&shard->ecache_muzzy, i);
		retained_bytes = ecache_nbytes_get(&shard->ecache_retained, i);

		extent_stats_out[i].ndirty = dirty;
		extent_stats_out[i].nmuzzy = muzzy;
		extent_stats_out[i].nretained = retained;
		extent_stats_out[i].dirty_bytes = dirty_bytes;
		extent_stats_out[i].muzzy_bytes = muzzy_bytes;
		extent_stats_out[i].retained_bytes = retained_bytes;
	}
}

static void
pa_shard_mtx_stats_read_single(tsdn_t *tsdn, mutex_prof_data_t *mutex_prof_data,
    malloc_mutex_t *mtx, int ind) {
	malloc_mutex_lock(tsdn, mtx);
	malloc_mutex_prof_read(tsdn, &mutex_prof_data[ind], mtx);
	malloc_mutex_unlock(tsdn, mtx);
}

void
pa_shard_mtx_stats_read(tsdn_t *tsdn, pa_shard_t *shard,
    mutex_prof_data_t mutex_prof_data[mutex_prof_num_arena_mutexes]) {
	pa_shard_mtx_stats_read_single(tsdn, mutex_prof_data,
	    &shard->edata_cache.mtx, arena_prof_mutex_extent_avail);
	pa_shard_mtx_stats_read_single(tsdn, mutex_prof_data,
	    &shard->ecache_dirty.mtx, arena_prof_mutex_extents_dirty);
	pa_shard_mtx_stats_read_single(tsdn, mutex_prof_data,
	    &shard->ecache_muzzy.mtx, arena_prof_mutex_extents_muzzy);
	pa_shard_mtx_stats_read_single(tsdn, mutex_prof_data,
	    &shard->ecache_retained.mtx, arena_prof_mutex_extents_retained);
	pa_shard_mtx_stats_read_single(tsdn, mutex_prof_data,
	    &shard->decay_dirty.mtx, arena_prof_mutex_decay_dirty);
	pa_shard_mtx_stats_read_single(tsdn, mutex_prof_data,
	    &shard->decay_muzzy.mtx, arena_prof_mutex_decay_muzzy);
}