Branch data Line data Source code
1 : : /*
2 : : * Copyright (c) 2016, Citrix Systems, Inc.
3 : : *
4 : : * All rights reserved.
5 : : *
6 : : * Redistribution and use in source and binary forms, with or without
7 : : * modification, are permitted provided that the following conditions are met:
8 : : *
9 : : * 1. Redistributions of source code must retain the above copyright
10 : : * notice, this list of conditions and the following disclaimer.
11 : : * 2. Redistributions in binary form must reproduce the above copyright
12 : : * notice, this list of conditions and the following disclaimer in the
13 : : * documentation and/or other materials provided with the distribution.
14 : : * 3. Neither the name of the copyright holder nor the names of its
15 : : * contributors may be used to endorse or promote products derived from
16 : : * this software without specific prior written permission.
17 : : *
18 : : * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 : : * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 : : * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 : : * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
22 : : * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
23 : : * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
24 : : * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
25 : : * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
26 : : * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
27 : : * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
28 : : * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 : : */
30 : :
31 : : #ifdef HAVE_CONFIG_H
32 : : #include "config.h"
33 : : #endif
34 : :
35 : : #include <stdio.h>
36 : : #include <stdarg.h>
37 : : #include <stdlib.h>
38 : : #include <errno.h>
39 : :
40 : : #include "tapdisk.h"
41 : : #include "tapdisk-stats.h"
42 : :
43 : : #define BUG_ON(_cond) if (_cond) { td_panic(); }
44 : :
45 : : static void
46 : 37 : __stats_vsprintf(td_stats_t *st,
47 : : const char *fmt, va_list ap)
48 : : {
49 : : char *buf;
50 : : int written, new_size, off;
51 : 37 : size_t size = 0;
52 : 37 : written = 1;
53 [ + - ]: 39 : while (written >= size) {
54 : : va_list aq;
55 : :
56 : 39 : size = st->buf + st->size - st->pos;
57 : 39 : va_copy(aq, ap);
58 : 78 : written = vsnprintf(st->pos, size, fmt, aq);
59 : 39 : va_end(aq);
60 [ + + ]: 39 : if (written < size)
61 : : break;
62 : 2 : new_size = st->size * 2;
63 : 2 : buf = realloc(st->buf, new_size);
64 [ - + ]: 2 : if (!buf) {
65 : 0 : st->err = -ENOMEM;
66 : 0 : written = size;
67 : 0 : break;
68 : : }
69 : 2 : off = st->pos - st->buf;
70 : 2 : st->buf = buf;
71 : 2 : st->size = new_size;
72 : 39 : st->pos = st->buf + off;
73 : : }
74 : 37 : st->pos += written;
75 : 37 : }
76 : :
77 : : static void __printf(2, 3)
78 : 18 : __stats_sprintf(td_stats_t *st,
79 : : const char *fmt, ...)
80 : : {
81 : : va_list ap;
82 : :
83 : 18 : va_start(ap, fmt);
84 : 18 : __stats_vsprintf(st, fmt, ap);
85 : 18 : va_end(ap);
86 : 18 : }
87 : :
88 : : static void
89 : : __stats_enter(td_stats_t *st)
90 : : {
91 : 1 : st->depth++;
92 [ # # ][ - + ]: 1 : BUG_ON(st->depth > TD_STATS_MAX_DEPTH);
93 : 1 : st->n_elem[st->depth] = 0;
94 : : }
95 : :
96 : : static void
97 : 0 : __stats_leave(td_stats_t *st)
98 : : {
99 : 1 : st->depth--;
100 : 0 : }
101 : :
102 : : static void
103 : 20 : __stats_next(td_stats_t *st)
104 : : {
105 : : int n_elem;
106 : :
107 : 20 : n_elem = st->n_elem[st->depth];
108 [ + + ]: 20 : if (n_elem > 0)
109 : 16 : __stats_sprintf(st, ", ");
110 : 20 : st->n_elem[st->depth]++;
111 : 20 : }
112 : :
113 : : static void
114 : 1 : __tapdisk_stats_enter(td_stats_t *st, char t)
115 : : {
116 : 1 : __stats_sprintf(st, "%c ", t);
117 : : __stats_enter(st);
118 : 1 : }
119 : :
120 : : void
121 : 1 : tapdisk_stats_enter(td_stats_t *st, char t)
122 : : {
123 : 1 : __stats_next(st);
124 : 1 : __tapdisk_stats_enter(st, t);
125 : 1 : }
126 : :
127 : : void
128 : 1 : tapdisk_stats_leave(td_stats_t *st, char t)
129 : : {
130 : 1 : __stats_leave(st);
131 : 1 : __stats_sprintf(st, " %c", t);
132 : 1 : }
133 : :
134 : : static void
135 : 19 : tapdisk_stats_vval(td_stats_t *st, const char *conv, va_list ap)
136 : : {
137 : 19 : char t = conv[0], fmt[32];
138 : :
139 : 19 : __stats_next(st);
140 : :
141 [ - + ]: 19 : switch (t) {
142 : : case 's':
143 : 0 : __stats_vsprintf(st, "\"%s\"", ap);
144 : : break;
145 : :
146 : : default:
147 : : snprintf(fmt, sizeof(fmt), "%%%s", conv);
148 : 19 : __stats_vsprintf(st, fmt, ap);
149 : : break;
150 : : }
151 : 19 : }
152 : :
153 : : void
154 : 19 : tapdisk_stats_val(td_stats_t *st, const char *conv, ...)
155 : : {
156 : : va_list ap;
157 : :
158 : 19 : va_start(ap, conv);
159 : 19 : tapdisk_stats_vval(st, conv, ap);
160 : 19 : va_end(ap);
161 : 19 : }
162 : :
163 : : void
164 : 0 : tapdisk_stats_field(td_stats_t *st, const char *key, const char *conv, ...)
165 : : {
166 : : va_list ap;
167 : : int n_elem;
168 : : char t;
169 : :
170 : 0 : n_elem = st->n_elem[st->depth]++;
171 [ # # ]: 0 : if (n_elem > 0)
172 : 0 : __stats_sprintf(st, ", ");
173 : :
174 : 0 : __stats_sprintf(st, "\"%s\": ", key);
175 : :
176 [ # # ]: 0 : if (!conv) {
177 : 0 : __stats_sprintf(st, "null");
178 : 0 : return;
179 : : }
180 : :
181 : 0 : t = conv[0];
182 [ # # ]: 0 : switch (t) {
183 : : case '[':
184 : : case '{':
185 : 0 : __tapdisk_stats_enter(st, t);
186 : : break;
187 : : default:
188 : 0 : va_start(ap, conv);
189 : : __stats_enter(st);
190 : 0 : tapdisk_stats_vval(st, conv, ap);
191 : 0 : __stats_leave(st);
192 : 0 : va_end(ap);
193 : : }
194 : : }
|