1 #include "test/jemalloc_test.h"
2
3 #define NTHREADS 4
4 #define NALLOCS_PER_THREAD 50
5 #define DUMP_INTERVAL 1
6 #define BT_COUNT_CHECK_INTERVAL 5
7
8 #ifdef JEMALLOC_PROF
9 const char *malloc_conf =
10 "prof:true,prof_accum:true,prof_active:false,lg_prof_sample:0";
11 #endif
12
13 static int
prof_dump_open_intercept(bool propagate_err,const char * filename)14 prof_dump_open_intercept(bool propagate_err, const char *filename)
15 {
16 int fd;
17
18 fd = open("/dev/null", O_WRONLY);
19 assert_d_ne(fd, -1, "Unexpected open() failure");
20
21 return (fd);
22 }
23
24 static void *
alloc_from_permuted_backtrace(unsigned thd_ind,unsigned iteration)25 alloc_from_permuted_backtrace(unsigned thd_ind, unsigned iteration)
26 {
27
28 return (btalloc(1, thd_ind*NALLOCS_PER_THREAD + iteration));
29 }
30
31 static void *
thd_start(void * varg)32 thd_start(void *varg)
33 {
34 unsigned thd_ind = *(unsigned *)varg;
35 size_t bt_count_prev, bt_count;
36 unsigned i_prev, i;
37
38 i_prev = 0;
39 bt_count_prev = 0;
40 for (i = 0; i < NALLOCS_PER_THREAD; i++) {
41 void *p = alloc_from_permuted_backtrace(thd_ind, i);
42 dallocx(p, 0);
43 if (i % DUMP_INTERVAL == 0) {
44 assert_d_eq(mallctl("prof.dump", NULL, NULL, NULL, 0),
45 0, "Unexpected error while dumping heap profile");
46 }
47
48 if (i % BT_COUNT_CHECK_INTERVAL == 0 ||
49 i+1 == NALLOCS_PER_THREAD) {
50 bt_count = prof_bt_count();
51 assert_zu_le(bt_count_prev+(i-i_prev), bt_count,
52 "Expected larger backtrace count increase");
53 i_prev = i;
54 bt_count_prev = bt_count;
55 }
56 }
57
58 return (NULL);
59 }
60
TEST_BEGIN(test_idump)61 TEST_BEGIN(test_idump)
62 {
63 bool active;
64 thd_t thds[NTHREADS];
65 unsigned thd_args[NTHREADS];
66 unsigned i;
67
68 test_skip_if(!config_prof);
69
70 active = true;
71 assert_d_eq(mallctl("prof.active", NULL, NULL, &active, sizeof(active)),
72 0, "Unexpected mallctl failure while activating profiling");
73
74 prof_dump_open = prof_dump_open_intercept;
75
76 for (i = 0; i < NTHREADS; i++) {
77 thd_args[i] = i;
78 thd_create(&thds[i], thd_start, (void *)&thd_args[i]);
79 }
80 for (i = 0; i < NTHREADS; i++)
81 thd_join(thds[i], NULL);
82 }
83 TEST_END
84
85 int
main(void)86 main(void)
87 {
88
89 return (test(
90 test_idump));
91 }
92