xref: /aosp_15_r20/external/jemalloc_new/src/prof.c (revision 1208bc7e437ced7eb82efac44ba17e3beba411da)
1*1208bc7eSAndroid Build Coastguard Worker #define JEMALLOC_PROF_C_
2*1208bc7eSAndroid Build Coastguard Worker #include "jemalloc/internal/jemalloc_preamble.h"
3*1208bc7eSAndroid Build Coastguard Worker #include "jemalloc/internal/jemalloc_internal_includes.h"
4*1208bc7eSAndroid Build Coastguard Worker 
5*1208bc7eSAndroid Build Coastguard Worker #include "jemalloc/internal/assert.h"
6*1208bc7eSAndroid Build Coastguard Worker #include "jemalloc/internal/ckh.h"
7*1208bc7eSAndroid Build Coastguard Worker #include "jemalloc/internal/hash.h"
8*1208bc7eSAndroid Build Coastguard Worker #include "jemalloc/internal/malloc_io.h"
9*1208bc7eSAndroid Build Coastguard Worker #include "jemalloc/internal/mutex.h"
10*1208bc7eSAndroid Build Coastguard Worker 
11*1208bc7eSAndroid Build Coastguard Worker /******************************************************************************/
12*1208bc7eSAndroid Build Coastguard Worker 
13*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF_LIBUNWIND
14*1208bc7eSAndroid Build Coastguard Worker #define UNW_LOCAL_ONLY
15*1208bc7eSAndroid Build Coastguard Worker #include <libunwind.h>
16*1208bc7eSAndroid Build Coastguard Worker #endif
17*1208bc7eSAndroid Build Coastguard Worker 
18*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF_LIBGCC
19*1208bc7eSAndroid Build Coastguard Worker /*
20*1208bc7eSAndroid Build Coastguard Worker  * We have a circular dependency -- jemalloc_internal.h tells us if we should
21*1208bc7eSAndroid Build Coastguard Worker  * use libgcc's unwinding functionality, but after we've included that, we've
22*1208bc7eSAndroid Build Coastguard Worker  * already hooked _Unwind_Backtrace.  We'll temporarily disable hooking.
23*1208bc7eSAndroid Build Coastguard Worker  */
24*1208bc7eSAndroid Build Coastguard Worker #undef _Unwind_Backtrace
25*1208bc7eSAndroid Build Coastguard Worker #include <unwind.h>
26*1208bc7eSAndroid Build Coastguard Worker #define _Unwind_Backtrace JEMALLOC_HOOK(_Unwind_Backtrace, hooks_libc_hook)
27*1208bc7eSAndroid Build Coastguard Worker #endif
28*1208bc7eSAndroid Build Coastguard Worker 
29*1208bc7eSAndroid Build Coastguard Worker /******************************************************************************/
30*1208bc7eSAndroid Build Coastguard Worker /* Data. */
31*1208bc7eSAndroid Build Coastguard Worker 
32*1208bc7eSAndroid Build Coastguard Worker bool		opt_prof = false;
33*1208bc7eSAndroid Build Coastguard Worker bool		opt_prof_active = true;
34*1208bc7eSAndroid Build Coastguard Worker bool		opt_prof_thread_active_init = true;
35*1208bc7eSAndroid Build Coastguard Worker size_t		opt_lg_prof_sample = LG_PROF_SAMPLE_DEFAULT;
36*1208bc7eSAndroid Build Coastguard Worker ssize_t		opt_lg_prof_interval = LG_PROF_INTERVAL_DEFAULT;
37*1208bc7eSAndroid Build Coastguard Worker bool		opt_prof_gdump = false;
38*1208bc7eSAndroid Build Coastguard Worker bool		opt_prof_final = false;
39*1208bc7eSAndroid Build Coastguard Worker bool		opt_prof_leak = false;
40*1208bc7eSAndroid Build Coastguard Worker bool		opt_prof_accum = false;
41*1208bc7eSAndroid Build Coastguard Worker char		opt_prof_prefix[
42*1208bc7eSAndroid Build Coastguard Worker     /* Minimize memory bloat for non-prof builds. */
43*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF
44*1208bc7eSAndroid Build Coastguard Worker     PATH_MAX +
45*1208bc7eSAndroid Build Coastguard Worker #endif
46*1208bc7eSAndroid Build Coastguard Worker     1];
47*1208bc7eSAndroid Build Coastguard Worker 
48*1208bc7eSAndroid Build Coastguard Worker /*
49*1208bc7eSAndroid Build Coastguard Worker  * Initialized as opt_prof_active, and accessed via
50*1208bc7eSAndroid Build Coastguard Worker  * prof_active_[gs]et{_unlocked,}().
51*1208bc7eSAndroid Build Coastguard Worker  */
52*1208bc7eSAndroid Build Coastguard Worker bool			prof_active;
53*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	prof_active_mtx;
54*1208bc7eSAndroid Build Coastguard Worker 
55*1208bc7eSAndroid Build Coastguard Worker /*
56*1208bc7eSAndroid Build Coastguard Worker  * Initialized as opt_prof_thread_active_init, and accessed via
57*1208bc7eSAndroid Build Coastguard Worker  * prof_thread_active_init_[gs]et().
58*1208bc7eSAndroid Build Coastguard Worker  */
59*1208bc7eSAndroid Build Coastguard Worker static bool		prof_thread_active_init;
60*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	prof_thread_active_init_mtx;
61*1208bc7eSAndroid Build Coastguard Worker 
62*1208bc7eSAndroid Build Coastguard Worker /*
63*1208bc7eSAndroid Build Coastguard Worker  * Initialized as opt_prof_gdump, and accessed via
64*1208bc7eSAndroid Build Coastguard Worker  * prof_gdump_[gs]et{_unlocked,}().
65*1208bc7eSAndroid Build Coastguard Worker  */
66*1208bc7eSAndroid Build Coastguard Worker bool			prof_gdump_val;
67*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	prof_gdump_mtx;
68*1208bc7eSAndroid Build Coastguard Worker 
69*1208bc7eSAndroid Build Coastguard Worker uint64_t	prof_interval = 0;
70*1208bc7eSAndroid Build Coastguard Worker 
71*1208bc7eSAndroid Build Coastguard Worker size_t		lg_prof_sample;
72*1208bc7eSAndroid Build Coastguard Worker 
73*1208bc7eSAndroid Build Coastguard Worker /*
74*1208bc7eSAndroid Build Coastguard Worker  * Table of mutexes that are shared among gctx's.  These are leaf locks, so
75*1208bc7eSAndroid Build Coastguard Worker  * there is no problem with using them for more than one gctx at the same time.
76*1208bc7eSAndroid Build Coastguard Worker  * The primary motivation for this sharing though is that gctx's are ephemeral,
77*1208bc7eSAndroid Build Coastguard Worker  * and destroying mutexes causes complications for systems that allocate when
78*1208bc7eSAndroid Build Coastguard Worker  * creating/destroying mutexes.
79*1208bc7eSAndroid Build Coastguard Worker  */
80*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	*gctx_locks;
81*1208bc7eSAndroid Build Coastguard Worker static atomic_u_t	cum_gctxs; /* Atomic counter. */
82*1208bc7eSAndroid Build Coastguard Worker 
83*1208bc7eSAndroid Build Coastguard Worker /*
84*1208bc7eSAndroid Build Coastguard Worker  * Table of mutexes that are shared among tdata's.  No operations require
85*1208bc7eSAndroid Build Coastguard Worker  * holding multiple tdata locks, so there is no problem with using them for more
86*1208bc7eSAndroid Build Coastguard Worker  * than one tdata at the same time, even though a gctx lock may be acquired
87*1208bc7eSAndroid Build Coastguard Worker  * while holding a tdata lock.
88*1208bc7eSAndroid Build Coastguard Worker  */
89*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	*tdata_locks;
90*1208bc7eSAndroid Build Coastguard Worker 
91*1208bc7eSAndroid Build Coastguard Worker /*
92*1208bc7eSAndroid Build Coastguard Worker  * Global hash of (prof_bt_t *)-->(prof_gctx_t *).  This is the master data
93*1208bc7eSAndroid Build Coastguard Worker  * structure that knows about all backtraces currently captured.
94*1208bc7eSAndroid Build Coastguard Worker  */
95*1208bc7eSAndroid Build Coastguard Worker static ckh_t		bt2gctx;
96*1208bc7eSAndroid Build Coastguard Worker /* Non static to enable profiling. */
97*1208bc7eSAndroid Build Coastguard Worker malloc_mutex_t		bt2gctx_mtx;
98*1208bc7eSAndroid Build Coastguard Worker 
99*1208bc7eSAndroid Build Coastguard Worker /*
100*1208bc7eSAndroid Build Coastguard Worker  * Tree of all extant prof_tdata_t structures, regardless of state,
101*1208bc7eSAndroid Build Coastguard Worker  * {attached,detached,expired}.
102*1208bc7eSAndroid Build Coastguard Worker  */
103*1208bc7eSAndroid Build Coastguard Worker static prof_tdata_tree_t	tdatas;
104*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	tdatas_mtx;
105*1208bc7eSAndroid Build Coastguard Worker 
106*1208bc7eSAndroid Build Coastguard Worker static uint64_t		next_thr_uid;
107*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	next_thr_uid_mtx;
108*1208bc7eSAndroid Build Coastguard Worker 
109*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	prof_dump_seq_mtx;
110*1208bc7eSAndroid Build Coastguard Worker static uint64_t		prof_dump_seq;
111*1208bc7eSAndroid Build Coastguard Worker static uint64_t		prof_dump_iseq;
112*1208bc7eSAndroid Build Coastguard Worker static uint64_t		prof_dump_mseq;
113*1208bc7eSAndroid Build Coastguard Worker static uint64_t		prof_dump_useq;
114*1208bc7eSAndroid Build Coastguard Worker 
115*1208bc7eSAndroid Build Coastguard Worker /*
116*1208bc7eSAndroid Build Coastguard Worker  * This buffer is rather large for stack allocation, so use a single buffer for
117*1208bc7eSAndroid Build Coastguard Worker  * all profile dumps.
118*1208bc7eSAndroid Build Coastguard Worker  */
119*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t	prof_dump_mtx;
120*1208bc7eSAndroid Build Coastguard Worker static char		prof_dump_buf[
121*1208bc7eSAndroid Build Coastguard Worker     /* Minimize memory bloat for non-prof builds. */
122*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF
123*1208bc7eSAndroid Build Coastguard Worker     PROF_DUMP_BUFSIZE
124*1208bc7eSAndroid Build Coastguard Worker #else
125*1208bc7eSAndroid Build Coastguard Worker     1
126*1208bc7eSAndroid Build Coastguard Worker #endif
127*1208bc7eSAndroid Build Coastguard Worker ];
128*1208bc7eSAndroid Build Coastguard Worker static size_t		prof_dump_buf_end;
129*1208bc7eSAndroid Build Coastguard Worker static int		prof_dump_fd;
130*1208bc7eSAndroid Build Coastguard Worker 
131*1208bc7eSAndroid Build Coastguard Worker /* Do not dump any profiles until bootstrapping is complete. */
132*1208bc7eSAndroid Build Coastguard Worker static bool		prof_booted = false;
133*1208bc7eSAndroid Build Coastguard Worker 
134*1208bc7eSAndroid Build Coastguard Worker /******************************************************************************/
135*1208bc7eSAndroid Build Coastguard Worker /*
136*1208bc7eSAndroid Build Coastguard Worker  * Function prototypes for static functions that are referenced prior to
137*1208bc7eSAndroid Build Coastguard Worker  * definition.
138*1208bc7eSAndroid Build Coastguard Worker  */
139*1208bc7eSAndroid Build Coastguard Worker 
140*1208bc7eSAndroid Build Coastguard Worker static bool	prof_tctx_should_destroy(tsdn_t *tsdn, prof_tctx_t *tctx);
141*1208bc7eSAndroid Build Coastguard Worker static void	prof_tctx_destroy(tsd_t *tsd, prof_tctx_t *tctx);
142*1208bc7eSAndroid Build Coastguard Worker static bool	prof_tdata_should_destroy(tsdn_t *tsdn, prof_tdata_t *tdata,
143*1208bc7eSAndroid Build Coastguard Worker     bool even_if_attached);
144*1208bc7eSAndroid Build Coastguard Worker static void	prof_tdata_destroy(tsd_t *tsd, prof_tdata_t *tdata,
145*1208bc7eSAndroid Build Coastguard Worker     bool even_if_attached);
146*1208bc7eSAndroid Build Coastguard Worker static char	*prof_thread_name_alloc(tsdn_t *tsdn, const char *thread_name);
147*1208bc7eSAndroid Build Coastguard Worker 
148*1208bc7eSAndroid Build Coastguard Worker /******************************************************************************/
149*1208bc7eSAndroid Build Coastguard Worker /* Red-black trees. */
150*1208bc7eSAndroid Build Coastguard Worker 
151*1208bc7eSAndroid Build Coastguard Worker static int
prof_tctx_comp(const prof_tctx_t * a,const prof_tctx_t * b)152*1208bc7eSAndroid Build Coastguard Worker prof_tctx_comp(const prof_tctx_t *a, const prof_tctx_t *b) {
153*1208bc7eSAndroid Build Coastguard Worker 	uint64_t a_thr_uid = a->thr_uid;
154*1208bc7eSAndroid Build Coastguard Worker 	uint64_t b_thr_uid = b->thr_uid;
155*1208bc7eSAndroid Build Coastguard Worker 	int ret = (a_thr_uid > b_thr_uid) - (a_thr_uid < b_thr_uid);
156*1208bc7eSAndroid Build Coastguard Worker 	if (ret == 0) {
157*1208bc7eSAndroid Build Coastguard Worker 		uint64_t a_thr_discrim = a->thr_discrim;
158*1208bc7eSAndroid Build Coastguard Worker 		uint64_t b_thr_discrim = b->thr_discrim;
159*1208bc7eSAndroid Build Coastguard Worker 		ret = (a_thr_discrim > b_thr_discrim) - (a_thr_discrim <
160*1208bc7eSAndroid Build Coastguard Worker 		    b_thr_discrim);
161*1208bc7eSAndroid Build Coastguard Worker 		if (ret == 0) {
162*1208bc7eSAndroid Build Coastguard Worker 			uint64_t a_tctx_uid = a->tctx_uid;
163*1208bc7eSAndroid Build Coastguard Worker 			uint64_t b_tctx_uid = b->tctx_uid;
164*1208bc7eSAndroid Build Coastguard Worker 			ret = (a_tctx_uid > b_tctx_uid) - (a_tctx_uid <
165*1208bc7eSAndroid Build Coastguard Worker 			    b_tctx_uid);
166*1208bc7eSAndroid Build Coastguard Worker 		}
167*1208bc7eSAndroid Build Coastguard Worker 	}
168*1208bc7eSAndroid Build Coastguard Worker 	return ret;
169*1208bc7eSAndroid Build Coastguard Worker }
170*1208bc7eSAndroid Build Coastguard Worker 
rb_gen(static UNUSED,tctx_tree_,prof_tctx_tree_t,prof_tctx_t,tctx_link,prof_tctx_comp)171*1208bc7eSAndroid Build Coastguard Worker rb_gen(static UNUSED, tctx_tree_, prof_tctx_tree_t, prof_tctx_t,
172*1208bc7eSAndroid Build Coastguard Worker     tctx_link, prof_tctx_comp)
173*1208bc7eSAndroid Build Coastguard Worker 
174*1208bc7eSAndroid Build Coastguard Worker static int
175*1208bc7eSAndroid Build Coastguard Worker prof_gctx_comp(const prof_gctx_t *a, const prof_gctx_t *b) {
176*1208bc7eSAndroid Build Coastguard Worker 	unsigned a_len = a->bt.len;
177*1208bc7eSAndroid Build Coastguard Worker 	unsigned b_len = b->bt.len;
178*1208bc7eSAndroid Build Coastguard Worker 	unsigned comp_len = (a_len < b_len) ? a_len : b_len;
179*1208bc7eSAndroid Build Coastguard Worker 	int ret = memcmp(a->bt.vec, b->bt.vec, comp_len * sizeof(void *));
180*1208bc7eSAndroid Build Coastguard Worker 	if (ret == 0) {
181*1208bc7eSAndroid Build Coastguard Worker 		ret = (a_len > b_len) - (a_len < b_len);
182*1208bc7eSAndroid Build Coastguard Worker 	}
183*1208bc7eSAndroid Build Coastguard Worker 	return ret;
184*1208bc7eSAndroid Build Coastguard Worker }
185*1208bc7eSAndroid Build Coastguard Worker 
rb_gen(static UNUSED,gctx_tree_,prof_gctx_tree_t,prof_gctx_t,dump_link,prof_gctx_comp)186*1208bc7eSAndroid Build Coastguard Worker rb_gen(static UNUSED, gctx_tree_, prof_gctx_tree_t, prof_gctx_t, dump_link,
187*1208bc7eSAndroid Build Coastguard Worker     prof_gctx_comp)
188*1208bc7eSAndroid Build Coastguard Worker 
189*1208bc7eSAndroid Build Coastguard Worker static int
190*1208bc7eSAndroid Build Coastguard Worker prof_tdata_comp(const prof_tdata_t *a, const prof_tdata_t *b) {
191*1208bc7eSAndroid Build Coastguard Worker 	int ret;
192*1208bc7eSAndroid Build Coastguard Worker 	uint64_t a_uid = a->thr_uid;
193*1208bc7eSAndroid Build Coastguard Worker 	uint64_t b_uid = b->thr_uid;
194*1208bc7eSAndroid Build Coastguard Worker 
195*1208bc7eSAndroid Build Coastguard Worker 	ret = ((a_uid > b_uid) - (a_uid < b_uid));
196*1208bc7eSAndroid Build Coastguard Worker 	if (ret == 0) {
197*1208bc7eSAndroid Build Coastguard Worker 		uint64_t a_discrim = a->thr_discrim;
198*1208bc7eSAndroid Build Coastguard Worker 		uint64_t b_discrim = b->thr_discrim;
199*1208bc7eSAndroid Build Coastguard Worker 
200*1208bc7eSAndroid Build Coastguard Worker 		ret = ((a_discrim > b_discrim) - (a_discrim < b_discrim));
201*1208bc7eSAndroid Build Coastguard Worker 	}
202*1208bc7eSAndroid Build Coastguard Worker 	return ret;
203*1208bc7eSAndroid Build Coastguard Worker }
204*1208bc7eSAndroid Build Coastguard Worker 
rb_gen(static UNUSED,tdata_tree_,prof_tdata_tree_t,prof_tdata_t,tdata_link,prof_tdata_comp)205*1208bc7eSAndroid Build Coastguard Worker rb_gen(static UNUSED, tdata_tree_, prof_tdata_tree_t, prof_tdata_t, tdata_link,
206*1208bc7eSAndroid Build Coastguard Worker     prof_tdata_comp)
207*1208bc7eSAndroid Build Coastguard Worker 
208*1208bc7eSAndroid Build Coastguard Worker /******************************************************************************/
209*1208bc7eSAndroid Build Coastguard Worker 
210*1208bc7eSAndroid Build Coastguard Worker void
211*1208bc7eSAndroid Build Coastguard Worker prof_alloc_rollback(tsd_t *tsd, prof_tctx_t *tctx, bool updated) {
212*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
213*1208bc7eSAndroid Build Coastguard Worker 
214*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
215*1208bc7eSAndroid Build Coastguard Worker 
216*1208bc7eSAndroid Build Coastguard Worker 	if (updated) {
217*1208bc7eSAndroid Build Coastguard Worker 		/*
218*1208bc7eSAndroid Build Coastguard Worker 		 * Compute a new sample threshold.  This isn't very important in
219*1208bc7eSAndroid Build Coastguard Worker 		 * practice, because this function is rarely executed, so the
220*1208bc7eSAndroid Build Coastguard Worker 		 * potential for sample bias is minimal except in contrived
221*1208bc7eSAndroid Build Coastguard Worker 		 * programs.
222*1208bc7eSAndroid Build Coastguard Worker 		 */
223*1208bc7eSAndroid Build Coastguard Worker 		tdata = prof_tdata_get(tsd, true);
224*1208bc7eSAndroid Build Coastguard Worker 		if (tdata != NULL) {
225*1208bc7eSAndroid Build Coastguard Worker 			prof_sample_threshold_update(tdata);
226*1208bc7eSAndroid Build Coastguard Worker 		}
227*1208bc7eSAndroid Build Coastguard Worker 	}
228*1208bc7eSAndroid Build Coastguard Worker 
229*1208bc7eSAndroid Build Coastguard Worker 	if ((uintptr_t)tctx > (uintptr_t)1U) {
230*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsd_tsdn(tsd), tctx->tdata->lock);
231*1208bc7eSAndroid Build Coastguard Worker 		tctx->prepared = false;
232*1208bc7eSAndroid Build Coastguard Worker 		if (prof_tctx_should_destroy(tsd_tsdn(tsd), tctx)) {
233*1208bc7eSAndroid Build Coastguard Worker 			prof_tctx_destroy(tsd, tctx);
234*1208bc7eSAndroid Build Coastguard Worker 		} else {
235*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_unlock(tsd_tsdn(tsd), tctx->tdata->lock);
236*1208bc7eSAndroid Build Coastguard Worker 		}
237*1208bc7eSAndroid Build Coastguard Worker 	}
238*1208bc7eSAndroid Build Coastguard Worker }
239*1208bc7eSAndroid Build Coastguard Worker 
240*1208bc7eSAndroid Build Coastguard Worker void
prof_malloc_sample_object(tsdn_t * tsdn,const void * ptr,size_t usize,prof_tctx_t * tctx)241*1208bc7eSAndroid Build Coastguard Worker prof_malloc_sample_object(tsdn_t *tsdn, const void *ptr, size_t usize,
242*1208bc7eSAndroid Build Coastguard Worker     prof_tctx_t *tctx) {
243*1208bc7eSAndroid Build Coastguard Worker 	prof_tctx_set(tsdn, ptr, usize, NULL, tctx);
244*1208bc7eSAndroid Build Coastguard Worker 
245*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, tctx->tdata->lock);
246*1208bc7eSAndroid Build Coastguard Worker 	tctx->cnts.curobjs++;
247*1208bc7eSAndroid Build Coastguard Worker 	tctx->cnts.curbytes += usize;
248*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof_accum) {
249*1208bc7eSAndroid Build Coastguard Worker 		tctx->cnts.accumobjs++;
250*1208bc7eSAndroid Build Coastguard Worker 		tctx->cnts.accumbytes += usize;
251*1208bc7eSAndroid Build Coastguard Worker 	}
252*1208bc7eSAndroid Build Coastguard Worker 	tctx->prepared = false;
253*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, tctx->tdata->lock);
254*1208bc7eSAndroid Build Coastguard Worker }
255*1208bc7eSAndroid Build Coastguard Worker 
256*1208bc7eSAndroid Build Coastguard Worker void
prof_free_sampled_object(tsd_t * tsd,size_t usize,prof_tctx_t * tctx)257*1208bc7eSAndroid Build Coastguard Worker prof_free_sampled_object(tsd_t *tsd, size_t usize, prof_tctx_t *tctx) {
258*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), tctx->tdata->lock);
259*1208bc7eSAndroid Build Coastguard Worker 	assert(tctx->cnts.curobjs > 0);
260*1208bc7eSAndroid Build Coastguard Worker 	assert(tctx->cnts.curbytes >= usize);
261*1208bc7eSAndroid Build Coastguard Worker 	tctx->cnts.curobjs--;
262*1208bc7eSAndroid Build Coastguard Worker 	tctx->cnts.curbytes -= usize;
263*1208bc7eSAndroid Build Coastguard Worker 
264*1208bc7eSAndroid Build Coastguard Worker 	if (prof_tctx_should_destroy(tsd_tsdn(tsd), tctx)) {
265*1208bc7eSAndroid Build Coastguard Worker 		prof_tctx_destroy(tsd, tctx);
266*1208bc7eSAndroid Build Coastguard Worker 	} else {
267*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), tctx->tdata->lock);
268*1208bc7eSAndroid Build Coastguard Worker 	}
269*1208bc7eSAndroid Build Coastguard Worker }
270*1208bc7eSAndroid Build Coastguard Worker 
271*1208bc7eSAndroid Build Coastguard Worker void
bt_init(prof_bt_t * bt,void ** vec)272*1208bc7eSAndroid Build Coastguard Worker bt_init(prof_bt_t *bt, void **vec) {
273*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
274*1208bc7eSAndroid Build Coastguard Worker 
275*1208bc7eSAndroid Build Coastguard Worker 	bt->vec = vec;
276*1208bc7eSAndroid Build Coastguard Worker 	bt->len = 0;
277*1208bc7eSAndroid Build Coastguard Worker }
278*1208bc7eSAndroid Build Coastguard Worker 
279*1208bc7eSAndroid Build Coastguard Worker static void
prof_enter(tsd_t * tsd,prof_tdata_t * tdata)280*1208bc7eSAndroid Build Coastguard Worker prof_enter(tsd_t *tsd, prof_tdata_t *tdata) {
281*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
282*1208bc7eSAndroid Build Coastguard Worker 	assert(tdata == prof_tdata_get(tsd, false));
283*1208bc7eSAndroid Build Coastguard Worker 
284*1208bc7eSAndroid Build Coastguard Worker 	if (tdata != NULL) {
285*1208bc7eSAndroid Build Coastguard Worker 		assert(!tdata->enq);
286*1208bc7eSAndroid Build Coastguard Worker 		tdata->enq = true;
287*1208bc7eSAndroid Build Coastguard Worker 	}
288*1208bc7eSAndroid Build Coastguard Worker 
289*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &bt2gctx_mtx);
290*1208bc7eSAndroid Build Coastguard Worker }
291*1208bc7eSAndroid Build Coastguard Worker 
292*1208bc7eSAndroid Build Coastguard Worker static void
prof_leave(tsd_t * tsd,prof_tdata_t * tdata)293*1208bc7eSAndroid Build Coastguard Worker prof_leave(tsd_t *tsd, prof_tdata_t *tdata) {
294*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
295*1208bc7eSAndroid Build Coastguard Worker 	assert(tdata == prof_tdata_get(tsd, false));
296*1208bc7eSAndroid Build Coastguard Worker 
297*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &bt2gctx_mtx);
298*1208bc7eSAndroid Build Coastguard Worker 
299*1208bc7eSAndroid Build Coastguard Worker 	if (tdata != NULL) {
300*1208bc7eSAndroid Build Coastguard Worker 		bool idump, gdump;
301*1208bc7eSAndroid Build Coastguard Worker 
302*1208bc7eSAndroid Build Coastguard Worker 		assert(tdata->enq);
303*1208bc7eSAndroid Build Coastguard Worker 		tdata->enq = false;
304*1208bc7eSAndroid Build Coastguard Worker 		idump = tdata->enq_idump;
305*1208bc7eSAndroid Build Coastguard Worker 		tdata->enq_idump = false;
306*1208bc7eSAndroid Build Coastguard Worker 		gdump = tdata->enq_gdump;
307*1208bc7eSAndroid Build Coastguard Worker 		tdata->enq_gdump = false;
308*1208bc7eSAndroid Build Coastguard Worker 
309*1208bc7eSAndroid Build Coastguard Worker 		if (idump) {
310*1208bc7eSAndroid Build Coastguard Worker 			prof_idump(tsd_tsdn(tsd));
311*1208bc7eSAndroid Build Coastguard Worker 		}
312*1208bc7eSAndroid Build Coastguard Worker 		if (gdump) {
313*1208bc7eSAndroid Build Coastguard Worker 			prof_gdump(tsd_tsdn(tsd));
314*1208bc7eSAndroid Build Coastguard Worker 		}
315*1208bc7eSAndroid Build Coastguard Worker 	}
316*1208bc7eSAndroid Build Coastguard Worker }
317*1208bc7eSAndroid Build Coastguard Worker 
318*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF_LIBUNWIND
319*1208bc7eSAndroid Build Coastguard Worker void
prof_backtrace(prof_bt_t * bt)320*1208bc7eSAndroid Build Coastguard Worker prof_backtrace(prof_bt_t *bt) {
321*1208bc7eSAndroid Build Coastguard Worker 	int nframes;
322*1208bc7eSAndroid Build Coastguard Worker 
323*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
324*1208bc7eSAndroid Build Coastguard Worker 	assert(bt->len == 0);
325*1208bc7eSAndroid Build Coastguard Worker 	assert(bt->vec != NULL);
326*1208bc7eSAndroid Build Coastguard Worker 
327*1208bc7eSAndroid Build Coastguard Worker 	nframes = unw_backtrace(bt->vec, PROF_BT_MAX);
328*1208bc7eSAndroid Build Coastguard Worker 	if (nframes <= 0) {
329*1208bc7eSAndroid Build Coastguard Worker 		return;
330*1208bc7eSAndroid Build Coastguard Worker 	}
331*1208bc7eSAndroid Build Coastguard Worker 	bt->len = nframes;
332*1208bc7eSAndroid Build Coastguard Worker }
333*1208bc7eSAndroid Build Coastguard Worker #elif (defined(JEMALLOC_PROF_LIBGCC))
334*1208bc7eSAndroid Build Coastguard Worker static _Unwind_Reason_Code
prof_unwind_init_callback(struct _Unwind_Context * context,void * arg)335*1208bc7eSAndroid Build Coastguard Worker prof_unwind_init_callback(struct _Unwind_Context *context, void *arg) {
336*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
337*1208bc7eSAndroid Build Coastguard Worker 
338*1208bc7eSAndroid Build Coastguard Worker 	return _URC_NO_REASON;
339*1208bc7eSAndroid Build Coastguard Worker }
340*1208bc7eSAndroid Build Coastguard Worker 
341*1208bc7eSAndroid Build Coastguard Worker static _Unwind_Reason_Code
prof_unwind_callback(struct _Unwind_Context * context,void * arg)342*1208bc7eSAndroid Build Coastguard Worker prof_unwind_callback(struct _Unwind_Context *context, void *arg) {
343*1208bc7eSAndroid Build Coastguard Worker 	prof_unwind_data_t *data = (prof_unwind_data_t *)arg;
344*1208bc7eSAndroid Build Coastguard Worker 	void *ip;
345*1208bc7eSAndroid Build Coastguard Worker 
346*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
347*1208bc7eSAndroid Build Coastguard Worker 
348*1208bc7eSAndroid Build Coastguard Worker 	ip = (void *)_Unwind_GetIP(context);
349*1208bc7eSAndroid Build Coastguard Worker 	if (ip == NULL) {
350*1208bc7eSAndroid Build Coastguard Worker 		return _URC_END_OF_STACK;
351*1208bc7eSAndroid Build Coastguard Worker 	}
352*1208bc7eSAndroid Build Coastguard Worker 	data->bt->vec[data->bt->len] = ip;
353*1208bc7eSAndroid Build Coastguard Worker 	data->bt->len++;
354*1208bc7eSAndroid Build Coastguard Worker 	if (data->bt->len == data->max) {
355*1208bc7eSAndroid Build Coastguard Worker 		return _URC_END_OF_STACK;
356*1208bc7eSAndroid Build Coastguard Worker 	}
357*1208bc7eSAndroid Build Coastguard Worker 
358*1208bc7eSAndroid Build Coastguard Worker 	return _URC_NO_REASON;
359*1208bc7eSAndroid Build Coastguard Worker }
360*1208bc7eSAndroid Build Coastguard Worker 
361*1208bc7eSAndroid Build Coastguard Worker void
prof_backtrace(prof_bt_t * bt)362*1208bc7eSAndroid Build Coastguard Worker prof_backtrace(prof_bt_t *bt) {
363*1208bc7eSAndroid Build Coastguard Worker 	prof_unwind_data_t data = {bt, PROF_BT_MAX};
364*1208bc7eSAndroid Build Coastguard Worker 
365*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
366*1208bc7eSAndroid Build Coastguard Worker 
367*1208bc7eSAndroid Build Coastguard Worker 	_Unwind_Backtrace(prof_unwind_callback, &data);
368*1208bc7eSAndroid Build Coastguard Worker }
369*1208bc7eSAndroid Build Coastguard Worker #elif (defined(JEMALLOC_PROF_GCC))
370*1208bc7eSAndroid Build Coastguard Worker void
prof_backtrace(prof_bt_t * bt)371*1208bc7eSAndroid Build Coastguard Worker prof_backtrace(prof_bt_t *bt) {
372*1208bc7eSAndroid Build Coastguard Worker #define BT_FRAME(i)							\
373*1208bc7eSAndroid Build Coastguard Worker 	if ((i) < PROF_BT_MAX) {					\
374*1208bc7eSAndroid Build Coastguard Worker 		void *p;						\
375*1208bc7eSAndroid Build Coastguard Worker 		if (__builtin_frame_address(i) == 0) {			\
376*1208bc7eSAndroid Build Coastguard Worker 			return;						\
377*1208bc7eSAndroid Build Coastguard Worker 		}							\
378*1208bc7eSAndroid Build Coastguard Worker 		p = __builtin_return_address(i);			\
379*1208bc7eSAndroid Build Coastguard Worker 		if (p == NULL) {					\
380*1208bc7eSAndroid Build Coastguard Worker 			return;						\
381*1208bc7eSAndroid Build Coastguard Worker 		}							\
382*1208bc7eSAndroid Build Coastguard Worker 		bt->vec[(i)] = p;					\
383*1208bc7eSAndroid Build Coastguard Worker 		bt->len = (i) + 1;					\
384*1208bc7eSAndroid Build Coastguard Worker 	} else {							\
385*1208bc7eSAndroid Build Coastguard Worker 		return;							\
386*1208bc7eSAndroid Build Coastguard Worker 	}
387*1208bc7eSAndroid Build Coastguard Worker 
388*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
389*1208bc7eSAndroid Build Coastguard Worker 
390*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(0)
391*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(1)
392*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(2)
393*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(3)
394*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(4)
395*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(5)
396*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(6)
397*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(7)
398*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(8)
399*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(9)
400*1208bc7eSAndroid Build Coastguard Worker 
401*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(10)
402*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(11)
403*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(12)
404*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(13)
405*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(14)
406*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(15)
407*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(16)
408*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(17)
409*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(18)
410*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(19)
411*1208bc7eSAndroid Build Coastguard Worker 
412*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(20)
413*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(21)
414*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(22)
415*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(23)
416*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(24)
417*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(25)
418*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(26)
419*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(27)
420*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(28)
421*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(29)
422*1208bc7eSAndroid Build Coastguard Worker 
423*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(30)
424*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(31)
425*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(32)
426*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(33)
427*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(34)
428*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(35)
429*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(36)
430*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(37)
431*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(38)
432*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(39)
433*1208bc7eSAndroid Build Coastguard Worker 
434*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(40)
435*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(41)
436*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(42)
437*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(43)
438*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(44)
439*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(45)
440*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(46)
441*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(47)
442*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(48)
443*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(49)
444*1208bc7eSAndroid Build Coastguard Worker 
445*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(50)
446*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(51)
447*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(52)
448*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(53)
449*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(54)
450*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(55)
451*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(56)
452*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(57)
453*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(58)
454*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(59)
455*1208bc7eSAndroid Build Coastguard Worker 
456*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(60)
457*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(61)
458*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(62)
459*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(63)
460*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(64)
461*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(65)
462*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(66)
463*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(67)
464*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(68)
465*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(69)
466*1208bc7eSAndroid Build Coastguard Worker 
467*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(70)
468*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(71)
469*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(72)
470*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(73)
471*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(74)
472*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(75)
473*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(76)
474*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(77)
475*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(78)
476*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(79)
477*1208bc7eSAndroid Build Coastguard Worker 
478*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(80)
479*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(81)
480*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(82)
481*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(83)
482*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(84)
483*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(85)
484*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(86)
485*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(87)
486*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(88)
487*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(89)
488*1208bc7eSAndroid Build Coastguard Worker 
489*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(90)
490*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(91)
491*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(92)
492*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(93)
493*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(94)
494*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(95)
495*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(96)
496*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(97)
497*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(98)
498*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(99)
499*1208bc7eSAndroid Build Coastguard Worker 
500*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(100)
501*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(101)
502*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(102)
503*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(103)
504*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(104)
505*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(105)
506*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(106)
507*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(107)
508*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(108)
509*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(109)
510*1208bc7eSAndroid Build Coastguard Worker 
511*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(110)
512*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(111)
513*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(112)
514*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(113)
515*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(114)
516*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(115)
517*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(116)
518*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(117)
519*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(118)
520*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(119)
521*1208bc7eSAndroid Build Coastguard Worker 
522*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(120)
523*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(121)
524*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(122)
525*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(123)
526*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(124)
527*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(125)
528*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(126)
529*1208bc7eSAndroid Build Coastguard Worker 	BT_FRAME(127)
530*1208bc7eSAndroid Build Coastguard Worker #undef BT_FRAME
531*1208bc7eSAndroid Build Coastguard Worker }
532*1208bc7eSAndroid Build Coastguard Worker #else
533*1208bc7eSAndroid Build Coastguard Worker void
prof_backtrace(prof_bt_t * bt)534*1208bc7eSAndroid Build Coastguard Worker prof_backtrace(prof_bt_t *bt) {
535*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
536*1208bc7eSAndroid Build Coastguard Worker 	not_reached();
537*1208bc7eSAndroid Build Coastguard Worker }
538*1208bc7eSAndroid Build Coastguard Worker #endif
539*1208bc7eSAndroid Build Coastguard Worker 
540*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t *
prof_gctx_mutex_choose(void)541*1208bc7eSAndroid Build Coastguard Worker prof_gctx_mutex_choose(void) {
542*1208bc7eSAndroid Build Coastguard Worker 	unsigned ngctxs = atomic_fetch_add_u(&cum_gctxs, 1, ATOMIC_RELAXED);
543*1208bc7eSAndroid Build Coastguard Worker 
544*1208bc7eSAndroid Build Coastguard Worker 	return &gctx_locks[(ngctxs - 1) % PROF_NCTX_LOCKS];
545*1208bc7eSAndroid Build Coastguard Worker }
546*1208bc7eSAndroid Build Coastguard Worker 
547*1208bc7eSAndroid Build Coastguard Worker static malloc_mutex_t *
prof_tdata_mutex_choose(uint64_t thr_uid)548*1208bc7eSAndroid Build Coastguard Worker prof_tdata_mutex_choose(uint64_t thr_uid) {
549*1208bc7eSAndroid Build Coastguard Worker 	return &tdata_locks[thr_uid % PROF_NTDATA_LOCKS];
550*1208bc7eSAndroid Build Coastguard Worker }
551*1208bc7eSAndroid Build Coastguard Worker 
552*1208bc7eSAndroid Build Coastguard Worker static prof_gctx_t *
prof_gctx_create(tsdn_t * tsdn,prof_bt_t * bt)553*1208bc7eSAndroid Build Coastguard Worker prof_gctx_create(tsdn_t *tsdn, prof_bt_t *bt) {
554*1208bc7eSAndroid Build Coastguard Worker 	/*
555*1208bc7eSAndroid Build Coastguard Worker 	 * Create a single allocation that has space for vec of length bt->len.
556*1208bc7eSAndroid Build Coastguard Worker 	 */
557*1208bc7eSAndroid Build Coastguard Worker 	size_t size = offsetof(prof_gctx_t, vec) + (bt->len * sizeof(void *));
558*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_t *gctx = (prof_gctx_t *)iallocztm(tsdn, size,
559*1208bc7eSAndroid Build Coastguard Worker 	    sz_size2index(size), false, NULL, true, arena_get(TSDN_NULL, 0, true),
560*1208bc7eSAndroid Build Coastguard Worker 	    true);
561*1208bc7eSAndroid Build Coastguard Worker 	if (gctx == NULL) {
562*1208bc7eSAndroid Build Coastguard Worker 		return NULL;
563*1208bc7eSAndroid Build Coastguard Worker 	}
564*1208bc7eSAndroid Build Coastguard Worker 	gctx->lock = prof_gctx_mutex_choose();
565*1208bc7eSAndroid Build Coastguard Worker 	/*
566*1208bc7eSAndroid Build Coastguard Worker 	 * Set nlimbo to 1, in order to avoid a race condition with
567*1208bc7eSAndroid Build Coastguard Worker 	 * prof_tctx_destroy()/prof_gctx_try_destroy().
568*1208bc7eSAndroid Build Coastguard Worker 	 */
569*1208bc7eSAndroid Build Coastguard Worker 	gctx->nlimbo = 1;
570*1208bc7eSAndroid Build Coastguard Worker 	tctx_tree_new(&gctx->tctxs);
571*1208bc7eSAndroid Build Coastguard Worker 	/* Duplicate bt. */
572*1208bc7eSAndroid Build Coastguard Worker 	memcpy(gctx->vec, bt->vec, bt->len * sizeof(void *));
573*1208bc7eSAndroid Build Coastguard Worker 	gctx->bt.vec = gctx->vec;
574*1208bc7eSAndroid Build Coastguard Worker 	gctx->bt.len = bt->len;
575*1208bc7eSAndroid Build Coastguard Worker 	return gctx;
576*1208bc7eSAndroid Build Coastguard Worker }
577*1208bc7eSAndroid Build Coastguard Worker 
578*1208bc7eSAndroid Build Coastguard Worker static void
prof_gctx_try_destroy(tsd_t * tsd,prof_tdata_t * tdata_self,prof_gctx_t * gctx,prof_tdata_t * tdata)579*1208bc7eSAndroid Build Coastguard Worker prof_gctx_try_destroy(tsd_t *tsd, prof_tdata_t *tdata_self, prof_gctx_t *gctx,
580*1208bc7eSAndroid Build Coastguard Worker     prof_tdata_t *tdata) {
581*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
582*1208bc7eSAndroid Build Coastguard Worker 
583*1208bc7eSAndroid Build Coastguard Worker 	/*
584*1208bc7eSAndroid Build Coastguard Worker 	 * Check that gctx is still unused by any thread cache before destroying
585*1208bc7eSAndroid Build Coastguard Worker 	 * it.  prof_lookup() increments gctx->nlimbo in order to avoid a race
586*1208bc7eSAndroid Build Coastguard Worker 	 * condition with this function, as does prof_tctx_destroy() in order to
587*1208bc7eSAndroid Build Coastguard Worker 	 * avoid a race between the main body of prof_tctx_destroy() and entry
588*1208bc7eSAndroid Build Coastguard Worker 	 * into this function.
589*1208bc7eSAndroid Build Coastguard Worker 	 */
590*1208bc7eSAndroid Build Coastguard Worker 	prof_enter(tsd, tdata_self);
591*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), gctx->lock);
592*1208bc7eSAndroid Build Coastguard Worker 	assert(gctx->nlimbo != 0);
593*1208bc7eSAndroid Build Coastguard Worker 	if (tctx_tree_empty(&gctx->tctxs) && gctx->nlimbo == 1) {
594*1208bc7eSAndroid Build Coastguard Worker 		/* Remove gctx from bt2gctx. */
595*1208bc7eSAndroid Build Coastguard Worker 		if (ckh_remove(tsd, &bt2gctx, &gctx->bt, NULL, NULL)) {
596*1208bc7eSAndroid Build Coastguard Worker 			not_reached();
597*1208bc7eSAndroid Build Coastguard Worker 		}
598*1208bc7eSAndroid Build Coastguard Worker 		prof_leave(tsd, tdata_self);
599*1208bc7eSAndroid Build Coastguard Worker 		/* Destroy gctx. */
600*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), gctx->lock);
601*1208bc7eSAndroid Build Coastguard Worker 		idalloctm(tsd_tsdn(tsd), gctx, NULL, NULL, true, true);
602*1208bc7eSAndroid Build Coastguard Worker 	} else {
603*1208bc7eSAndroid Build Coastguard Worker 		/*
604*1208bc7eSAndroid Build Coastguard Worker 		 * Compensate for increment in prof_tctx_destroy() or
605*1208bc7eSAndroid Build Coastguard Worker 		 * prof_lookup().
606*1208bc7eSAndroid Build Coastguard Worker 		 */
607*1208bc7eSAndroid Build Coastguard Worker 		gctx->nlimbo--;
608*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), gctx->lock);
609*1208bc7eSAndroid Build Coastguard Worker 		prof_leave(tsd, tdata_self);
610*1208bc7eSAndroid Build Coastguard Worker 	}
611*1208bc7eSAndroid Build Coastguard Worker }
612*1208bc7eSAndroid Build Coastguard Worker 
613*1208bc7eSAndroid Build Coastguard Worker static bool
prof_tctx_should_destroy(tsdn_t * tsdn,prof_tctx_t * tctx)614*1208bc7eSAndroid Build Coastguard Worker prof_tctx_should_destroy(tsdn_t *tsdn, prof_tctx_t *tctx) {
615*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsdn, tctx->tdata->lock);
616*1208bc7eSAndroid Build Coastguard Worker 
617*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof_accum) {
618*1208bc7eSAndroid Build Coastguard Worker 		return false;
619*1208bc7eSAndroid Build Coastguard Worker 	}
620*1208bc7eSAndroid Build Coastguard Worker 	if (tctx->cnts.curobjs != 0) {
621*1208bc7eSAndroid Build Coastguard Worker 		return false;
622*1208bc7eSAndroid Build Coastguard Worker 	}
623*1208bc7eSAndroid Build Coastguard Worker 	if (tctx->prepared) {
624*1208bc7eSAndroid Build Coastguard Worker 		return false;
625*1208bc7eSAndroid Build Coastguard Worker 	}
626*1208bc7eSAndroid Build Coastguard Worker 	return true;
627*1208bc7eSAndroid Build Coastguard Worker }
628*1208bc7eSAndroid Build Coastguard Worker 
629*1208bc7eSAndroid Build Coastguard Worker static bool
prof_gctx_should_destroy(prof_gctx_t * gctx)630*1208bc7eSAndroid Build Coastguard Worker prof_gctx_should_destroy(prof_gctx_t *gctx) {
631*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof_accum) {
632*1208bc7eSAndroid Build Coastguard Worker 		return false;
633*1208bc7eSAndroid Build Coastguard Worker 	}
634*1208bc7eSAndroid Build Coastguard Worker 	if (!tctx_tree_empty(&gctx->tctxs)) {
635*1208bc7eSAndroid Build Coastguard Worker 		return false;
636*1208bc7eSAndroid Build Coastguard Worker 	}
637*1208bc7eSAndroid Build Coastguard Worker 	if (gctx->nlimbo != 0) {
638*1208bc7eSAndroid Build Coastguard Worker 		return false;
639*1208bc7eSAndroid Build Coastguard Worker 	}
640*1208bc7eSAndroid Build Coastguard Worker 	return true;
641*1208bc7eSAndroid Build Coastguard Worker }
642*1208bc7eSAndroid Build Coastguard Worker 
643*1208bc7eSAndroid Build Coastguard Worker static void
prof_tctx_destroy(tsd_t * tsd,prof_tctx_t * tctx)644*1208bc7eSAndroid Build Coastguard Worker prof_tctx_destroy(tsd_t *tsd, prof_tctx_t *tctx) {
645*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata = tctx->tdata;
646*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_t *gctx = tctx->gctx;
647*1208bc7eSAndroid Build Coastguard Worker 	bool destroy_tdata, destroy_tctx, destroy_gctx;
648*1208bc7eSAndroid Build Coastguard Worker 
649*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsd_tsdn(tsd), tctx->tdata->lock);
650*1208bc7eSAndroid Build Coastguard Worker 
651*1208bc7eSAndroid Build Coastguard Worker 	assert(tctx->cnts.curobjs == 0);
652*1208bc7eSAndroid Build Coastguard Worker 	assert(tctx->cnts.curbytes == 0);
653*1208bc7eSAndroid Build Coastguard Worker 	assert(!opt_prof_accum);
654*1208bc7eSAndroid Build Coastguard Worker 	assert(tctx->cnts.accumobjs == 0);
655*1208bc7eSAndroid Build Coastguard Worker 	assert(tctx->cnts.accumbytes == 0);
656*1208bc7eSAndroid Build Coastguard Worker 
657*1208bc7eSAndroid Build Coastguard Worker 	ckh_remove(tsd, &tdata->bt2tctx, &gctx->bt, NULL, NULL);
658*1208bc7eSAndroid Build Coastguard Worker 	destroy_tdata = prof_tdata_should_destroy(tsd_tsdn(tsd), tdata, false);
659*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), tdata->lock);
660*1208bc7eSAndroid Build Coastguard Worker 
661*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), gctx->lock);
662*1208bc7eSAndroid Build Coastguard Worker 	switch (tctx->state) {
663*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_nominal:
664*1208bc7eSAndroid Build Coastguard Worker 		tctx_tree_remove(&gctx->tctxs, tctx);
665*1208bc7eSAndroid Build Coastguard Worker 		destroy_tctx = true;
666*1208bc7eSAndroid Build Coastguard Worker 		if (prof_gctx_should_destroy(gctx)) {
667*1208bc7eSAndroid Build Coastguard Worker 			/*
668*1208bc7eSAndroid Build Coastguard Worker 			 * Increment gctx->nlimbo in order to keep another
669*1208bc7eSAndroid Build Coastguard Worker 			 * thread from winning the race to destroy gctx while
670*1208bc7eSAndroid Build Coastguard Worker 			 * this one has gctx->lock dropped.  Without this, it
671*1208bc7eSAndroid Build Coastguard Worker 			 * would be possible for another thread to:
672*1208bc7eSAndroid Build Coastguard Worker 			 *
673*1208bc7eSAndroid Build Coastguard Worker 			 * 1) Sample an allocation associated with gctx.
674*1208bc7eSAndroid Build Coastguard Worker 			 * 2) Deallocate the sampled object.
675*1208bc7eSAndroid Build Coastguard Worker 			 * 3) Successfully prof_gctx_try_destroy(gctx).
676*1208bc7eSAndroid Build Coastguard Worker 			 *
677*1208bc7eSAndroid Build Coastguard Worker 			 * The result would be that gctx no longer exists by the
678*1208bc7eSAndroid Build Coastguard Worker 			 * time this thread accesses it in
679*1208bc7eSAndroid Build Coastguard Worker 			 * prof_gctx_try_destroy().
680*1208bc7eSAndroid Build Coastguard Worker 			 */
681*1208bc7eSAndroid Build Coastguard Worker 			gctx->nlimbo++;
682*1208bc7eSAndroid Build Coastguard Worker 			destroy_gctx = true;
683*1208bc7eSAndroid Build Coastguard Worker 		} else {
684*1208bc7eSAndroid Build Coastguard Worker 			destroy_gctx = false;
685*1208bc7eSAndroid Build Coastguard Worker 		}
686*1208bc7eSAndroid Build Coastguard Worker 		break;
687*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_dumping:
688*1208bc7eSAndroid Build Coastguard Worker 		/*
689*1208bc7eSAndroid Build Coastguard Worker 		 * A dumping thread needs tctx to remain valid until dumping
690*1208bc7eSAndroid Build Coastguard Worker 		 * has finished.  Change state such that the dumping thread will
691*1208bc7eSAndroid Build Coastguard Worker 		 * complete destruction during a late dump iteration phase.
692*1208bc7eSAndroid Build Coastguard Worker 		 */
693*1208bc7eSAndroid Build Coastguard Worker 		tctx->state = prof_tctx_state_purgatory;
694*1208bc7eSAndroid Build Coastguard Worker 		destroy_tctx = false;
695*1208bc7eSAndroid Build Coastguard Worker 		destroy_gctx = false;
696*1208bc7eSAndroid Build Coastguard Worker 		break;
697*1208bc7eSAndroid Build Coastguard Worker 	default:
698*1208bc7eSAndroid Build Coastguard Worker 		not_reached();
699*1208bc7eSAndroid Build Coastguard Worker 		destroy_tctx = false;
700*1208bc7eSAndroid Build Coastguard Worker 		destroy_gctx = false;
701*1208bc7eSAndroid Build Coastguard Worker 	}
702*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), gctx->lock);
703*1208bc7eSAndroid Build Coastguard Worker 	if (destroy_gctx) {
704*1208bc7eSAndroid Build Coastguard Worker 		prof_gctx_try_destroy(tsd, prof_tdata_get(tsd, false), gctx,
705*1208bc7eSAndroid Build Coastguard Worker 		    tdata);
706*1208bc7eSAndroid Build Coastguard Worker 	}
707*1208bc7eSAndroid Build Coastguard Worker 
708*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_not_owner(tsd_tsdn(tsd), tctx->tdata->lock);
709*1208bc7eSAndroid Build Coastguard Worker 
710*1208bc7eSAndroid Build Coastguard Worker 	if (destroy_tdata) {
711*1208bc7eSAndroid Build Coastguard Worker 		prof_tdata_destroy(tsd, tdata, false);
712*1208bc7eSAndroid Build Coastguard Worker 	}
713*1208bc7eSAndroid Build Coastguard Worker 
714*1208bc7eSAndroid Build Coastguard Worker 	if (destroy_tctx) {
715*1208bc7eSAndroid Build Coastguard Worker 		idalloctm(tsd_tsdn(tsd), tctx, NULL, NULL, true, true);
716*1208bc7eSAndroid Build Coastguard Worker 	}
717*1208bc7eSAndroid Build Coastguard Worker }
718*1208bc7eSAndroid Build Coastguard Worker 
719*1208bc7eSAndroid Build Coastguard Worker static bool
prof_lookup_global(tsd_t * tsd,prof_bt_t * bt,prof_tdata_t * tdata,void ** p_btkey,prof_gctx_t ** p_gctx,bool * p_new_gctx)720*1208bc7eSAndroid Build Coastguard Worker prof_lookup_global(tsd_t *tsd, prof_bt_t *bt, prof_tdata_t *tdata,
721*1208bc7eSAndroid Build Coastguard Worker     void **p_btkey, prof_gctx_t **p_gctx, bool *p_new_gctx) {
722*1208bc7eSAndroid Build Coastguard Worker 	union {
723*1208bc7eSAndroid Build Coastguard Worker 		prof_gctx_t	*p;
724*1208bc7eSAndroid Build Coastguard Worker 		void		*v;
725*1208bc7eSAndroid Build Coastguard Worker 	} gctx, tgctx;
726*1208bc7eSAndroid Build Coastguard Worker 	union {
727*1208bc7eSAndroid Build Coastguard Worker 		prof_bt_t	*p;
728*1208bc7eSAndroid Build Coastguard Worker 		void		*v;
729*1208bc7eSAndroid Build Coastguard Worker 	} btkey;
730*1208bc7eSAndroid Build Coastguard Worker 	bool new_gctx;
731*1208bc7eSAndroid Build Coastguard Worker 
732*1208bc7eSAndroid Build Coastguard Worker 	prof_enter(tsd, tdata);
733*1208bc7eSAndroid Build Coastguard Worker 	if (ckh_search(&bt2gctx, bt, &btkey.v, &gctx.v)) {
734*1208bc7eSAndroid Build Coastguard Worker 		/* bt has never been seen before.  Insert it. */
735*1208bc7eSAndroid Build Coastguard Worker 		prof_leave(tsd, tdata);
736*1208bc7eSAndroid Build Coastguard Worker 		tgctx.p = prof_gctx_create(tsd_tsdn(tsd), bt);
737*1208bc7eSAndroid Build Coastguard Worker 		if (tgctx.v == NULL) {
738*1208bc7eSAndroid Build Coastguard Worker 			return true;
739*1208bc7eSAndroid Build Coastguard Worker 		}
740*1208bc7eSAndroid Build Coastguard Worker 		prof_enter(tsd, tdata);
741*1208bc7eSAndroid Build Coastguard Worker 		if (ckh_search(&bt2gctx, bt, &btkey.v, &gctx.v)) {
742*1208bc7eSAndroid Build Coastguard Worker 			gctx.p = tgctx.p;
743*1208bc7eSAndroid Build Coastguard Worker 			btkey.p = &gctx.p->bt;
744*1208bc7eSAndroid Build Coastguard Worker 			if (ckh_insert(tsd, &bt2gctx, btkey.v, gctx.v)) {
745*1208bc7eSAndroid Build Coastguard Worker 				/* OOM. */
746*1208bc7eSAndroid Build Coastguard Worker 				prof_leave(tsd, tdata);
747*1208bc7eSAndroid Build Coastguard Worker 				idalloctm(tsd_tsdn(tsd), gctx.v, NULL, NULL,
748*1208bc7eSAndroid Build Coastguard Worker 				    true, true);
749*1208bc7eSAndroid Build Coastguard Worker 				return true;
750*1208bc7eSAndroid Build Coastguard Worker 			}
751*1208bc7eSAndroid Build Coastguard Worker 			new_gctx = true;
752*1208bc7eSAndroid Build Coastguard Worker 		} else {
753*1208bc7eSAndroid Build Coastguard Worker 			new_gctx = false;
754*1208bc7eSAndroid Build Coastguard Worker 		}
755*1208bc7eSAndroid Build Coastguard Worker 	} else {
756*1208bc7eSAndroid Build Coastguard Worker 		tgctx.v = NULL;
757*1208bc7eSAndroid Build Coastguard Worker 		new_gctx = false;
758*1208bc7eSAndroid Build Coastguard Worker 	}
759*1208bc7eSAndroid Build Coastguard Worker 
760*1208bc7eSAndroid Build Coastguard Worker 	if (!new_gctx) {
761*1208bc7eSAndroid Build Coastguard Worker 		/*
762*1208bc7eSAndroid Build Coastguard Worker 		 * Increment nlimbo, in order to avoid a race condition with
763*1208bc7eSAndroid Build Coastguard Worker 		 * prof_tctx_destroy()/prof_gctx_try_destroy().
764*1208bc7eSAndroid Build Coastguard Worker 		 */
765*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsd_tsdn(tsd), gctx.p->lock);
766*1208bc7eSAndroid Build Coastguard Worker 		gctx.p->nlimbo++;
767*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), gctx.p->lock);
768*1208bc7eSAndroid Build Coastguard Worker 		new_gctx = false;
769*1208bc7eSAndroid Build Coastguard Worker 
770*1208bc7eSAndroid Build Coastguard Worker 		if (tgctx.v != NULL) {
771*1208bc7eSAndroid Build Coastguard Worker 			/* Lost race to insert. */
772*1208bc7eSAndroid Build Coastguard Worker 			idalloctm(tsd_tsdn(tsd), tgctx.v, NULL, NULL, true,
773*1208bc7eSAndroid Build Coastguard Worker 			    true);
774*1208bc7eSAndroid Build Coastguard Worker 		}
775*1208bc7eSAndroid Build Coastguard Worker 	}
776*1208bc7eSAndroid Build Coastguard Worker 	prof_leave(tsd, tdata);
777*1208bc7eSAndroid Build Coastguard Worker 
778*1208bc7eSAndroid Build Coastguard Worker 	*p_btkey = btkey.v;
779*1208bc7eSAndroid Build Coastguard Worker 	*p_gctx = gctx.p;
780*1208bc7eSAndroid Build Coastguard Worker 	*p_new_gctx = new_gctx;
781*1208bc7eSAndroid Build Coastguard Worker 	return false;
782*1208bc7eSAndroid Build Coastguard Worker }
783*1208bc7eSAndroid Build Coastguard Worker 
784*1208bc7eSAndroid Build Coastguard Worker prof_tctx_t *
prof_lookup(tsd_t * tsd,prof_bt_t * bt)785*1208bc7eSAndroid Build Coastguard Worker prof_lookup(tsd_t *tsd, prof_bt_t *bt) {
786*1208bc7eSAndroid Build Coastguard Worker 	union {
787*1208bc7eSAndroid Build Coastguard Worker 		prof_tctx_t	*p;
788*1208bc7eSAndroid Build Coastguard Worker 		void		*v;
789*1208bc7eSAndroid Build Coastguard Worker 	} ret;
790*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
791*1208bc7eSAndroid Build Coastguard Worker 	bool not_found;
792*1208bc7eSAndroid Build Coastguard Worker 
793*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
794*1208bc7eSAndroid Build Coastguard Worker 
795*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, false);
796*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
797*1208bc7eSAndroid Build Coastguard Worker 		return NULL;
798*1208bc7eSAndroid Build Coastguard Worker 	}
799*1208bc7eSAndroid Build Coastguard Worker 
800*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), tdata->lock);
801*1208bc7eSAndroid Build Coastguard Worker 	not_found = ckh_search(&tdata->bt2tctx, bt, NULL, &ret.v);
802*1208bc7eSAndroid Build Coastguard Worker 	if (!not_found) { /* Note double negative! */
803*1208bc7eSAndroid Build Coastguard Worker 		ret.p->prepared = true;
804*1208bc7eSAndroid Build Coastguard Worker 	}
805*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), tdata->lock);
806*1208bc7eSAndroid Build Coastguard Worker 	if (not_found) {
807*1208bc7eSAndroid Build Coastguard Worker 		void *btkey;
808*1208bc7eSAndroid Build Coastguard Worker 		prof_gctx_t *gctx;
809*1208bc7eSAndroid Build Coastguard Worker 		bool new_gctx, error;
810*1208bc7eSAndroid Build Coastguard Worker 
811*1208bc7eSAndroid Build Coastguard Worker 		/*
812*1208bc7eSAndroid Build Coastguard Worker 		 * This thread's cache lacks bt.  Look for it in the global
813*1208bc7eSAndroid Build Coastguard Worker 		 * cache.
814*1208bc7eSAndroid Build Coastguard Worker 		 */
815*1208bc7eSAndroid Build Coastguard Worker 		if (prof_lookup_global(tsd, bt, tdata, &btkey, &gctx,
816*1208bc7eSAndroid Build Coastguard Worker 		    &new_gctx)) {
817*1208bc7eSAndroid Build Coastguard Worker 			return NULL;
818*1208bc7eSAndroid Build Coastguard Worker 		}
819*1208bc7eSAndroid Build Coastguard Worker 
820*1208bc7eSAndroid Build Coastguard Worker 		/* Link a prof_tctx_t into gctx for this thread. */
821*1208bc7eSAndroid Build Coastguard Worker 		ret.v = iallocztm(tsd_tsdn(tsd), sizeof(prof_tctx_t),
822*1208bc7eSAndroid Build Coastguard Worker 		    sz_size2index(sizeof(prof_tctx_t)), false, NULL, true,
823*1208bc7eSAndroid Build Coastguard Worker 		    arena_ichoose(tsd, NULL), true);
824*1208bc7eSAndroid Build Coastguard Worker 		if (ret.p == NULL) {
825*1208bc7eSAndroid Build Coastguard Worker 			if (new_gctx) {
826*1208bc7eSAndroid Build Coastguard Worker 				prof_gctx_try_destroy(tsd, tdata, gctx, tdata);
827*1208bc7eSAndroid Build Coastguard Worker 			}
828*1208bc7eSAndroid Build Coastguard Worker 			return NULL;
829*1208bc7eSAndroid Build Coastguard Worker 		}
830*1208bc7eSAndroid Build Coastguard Worker 		ret.p->tdata = tdata;
831*1208bc7eSAndroid Build Coastguard Worker 		ret.p->thr_uid = tdata->thr_uid;
832*1208bc7eSAndroid Build Coastguard Worker 		ret.p->thr_discrim = tdata->thr_discrim;
833*1208bc7eSAndroid Build Coastguard Worker 		memset(&ret.p->cnts, 0, sizeof(prof_cnt_t));
834*1208bc7eSAndroid Build Coastguard Worker 		ret.p->gctx = gctx;
835*1208bc7eSAndroid Build Coastguard Worker 		ret.p->tctx_uid = tdata->tctx_uid_next++;
836*1208bc7eSAndroid Build Coastguard Worker 		ret.p->prepared = true;
837*1208bc7eSAndroid Build Coastguard Worker 		ret.p->state = prof_tctx_state_initializing;
838*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsd_tsdn(tsd), tdata->lock);
839*1208bc7eSAndroid Build Coastguard Worker 		error = ckh_insert(tsd, &tdata->bt2tctx, btkey, ret.v);
840*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), tdata->lock);
841*1208bc7eSAndroid Build Coastguard Worker 		if (error) {
842*1208bc7eSAndroid Build Coastguard Worker 			if (new_gctx) {
843*1208bc7eSAndroid Build Coastguard Worker 				prof_gctx_try_destroy(tsd, tdata, gctx, tdata);
844*1208bc7eSAndroid Build Coastguard Worker 			}
845*1208bc7eSAndroid Build Coastguard Worker 			idalloctm(tsd_tsdn(tsd), ret.v, NULL, NULL, true, true);
846*1208bc7eSAndroid Build Coastguard Worker 			return NULL;
847*1208bc7eSAndroid Build Coastguard Worker 		}
848*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsd_tsdn(tsd), gctx->lock);
849*1208bc7eSAndroid Build Coastguard Worker 		ret.p->state = prof_tctx_state_nominal;
850*1208bc7eSAndroid Build Coastguard Worker 		tctx_tree_insert(&gctx->tctxs, ret.p);
851*1208bc7eSAndroid Build Coastguard Worker 		gctx->nlimbo--;
852*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), gctx->lock);
853*1208bc7eSAndroid Build Coastguard Worker 	}
854*1208bc7eSAndroid Build Coastguard Worker 
855*1208bc7eSAndroid Build Coastguard Worker 	return ret.p;
856*1208bc7eSAndroid Build Coastguard Worker }
857*1208bc7eSAndroid Build Coastguard Worker 
858*1208bc7eSAndroid Build Coastguard Worker /*
859*1208bc7eSAndroid Build Coastguard Worker  * The bodies of this function and prof_leakcheck() are compiled out unless heap
860*1208bc7eSAndroid Build Coastguard Worker  * profiling is enabled, so that it is possible to compile jemalloc with
861*1208bc7eSAndroid Build Coastguard Worker  * floating point support completely disabled.  Avoiding floating point code is
862*1208bc7eSAndroid Build Coastguard Worker  * important on memory-constrained systems, but it also enables a workaround for
863*1208bc7eSAndroid Build Coastguard Worker  * versions of glibc that don't properly save/restore floating point registers
864*1208bc7eSAndroid Build Coastguard Worker  * during dynamic lazy symbol loading (which internally calls into whatever
865*1208bc7eSAndroid Build Coastguard Worker  * malloc implementation happens to be integrated into the application).  Note
866*1208bc7eSAndroid Build Coastguard Worker  * that some compilers (e.g.  gcc 4.8) may use floating point registers for fast
867*1208bc7eSAndroid Build Coastguard Worker  * memory moves, so jemalloc must be compiled with such optimizations disabled
868*1208bc7eSAndroid Build Coastguard Worker  * (e.g.
869*1208bc7eSAndroid Build Coastguard Worker  * -mno-sse) in order for the workaround to be complete.
870*1208bc7eSAndroid Build Coastguard Worker  */
871*1208bc7eSAndroid Build Coastguard Worker void
prof_sample_threshold_update(prof_tdata_t * tdata)872*1208bc7eSAndroid Build Coastguard Worker prof_sample_threshold_update(prof_tdata_t *tdata) {
873*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF
874*1208bc7eSAndroid Build Coastguard Worker 	uint64_t r;
875*1208bc7eSAndroid Build Coastguard Worker 	double u;
876*1208bc7eSAndroid Build Coastguard Worker 
877*1208bc7eSAndroid Build Coastguard Worker 	if (!config_prof) {
878*1208bc7eSAndroid Build Coastguard Worker 		return;
879*1208bc7eSAndroid Build Coastguard Worker 	}
880*1208bc7eSAndroid Build Coastguard Worker 
881*1208bc7eSAndroid Build Coastguard Worker 	if (lg_prof_sample == 0) {
882*1208bc7eSAndroid Build Coastguard Worker 		tdata->bytes_until_sample = 0;
883*1208bc7eSAndroid Build Coastguard Worker 		return;
884*1208bc7eSAndroid Build Coastguard Worker 	}
885*1208bc7eSAndroid Build Coastguard Worker 
886*1208bc7eSAndroid Build Coastguard Worker 	/*
887*1208bc7eSAndroid Build Coastguard Worker 	 * Compute sample interval as a geometrically distributed random
888*1208bc7eSAndroid Build Coastguard Worker 	 * variable with mean (2^lg_prof_sample).
889*1208bc7eSAndroid Build Coastguard Worker 	 *
890*1208bc7eSAndroid Build Coastguard Worker 	 *                             __        __
891*1208bc7eSAndroid Build Coastguard Worker 	 *                             |  log(u)  |                     1
892*1208bc7eSAndroid Build Coastguard Worker 	 * tdata->bytes_until_sample = | -------- |, where p = ---------------
893*1208bc7eSAndroid Build Coastguard Worker 	 *                             | log(1-p) |             lg_prof_sample
894*1208bc7eSAndroid Build Coastguard Worker 	 *                                                     2
895*1208bc7eSAndroid Build Coastguard Worker 	 *
896*1208bc7eSAndroid Build Coastguard Worker 	 * For more information on the math, see:
897*1208bc7eSAndroid Build Coastguard Worker 	 *
898*1208bc7eSAndroid Build Coastguard Worker 	 *   Non-Uniform Random Variate Generation
899*1208bc7eSAndroid Build Coastguard Worker 	 *   Luc Devroye
900*1208bc7eSAndroid Build Coastguard Worker 	 *   Springer-Verlag, New York, 1986
901*1208bc7eSAndroid Build Coastguard Worker 	 *   pp 500
902*1208bc7eSAndroid Build Coastguard Worker 	 *   (http://luc.devroye.org/rnbookindex.html)
903*1208bc7eSAndroid Build Coastguard Worker 	 */
904*1208bc7eSAndroid Build Coastguard Worker 	r = prng_lg_range_u64(&tdata->prng_state, 53);
905*1208bc7eSAndroid Build Coastguard Worker 	u = (double)r * (1.0/9007199254740992.0L);
906*1208bc7eSAndroid Build Coastguard Worker 	tdata->bytes_until_sample = (uint64_t)(log(u) /
907*1208bc7eSAndroid Build Coastguard Worker 	    log(1.0 - (1.0 / (double)((uint64_t)1U << lg_prof_sample))))
908*1208bc7eSAndroid Build Coastguard Worker 	    + (uint64_t)1U;
909*1208bc7eSAndroid Build Coastguard Worker #endif
910*1208bc7eSAndroid Build Coastguard Worker }
911*1208bc7eSAndroid Build Coastguard Worker 
912*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_JET
913*1208bc7eSAndroid Build Coastguard Worker static prof_tdata_t *
prof_tdata_count_iter(prof_tdata_tree_t * tdatas,prof_tdata_t * tdata,void * arg)914*1208bc7eSAndroid Build Coastguard Worker prof_tdata_count_iter(prof_tdata_tree_t *tdatas, prof_tdata_t *tdata,
915*1208bc7eSAndroid Build Coastguard Worker     void *arg) {
916*1208bc7eSAndroid Build Coastguard Worker 	size_t *tdata_count = (size_t *)arg;
917*1208bc7eSAndroid Build Coastguard Worker 
918*1208bc7eSAndroid Build Coastguard Worker 	(*tdata_count)++;
919*1208bc7eSAndroid Build Coastguard Worker 
920*1208bc7eSAndroid Build Coastguard Worker 	return NULL;
921*1208bc7eSAndroid Build Coastguard Worker }
922*1208bc7eSAndroid Build Coastguard Worker 
923*1208bc7eSAndroid Build Coastguard Worker size_t
prof_tdata_count(void)924*1208bc7eSAndroid Build Coastguard Worker prof_tdata_count(void) {
925*1208bc7eSAndroid Build Coastguard Worker 	size_t tdata_count = 0;
926*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t *tsdn;
927*1208bc7eSAndroid Build Coastguard Worker 
928*1208bc7eSAndroid Build Coastguard Worker 	tsdn = tsdn_fetch();
929*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &tdatas_mtx);
930*1208bc7eSAndroid Build Coastguard Worker 	tdata_tree_iter(&tdatas, NULL, prof_tdata_count_iter,
931*1208bc7eSAndroid Build Coastguard Worker 	    (void *)&tdata_count);
932*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &tdatas_mtx);
933*1208bc7eSAndroid Build Coastguard Worker 
934*1208bc7eSAndroid Build Coastguard Worker 	return tdata_count;
935*1208bc7eSAndroid Build Coastguard Worker }
936*1208bc7eSAndroid Build Coastguard Worker 
937*1208bc7eSAndroid Build Coastguard Worker size_t
prof_bt_count(void)938*1208bc7eSAndroid Build Coastguard Worker prof_bt_count(void) {
939*1208bc7eSAndroid Build Coastguard Worker 	size_t bt_count;
940*1208bc7eSAndroid Build Coastguard Worker 	tsd_t *tsd;
941*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
942*1208bc7eSAndroid Build Coastguard Worker 
943*1208bc7eSAndroid Build Coastguard Worker 	tsd = tsd_fetch();
944*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, false);
945*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
946*1208bc7eSAndroid Build Coastguard Worker 		return 0;
947*1208bc7eSAndroid Build Coastguard Worker 	}
948*1208bc7eSAndroid Build Coastguard Worker 
949*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &bt2gctx_mtx);
950*1208bc7eSAndroid Build Coastguard Worker 	bt_count = ckh_count(&bt2gctx);
951*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &bt2gctx_mtx);
952*1208bc7eSAndroid Build Coastguard Worker 
953*1208bc7eSAndroid Build Coastguard Worker 	return bt_count;
954*1208bc7eSAndroid Build Coastguard Worker }
955*1208bc7eSAndroid Build Coastguard Worker #endif
956*1208bc7eSAndroid Build Coastguard Worker 
957*1208bc7eSAndroid Build Coastguard Worker static int
prof_dump_open_impl(bool propagate_err,const char * filename)958*1208bc7eSAndroid Build Coastguard Worker prof_dump_open_impl(bool propagate_err, const char *filename) {
959*1208bc7eSAndroid Build Coastguard Worker 	int fd;
960*1208bc7eSAndroid Build Coastguard Worker 
961*1208bc7eSAndroid Build Coastguard Worker 	fd = creat(filename, 0644);
962*1208bc7eSAndroid Build Coastguard Worker 	if (fd == -1 && !propagate_err) {
963*1208bc7eSAndroid Build Coastguard Worker 		malloc_printf("<jemalloc>: creat(\"%s\"), 0644) failed\n",
964*1208bc7eSAndroid Build Coastguard Worker 		    filename);
965*1208bc7eSAndroid Build Coastguard Worker 		if (opt_abort) {
966*1208bc7eSAndroid Build Coastguard Worker 			abort();
967*1208bc7eSAndroid Build Coastguard Worker 		}
968*1208bc7eSAndroid Build Coastguard Worker 	}
969*1208bc7eSAndroid Build Coastguard Worker 
970*1208bc7eSAndroid Build Coastguard Worker 	return fd;
971*1208bc7eSAndroid Build Coastguard Worker }
972*1208bc7eSAndroid Build Coastguard Worker prof_dump_open_t *JET_MUTABLE prof_dump_open = prof_dump_open_impl;
973*1208bc7eSAndroid Build Coastguard Worker 
974*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_flush(bool propagate_err)975*1208bc7eSAndroid Build Coastguard Worker prof_dump_flush(bool propagate_err) {
976*1208bc7eSAndroid Build Coastguard Worker 	bool ret = false;
977*1208bc7eSAndroid Build Coastguard Worker 	ssize_t err;
978*1208bc7eSAndroid Build Coastguard Worker 
979*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
980*1208bc7eSAndroid Build Coastguard Worker 
981*1208bc7eSAndroid Build Coastguard Worker 	err = malloc_write_fd(prof_dump_fd, prof_dump_buf, prof_dump_buf_end);
982*1208bc7eSAndroid Build Coastguard Worker 	if (err == -1) {
983*1208bc7eSAndroid Build Coastguard Worker 		if (!propagate_err) {
984*1208bc7eSAndroid Build Coastguard Worker 			malloc_write("<jemalloc>: write() failed during heap "
985*1208bc7eSAndroid Build Coastguard Worker 			    "profile flush\n");
986*1208bc7eSAndroid Build Coastguard Worker 			if (opt_abort) {
987*1208bc7eSAndroid Build Coastguard Worker 				abort();
988*1208bc7eSAndroid Build Coastguard Worker 			}
989*1208bc7eSAndroid Build Coastguard Worker 		}
990*1208bc7eSAndroid Build Coastguard Worker 		ret = true;
991*1208bc7eSAndroid Build Coastguard Worker 	}
992*1208bc7eSAndroid Build Coastguard Worker 	prof_dump_buf_end = 0;
993*1208bc7eSAndroid Build Coastguard Worker 
994*1208bc7eSAndroid Build Coastguard Worker 	return ret;
995*1208bc7eSAndroid Build Coastguard Worker }
996*1208bc7eSAndroid Build Coastguard Worker 
997*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_close(bool propagate_err)998*1208bc7eSAndroid Build Coastguard Worker prof_dump_close(bool propagate_err) {
999*1208bc7eSAndroid Build Coastguard Worker 	bool ret;
1000*1208bc7eSAndroid Build Coastguard Worker 
1001*1208bc7eSAndroid Build Coastguard Worker 	assert(prof_dump_fd != -1);
1002*1208bc7eSAndroid Build Coastguard Worker 	ret = prof_dump_flush(propagate_err);
1003*1208bc7eSAndroid Build Coastguard Worker 	close(prof_dump_fd);
1004*1208bc7eSAndroid Build Coastguard Worker 	prof_dump_fd = -1;
1005*1208bc7eSAndroid Build Coastguard Worker 
1006*1208bc7eSAndroid Build Coastguard Worker 	return ret;
1007*1208bc7eSAndroid Build Coastguard Worker }
1008*1208bc7eSAndroid Build Coastguard Worker 
1009*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_write(bool propagate_err,const char * s)1010*1208bc7eSAndroid Build Coastguard Worker prof_dump_write(bool propagate_err, const char *s) {
1011*1208bc7eSAndroid Build Coastguard Worker 	size_t i, slen, n;
1012*1208bc7eSAndroid Build Coastguard Worker 
1013*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1014*1208bc7eSAndroid Build Coastguard Worker 
1015*1208bc7eSAndroid Build Coastguard Worker 	i = 0;
1016*1208bc7eSAndroid Build Coastguard Worker 	slen = strlen(s);
1017*1208bc7eSAndroid Build Coastguard Worker 	while (i < slen) {
1018*1208bc7eSAndroid Build Coastguard Worker 		/* Flush the buffer if it is full. */
1019*1208bc7eSAndroid Build Coastguard Worker 		if (prof_dump_buf_end == PROF_DUMP_BUFSIZE) {
1020*1208bc7eSAndroid Build Coastguard Worker 			if (prof_dump_flush(propagate_err) && propagate_err) {
1021*1208bc7eSAndroid Build Coastguard Worker 				return true;
1022*1208bc7eSAndroid Build Coastguard Worker 			}
1023*1208bc7eSAndroid Build Coastguard Worker 		}
1024*1208bc7eSAndroid Build Coastguard Worker 
1025*1208bc7eSAndroid Build Coastguard Worker 		if (prof_dump_buf_end + slen <= PROF_DUMP_BUFSIZE) {
1026*1208bc7eSAndroid Build Coastguard Worker 			/* Finish writing. */
1027*1208bc7eSAndroid Build Coastguard Worker 			n = slen - i;
1028*1208bc7eSAndroid Build Coastguard Worker 		} else {
1029*1208bc7eSAndroid Build Coastguard Worker 			/* Write as much of s as will fit. */
1030*1208bc7eSAndroid Build Coastguard Worker 			n = PROF_DUMP_BUFSIZE - prof_dump_buf_end;
1031*1208bc7eSAndroid Build Coastguard Worker 		}
1032*1208bc7eSAndroid Build Coastguard Worker 		memcpy(&prof_dump_buf[prof_dump_buf_end], &s[i], n);
1033*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_buf_end += n;
1034*1208bc7eSAndroid Build Coastguard Worker 		i += n;
1035*1208bc7eSAndroid Build Coastguard Worker 	}
1036*1208bc7eSAndroid Build Coastguard Worker 
1037*1208bc7eSAndroid Build Coastguard Worker 	return false;
1038*1208bc7eSAndroid Build Coastguard Worker }
1039*1208bc7eSAndroid Build Coastguard Worker 
1040*1208bc7eSAndroid Build Coastguard Worker JEMALLOC_FORMAT_PRINTF(2, 3)
1041*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_printf(bool propagate_err,const char * format,...)1042*1208bc7eSAndroid Build Coastguard Worker prof_dump_printf(bool propagate_err, const char *format, ...) {
1043*1208bc7eSAndroid Build Coastguard Worker 	bool ret;
1044*1208bc7eSAndroid Build Coastguard Worker 	va_list ap;
1045*1208bc7eSAndroid Build Coastguard Worker 	char buf[PROF_PRINTF_BUFSIZE];
1046*1208bc7eSAndroid Build Coastguard Worker 
1047*1208bc7eSAndroid Build Coastguard Worker 	va_start(ap, format);
1048*1208bc7eSAndroid Build Coastguard Worker 	malloc_vsnprintf(buf, sizeof(buf), format, ap);
1049*1208bc7eSAndroid Build Coastguard Worker 	va_end(ap);
1050*1208bc7eSAndroid Build Coastguard Worker 	ret = prof_dump_write(propagate_err, buf);
1051*1208bc7eSAndroid Build Coastguard Worker 
1052*1208bc7eSAndroid Build Coastguard Worker 	return ret;
1053*1208bc7eSAndroid Build Coastguard Worker }
1054*1208bc7eSAndroid Build Coastguard Worker 
1055*1208bc7eSAndroid Build Coastguard Worker static void
prof_tctx_merge_tdata(tsdn_t * tsdn,prof_tctx_t * tctx,prof_tdata_t * tdata)1056*1208bc7eSAndroid Build Coastguard Worker prof_tctx_merge_tdata(tsdn_t *tsdn, prof_tctx_t *tctx, prof_tdata_t *tdata) {
1057*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsdn, tctx->tdata->lock);
1058*1208bc7eSAndroid Build Coastguard Worker 
1059*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, tctx->gctx->lock);
1060*1208bc7eSAndroid Build Coastguard Worker 
1061*1208bc7eSAndroid Build Coastguard Worker 	switch (tctx->state) {
1062*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_initializing:
1063*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsdn, tctx->gctx->lock);
1064*1208bc7eSAndroid Build Coastguard Worker 		return;
1065*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_nominal:
1066*1208bc7eSAndroid Build Coastguard Worker 		tctx->state = prof_tctx_state_dumping;
1067*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsdn, tctx->gctx->lock);
1068*1208bc7eSAndroid Build Coastguard Worker 
1069*1208bc7eSAndroid Build Coastguard Worker 		memcpy(&tctx->dump_cnts, &tctx->cnts, sizeof(prof_cnt_t));
1070*1208bc7eSAndroid Build Coastguard Worker 
1071*1208bc7eSAndroid Build Coastguard Worker 		tdata->cnt_summed.curobjs += tctx->dump_cnts.curobjs;
1072*1208bc7eSAndroid Build Coastguard Worker 		tdata->cnt_summed.curbytes += tctx->dump_cnts.curbytes;
1073*1208bc7eSAndroid Build Coastguard Worker 		if (opt_prof_accum) {
1074*1208bc7eSAndroid Build Coastguard Worker 			tdata->cnt_summed.accumobjs +=
1075*1208bc7eSAndroid Build Coastguard Worker 			    tctx->dump_cnts.accumobjs;
1076*1208bc7eSAndroid Build Coastguard Worker 			tdata->cnt_summed.accumbytes +=
1077*1208bc7eSAndroid Build Coastguard Worker 			    tctx->dump_cnts.accumbytes;
1078*1208bc7eSAndroid Build Coastguard Worker 		}
1079*1208bc7eSAndroid Build Coastguard Worker 		break;
1080*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_dumping:
1081*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_purgatory:
1082*1208bc7eSAndroid Build Coastguard Worker 		not_reached();
1083*1208bc7eSAndroid Build Coastguard Worker 	}
1084*1208bc7eSAndroid Build Coastguard Worker }
1085*1208bc7eSAndroid Build Coastguard Worker 
1086*1208bc7eSAndroid Build Coastguard Worker static void
prof_tctx_merge_gctx(tsdn_t * tsdn,prof_tctx_t * tctx,prof_gctx_t * gctx)1087*1208bc7eSAndroid Build Coastguard Worker prof_tctx_merge_gctx(tsdn_t *tsdn, prof_tctx_t *tctx, prof_gctx_t *gctx) {
1088*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsdn, gctx->lock);
1089*1208bc7eSAndroid Build Coastguard Worker 
1090*1208bc7eSAndroid Build Coastguard Worker 	gctx->cnt_summed.curobjs += tctx->dump_cnts.curobjs;
1091*1208bc7eSAndroid Build Coastguard Worker 	gctx->cnt_summed.curbytes += tctx->dump_cnts.curbytes;
1092*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof_accum) {
1093*1208bc7eSAndroid Build Coastguard Worker 		gctx->cnt_summed.accumobjs += tctx->dump_cnts.accumobjs;
1094*1208bc7eSAndroid Build Coastguard Worker 		gctx->cnt_summed.accumbytes += tctx->dump_cnts.accumbytes;
1095*1208bc7eSAndroid Build Coastguard Worker 	}
1096*1208bc7eSAndroid Build Coastguard Worker }
1097*1208bc7eSAndroid Build Coastguard Worker 
1098*1208bc7eSAndroid Build Coastguard Worker static prof_tctx_t *
prof_tctx_merge_iter(prof_tctx_tree_t * tctxs,prof_tctx_t * tctx,void * arg)1099*1208bc7eSAndroid Build Coastguard Worker prof_tctx_merge_iter(prof_tctx_tree_t *tctxs, prof_tctx_t *tctx, void *arg) {
1100*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t *tsdn = (tsdn_t *)arg;
1101*1208bc7eSAndroid Build Coastguard Worker 
1102*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsdn, tctx->gctx->lock);
1103*1208bc7eSAndroid Build Coastguard Worker 
1104*1208bc7eSAndroid Build Coastguard Worker 	switch (tctx->state) {
1105*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_nominal:
1106*1208bc7eSAndroid Build Coastguard Worker 		/* New since dumping started; ignore. */
1107*1208bc7eSAndroid Build Coastguard Worker 		break;
1108*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_dumping:
1109*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_purgatory:
1110*1208bc7eSAndroid Build Coastguard Worker 		prof_tctx_merge_gctx(tsdn, tctx, tctx->gctx);
1111*1208bc7eSAndroid Build Coastguard Worker 		break;
1112*1208bc7eSAndroid Build Coastguard Worker 	default:
1113*1208bc7eSAndroid Build Coastguard Worker 		not_reached();
1114*1208bc7eSAndroid Build Coastguard Worker 	}
1115*1208bc7eSAndroid Build Coastguard Worker 
1116*1208bc7eSAndroid Build Coastguard Worker 	return NULL;
1117*1208bc7eSAndroid Build Coastguard Worker }
1118*1208bc7eSAndroid Build Coastguard Worker 
1119*1208bc7eSAndroid Build Coastguard Worker struct prof_tctx_dump_iter_arg_s {
1120*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t	*tsdn;
1121*1208bc7eSAndroid Build Coastguard Worker 	bool	propagate_err;
1122*1208bc7eSAndroid Build Coastguard Worker };
1123*1208bc7eSAndroid Build Coastguard Worker 
1124*1208bc7eSAndroid Build Coastguard Worker static prof_tctx_t *
prof_tctx_dump_iter(prof_tctx_tree_t * tctxs,prof_tctx_t * tctx,void * opaque)1125*1208bc7eSAndroid Build Coastguard Worker prof_tctx_dump_iter(prof_tctx_tree_t *tctxs, prof_tctx_t *tctx, void *opaque) {
1126*1208bc7eSAndroid Build Coastguard Worker 	struct prof_tctx_dump_iter_arg_s *arg =
1127*1208bc7eSAndroid Build Coastguard Worker 	    (struct prof_tctx_dump_iter_arg_s *)opaque;
1128*1208bc7eSAndroid Build Coastguard Worker 
1129*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(arg->tsdn, tctx->gctx->lock);
1130*1208bc7eSAndroid Build Coastguard Worker 
1131*1208bc7eSAndroid Build Coastguard Worker 	switch (tctx->state) {
1132*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_initializing:
1133*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_nominal:
1134*1208bc7eSAndroid Build Coastguard Worker 		/* Not captured by this dump. */
1135*1208bc7eSAndroid Build Coastguard Worker 		break;
1136*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_dumping:
1137*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_purgatory:
1138*1208bc7eSAndroid Build Coastguard Worker 		if (prof_dump_printf(arg->propagate_err,
1139*1208bc7eSAndroid Build Coastguard Worker 		    "  t%"FMTu64": %"FMTu64": %"FMTu64" [%"FMTu64": "
1140*1208bc7eSAndroid Build Coastguard Worker 		    "%"FMTu64"]\n", tctx->thr_uid, tctx->dump_cnts.curobjs,
1141*1208bc7eSAndroid Build Coastguard Worker 		    tctx->dump_cnts.curbytes, tctx->dump_cnts.accumobjs,
1142*1208bc7eSAndroid Build Coastguard Worker 		    tctx->dump_cnts.accumbytes)) {
1143*1208bc7eSAndroid Build Coastguard Worker 			return tctx;
1144*1208bc7eSAndroid Build Coastguard Worker 		}
1145*1208bc7eSAndroid Build Coastguard Worker 		break;
1146*1208bc7eSAndroid Build Coastguard Worker 	default:
1147*1208bc7eSAndroid Build Coastguard Worker 		not_reached();
1148*1208bc7eSAndroid Build Coastguard Worker 	}
1149*1208bc7eSAndroid Build Coastguard Worker 	return NULL;
1150*1208bc7eSAndroid Build Coastguard Worker }
1151*1208bc7eSAndroid Build Coastguard Worker 
1152*1208bc7eSAndroid Build Coastguard Worker static prof_tctx_t *
prof_tctx_finish_iter(prof_tctx_tree_t * tctxs,prof_tctx_t * tctx,void * arg)1153*1208bc7eSAndroid Build Coastguard Worker prof_tctx_finish_iter(prof_tctx_tree_t *tctxs, prof_tctx_t *tctx, void *arg) {
1154*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t *tsdn = (tsdn_t *)arg;
1155*1208bc7eSAndroid Build Coastguard Worker 	prof_tctx_t *ret;
1156*1208bc7eSAndroid Build Coastguard Worker 
1157*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsdn, tctx->gctx->lock);
1158*1208bc7eSAndroid Build Coastguard Worker 
1159*1208bc7eSAndroid Build Coastguard Worker 	switch (tctx->state) {
1160*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_nominal:
1161*1208bc7eSAndroid Build Coastguard Worker 		/* New since dumping started; ignore. */
1162*1208bc7eSAndroid Build Coastguard Worker 		break;
1163*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_dumping:
1164*1208bc7eSAndroid Build Coastguard Worker 		tctx->state = prof_tctx_state_nominal;
1165*1208bc7eSAndroid Build Coastguard Worker 		break;
1166*1208bc7eSAndroid Build Coastguard Worker 	case prof_tctx_state_purgatory:
1167*1208bc7eSAndroid Build Coastguard Worker 		ret = tctx;
1168*1208bc7eSAndroid Build Coastguard Worker 		goto label_return;
1169*1208bc7eSAndroid Build Coastguard Worker 	default:
1170*1208bc7eSAndroid Build Coastguard Worker 		not_reached();
1171*1208bc7eSAndroid Build Coastguard Worker 	}
1172*1208bc7eSAndroid Build Coastguard Worker 
1173*1208bc7eSAndroid Build Coastguard Worker 	ret = NULL;
1174*1208bc7eSAndroid Build Coastguard Worker label_return:
1175*1208bc7eSAndroid Build Coastguard Worker 	return ret;
1176*1208bc7eSAndroid Build Coastguard Worker }
1177*1208bc7eSAndroid Build Coastguard Worker 
1178*1208bc7eSAndroid Build Coastguard Worker static void
prof_dump_gctx_prep(tsdn_t * tsdn,prof_gctx_t * gctx,prof_gctx_tree_t * gctxs)1179*1208bc7eSAndroid Build Coastguard Worker prof_dump_gctx_prep(tsdn_t *tsdn, prof_gctx_t *gctx, prof_gctx_tree_t *gctxs) {
1180*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1181*1208bc7eSAndroid Build Coastguard Worker 
1182*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, gctx->lock);
1183*1208bc7eSAndroid Build Coastguard Worker 
1184*1208bc7eSAndroid Build Coastguard Worker 	/*
1185*1208bc7eSAndroid Build Coastguard Worker 	 * Increment nlimbo so that gctx won't go away before dump.
1186*1208bc7eSAndroid Build Coastguard Worker 	 * Additionally, link gctx into the dump list so that it is included in
1187*1208bc7eSAndroid Build Coastguard Worker 	 * prof_dump()'s second pass.
1188*1208bc7eSAndroid Build Coastguard Worker 	 */
1189*1208bc7eSAndroid Build Coastguard Worker 	gctx->nlimbo++;
1190*1208bc7eSAndroid Build Coastguard Worker 	gctx_tree_insert(gctxs, gctx);
1191*1208bc7eSAndroid Build Coastguard Worker 
1192*1208bc7eSAndroid Build Coastguard Worker 	memset(&gctx->cnt_summed, 0, sizeof(prof_cnt_t));
1193*1208bc7eSAndroid Build Coastguard Worker 
1194*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, gctx->lock);
1195*1208bc7eSAndroid Build Coastguard Worker }
1196*1208bc7eSAndroid Build Coastguard Worker 
1197*1208bc7eSAndroid Build Coastguard Worker struct prof_gctx_merge_iter_arg_s {
1198*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t	*tsdn;
1199*1208bc7eSAndroid Build Coastguard Worker 	size_t	leak_ngctx;
1200*1208bc7eSAndroid Build Coastguard Worker };
1201*1208bc7eSAndroid Build Coastguard Worker 
1202*1208bc7eSAndroid Build Coastguard Worker static prof_gctx_t *
prof_gctx_merge_iter(prof_gctx_tree_t * gctxs,prof_gctx_t * gctx,void * opaque)1203*1208bc7eSAndroid Build Coastguard Worker prof_gctx_merge_iter(prof_gctx_tree_t *gctxs, prof_gctx_t *gctx, void *opaque) {
1204*1208bc7eSAndroid Build Coastguard Worker 	struct prof_gctx_merge_iter_arg_s *arg =
1205*1208bc7eSAndroid Build Coastguard Worker 	    (struct prof_gctx_merge_iter_arg_s *)opaque;
1206*1208bc7eSAndroid Build Coastguard Worker 
1207*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(arg->tsdn, gctx->lock);
1208*1208bc7eSAndroid Build Coastguard Worker 	tctx_tree_iter(&gctx->tctxs, NULL, prof_tctx_merge_iter,
1209*1208bc7eSAndroid Build Coastguard Worker 	    (void *)arg->tsdn);
1210*1208bc7eSAndroid Build Coastguard Worker 	if (gctx->cnt_summed.curobjs != 0) {
1211*1208bc7eSAndroid Build Coastguard Worker 		arg->leak_ngctx++;
1212*1208bc7eSAndroid Build Coastguard Worker 	}
1213*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(arg->tsdn, gctx->lock);
1214*1208bc7eSAndroid Build Coastguard Worker 
1215*1208bc7eSAndroid Build Coastguard Worker 	return NULL;
1216*1208bc7eSAndroid Build Coastguard Worker }
1217*1208bc7eSAndroid Build Coastguard Worker 
1218*1208bc7eSAndroid Build Coastguard Worker static void
prof_gctx_finish(tsd_t * tsd,prof_gctx_tree_t * gctxs)1219*1208bc7eSAndroid Build Coastguard Worker prof_gctx_finish(tsd_t *tsd, prof_gctx_tree_t *gctxs) {
1220*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata = prof_tdata_get(tsd, false);
1221*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_t *gctx;
1222*1208bc7eSAndroid Build Coastguard Worker 
1223*1208bc7eSAndroid Build Coastguard Worker 	/*
1224*1208bc7eSAndroid Build Coastguard Worker 	 * Standard tree iteration won't work here, because as soon as we
1225*1208bc7eSAndroid Build Coastguard Worker 	 * decrement gctx->nlimbo and unlock gctx, another thread can
1226*1208bc7eSAndroid Build Coastguard Worker 	 * concurrently destroy it, which will corrupt the tree.  Therefore,
1227*1208bc7eSAndroid Build Coastguard Worker 	 * tear down the tree one node at a time during iteration.
1228*1208bc7eSAndroid Build Coastguard Worker 	 */
1229*1208bc7eSAndroid Build Coastguard Worker 	while ((gctx = gctx_tree_first(gctxs)) != NULL) {
1230*1208bc7eSAndroid Build Coastguard Worker 		gctx_tree_remove(gctxs, gctx);
1231*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsd_tsdn(tsd), gctx->lock);
1232*1208bc7eSAndroid Build Coastguard Worker 		{
1233*1208bc7eSAndroid Build Coastguard Worker 			prof_tctx_t *next;
1234*1208bc7eSAndroid Build Coastguard Worker 
1235*1208bc7eSAndroid Build Coastguard Worker 			next = NULL;
1236*1208bc7eSAndroid Build Coastguard Worker 			do {
1237*1208bc7eSAndroid Build Coastguard Worker 				prof_tctx_t *to_destroy =
1238*1208bc7eSAndroid Build Coastguard Worker 				    tctx_tree_iter(&gctx->tctxs, next,
1239*1208bc7eSAndroid Build Coastguard Worker 				    prof_tctx_finish_iter,
1240*1208bc7eSAndroid Build Coastguard Worker 				    (void *)tsd_tsdn(tsd));
1241*1208bc7eSAndroid Build Coastguard Worker 				if (to_destroy != NULL) {
1242*1208bc7eSAndroid Build Coastguard Worker 					next = tctx_tree_next(&gctx->tctxs,
1243*1208bc7eSAndroid Build Coastguard Worker 					    to_destroy);
1244*1208bc7eSAndroid Build Coastguard Worker 					tctx_tree_remove(&gctx->tctxs,
1245*1208bc7eSAndroid Build Coastguard Worker 					    to_destroy);
1246*1208bc7eSAndroid Build Coastguard Worker 					idalloctm(tsd_tsdn(tsd), to_destroy,
1247*1208bc7eSAndroid Build Coastguard Worker 					    NULL, NULL, true, true);
1248*1208bc7eSAndroid Build Coastguard Worker 				} else {
1249*1208bc7eSAndroid Build Coastguard Worker 					next = NULL;
1250*1208bc7eSAndroid Build Coastguard Worker 				}
1251*1208bc7eSAndroid Build Coastguard Worker 			} while (next != NULL);
1252*1208bc7eSAndroid Build Coastguard Worker 		}
1253*1208bc7eSAndroid Build Coastguard Worker 		gctx->nlimbo--;
1254*1208bc7eSAndroid Build Coastguard Worker 		if (prof_gctx_should_destroy(gctx)) {
1255*1208bc7eSAndroid Build Coastguard Worker 			gctx->nlimbo++;
1256*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_unlock(tsd_tsdn(tsd), gctx->lock);
1257*1208bc7eSAndroid Build Coastguard Worker 			prof_gctx_try_destroy(tsd, tdata, gctx, tdata);
1258*1208bc7eSAndroid Build Coastguard Worker 		} else {
1259*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_unlock(tsd_tsdn(tsd), gctx->lock);
1260*1208bc7eSAndroid Build Coastguard Worker 		}
1261*1208bc7eSAndroid Build Coastguard Worker 	}
1262*1208bc7eSAndroid Build Coastguard Worker }
1263*1208bc7eSAndroid Build Coastguard Worker 
1264*1208bc7eSAndroid Build Coastguard Worker struct prof_tdata_merge_iter_arg_s {
1265*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t		*tsdn;
1266*1208bc7eSAndroid Build Coastguard Worker 	prof_cnt_t	cnt_all;
1267*1208bc7eSAndroid Build Coastguard Worker };
1268*1208bc7eSAndroid Build Coastguard Worker 
1269*1208bc7eSAndroid Build Coastguard Worker static prof_tdata_t *
prof_tdata_merge_iter(prof_tdata_tree_t * tdatas,prof_tdata_t * tdata,void * opaque)1270*1208bc7eSAndroid Build Coastguard Worker prof_tdata_merge_iter(prof_tdata_tree_t *tdatas, prof_tdata_t *tdata,
1271*1208bc7eSAndroid Build Coastguard Worker     void *opaque) {
1272*1208bc7eSAndroid Build Coastguard Worker 	struct prof_tdata_merge_iter_arg_s *arg =
1273*1208bc7eSAndroid Build Coastguard Worker 	    (struct prof_tdata_merge_iter_arg_s *)opaque;
1274*1208bc7eSAndroid Build Coastguard Worker 
1275*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(arg->tsdn, tdata->lock);
1276*1208bc7eSAndroid Build Coastguard Worker 	if (!tdata->expired) {
1277*1208bc7eSAndroid Build Coastguard Worker 		size_t tabind;
1278*1208bc7eSAndroid Build Coastguard Worker 		union {
1279*1208bc7eSAndroid Build Coastguard Worker 			prof_tctx_t	*p;
1280*1208bc7eSAndroid Build Coastguard Worker 			void		*v;
1281*1208bc7eSAndroid Build Coastguard Worker 		} tctx;
1282*1208bc7eSAndroid Build Coastguard Worker 
1283*1208bc7eSAndroid Build Coastguard Worker 		tdata->dumping = true;
1284*1208bc7eSAndroid Build Coastguard Worker 		memset(&tdata->cnt_summed, 0, sizeof(prof_cnt_t));
1285*1208bc7eSAndroid Build Coastguard Worker 		for (tabind = 0; !ckh_iter(&tdata->bt2tctx, &tabind, NULL,
1286*1208bc7eSAndroid Build Coastguard Worker 		    &tctx.v);) {
1287*1208bc7eSAndroid Build Coastguard Worker 			prof_tctx_merge_tdata(arg->tsdn, tctx.p, tdata);
1288*1208bc7eSAndroid Build Coastguard Worker 		}
1289*1208bc7eSAndroid Build Coastguard Worker 
1290*1208bc7eSAndroid Build Coastguard Worker 		arg->cnt_all.curobjs += tdata->cnt_summed.curobjs;
1291*1208bc7eSAndroid Build Coastguard Worker 		arg->cnt_all.curbytes += tdata->cnt_summed.curbytes;
1292*1208bc7eSAndroid Build Coastguard Worker 		if (opt_prof_accum) {
1293*1208bc7eSAndroid Build Coastguard Worker 			arg->cnt_all.accumobjs += tdata->cnt_summed.accumobjs;
1294*1208bc7eSAndroid Build Coastguard Worker 			arg->cnt_all.accumbytes += tdata->cnt_summed.accumbytes;
1295*1208bc7eSAndroid Build Coastguard Worker 		}
1296*1208bc7eSAndroid Build Coastguard Worker 	} else {
1297*1208bc7eSAndroid Build Coastguard Worker 		tdata->dumping = false;
1298*1208bc7eSAndroid Build Coastguard Worker 	}
1299*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(arg->tsdn, tdata->lock);
1300*1208bc7eSAndroid Build Coastguard Worker 
1301*1208bc7eSAndroid Build Coastguard Worker 	return NULL;
1302*1208bc7eSAndroid Build Coastguard Worker }
1303*1208bc7eSAndroid Build Coastguard Worker 
1304*1208bc7eSAndroid Build Coastguard Worker static prof_tdata_t *
prof_tdata_dump_iter(prof_tdata_tree_t * tdatas,prof_tdata_t * tdata,void * arg)1305*1208bc7eSAndroid Build Coastguard Worker prof_tdata_dump_iter(prof_tdata_tree_t *tdatas, prof_tdata_t *tdata,
1306*1208bc7eSAndroid Build Coastguard Worker     void *arg) {
1307*1208bc7eSAndroid Build Coastguard Worker 	bool propagate_err = *(bool *)arg;
1308*1208bc7eSAndroid Build Coastguard Worker 
1309*1208bc7eSAndroid Build Coastguard Worker 	if (!tdata->dumping) {
1310*1208bc7eSAndroid Build Coastguard Worker 		return NULL;
1311*1208bc7eSAndroid Build Coastguard Worker 	}
1312*1208bc7eSAndroid Build Coastguard Worker 
1313*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_printf(propagate_err,
1314*1208bc7eSAndroid Build Coastguard Worker 	    "  t%"FMTu64": %"FMTu64": %"FMTu64" [%"FMTu64": %"FMTu64"]%s%s\n",
1315*1208bc7eSAndroid Build Coastguard Worker 	    tdata->thr_uid, tdata->cnt_summed.curobjs,
1316*1208bc7eSAndroid Build Coastguard Worker 	    tdata->cnt_summed.curbytes, tdata->cnt_summed.accumobjs,
1317*1208bc7eSAndroid Build Coastguard Worker 	    tdata->cnt_summed.accumbytes,
1318*1208bc7eSAndroid Build Coastguard Worker 	    (tdata->thread_name != NULL) ? " " : "",
1319*1208bc7eSAndroid Build Coastguard Worker 	    (tdata->thread_name != NULL) ? tdata->thread_name : "")) {
1320*1208bc7eSAndroid Build Coastguard Worker 		return tdata;
1321*1208bc7eSAndroid Build Coastguard Worker 	}
1322*1208bc7eSAndroid Build Coastguard Worker 	return NULL;
1323*1208bc7eSAndroid Build Coastguard Worker }
1324*1208bc7eSAndroid Build Coastguard Worker 
1325*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_header_impl(tsdn_t * tsdn,bool propagate_err,const prof_cnt_t * cnt_all)1326*1208bc7eSAndroid Build Coastguard Worker prof_dump_header_impl(tsdn_t *tsdn, bool propagate_err,
1327*1208bc7eSAndroid Build Coastguard Worker     const prof_cnt_t *cnt_all) {
1328*1208bc7eSAndroid Build Coastguard Worker 	bool ret;
1329*1208bc7eSAndroid Build Coastguard Worker 
1330*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_printf(propagate_err,
1331*1208bc7eSAndroid Build Coastguard Worker 	    "heap_v2/%"FMTu64"\n"
1332*1208bc7eSAndroid Build Coastguard Worker 	    "  t*: %"FMTu64": %"FMTu64" [%"FMTu64": %"FMTu64"]\n",
1333*1208bc7eSAndroid Build Coastguard Worker 	    ((uint64_t)1U << lg_prof_sample), cnt_all->curobjs,
1334*1208bc7eSAndroid Build Coastguard Worker 	    cnt_all->curbytes, cnt_all->accumobjs, cnt_all->accumbytes)) {
1335*1208bc7eSAndroid Build Coastguard Worker 		return true;
1336*1208bc7eSAndroid Build Coastguard Worker 	}
1337*1208bc7eSAndroid Build Coastguard Worker 
1338*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &tdatas_mtx);
1339*1208bc7eSAndroid Build Coastguard Worker 	ret = (tdata_tree_iter(&tdatas, NULL, prof_tdata_dump_iter,
1340*1208bc7eSAndroid Build Coastguard Worker 	    (void *)&propagate_err) != NULL);
1341*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &tdatas_mtx);
1342*1208bc7eSAndroid Build Coastguard Worker 	return ret;
1343*1208bc7eSAndroid Build Coastguard Worker }
1344*1208bc7eSAndroid Build Coastguard Worker prof_dump_header_t *JET_MUTABLE prof_dump_header = prof_dump_header_impl;
1345*1208bc7eSAndroid Build Coastguard Worker 
1346*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_gctx(tsdn_t * tsdn,bool propagate_err,prof_gctx_t * gctx,const prof_bt_t * bt,prof_gctx_tree_t * gctxs)1347*1208bc7eSAndroid Build Coastguard Worker prof_dump_gctx(tsdn_t *tsdn, bool propagate_err, prof_gctx_t *gctx,
1348*1208bc7eSAndroid Build Coastguard Worker     const prof_bt_t *bt, prof_gctx_tree_t *gctxs) {
1349*1208bc7eSAndroid Build Coastguard Worker 	bool ret;
1350*1208bc7eSAndroid Build Coastguard Worker 	unsigned i;
1351*1208bc7eSAndroid Build Coastguard Worker 	struct prof_tctx_dump_iter_arg_s prof_tctx_dump_iter_arg;
1352*1208bc7eSAndroid Build Coastguard Worker 
1353*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1354*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsdn, gctx->lock);
1355*1208bc7eSAndroid Build Coastguard Worker 
1356*1208bc7eSAndroid Build Coastguard Worker 	/* Avoid dumping such gctx's that have no useful data. */
1357*1208bc7eSAndroid Build Coastguard Worker 	if ((!opt_prof_accum && gctx->cnt_summed.curobjs == 0) ||
1358*1208bc7eSAndroid Build Coastguard Worker 	    (opt_prof_accum && gctx->cnt_summed.accumobjs == 0)) {
1359*1208bc7eSAndroid Build Coastguard Worker 		assert(gctx->cnt_summed.curobjs == 0);
1360*1208bc7eSAndroid Build Coastguard Worker 		assert(gctx->cnt_summed.curbytes == 0);
1361*1208bc7eSAndroid Build Coastguard Worker 		assert(gctx->cnt_summed.accumobjs == 0);
1362*1208bc7eSAndroid Build Coastguard Worker 		assert(gctx->cnt_summed.accumbytes == 0);
1363*1208bc7eSAndroid Build Coastguard Worker 		ret = false;
1364*1208bc7eSAndroid Build Coastguard Worker 		goto label_return;
1365*1208bc7eSAndroid Build Coastguard Worker 	}
1366*1208bc7eSAndroid Build Coastguard Worker 
1367*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_printf(propagate_err, "@")) {
1368*1208bc7eSAndroid Build Coastguard Worker 		ret = true;
1369*1208bc7eSAndroid Build Coastguard Worker 		goto label_return;
1370*1208bc7eSAndroid Build Coastguard Worker 	}
1371*1208bc7eSAndroid Build Coastguard Worker 	for (i = 0; i < bt->len; i++) {
1372*1208bc7eSAndroid Build Coastguard Worker 		if (prof_dump_printf(propagate_err, " %#"FMTxPTR,
1373*1208bc7eSAndroid Build Coastguard Worker 		    (uintptr_t)bt->vec[i])) {
1374*1208bc7eSAndroid Build Coastguard Worker 			ret = true;
1375*1208bc7eSAndroid Build Coastguard Worker 			goto label_return;
1376*1208bc7eSAndroid Build Coastguard Worker 		}
1377*1208bc7eSAndroid Build Coastguard Worker 	}
1378*1208bc7eSAndroid Build Coastguard Worker 
1379*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_printf(propagate_err,
1380*1208bc7eSAndroid Build Coastguard Worker 	    "\n"
1381*1208bc7eSAndroid Build Coastguard Worker 	    "  t*: %"FMTu64": %"FMTu64" [%"FMTu64": %"FMTu64"]\n",
1382*1208bc7eSAndroid Build Coastguard Worker 	    gctx->cnt_summed.curobjs, gctx->cnt_summed.curbytes,
1383*1208bc7eSAndroid Build Coastguard Worker 	    gctx->cnt_summed.accumobjs, gctx->cnt_summed.accumbytes)) {
1384*1208bc7eSAndroid Build Coastguard Worker 		ret = true;
1385*1208bc7eSAndroid Build Coastguard Worker 		goto label_return;
1386*1208bc7eSAndroid Build Coastguard Worker 	}
1387*1208bc7eSAndroid Build Coastguard Worker 
1388*1208bc7eSAndroid Build Coastguard Worker 	prof_tctx_dump_iter_arg.tsdn = tsdn;
1389*1208bc7eSAndroid Build Coastguard Worker 	prof_tctx_dump_iter_arg.propagate_err = propagate_err;
1390*1208bc7eSAndroid Build Coastguard Worker 	if (tctx_tree_iter(&gctx->tctxs, NULL, prof_tctx_dump_iter,
1391*1208bc7eSAndroid Build Coastguard Worker 	    (void *)&prof_tctx_dump_iter_arg) != NULL) {
1392*1208bc7eSAndroid Build Coastguard Worker 		ret = true;
1393*1208bc7eSAndroid Build Coastguard Worker 		goto label_return;
1394*1208bc7eSAndroid Build Coastguard Worker 	}
1395*1208bc7eSAndroid Build Coastguard Worker 
1396*1208bc7eSAndroid Build Coastguard Worker 	ret = false;
1397*1208bc7eSAndroid Build Coastguard Worker label_return:
1398*1208bc7eSAndroid Build Coastguard Worker 	return ret;
1399*1208bc7eSAndroid Build Coastguard Worker }
1400*1208bc7eSAndroid Build Coastguard Worker 
1401*1208bc7eSAndroid Build Coastguard Worker #ifndef _WIN32
1402*1208bc7eSAndroid Build Coastguard Worker JEMALLOC_FORMAT_PRINTF(1, 2)
1403*1208bc7eSAndroid Build Coastguard Worker static int
prof_open_maps(const char * format,...)1404*1208bc7eSAndroid Build Coastguard Worker prof_open_maps(const char *format, ...) {
1405*1208bc7eSAndroid Build Coastguard Worker 	int mfd;
1406*1208bc7eSAndroid Build Coastguard Worker 	va_list ap;
1407*1208bc7eSAndroid Build Coastguard Worker 	char filename[PATH_MAX + 1];
1408*1208bc7eSAndroid Build Coastguard Worker 
1409*1208bc7eSAndroid Build Coastguard Worker 	va_start(ap, format);
1410*1208bc7eSAndroid Build Coastguard Worker 	malloc_vsnprintf(filename, sizeof(filename), format, ap);
1411*1208bc7eSAndroid Build Coastguard Worker 	va_end(ap);
1412*1208bc7eSAndroid Build Coastguard Worker 
1413*1208bc7eSAndroid Build Coastguard Worker #if defined(O_CLOEXEC)
1414*1208bc7eSAndroid Build Coastguard Worker 	mfd = open(filename, O_RDONLY | O_CLOEXEC);
1415*1208bc7eSAndroid Build Coastguard Worker #else
1416*1208bc7eSAndroid Build Coastguard Worker 	mfd = open(filename, O_RDONLY);
1417*1208bc7eSAndroid Build Coastguard Worker 	if (mfd != -1) {
1418*1208bc7eSAndroid Build Coastguard Worker 		fcntl(mfd, F_SETFD, fcntl(mfd, F_GETFD) | FD_CLOEXEC);
1419*1208bc7eSAndroid Build Coastguard Worker 	}
1420*1208bc7eSAndroid Build Coastguard Worker #endif
1421*1208bc7eSAndroid Build Coastguard Worker 
1422*1208bc7eSAndroid Build Coastguard Worker 	return mfd;
1423*1208bc7eSAndroid Build Coastguard Worker }
1424*1208bc7eSAndroid Build Coastguard Worker #endif
1425*1208bc7eSAndroid Build Coastguard Worker 
1426*1208bc7eSAndroid Build Coastguard Worker static int
prof_getpid(void)1427*1208bc7eSAndroid Build Coastguard Worker prof_getpid(void) {
1428*1208bc7eSAndroid Build Coastguard Worker #ifdef _WIN32
1429*1208bc7eSAndroid Build Coastguard Worker 	return GetCurrentProcessId();
1430*1208bc7eSAndroid Build Coastguard Worker #else
1431*1208bc7eSAndroid Build Coastguard Worker 	return getpid();
1432*1208bc7eSAndroid Build Coastguard Worker #endif
1433*1208bc7eSAndroid Build Coastguard Worker }
1434*1208bc7eSAndroid Build Coastguard Worker 
1435*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_maps(bool propagate_err)1436*1208bc7eSAndroid Build Coastguard Worker prof_dump_maps(bool propagate_err) {
1437*1208bc7eSAndroid Build Coastguard Worker 	bool ret;
1438*1208bc7eSAndroid Build Coastguard Worker 	int mfd;
1439*1208bc7eSAndroid Build Coastguard Worker 
1440*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1441*1208bc7eSAndroid Build Coastguard Worker #ifdef __FreeBSD__
1442*1208bc7eSAndroid Build Coastguard Worker 	mfd = prof_open_maps("/proc/curproc/map");
1443*1208bc7eSAndroid Build Coastguard Worker #elif defined(_WIN32)
1444*1208bc7eSAndroid Build Coastguard Worker 	mfd = -1; // Not implemented
1445*1208bc7eSAndroid Build Coastguard Worker #else
1446*1208bc7eSAndroid Build Coastguard Worker 	{
1447*1208bc7eSAndroid Build Coastguard Worker 		int pid = prof_getpid();
1448*1208bc7eSAndroid Build Coastguard Worker 
1449*1208bc7eSAndroid Build Coastguard Worker 		mfd = prof_open_maps("/proc/%d/task/%d/maps", pid, pid);
1450*1208bc7eSAndroid Build Coastguard Worker 		if (mfd == -1) {
1451*1208bc7eSAndroid Build Coastguard Worker 			mfd = prof_open_maps("/proc/%d/maps", pid);
1452*1208bc7eSAndroid Build Coastguard Worker 		}
1453*1208bc7eSAndroid Build Coastguard Worker 	}
1454*1208bc7eSAndroid Build Coastguard Worker #endif
1455*1208bc7eSAndroid Build Coastguard Worker 	if (mfd != -1) {
1456*1208bc7eSAndroid Build Coastguard Worker 		ssize_t nread;
1457*1208bc7eSAndroid Build Coastguard Worker 
1458*1208bc7eSAndroid Build Coastguard Worker 		if (prof_dump_write(propagate_err, "\nMAPPED_LIBRARIES:\n") &&
1459*1208bc7eSAndroid Build Coastguard Worker 		    propagate_err) {
1460*1208bc7eSAndroid Build Coastguard Worker 			ret = true;
1461*1208bc7eSAndroid Build Coastguard Worker 			goto label_return;
1462*1208bc7eSAndroid Build Coastguard Worker 		}
1463*1208bc7eSAndroid Build Coastguard Worker 		nread = 0;
1464*1208bc7eSAndroid Build Coastguard Worker 		do {
1465*1208bc7eSAndroid Build Coastguard Worker 			prof_dump_buf_end += nread;
1466*1208bc7eSAndroid Build Coastguard Worker 			if (prof_dump_buf_end == PROF_DUMP_BUFSIZE) {
1467*1208bc7eSAndroid Build Coastguard Worker 				/* Make space in prof_dump_buf before read(). */
1468*1208bc7eSAndroid Build Coastguard Worker 				if (prof_dump_flush(propagate_err) &&
1469*1208bc7eSAndroid Build Coastguard Worker 				    propagate_err) {
1470*1208bc7eSAndroid Build Coastguard Worker 					ret = true;
1471*1208bc7eSAndroid Build Coastguard Worker 					goto label_return;
1472*1208bc7eSAndroid Build Coastguard Worker 				}
1473*1208bc7eSAndroid Build Coastguard Worker 			}
1474*1208bc7eSAndroid Build Coastguard Worker 			nread = malloc_read_fd(mfd,
1475*1208bc7eSAndroid Build Coastguard Worker 			    &prof_dump_buf[prof_dump_buf_end], PROF_DUMP_BUFSIZE
1476*1208bc7eSAndroid Build Coastguard Worker 			    - prof_dump_buf_end);
1477*1208bc7eSAndroid Build Coastguard Worker 		} while (nread > 0);
1478*1208bc7eSAndroid Build Coastguard Worker 	} else {
1479*1208bc7eSAndroid Build Coastguard Worker 		ret = true;
1480*1208bc7eSAndroid Build Coastguard Worker 		goto label_return;
1481*1208bc7eSAndroid Build Coastguard Worker 	}
1482*1208bc7eSAndroid Build Coastguard Worker 
1483*1208bc7eSAndroid Build Coastguard Worker 	ret = false;
1484*1208bc7eSAndroid Build Coastguard Worker label_return:
1485*1208bc7eSAndroid Build Coastguard Worker 	if (mfd != -1) {
1486*1208bc7eSAndroid Build Coastguard Worker 		close(mfd);
1487*1208bc7eSAndroid Build Coastguard Worker 	}
1488*1208bc7eSAndroid Build Coastguard Worker 	return ret;
1489*1208bc7eSAndroid Build Coastguard Worker }
1490*1208bc7eSAndroid Build Coastguard Worker 
1491*1208bc7eSAndroid Build Coastguard Worker /*
1492*1208bc7eSAndroid Build Coastguard Worker  * See prof_sample_threshold_update() comment for why the body of this function
1493*1208bc7eSAndroid Build Coastguard Worker  * is conditionally compiled.
1494*1208bc7eSAndroid Build Coastguard Worker  */
1495*1208bc7eSAndroid Build Coastguard Worker static void
prof_leakcheck(const prof_cnt_t * cnt_all,size_t leak_ngctx,const char * filename)1496*1208bc7eSAndroid Build Coastguard Worker prof_leakcheck(const prof_cnt_t *cnt_all, size_t leak_ngctx,
1497*1208bc7eSAndroid Build Coastguard Worker     const char *filename) {
1498*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF
1499*1208bc7eSAndroid Build Coastguard Worker 	/*
1500*1208bc7eSAndroid Build Coastguard Worker 	 * Scaling is equivalent AdjustSamples() in jeprof, but the result may
1501*1208bc7eSAndroid Build Coastguard Worker 	 * differ slightly from what jeprof reports, because here we scale the
1502*1208bc7eSAndroid Build Coastguard Worker 	 * summary values, whereas jeprof scales each context individually and
1503*1208bc7eSAndroid Build Coastguard Worker 	 * reports the sums of the scaled values.
1504*1208bc7eSAndroid Build Coastguard Worker 	 */
1505*1208bc7eSAndroid Build Coastguard Worker 	if (cnt_all->curbytes != 0) {
1506*1208bc7eSAndroid Build Coastguard Worker 		double sample_period = (double)((uint64_t)1 << lg_prof_sample);
1507*1208bc7eSAndroid Build Coastguard Worker 		double ratio = (((double)cnt_all->curbytes) /
1508*1208bc7eSAndroid Build Coastguard Worker 		    (double)cnt_all->curobjs) / sample_period;
1509*1208bc7eSAndroid Build Coastguard Worker 		double scale_factor = 1.0 / (1.0 - exp(-ratio));
1510*1208bc7eSAndroid Build Coastguard Worker 		uint64_t curbytes = (uint64_t)round(((double)cnt_all->curbytes)
1511*1208bc7eSAndroid Build Coastguard Worker 		    * scale_factor);
1512*1208bc7eSAndroid Build Coastguard Worker 		uint64_t curobjs = (uint64_t)round(((double)cnt_all->curobjs) *
1513*1208bc7eSAndroid Build Coastguard Worker 		    scale_factor);
1514*1208bc7eSAndroid Build Coastguard Worker 
1515*1208bc7eSAndroid Build Coastguard Worker 		malloc_printf("<jemalloc>: Leak approximation summary: ~%"FMTu64
1516*1208bc7eSAndroid Build Coastguard Worker 		    " byte%s, ~%"FMTu64" object%s, >= %zu context%s\n",
1517*1208bc7eSAndroid Build Coastguard Worker 		    curbytes, (curbytes != 1) ? "s" : "", curobjs, (curobjs !=
1518*1208bc7eSAndroid Build Coastguard Worker 		    1) ? "s" : "", leak_ngctx, (leak_ngctx != 1) ? "s" : "");
1519*1208bc7eSAndroid Build Coastguard Worker 		malloc_printf(
1520*1208bc7eSAndroid Build Coastguard Worker 		    "<jemalloc>: Run jeprof on \"%s\" for leak detail\n",
1521*1208bc7eSAndroid Build Coastguard Worker 		    filename);
1522*1208bc7eSAndroid Build Coastguard Worker 	}
1523*1208bc7eSAndroid Build Coastguard Worker #endif
1524*1208bc7eSAndroid Build Coastguard Worker }
1525*1208bc7eSAndroid Build Coastguard Worker 
1526*1208bc7eSAndroid Build Coastguard Worker struct prof_gctx_dump_iter_arg_s {
1527*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t	*tsdn;
1528*1208bc7eSAndroid Build Coastguard Worker 	bool	propagate_err;
1529*1208bc7eSAndroid Build Coastguard Worker };
1530*1208bc7eSAndroid Build Coastguard Worker 
1531*1208bc7eSAndroid Build Coastguard Worker static prof_gctx_t *
prof_gctx_dump_iter(prof_gctx_tree_t * gctxs,prof_gctx_t * gctx,void * opaque)1532*1208bc7eSAndroid Build Coastguard Worker prof_gctx_dump_iter(prof_gctx_tree_t *gctxs, prof_gctx_t *gctx, void *opaque) {
1533*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_t *ret;
1534*1208bc7eSAndroid Build Coastguard Worker 	struct prof_gctx_dump_iter_arg_s *arg =
1535*1208bc7eSAndroid Build Coastguard Worker 	    (struct prof_gctx_dump_iter_arg_s *)opaque;
1536*1208bc7eSAndroid Build Coastguard Worker 
1537*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(arg->tsdn, gctx->lock);
1538*1208bc7eSAndroid Build Coastguard Worker 
1539*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_gctx(arg->tsdn, arg->propagate_err, gctx, &gctx->bt,
1540*1208bc7eSAndroid Build Coastguard Worker 	    gctxs)) {
1541*1208bc7eSAndroid Build Coastguard Worker 		ret = gctx;
1542*1208bc7eSAndroid Build Coastguard Worker 		goto label_return;
1543*1208bc7eSAndroid Build Coastguard Worker 	}
1544*1208bc7eSAndroid Build Coastguard Worker 
1545*1208bc7eSAndroid Build Coastguard Worker 	ret = NULL;
1546*1208bc7eSAndroid Build Coastguard Worker label_return:
1547*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(arg->tsdn, gctx->lock);
1548*1208bc7eSAndroid Build Coastguard Worker 	return ret;
1549*1208bc7eSAndroid Build Coastguard Worker }
1550*1208bc7eSAndroid Build Coastguard Worker 
1551*1208bc7eSAndroid Build Coastguard Worker static void
prof_dump_prep(tsd_t * tsd,prof_tdata_t * tdata,struct prof_tdata_merge_iter_arg_s * prof_tdata_merge_iter_arg,struct prof_gctx_merge_iter_arg_s * prof_gctx_merge_iter_arg,prof_gctx_tree_t * gctxs)1552*1208bc7eSAndroid Build Coastguard Worker prof_dump_prep(tsd_t *tsd, prof_tdata_t *tdata,
1553*1208bc7eSAndroid Build Coastguard Worker     struct prof_tdata_merge_iter_arg_s *prof_tdata_merge_iter_arg,
1554*1208bc7eSAndroid Build Coastguard Worker     struct prof_gctx_merge_iter_arg_s *prof_gctx_merge_iter_arg,
1555*1208bc7eSAndroid Build Coastguard Worker     prof_gctx_tree_t *gctxs) {
1556*1208bc7eSAndroid Build Coastguard Worker 	size_t tabind;
1557*1208bc7eSAndroid Build Coastguard Worker 	union {
1558*1208bc7eSAndroid Build Coastguard Worker 		prof_gctx_t	*p;
1559*1208bc7eSAndroid Build Coastguard Worker 		void		*v;
1560*1208bc7eSAndroid Build Coastguard Worker 	} gctx;
1561*1208bc7eSAndroid Build Coastguard Worker 
1562*1208bc7eSAndroid Build Coastguard Worker 	prof_enter(tsd, tdata);
1563*1208bc7eSAndroid Build Coastguard Worker 
1564*1208bc7eSAndroid Build Coastguard Worker 	/*
1565*1208bc7eSAndroid Build Coastguard Worker 	 * Put gctx's in limbo and clear their counters in preparation for
1566*1208bc7eSAndroid Build Coastguard Worker 	 * summing.
1567*1208bc7eSAndroid Build Coastguard Worker 	 */
1568*1208bc7eSAndroid Build Coastguard Worker 	gctx_tree_new(gctxs);
1569*1208bc7eSAndroid Build Coastguard Worker 	for (tabind = 0; !ckh_iter(&bt2gctx, &tabind, NULL, &gctx.v);) {
1570*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_gctx_prep(tsd_tsdn(tsd), gctx.p, gctxs);
1571*1208bc7eSAndroid Build Coastguard Worker 	}
1572*1208bc7eSAndroid Build Coastguard Worker 
1573*1208bc7eSAndroid Build Coastguard Worker 	/*
1574*1208bc7eSAndroid Build Coastguard Worker 	 * Iterate over tdatas, and for the non-expired ones snapshot their tctx
1575*1208bc7eSAndroid Build Coastguard Worker 	 * stats and merge them into the associated gctx's.
1576*1208bc7eSAndroid Build Coastguard Worker 	 */
1577*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_merge_iter_arg->tsdn = tsd_tsdn(tsd);
1578*1208bc7eSAndroid Build Coastguard Worker 	memset(&prof_tdata_merge_iter_arg->cnt_all, 0, sizeof(prof_cnt_t));
1579*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &tdatas_mtx);
1580*1208bc7eSAndroid Build Coastguard Worker 	tdata_tree_iter(&tdatas, NULL, prof_tdata_merge_iter,
1581*1208bc7eSAndroid Build Coastguard Worker 	    (void *)prof_tdata_merge_iter_arg);
1582*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &tdatas_mtx);
1583*1208bc7eSAndroid Build Coastguard Worker 
1584*1208bc7eSAndroid Build Coastguard Worker 	/* Merge tctx stats into gctx's. */
1585*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_merge_iter_arg->tsdn = tsd_tsdn(tsd);
1586*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_merge_iter_arg->leak_ngctx = 0;
1587*1208bc7eSAndroid Build Coastguard Worker 	gctx_tree_iter(gctxs, NULL, prof_gctx_merge_iter,
1588*1208bc7eSAndroid Build Coastguard Worker 	    (void *)prof_gctx_merge_iter_arg);
1589*1208bc7eSAndroid Build Coastguard Worker 
1590*1208bc7eSAndroid Build Coastguard Worker 	prof_leave(tsd, tdata);
1591*1208bc7eSAndroid Build Coastguard Worker }
1592*1208bc7eSAndroid Build Coastguard Worker 
1593*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump_file(tsd_t * tsd,bool propagate_err,const char * filename,bool leakcheck,prof_tdata_t * tdata,struct prof_tdata_merge_iter_arg_s * prof_tdata_merge_iter_arg,struct prof_gctx_merge_iter_arg_s * prof_gctx_merge_iter_arg,struct prof_gctx_dump_iter_arg_s * prof_gctx_dump_iter_arg,prof_gctx_tree_t * gctxs)1594*1208bc7eSAndroid Build Coastguard Worker prof_dump_file(tsd_t *tsd, bool propagate_err, const char *filename,
1595*1208bc7eSAndroid Build Coastguard Worker     bool leakcheck, prof_tdata_t *tdata,
1596*1208bc7eSAndroid Build Coastguard Worker     struct prof_tdata_merge_iter_arg_s *prof_tdata_merge_iter_arg,
1597*1208bc7eSAndroid Build Coastguard Worker     struct prof_gctx_merge_iter_arg_s *prof_gctx_merge_iter_arg,
1598*1208bc7eSAndroid Build Coastguard Worker     struct prof_gctx_dump_iter_arg_s *prof_gctx_dump_iter_arg,
1599*1208bc7eSAndroid Build Coastguard Worker     prof_gctx_tree_t *gctxs) {
1600*1208bc7eSAndroid Build Coastguard Worker 	/* Create dump file. */
1601*1208bc7eSAndroid Build Coastguard Worker 	if ((prof_dump_fd = prof_dump_open(propagate_err, filename)) == -1) {
1602*1208bc7eSAndroid Build Coastguard Worker 		return true;
1603*1208bc7eSAndroid Build Coastguard Worker 	}
1604*1208bc7eSAndroid Build Coastguard Worker 
1605*1208bc7eSAndroid Build Coastguard Worker 	/* Dump profile header. */
1606*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_header(tsd_tsdn(tsd), propagate_err,
1607*1208bc7eSAndroid Build Coastguard Worker 	    &prof_tdata_merge_iter_arg->cnt_all)) {
1608*1208bc7eSAndroid Build Coastguard Worker 		goto label_write_error;
1609*1208bc7eSAndroid Build Coastguard Worker 	}
1610*1208bc7eSAndroid Build Coastguard Worker 
1611*1208bc7eSAndroid Build Coastguard Worker 	/* Dump per gctx profile stats. */
1612*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_dump_iter_arg->tsdn = tsd_tsdn(tsd);
1613*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_dump_iter_arg->propagate_err = propagate_err;
1614*1208bc7eSAndroid Build Coastguard Worker 	if (gctx_tree_iter(gctxs, NULL, prof_gctx_dump_iter,
1615*1208bc7eSAndroid Build Coastguard Worker 	    (void *)prof_gctx_dump_iter_arg) != NULL) {
1616*1208bc7eSAndroid Build Coastguard Worker 		goto label_write_error;
1617*1208bc7eSAndroid Build Coastguard Worker 	}
1618*1208bc7eSAndroid Build Coastguard Worker 
1619*1208bc7eSAndroid Build Coastguard Worker 	/* Dump /proc/<pid>/maps if possible. */
1620*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_maps(propagate_err)) {
1621*1208bc7eSAndroid Build Coastguard Worker 		goto label_write_error;
1622*1208bc7eSAndroid Build Coastguard Worker 	}
1623*1208bc7eSAndroid Build Coastguard Worker 
1624*1208bc7eSAndroid Build Coastguard Worker 	if (prof_dump_close(propagate_err)) {
1625*1208bc7eSAndroid Build Coastguard Worker 		return true;
1626*1208bc7eSAndroid Build Coastguard Worker 	}
1627*1208bc7eSAndroid Build Coastguard Worker 
1628*1208bc7eSAndroid Build Coastguard Worker 	return false;
1629*1208bc7eSAndroid Build Coastguard Worker label_write_error:
1630*1208bc7eSAndroid Build Coastguard Worker 	prof_dump_close(propagate_err);
1631*1208bc7eSAndroid Build Coastguard Worker 	return true;
1632*1208bc7eSAndroid Build Coastguard Worker }
1633*1208bc7eSAndroid Build Coastguard Worker 
1634*1208bc7eSAndroid Build Coastguard Worker static bool
prof_dump(tsd_t * tsd,bool propagate_err,const char * filename,bool leakcheck)1635*1208bc7eSAndroid Build Coastguard Worker prof_dump(tsd_t *tsd, bool propagate_err, const char *filename,
1636*1208bc7eSAndroid Build Coastguard Worker     bool leakcheck) {
1637*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1638*1208bc7eSAndroid Build Coastguard Worker 	assert(tsd_reentrancy_level_get(tsd) == 0);
1639*1208bc7eSAndroid Build Coastguard Worker 
1640*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t * tdata = prof_tdata_get(tsd, true);
1641*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
1642*1208bc7eSAndroid Build Coastguard Worker 		return true;
1643*1208bc7eSAndroid Build Coastguard Worker 	}
1644*1208bc7eSAndroid Build Coastguard Worker 
1645*1208bc7eSAndroid Build Coastguard Worker 	pre_reentrancy(tsd, NULL);
1646*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &prof_dump_mtx);
1647*1208bc7eSAndroid Build Coastguard Worker 
1648*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_tree_t gctxs;
1649*1208bc7eSAndroid Build Coastguard Worker 	struct prof_tdata_merge_iter_arg_s prof_tdata_merge_iter_arg;
1650*1208bc7eSAndroid Build Coastguard Worker 	struct prof_gctx_merge_iter_arg_s prof_gctx_merge_iter_arg;
1651*1208bc7eSAndroid Build Coastguard Worker 	struct prof_gctx_dump_iter_arg_s prof_gctx_dump_iter_arg;
1652*1208bc7eSAndroid Build Coastguard Worker 	prof_dump_prep(tsd, tdata, &prof_tdata_merge_iter_arg,
1653*1208bc7eSAndroid Build Coastguard Worker 	    &prof_gctx_merge_iter_arg, &gctxs);
1654*1208bc7eSAndroid Build Coastguard Worker 	bool err = prof_dump_file(tsd, propagate_err, filename, leakcheck, tdata,
1655*1208bc7eSAndroid Build Coastguard Worker 	    &prof_tdata_merge_iter_arg, &prof_gctx_merge_iter_arg,
1656*1208bc7eSAndroid Build Coastguard Worker 	    &prof_gctx_dump_iter_arg, &gctxs);
1657*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_finish(tsd, &gctxs);
1658*1208bc7eSAndroid Build Coastguard Worker 
1659*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &prof_dump_mtx);
1660*1208bc7eSAndroid Build Coastguard Worker 	post_reentrancy(tsd);
1661*1208bc7eSAndroid Build Coastguard Worker 
1662*1208bc7eSAndroid Build Coastguard Worker 	if (err) {
1663*1208bc7eSAndroid Build Coastguard Worker 		return true;
1664*1208bc7eSAndroid Build Coastguard Worker 	}
1665*1208bc7eSAndroid Build Coastguard Worker 
1666*1208bc7eSAndroid Build Coastguard Worker 	if (leakcheck) {
1667*1208bc7eSAndroid Build Coastguard Worker 		prof_leakcheck(&prof_tdata_merge_iter_arg.cnt_all,
1668*1208bc7eSAndroid Build Coastguard Worker 		    prof_gctx_merge_iter_arg.leak_ngctx, filename);
1669*1208bc7eSAndroid Build Coastguard Worker 	}
1670*1208bc7eSAndroid Build Coastguard Worker 	return false;
1671*1208bc7eSAndroid Build Coastguard Worker }
1672*1208bc7eSAndroid Build Coastguard Worker 
1673*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_JET
1674*1208bc7eSAndroid Build Coastguard Worker void
prof_cnt_all(uint64_t * curobjs,uint64_t * curbytes,uint64_t * accumobjs,uint64_t * accumbytes)1675*1208bc7eSAndroid Build Coastguard Worker prof_cnt_all(uint64_t *curobjs, uint64_t *curbytes, uint64_t *accumobjs,
1676*1208bc7eSAndroid Build Coastguard Worker     uint64_t *accumbytes) {
1677*1208bc7eSAndroid Build Coastguard Worker 	tsd_t *tsd;
1678*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
1679*1208bc7eSAndroid Build Coastguard Worker 	struct prof_tdata_merge_iter_arg_s prof_tdata_merge_iter_arg;
1680*1208bc7eSAndroid Build Coastguard Worker 	struct prof_gctx_merge_iter_arg_s prof_gctx_merge_iter_arg;
1681*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_tree_t gctxs;
1682*1208bc7eSAndroid Build Coastguard Worker 
1683*1208bc7eSAndroid Build Coastguard Worker 	tsd = tsd_fetch();
1684*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, false);
1685*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
1686*1208bc7eSAndroid Build Coastguard Worker 		if (curobjs != NULL) {
1687*1208bc7eSAndroid Build Coastguard Worker 			*curobjs = 0;
1688*1208bc7eSAndroid Build Coastguard Worker 		}
1689*1208bc7eSAndroid Build Coastguard Worker 		if (curbytes != NULL) {
1690*1208bc7eSAndroid Build Coastguard Worker 			*curbytes = 0;
1691*1208bc7eSAndroid Build Coastguard Worker 		}
1692*1208bc7eSAndroid Build Coastguard Worker 		if (accumobjs != NULL) {
1693*1208bc7eSAndroid Build Coastguard Worker 			*accumobjs = 0;
1694*1208bc7eSAndroid Build Coastguard Worker 		}
1695*1208bc7eSAndroid Build Coastguard Worker 		if (accumbytes != NULL) {
1696*1208bc7eSAndroid Build Coastguard Worker 			*accumbytes = 0;
1697*1208bc7eSAndroid Build Coastguard Worker 		}
1698*1208bc7eSAndroid Build Coastguard Worker 		return;
1699*1208bc7eSAndroid Build Coastguard Worker 	}
1700*1208bc7eSAndroid Build Coastguard Worker 
1701*1208bc7eSAndroid Build Coastguard Worker 	prof_dump_prep(tsd, tdata, &prof_tdata_merge_iter_arg,
1702*1208bc7eSAndroid Build Coastguard Worker 	    &prof_gctx_merge_iter_arg, &gctxs);
1703*1208bc7eSAndroid Build Coastguard Worker 	prof_gctx_finish(tsd, &gctxs);
1704*1208bc7eSAndroid Build Coastguard Worker 
1705*1208bc7eSAndroid Build Coastguard Worker 	if (curobjs != NULL) {
1706*1208bc7eSAndroid Build Coastguard Worker 		*curobjs = prof_tdata_merge_iter_arg.cnt_all.curobjs;
1707*1208bc7eSAndroid Build Coastguard Worker 	}
1708*1208bc7eSAndroid Build Coastguard Worker 	if (curbytes != NULL) {
1709*1208bc7eSAndroid Build Coastguard Worker 		*curbytes = prof_tdata_merge_iter_arg.cnt_all.curbytes;
1710*1208bc7eSAndroid Build Coastguard Worker 	}
1711*1208bc7eSAndroid Build Coastguard Worker 	if (accumobjs != NULL) {
1712*1208bc7eSAndroid Build Coastguard Worker 		*accumobjs = prof_tdata_merge_iter_arg.cnt_all.accumobjs;
1713*1208bc7eSAndroid Build Coastguard Worker 	}
1714*1208bc7eSAndroid Build Coastguard Worker 	if (accumbytes != NULL) {
1715*1208bc7eSAndroid Build Coastguard Worker 		*accumbytes = prof_tdata_merge_iter_arg.cnt_all.accumbytes;
1716*1208bc7eSAndroid Build Coastguard Worker 	}
1717*1208bc7eSAndroid Build Coastguard Worker }
1718*1208bc7eSAndroid Build Coastguard Worker #endif
1719*1208bc7eSAndroid Build Coastguard Worker 
1720*1208bc7eSAndroid Build Coastguard Worker #define DUMP_FILENAME_BUFSIZE	(PATH_MAX + 1)
1721*1208bc7eSAndroid Build Coastguard Worker #define VSEQ_INVALID		UINT64_C(0xffffffffffffffff)
1722*1208bc7eSAndroid Build Coastguard Worker static void
prof_dump_filename(char * filename,char v,uint64_t vseq)1723*1208bc7eSAndroid Build Coastguard Worker prof_dump_filename(char *filename, char v, uint64_t vseq) {
1724*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1725*1208bc7eSAndroid Build Coastguard Worker 
1726*1208bc7eSAndroid Build Coastguard Worker 	if (vseq != VSEQ_INVALID) {
1727*1208bc7eSAndroid Build Coastguard Worker 	        /* "<prefix>.<pid>.<seq>.v<vseq>.heap" */
1728*1208bc7eSAndroid Build Coastguard Worker 		malloc_snprintf(filename, DUMP_FILENAME_BUFSIZE,
1729*1208bc7eSAndroid Build Coastguard Worker 		    "%s.%d.%"FMTu64".%c%"FMTu64".heap",
1730*1208bc7eSAndroid Build Coastguard Worker 		    opt_prof_prefix, prof_getpid(), prof_dump_seq, v, vseq);
1731*1208bc7eSAndroid Build Coastguard Worker 	} else {
1732*1208bc7eSAndroid Build Coastguard Worker 	        /* "<prefix>.<pid>.<seq>.<v>.heap" */
1733*1208bc7eSAndroid Build Coastguard Worker 		malloc_snprintf(filename, DUMP_FILENAME_BUFSIZE,
1734*1208bc7eSAndroid Build Coastguard Worker 		    "%s.%d.%"FMTu64".%c.heap",
1735*1208bc7eSAndroid Build Coastguard Worker 		    opt_prof_prefix, prof_getpid(), prof_dump_seq, v);
1736*1208bc7eSAndroid Build Coastguard Worker 	}
1737*1208bc7eSAndroid Build Coastguard Worker 	prof_dump_seq++;
1738*1208bc7eSAndroid Build Coastguard Worker }
1739*1208bc7eSAndroid Build Coastguard Worker 
1740*1208bc7eSAndroid Build Coastguard Worker static void
prof_fdump(void)1741*1208bc7eSAndroid Build Coastguard Worker prof_fdump(void) {
1742*1208bc7eSAndroid Build Coastguard Worker 	tsd_t *tsd;
1743*1208bc7eSAndroid Build Coastguard Worker 	char filename[DUMP_FILENAME_BUFSIZE];
1744*1208bc7eSAndroid Build Coastguard Worker 
1745*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1746*1208bc7eSAndroid Build Coastguard Worker 	assert(opt_prof_final);
1747*1208bc7eSAndroid Build Coastguard Worker 	assert(opt_prof_prefix[0] != '\0');
1748*1208bc7eSAndroid Build Coastguard Worker 
1749*1208bc7eSAndroid Build Coastguard Worker 	if (!prof_booted) {
1750*1208bc7eSAndroid Build Coastguard Worker 		return;
1751*1208bc7eSAndroid Build Coastguard Worker 	}
1752*1208bc7eSAndroid Build Coastguard Worker 	tsd = tsd_fetch();
1753*1208bc7eSAndroid Build Coastguard Worker 	assert(tsd_reentrancy_level_get(tsd) == 0);
1754*1208bc7eSAndroid Build Coastguard Worker 
1755*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &prof_dump_seq_mtx);
1756*1208bc7eSAndroid Build Coastguard Worker 	prof_dump_filename(filename, 'f', VSEQ_INVALID);
1757*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &prof_dump_seq_mtx);
1758*1208bc7eSAndroid Build Coastguard Worker 	prof_dump(tsd, false, filename, opt_prof_leak);
1759*1208bc7eSAndroid Build Coastguard Worker }
1760*1208bc7eSAndroid Build Coastguard Worker 
1761*1208bc7eSAndroid Build Coastguard Worker bool
prof_accum_init(tsdn_t * tsdn,prof_accum_t * prof_accum)1762*1208bc7eSAndroid Build Coastguard Worker prof_accum_init(tsdn_t *tsdn, prof_accum_t *prof_accum) {
1763*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1764*1208bc7eSAndroid Build Coastguard Worker 
1765*1208bc7eSAndroid Build Coastguard Worker #ifndef JEMALLOC_ATOMIC_U64
1766*1208bc7eSAndroid Build Coastguard Worker 	if (malloc_mutex_init(&prof_accum->mtx, "prof_accum",
1767*1208bc7eSAndroid Build Coastguard Worker 	    WITNESS_RANK_PROF_ACCUM, malloc_mutex_rank_exclusive)) {
1768*1208bc7eSAndroid Build Coastguard Worker 		return true;
1769*1208bc7eSAndroid Build Coastguard Worker 	}
1770*1208bc7eSAndroid Build Coastguard Worker 	prof_accum->accumbytes = 0;
1771*1208bc7eSAndroid Build Coastguard Worker #else
1772*1208bc7eSAndroid Build Coastguard Worker 	atomic_store_u64(&prof_accum->accumbytes, 0, ATOMIC_RELAXED);
1773*1208bc7eSAndroid Build Coastguard Worker #endif
1774*1208bc7eSAndroid Build Coastguard Worker 	return false;
1775*1208bc7eSAndroid Build Coastguard Worker }
1776*1208bc7eSAndroid Build Coastguard Worker 
1777*1208bc7eSAndroid Build Coastguard Worker void
prof_idump(tsdn_t * tsdn)1778*1208bc7eSAndroid Build Coastguard Worker prof_idump(tsdn_t *tsdn) {
1779*1208bc7eSAndroid Build Coastguard Worker 	tsd_t *tsd;
1780*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
1781*1208bc7eSAndroid Build Coastguard Worker 
1782*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1783*1208bc7eSAndroid Build Coastguard Worker 
1784*1208bc7eSAndroid Build Coastguard Worker 	if (!prof_booted || tsdn_null(tsdn) || !prof_active_get_unlocked()) {
1785*1208bc7eSAndroid Build Coastguard Worker 		return;
1786*1208bc7eSAndroid Build Coastguard Worker 	}
1787*1208bc7eSAndroid Build Coastguard Worker 	tsd = tsdn_tsd(tsdn);
1788*1208bc7eSAndroid Build Coastguard Worker 	if (tsd_reentrancy_level_get(tsd) > 0) {
1789*1208bc7eSAndroid Build Coastguard Worker 		return;
1790*1208bc7eSAndroid Build Coastguard Worker 	}
1791*1208bc7eSAndroid Build Coastguard Worker 
1792*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, false);
1793*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
1794*1208bc7eSAndroid Build Coastguard Worker 		return;
1795*1208bc7eSAndroid Build Coastguard Worker 	}
1796*1208bc7eSAndroid Build Coastguard Worker 	if (tdata->enq) {
1797*1208bc7eSAndroid Build Coastguard Worker 		tdata->enq_idump = true;
1798*1208bc7eSAndroid Build Coastguard Worker 		return;
1799*1208bc7eSAndroid Build Coastguard Worker 	}
1800*1208bc7eSAndroid Build Coastguard Worker 
1801*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof_prefix[0] != '\0') {
1802*1208bc7eSAndroid Build Coastguard Worker 		char filename[PATH_MAX + 1];
1803*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsd_tsdn(tsd), &prof_dump_seq_mtx);
1804*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_filename(filename, 'i', prof_dump_iseq);
1805*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_iseq++;
1806*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), &prof_dump_seq_mtx);
1807*1208bc7eSAndroid Build Coastguard Worker 		prof_dump(tsd, false, filename, false);
1808*1208bc7eSAndroid Build Coastguard Worker 	}
1809*1208bc7eSAndroid Build Coastguard Worker }
1810*1208bc7eSAndroid Build Coastguard Worker 
1811*1208bc7eSAndroid Build Coastguard Worker bool
prof_mdump(tsd_t * tsd,const char * filename)1812*1208bc7eSAndroid Build Coastguard Worker prof_mdump(tsd_t *tsd, const char *filename) {
1813*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1814*1208bc7eSAndroid Build Coastguard Worker 	assert(tsd_reentrancy_level_get(tsd) == 0);
1815*1208bc7eSAndroid Build Coastguard Worker 
1816*1208bc7eSAndroid Build Coastguard Worker 	if (!opt_prof || !prof_booted) {
1817*1208bc7eSAndroid Build Coastguard Worker 		return true;
1818*1208bc7eSAndroid Build Coastguard Worker 	}
1819*1208bc7eSAndroid Build Coastguard Worker 	char filename_buf[DUMP_FILENAME_BUFSIZE];
1820*1208bc7eSAndroid Build Coastguard Worker 	if (filename == NULL) {
1821*1208bc7eSAndroid Build Coastguard Worker 		/* No filename specified, so automatically generate one. */
1822*1208bc7eSAndroid Build Coastguard Worker 		if (opt_prof_prefix[0] == '\0') {
1823*1208bc7eSAndroid Build Coastguard Worker 			return true;
1824*1208bc7eSAndroid Build Coastguard Worker 		}
1825*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsd_tsdn(tsd), &prof_dump_seq_mtx);
1826*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_filename(filename_buf, 'm', prof_dump_mseq);
1827*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_mseq++;
1828*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsd_tsdn(tsd), &prof_dump_seq_mtx);
1829*1208bc7eSAndroid Build Coastguard Worker 		filename = filename_buf;
1830*1208bc7eSAndroid Build Coastguard Worker 	}
1831*1208bc7eSAndroid Build Coastguard Worker 	return prof_dump(tsd, true, filename, false);
1832*1208bc7eSAndroid Build Coastguard Worker }
1833*1208bc7eSAndroid Build Coastguard Worker 
1834*1208bc7eSAndroid Build Coastguard Worker void
prof_gdump(tsdn_t * tsdn)1835*1208bc7eSAndroid Build Coastguard Worker prof_gdump(tsdn_t *tsdn) {
1836*1208bc7eSAndroid Build Coastguard Worker 	tsd_t *tsd;
1837*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
1838*1208bc7eSAndroid Build Coastguard Worker 
1839*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1840*1208bc7eSAndroid Build Coastguard Worker 
1841*1208bc7eSAndroid Build Coastguard Worker 	if (!prof_booted || tsdn_null(tsdn) || !prof_active_get_unlocked()) {
1842*1208bc7eSAndroid Build Coastguard Worker 		return;
1843*1208bc7eSAndroid Build Coastguard Worker 	}
1844*1208bc7eSAndroid Build Coastguard Worker 	tsd = tsdn_tsd(tsdn);
1845*1208bc7eSAndroid Build Coastguard Worker 	if (tsd_reentrancy_level_get(tsd) > 0) {
1846*1208bc7eSAndroid Build Coastguard Worker 		return;
1847*1208bc7eSAndroid Build Coastguard Worker 	}
1848*1208bc7eSAndroid Build Coastguard Worker 
1849*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, false);
1850*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
1851*1208bc7eSAndroid Build Coastguard Worker 		return;
1852*1208bc7eSAndroid Build Coastguard Worker 	}
1853*1208bc7eSAndroid Build Coastguard Worker 	if (tdata->enq) {
1854*1208bc7eSAndroid Build Coastguard Worker 		tdata->enq_gdump = true;
1855*1208bc7eSAndroid Build Coastguard Worker 		return;
1856*1208bc7eSAndroid Build Coastguard Worker 	}
1857*1208bc7eSAndroid Build Coastguard Worker 
1858*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof_prefix[0] != '\0') {
1859*1208bc7eSAndroid Build Coastguard Worker 		char filename[DUMP_FILENAME_BUFSIZE];
1860*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_lock(tsdn, &prof_dump_seq_mtx);
1861*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_filename(filename, 'u', prof_dump_useq);
1862*1208bc7eSAndroid Build Coastguard Worker 		prof_dump_useq++;
1863*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_unlock(tsdn, &prof_dump_seq_mtx);
1864*1208bc7eSAndroid Build Coastguard Worker 		prof_dump(tsd, false, filename, false);
1865*1208bc7eSAndroid Build Coastguard Worker 	}
1866*1208bc7eSAndroid Build Coastguard Worker }
1867*1208bc7eSAndroid Build Coastguard Worker 
1868*1208bc7eSAndroid Build Coastguard Worker static void
prof_bt_hash(const void * key,size_t r_hash[2])1869*1208bc7eSAndroid Build Coastguard Worker prof_bt_hash(const void *key, size_t r_hash[2]) {
1870*1208bc7eSAndroid Build Coastguard Worker 	prof_bt_t *bt = (prof_bt_t *)key;
1871*1208bc7eSAndroid Build Coastguard Worker 
1872*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1873*1208bc7eSAndroid Build Coastguard Worker 
1874*1208bc7eSAndroid Build Coastguard Worker 	hash(bt->vec, bt->len * sizeof(void *), 0x94122f33U, r_hash);
1875*1208bc7eSAndroid Build Coastguard Worker }
1876*1208bc7eSAndroid Build Coastguard Worker 
1877*1208bc7eSAndroid Build Coastguard Worker static bool
prof_bt_keycomp(const void * k1,const void * k2)1878*1208bc7eSAndroid Build Coastguard Worker prof_bt_keycomp(const void *k1, const void *k2) {
1879*1208bc7eSAndroid Build Coastguard Worker 	const prof_bt_t *bt1 = (prof_bt_t *)k1;
1880*1208bc7eSAndroid Build Coastguard Worker 	const prof_bt_t *bt2 = (prof_bt_t *)k2;
1881*1208bc7eSAndroid Build Coastguard Worker 
1882*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1883*1208bc7eSAndroid Build Coastguard Worker 
1884*1208bc7eSAndroid Build Coastguard Worker 	if (bt1->len != bt2->len) {
1885*1208bc7eSAndroid Build Coastguard Worker 		return false;
1886*1208bc7eSAndroid Build Coastguard Worker 	}
1887*1208bc7eSAndroid Build Coastguard Worker 	return (memcmp(bt1->vec, bt2->vec, bt1->len * sizeof(void *)) == 0);
1888*1208bc7eSAndroid Build Coastguard Worker }
1889*1208bc7eSAndroid Build Coastguard Worker 
1890*1208bc7eSAndroid Build Coastguard Worker static uint64_t
prof_thr_uid_alloc(tsdn_t * tsdn)1891*1208bc7eSAndroid Build Coastguard Worker prof_thr_uid_alloc(tsdn_t *tsdn) {
1892*1208bc7eSAndroid Build Coastguard Worker 	uint64_t thr_uid;
1893*1208bc7eSAndroid Build Coastguard Worker 
1894*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &next_thr_uid_mtx);
1895*1208bc7eSAndroid Build Coastguard Worker 	thr_uid = next_thr_uid;
1896*1208bc7eSAndroid Build Coastguard Worker 	next_thr_uid++;
1897*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &next_thr_uid_mtx);
1898*1208bc7eSAndroid Build Coastguard Worker 
1899*1208bc7eSAndroid Build Coastguard Worker 	return thr_uid;
1900*1208bc7eSAndroid Build Coastguard Worker }
1901*1208bc7eSAndroid Build Coastguard Worker 
1902*1208bc7eSAndroid Build Coastguard Worker static prof_tdata_t *
prof_tdata_init_impl(tsd_t * tsd,uint64_t thr_uid,uint64_t thr_discrim,char * thread_name,bool active)1903*1208bc7eSAndroid Build Coastguard Worker prof_tdata_init_impl(tsd_t *tsd, uint64_t thr_uid, uint64_t thr_discrim,
1904*1208bc7eSAndroid Build Coastguard Worker     char *thread_name, bool active) {
1905*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
1906*1208bc7eSAndroid Build Coastguard Worker 
1907*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
1908*1208bc7eSAndroid Build Coastguard Worker 
1909*1208bc7eSAndroid Build Coastguard Worker 	/* Initialize an empty cache for this thread. */
1910*1208bc7eSAndroid Build Coastguard Worker 	tdata = (prof_tdata_t *)iallocztm(tsd_tsdn(tsd), sizeof(prof_tdata_t),
1911*1208bc7eSAndroid Build Coastguard Worker 	    sz_size2index(sizeof(prof_tdata_t)), false, NULL, true,
1912*1208bc7eSAndroid Build Coastguard Worker 	    arena_get(TSDN_NULL, 0, true), true);
1913*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
1914*1208bc7eSAndroid Build Coastguard Worker 		return NULL;
1915*1208bc7eSAndroid Build Coastguard Worker 	}
1916*1208bc7eSAndroid Build Coastguard Worker 
1917*1208bc7eSAndroid Build Coastguard Worker 	tdata->lock = prof_tdata_mutex_choose(thr_uid);
1918*1208bc7eSAndroid Build Coastguard Worker 	tdata->thr_uid = thr_uid;
1919*1208bc7eSAndroid Build Coastguard Worker 	tdata->thr_discrim = thr_discrim;
1920*1208bc7eSAndroid Build Coastguard Worker 	tdata->thread_name = thread_name;
1921*1208bc7eSAndroid Build Coastguard Worker 	tdata->attached = true;
1922*1208bc7eSAndroid Build Coastguard Worker 	tdata->expired = false;
1923*1208bc7eSAndroid Build Coastguard Worker 	tdata->tctx_uid_next = 0;
1924*1208bc7eSAndroid Build Coastguard Worker 
1925*1208bc7eSAndroid Build Coastguard Worker 	if (ckh_new(tsd, &tdata->bt2tctx, PROF_CKH_MINITEMS, prof_bt_hash,
1926*1208bc7eSAndroid Build Coastguard Worker 	    prof_bt_keycomp)) {
1927*1208bc7eSAndroid Build Coastguard Worker 		idalloctm(tsd_tsdn(tsd), tdata, NULL, NULL, true, true);
1928*1208bc7eSAndroid Build Coastguard Worker 		return NULL;
1929*1208bc7eSAndroid Build Coastguard Worker 	}
1930*1208bc7eSAndroid Build Coastguard Worker 
1931*1208bc7eSAndroid Build Coastguard Worker 	tdata->prng_state = (uint64_t)(uintptr_t)tdata;
1932*1208bc7eSAndroid Build Coastguard Worker 	prof_sample_threshold_update(tdata);
1933*1208bc7eSAndroid Build Coastguard Worker 
1934*1208bc7eSAndroid Build Coastguard Worker 	tdata->enq = false;
1935*1208bc7eSAndroid Build Coastguard Worker 	tdata->enq_idump = false;
1936*1208bc7eSAndroid Build Coastguard Worker 	tdata->enq_gdump = false;
1937*1208bc7eSAndroid Build Coastguard Worker 
1938*1208bc7eSAndroid Build Coastguard Worker 	tdata->dumping = false;
1939*1208bc7eSAndroid Build Coastguard Worker 	tdata->active = active;
1940*1208bc7eSAndroid Build Coastguard Worker 
1941*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &tdatas_mtx);
1942*1208bc7eSAndroid Build Coastguard Worker 	tdata_tree_insert(&tdatas, tdata);
1943*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &tdatas_mtx);
1944*1208bc7eSAndroid Build Coastguard Worker 
1945*1208bc7eSAndroid Build Coastguard Worker 	return tdata;
1946*1208bc7eSAndroid Build Coastguard Worker }
1947*1208bc7eSAndroid Build Coastguard Worker 
1948*1208bc7eSAndroid Build Coastguard Worker prof_tdata_t *
prof_tdata_init(tsd_t * tsd)1949*1208bc7eSAndroid Build Coastguard Worker prof_tdata_init(tsd_t *tsd) {
1950*1208bc7eSAndroid Build Coastguard Worker 	return prof_tdata_init_impl(tsd, prof_thr_uid_alloc(tsd_tsdn(tsd)), 0,
1951*1208bc7eSAndroid Build Coastguard Worker 	    NULL, prof_thread_active_init_get(tsd_tsdn(tsd)));
1952*1208bc7eSAndroid Build Coastguard Worker }
1953*1208bc7eSAndroid Build Coastguard Worker 
1954*1208bc7eSAndroid Build Coastguard Worker static bool
prof_tdata_should_destroy_unlocked(prof_tdata_t * tdata,bool even_if_attached)1955*1208bc7eSAndroid Build Coastguard Worker prof_tdata_should_destroy_unlocked(prof_tdata_t *tdata, bool even_if_attached) {
1956*1208bc7eSAndroid Build Coastguard Worker 	if (tdata->attached && !even_if_attached) {
1957*1208bc7eSAndroid Build Coastguard Worker 		return false;
1958*1208bc7eSAndroid Build Coastguard Worker 	}
1959*1208bc7eSAndroid Build Coastguard Worker 	if (ckh_count(&tdata->bt2tctx) != 0) {
1960*1208bc7eSAndroid Build Coastguard Worker 		return false;
1961*1208bc7eSAndroid Build Coastguard Worker 	}
1962*1208bc7eSAndroid Build Coastguard Worker 	return true;
1963*1208bc7eSAndroid Build Coastguard Worker }
1964*1208bc7eSAndroid Build Coastguard Worker 
1965*1208bc7eSAndroid Build Coastguard Worker static bool
prof_tdata_should_destroy(tsdn_t * tsdn,prof_tdata_t * tdata,bool even_if_attached)1966*1208bc7eSAndroid Build Coastguard Worker prof_tdata_should_destroy(tsdn_t *tsdn, prof_tdata_t *tdata,
1967*1208bc7eSAndroid Build Coastguard Worker     bool even_if_attached) {
1968*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsdn, tdata->lock);
1969*1208bc7eSAndroid Build Coastguard Worker 
1970*1208bc7eSAndroid Build Coastguard Worker 	return prof_tdata_should_destroy_unlocked(tdata, even_if_attached);
1971*1208bc7eSAndroid Build Coastguard Worker }
1972*1208bc7eSAndroid Build Coastguard Worker 
1973*1208bc7eSAndroid Build Coastguard Worker static void
prof_tdata_destroy_locked(tsd_t * tsd,prof_tdata_t * tdata,bool even_if_attached)1974*1208bc7eSAndroid Build Coastguard Worker prof_tdata_destroy_locked(tsd_t *tsd, prof_tdata_t *tdata,
1975*1208bc7eSAndroid Build Coastguard Worker     bool even_if_attached) {
1976*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_assert_owner(tsd_tsdn(tsd), &tdatas_mtx);
1977*1208bc7eSAndroid Build Coastguard Worker 
1978*1208bc7eSAndroid Build Coastguard Worker 	tdata_tree_remove(&tdatas, tdata);
1979*1208bc7eSAndroid Build Coastguard Worker 
1980*1208bc7eSAndroid Build Coastguard Worker 	assert(prof_tdata_should_destroy_unlocked(tdata, even_if_attached));
1981*1208bc7eSAndroid Build Coastguard Worker 
1982*1208bc7eSAndroid Build Coastguard Worker 	if (tdata->thread_name != NULL) {
1983*1208bc7eSAndroid Build Coastguard Worker 		idalloctm(tsd_tsdn(tsd), tdata->thread_name, NULL, NULL, true,
1984*1208bc7eSAndroid Build Coastguard Worker 		    true);
1985*1208bc7eSAndroid Build Coastguard Worker 	}
1986*1208bc7eSAndroid Build Coastguard Worker 	ckh_delete(tsd, &tdata->bt2tctx);
1987*1208bc7eSAndroid Build Coastguard Worker 	idalloctm(tsd_tsdn(tsd), tdata, NULL, NULL, true, true);
1988*1208bc7eSAndroid Build Coastguard Worker }
1989*1208bc7eSAndroid Build Coastguard Worker 
1990*1208bc7eSAndroid Build Coastguard Worker static void
prof_tdata_destroy(tsd_t * tsd,prof_tdata_t * tdata,bool even_if_attached)1991*1208bc7eSAndroid Build Coastguard Worker prof_tdata_destroy(tsd_t *tsd, prof_tdata_t *tdata, bool even_if_attached) {
1992*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &tdatas_mtx);
1993*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_destroy_locked(tsd, tdata, even_if_attached);
1994*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &tdatas_mtx);
1995*1208bc7eSAndroid Build Coastguard Worker }
1996*1208bc7eSAndroid Build Coastguard Worker 
1997*1208bc7eSAndroid Build Coastguard Worker static void
prof_tdata_detach(tsd_t * tsd,prof_tdata_t * tdata)1998*1208bc7eSAndroid Build Coastguard Worker prof_tdata_detach(tsd_t *tsd, prof_tdata_t *tdata) {
1999*1208bc7eSAndroid Build Coastguard Worker 	bool destroy_tdata;
2000*1208bc7eSAndroid Build Coastguard Worker 
2001*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), tdata->lock);
2002*1208bc7eSAndroid Build Coastguard Worker 	if (tdata->attached) {
2003*1208bc7eSAndroid Build Coastguard Worker 		destroy_tdata = prof_tdata_should_destroy(tsd_tsdn(tsd), tdata,
2004*1208bc7eSAndroid Build Coastguard Worker 		    true);
2005*1208bc7eSAndroid Build Coastguard Worker 		/*
2006*1208bc7eSAndroid Build Coastguard Worker 		 * Only detach if !destroy_tdata, because detaching would allow
2007*1208bc7eSAndroid Build Coastguard Worker 		 * another thread to win the race to destroy tdata.
2008*1208bc7eSAndroid Build Coastguard Worker 		 */
2009*1208bc7eSAndroid Build Coastguard Worker 		if (!destroy_tdata) {
2010*1208bc7eSAndroid Build Coastguard Worker 			tdata->attached = false;
2011*1208bc7eSAndroid Build Coastguard Worker 		}
2012*1208bc7eSAndroid Build Coastguard Worker 		tsd_prof_tdata_set(tsd, NULL);
2013*1208bc7eSAndroid Build Coastguard Worker 	} else {
2014*1208bc7eSAndroid Build Coastguard Worker 		destroy_tdata = false;
2015*1208bc7eSAndroid Build Coastguard Worker 	}
2016*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), tdata->lock);
2017*1208bc7eSAndroid Build Coastguard Worker 	if (destroy_tdata) {
2018*1208bc7eSAndroid Build Coastguard Worker 		prof_tdata_destroy(tsd, tdata, true);
2019*1208bc7eSAndroid Build Coastguard Worker 	}
2020*1208bc7eSAndroid Build Coastguard Worker }
2021*1208bc7eSAndroid Build Coastguard Worker 
2022*1208bc7eSAndroid Build Coastguard Worker prof_tdata_t *
prof_tdata_reinit(tsd_t * tsd,prof_tdata_t * tdata)2023*1208bc7eSAndroid Build Coastguard Worker prof_tdata_reinit(tsd_t *tsd, prof_tdata_t *tdata) {
2024*1208bc7eSAndroid Build Coastguard Worker 	uint64_t thr_uid = tdata->thr_uid;
2025*1208bc7eSAndroid Build Coastguard Worker 	uint64_t thr_discrim = tdata->thr_discrim + 1;
2026*1208bc7eSAndroid Build Coastguard Worker 	char *thread_name = (tdata->thread_name != NULL) ?
2027*1208bc7eSAndroid Build Coastguard Worker 	    prof_thread_name_alloc(tsd_tsdn(tsd), tdata->thread_name) : NULL;
2028*1208bc7eSAndroid Build Coastguard Worker 	bool active = tdata->active;
2029*1208bc7eSAndroid Build Coastguard Worker 
2030*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_detach(tsd, tdata);
2031*1208bc7eSAndroid Build Coastguard Worker 	return prof_tdata_init_impl(tsd, thr_uid, thr_discrim, thread_name,
2032*1208bc7eSAndroid Build Coastguard Worker 	    active);
2033*1208bc7eSAndroid Build Coastguard Worker }
2034*1208bc7eSAndroid Build Coastguard Worker 
2035*1208bc7eSAndroid Build Coastguard Worker static bool
prof_tdata_expire(tsdn_t * tsdn,prof_tdata_t * tdata)2036*1208bc7eSAndroid Build Coastguard Worker prof_tdata_expire(tsdn_t *tsdn, prof_tdata_t *tdata) {
2037*1208bc7eSAndroid Build Coastguard Worker 	bool destroy_tdata;
2038*1208bc7eSAndroid Build Coastguard Worker 
2039*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, tdata->lock);
2040*1208bc7eSAndroid Build Coastguard Worker 	if (!tdata->expired) {
2041*1208bc7eSAndroid Build Coastguard Worker 		tdata->expired = true;
2042*1208bc7eSAndroid Build Coastguard Worker 		destroy_tdata = tdata->attached ? false :
2043*1208bc7eSAndroid Build Coastguard Worker 		    prof_tdata_should_destroy(tsdn, tdata, false);
2044*1208bc7eSAndroid Build Coastguard Worker 	} else {
2045*1208bc7eSAndroid Build Coastguard Worker 		destroy_tdata = false;
2046*1208bc7eSAndroid Build Coastguard Worker 	}
2047*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, tdata->lock);
2048*1208bc7eSAndroid Build Coastguard Worker 
2049*1208bc7eSAndroid Build Coastguard Worker 	return destroy_tdata;
2050*1208bc7eSAndroid Build Coastguard Worker }
2051*1208bc7eSAndroid Build Coastguard Worker 
2052*1208bc7eSAndroid Build Coastguard Worker static prof_tdata_t *
prof_tdata_reset_iter(prof_tdata_tree_t * tdatas,prof_tdata_t * tdata,void * arg)2053*1208bc7eSAndroid Build Coastguard Worker prof_tdata_reset_iter(prof_tdata_tree_t *tdatas, prof_tdata_t *tdata,
2054*1208bc7eSAndroid Build Coastguard Worker     void *arg) {
2055*1208bc7eSAndroid Build Coastguard Worker 	tsdn_t *tsdn = (tsdn_t *)arg;
2056*1208bc7eSAndroid Build Coastguard Worker 
2057*1208bc7eSAndroid Build Coastguard Worker 	return (prof_tdata_expire(tsdn, tdata) ? tdata : NULL);
2058*1208bc7eSAndroid Build Coastguard Worker }
2059*1208bc7eSAndroid Build Coastguard Worker 
2060*1208bc7eSAndroid Build Coastguard Worker void
prof_reset(tsd_t * tsd,size_t lg_sample)2061*1208bc7eSAndroid Build Coastguard Worker prof_reset(tsd_t *tsd, size_t lg_sample) {
2062*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *next;
2063*1208bc7eSAndroid Build Coastguard Worker 
2064*1208bc7eSAndroid Build Coastguard Worker 	assert(lg_sample < (sizeof(uint64_t) << 3));
2065*1208bc7eSAndroid Build Coastguard Worker 
2066*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &prof_dump_mtx);
2067*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsd_tsdn(tsd), &tdatas_mtx);
2068*1208bc7eSAndroid Build Coastguard Worker 
2069*1208bc7eSAndroid Build Coastguard Worker 	lg_prof_sample = lg_sample;
2070*1208bc7eSAndroid Build Coastguard Worker 
2071*1208bc7eSAndroid Build Coastguard Worker 	next = NULL;
2072*1208bc7eSAndroid Build Coastguard Worker 	do {
2073*1208bc7eSAndroid Build Coastguard Worker 		prof_tdata_t *to_destroy = tdata_tree_iter(&tdatas, next,
2074*1208bc7eSAndroid Build Coastguard Worker 		    prof_tdata_reset_iter, (void *)tsd);
2075*1208bc7eSAndroid Build Coastguard Worker 		if (to_destroy != NULL) {
2076*1208bc7eSAndroid Build Coastguard Worker 			next = tdata_tree_next(&tdatas, to_destroy);
2077*1208bc7eSAndroid Build Coastguard Worker 			prof_tdata_destroy_locked(tsd, to_destroy, false);
2078*1208bc7eSAndroid Build Coastguard Worker 		} else {
2079*1208bc7eSAndroid Build Coastguard Worker 			next = NULL;
2080*1208bc7eSAndroid Build Coastguard Worker 		}
2081*1208bc7eSAndroid Build Coastguard Worker 	} while (next != NULL);
2082*1208bc7eSAndroid Build Coastguard Worker 
2083*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &tdatas_mtx);
2084*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsd_tsdn(tsd), &prof_dump_mtx);
2085*1208bc7eSAndroid Build Coastguard Worker }
2086*1208bc7eSAndroid Build Coastguard Worker 
2087*1208bc7eSAndroid Build Coastguard Worker void
prof_tdata_cleanup(tsd_t * tsd)2088*1208bc7eSAndroid Build Coastguard Worker prof_tdata_cleanup(tsd_t *tsd) {
2089*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
2090*1208bc7eSAndroid Build Coastguard Worker 
2091*1208bc7eSAndroid Build Coastguard Worker 	if (!config_prof) {
2092*1208bc7eSAndroid Build Coastguard Worker 		return;
2093*1208bc7eSAndroid Build Coastguard Worker 	}
2094*1208bc7eSAndroid Build Coastguard Worker 
2095*1208bc7eSAndroid Build Coastguard Worker 	tdata = tsd_prof_tdata_get(tsd);
2096*1208bc7eSAndroid Build Coastguard Worker 	if (tdata != NULL) {
2097*1208bc7eSAndroid Build Coastguard Worker 		prof_tdata_detach(tsd, tdata);
2098*1208bc7eSAndroid Build Coastguard Worker 	}
2099*1208bc7eSAndroid Build Coastguard Worker }
2100*1208bc7eSAndroid Build Coastguard Worker 
2101*1208bc7eSAndroid Build Coastguard Worker bool
prof_active_get(tsdn_t * tsdn)2102*1208bc7eSAndroid Build Coastguard Worker prof_active_get(tsdn_t *tsdn) {
2103*1208bc7eSAndroid Build Coastguard Worker 	bool prof_active_current;
2104*1208bc7eSAndroid Build Coastguard Worker 
2105*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &prof_active_mtx);
2106*1208bc7eSAndroid Build Coastguard Worker 	prof_active_current = prof_active;
2107*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &prof_active_mtx);
2108*1208bc7eSAndroid Build Coastguard Worker 	return prof_active_current;
2109*1208bc7eSAndroid Build Coastguard Worker }
2110*1208bc7eSAndroid Build Coastguard Worker 
2111*1208bc7eSAndroid Build Coastguard Worker bool
prof_active_set(tsdn_t * tsdn,bool active)2112*1208bc7eSAndroid Build Coastguard Worker prof_active_set(tsdn_t *tsdn, bool active) {
2113*1208bc7eSAndroid Build Coastguard Worker 	bool prof_active_old;
2114*1208bc7eSAndroid Build Coastguard Worker 
2115*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &prof_active_mtx);
2116*1208bc7eSAndroid Build Coastguard Worker 	prof_active_old = prof_active;
2117*1208bc7eSAndroid Build Coastguard Worker 	prof_active = active;
2118*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &prof_active_mtx);
2119*1208bc7eSAndroid Build Coastguard Worker 	return prof_active_old;
2120*1208bc7eSAndroid Build Coastguard Worker }
2121*1208bc7eSAndroid Build Coastguard Worker 
2122*1208bc7eSAndroid Build Coastguard Worker const char *
prof_thread_name_get(tsd_t * tsd)2123*1208bc7eSAndroid Build Coastguard Worker prof_thread_name_get(tsd_t *tsd) {
2124*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
2125*1208bc7eSAndroid Build Coastguard Worker 
2126*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, true);
2127*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
2128*1208bc7eSAndroid Build Coastguard Worker 		return "";
2129*1208bc7eSAndroid Build Coastguard Worker 	}
2130*1208bc7eSAndroid Build Coastguard Worker 	return (tdata->thread_name != NULL ? tdata->thread_name : "");
2131*1208bc7eSAndroid Build Coastguard Worker }
2132*1208bc7eSAndroid Build Coastguard Worker 
2133*1208bc7eSAndroid Build Coastguard Worker static char *
prof_thread_name_alloc(tsdn_t * tsdn,const char * thread_name)2134*1208bc7eSAndroid Build Coastguard Worker prof_thread_name_alloc(tsdn_t *tsdn, const char *thread_name) {
2135*1208bc7eSAndroid Build Coastguard Worker 	char *ret;
2136*1208bc7eSAndroid Build Coastguard Worker 	size_t size;
2137*1208bc7eSAndroid Build Coastguard Worker 
2138*1208bc7eSAndroid Build Coastguard Worker 	if (thread_name == NULL) {
2139*1208bc7eSAndroid Build Coastguard Worker 		return NULL;
2140*1208bc7eSAndroid Build Coastguard Worker 	}
2141*1208bc7eSAndroid Build Coastguard Worker 
2142*1208bc7eSAndroid Build Coastguard Worker 	size = strlen(thread_name) + 1;
2143*1208bc7eSAndroid Build Coastguard Worker 	if (size == 1) {
2144*1208bc7eSAndroid Build Coastguard Worker 		return "";
2145*1208bc7eSAndroid Build Coastguard Worker 	}
2146*1208bc7eSAndroid Build Coastguard Worker 
2147*1208bc7eSAndroid Build Coastguard Worker 	ret = iallocztm(tsdn, size, sz_size2index(size), false, NULL, true,
2148*1208bc7eSAndroid Build Coastguard Worker 	    arena_get(TSDN_NULL, 0, true), true);
2149*1208bc7eSAndroid Build Coastguard Worker 	if (ret == NULL) {
2150*1208bc7eSAndroid Build Coastguard Worker 		return NULL;
2151*1208bc7eSAndroid Build Coastguard Worker 	}
2152*1208bc7eSAndroid Build Coastguard Worker 	memcpy(ret, thread_name, size);
2153*1208bc7eSAndroid Build Coastguard Worker 	return ret;
2154*1208bc7eSAndroid Build Coastguard Worker }
2155*1208bc7eSAndroid Build Coastguard Worker 
2156*1208bc7eSAndroid Build Coastguard Worker int
prof_thread_name_set(tsd_t * tsd,const char * thread_name)2157*1208bc7eSAndroid Build Coastguard Worker prof_thread_name_set(tsd_t *tsd, const char *thread_name) {
2158*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
2159*1208bc7eSAndroid Build Coastguard Worker 	unsigned i;
2160*1208bc7eSAndroid Build Coastguard Worker 	char *s;
2161*1208bc7eSAndroid Build Coastguard Worker 
2162*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, true);
2163*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
2164*1208bc7eSAndroid Build Coastguard Worker 		return EAGAIN;
2165*1208bc7eSAndroid Build Coastguard Worker 	}
2166*1208bc7eSAndroid Build Coastguard Worker 
2167*1208bc7eSAndroid Build Coastguard Worker 	/* Validate input. */
2168*1208bc7eSAndroid Build Coastguard Worker 	if (thread_name == NULL) {
2169*1208bc7eSAndroid Build Coastguard Worker 		return EFAULT;
2170*1208bc7eSAndroid Build Coastguard Worker 	}
2171*1208bc7eSAndroid Build Coastguard Worker 	for (i = 0; thread_name[i] != '\0'; i++) {
2172*1208bc7eSAndroid Build Coastguard Worker 		char c = thread_name[i];
2173*1208bc7eSAndroid Build Coastguard Worker 		if (!isgraph(c) && !isblank(c)) {
2174*1208bc7eSAndroid Build Coastguard Worker 			return EFAULT;
2175*1208bc7eSAndroid Build Coastguard Worker 		}
2176*1208bc7eSAndroid Build Coastguard Worker 	}
2177*1208bc7eSAndroid Build Coastguard Worker 
2178*1208bc7eSAndroid Build Coastguard Worker 	s = prof_thread_name_alloc(tsd_tsdn(tsd), thread_name);
2179*1208bc7eSAndroid Build Coastguard Worker 	if (s == NULL) {
2180*1208bc7eSAndroid Build Coastguard Worker 		return EAGAIN;
2181*1208bc7eSAndroid Build Coastguard Worker 	}
2182*1208bc7eSAndroid Build Coastguard Worker 
2183*1208bc7eSAndroid Build Coastguard Worker 	if (tdata->thread_name != NULL) {
2184*1208bc7eSAndroid Build Coastguard Worker 		idalloctm(tsd_tsdn(tsd), tdata->thread_name, NULL, NULL, true,
2185*1208bc7eSAndroid Build Coastguard Worker 		    true);
2186*1208bc7eSAndroid Build Coastguard Worker 		tdata->thread_name = NULL;
2187*1208bc7eSAndroid Build Coastguard Worker 	}
2188*1208bc7eSAndroid Build Coastguard Worker 	if (strlen(s) > 0) {
2189*1208bc7eSAndroid Build Coastguard Worker 		tdata->thread_name = s;
2190*1208bc7eSAndroid Build Coastguard Worker 	}
2191*1208bc7eSAndroid Build Coastguard Worker 	return 0;
2192*1208bc7eSAndroid Build Coastguard Worker }
2193*1208bc7eSAndroid Build Coastguard Worker 
2194*1208bc7eSAndroid Build Coastguard Worker bool
prof_thread_active_get(tsd_t * tsd)2195*1208bc7eSAndroid Build Coastguard Worker prof_thread_active_get(tsd_t *tsd) {
2196*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
2197*1208bc7eSAndroid Build Coastguard Worker 
2198*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, true);
2199*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
2200*1208bc7eSAndroid Build Coastguard Worker 		return false;
2201*1208bc7eSAndroid Build Coastguard Worker 	}
2202*1208bc7eSAndroid Build Coastguard Worker 	return tdata->active;
2203*1208bc7eSAndroid Build Coastguard Worker }
2204*1208bc7eSAndroid Build Coastguard Worker 
2205*1208bc7eSAndroid Build Coastguard Worker bool
prof_thread_active_set(tsd_t * tsd,bool active)2206*1208bc7eSAndroid Build Coastguard Worker prof_thread_active_set(tsd_t *tsd, bool active) {
2207*1208bc7eSAndroid Build Coastguard Worker 	prof_tdata_t *tdata;
2208*1208bc7eSAndroid Build Coastguard Worker 
2209*1208bc7eSAndroid Build Coastguard Worker 	tdata = prof_tdata_get(tsd, true);
2210*1208bc7eSAndroid Build Coastguard Worker 	if (tdata == NULL) {
2211*1208bc7eSAndroid Build Coastguard Worker 		return true;
2212*1208bc7eSAndroid Build Coastguard Worker 	}
2213*1208bc7eSAndroid Build Coastguard Worker 	tdata->active = active;
2214*1208bc7eSAndroid Build Coastguard Worker 	return false;
2215*1208bc7eSAndroid Build Coastguard Worker }
2216*1208bc7eSAndroid Build Coastguard Worker 
2217*1208bc7eSAndroid Build Coastguard Worker bool
prof_thread_active_init_get(tsdn_t * tsdn)2218*1208bc7eSAndroid Build Coastguard Worker prof_thread_active_init_get(tsdn_t *tsdn) {
2219*1208bc7eSAndroid Build Coastguard Worker 	bool active_init;
2220*1208bc7eSAndroid Build Coastguard Worker 
2221*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &prof_thread_active_init_mtx);
2222*1208bc7eSAndroid Build Coastguard Worker 	active_init = prof_thread_active_init;
2223*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &prof_thread_active_init_mtx);
2224*1208bc7eSAndroid Build Coastguard Worker 	return active_init;
2225*1208bc7eSAndroid Build Coastguard Worker }
2226*1208bc7eSAndroid Build Coastguard Worker 
2227*1208bc7eSAndroid Build Coastguard Worker bool
prof_thread_active_init_set(tsdn_t * tsdn,bool active_init)2228*1208bc7eSAndroid Build Coastguard Worker prof_thread_active_init_set(tsdn_t *tsdn, bool active_init) {
2229*1208bc7eSAndroid Build Coastguard Worker 	bool active_init_old;
2230*1208bc7eSAndroid Build Coastguard Worker 
2231*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &prof_thread_active_init_mtx);
2232*1208bc7eSAndroid Build Coastguard Worker 	active_init_old = prof_thread_active_init;
2233*1208bc7eSAndroid Build Coastguard Worker 	prof_thread_active_init = active_init;
2234*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &prof_thread_active_init_mtx);
2235*1208bc7eSAndroid Build Coastguard Worker 	return active_init_old;
2236*1208bc7eSAndroid Build Coastguard Worker }
2237*1208bc7eSAndroid Build Coastguard Worker 
2238*1208bc7eSAndroid Build Coastguard Worker bool
prof_gdump_get(tsdn_t * tsdn)2239*1208bc7eSAndroid Build Coastguard Worker prof_gdump_get(tsdn_t *tsdn) {
2240*1208bc7eSAndroid Build Coastguard Worker 	bool prof_gdump_current;
2241*1208bc7eSAndroid Build Coastguard Worker 
2242*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &prof_gdump_mtx);
2243*1208bc7eSAndroid Build Coastguard Worker 	prof_gdump_current = prof_gdump_val;
2244*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &prof_gdump_mtx);
2245*1208bc7eSAndroid Build Coastguard Worker 	return prof_gdump_current;
2246*1208bc7eSAndroid Build Coastguard Worker }
2247*1208bc7eSAndroid Build Coastguard Worker 
2248*1208bc7eSAndroid Build Coastguard Worker bool
prof_gdump_set(tsdn_t * tsdn,bool gdump)2249*1208bc7eSAndroid Build Coastguard Worker prof_gdump_set(tsdn_t *tsdn, bool gdump) {
2250*1208bc7eSAndroid Build Coastguard Worker 	bool prof_gdump_old;
2251*1208bc7eSAndroid Build Coastguard Worker 
2252*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_lock(tsdn, &prof_gdump_mtx);
2253*1208bc7eSAndroid Build Coastguard Worker 	prof_gdump_old = prof_gdump_val;
2254*1208bc7eSAndroid Build Coastguard Worker 	prof_gdump_val = gdump;
2255*1208bc7eSAndroid Build Coastguard Worker 	malloc_mutex_unlock(tsdn, &prof_gdump_mtx);
2256*1208bc7eSAndroid Build Coastguard Worker 	return prof_gdump_old;
2257*1208bc7eSAndroid Build Coastguard Worker }
2258*1208bc7eSAndroid Build Coastguard Worker 
2259*1208bc7eSAndroid Build Coastguard Worker void
prof_boot0(void)2260*1208bc7eSAndroid Build Coastguard Worker prof_boot0(void) {
2261*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
2262*1208bc7eSAndroid Build Coastguard Worker 
2263*1208bc7eSAndroid Build Coastguard Worker 	memcpy(opt_prof_prefix, PROF_PREFIX_DEFAULT,
2264*1208bc7eSAndroid Build Coastguard Worker 	    sizeof(PROF_PREFIX_DEFAULT));
2265*1208bc7eSAndroid Build Coastguard Worker }
2266*1208bc7eSAndroid Build Coastguard Worker 
2267*1208bc7eSAndroid Build Coastguard Worker void
prof_boot1(void)2268*1208bc7eSAndroid Build Coastguard Worker prof_boot1(void) {
2269*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
2270*1208bc7eSAndroid Build Coastguard Worker 
2271*1208bc7eSAndroid Build Coastguard Worker 	/*
2272*1208bc7eSAndroid Build Coastguard Worker 	 * opt_prof must be in its final state before any arenas are
2273*1208bc7eSAndroid Build Coastguard Worker 	 * initialized, so this function must be executed early.
2274*1208bc7eSAndroid Build Coastguard Worker 	 */
2275*1208bc7eSAndroid Build Coastguard Worker 
2276*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof_leak && !opt_prof) {
2277*1208bc7eSAndroid Build Coastguard Worker 		/*
2278*1208bc7eSAndroid Build Coastguard Worker 		 * Enable opt_prof, but in such a way that profiles are never
2279*1208bc7eSAndroid Build Coastguard Worker 		 * automatically dumped.
2280*1208bc7eSAndroid Build Coastguard Worker 		 */
2281*1208bc7eSAndroid Build Coastguard Worker 		opt_prof = true;
2282*1208bc7eSAndroid Build Coastguard Worker 		opt_prof_gdump = false;
2283*1208bc7eSAndroid Build Coastguard Worker 	} else if (opt_prof) {
2284*1208bc7eSAndroid Build Coastguard Worker 		if (opt_lg_prof_interval >= 0) {
2285*1208bc7eSAndroid Build Coastguard Worker 			prof_interval = (((uint64_t)1U) <<
2286*1208bc7eSAndroid Build Coastguard Worker 			    opt_lg_prof_interval);
2287*1208bc7eSAndroid Build Coastguard Worker 		}
2288*1208bc7eSAndroid Build Coastguard Worker 	}
2289*1208bc7eSAndroid Build Coastguard Worker }
2290*1208bc7eSAndroid Build Coastguard Worker 
2291*1208bc7eSAndroid Build Coastguard Worker bool
prof_boot2(tsd_t * tsd)2292*1208bc7eSAndroid Build Coastguard Worker prof_boot2(tsd_t *tsd) {
2293*1208bc7eSAndroid Build Coastguard Worker 	cassert(config_prof);
2294*1208bc7eSAndroid Build Coastguard Worker 
2295*1208bc7eSAndroid Build Coastguard Worker 	if (opt_prof) {
2296*1208bc7eSAndroid Build Coastguard Worker 		unsigned i;
2297*1208bc7eSAndroid Build Coastguard Worker 
2298*1208bc7eSAndroid Build Coastguard Worker 		lg_prof_sample = opt_lg_prof_sample;
2299*1208bc7eSAndroid Build Coastguard Worker 
2300*1208bc7eSAndroid Build Coastguard Worker 		prof_active = opt_prof_active;
2301*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&prof_active_mtx, "prof_active",
2302*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_ACTIVE, malloc_mutex_rank_exclusive)) {
2303*1208bc7eSAndroid Build Coastguard Worker 			return true;
2304*1208bc7eSAndroid Build Coastguard Worker 		}
2305*1208bc7eSAndroid Build Coastguard Worker 
2306*1208bc7eSAndroid Build Coastguard Worker 		prof_gdump_val = opt_prof_gdump;
2307*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&prof_gdump_mtx, "prof_gdump",
2308*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_GDUMP, malloc_mutex_rank_exclusive)) {
2309*1208bc7eSAndroid Build Coastguard Worker 			return true;
2310*1208bc7eSAndroid Build Coastguard Worker 		}
2311*1208bc7eSAndroid Build Coastguard Worker 
2312*1208bc7eSAndroid Build Coastguard Worker 		prof_thread_active_init = opt_prof_thread_active_init;
2313*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&prof_thread_active_init_mtx,
2314*1208bc7eSAndroid Build Coastguard Worker 		    "prof_thread_active_init",
2315*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_THREAD_ACTIVE_INIT,
2316*1208bc7eSAndroid Build Coastguard Worker 		    malloc_mutex_rank_exclusive)) {
2317*1208bc7eSAndroid Build Coastguard Worker 			return true;
2318*1208bc7eSAndroid Build Coastguard Worker 		}
2319*1208bc7eSAndroid Build Coastguard Worker 
2320*1208bc7eSAndroid Build Coastguard Worker 		if (ckh_new(tsd, &bt2gctx, PROF_CKH_MINITEMS, prof_bt_hash,
2321*1208bc7eSAndroid Build Coastguard Worker 		    prof_bt_keycomp)) {
2322*1208bc7eSAndroid Build Coastguard Worker 			return true;
2323*1208bc7eSAndroid Build Coastguard Worker 		}
2324*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&bt2gctx_mtx, "prof_bt2gctx",
2325*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_BT2GCTX, malloc_mutex_rank_exclusive)) {
2326*1208bc7eSAndroid Build Coastguard Worker 			return true;
2327*1208bc7eSAndroid Build Coastguard Worker 		}
2328*1208bc7eSAndroid Build Coastguard Worker 
2329*1208bc7eSAndroid Build Coastguard Worker 		tdata_tree_new(&tdatas);
2330*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&tdatas_mtx, "prof_tdatas",
2331*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_TDATAS, malloc_mutex_rank_exclusive)) {
2332*1208bc7eSAndroid Build Coastguard Worker 			return true;
2333*1208bc7eSAndroid Build Coastguard Worker 		}
2334*1208bc7eSAndroid Build Coastguard Worker 
2335*1208bc7eSAndroid Build Coastguard Worker 		next_thr_uid = 0;
2336*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&next_thr_uid_mtx, "prof_next_thr_uid",
2337*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_NEXT_THR_UID, malloc_mutex_rank_exclusive)) {
2338*1208bc7eSAndroid Build Coastguard Worker 			return true;
2339*1208bc7eSAndroid Build Coastguard Worker 		}
2340*1208bc7eSAndroid Build Coastguard Worker 
2341*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&prof_dump_seq_mtx, "prof_dump_seq",
2342*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_DUMP_SEQ, malloc_mutex_rank_exclusive)) {
2343*1208bc7eSAndroid Build Coastguard Worker 			return true;
2344*1208bc7eSAndroid Build Coastguard Worker 		}
2345*1208bc7eSAndroid Build Coastguard Worker 		if (malloc_mutex_init(&prof_dump_mtx, "prof_dump",
2346*1208bc7eSAndroid Build Coastguard Worker 		    WITNESS_RANK_PROF_DUMP, malloc_mutex_rank_exclusive)) {
2347*1208bc7eSAndroid Build Coastguard Worker 			return true;
2348*1208bc7eSAndroid Build Coastguard Worker 		}
2349*1208bc7eSAndroid Build Coastguard Worker 
2350*1208bc7eSAndroid Build Coastguard Worker 		if (opt_prof_final && opt_prof_prefix[0] != '\0' &&
2351*1208bc7eSAndroid Build Coastguard Worker 		    atexit(prof_fdump) != 0) {
2352*1208bc7eSAndroid Build Coastguard Worker 			malloc_write("<jemalloc>: Error in atexit()\n");
2353*1208bc7eSAndroid Build Coastguard Worker 			if (opt_abort) {
2354*1208bc7eSAndroid Build Coastguard Worker 				abort();
2355*1208bc7eSAndroid Build Coastguard Worker 			}
2356*1208bc7eSAndroid Build Coastguard Worker 		}
2357*1208bc7eSAndroid Build Coastguard Worker 
2358*1208bc7eSAndroid Build Coastguard Worker 		gctx_locks = (malloc_mutex_t *)base_alloc(tsd_tsdn(tsd),
2359*1208bc7eSAndroid Build Coastguard Worker 		    b0get(), PROF_NCTX_LOCKS * sizeof(malloc_mutex_t),
2360*1208bc7eSAndroid Build Coastguard Worker 		    CACHELINE);
2361*1208bc7eSAndroid Build Coastguard Worker 		if (gctx_locks == NULL) {
2362*1208bc7eSAndroid Build Coastguard Worker 			return true;
2363*1208bc7eSAndroid Build Coastguard Worker 		}
2364*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NCTX_LOCKS; i++) {
2365*1208bc7eSAndroid Build Coastguard Worker 			if (malloc_mutex_init(&gctx_locks[i], "prof_gctx",
2366*1208bc7eSAndroid Build Coastguard Worker 			    WITNESS_RANK_PROF_GCTX,
2367*1208bc7eSAndroid Build Coastguard Worker 			    malloc_mutex_rank_exclusive)) {
2368*1208bc7eSAndroid Build Coastguard Worker 				return true;
2369*1208bc7eSAndroid Build Coastguard Worker 			}
2370*1208bc7eSAndroid Build Coastguard Worker 		}
2371*1208bc7eSAndroid Build Coastguard Worker 
2372*1208bc7eSAndroid Build Coastguard Worker 		tdata_locks = (malloc_mutex_t *)base_alloc(tsd_tsdn(tsd),
2373*1208bc7eSAndroid Build Coastguard Worker 		    b0get(), PROF_NTDATA_LOCKS * sizeof(malloc_mutex_t),
2374*1208bc7eSAndroid Build Coastguard Worker 		    CACHELINE);
2375*1208bc7eSAndroid Build Coastguard Worker 		if (tdata_locks == NULL) {
2376*1208bc7eSAndroid Build Coastguard Worker 			return true;
2377*1208bc7eSAndroid Build Coastguard Worker 		}
2378*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NTDATA_LOCKS; i++) {
2379*1208bc7eSAndroid Build Coastguard Worker 			if (malloc_mutex_init(&tdata_locks[i], "prof_tdata",
2380*1208bc7eSAndroid Build Coastguard Worker 			    WITNESS_RANK_PROF_TDATA,
2381*1208bc7eSAndroid Build Coastguard Worker 			    malloc_mutex_rank_exclusive)) {
2382*1208bc7eSAndroid Build Coastguard Worker 				return true;
2383*1208bc7eSAndroid Build Coastguard Worker 			}
2384*1208bc7eSAndroid Build Coastguard Worker 		}
2385*1208bc7eSAndroid Build Coastguard Worker 	}
2386*1208bc7eSAndroid Build Coastguard Worker 
2387*1208bc7eSAndroid Build Coastguard Worker #ifdef JEMALLOC_PROF_LIBGCC
2388*1208bc7eSAndroid Build Coastguard Worker 	/*
2389*1208bc7eSAndroid Build Coastguard Worker 	 * Cause the backtracing machinery to allocate its internal state
2390*1208bc7eSAndroid Build Coastguard Worker 	 * before enabling profiling.
2391*1208bc7eSAndroid Build Coastguard Worker 	 */
2392*1208bc7eSAndroid Build Coastguard Worker 	_Unwind_Backtrace(prof_unwind_init_callback, NULL);
2393*1208bc7eSAndroid Build Coastguard Worker #endif
2394*1208bc7eSAndroid Build Coastguard Worker 
2395*1208bc7eSAndroid Build Coastguard Worker 	prof_booted = true;
2396*1208bc7eSAndroid Build Coastguard Worker 
2397*1208bc7eSAndroid Build Coastguard Worker 	return false;
2398*1208bc7eSAndroid Build Coastguard Worker }
2399*1208bc7eSAndroid Build Coastguard Worker 
2400*1208bc7eSAndroid Build Coastguard Worker void
prof_prefork0(tsdn_t * tsdn)2401*1208bc7eSAndroid Build Coastguard Worker prof_prefork0(tsdn_t *tsdn) {
2402*1208bc7eSAndroid Build Coastguard Worker 	if (config_prof && opt_prof) {
2403*1208bc7eSAndroid Build Coastguard Worker 		unsigned i;
2404*1208bc7eSAndroid Build Coastguard Worker 
2405*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &prof_dump_mtx);
2406*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &bt2gctx_mtx);
2407*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &tdatas_mtx);
2408*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NTDATA_LOCKS; i++) {
2409*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_prefork(tsdn, &tdata_locks[i]);
2410*1208bc7eSAndroid Build Coastguard Worker 		}
2411*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NCTX_LOCKS; i++) {
2412*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_prefork(tsdn, &gctx_locks[i]);
2413*1208bc7eSAndroid Build Coastguard Worker 		}
2414*1208bc7eSAndroid Build Coastguard Worker 	}
2415*1208bc7eSAndroid Build Coastguard Worker }
2416*1208bc7eSAndroid Build Coastguard Worker 
2417*1208bc7eSAndroid Build Coastguard Worker void
prof_prefork1(tsdn_t * tsdn)2418*1208bc7eSAndroid Build Coastguard Worker prof_prefork1(tsdn_t *tsdn) {
2419*1208bc7eSAndroid Build Coastguard Worker 	if (config_prof && opt_prof) {
2420*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &prof_active_mtx);
2421*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &prof_dump_seq_mtx);
2422*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &prof_gdump_mtx);
2423*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &next_thr_uid_mtx);
2424*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_prefork(tsdn, &prof_thread_active_init_mtx);
2425*1208bc7eSAndroid Build Coastguard Worker 	}
2426*1208bc7eSAndroid Build Coastguard Worker }
2427*1208bc7eSAndroid Build Coastguard Worker 
2428*1208bc7eSAndroid Build Coastguard Worker void
prof_postfork_parent(tsdn_t * tsdn)2429*1208bc7eSAndroid Build Coastguard Worker prof_postfork_parent(tsdn_t *tsdn) {
2430*1208bc7eSAndroid Build Coastguard Worker 	if (config_prof && opt_prof) {
2431*1208bc7eSAndroid Build Coastguard Worker 		unsigned i;
2432*1208bc7eSAndroid Build Coastguard Worker 
2433*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn,
2434*1208bc7eSAndroid Build Coastguard Worker 		    &prof_thread_active_init_mtx);
2435*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn, &next_thr_uid_mtx);
2436*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn, &prof_gdump_mtx);
2437*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn, &prof_dump_seq_mtx);
2438*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn, &prof_active_mtx);
2439*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NCTX_LOCKS; i++) {
2440*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_postfork_parent(tsdn, &gctx_locks[i]);
2441*1208bc7eSAndroid Build Coastguard Worker 		}
2442*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NTDATA_LOCKS; i++) {
2443*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_postfork_parent(tsdn, &tdata_locks[i]);
2444*1208bc7eSAndroid Build Coastguard Worker 		}
2445*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn, &tdatas_mtx);
2446*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn, &bt2gctx_mtx);
2447*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_parent(tsdn, &prof_dump_mtx);
2448*1208bc7eSAndroid Build Coastguard Worker 	}
2449*1208bc7eSAndroid Build Coastguard Worker }
2450*1208bc7eSAndroid Build Coastguard Worker 
2451*1208bc7eSAndroid Build Coastguard Worker void
prof_postfork_child(tsdn_t * tsdn)2452*1208bc7eSAndroid Build Coastguard Worker prof_postfork_child(tsdn_t *tsdn) {
2453*1208bc7eSAndroid Build Coastguard Worker 	if (config_prof && opt_prof) {
2454*1208bc7eSAndroid Build Coastguard Worker 		unsigned i;
2455*1208bc7eSAndroid Build Coastguard Worker 
2456*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &prof_thread_active_init_mtx);
2457*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &next_thr_uid_mtx);
2458*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &prof_gdump_mtx);
2459*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &prof_dump_seq_mtx);
2460*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &prof_active_mtx);
2461*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NCTX_LOCKS; i++) {
2462*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_postfork_child(tsdn, &gctx_locks[i]);
2463*1208bc7eSAndroid Build Coastguard Worker 		}
2464*1208bc7eSAndroid Build Coastguard Worker 		for (i = 0; i < PROF_NTDATA_LOCKS; i++) {
2465*1208bc7eSAndroid Build Coastguard Worker 			malloc_mutex_postfork_child(tsdn, &tdata_locks[i]);
2466*1208bc7eSAndroid Build Coastguard Worker 		}
2467*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &tdatas_mtx);
2468*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &bt2gctx_mtx);
2469*1208bc7eSAndroid Build Coastguard Worker 		malloc_mutex_postfork_child(tsdn, &prof_dump_mtx);
2470*1208bc7eSAndroid Build Coastguard Worker 	}
2471*1208bc7eSAndroid Build Coastguard Worker }
2472*1208bc7eSAndroid Build Coastguard Worker 
2473*1208bc7eSAndroid Build Coastguard Worker /******************************************************************************/
2474