Commit | Line | Data |
---|---|---|
f3ed775e DG |
1 | /* |
2 | * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca> | |
3 | * | |
d14d33bf AM |
4 | * This program is free software; you can redistribute it and/or modify |
5 | * it under the terms of the GNU General Public License, version 2 only, | |
6 | * as published by the Free Software Foundation. | |
f3ed775e DG |
7 | * |
8 | * This program is distributed in the hope that it will be useful, | |
9 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
10 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
11 | * GNU General Public License for more details. | |
12 | * | |
d14d33bf AM |
13 | * You should have received a copy of the GNU General Public License along |
14 | * with this program; if not, write to the Free Software Foundation, Inc., | |
15 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | |
f3ed775e DG |
16 | */ |
17 | ||
6c1c0768 | 18 | #define _LGPL_SOURCE |
b2064f54 | 19 | #include <assert.h> |
f3ed775e DG |
20 | #include <popt.h> |
21 | #include <stdio.h> | |
22 | #include <stdlib.h> | |
f3ed775e DG |
23 | #include <sys/stat.h> |
24 | #include <sys/types.h> | |
25 | #include <unistd.h> | |
5a0de755 | 26 | #include <inttypes.h> |
8f0d098b | 27 | #include <ctype.h> |
f3ed775e | 28 | |
42224349 | 29 | #include <src/common/sessiond-comm/sessiond-comm.h> |
f5436bfc | 30 | #include <common/compat/string.h> |
f3ed775e | 31 | |
89476427 JRJ |
32 | /* Mi dependancy */ |
33 | #include <common/mi-lttng.h> | |
34 | ||
35 | #include "../command.h" | |
36 | ||
8ab7c0d9 MD |
37 | #if (LTTNG_SYMBOL_NAME_LEN == 256) |
38 | #define LTTNG_SYMBOL_NAME_LEN_SCANF_IS_A_BROKEN_API "255" | |
39 | #endif | |
40 | ||
f3ed775e DG |
41 | static char *opt_event_list; |
42 | static int opt_event_type; | |
0cda4f28 MD |
43 | static const char *opt_loglevel; |
44 | static int opt_loglevel_type; | |
6181537c | 45 | static int opt_kernel; |
5440dc42 | 46 | static char *opt_session_name; |
721b94fe | 47 | static int opt_domain; |
f3ed775e | 48 | static int opt_enable_all; |
cf0e5467 | 49 | static char *opt_probe; |
8f0d098b | 50 | static char *opt_function; |
f3ed775e | 51 | static char *opt_channel_name; |
53a80697 | 52 | static char *opt_filter; |
fac3366c | 53 | static char *opt_exclude; |
f3ed775e DG |
54 | |
55 | enum { | |
56 | OPT_HELP = 1, | |
f3ed775e | 57 | OPT_TRACEPOINT, |
cf0e5467 | 58 | OPT_PROBE, |
f3ed775e | 59 | OPT_FUNCTION, |
a54bd42d | 60 | OPT_SYSCALL, |
0cda4f28 MD |
61 | OPT_LOGLEVEL, |
62 | OPT_LOGLEVEL_ONLY, | |
679b4943 | 63 | OPT_LIST_OPTIONS, |
53a80697 | 64 | OPT_FILTER, |
fac3366c | 65 | OPT_EXCLUDE, |
f3ed775e DG |
66 | }; |
67 | ||
cd80958d | 68 | static struct lttng_handle *handle; |
89476427 | 69 | static struct mi_writer *writer; |
cd80958d | 70 | |
f3ed775e DG |
71 | static struct poptOption long_options[] = { |
72 | /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */ | |
e14f64a8 | 73 | {"all", 'a', POPT_ARG_VAL, &opt_enable_all, 1, 0, 0}, |
f3ed775e | 74 | {"channel", 'c', POPT_ARG_STRING, &opt_channel_name, 0, 0, 0}, |
721b94fe JR |
75 | {"kernel", 'k', POPT_ARG_VAL, &opt_domain, LTTNG_DOMAIN_KERNEL, 0, 0}, |
76 | {"userspace", 'u', POPT_ARG_VAL, &opt_domain, LTTNG_DOMAIN_UST, 0, 0}, | |
77 | {"jul", 'j', POPT_ARG_VAL, &opt_domain, LTTNG_DOMAIN_JUL, 0, 0}, | |
78 | {"log4j", 'l', POPT_ARG_VAL, &opt_domain, LTTNG_DOMAIN_LOG4J, 0, 0}, | |
79 | {"python", 'p', POPT_ARG_VAL, &opt_domain, LTTNG_DOMAIN_PYTHON, 0, 0}, | |
f3ed775e | 80 | {"tracepoint", 0, POPT_ARG_NONE, 0, OPT_TRACEPOINT, 0, 0}, |
7ebae521 | 81 | {"probe", 0, POPT_ARG_STRING, &opt_probe, OPT_PROBE, 0, 0}, |
40e9d5d3 | 82 | {"function", 0, POPT_ARG_STRING, &opt_function, OPT_FUNCTION, 0, 0}, |
7ebae521 | 83 | {"syscall", 0, POPT_ARG_NONE, 0, OPT_SYSCALL, 0, 0}, |
0cda4f28 MD |
84 | {"loglevel", 0, POPT_ARG_STRING, 0, OPT_LOGLEVEL, 0, 0}, |
85 | {"loglevel-only", 0, POPT_ARG_STRING, 0, OPT_LOGLEVEL_ONLY, 0, 0}, | |
53a80697 | 86 | {"filter", 'f', POPT_ARG_STRING, &opt_filter, OPT_FILTER, 0, 0}, |
fac3366c | 87 | {"exclude", 'x', POPT_ARG_STRING, &opt_exclude, OPT_EXCLUDE, 0, 0}, |
721b94fe | 88 | {"session", 's', POPT_ARG_NONE, 0, 0, 0, 0}, |
f3ed775e DG |
89 | {0, 0, 0, 0, 0, 0, 0} |
90 | }; | |
91 | ||
721b94fe JR |
92 | static struct poptOption global_long_options[] = { |
93 | /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */ | |
94 | {"help", 'h', POPT_ARG_NONE, 0, OPT_HELP, 0, 0}, | |
95 | {"session", 's', POPT_ARG_STRING, &opt_session_name, 0, 0, 0}, | |
96 | {"list-options", 0, POPT_ARG_NONE, NULL, OPT_LIST_OPTIONS, NULL, NULL}, | |
97 | {0, 0, 0, 0, 0, 0, 0} | |
98 | }; | |
0d63dd19 | 99 | /* |
6181537c | 100 | * Parse probe options. |
0d63dd19 | 101 | */ |
cf0e5467 | 102 | static int parse_probe_opts(struct lttng_event *ev, char *opt) |
0d63dd19 | 103 | { |
49d4e302 JRJ |
104 | int ret = CMD_SUCCESS; |
105 | int match; | |
8ff0bbd0 | 106 | char s_hex[19]; |
8ab7c0d9 | 107 | #define S_HEX_LEN_SCANF_IS_A_BROKEN_API "18" /* 18 is (19 - 1) (\0 is extra) */ |
0d63dd19 DG |
108 | char name[LTTNG_SYMBOL_NAME_LEN]; |
109 | ||
110 | if (opt == NULL) { | |
49d4e302 | 111 | ret = CMD_ERROR; |
8f0d098b | 112 | goto end; |
0d63dd19 DG |
113 | } |
114 | ||
115 | /* Check for symbol+offset */ | |
49d4e302 | 116 | match = sscanf(opt, "%" LTTNG_SYMBOL_NAME_LEN_SCANF_IS_A_BROKEN_API |
8ab7c0d9 | 117 | "[^'+']+%" S_HEX_LEN_SCANF_IS_A_BROKEN_API "s", name, s_hex); |
49d4e302 | 118 | if (match == 2) { |
7d29a247 | 119 | strncpy(ev->attr.probe.symbol_name, name, LTTNG_SYMBOL_NAME_LEN); |
99497cd0 | 120 | ev->attr.probe.symbol_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; |
cf0e5467 | 121 | DBG("probe symbol %s", ev->attr.probe.symbol_name); |
9d035200 | 122 | if (*s_hex == '\0') { |
8ff0bbd0 | 123 | ERR("Invalid probe offset %s", s_hex); |
49d4e302 | 124 | ret = CMD_ERROR; |
8f0d098b | 125 | goto end; |
0d63dd19 | 126 | } |
8ff0bbd0 | 127 | ev->attr.probe.offset = strtoul(s_hex, NULL, 0); |
cf0e5467 | 128 | DBG("probe offset %" PRIu64, ev->attr.probe.offset); |
3000dc78 | 129 | ev->attr.probe.addr = 0; |
8f0d098b MD |
130 | goto end; |
131 | } | |
132 | ||
133 | /* Check for symbol */ | |
134 | if (isalpha(name[0])) { | |
49d4e302 | 135 | match = sscanf(opt, "%" LTTNG_SYMBOL_NAME_LEN_SCANF_IS_A_BROKEN_API "s", |
8ab7c0d9 | 136 | name); |
49d4e302 | 137 | if (match == 1) { |
8f0d098b | 138 | strncpy(ev->attr.probe.symbol_name, name, LTTNG_SYMBOL_NAME_LEN); |
99497cd0 | 139 | ev->attr.probe.symbol_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; |
8f0d098b MD |
140 | DBG("probe symbol %s", ev->attr.probe.symbol_name); |
141 | ev->attr.probe.offset = 0; | |
142 | DBG("probe offset %" PRIu64, ev->attr.probe.offset); | |
143 | ev->attr.probe.addr = 0; | |
144 | goto end; | |
145 | } | |
0d63dd19 DG |
146 | } |
147 | ||
148 | /* Check for address */ | |
49d4e302 JRJ |
149 | match = sscanf(opt, "%" S_HEX_LEN_SCANF_IS_A_BROKEN_API "s", s_hex); |
150 | if (match > 0) { | |
9d035200 | 151 | if (*s_hex == '\0') { |
8ff0bbd0 | 152 | ERR("Invalid probe address %s", s_hex); |
49d4e302 | 153 | ret = CMD_ERROR; |
8f0d098b | 154 | goto end; |
0d63dd19 | 155 | } |
8ff0bbd0 | 156 | ev->attr.probe.addr = strtoul(s_hex, NULL, 0); |
cf0e5467 | 157 | DBG("probe addr %" PRIu64, ev->attr.probe.addr); |
3000dc78 DG |
158 | ev->attr.probe.offset = 0; |
159 | memset(ev->attr.probe.symbol_name, 0, LTTNG_SYMBOL_NAME_LEN); | |
8f0d098b | 160 | goto end; |
0d63dd19 DG |
161 | } |
162 | ||
163 | /* No match */ | |
49d4e302 | 164 | ret = CMD_ERROR; |
0d63dd19 | 165 | |
8f0d098b | 166 | end: |
0d63dd19 DG |
167 | return ret; |
168 | } | |
169 | ||
5cdb6027 DG |
170 | /* |
171 | * Maps LOG4j loglevel from string to value | |
172 | */ | |
173 | static int loglevel_log4j_str_to_value(const char *inputstr) | |
174 | { | |
175 | int i = 0; | |
176 | char str[LTTNG_SYMBOL_NAME_LEN]; | |
177 | ||
3712b110 JG |
178 | if (!inputstr || strlen(inputstr) == 0) { |
179 | return -1; | |
180 | } | |
181 | ||
5cdb6027 DG |
182 | /* |
183 | * Loop up to LTTNG_SYMBOL_NAME_LEN minus one because the NULL bytes is | |
184 | * added at the end of the loop so a the upper bound we avoid the overflow. | |
185 | */ | |
186 | while (i < (LTTNG_SYMBOL_NAME_LEN - 1) && inputstr[i] != '\0') { | |
187 | str[i] = toupper(inputstr[i]); | |
188 | i++; | |
189 | } | |
190 | str[i] = '\0'; | |
191 | ||
192 | if (!strcmp(str, "LOG4J_OFF") || !strcmp(str, "OFF")) { | |
193 | return LTTNG_LOGLEVEL_LOG4J_OFF; | |
194 | } else if (!strcmp(str, "LOG4J_FATAL") || !strcmp(str, "FATAL")) { | |
195 | return LTTNG_LOGLEVEL_LOG4J_FATAL; | |
196 | } else if (!strcmp(str, "LOG4J_ERROR") || !strcmp(str, "ERROR")) { | |
197 | return LTTNG_LOGLEVEL_LOG4J_ERROR; | |
198 | } else if (!strcmp(str, "LOG4J_WARN") || !strcmp(str, "WARN")) { | |
199 | return LTTNG_LOGLEVEL_LOG4J_WARN; | |
200 | } else if (!strcmp(str, "LOG4J_INFO") || !strcmp(str, "INFO")) { | |
201 | return LTTNG_LOGLEVEL_LOG4J_INFO; | |
202 | } else if (!strcmp(str, "LOG4J_DEBUG") || !strcmp(str, "DEBUG")) { | |
203 | return LTTNG_LOGLEVEL_LOG4J_DEBUG; | |
204 | } else if (!strcmp(str, "LOG4J_TRACE") || !strcmp(str, "TRACE")) { | |
205 | return LTTNG_LOGLEVEL_LOG4J_TRACE; | |
206 | } else if (!strcmp(str, "LOG4J_ALL") || !strcmp(str, "ALL")) { | |
207 | return LTTNG_LOGLEVEL_LOG4J_ALL; | |
208 | } else { | |
209 | return -1; | |
210 | } | |
211 | } | |
212 | ||
b2064f54 DG |
213 | /* |
214 | * Maps JUL loglevel from string to value | |
215 | */ | |
216 | static int loglevel_jul_str_to_value(const char *inputstr) | |
217 | { | |
218 | int i = 0; | |
219 | char str[LTTNG_SYMBOL_NAME_LEN]; | |
220 | ||
3712b110 JG |
221 | if (!inputstr || strlen(inputstr) == 0) { |
222 | return -1; | |
223 | } | |
224 | ||
b2064f54 DG |
225 | /* |
226 | * Loop up to LTTNG_SYMBOL_NAME_LEN minus one because the NULL bytes is | |
227 | * added at the end of the loop so a the upper bound we avoid the overflow. | |
228 | */ | |
229 | while (i < (LTTNG_SYMBOL_NAME_LEN - 1) && inputstr[i] != '\0') { | |
230 | str[i] = toupper(inputstr[i]); | |
231 | i++; | |
232 | } | |
233 | str[i] = '\0'; | |
234 | ||
235 | if (!strcmp(str, "JUL_OFF") || !strcmp(str, "OFF")) { | |
236 | return LTTNG_LOGLEVEL_JUL_OFF; | |
237 | } else if (!strcmp(str, "JUL_SEVERE") || !strcmp(str, "SEVERE")) { | |
238 | return LTTNG_LOGLEVEL_JUL_SEVERE; | |
239 | } else if (!strcmp(str, "JUL_WARNING") || !strcmp(str, "WARNING")) { | |
240 | return LTTNG_LOGLEVEL_JUL_WARNING; | |
241 | } else if (!strcmp(str, "JUL_INFO") || !strcmp(str, "INFO")) { | |
242 | return LTTNG_LOGLEVEL_JUL_INFO; | |
243 | } else if (!strcmp(str, "JUL_CONFIG") || !strcmp(str, "CONFIG")) { | |
244 | return LTTNG_LOGLEVEL_JUL_CONFIG; | |
245 | } else if (!strcmp(str, "JUL_FINE") || !strcmp(str, "FINE")) { | |
246 | return LTTNG_LOGLEVEL_JUL_FINE; | |
247 | } else if (!strcmp(str, "JUL_FINER") || !strcmp(str, "FINER")) { | |
248 | return LTTNG_LOGLEVEL_JUL_FINER; | |
249 | } else if (!strcmp(str, "JUL_FINEST") || !strcmp(str, "FINEST")) { | |
250 | return LTTNG_LOGLEVEL_JUL_FINEST; | |
251 | } else if (!strcmp(str, "JUL_ALL") || !strcmp(str, "ALL")) { | |
252 | return LTTNG_LOGLEVEL_JUL_ALL; | |
253 | } else { | |
254 | return -1; | |
255 | } | |
256 | } | |
257 | ||
0e115563 DG |
258 | /* |
259 | * Maps Python loglevel from string to value | |
260 | */ | |
261 | static int loglevel_python_str_to_value(const char *inputstr) | |
262 | { | |
263 | int i = 0; | |
264 | char str[LTTNG_SYMBOL_NAME_LEN]; | |
265 | ||
3712b110 JG |
266 | if (!inputstr || strlen(inputstr) == 0) { |
267 | return -1; | |
268 | } | |
269 | ||
0e115563 DG |
270 | /* |
271 | * Loop up to LTTNG_SYMBOL_NAME_LEN minus one because the NULL bytes is | |
272 | * added at the end of the loop so a the upper bound we avoid the overflow. | |
273 | */ | |
274 | while (i < (LTTNG_SYMBOL_NAME_LEN - 1) && inputstr[i] != '\0') { | |
275 | str[i] = toupper(inputstr[i]); | |
276 | i++; | |
277 | } | |
278 | str[i] = '\0'; | |
279 | ||
280 | if (!strcmp(str, "PYTHON_CRITICAL") || !strcmp(str, "CRITICAL")) { | |
281 | return LTTNG_LOGLEVEL_PYTHON_CRITICAL; | |
282 | } else if (!strcmp(str, "PYTHON_ERROR") || !strcmp(str, "ERROR")) { | |
283 | return LTTNG_LOGLEVEL_PYTHON_ERROR; | |
284 | } else if (!strcmp(str, "PYTHON_WARNING") || !strcmp(str, "WARNING")) { | |
285 | return LTTNG_LOGLEVEL_PYTHON_WARNING; | |
286 | } else if (!strcmp(str, "PYTHON_INFO") || !strcmp(str, "INFO")) { | |
287 | return LTTNG_LOGLEVEL_PYTHON_INFO; | |
288 | } else if (!strcmp(str, "PYTNON_DEBUG") || !strcmp(str, "DEBUG")) { | |
289 | return LTTNG_LOGLEVEL_PYTHON_DEBUG; | |
290 | } else if (!strcmp(str, "PYTHON_NOTSET") || !strcmp(str, "NOTSET")) { | |
291 | return LTTNG_LOGLEVEL_PYTHON_NOTSET; | |
292 | } else { | |
293 | return -1; | |
294 | } | |
295 | } | |
296 | ||
8005f29a | 297 | /* |
0c8477c8 | 298 | * Maps loglevel from string to value |
8005f29a MD |
299 | */ |
300 | static | |
46839cc2 | 301 | int loglevel_str_to_value(const char *inputstr) |
8005f29a | 302 | { |
46839cc2 MD |
303 | int i = 0; |
304 | char str[LTTNG_SYMBOL_NAME_LEN]; | |
305 | ||
3712b110 JG |
306 | if (!inputstr || strlen(inputstr) == 0) { |
307 | return -1; | |
308 | } | |
309 | ||
e051e07c DG |
310 | /* |
311 | * Loop up to LTTNG_SYMBOL_NAME_LEN minus one because the NULL bytes is | |
312 | * added at the end of the loop so a the upper bound we avoid the overflow. | |
313 | */ | |
314 | while (i < (LTTNG_SYMBOL_NAME_LEN - 1) && inputstr[i] != '\0') { | |
46839cc2 MD |
315 | str[i] = toupper(inputstr[i]); |
316 | i++; | |
317 | } | |
318 | str[i] = '\0'; | |
319 | if (!strcmp(str, "TRACE_EMERG") || !strcmp(str, "EMERG")) { | |
320 | return LTTNG_LOGLEVEL_EMERG; | |
321 | } else if (!strcmp(str, "TRACE_ALERT") || !strcmp(str, "ALERT")) { | |
322 | return LTTNG_LOGLEVEL_ALERT; | |
323 | } else if (!strcmp(str, "TRACE_CRIT") || !strcmp(str, "CRIT")) { | |
324 | return LTTNG_LOGLEVEL_CRIT; | |
325 | } else if (!strcmp(str, "TRACE_ERR") || !strcmp(str, "ERR")) { | |
326 | return LTTNG_LOGLEVEL_ERR; | |
327 | } else if (!strcmp(str, "TRACE_WARNING") || !strcmp(str, "WARNING")) { | |
328 | return LTTNG_LOGLEVEL_WARNING; | |
329 | } else if (!strcmp(str, "TRACE_NOTICE") || !strcmp(str, "NOTICE")) { | |
330 | return LTTNG_LOGLEVEL_NOTICE; | |
331 | } else if (!strcmp(str, "TRACE_INFO") || !strcmp(str, "INFO")) { | |
332 | return LTTNG_LOGLEVEL_INFO; | |
333 | } else if (!strcmp(str, "TRACE_DEBUG_SYSTEM") || !strcmp(str, "DEBUG_SYSTEM") || !strcmp(str, "SYSTEM")) { | |
334 | return LTTNG_LOGLEVEL_DEBUG_SYSTEM; | |
335 | } else if (!strcmp(str, "TRACE_DEBUG_PROGRAM") || !strcmp(str, "DEBUG_PROGRAM") || !strcmp(str, "PROGRAM")) { | |
336 | return LTTNG_LOGLEVEL_DEBUG_PROGRAM; | |
337 | } else if (!strcmp(str, "TRACE_DEBUG_PROCESS") || !strcmp(str, "DEBUG_PROCESS") || !strcmp(str, "PROCESS")) { | |
338 | return LTTNG_LOGLEVEL_DEBUG_PROCESS; | |
339 | } else if (!strcmp(str, "TRACE_DEBUG_MODULE") || !strcmp(str, "DEBUG_MODULE") || !strcmp(str, "MODULE")) { | |
340 | return LTTNG_LOGLEVEL_DEBUG_MODULE; | |
341 | } else if (!strcmp(str, "TRACE_DEBUG_UNIT") || !strcmp(str, "DEBUG_UNIT") || !strcmp(str, "UNIT")) { | |
342 | return LTTNG_LOGLEVEL_DEBUG_UNIT; | |
343 | } else if (!strcmp(str, "TRACE_DEBUG_FUNCTION") || !strcmp(str, "DEBUG_FUNCTION") || !strcmp(str, "FUNCTION")) { | |
344 | return LTTNG_LOGLEVEL_DEBUG_FUNCTION; | |
345 | } else if (!strcmp(str, "TRACE_DEBUG_LINE") || !strcmp(str, "DEBUG_LINE") || !strcmp(str, "LINE")) { | |
346 | return LTTNG_LOGLEVEL_DEBUG_LINE; | |
347 | } else if (!strcmp(str, "TRACE_DEBUG") || !strcmp(str, "DEBUG")) { | |
348 | return LTTNG_LOGLEVEL_DEBUG; | |
8005f29a MD |
349 | } else { |
350 | return -1; | |
351 | } | |
352 | } | |
353 | ||
85076754 MD |
354 | static |
355 | const char *print_channel_name(const char *name) | |
356 | { | |
357 | return name ? : DEFAULT_CHANNEL_NAME; | |
358 | } | |
359 | ||
360 | static | |
361 | const char *print_raw_channel_name(const char *name) | |
362 | { | |
363 | return name ? : "<default>"; | |
364 | } | |
365 | ||
89476427 JRJ |
366 | /* |
367 | * Mi print exlcusion list | |
368 | */ | |
369 | static | |
370 | int mi_print_exclusion(int count, char **names) | |
371 | { | |
372 | int i, ret; | |
373 | ||
374 | assert(writer); | |
375 | ||
376 | if (count == 0) { | |
377 | ret = 0; | |
378 | goto end; | |
379 | } | |
380 | ret = mi_lttng_writer_open_element(writer, config_element_exclusions); | |
381 | if (ret) { | |
382 | goto end; | |
383 | } | |
384 | ||
385 | for (i = 0; i < count; i++) { | |
386 | ret = mi_lttng_writer_write_element_string(writer, | |
387 | config_element_exclusion, names[i]); | |
388 | if (ret) { | |
389 | goto end; | |
390 | } | |
391 | } | |
392 | ||
393 | /* Close exclusions element */ | |
394 | ret = mi_lttng_writer_close_element(writer); | |
395 | ||
396 | end: | |
397 | return ret; | |
398 | } | |
399 | ||
9c48cab3 JI |
400 | /* |
401 | * Return allocated string for pretty-printing exclusion names. | |
402 | */ | |
403 | static | |
404 | char *print_exclusions(int count, char **names) | |
405 | { | |
406 | int length = 0; | |
407 | int i; | |
408 | const char *preamble = " excluding "; | |
409 | char *ret; | |
410 | ||
411 | if (count == 0) { | |
412 | return strdup(""); | |
413 | } | |
414 | ||
415 | /* calculate total required length */ | |
416 | for (i = 0; i < count; i++) { | |
417 | length += strlen(names[i]) + 1; | |
418 | } | |
419 | ||
420 | /* add length of preamble + one for NUL - one for last (missing) comma */ | |
421 | length += strlen(preamble); | |
fd591b18 MD |
422 | ret = zmalloc(length); |
423 | if (!ret) { | |
424 | return NULL; | |
425 | } | |
9c48cab3 JI |
426 | strncpy(ret, preamble, length); |
427 | for (i = 0; i < count; i++) { | |
428 | strcat(ret, names[i]); | |
429 | if (i != count - 1) { | |
430 | strcat(ret, ","); | |
431 | } | |
432 | } | |
89476427 | 433 | |
9c48cab3 JI |
434 | return ret; |
435 | } | |
436 | ||
748bde76 JI |
437 | /* |
438 | * Compare list of exclusions against an event name. | |
439 | * Return a list of legal exclusion names. | |
440 | * Produce an error or a warning about others (depending on the situation) | |
441 | */ | |
442 | static | |
443 | int check_exclusion_subsets(const char *event_name, | |
444 | const char *exclusions, | |
445 | int *exclusion_count_ptr, | |
446 | char ***exclusion_list_ptr) | |
447 | { | |
448 | const char *excluder_ptr; | |
449 | const char *event_ptr; | |
450 | const char *next_excluder; | |
451 | int excluder_length; | |
452 | int exclusion_count = 0; | |
453 | char **exclusion_list = NULL; | |
454 | int ret = CMD_SUCCESS; | |
455 | ||
456 | if (event_name[strlen(event_name) - 1] != '*') { | |
457 | ERR("Event %s: Excluders can only be used with wildcarded events", event_name); | |
458 | goto error; | |
459 | } | |
460 | ||
461 | next_excluder = exclusions; | |
462 | while (*next_excluder != 0) { | |
463 | event_ptr = event_name; | |
464 | excluder_ptr = next_excluder; | |
465 | excluder_length = strcspn(next_excluder, ","); | |
466 | ||
467 | /* Scan both the excluder and the event letter by letter */ | |
468 | while (1) { | |
469 | char e, x; | |
470 | ||
471 | e = *event_ptr; | |
472 | x = *excluder_ptr; | |
473 | ||
474 | if (x == '*') { | |
475 | /* Event is a subset of the excluder */ | |
476 | ERR("Event %s: %.*s excludes all events from %s", | |
477 | event_name, | |
478 | excluder_length, | |
479 | next_excluder, | |
480 | event_name); | |
481 | goto error; | |
482 | } | |
483 | if (e == '*') { | |
5ef79758 MD |
484 | char *string; |
485 | char **new_exclusion_list; | |
486 | ||
748bde76 | 487 | /* Excluder is a proper subset of event */ |
f5436bfc | 488 | string = lttng_strndup(next_excluder, excluder_length); |
5ef79758 | 489 | if (!string) { |
f5436bfc | 490 | PERROR("lttng_strndup error"); |
5ef79758 MD |
491 | goto error; |
492 | } | |
493 | new_exclusion_list = realloc(exclusion_list, | |
1940b29e | 494 | sizeof(char *) * (exclusion_count + 1)); |
5ef79758 MD |
495 | if (!new_exclusion_list) { |
496 | PERROR("realloc"); | |
497 | free(string); | |
498 | goto error; | |
499 | } | |
500 | exclusion_list = new_exclusion_list; | |
748bde76 | 501 | exclusion_count++; |
5ef79758 | 502 | exclusion_list[exclusion_count - 1] = string; |
748bde76 JI |
503 | break; |
504 | } | |
505 | if (x != e) { | |
506 | /* Excluder and event sets have no common elements */ | |
507 | WARN("Event %s: %.*s does not exclude any events from %s", | |
508 | event_name, | |
509 | excluder_length, | |
510 | next_excluder, | |
511 | event_name); | |
512 | break; | |
513 | } | |
514 | excluder_ptr++; | |
515 | event_ptr++; | |
516 | } | |
517 | /* next excluder */ | |
518 | next_excluder += excluder_length; | |
519 | if (*next_excluder == ',') { | |
520 | next_excluder++; | |
521 | } | |
522 | } | |
523 | goto end; | |
524 | error: | |
525 | while (exclusion_count--) { | |
526 | free(exclusion_list[exclusion_count]); | |
527 | } | |
528 | if (exclusion_list != NULL) { | |
529 | free(exclusion_list); | |
530 | } | |
531 | exclusion_list = NULL; | |
532 | exclusion_count = 0; | |
533 | ret = CMD_ERROR; | |
534 | end: | |
535 | *exclusion_count_ptr = exclusion_count; | |
536 | *exclusion_list_ptr = exclusion_list; | |
537 | return ret; | |
538 | } | |
502bbe89 PP |
539 | |
540 | static void warn_on_truncated_exclusion_names(char **exclusion_list, | |
541 | int exclusion_count, int *warn) | |
542 | { | |
543 | size_t i = 0; | |
544 | ||
545 | for (i = 0; i < exclusion_count; ++i) { | |
546 | const char *name = exclusion_list[i]; | |
547 | size_t len = strlen(name); | |
548 | ||
549 | if (len >= LTTNG_SYMBOL_NAME_LEN) { | |
550 | WARN("Event exclusion \"%s\" will be truncated", | |
551 | name); | |
552 | *warn = 1; | |
553 | } | |
554 | } | |
555 | } | |
556 | ||
f3ed775e | 557 | /* |
6181537c | 558 | * Enabling event using the lttng API. |
89476427 | 559 | * Note: in case of error only the last error code will be return. |
f3ed775e | 560 | */ |
cd80958d | 561 | static int enable_events(char *session_name) |
f3ed775e | 562 | { |
89476427 JRJ |
563 | int ret = CMD_SUCCESS, command_ret = CMD_SUCCESS; |
564 | int error_holder = CMD_SUCCESS, warn = 0, error = 0, success = 1; | |
b73d0b29 | 565 | char *event_name, *channel_name = NULL; |
f3ed775e | 566 | struct lttng_event ev; |
7d29a247 | 567 | struct lttng_domain dom; |
7ed70bc9 JI |
568 | int exclusion_count = 0; |
569 | char **exclusion_list = NULL; | |
f3ed775e | 570 | |
441c16a7 MD |
571 | memset(&ev, 0, sizeof(ev)); |
572 | memset(&dom, 0, sizeof(dom)); | |
573 | ||
53a80697 | 574 | if (opt_kernel) { |
67b58630 JG |
575 | if (opt_loglevel) { |
576 | WARN("Kernel loglevels are not supported."); | |
577 | } | |
53a80697 MD |
578 | } |
579 | ||
7d29a247 DG |
580 | /* Create lttng domain */ |
581 | if (opt_kernel) { | |
582 | dom.type = LTTNG_DOMAIN_KERNEL; | |
7972aab2 | 583 | dom.buf_type = LTTNG_BUFFER_GLOBAL; |
d78d6610 | 584 | } else if (opt_userspace) { |
2bdd86d4 | 585 | dom.type = LTTNG_DOMAIN_UST; |
7972aab2 | 586 | /* Default. */ |
8692d4e5 | 587 | dom.buf_type = LTTNG_BUFFER_PER_UID; |
b9dfb167 DG |
588 | } else if (opt_jul) { |
589 | dom.type = LTTNG_DOMAIN_JUL; | |
590 | /* Default. */ | |
591 | dom.buf_type = LTTNG_BUFFER_PER_UID; | |
5cdb6027 DG |
592 | } else if (opt_log4j) { |
593 | dom.type = LTTNG_DOMAIN_LOG4J; | |
594 | /* Default. */ | |
595 | dom.buf_type = LTTNG_BUFFER_PER_UID; | |
0e115563 DG |
596 | } else if (opt_python) { |
597 | dom.type = LTTNG_DOMAIN_PYTHON; | |
598 | /* Default. */ | |
599 | dom.buf_type = LTTNG_BUFFER_PER_UID; | |
6181537c | 600 | } else { |
3ecec76a PP |
601 | /* Checked by the caller. */ |
602 | assert(0); | |
2bdd86d4 | 603 | } |
7d29a247 | 604 | |
8b08c525 AB |
605 | if (opt_exclude) { |
606 | switch (dom.type) { | |
607 | case LTTNG_DOMAIN_KERNEL: | |
608 | case LTTNG_DOMAIN_JUL: | |
609 | case LTTNG_DOMAIN_LOG4J: | |
610 | case LTTNG_DOMAIN_PYTHON: | |
611 | ERR("Event name exclusions are not yet implemented for %s events", | |
612 | get_domain_str(dom.type)); | |
613 | ret = CMD_ERROR; | |
614 | goto error; | |
615 | case LTTNG_DOMAIN_UST: | |
616 | /* Exclusions supported */ | |
617 | break; | |
618 | default: | |
619 | assert(0); | |
620 | } | |
d5dd17fd JI |
621 | } |
622 | ||
85076754 | 623 | channel_name = opt_channel_name; |
ae856491 | 624 | |
cd80958d DG |
625 | handle = lttng_create_handle(session_name, &dom); |
626 | if (handle == NULL) { | |
627 | ret = -1; | |
628 | goto error; | |
629 | } | |
1aef21b6 | 630 | |
89476427 JRJ |
631 | /* Prepare Mi */ |
632 | if (lttng_opt_mi) { | |
633 | /* Open a events element */ | |
634 | ret = mi_lttng_writer_open_element(writer, config_element_events); | |
635 | if (ret) { | |
636 | ret = CMD_ERROR; | |
637 | goto error; | |
638 | } | |
639 | } | |
640 | ||
cd80958d | 641 | if (opt_enable_all) { |
8c9ae521 | 642 | /* Default setup for enable all */ |
75e8c5ab MD |
643 | if (opt_kernel) { |
644 | ev.type = opt_event_type; | |
29c62722 | 645 | strcpy(ev.name, "*"); |
300b8fd5 MD |
646 | /* kernel loglevels not implemented */ |
647 | ev.loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; | |
75e8c5ab MD |
648 | } else { |
649 | ev.type = LTTNG_EVENT_TRACEPOINT; | |
650 | strcpy(ev.name, "*"); | |
300b8fd5 MD |
651 | ev.loglevel_type = opt_loglevel_type; |
652 | if (opt_loglevel) { | |
0e115563 | 653 | assert(opt_userspace || opt_jul || opt_log4j || opt_python); |
b2064f54 DG |
654 | if (opt_userspace) { |
655 | ev.loglevel = loglevel_str_to_value(opt_loglevel); | |
656 | } else if (opt_jul) { | |
657 | ev.loglevel = loglevel_jul_str_to_value(opt_loglevel); | |
5cdb6027 DG |
658 | } else if (opt_log4j) { |
659 | ev.loglevel = loglevel_log4j_str_to_value(opt_loglevel); | |
0e115563 DG |
660 | } else if (opt_python) { |
661 | ev.loglevel = loglevel_python_str_to_value(opt_loglevel); | |
b2064f54 | 662 | } |
300b8fd5 MD |
663 | if (ev.loglevel == -1) { |
664 | ERR("Unknown loglevel %s", opt_loglevel); | |
2f70b271 | 665 | ret = -LTTNG_ERR_INVALID; |
300b8fd5 MD |
666 | goto error; |
667 | } | |
22e25b71 | 668 | } else { |
0e115563 | 669 | assert(opt_userspace || opt_jul || opt_log4j || opt_python); |
b2064f54 DG |
670 | if (opt_userspace) { |
671 | ev.loglevel = -1; | |
34aa3685 | 672 | } else if (opt_jul) { |
b2064f54 | 673 | ev.loglevel = LTTNG_LOGLEVEL_JUL_ALL; |
34aa3685 JG |
674 | } else if (opt_log4j) { |
675 | ev.loglevel = LTTNG_LOGLEVEL_LOG4J_ALL; | |
0e115563 DG |
676 | } else if (opt_python) { |
677 | ev.loglevel = LTTNG_LOGLEVEL_PYTHON_DEBUG; | |
b2064f54 | 678 | } |
300b8fd5 | 679 | } |
75e8c5ab | 680 | } |
8c9ae521 | 681 | |
7ed70bc9 JI |
682 | if (opt_exclude) { |
683 | ret = check_exclusion_subsets("*", opt_exclude, | |
684 | &exclusion_count, &exclusion_list); | |
685 | if (ret == CMD_ERROR) { | |
686 | goto error; | |
687 | } | |
89476427 | 688 | ev.exclusion = 1; |
502bbe89 PP |
689 | |
690 | warn_on_truncated_exclusion_names(exclusion_list, | |
691 | exclusion_count, &warn); | |
7ed70bc9 | 692 | } |
025faf73 | 693 | if (!opt_filter) { |
7ed70bc9 JI |
694 | ret = lttng_enable_event_with_exclusions(handle, |
695 | &ev, channel_name, | |
696 | NULL, | |
697 | exclusion_count, exclusion_list); | |
025faf73 DG |
698 | if (ret < 0) { |
699 | switch (-ret) { | |
700 | case LTTNG_ERR_KERN_EVENT_EXIST: | |
701 | WARN("Kernel events already enabled (channel %s, session %s)", | |
85076754 | 702 | print_channel_name(channel_name), session_name); |
89476427 | 703 | warn = 1; |
025faf73 | 704 | break; |
45d5d421 CB |
705 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
706 | { | |
707 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 708 | ERR("Events: %s (domain %s, channel %s, session %s)", |
45d5d421 | 709 | msg, |
721b94fe | 710 | get_domain_str(dom.type), |
45d5d421 CB |
711 | print_channel_name(channel_name), |
712 | session_name); | |
713 | error = 1; | |
714 | break; | |
715 | } | |
025faf73 | 716 | default: |
721b94fe | 717 | ERR("Events: %s (domain %s, channel %s, session %s)", |
85076754 | 718 | lttng_strerror(ret), |
721b94fe | 719 | get_domain_str(dom.type), |
85076754 MD |
720 | ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
721 | ? print_raw_channel_name(channel_name) | |
722 | : print_channel_name(channel_name), | |
723 | session_name); | |
89476427 | 724 | error = 1; |
025faf73 DG |
725 | break; |
726 | } | |
727 | goto end; | |
42224349 | 728 | } |
8c9ae521 | 729 | |
025faf73 DG |
730 | switch (opt_event_type) { |
731 | case LTTNG_EVENT_TRACEPOINT: | |
67b58630 | 732 | if (opt_loglevel && dom.type != LTTNG_DOMAIN_KERNEL) { |
9c48cab3 | 733 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
734 | |
735 | if (!exclusion_string) { | |
736 | PERROR("Cannot allocate exclusion_string"); | |
737 | error = 1; | |
738 | goto end; | |
739 | } | |
9c48cab3 | 740 | MSG("All %s tracepoints%s are enabled in channel %s for loglevel %s", |
b9dfb167 | 741 | get_domain_str(dom.type), |
9c48cab3 | 742 | exclusion_string, |
85076754 | 743 | print_channel_name(channel_name), |
025faf73 | 744 | opt_loglevel); |
9c48cab3 | 745 | free(exclusion_string); |
025faf73 | 746 | } else { |
9c48cab3 | 747 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
748 | |
749 | if (!exclusion_string) { | |
750 | PERROR("Cannot allocate exclusion_string"); | |
751 | error = 1; | |
752 | goto end; | |
753 | } | |
9c48cab3 | 754 | MSG("All %s tracepoints%s are enabled in channel %s", |
b9dfb167 | 755 | get_domain_str(dom.type), |
9c48cab3 | 756 | exclusion_string, |
85076754 | 757 | print_channel_name(channel_name)); |
9c48cab3 | 758 | free(exclusion_string); |
025faf73 DG |
759 | } |
760 | break; | |
761 | case LTTNG_EVENT_SYSCALL: | |
762 | if (opt_kernel) { | |
6e911cad MD |
763 | MSG("All %s system calls are enabled in channel %s", |
764 | get_domain_str(dom.type), | |
85076754 | 765 | print_channel_name(channel_name)); |
025faf73 DG |
766 | } |
767 | break; | |
768 | case LTTNG_EVENT_ALL: | |
67b58630 | 769 | if (opt_loglevel && dom.type != LTTNG_DOMAIN_KERNEL) { |
9c48cab3 | 770 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
771 | |
772 | if (!exclusion_string) { | |
773 | PERROR("Cannot allocate exclusion_string"); | |
774 | error = 1; | |
775 | goto end; | |
776 | } | |
9c48cab3 | 777 | MSG("All %s events%s are enabled in channel %s for loglevel %s", |
b9dfb167 | 778 | get_domain_str(dom.type), |
9c48cab3 | 779 | exclusion_string, |
85076754 | 780 | print_channel_name(channel_name), |
025faf73 | 781 | opt_loglevel); |
9c48cab3 | 782 | free(exclusion_string); |
025faf73 | 783 | } else { |
9c48cab3 | 784 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
785 | |
786 | if (!exclusion_string) { | |
787 | PERROR("Cannot allocate exclusion_string"); | |
788 | error = 1; | |
789 | goto end; | |
790 | } | |
9c48cab3 | 791 | MSG("All %s events%s are enabled in channel %s", |
b9dfb167 | 792 | get_domain_str(dom.type), |
9c48cab3 | 793 | exclusion_string, |
85076754 | 794 | print_channel_name(channel_name)); |
9c48cab3 | 795 | free(exclusion_string); |
025faf73 DG |
796 | } |
797 | break; | |
798 | default: | |
799 | /* | |
800 | * We should not be here since lttng_enable_event should have | |
801 | * failed on the event type. | |
802 | */ | |
803 | goto error; | |
57064ada | 804 | } |
f3ed775e | 805 | } |
89476427 | 806 | |
025faf73 | 807 | if (opt_filter) { |
89476427 | 808 | command_ret = lttng_enable_event_with_exclusions(handle, &ev, channel_name, |
7ed70bc9 | 809 | opt_filter, exclusion_count, exclusion_list); |
89476427 JRJ |
810 | if (command_ret < 0) { |
811 | switch (-command_ret) { | |
16363652 | 812 | case LTTNG_ERR_FILTER_EXIST: |
85076754 | 813 | WARN("Filter on all events is already enabled" |
721b94fe JR |
814 | " (domain %s, channel %s, session %s)", |
815 | get_domain_str(dom.type), | |
85076754 | 816 | print_channel_name(channel_name), session_name); |
89476427 | 817 | warn = 1; |
16363652 | 818 | break; |
45d5d421 CB |
819 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
820 | { | |
821 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 822 | ERR("All events: %s (domain %s, channel %s, session %s, filter \'%s\')", |
45d5d421 | 823 | msg, |
721b94fe | 824 | get_domain_str(dom.type), |
45d5d421 CB |
825 | print_channel_name(channel_name), |
826 | session_name, opt_filter); | |
827 | error = 1; | |
828 | break; | |
829 | } | |
16363652 | 830 | default: |
721b94fe | 831 | ERR("All events: %s (domain %s, channel %s, session %s, filter \'%s\')", |
da3d7d0e | 832 | lttng_strerror(command_ret), |
721b94fe | 833 | get_domain_str(dom.type), |
da3d7d0e | 834 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
85076754 MD |
835 | ? print_raw_channel_name(channel_name) |
836 | : print_channel_name(channel_name), | |
837 | session_name, opt_filter); | |
89476427 | 838 | error = 1; |
16363652 DG |
839 | break; |
840 | } | |
89476427 | 841 | error_holder = command_ret; |
16363652 | 842 | } else { |
89476427 | 843 | ev.filter = 1; |
16363652 DG |
844 | MSG("Filter '%s' successfully set", opt_filter); |
845 | } | |
846 | } | |
89476427 JRJ |
847 | |
848 | if (lttng_opt_mi) { | |
849 | /* The wildcard * is used for kernel and ust domain to | |
850 | * represent ALL. We copy * in event name to force the wildcard use | |
851 | * for kernel domain | |
852 | * | |
853 | * Note: this is strictly for semantic and printing while in | |
854 | * machine interface mode. | |
855 | */ | |
856 | strcpy(ev.name, "*"); | |
857 | ||
858 | /* If we reach here the events are enabled */ | |
859 | if (!error && !warn) { | |
860 | ev.enabled = 1; | |
861 | } else { | |
862 | ev.enabled = 0; | |
863 | success = 0; | |
864 | } | |
970d848b | 865 | ret = mi_lttng_event(writer, &ev, 1, handle->domain.type); |
89476427 JRJ |
866 | if (ret) { |
867 | ret = CMD_ERROR; | |
868 | goto error; | |
869 | } | |
870 | ||
871 | /* print exclusion */ | |
872 | ret = mi_print_exclusion(exclusion_count, exclusion_list); | |
873 | if (ret) { | |
874 | ret = CMD_ERROR; | |
875 | goto error; | |
876 | } | |
877 | ||
878 | /* Success ? */ | |
879 | ret = mi_lttng_writer_write_element_bool(writer, | |
880 | mi_lttng_element_command_success, success); | |
881 | if (ret) { | |
882 | ret = CMD_ERROR; | |
883 | goto error; | |
884 | } | |
885 | ||
886 | /* Close event element */ | |
887 | ret = mi_lttng_writer_close_element(writer); | |
888 | if (ret) { | |
889 | ret = CMD_ERROR; | |
890 | goto error; | |
891 | } | |
892 | } | |
893 | ||
8c9ae521 | 894 | goto end; |
f3ed775e DG |
895 | } |
896 | ||
897 | /* Strip event list */ | |
898 | event_name = strtok(opt_event_list, ","); | |
899 | while (event_name != NULL) { | |
6181537c DG |
900 | /* Copy name and type of the event */ |
901 | strncpy(ev.name, event_name, LTTNG_SYMBOL_NAME_LEN); | |
902 | ev.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; | |
903 | ev.type = opt_event_type; | |
904 | ||
f3ed775e DG |
905 | /* Kernel tracer action */ |
906 | if (opt_kernel) { | |
907 | DBG("Enabling kernel event %s for channel %s", | |
85076754 MD |
908 | event_name, |
909 | print_channel_name(channel_name)); | |
f3ed775e DG |
910 | |
911 | switch (opt_event_type) { | |
29c62722 MD |
912 | case LTTNG_EVENT_ALL: /* Enable tracepoints and syscalls */ |
913 | /* If event name differs from *, select tracepoint. */ | |
914 | if (strcmp(ev.name, "*")) { | |
915 | ev.type = LTTNG_EVENT_TRACEPOINT; | |
916 | } | |
917 | break; | |
e6ddca71 | 918 | case LTTNG_EVENT_TRACEPOINT: |
f3ed775e | 919 | break; |
7d29a247 | 920 | case LTTNG_EVENT_PROBE: |
cf0e5467 | 921 | ret = parse_probe_opts(&ev, opt_probe); |
49d4e302 | 922 | if (ret) { |
cf0e5467 | 923 | ERR("Unable to parse probe options"); |
0d63dd19 DG |
924 | ret = 0; |
925 | goto error; | |
926 | } | |
f3ed775e DG |
927 | break; |
928 | case LTTNG_EVENT_FUNCTION: | |
8f0d098b | 929 | ret = parse_probe_opts(&ev, opt_function); |
49d4e302 | 930 | if (ret) { |
8f0d098b MD |
931 | ERR("Unable to parse function probe options"); |
932 | ret = 0; | |
933 | goto error; | |
934 | } | |
935 | break; | |
a54bd42d | 936 | case LTTNG_EVENT_SYSCALL: |
c6aa2d41 MD |
937 | ev.type = LTTNG_EVENT_SYSCALL; |
938 | break; | |
f3ed775e | 939 | default: |
1ab1ea0b | 940 | ret = CMD_UNDEFINED; |
f3ed775e DG |
941 | goto error; |
942 | } | |
0cda4f28 | 943 | |
0cda4f28 | 944 | /* kernel loglevels not implemented */ |
8005f29a | 945 | ev.loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; |
f3ed775e | 946 | } else if (opt_userspace) { /* User-space tracer action */ |
300b8fd5 | 947 | DBG("Enabling UST event %s for channel %s, loglevel %s", event_name, |
85076754 | 948 | print_channel_name(channel_name), opt_loglevel ? : "<all>"); |
2bdd86d4 MD |
949 | |
950 | switch (opt_event_type) { | |
951 | case LTTNG_EVENT_ALL: /* Default behavior is tracepoint */ | |
2bdd86d4 MD |
952 | /* Fall-through */ |
953 | case LTTNG_EVENT_TRACEPOINT: | |
e4baff1e MD |
954 | /* Copy name and type of the event */ |
955 | ev.type = LTTNG_EVENT_TRACEPOINT; | |
956 | strncpy(ev.name, event_name, LTTNG_SYMBOL_NAME_LEN); | |
957 | ev.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; | |
2bdd86d4 MD |
958 | break; |
959 | case LTTNG_EVENT_PROBE: | |
960 | case LTTNG_EVENT_FUNCTION: | |
2bdd86d4 MD |
961 | case LTTNG_EVENT_SYSCALL: |
962 | default: | |
cc62c0c0 | 963 | ERR("Event type not available for user-space tracing"); |
4ce78777 | 964 | ret = CMD_UNSUPPORTED; |
2bdd86d4 MD |
965 | goto error; |
966 | } | |
0cda4f28 | 967 | |
7ed70bc9 | 968 | if (opt_exclude) { |
89476427 | 969 | ev.exclusion = 1; |
d5dd17fd JI |
970 | if (opt_event_type != LTTNG_EVENT_ALL && opt_event_type != LTTNG_EVENT_TRACEPOINT) { |
971 | ERR("Exclusion option can only be used with tracepoint events"); | |
972 | ret = CMD_ERROR; | |
973 | goto error; | |
974 | } | |
7ed70bc9 JI |
975 | /* Free previously allocated items */ |
976 | if (exclusion_list != NULL) { | |
977 | while (exclusion_count--) { | |
978 | free(exclusion_list[exclusion_count]); | |
979 | } | |
980 | free(exclusion_list); | |
981 | exclusion_list = NULL; | |
982 | } | |
983 | /* Check for proper subsets */ | |
984 | ret = check_exclusion_subsets(event_name, opt_exclude, | |
985 | &exclusion_count, &exclusion_list); | |
986 | if (ret == CMD_ERROR) { | |
987 | goto error; | |
988 | } | |
502bbe89 PP |
989 | |
990 | warn_on_truncated_exclusion_names( | |
991 | exclusion_list, exclusion_count, &warn); | |
7ed70bc9 JI |
992 | } |
993 | ||
0cda4f28 | 994 | ev.loglevel_type = opt_loglevel_type; |
ed7f4083 | 995 | if (opt_loglevel) { |
8005f29a MD |
996 | ev.loglevel = loglevel_str_to_value(opt_loglevel); |
997 | if (ev.loglevel == -1) { | |
998 | ERR("Unknown loglevel %s", opt_loglevel); | |
2f70b271 | 999 | ret = -LTTNG_ERR_INVALID; |
8005f29a MD |
1000 | goto error; |
1001 | } | |
22e25b71 MD |
1002 | } else { |
1003 | ev.loglevel = -1; | |
ed7f4083 | 1004 | } |
0e115563 | 1005 | } else if (opt_jul || opt_log4j || opt_python) { |
b9dfb167 DG |
1006 | if (opt_event_type != LTTNG_EVENT_ALL && |
1007 | opt_event_type != LTTNG_EVENT_TRACEPOINT) { | |
5cdb6027 | 1008 | ERR("Event type not supported for domain."); |
b9dfb167 DG |
1009 | ret = CMD_UNSUPPORTED; |
1010 | goto error; | |
1011 | } | |
b2064f54 DG |
1012 | |
1013 | ev.loglevel_type = opt_loglevel_type; | |
1014 | if (opt_loglevel) { | |
5cdb6027 DG |
1015 | if (opt_jul) { |
1016 | ev.loglevel = loglevel_jul_str_to_value(opt_loglevel); | |
1017 | } else if (opt_log4j) { | |
1018 | ev.loglevel = loglevel_log4j_str_to_value(opt_loglevel); | |
0e115563 DG |
1019 | } else if (opt_python) { |
1020 | ev.loglevel = loglevel_python_str_to_value(opt_loglevel); | |
5cdb6027 | 1021 | } |
b2064f54 DG |
1022 | if (ev.loglevel == -1) { |
1023 | ERR("Unknown loglevel %s", opt_loglevel); | |
1024 | ret = -LTTNG_ERR_INVALID; | |
1025 | goto error; | |
1026 | } | |
1027 | } else { | |
5cdb6027 DG |
1028 | if (opt_jul) { |
1029 | ev.loglevel = LTTNG_LOGLEVEL_JUL_ALL; | |
1030 | } else if (opt_log4j) { | |
1031 | ev.loglevel = LTTNG_LOGLEVEL_LOG4J_ALL; | |
0e115563 DG |
1032 | } else if (opt_python) { |
1033 | ev.loglevel = LTTNG_LOGLEVEL_PYTHON_DEBUG; | |
5cdb6027 | 1034 | } |
b2064f54 | 1035 | } |
b9dfb167 DG |
1036 | ev.type = LTTNG_EVENT_TRACEPOINT; |
1037 | strncpy(ev.name, event_name, LTTNG_SYMBOL_NAME_LEN); | |
1038 | ev.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; | |
f3ed775e | 1039 | } else { |
3ecec76a | 1040 | assert(0); |
f3ed775e DG |
1041 | } |
1042 | ||
025faf73 | 1043 | if (!opt_filter) { |
9c48cab3 JI |
1044 | char *exclusion_string; |
1045 | ||
89476427 | 1046 | command_ret = lttng_enable_event_with_exclusions(handle, |
7ed70bc9 JI |
1047 | &ev, channel_name, |
1048 | NULL, exclusion_count, exclusion_list); | |
9c48cab3 | 1049 | exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
1050 | if (!exclusion_string) { |
1051 | PERROR("Cannot allocate exclusion_string"); | |
1052 | error = 1; | |
1053 | goto end; | |
1054 | } | |
89476427 | 1055 | if (command_ret < 0) { |
025faf73 | 1056 | /* Turn ret to positive value to handle the positive error code */ |
89476427 | 1057 | switch (-command_ret) { |
025faf73 | 1058 | case LTTNG_ERR_KERN_EVENT_EXIST: |
9c48cab3 | 1059 | WARN("Kernel event %s%s already enabled (channel %s, session %s)", |
85076754 | 1060 | event_name, |
9c48cab3 | 1061 | exclusion_string, |
85076754 | 1062 | print_channel_name(channel_name), session_name); |
89476427 | 1063 | warn = 1; |
025faf73 | 1064 | break; |
45d5d421 CB |
1065 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
1066 | { | |
1067 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 1068 | ERR("Event %s%s: %s (domain %s,channel %s, session %s)", event_name, |
45d5d421 CB |
1069 | exclusion_string, |
1070 | msg, | |
721b94fe | 1071 | get_domain_str(dom.type), |
45d5d421 CB |
1072 | print_channel_name(channel_name), |
1073 | session_name); | |
1074 | error = 1; | |
1075 | break; | |
1076 | } | |
025faf73 | 1077 | default: |
721b94fe | 1078 | ERR("Event %s%s: %s (domain %s, channel %s, session %s)", event_name, |
9c48cab3 | 1079 | exclusion_string, |
da3d7d0e | 1080 | lttng_strerror(command_ret), |
721b94fe | 1081 | get_domain_str(dom.type), |
da3d7d0e | 1082 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
85076754 MD |
1083 | ? print_raw_channel_name(channel_name) |
1084 | : print_channel_name(channel_name), | |
1085 | session_name); | |
89476427 | 1086 | error = 1; |
025faf73 DG |
1087 | break; |
1088 | } | |
89476427 | 1089 | error_holder = command_ret; |
025faf73 | 1090 | } else { |
8274eeba AB |
1091 | switch (dom.type) { |
1092 | case LTTNG_DOMAIN_KERNEL: | |
1093 | case LTTNG_DOMAIN_UST: | |
49ceaa70 | 1094 | MSG("%s event %s%s created in channel %s", |
8274eeba AB |
1095 | get_domain_str(dom.type), |
1096 | event_name, | |
1097 | exclusion_string, | |
1098 | print_channel_name(channel_name)); | |
1099 | break; | |
1100 | case LTTNG_DOMAIN_JUL: | |
1101 | case LTTNG_DOMAIN_LOG4J: | |
1102 | case LTTNG_DOMAIN_PYTHON: | |
1103 | /* | |
1104 | * Don't print the default channel | |
1105 | * name for agent domains. | |
1106 | */ | |
895707da | 1107 | MSG("%s event %s%s enabled", |
8274eeba AB |
1108 | get_domain_str(dom.type), |
1109 | event_name, | |
1110 | exclusion_string); | |
1111 | break; | |
1112 | default: | |
1113 | assert(0); | |
49ceaa70 | 1114 | } |
42224349 | 1115 | } |
9c48cab3 | 1116 | free(exclusion_string); |
6181537c | 1117 | } |
025faf73 DG |
1118 | |
1119 | if (opt_filter) { | |
9c48cab3 JI |
1120 | char *exclusion_string; |
1121 | ||
89476427 JRJ |
1122 | /* Filter present */ |
1123 | ev.filter = 1; | |
1124 | ||
1125 | command_ret = lttng_enable_event_with_exclusions(handle, &ev, channel_name, | |
7ed70bc9 | 1126 | opt_filter, exclusion_count, exclusion_list); |
9c48cab3 | 1127 | exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
1128 | if (!exclusion_string) { |
1129 | PERROR("Cannot allocate exclusion_string"); | |
1130 | error = 1; | |
1131 | goto end; | |
1132 | } | |
89476427 JRJ |
1133 | if (command_ret < 0) { |
1134 | switch (-command_ret) { | |
7671f53c | 1135 | case LTTNG_ERR_FILTER_EXIST: |
9c48cab3 | 1136 | WARN("Filter on event %s%s is already enabled" |
721b94fe | 1137 | " (domain %s, channel %s, session %s)", |
85076754 | 1138 | event_name, |
9c48cab3 | 1139 | exclusion_string, |
721b94fe | 1140 | get_domain_str(dom.type), |
85076754 | 1141 | print_channel_name(channel_name), session_name); |
89476427 | 1142 | warn = 1; |
7671f53c | 1143 | break; |
45d5d421 CB |
1144 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
1145 | { | |
1146 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 1147 | ERR("Event %s%s: %s (domain %s, channel %s, session %s, filter \'%s\')", ev.name, |
45d5d421 CB |
1148 | exclusion_string, |
1149 | msg, | |
721b94fe | 1150 | get_domain_str(dom.type), |
45d5d421 CB |
1151 | print_channel_name(channel_name), |
1152 | session_name, opt_filter); | |
1153 | error = 1; | |
1154 | break; | |
1155 | } | |
7671f53c | 1156 | default: |
721b94fe | 1157 | ERR("Event %s%s: %s (domain %s, channel %s, session %s, filter \'%s\')", ev.name, |
9c48cab3 | 1158 | exclusion_string, |
da3d7d0e | 1159 | lttng_strerror(command_ret), |
721b94fe | 1160 | get_domain_str(dom.type), |
da3d7d0e | 1161 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
85076754 MD |
1162 | ? print_raw_channel_name(channel_name) |
1163 | : print_channel_name(channel_name), | |
1164 | session_name, opt_filter); | |
89476427 | 1165 | error = 1; |
7671f53c CB |
1166 | break; |
1167 | } | |
89476427 JRJ |
1168 | error_holder = command_ret; |
1169 | ||
16363652 | 1170 | } else { |
721b94fe | 1171 | MSG("Event %s%s: Filter '%s' for domain %s successfully set", |
9c48cab3 | 1172 | event_name, exclusion_string, |
721b94fe JR |
1173 | opt_filter, |
1174 | get_domain_str(dom.type)); | |
53a80697 | 1175 | } |
9c48cab3 | 1176 | free(exclusion_string); |
53a80697 | 1177 | } |
6181537c | 1178 | |
89476427 JRJ |
1179 | if (lttng_opt_mi) { |
1180 | if (command_ret) { | |
1181 | success = 0; | |
1182 | ev.enabled = 0; | |
1183 | } else { | |
1184 | ev.enabled = 1; | |
1185 | } | |
1186 | ||
970d848b | 1187 | ret = mi_lttng_event(writer, &ev, 1, handle->domain.type); |
89476427 JRJ |
1188 | if (ret) { |
1189 | ret = CMD_ERROR; | |
1190 | goto error; | |
1191 | } | |
1192 | ||
1193 | /* print exclusion */ | |
1194 | ret = mi_print_exclusion(exclusion_count, exclusion_list); | |
1195 | if (ret) { | |
1196 | ret = CMD_ERROR; | |
1197 | goto error; | |
1198 | } | |
1199 | ||
1200 | /* Success ? */ | |
1201 | ret = mi_lttng_writer_write_element_bool(writer, | |
1202 | mi_lttng_element_command_success, success); | |
1203 | if (ret) { | |
1204 | ret = CMD_ERROR; | |
1205 | goto end; | |
1206 | } | |
1207 | ||
1208 | /* Close event element */ | |
1209 | ret = mi_lttng_writer_close_element(writer); | |
1210 | if (ret) { | |
1211 | ret = CMD_ERROR; | |
1212 | goto end; | |
1213 | } | |
1214 | } | |
1215 | ||
f3ed775e DG |
1216 | /* Next event */ |
1217 | event_name = strtok(NULL, ","); | |
89476427 JRJ |
1218 | /* Reset warn, error and success */ |
1219 | success = 1; | |
f3ed775e DG |
1220 | } |
1221 | ||
8c9ae521 | 1222 | end: |
89476427 JRJ |
1223 | /* Close Mi */ |
1224 | if (lttng_opt_mi) { | |
1225 | /* Close events element */ | |
1226 | ret = mi_lttng_writer_close_element(writer); | |
1227 | if (ret) { | |
1228 | ret = CMD_ERROR; | |
1229 | goto error; | |
1230 | } | |
1231 | } | |
f3ed775e | 1232 | error: |
ae856491 DG |
1233 | if (warn) { |
1234 | ret = CMD_WARNING; | |
1235 | } | |
89476427 JRJ |
1236 | if (error) { |
1237 | ret = CMD_ERROR; | |
1238 | } | |
cd80958d DG |
1239 | lttng_destroy_handle(handle); |
1240 | ||
7ed70bc9 JI |
1241 | if (exclusion_list != NULL) { |
1242 | while (exclusion_count--) { | |
1243 | free(exclusion_list[exclusion_count]); | |
1244 | } | |
1245 | free(exclusion_list); | |
1246 | } | |
1247 | ||
89476427 JRJ |
1248 | /* Overwrite ret with error_holder if there was an actual error with |
1249 | * enabling an event. | |
1250 | */ | |
1251 | ret = error_holder ? error_holder : ret; | |
1252 | ||
f3ed775e DG |
1253 | return ret; |
1254 | } | |
1255 | ||
1256 | /* | |
6181537c | 1257 | * Add event to trace session |
f3ed775e | 1258 | */ |
721b94fe JR |
1259 | |
1260 | struct args_tuple { | |
1261 | int argv_index_start; | |
1262 | int argv_index_end; | |
1263 | }; | |
1264 | ||
1265 | struct domain_configuration { | |
1266 | int enable_all; | |
1267 | int event_type; | |
1268 | char *event_list; | |
1269 | char *loglevel; | |
1270 | int loglevel_type; | |
1271 | enum lttng_domain_type domain_type; | |
1272 | char *probe; | |
1273 | char *function; | |
1274 | char *channel_name; | |
1275 | char *filter; | |
1276 | char *exclude; | |
1277 | }; | |
1278 | ||
1279 | static struct domain_configuration *initialize_domain_configuration(enum lttng_domain_type type) | |
f3ed775e | 1280 | { |
f3ed775e | 1281 | |
721b94fe | 1282 | struct domain_configuration *config = malloc(sizeof(struct domain_configuration)); |
f3ed775e | 1283 | |
721b94fe JR |
1284 | if (!config) { |
1285 | goto end; | |
1286 | } | |
f3ed775e | 1287 | |
721b94fe JR |
1288 | switch(type) { |
1289 | case LTTNG_DOMAIN_KERNEL: | |
1290 | config->domain_type = LTTNG_DOMAIN_KERNEL; | |
eeac7d46 | 1291 | break; |
721b94fe JR |
1292 | case LTTNG_DOMAIN_UST: |
1293 | config->domain_type = LTTNG_DOMAIN_UST; | |
0cda4f28 | 1294 | break; |
721b94fe JR |
1295 | case LTTNG_DOMAIN_JUL: |
1296 | config->domain_type = LTTNG_DOMAIN_JUL; | |
13dce3b7 | 1297 | break; |
721b94fe JR |
1298 | case LTTNG_DOMAIN_LOG4J: |
1299 | config->domain_type = LTTNG_DOMAIN_LOG4J; | |
53a80697 | 1300 | break; |
721b94fe JR |
1301 | case LTTNG_DOMAIN_PYTHON: |
1302 | config->domain_type = LTTNG_DOMAIN_PYTHON; | |
fac3366c | 1303 | break; |
721b94fe | 1304 | case LTTNG_DOMAIN_NONE: |
f3ed775e | 1305 | default: |
721b94fe JR |
1306 | free(config); |
1307 | config=NULL; | |
f3ed775e | 1308 | goto end; |
721b94fe JR |
1309 | }; |
1310 | ||
1311 | config->event_type = LTTNG_EVENT_ALL ; | |
1312 | config->enable_all = 0; | |
1313 | config->loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; | |
1314 | config->loglevel = NULL; | |
1315 | config->probe = NULL; | |
1316 | config->function = NULL; | |
1317 | config->channel_name = NULL; | |
1318 | config->filter = NULL; | |
1319 | config->exclude = NULL; | |
1320 | config->event_list = NULL; | |
1321 | end: | |
1322 | return config; | |
1323 | } | |
de044b7a | 1324 | |
721b94fe JR |
1325 | int cmd_enable_events(int argc, const char **argv) |
1326 | { | |
1327 | int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1; | |
1328 | static poptContext pc; | |
1329 | char *session_name = NULL; | |
1330 | int event_type = -1; | |
f3ed775e | 1331 | |
721b94fe JR |
1332 | struct domain_configuration *userspace_config = NULL; |
1333 | struct domain_configuration *kernel_config = NULL; | |
1334 | struct domain_configuration *jul_config = NULL; | |
1335 | struct domain_configuration *log4j_config = NULL; | |
1336 | struct domain_configuration *python_config = NULL; | |
1337 | struct domain_configuration *domain_configuration[5] = { userspace_config, kernel_config, jul_config, log4j_config, python_config} | |
3ecec76a | 1338 | |
89476427 JRJ |
1339 | /* Mi check */ |
1340 | if (lttng_opt_mi) { | |
1341 | writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi); | |
1342 | if (!writer) { | |
1343 | ret = -LTTNG_ERR_NOMEM; | |
1344 | goto end; | |
1345 | } | |
1346 | ||
1347 | /* Open command element */ | |
1348 | ret = mi_lttng_writer_command_open(writer, | |
1349 | mi_lttng_element_command_enable_event); | |
1350 | if (ret) { | |
1351 | ret = CMD_ERROR; | |
1352 | goto end; | |
1353 | } | |
1354 | ||
1355 | /* Open output element */ | |
1356 | ret = mi_lttng_writer_open_element(writer, | |
1357 | mi_lttng_element_command_output); | |
1358 | if (ret) { | |
1359 | ret = CMD_ERROR; | |
1360 | goto end; | |
1361 | } | |
1362 | } | |
1363 | ||
721b94fe JR |
1364 | /* Parse global arguments */ |
1365 | pc = poptGetContext(NULL, argc, argv, global_long_options, 0); | |
1366 | poptReadDefaultConfig(pc, 0); | |
1367 | ||
1368 | ||
1369 | while ((opt = poptGetNextOpt(pc)) != -1) { | |
1370 | switch (opt) { | |
1371 | case OPT_HELP: | |
1372 | SHOW_HELP(); | |
1373 | goto end; | |
1374 | case OPT_LIST_OPTIONS: | |
1375 | list_cmd_options(stdout, long_options); | |
1376 | goto end; | |
1377 | default: | |
1378 | break; | |
1379 | } | |
f3ed775e DG |
1380 | } |
1381 | ||
721b94fe JR |
1382 | /* Dispose the global arguments context */ |
1383 | poptFreeContext(pc); | |
1384 | pc = NULL; | |
1385 | ||
1386 | /* Find the number of domain based on the passed arguments */ | |
1387 | int i; | |
1388 | int args_tuple_count = 0; | |
1389 | int arg_state_looking_for_end = 0; | |
1390 | struct args_tuple *args_tuple_list = NULL; | |
1391 | for (i = 1; i < argc ; i++) { | |
1392 | ||
1393 | if (strcmp("-u", argv[i]) && strcmp("--userspace", argv[i]) && | |
1394 | strcmp("-j", argv[i]) && strcmp("--jul", argv[i]) && | |
1395 | strcmp("-l", argv[i]) && strcmp("--log4j", argv[i]) && | |
1396 | strcmp("-p", argv[i]) && strcmp("--python", argv[i]) && | |
1397 | strcmp("-k", argv[i]) && strcmp("--kernel", argv[i])) { | |
1398 | continue; | |
1399 | } | |
1400 | ||
1401 | ||
1402 | struct args_tuple *tmp_pointer = NULL; | |
1403 | args_tuple_count++; | |
1404 | tmp_pointer = realloc(args_tuple_list, sizeof(struct args_tuple) * args_tuple_count); | |
1405 | if (!tmp_pointer) { | |
1406 | ERR("Realoc of args tuple failed"); | |
1407 | ret = CMD_ERROR; | |
1408 | goto end; | |
1409 | } | |
1410 | args_tuple_list = tmp_pointer; | |
1411 | ||
1412 | if (!arg_state_looking_for_end) { | |
1413 | if (args_tuple_count -1 < 0) { | |
1414 | ERR("Args parsing illegal state"); | |
1415 | ret = CMD_ERROR; | |
1416 | goto end; | |
1417 | } | |
1418 | args_tuple_list[args_tuple_count-1].argv_index_start = i; | |
1419 | arg_state_looking_for_end = 1; | |
1420 | } else { | |
1421 | if (args_tuple_count - 2 < 0 || args_tuple_count -1 < 0) { | |
1422 | ERR("Args parsing illegal state"); | |
1423 | ret = CMD_ERROR; | |
1424 | goto end; | |
1425 | } | |
1426 | ||
1427 | /* Close the previous tuple */ | |
1428 | args_tuple_list[args_tuple_count-2].argv_index_end = i - 1; | |
1429 | ||
1430 | /* Start the new tuple */ | |
1431 | args_tuple_list[args_tuple_count-1].argv_index_start = i; | |
cd80958d | 1432 | } |
cd80958d DG |
1433 | } |
1434 | ||
721b94fe JR |
1435 | if (args_tuple_count <= 0) { |
1436 | ret = print_missing_or_multiple_domains(0); | |
1437 | if (ret) { | |
1438 | ret = CMD_ERROR; | |
1439 | goto end; | |
1440 | } | |
1441 | goto end; | |
89476427 JRJ |
1442 | } |
1443 | ||
721b94fe JR |
1444 | /* Close the last tuple */ |
1445 | args_tuple_list[args_tuple_count-1].argv_index_end = i - 1; | |
1446 | ||
1447 | if (args_tuple_count == 1) { | |
1448 | /* Preserve the old way with a domain flag that can be anywhere */ | |
1449 | args_tuple_list[0].argv_index_start = 1; | |
1450 | } | |
1451 | ||
1452 | for (i = 0; i < args_tuple_count; i++) { | |
1453 | struct args_tuple *tuple = &args_tuple_list[i]; | |
1454 | int cur_argc = tuple->argv_index_end - tuple-> argv_index_start + 1; | |
1455 | const char **cur_argv = &argv[tuple->argv_index_start]; | |
1456 | ||
1457 | /* Default options */ | |
1458 | ||
1459 | /* Domain */ | |
1460 | opt_domain = LTTNG_DOMAIN_NONE; | |
1461 | ||
1462 | /* Event options */ | |
1463 | opt_enable_all = 0; | |
1464 | opt_event_type = LTTNG_EVENT_ALL; | |
1465 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; | |
1466 | opt_event_list = NULL; | |
1467 | opt_loglevel = NULL; | |
1468 | opt_probe = NULL; | |
1469 | opt_function = NULL; | |
1470 | opt_channel_name = NULL; | |
1471 | opt_filter = NULL; | |
1472 | opt_exclude = NULL; | |
1473 | ||
1474 | pc = poptGetContext(NULL, cur_argc, cur_argv, long_options, POPT_CONTEXT_KEEP_FIRST); | |
1475 | poptReadDefaultConfig(pc, 0); | |
1476 | ||
1477 | /* Default event type */ | |
1478 | opt_event_type = LTTNG_EVENT_ALL; | |
1479 | ||
1480 | while ((opt = poptGetNextOpt(pc)) != -1) { | |
1481 | switch (opt) { | |
1482 | case OPT_HELP: | |
1483 | SHOW_HELP(); | |
1484 | goto end; | |
1485 | case OPT_TRACEPOINT: | |
1486 | opt_event_type = LTTNG_EVENT_TRACEPOINT; | |
1487 | break; | |
1488 | case OPT_PROBE: | |
1489 | opt_event_type = LTTNG_EVENT_PROBE; | |
1490 | break; | |
1491 | case OPT_FUNCTION: | |
1492 | opt_event_type = LTTNG_EVENT_FUNCTION; | |
1493 | break; | |
1494 | case OPT_SYSCALL: | |
1495 | opt_event_type = LTTNG_EVENT_SYSCALL; | |
1496 | break; | |
1497 | case OPT_LOGLEVEL: | |
1498 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_RANGE; | |
1499 | opt_loglevel = poptGetOptArg(pc); | |
1500 | break; | |
1501 | case OPT_LOGLEVEL_ONLY: | |
1502 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_SINGLE; | |
1503 | opt_loglevel = poptGetOptArg(pc); | |
1504 | break; | |
1505 | case OPT_LIST_OPTIONS: | |
1506 | list_cmd_options(stdout, long_options); | |
1507 | goto end; | |
1508 | case OPT_FILTER: | |
1509 | break; | |
1510 | case OPT_EXCLUDE: | |
1511 | break; | |
1512 | default: | |
1513 | ret = CMD_UNDEFINED; | |
1514 | goto end; | |
1515 | } | |
1516 | ||
1517 | /* Validate event type. Multiple event type are not supported. */ | |
1518 | if (event_type == -1) { | |
1519 | event_type = opt_event_type; | |
1520 | } else { | |
1521 | if (event_type != opt_event_type) { | |
1522 | ERR("Multiple event type not supported."); | |
1523 | ret = CMD_ERROR; | |
1524 | goto end; | |
1525 | } | |
1526 | } | |
1527 | } | |
1528 | ||
1529 | struct domain_configuration *tmp_config = initialize_domain_configuration(opt_domain); | |
1530 | ||
1531 | opt_event_list = (char*) poptGetArg(pc); | |
1532 | if (opt_event_list == NULL && opt_enable_all == 0) { | |
1533 | ERR("Missing event name(s).\n"); | |
1534 | ret = CMD_ERROR; | |
1535 | goto end; | |
1536 | } | |
1537 | ||
1538 | if (!opt_session_name) { | |
1539 | session_name = get_session_name(); | |
1540 | if (session_name == NULL) { | |
1541 | command_ret = CMD_ERROR; | |
1542 | success = 0; | |
1543 | goto mi_closing; | |
1544 | } | |
1545 | } else { | |
1546 | session_name = opt_session_name; | |
1547 | } | |
1548 | ||
1549 | config->event_type = opt_event_type ; | |
1550 | config->enable_all = opt_enable_all; | |
1551 | config->loglevel_type = opt_loglevel_type; | |
1552 | config->loglevel = opt_loglevel; | |
1553 | config->probe = opt_probe; | |
1554 | config->function = opt_function; | |
1555 | config->channel_name = opt_channel_name; | |
1556 | config->filter = opt_filter; | |
1557 | config->exclude = opt_exclude; | |
1558 | config->event_list = opt_event_list; | |
1559 | ||
1560 | switch(opt_domain) { | |
1561 | case LTTNG_DOMAIN_KERNEL: | |
1562 | if (kernel_config) { | |
1563 | ERR("Only one -k option is permitted per command"); | |
1564 | ret = CMD_ERROR; | |
1565 | goto end; | |
1566 | } | |
1567 | kernel_config = tmp_config; | |
1568 | break; | |
1569 | case LTTNG_DOMAIN_UST: | |
1570 | if (ust_config) { | |
1571 | ERR("Only one -u option is permitted per command"); | |
1572 | ret = CMD_ERROR; | |
1573 | goto end; | |
1574 | } | |
1575 | ust_config = tmp_config; | |
1576 | break; | |
1577 | case LTTNG_DOMAIN_JUL: | |
1578 | if (jul_config) { | |
1579 | ERR("Only one -j option is permitted per command"); | |
1580 | ret = CMD_ERROR; | |
1581 | goto end; | |
1582 | } | |
1583 | jul_config = tmp_config; | |
1584 | break; | |
1585 | case LTTNG_DOMAIN_LOG4J: | |
1586 | if (log4j_config) { | |
1587 | ERR("Only one -l option is permitted per command"); | |
1588 | ret = CMD_ERROR; | |
1589 | goto end; | |
1590 | } | |
1591 | log4j_config = tmp_config; | |
1592 | break; | |
1593 | case LTTNG_DOMAIN_PYTHON: | |
1594 | if (python_config) { | |
1595 | ERR("Only one -p option is permitted per command"); | |
1596 | ret = CMD_ERROR; | |
1597 | goto end; | |
1598 | } | |
1599 | python_config = tmp_config; | |
1600 | break; | |
1601 | case LTTNG_DOMAIN_NONE: | |
1602 | default: | |
1603 | ret = CMD_ERROR; | |
1604 | goto end | |
1605 | }; | |
1606 | ||
1607 | tmp_config = NULL; | |
1608 | ||
1609 | } | |
1610 | ||
1611 | command_ret = enable_events_1(session_name, configurations, sizeof(configurations)/sizeof(struct *domain_configuration); | |
1612 | if (command_ret) { | |
1613 | success = 0; | |
1614 | } | |
1615 | ||
89476427 JRJ |
1616 | mi_closing: |
1617 | /* Mi closing */ | |
1618 | if (lttng_opt_mi) { | |
1619 | /* Close output element */ | |
1620 | ret = mi_lttng_writer_close_element(writer); | |
1621 | if (ret) { | |
1622 | ret = CMD_ERROR; | |
1623 | goto end; | |
1624 | } | |
1625 | ||
1626 | ret = mi_lttng_writer_write_element_bool(writer, | |
1627 | mi_lttng_element_command_success, success); | |
1628 | if (ret) { | |
1629 | ret = CMD_ERROR; | |
1630 | goto end; | |
1631 | } | |
1632 | ||
1633 | /* Command element close */ | |
1634 | ret = mi_lttng_writer_command_close(writer); | |
1635 | if (ret) { | |
1636 | ret = CMD_ERROR; | |
1637 | goto end; | |
1638 | } | |
1639 | } | |
f3ed775e DG |
1640 | |
1641 | end: | |
89476427 JRJ |
1642 | /* Mi clean-up */ |
1643 | if (writer && mi_lttng_writer_destroy(writer)) { | |
1644 | /* Preserve original error code */ | |
1645 | ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL; | |
1646 | } | |
1647 | ||
cd80958d DG |
1648 | if (opt_session_name == NULL) { |
1649 | free(session_name); | |
1650 | } | |
1651 | ||
89476427 JRJ |
1652 | /* Overwrite ret if an error occurred in enable_events */ |
1653 | ret = command_ret ? command_ret : ret; | |
1654 | ||
721b94fe JR |
1655 | free(userspace_config); |
1656 | free(kernel_config); | |
1657 | free(jul_config); | |
1658 | free(log4j_config); | |
1659 | free(python_config); | |
ca1c3607 | 1660 | poptFreeContext(pc); |
f3ed775e DG |
1661 | return ret; |
1662 | } |