1*a0698ed9Schristos #include "jemalloc/internal/jemalloc_preamble.h"
2*a0698ed9Schristos #include "jemalloc/internal/jemalloc_internal_includes.h"
3*a0698ed9Schristos
4*a0698ed9Schristos #include "jemalloc/internal/log.h"
5*a0698ed9Schristos
6*a0698ed9Schristos char log_var_names[JEMALLOC_LOG_VAR_BUFSIZE];
7*a0698ed9Schristos atomic_b_t log_init_done = ATOMIC_INIT(false);
8*a0698ed9Schristos
9*a0698ed9Schristos /*
10*a0698ed9Schristos * Returns true if we were able to pick out a segment. Fills in r_segment_end
11*a0698ed9Schristos * with a pointer to the first character after the end of the string.
12*a0698ed9Schristos */
13*a0698ed9Schristos static const char *
log_var_extract_segment(const char * segment_begin)14*a0698ed9Schristos log_var_extract_segment(const char* segment_begin) {
15*a0698ed9Schristos const char *end;
16*a0698ed9Schristos for (end = segment_begin; *end != '\0' && *end != '|'; end++) {
17*a0698ed9Schristos }
18*a0698ed9Schristos return end;
19*a0698ed9Schristos }
20*a0698ed9Schristos
21*a0698ed9Schristos static bool
log_var_matches_segment(const char * segment_begin,const char * segment_end,const char * log_var_begin,const char * log_var_end)22*a0698ed9Schristos log_var_matches_segment(const char *segment_begin, const char *segment_end,
23*a0698ed9Schristos const char *log_var_begin, const char *log_var_end) {
24*a0698ed9Schristos assert(segment_begin <= segment_end);
25*a0698ed9Schristos assert(log_var_begin < log_var_end);
26*a0698ed9Schristos
27*a0698ed9Schristos ptrdiff_t segment_len = segment_end - segment_begin;
28*a0698ed9Schristos ptrdiff_t log_var_len = log_var_end - log_var_begin;
29*a0698ed9Schristos /* The special '.' segment matches everything. */
30*a0698ed9Schristos if (segment_len == 1 && *segment_begin == '.') {
31*a0698ed9Schristos return true;
32*a0698ed9Schristos }
33*a0698ed9Schristos if (segment_len == log_var_len) {
34*a0698ed9Schristos return strncmp(segment_begin, log_var_begin, segment_len) == 0;
35*a0698ed9Schristos } else if (segment_len < log_var_len) {
36*a0698ed9Schristos return strncmp(segment_begin, log_var_begin, segment_len) == 0
37*a0698ed9Schristos && log_var_begin[segment_len] == '.';
38*a0698ed9Schristos } else {
39*a0698ed9Schristos return false;
40*a0698ed9Schristos }
41*a0698ed9Schristos }
42*a0698ed9Schristos
43*a0698ed9Schristos unsigned
log_var_update_state(log_var_t * log_var)44*a0698ed9Schristos log_var_update_state(log_var_t *log_var) {
45*a0698ed9Schristos const char *log_var_begin = log_var->name;
46*a0698ed9Schristos const char *log_var_end = log_var->name + strlen(log_var->name);
47*a0698ed9Schristos
48*a0698ed9Schristos /* Pointer to one before the beginning of the current segment. */
49*a0698ed9Schristos const char *segment_begin = log_var_names;
50*a0698ed9Schristos
51*a0698ed9Schristos /*
52*a0698ed9Schristos * If log_init done is false, we haven't parsed the malloc conf yet. To
53*a0698ed9Schristos * avoid log-spew, we default to not displaying anything.
54*a0698ed9Schristos */
55*a0698ed9Schristos if (!atomic_load_b(&log_init_done, ATOMIC_ACQUIRE)) {
56*a0698ed9Schristos return LOG_INITIALIZED_NOT_ENABLED;
57*a0698ed9Schristos }
58*a0698ed9Schristos
59*a0698ed9Schristos while (true) {
60*a0698ed9Schristos const char *segment_end = log_var_extract_segment(
61*a0698ed9Schristos segment_begin);
62*a0698ed9Schristos assert(segment_end < log_var_names + JEMALLOC_LOG_VAR_BUFSIZE);
63*a0698ed9Schristos if (log_var_matches_segment(segment_begin, segment_end,
64*a0698ed9Schristos log_var_begin, log_var_end)) {
65*a0698ed9Schristos atomic_store_u(&log_var->state, LOG_ENABLED,
66*a0698ed9Schristos ATOMIC_RELAXED);
67*a0698ed9Schristos return LOG_ENABLED;
68*a0698ed9Schristos }
69*a0698ed9Schristos if (*segment_end == '\0') {
70*a0698ed9Schristos /* Hit the end of the segment string with no match. */
71*a0698ed9Schristos atomic_store_u(&log_var->state,
72*a0698ed9Schristos LOG_INITIALIZED_NOT_ENABLED, ATOMIC_RELAXED);
73*a0698ed9Schristos return LOG_INITIALIZED_NOT_ENABLED;
74*a0698ed9Schristos }
75*a0698ed9Schristos /* Otherwise, skip the delimiter and continue. */
76*a0698ed9Schristos segment_begin = segment_end + 1;
77*a0698ed9Schristos }
78*a0698ed9Schristos }
79