1 #include "test/jemalloc_test.h"
2
3 #ifdef JEMALLOC_PROF
4 const char *malloc_conf =
5 "prof:true,prof_thread_active_init:false,lg_prof_sample:0";
6 #endif
7
8 static void
mallctl_bool_get(const char * name,bool expected,const char * func,int line)9 mallctl_bool_get(const char *name, bool expected, const char *func, int line)
10 {
11 bool old;
12 size_t sz;
13
14 sz = sizeof(old);
15 assert_d_eq(mallctl(name, (void *)&old, &sz, NULL, 0), 0,
16 "%s():%d: Unexpected mallctl failure reading %s", func, line, name);
17 assert_b_eq(old, expected, "%s():%d: Unexpected %s value", func, line,
18 name);
19 }
20
21 static void
mallctl_bool_set(const char * name,bool old_expected,bool val_new,const char * func,int line)22 mallctl_bool_set(const char *name, bool old_expected, bool val_new,
23 const char *func, int line)
24 {
25 bool old;
26 size_t sz;
27
28 sz = sizeof(old);
29 assert_d_eq(mallctl(name, (void *)&old, &sz, (void *)&val_new,
30 sizeof(val_new)), 0,
31 "%s():%d: Unexpected mallctl failure reading/writing %s", func,
32 line, name);
33 assert_b_eq(old, old_expected, "%s():%d: Unexpected %s value", func,
34 line, name);
35 }
36
37 static void
mallctl_prof_active_get_impl(bool prof_active_old_expected,const char * func,int line)38 mallctl_prof_active_get_impl(bool prof_active_old_expected, const char *func,
39 int line)
40 {
41
42 mallctl_bool_get("prof.active", prof_active_old_expected, func, line);
43 }
44 #define mallctl_prof_active_get(a) \
45 mallctl_prof_active_get_impl(a, __func__, __LINE__)
46
47 static void
mallctl_prof_active_set_impl(bool prof_active_old_expected,bool prof_active_new,const char * func,int line)48 mallctl_prof_active_set_impl(bool prof_active_old_expected,
49 bool prof_active_new, const char *func, int line)
50 {
51
52 mallctl_bool_set("prof.active", prof_active_old_expected,
53 prof_active_new, func, line);
54 }
55 #define mallctl_prof_active_set(a, b) \
56 mallctl_prof_active_set_impl(a, b, __func__, __LINE__)
57
58 static void
mallctl_thread_prof_active_get_impl(bool thread_prof_active_old_expected,const char * func,int line)59 mallctl_thread_prof_active_get_impl(bool thread_prof_active_old_expected,
60 const char *func, int line)
61 {
62
63 mallctl_bool_get("thread.prof.active", thread_prof_active_old_expected,
64 func, line);
65 }
66 #define mallctl_thread_prof_active_get(a) \
67 mallctl_thread_prof_active_get_impl(a, __func__, __LINE__)
68
69 static void
mallctl_thread_prof_active_set_impl(bool thread_prof_active_old_expected,bool thread_prof_active_new,const char * func,int line)70 mallctl_thread_prof_active_set_impl(bool thread_prof_active_old_expected,
71 bool thread_prof_active_new, const char *func, int line)
72 {
73
74 mallctl_bool_set("thread.prof.active", thread_prof_active_old_expected,
75 thread_prof_active_new, func, line);
76 }
77 #define mallctl_thread_prof_active_set(a, b) \
78 mallctl_thread_prof_active_set_impl(a, b, __func__, __LINE__)
79
80 static void
prof_sampling_probe_impl(bool expect_sample,const char * func,int line)81 prof_sampling_probe_impl(bool expect_sample, const char *func, int line)
82 {
83 void *p;
84 size_t expected_backtraces = expect_sample ? 1 : 0;
85
86 assert_zu_eq(prof_bt_count(), 0, "%s():%d: Expected 0 backtraces", func,
87 line);
88 p = mallocx(1, 0);
89 assert_ptr_not_null(p, "Unexpected mallocx() failure");
90 assert_zu_eq(prof_bt_count(), expected_backtraces,
91 "%s():%d: Unexpected backtrace count", func, line);
92 dallocx(p, 0);
93 }
94 #define prof_sampling_probe(a) \
95 prof_sampling_probe_impl(a, __func__, __LINE__)
96
TEST_BEGIN(test_prof_active)97 TEST_BEGIN(test_prof_active)
98 {
99
100 test_skip_if(!config_prof);
101
102 mallctl_prof_active_get(true);
103 mallctl_thread_prof_active_get(false);
104
105 mallctl_prof_active_set(true, true);
106 mallctl_thread_prof_active_set(false, false);
107 /* prof.active, !thread.prof.active. */
108 prof_sampling_probe(false);
109
110 mallctl_prof_active_set(true, false);
111 mallctl_thread_prof_active_set(false, false);
112 /* !prof.active, !thread.prof.active. */
113 prof_sampling_probe(false);
114
115 mallctl_prof_active_set(false, false);
116 mallctl_thread_prof_active_set(false, true);
117 /* !prof.active, thread.prof.active. */
118 prof_sampling_probe(false);
119
120 mallctl_prof_active_set(false, true);
121 mallctl_thread_prof_active_set(true, true);
122 /* prof.active, thread.prof.active. */
123 prof_sampling_probe(true);
124
125 /* Restore settings. */
126 mallctl_prof_active_set(true, true);
127 mallctl_thread_prof_active_set(true, false);
128 }
129 TEST_END
130
131 int
main(void)132 main(void)
133 {
134
135 return (test(
136 test_prof_active));
137 }
138