1 #include "test/jemalloc_test.h"
2
3 #include "jemalloc/internal/util.h"
4
5 static void
test_switch_background_thread_ctl(bool new_val)6 test_switch_background_thread_ctl(bool new_val) {
7 bool e0, e1;
8 size_t sz = sizeof(bool);
9
10 e1 = new_val;
11 assert_d_eq(mallctl("background_thread", (void *)&e0, &sz,
12 &e1, sz), 0, "Unexpected mallctl() failure");
13 assert_b_eq(e0, !e1,
14 "background_thread should be %d before.\n", !e1);
15 if (e1) {
16 assert_zu_gt(n_background_threads, 0,
17 "Number of background threads should be non zero.\n");
18 } else {
19 assert_zu_eq(n_background_threads, 0,
20 "Number of background threads should be zero.\n");
21 }
22 }
23
24 static void
test_repeat_background_thread_ctl(bool before)25 test_repeat_background_thread_ctl(bool before) {
26 bool e0, e1;
27 size_t sz = sizeof(bool);
28
29 e1 = before;
30 assert_d_eq(mallctl("background_thread", (void *)&e0, &sz,
31 &e1, sz), 0, "Unexpected mallctl() failure");
32 assert_b_eq(e0, before,
33 "background_thread should be %d.\n", before);
34 if (e1) {
35 assert_zu_gt(n_background_threads, 0,
36 "Number of background threads should be non zero.\n");
37 } else {
38 assert_zu_eq(n_background_threads, 0,
39 "Number of background threads should be zero.\n");
40 }
41 }
42
TEST_BEGIN(test_background_thread_ctl)43 TEST_BEGIN(test_background_thread_ctl) {
44 test_skip_if(!have_background_thread);
45
46 bool e0, e1;
47 size_t sz = sizeof(bool);
48
49 assert_d_eq(mallctl("opt.background_thread", (void *)&e0, &sz,
50 NULL, 0), 0, "Unexpected mallctl() failure");
51 assert_d_eq(mallctl("background_thread", (void *)&e1, &sz,
52 NULL, 0), 0, "Unexpected mallctl() failure");
53 assert_b_eq(e0, e1,
54 "Default and opt.background_thread does not match.\n");
55 if (e0) {
56 test_switch_background_thread_ctl(false);
57 }
58 assert_zu_eq(n_background_threads, 0,
59 "Number of background threads should be 0.\n");
60
61 for (unsigned i = 0; i < 4; i++) {
62 test_switch_background_thread_ctl(true);
63 test_repeat_background_thread_ctl(true);
64 test_repeat_background_thread_ctl(true);
65
66 test_switch_background_thread_ctl(false);
67 test_repeat_background_thread_ctl(false);
68 test_repeat_background_thread_ctl(false);
69 }
70 }
71 TEST_END
72
TEST_BEGIN(test_background_thread_running)73 TEST_BEGIN(test_background_thread_running) {
74 test_skip_if(!have_background_thread);
75 test_skip_if(!config_stats);
76
77 #if defined(JEMALLOC_BACKGROUND_THREAD)
78 tsd_t *tsd = tsd_fetch();
79 background_thread_info_t *info = &background_thread_info[0];
80
81 test_repeat_background_thread_ctl(false);
82 test_switch_background_thread_ctl(true);
83 assert_b_eq(info->state, background_thread_started,
84 "Background_thread did not start.\n");
85
86 nstime_t start, now;
87 nstime_init(&start, 0);
88 nstime_update(&start);
89
90 bool ran = false;
91 while (true) {
92 malloc_mutex_lock(tsd_tsdn(tsd), &info->mtx);
93 if (info->tot_n_runs > 0) {
94 ran = true;
95 }
96 malloc_mutex_unlock(tsd_tsdn(tsd), &info->mtx);
97 if (ran) {
98 break;
99 }
100
101 nstime_init(&now, 0);
102 nstime_update(&now);
103 nstime_subtract(&now, &start);
104 assert_u64_lt(nstime_sec(&now), 1000,
105 "Background threads did not run for 1000 seconds.");
106 sleep(1);
107 }
108 test_switch_background_thread_ctl(false);
109 #endif
110 }
111 TEST_END
112
113 int
main(void)114 main(void) {
115 /* Background_thread creation tests reentrancy naturally. */
116 return test_no_reentrancy(
117 test_background_thread_ctl,
118 test_background_thread_running);
119 }
120