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; | |
c7dfbc0e | 43 | static char *opt_loglevel; |
0cda4f28 | 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 | |
1036985c | 301 | int loglevel_ust_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 | ||
1036985c JR |
354 | /* |
355 | * Map a userspace agent loglevel to it's value based on the domain type. | |
356 | * | |
357 | * Assert when loglevel is NULL and domain type is LTTNG_DOMAIN_NONE || | |
358 | * LTTNG_DOMAIN_KERNEL. | |
359 | * | |
360 | * return -1 on invalid loglevel. | |
361 | */ | |
362 | static int loglevel_str_to_value(const char* loglevel, enum lttng_domain_type type) | |
363 | { | |
364 | int ret = -1; | |
365 | switch (type) { | |
366 | case LTTNG_DOMAIN_UST: | |
367 | ret = loglevel_ust_str_to_value(loglevel); | |
368 | break; | |
369 | case LTTNG_DOMAIN_JUL: | |
370 | ret = loglevel_jul_str_to_value(loglevel); | |
371 | break; | |
372 | case LTTNG_DOMAIN_LOG4J: | |
373 | ret = loglevel_log4j_str_to_value(loglevel); | |
374 | break; | |
375 | case LTTNG_DOMAIN_PYTHON: | |
376 | ret = loglevel_python_str_to_value(loglevel); | |
377 | break; | |
378 | default: | |
379 | assert(0); | |
380 | } | |
381 | ||
382 | return ret; | |
383 | } | |
384 | ||
85076754 MD |
385 | static |
386 | const char *print_channel_name(const char *name) | |
387 | { | |
388 | return name ? : DEFAULT_CHANNEL_NAME; | |
389 | } | |
390 | ||
391 | static | |
392 | const char *print_raw_channel_name(const char *name) | |
393 | { | |
394 | return name ? : "<default>"; | |
395 | } | |
396 | ||
89476427 JRJ |
397 | /* |
398 | * Mi print exlcusion list | |
399 | */ | |
400 | static | |
401 | int mi_print_exclusion(int count, char **names) | |
402 | { | |
403 | int i, ret; | |
404 | ||
405 | assert(writer); | |
406 | ||
407 | if (count == 0) { | |
408 | ret = 0; | |
409 | goto end; | |
410 | } | |
411 | ret = mi_lttng_writer_open_element(writer, config_element_exclusions); | |
412 | if (ret) { | |
413 | goto end; | |
414 | } | |
415 | ||
416 | for (i = 0; i < count; i++) { | |
417 | ret = mi_lttng_writer_write_element_string(writer, | |
418 | config_element_exclusion, names[i]); | |
419 | if (ret) { | |
420 | goto end; | |
421 | } | |
422 | } | |
423 | ||
424 | /* Close exclusions element */ | |
425 | ret = mi_lttng_writer_close_element(writer); | |
426 | ||
427 | end: | |
428 | return ret; | |
429 | } | |
430 | ||
9c48cab3 JI |
431 | /* |
432 | * Return allocated string for pretty-printing exclusion names. | |
433 | */ | |
434 | static | |
435 | char *print_exclusions(int count, char **names) | |
436 | { | |
437 | int length = 0; | |
438 | int i; | |
439 | const char *preamble = " excluding "; | |
440 | char *ret; | |
441 | ||
442 | if (count == 0) { | |
443 | return strdup(""); | |
444 | } | |
445 | ||
446 | /* calculate total required length */ | |
447 | for (i = 0; i < count; i++) { | |
448 | length += strlen(names[i]) + 1; | |
449 | } | |
450 | ||
451 | /* add length of preamble + one for NUL - one for last (missing) comma */ | |
452 | length += strlen(preamble); | |
fd591b18 MD |
453 | ret = zmalloc(length); |
454 | if (!ret) { | |
455 | return NULL; | |
456 | } | |
9c48cab3 JI |
457 | strncpy(ret, preamble, length); |
458 | for (i = 0; i < count; i++) { | |
459 | strcat(ret, names[i]); | |
460 | if (i != count - 1) { | |
461 | strcat(ret, ","); | |
462 | } | |
463 | } | |
89476427 | 464 | |
9c48cab3 JI |
465 | return ret; |
466 | } | |
467 | ||
748bde76 JI |
468 | /* |
469 | * Compare list of exclusions against an event name. | |
470 | * Return a list of legal exclusion names. | |
471 | * Produce an error or a warning about others (depending on the situation) | |
472 | */ | |
473 | static | |
474 | int check_exclusion_subsets(const char *event_name, | |
475 | const char *exclusions, | |
476 | int *exclusion_count_ptr, | |
477 | char ***exclusion_list_ptr) | |
478 | { | |
479 | const char *excluder_ptr; | |
480 | const char *event_ptr; | |
481 | const char *next_excluder; | |
482 | int excluder_length; | |
483 | int exclusion_count = 0; | |
484 | char **exclusion_list = NULL; | |
485 | int ret = CMD_SUCCESS; | |
486 | ||
487 | if (event_name[strlen(event_name) - 1] != '*') { | |
488 | ERR("Event %s: Excluders can only be used with wildcarded events", event_name); | |
489 | goto error; | |
490 | } | |
491 | ||
492 | next_excluder = exclusions; | |
493 | while (*next_excluder != 0) { | |
494 | event_ptr = event_name; | |
495 | excluder_ptr = next_excluder; | |
496 | excluder_length = strcspn(next_excluder, ","); | |
497 | ||
498 | /* Scan both the excluder and the event letter by letter */ | |
499 | while (1) { | |
500 | char e, x; | |
501 | ||
502 | e = *event_ptr; | |
503 | x = *excluder_ptr; | |
504 | ||
505 | if (x == '*') { | |
506 | /* Event is a subset of the excluder */ | |
507 | ERR("Event %s: %.*s excludes all events from %s", | |
508 | event_name, | |
509 | excluder_length, | |
510 | next_excluder, | |
511 | event_name); | |
512 | goto error; | |
513 | } | |
514 | if (e == '*') { | |
5ef79758 MD |
515 | char *string; |
516 | char **new_exclusion_list; | |
517 | ||
748bde76 | 518 | /* Excluder is a proper subset of event */ |
f5436bfc | 519 | string = lttng_strndup(next_excluder, excluder_length); |
5ef79758 | 520 | if (!string) { |
f5436bfc | 521 | PERROR("lttng_strndup error"); |
5ef79758 MD |
522 | goto error; |
523 | } | |
524 | new_exclusion_list = realloc(exclusion_list, | |
1940b29e | 525 | sizeof(char *) * (exclusion_count + 1)); |
5ef79758 MD |
526 | if (!new_exclusion_list) { |
527 | PERROR("realloc"); | |
528 | free(string); | |
529 | goto error; | |
530 | } | |
531 | exclusion_list = new_exclusion_list; | |
748bde76 | 532 | exclusion_count++; |
5ef79758 | 533 | exclusion_list[exclusion_count - 1] = string; |
748bde76 JI |
534 | break; |
535 | } | |
536 | if (x != e) { | |
537 | /* Excluder and event sets have no common elements */ | |
538 | WARN("Event %s: %.*s does not exclude any events from %s", | |
539 | event_name, | |
540 | excluder_length, | |
541 | next_excluder, | |
542 | event_name); | |
543 | break; | |
544 | } | |
545 | excluder_ptr++; | |
546 | event_ptr++; | |
547 | } | |
548 | /* next excluder */ | |
549 | next_excluder += excluder_length; | |
550 | if (*next_excluder == ',') { | |
551 | next_excluder++; | |
552 | } | |
553 | } | |
554 | goto end; | |
555 | error: | |
556 | while (exclusion_count--) { | |
557 | free(exclusion_list[exclusion_count]); | |
558 | } | |
559 | if (exclusion_list != NULL) { | |
560 | free(exclusion_list); | |
561 | } | |
562 | exclusion_list = NULL; | |
563 | exclusion_count = 0; | |
564 | ret = CMD_ERROR; | |
565 | end: | |
566 | *exclusion_count_ptr = exclusion_count; | |
567 | *exclusion_list_ptr = exclusion_list; | |
568 | return ret; | |
569 | } | |
502bbe89 PP |
570 | |
571 | static void warn_on_truncated_exclusion_names(char **exclusion_list, | |
572 | int exclusion_count, int *warn) | |
573 | { | |
574 | size_t i = 0; | |
575 | ||
576 | for (i = 0; i < exclusion_count; ++i) { | |
577 | const char *name = exclusion_list[i]; | |
578 | size_t len = strlen(name); | |
579 | ||
580 | if (len >= LTTNG_SYMBOL_NAME_LEN) { | |
581 | WARN("Event exclusion \"%s\" will be truncated", | |
582 | name); | |
583 | *warn = 1; | |
584 | } | |
585 | } | |
586 | } | |
587 | ||
c7dfbc0e JR |
588 | struct domain_configuration { |
589 | int enable_all; | |
590 | int event_type; | |
591 | char *event_list; | |
592 | char *loglevel; | |
593 | int loglevel_type; | |
594 | enum lttng_domain_type domain_type; | |
595 | char *probe; | |
596 | char *function; | |
597 | char *channel_name; | |
598 | char *filter; | |
599 | char *exclude; | |
600 | }; | |
601 | ||
f3ed775e | 602 | /* |
6181537c | 603 | * Enabling event using the lttng API. |
89476427 | 604 | * Note: in case of error only the last error code will be return. |
f3ed775e | 605 | */ |
c7dfbc0e | 606 | static int enable_events(char *session_name, struct domain_configuration *config) |
f3ed775e | 607 | { |
89476427 JRJ |
608 | int ret = CMD_SUCCESS, command_ret = CMD_SUCCESS; |
609 | int error_holder = CMD_SUCCESS, warn = 0, error = 0, success = 1; | |
b73d0b29 | 610 | char *event_name, *channel_name = NULL; |
f3ed775e | 611 | struct lttng_event ev; |
7d29a247 | 612 | struct lttng_domain dom; |
7ed70bc9 JI |
613 | int exclusion_count = 0; |
614 | char **exclusion_list = NULL; | |
f3ed775e | 615 | |
c7dfbc0e JR |
616 | int config_enable_all; |
617 | int config_event_type; | |
618 | char *config_event_list; | |
619 | char *config_loglevel; | |
620 | int config_loglevel_type; | |
621 | enum lttng_domain_type config_domain_type; | |
622 | char *config_probe; | |
623 | char *config_function; | |
624 | char *config_channel_name; | |
625 | char *config_filter; | |
626 | char *config_exclude; | |
627 | ||
628 | assert(config); | |
629 | ||
441c16a7 MD |
630 | memset(&ev, 0, sizeof(ev)); |
631 | memset(&dom, 0, sizeof(dom)); | |
632 | ||
c7dfbc0e JR |
633 | config_enable_all = config->enable_all; |
634 | config_event_type = config->event_type; | |
635 | config_event_list = config->event_list; | |
636 | config_loglevel = config->loglevel; | |
637 | config_loglevel_type = config->loglevel_type; | |
638 | config_domain_type = config->domain_type; | |
639 | config_probe = config->probe; | |
640 | config_function = config->function; | |
641 | config_channel_name = config->channel_name; | |
642 | config_filter = config->filter; | |
643 | config_exclude = config->exclude; | |
644 | ||
645 | if (config_domain_type == LTTNG_DOMAIN_KERNEL) { | |
646 | if (config_loglevel) { | |
67b58630 JG |
647 | WARN("Kernel loglevels are not supported."); |
648 | } | |
53a80697 MD |
649 | } |
650 | ||
7d29a247 | 651 | /* Create lttng domain */ |
c7dfbc0e JR |
652 | dom.type = config_domain_type; |
653 | switch (config_domain_type) { | |
654 | case LTTNG_DOMAIN_KERNEL: | |
7972aab2 | 655 | dom.buf_type = LTTNG_BUFFER_GLOBAL; |
c7dfbc0e JR |
656 | break; |
657 | case LTTNG_DOMAIN_UST: | |
658 | case LTTNG_DOMAIN_JUL: | |
659 | case LTTNG_DOMAIN_LOG4J: | |
660 | case LTTNG_DOMAIN_PYTHON: | |
0e115563 | 661 | dom.buf_type = LTTNG_BUFFER_PER_UID; |
c7dfbc0e JR |
662 | break; |
663 | case LTTNG_DOMAIN_NONE: | |
664 | default: | |
3ecec76a | 665 | assert(0); |
2bdd86d4 | 666 | } |
7d29a247 | 667 | |
c7dfbc0e JR |
668 | |
669 | if (config_exclude) { | |
8b08c525 AB |
670 | switch (dom.type) { |
671 | case LTTNG_DOMAIN_KERNEL: | |
672 | case LTTNG_DOMAIN_JUL: | |
673 | case LTTNG_DOMAIN_LOG4J: | |
674 | case LTTNG_DOMAIN_PYTHON: | |
675 | ERR("Event name exclusions are not yet implemented for %s events", | |
676 | get_domain_str(dom.type)); | |
677 | ret = CMD_ERROR; | |
678 | goto error; | |
679 | case LTTNG_DOMAIN_UST: | |
680 | /* Exclusions supported */ | |
681 | break; | |
682 | default: | |
683 | assert(0); | |
684 | } | |
d5dd17fd JI |
685 | } |
686 | ||
c7dfbc0e | 687 | channel_name = config_channel_name; |
ae856491 | 688 | |
cd80958d DG |
689 | handle = lttng_create_handle(session_name, &dom); |
690 | if (handle == NULL) { | |
691 | ret = -1; | |
692 | goto error; | |
693 | } | |
1aef21b6 | 694 | |
89476427 JRJ |
695 | /* Prepare Mi */ |
696 | if (lttng_opt_mi) { | |
c7dfbc0e JR |
697 | /* Open a domain element */ |
698 | ret = mi_lttng_writer_open_element(writer, config_element_domain); | |
699 | if (ret) { | |
700 | ret = CMD_ERROR; | |
701 | goto error; | |
702 | } | |
703 | ||
704 | /* Specify the domain type */ | |
705 | ret = mi_lttng_writer_write_element_string(writer, | |
706 | config_element_type, | |
707 | mi_lttng_domaintype_string(config_domain_type)); | |
708 | if (ret) { | |
709 | ret = CMD_ERROR; | |
710 | goto error; | |
711 | } | |
712 | ||
89476427 JRJ |
713 | /* Open a events element */ |
714 | ret = mi_lttng_writer_open_element(writer, config_element_events); | |
715 | if (ret) { | |
716 | ret = CMD_ERROR; | |
717 | goto error; | |
718 | } | |
719 | } | |
720 | ||
c7dfbc0e | 721 | if (config_enable_all) { |
8c9ae521 | 722 | /* Default setup for enable all */ |
c7dfbc0e JR |
723 | if (config_domain_type == LTTNG_DOMAIN_KERNEL) { |
724 | ev.type = config_event_type; | |
29c62722 | 725 | strcpy(ev.name, "*"); |
300b8fd5 MD |
726 | /* kernel loglevels not implemented */ |
727 | ev.loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; | |
75e8c5ab MD |
728 | } else { |
729 | ev.type = LTTNG_EVENT_TRACEPOINT; | |
730 | strcpy(ev.name, "*"); | |
c7dfbc0e JR |
731 | ev.loglevel_type = config_loglevel_type; |
732 | if (config_loglevel) { | |
1036985c | 733 | ev.loglevel = loglevel_str_to_value(config_loglevel, config_domain_type); |
300b8fd5 | 734 | if (ev.loglevel == -1) { |
c7dfbc0e | 735 | ERR("Unknown loglevel %s", config_loglevel); |
2f70b271 | 736 | ret = -LTTNG_ERR_INVALID; |
300b8fd5 MD |
737 | goto error; |
738 | } | |
22e25b71 | 739 | } else { |
c7dfbc0e JR |
740 | assert(config_domain_type != LTTNG_DOMAIN_NONE || config_domain_type != LTTNG_DOMAIN_KERNEL); |
741 | switch (config_domain_type) { | |
742 | case LTTNG_DOMAIN_UST: | |
b2064f54 | 743 | ev.loglevel = -1; |
c7dfbc0e JR |
744 | break; |
745 | case LTTNG_DOMAIN_JUL: | |
b2064f54 | 746 | ev.loglevel = LTTNG_LOGLEVEL_JUL_ALL; |
c7dfbc0e JR |
747 | break; |
748 | case LTTNG_DOMAIN_LOG4J: | |
34aa3685 | 749 | ev.loglevel = LTTNG_LOGLEVEL_LOG4J_ALL; |
c7dfbc0e JR |
750 | break; |
751 | case LTTNG_DOMAIN_PYTHON: | |
0e115563 | 752 | ev.loglevel = LTTNG_LOGLEVEL_PYTHON_DEBUG; |
c7dfbc0e JR |
753 | break; |
754 | default: | |
755 | assert(0); | |
b2064f54 | 756 | } |
300b8fd5 | 757 | } |
75e8c5ab | 758 | } |
8c9ae521 | 759 | |
c7dfbc0e JR |
760 | if (config_exclude) { |
761 | ret = check_exclusion_subsets("*", config_exclude, | |
7ed70bc9 JI |
762 | &exclusion_count, &exclusion_list); |
763 | if (ret == CMD_ERROR) { | |
764 | goto error; | |
765 | } | |
89476427 | 766 | ev.exclusion = 1; |
502bbe89 PP |
767 | |
768 | warn_on_truncated_exclusion_names(exclusion_list, | |
769 | exclusion_count, &warn); | |
7ed70bc9 | 770 | } |
c7dfbc0e | 771 | if (!config_filter) { |
7ed70bc9 JI |
772 | ret = lttng_enable_event_with_exclusions(handle, |
773 | &ev, channel_name, | |
774 | NULL, | |
775 | exclusion_count, exclusion_list); | |
025faf73 DG |
776 | if (ret < 0) { |
777 | switch (-ret) { | |
778 | case LTTNG_ERR_KERN_EVENT_EXIST: | |
779 | WARN("Kernel events already enabled (channel %s, session %s)", | |
85076754 | 780 | print_channel_name(channel_name), session_name); |
89476427 | 781 | warn = 1; |
025faf73 | 782 | break; |
45d5d421 CB |
783 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
784 | { | |
785 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 786 | ERR("Events: %s (domain %s, channel %s, session %s)", |
45d5d421 | 787 | msg, |
721b94fe | 788 | get_domain_str(dom.type), |
45d5d421 CB |
789 | print_channel_name(channel_name), |
790 | session_name); | |
791 | error = 1; | |
792 | break; | |
793 | } | |
025faf73 | 794 | default: |
721b94fe | 795 | ERR("Events: %s (domain %s, channel %s, session %s)", |
85076754 | 796 | lttng_strerror(ret), |
721b94fe | 797 | get_domain_str(dom.type), |
85076754 MD |
798 | ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
799 | ? print_raw_channel_name(channel_name) | |
800 | : print_channel_name(channel_name), | |
801 | session_name); | |
89476427 | 802 | error = 1; |
025faf73 DG |
803 | break; |
804 | } | |
805 | goto end; | |
42224349 | 806 | } |
8c9ae521 | 807 | |
c7dfbc0e | 808 | switch (config_event_type) { |
025faf73 | 809 | case LTTNG_EVENT_TRACEPOINT: |
c7dfbc0e | 810 | if (config_loglevel && dom.type != LTTNG_DOMAIN_KERNEL) { |
9c48cab3 | 811 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
812 | |
813 | if (!exclusion_string) { | |
814 | PERROR("Cannot allocate exclusion_string"); | |
815 | error = 1; | |
816 | goto end; | |
817 | } | |
9c48cab3 | 818 | MSG("All %s tracepoints%s are enabled in channel %s for loglevel %s", |
b9dfb167 | 819 | get_domain_str(dom.type), |
9c48cab3 | 820 | exclusion_string, |
85076754 | 821 | print_channel_name(channel_name), |
c7dfbc0e | 822 | config_loglevel); |
9c48cab3 | 823 | free(exclusion_string); |
025faf73 | 824 | } else { |
9c48cab3 | 825 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
826 | |
827 | if (!exclusion_string) { | |
828 | PERROR("Cannot allocate exclusion_string"); | |
829 | error = 1; | |
830 | goto end; | |
831 | } | |
9c48cab3 | 832 | MSG("All %s tracepoints%s are enabled in channel %s", |
b9dfb167 | 833 | get_domain_str(dom.type), |
9c48cab3 | 834 | exclusion_string, |
85076754 | 835 | print_channel_name(channel_name)); |
9c48cab3 | 836 | free(exclusion_string); |
025faf73 DG |
837 | } |
838 | break; | |
839 | case LTTNG_EVENT_SYSCALL: | |
840 | if (opt_kernel) { | |
6e911cad MD |
841 | MSG("All %s system calls are enabled in channel %s", |
842 | get_domain_str(dom.type), | |
85076754 | 843 | print_channel_name(channel_name)); |
025faf73 DG |
844 | } |
845 | break; | |
846 | case LTTNG_EVENT_ALL: | |
c7dfbc0e | 847 | if (config_loglevel && dom.type != LTTNG_DOMAIN_KERNEL) { |
9c48cab3 | 848 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
849 | |
850 | if (!exclusion_string) { | |
851 | PERROR("Cannot allocate exclusion_string"); | |
852 | error = 1; | |
853 | goto end; | |
854 | } | |
9c48cab3 | 855 | MSG("All %s events%s are enabled in channel %s for loglevel %s", |
b9dfb167 | 856 | get_domain_str(dom.type), |
9c48cab3 | 857 | exclusion_string, |
85076754 | 858 | print_channel_name(channel_name), |
c7dfbc0e | 859 | config_loglevel); |
9c48cab3 | 860 | free(exclusion_string); |
025faf73 | 861 | } else { |
9c48cab3 | 862 | char *exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
863 | |
864 | if (!exclusion_string) { | |
865 | PERROR("Cannot allocate exclusion_string"); | |
866 | error = 1; | |
867 | goto end; | |
868 | } | |
9c48cab3 | 869 | MSG("All %s events%s are enabled in channel %s", |
b9dfb167 | 870 | get_domain_str(dom.type), |
9c48cab3 | 871 | exclusion_string, |
85076754 | 872 | print_channel_name(channel_name)); |
9c48cab3 | 873 | free(exclusion_string); |
025faf73 DG |
874 | } |
875 | break; | |
876 | default: | |
877 | /* | |
878 | * We should not be here since lttng_enable_event should have | |
879 | * failed on the event type. | |
880 | */ | |
881 | goto error; | |
57064ada | 882 | } |
f3ed775e | 883 | } |
89476427 | 884 | |
c7dfbc0e | 885 | if (config_filter) { |
89476427 | 886 | command_ret = lttng_enable_event_with_exclusions(handle, &ev, channel_name, |
c7dfbc0e | 887 | config_filter, exclusion_count, exclusion_list); |
89476427 JRJ |
888 | if (command_ret < 0) { |
889 | switch (-command_ret) { | |
16363652 | 890 | case LTTNG_ERR_FILTER_EXIST: |
85076754 | 891 | WARN("Filter on all events is already enabled" |
721b94fe JR |
892 | " (domain %s, channel %s, session %s)", |
893 | get_domain_str(dom.type), | |
85076754 | 894 | print_channel_name(channel_name), session_name); |
89476427 | 895 | warn = 1; |
16363652 | 896 | break; |
45d5d421 CB |
897 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
898 | { | |
899 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 900 | ERR("All events: %s (domain %s, channel %s, session %s, filter \'%s\')", |
45d5d421 | 901 | msg, |
721b94fe | 902 | get_domain_str(dom.type), |
45d5d421 | 903 | print_channel_name(channel_name), |
c7dfbc0e | 904 | session_name, config_filter); |
45d5d421 CB |
905 | error = 1; |
906 | break; | |
907 | } | |
16363652 | 908 | default: |
721b94fe | 909 | ERR("All events: %s (domain %s, channel %s, session %s, filter \'%s\')", |
da3d7d0e | 910 | lttng_strerror(command_ret), |
721b94fe | 911 | get_domain_str(dom.type), |
da3d7d0e | 912 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
85076754 MD |
913 | ? print_raw_channel_name(channel_name) |
914 | : print_channel_name(channel_name), | |
c7dfbc0e | 915 | session_name, config_filter); |
89476427 | 916 | error = 1; |
16363652 DG |
917 | break; |
918 | } | |
89476427 | 919 | error_holder = command_ret; |
16363652 | 920 | } else { |
89476427 | 921 | ev.filter = 1; |
c7dfbc0e | 922 | MSG("Filter '%s' successfully set", config_filter); |
16363652 DG |
923 | } |
924 | } | |
89476427 JRJ |
925 | |
926 | if (lttng_opt_mi) { | |
927 | /* The wildcard * is used for kernel and ust domain to | |
928 | * represent ALL. We copy * in event name to force the wildcard use | |
929 | * for kernel domain | |
930 | * | |
931 | * Note: this is strictly for semantic and printing while in | |
932 | * machine interface mode. | |
933 | */ | |
934 | strcpy(ev.name, "*"); | |
935 | ||
936 | /* If we reach here the events are enabled */ | |
937 | if (!error && !warn) { | |
938 | ev.enabled = 1; | |
939 | } else { | |
940 | ev.enabled = 0; | |
941 | success = 0; | |
942 | } | |
970d848b | 943 | ret = mi_lttng_event(writer, &ev, 1, handle->domain.type); |
89476427 JRJ |
944 | if (ret) { |
945 | ret = CMD_ERROR; | |
946 | goto error; | |
947 | } | |
948 | ||
949 | /* print exclusion */ | |
950 | ret = mi_print_exclusion(exclusion_count, exclusion_list); | |
951 | if (ret) { | |
952 | ret = CMD_ERROR; | |
953 | goto error; | |
954 | } | |
955 | ||
956 | /* Success ? */ | |
957 | ret = mi_lttng_writer_write_element_bool(writer, | |
958 | mi_lttng_element_command_success, success); | |
959 | if (ret) { | |
960 | ret = CMD_ERROR; | |
961 | goto error; | |
962 | } | |
963 | ||
964 | /* Close event element */ | |
965 | ret = mi_lttng_writer_close_element(writer); | |
966 | if (ret) { | |
967 | ret = CMD_ERROR; | |
968 | goto error; | |
969 | } | |
970 | } | |
971 | ||
8c9ae521 | 972 | goto end; |
f3ed775e DG |
973 | } |
974 | ||
975 | /* Strip event list */ | |
c7dfbc0e | 976 | event_name = strtok(config_event_list, ","); |
f3ed775e | 977 | while (event_name != NULL) { |
6181537c DG |
978 | /* Copy name and type of the event */ |
979 | strncpy(ev.name, event_name, LTTNG_SYMBOL_NAME_LEN); | |
980 | ev.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; | |
c7dfbc0e | 981 | ev.type = config_event_type; |
6181537c | 982 | |
f3ed775e | 983 | /* Kernel tracer action */ |
c7dfbc0e JR |
984 | switch (config_domain_type) { |
985 | case LTTNG_DOMAIN_KERNEL: | |
f3ed775e | 986 | DBG("Enabling kernel event %s for channel %s", |
85076754 MD |
987 | event_name, |
988 | print_channel_name(channel_name)); | |
f3ed775e | 989 | |
c7dfbc0e | 990 | switch (config_event_type) { |
29c62722 MD |
991 | case LTTNG_EVENT_ALL: /* Enable tracepoints and syscalls */ |
992 | /* If event name differs from *, select tracepoint. */ | |
993 | if (strcmp(ev.name, "*")) { | |
994 | ev.type = LTTNG_EVENT_TRACEPOINT; | |
995 | } | |
996 | break; | |
e6ddca71 | 997 | case LTTNG_EVENT_TRACEPOINT: |
f3ed775e | 998 | break; |
7d29a247 | 999 | case LTTNG_EVENT_PROBE: |
c7dfbc0e | 1000 | ret = parse_probe_opts(&ev, config_probe); |
49d4e302 | 1001 | if (ret) { |
cf0e5467 | 1002 | ERR("Unable to parse probe options"); |
0d63dd19 DG |
1003 | ret = 0; |
1004 | goto error; | |
1005 | } | |
f3ed775e DG |
1006 | break; |
1007 | case LTTNG_EVENT_FUNCTION: | |
c7dfbc0e | 1008 | ret = parse_probe_opts(&ev, config_function); |
49d4e302 | 1009 | if (ret) { |
8f0d098b MD |
1010 | ERR("Unable to parse function probe options"); |
1011 | ret = 0; | |
1012 | goto error; | |
1013 | } | |
1014 | break; | |
a54bd42d | 1015 | case LTTNG_EVENT_SYSCALL: |
c6aa2d41 MD |
1016 | ev.type = LTTNG_EVENT_SYSCALL; |
1017 | break; | |
f3ed775e | 1018 | default: |
1ab1ea0b | 1019 | ret = CMD_UNDEFINED; |
f3ed775e DG |
1020 | goto error; |
1021 | } | |
0cda4f28 | 1022 | |
0cda4f28 | 1023 | /* kernel loglevels not implemented */ |
8005f29a | 1024 | ev.loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; |
c7dfbc0e JR |
1025 | break; |
1026 | case LTTNG_DOMAIN_UST: | |
1027 | /* User-space tracer action */ | |
300b8fd5 | 1028 | DBG("Enabling UST event %s for channel %s, loglevel %s", event_name, |
c7dfbc0e | 1029 | print_channel_name(channel_name), config_loglevel ? : "<all>"); |
2bdd86d4 | 1030 | |
c7dfbc0e | 1031 | switch (config_event_type) { |
2bdd86d4 | 1032 | case LTTNG_EVENT_ALL: /* Default behavior is tracepoint */ |
2bdd86d4 MD |
1033 | /* Fall-through */ |
1034 | case LTTNG_EVENT_TRACEPOINT: | |
e4baff1e MD |
1035 | /* Copy name and type of the event */ |
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'; | |
2bdd86d4 MD |
1039 | break; |
1040 | case LTTNG_EVENT_PROBE: | |
1041 | case LTTNG_EVENT_FUNCTION: | |
2bdd86d4 MD |
1042 | case LTTNG_EVENT_SYSCALL: |
1043 | default: | |
cc62c0c0 | 1044 | ERR("Event type not available for user-space tracing"); |
4ce78777 | 1045 | ret = CMD_UNSUPPORTED; |
2bdd86d4 MD |
1046 | goto error; |
1047 | } | |
0cda4f28 | 1048 | |
c7dfbc0e | 1049 | if (config_exclude) { |
89476427 | 1050 | ev.exclusion = 1; |
d0f1c989 | 1051 | if (config_event_type != LTTNG_EVENT_ALL && config_event_type != LTTNG_EVENT_TRACEPOINT) { |
d5dd17fd JI |
1052 | ERR("Exclusion option can only be used with tracepoint events"); |
1053 | ret = CMD_ERROR; | |
1054 | goto error; | |
1055 | } | |
7ed70bc9 JI |
1056 | /* Free previously allocated items */ |
1057 | if (exclusion_list != NULL) { | |
1058 | while (exclusion_count--) { | |
1059 | free(exclusion_list[exclusion_count]); | |
1060 | } | |
1061 | free(exclusion_list); | |
1062 | exclusion_list = NULL; | |
1063 | } | |
1064 | /* Check for proper subsets */ | |
c7dfbc0e | 1065 | ret = check_exclusion_subsets(event_name, config_exclude, |
7ed70bc9 JI |
1066 | &exclusion_count, &exclusion_list); |
1067 | if (ret == CMD_ERROR) { | |
1068 | goto error; | |
1069 | } | |
502bbe89 PP |
1070 | |
1071 | warn_on_truncated_exclusion_names( | |
1072 | exclusion_list, exclusion_count, &warn); | |
7ed70bc9 JI |
1073 | } |
1074 | ||
c7dfbc0e JR |
1075 | ev.loglevel_type = config_loglevel_type; |
1076 | if (config_loglevel) { | |
1036985c | 1077 | ev.loglevel = loglevel_ust_str_to_value(config_loglevel); |
8005f29a | 1078 | if (ev.loglevel == -1) { |
c7dfbc0e | 1079 | ERR("Unknown loglevel %s", config_loglevel); |
2f70b271 | 1080 | ret = -LTTNG_ERR_INVALID; |
8005f29a MD |
1081 | goto error; |
1082 | } | |
22e25b71 MD |
1083 | } else { |
1084 | ev.loglevel = -1; | |
ed7f4083 | 1085 | } |
c7dfbc0e JR |
1086 | break; |
1087 | case LTTNG_DOMAIN_JUL: | |
1088 | case LTTNG_DOMAIN_LOG4J: | |
1089 | case LTTNG_DOMAIN_PYTHON: | |
1090 | if (config_event_type != LTTNG_EVENT_ALL && | |
1091 | config_event_type != LTTNG_EVENT_TRACEPOINT) { | |
5cdb6027 | 1092 | ERR("Event type not supported for domain."); |
b9dfb167 DG |
1093 | ret = CMD_UNSUPPORTED; |
1094 | goto error; | |
1095 | } | |
b2064f54 | 1096 | |
c7dfbc0e JR |
1097 | ev.loglevel_type = config_loglevel_type; |
1098 | if (config_loglevel) { | |
1036985c | 1099 | ev.loglevel = loglevel_str_to_value(config_loglevel, config_domain_type); |
b2064f54 | 1100 | if (ev.loglevel == -1) { |
c7dfbc0e | 1101 | ERR("Unknown loglevel %s", config_loglevel); |
b2064f54 DG |
1102 | ret = -LTTNG_ERR_INVALID; |
1103 | goto error; | |
1104 | } | |
1105 | } else { | |
c7dfbc0e JR |
1106 | switch (config_domain_type) { |
1107 | case LTTNG_DOMAIN_JUL: | |
5cdb6027 | 1108 | ev.loglevel = LTTNG_LOGLEVEL_JUL_ALL; |
c7dfbc0e JR |
1109 | break; |
1110 | case LTTNG_DOMAIN_LOG4J: | |
5cdb6027 | 1111 | ev.loglevel = LTTNG_LOGLEVEL_LOG4J_ALL; |
c7dfbc0e JR |
1112 | break; |
1113 | case LTTNG_DOMAIN_PYTHON: | |
0e115563 | 1114 | ev.loglevel = LTTNG_LOGLEVEL_PYTHON_DEBUG; |
c7dfbc0e JR |
1115 | break; |
1116 | default: | |
1117 | assert(0); | |
1118 | break; | |
5cdb6027 | 1119 | } |
b2064f54 | 1120 | } |
b9dfb167 DG |
1121 | ev.type = LTTNG_EVENT_TRACEPOINT; |
1122 | strncpy(ev.name, event_name, LTTNG_SYMBOL_NAME_LEN); | |
1123 | ev.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; | |
c7dfbc0e JR |
1124 | break; |
1125 | default: | |
3ecec76a | 1126 | assert(0); |
f3ed775e DG |
1127 | } |
1128 | ||
c7dfbc0e | 1129 | if (!config_filter) { |
9c48cab3 JI |
1130 | char *exclusion_string; |
1131 | ||
89476427 | 1132 | command_ret = lttng_enable_event_with_exclusions(handle, |
7ed70bc9 JI |
1133 | &ev, channel_name, |
1134 | NULL, exclusion_count, exclusion_list); | |
9c48cab3 | 1135 | exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
1136 | if (!exclusion_string) { |
1137 | PERROR("Cannot allocate exclusion_string"); | |
1138 | error = 1; | |
1139 | goto end; | |
1140 | } | |
89476427 | 1141 | if (command_ret < 0) { |
025faf73 | 1142 | /* Turn ret to positive value to handle the positive error code */ |
89476427 | 1143 | switch (-command_ret) { |
025faf73 | 1144 | case LTTNG_ERR_KERN_EVENT_EXIST: |
9c48cab3 | 1145 | WARN("Kernel event %s%s already enabled (channel %s, session %s)", |
85076754 | 1146 | event_name, |
9c48cab3 | 1147 | exclusion_string, |
85076754 | 1148 | print_channel_name(channel_name), session_name); |
89476427 | 1149 | warn = 1; |
025faf73 | 1150 | break; |
45d5d421 CB |
1151 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
1152 | { | |
1153 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 1154 | ERR("Event %s%s: %s (domain %s,channel %s, session %s)", event_name, |
45d5d421 CB |
1155 | exclusion_string, |
1156 | msg, | |
721b94fe | 1157 | get_domain_str(dom.type), |
45d5d421 CB |
1158 | print_channel_name(channel_name), |
1159 | session_name); | |
1160 | error = 1; | |
1161 | break; | |
1162 | } | |
025faf73 | 1163 | default: |
721b94fe | 1164 | ERR("Event %s%s: %s (domain %s, channel %s, session %s)", event_name, |
9c48cab3 | 1165 | exclusion_string, |
da3d7d0e | 1166 | lttng_strerror(command_ret), |
721b94fe | 1167 | get_domain_str(dom.type), |
da3d7d0e | 1168 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
85076754 MD |
1169 | ? print_raw_channel_name(channel_name) |
1170 | : print_channel_name(channel_name), | |
1171 | session_name); | |
89476427 | 1172 | error = 1; |
025faf73 DG |
1173 | break; |
1174 | } | |
89476427 | 1175 | error_holder = command_ret; |
025faf73 | 1176 | } else { |
8274eeba AB |
1177 | switch (dom.type) { |
1178 | case LTTNG_DOMAIN_KERNEL: | |
1179 | case LTTNG_DOMAIN_UST: | |
49ceaa70 | 1180 | MSG("%s event %s%s created in channel %s", |
8274eeba AB |
1181 | get_domain_str(dom.type), |
1182 | event_name, | |
1183 | exclusion_string, | |
1184 | print_channel_name(channel_name)); | |
1185 | break; | |
1186 | case LTTNG_DOMAIN_JUL: | |
1187 | case LTTNG_DOMAIN_LOG4J: | |
1188 | case LTTNG_DOMAIN_PYTHON: | |
1189 | /* | |
1190 | * Don't print the default channel | |
1191 | * name for agent domains. | |
1192 | */ | |
895707da | 1193 | MSG("%s event %s%s enabled", |
8274eeba AB |
1194 | get_domain_str(dom.type), |
1195 | event_name, | |
1196 | exclusion_string); | |
1197 | break; | |
1198 | default: | |
1199 | assert(0); | |
49ceaa70 | 1200 | } |
42224349 | 1201 | } |
9c48cab3 | 1202 | free(exclusion_string); |
6181537c | 1203 | } |
025faf73 | 1204 | |
c7dfbc0e | 1205 | if (config_filter) { |
9c48cab3 JI |
1206 | char *exclusion_string; |
1207 | ||
89476427 JRJ |
1208 | /* Filter present */ |
1209 | ev.filter = 1; | |
1210 | ||
1211 | command_ret = lttng_enable_event_with_exclusions(handle, &ev, channel_name, | |
c7dfbc0e | 1212 | config_filter, exclusion_count, exclusion_list); |
9c48cab3 | 1213 | exclusion_string = print_exclusions(exclusion_count, exclusion_list); |
5ef79758 MD |
1214 | if (!exclusion_string) { |
1215 | PERROR("Cannot allocate exclusion_string"); | |
1216 | error = 1; | |
1217 | goto end; | |
1218 | } | |
89476427 JRJ |
1219 | if (command_ret < 0) { |
1220 | switch (-command_ret) { | |
7671f53c | 1221 | case LTTNG_ERR_FILTER_EXIST: |
9c48cab3 | 1222 | WARN("Filter on event %s%s is already enabled" |
721b94fe | 1223 | " (domain %s, channel %s, session %s)", |
85076754 | 1224 | event_name, |
9c48cab3 | 1225 | exclusion_string, |
721b94fe | 1226 | get_domain_str(dom.type), |
85076754 | 1227 | print_channel_name(channel_name), session_name); |
89476427 | 1228 | warn = 1; |
7671f53c | 1229 | break; |
45d5d421 CB |
1230 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
1231 | { | |
1232 | const char *msg = "The command tried to enable an event in a new domain for a session that has already been started once."; | |
721b94fe | 1233 | ERR("Event %s%s: %s (domain %s, channel %s, session %s, filter \'%s\')", ev.name, |
45d5d421 CB |
1234 | exclusion_string, |
1235 | msg, | |
721b94fe | 1236 | get_domain_str(dom.type), |
45d5d421 | 1237 | print_channel_name(channel_name), |
c7dfbc0e | 1238 | session_name, config_filter); |
45d5d421 CB |
1239 | error = 1; |
1240 | break; | |
1241 | } | |
7671f53c | 1242 | default: |
721b94fe | 1243 | ERR("Event %s%s: %s (domain %s, channel %s, session %s, filter \'%s\')", ev.name, |
9c48cab3 | 1244 | exclusion_string, |
da3d7d0e | 1245 | lttng_strerror(command_ret), |
721b94fe | 1246 | get_domain_str(dom.type), |
da3d7d0e | 1247 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME |
85076754 MD |
1248 | ? print_raw_channel_name(channel_name) |
1249 | : print_channel_name(channel_name), | |
c7dfbc0e | 1250 | session_name, config_filter); |
89476427 | 1251 | error = 1; |
7671f53c CB |
1252 | break; |
1253 | } | |
89476427 JRJ |
1254 | error_holder = command_ret; |
1255 | ||
16363652 | 1256 | } else { |
721b94fe | 1257 | MSG("Event %s%s: Filter '%s' for domain %s successfully set", |
9c48cab3 | 1258 | event_name, exclusion_string, |
c7dfbc0e | 1259 | config_filter, |
721b94fe | 1260 | get_domain_str(dom.type)); |
53a80697 | 1261 | } |
9c48cab3 | 1262 | free(exclusion_string); |
53a80697 | 1263 | } |
6181537c | 1264 | |
89476427 JRJ |
1265 | if (lttng_opt_mi) { |
1266 | if (command_ret) { | |
1267 | success = 0; | |
1268 | ev.enabled = 0; | |
1269 | } else { | |
1270 | ev.enabled = 1; | |
1271 | } | |
1272 | ||
970d848b | 1273 | ret = mi_lttng_event(writer, &ev, 1, handle->domain.type); |
89476427 JRJ |
1274 | if (ret) { |
1275 | ret = CMD_ERROR; | |
1276 | goto error; | |
1277 | } | |
1278 | ||
1279 | /* print exclusion */ | |
1280 | ret = mi_print_exclusion(exclusion_count, exclusion_list); | |
1281 | if (ret) { | |
1282 | ret = CMD_ERROR; | |
1283 | goto error; | |
1284 | } | |
1285 | ||
1286 | /* Success ? */ | |
1287 | ret = mi_lttng_writer_write_element_bool(writer, | |
1288 | mi_lttng_element_command_success, success); | |
1289 | if (ret) { | |
1290 | ret = CMD_ERROR; | |
1291 | goto end; | |
1292 | } | |
1293 | ||
1294 | /* Close event element */ | |
1295 | ret = mi_lttng_writer_close_element(writer); | |
1296 | if (ret) { | |
1297 | ret = CMD_ERROR; | |
1298 | goto end; | |
1299 | } | |
1300 | } | |
1301 | ||
f3ed775e DG |
1302 | /* Next event */ |
1303 | event_name = strtok(NULL, ","); | |
89476427 JRJ |
1304 | /* Reset warn, error and success */ |
1305 | success = 1; | |
f3ed775e DG |
1306 | } |
1307 | ||
8c9ae521 | 1308 | end: |
89476427 JRJ |
1309 | /* Close Mi */ |
1310 | if (lttng_opt_mi) { | |
c7dfbc0e JR |
1311 | /* Close events and domain element */ |
1312 | ret = mi_lttng_close_multi_element(writer, 2); | |
89476427 JRJ |
1313 | if (ret) { |
1314 | ret = CMD_ERROR; | |
1315 | goto error; | |
1316 | } | |
1317 | } | |
f3ed775e | 1318 | error: |
ae856491 DG |
1319 | if (warn) { |
1320 | ret = CMD_WARNING; | |
1321 | } | |
89476427 JRJ |
1322 | if (error) { |
1323 | ret = CMD_ERROR; | |
1324 | } | |
cd80958d DG |
1325 | lttng_destroy_handle(handle); |
1326 | ||
7ed70bc9 JI |
1327 | if (exclusion_list != NULL) { |
1328 | while (exclusion_count--) { | |
1329 | free(exclusion_list[exclusion_count]); | |
1330 | } | |
1331 | free(exclusion_list); | |
1332 | } | |
1333 | ||
89476427 JRJ |
1334 | /* Overwrite ret with error_holder if there was an actual error with |
1335 | * enabling an event. | |
1336 | */ | |
1337 | ret = error_holder ? error_holder : ret; | |
1338 | ||
f3ed775e DG |
1339 | return ret; |
1340 | } | |
1341 | ||
1342 | /* | |
6181537c | 1343 | * Add event to trace session |
f3ed775e | 1344 | */ |
721b94fe JR |
1345 | |
1346 | struct args_tuple { | |
1347 | int argv_index_start; | |
1348 | int argv_index_end; | |
1349 | }; | |
1350 | ||
721b94fe JR |
1351 | |
1352 | static struct domain_configuration *initialize_domain_configuration(enum lttng_domain_type type) | |
f3ed775e | 1353 | { |
f3ed775e | 1354 | |
721b94fe | 1355 | struct domain_configuration *config = malloc(sizeof(struct domain_configuration)); |
f3ed775e | 1356 | |
721b94fe JR |
1357 | if (!config) { |
1358 | goto end; | |
1359 | } | |
f3ed775e | 1360 | |
721b94fe | 1361 | switch(type) { |
c7dfbc0e JR |
1362 | case LTTNG_DOMAIN_KERNEL: |
1363 | config->domain_type = LTTNG_DOMAIN_KERNEL; | |
1364 | break; | |
1365 | case LTTNG_DOMAIN_UST: | |
1366 | config->domain_type = LTTNG_DOMAIN_UST; | |
1367 | break; | |
1368 | case LTTNG_DOMAIN_JUL: | |
1369 | config->domain_type = LTTNG_DOMAIN_JUL; | |
1370 | break; | |
1371 | case LTTNG_DOMAIN_LOG4J: | |
1372 | config->domain_type = LTTNG_DOMAIN_LOG4J; | |
1373 | break; | |
1374 | case LTTNG_DOMAIN_PYTHON: | |
1375 | config->domain_type = LTTNG_DOMAIN_PYTHON; | |
1376 | break; | |
1377 | case LTTNG_DOMAIN_NONE: | |
1378 | default: | |
1379 | free(config); | |
1380 | config=NULL; | |
1381 | goto end; | |
721b94fe JR |
1382 | }; |
1383 | ||
1384 | config->event_type = LTTNG_EVENT_ALL ; | |
1385 | config->enable_all = 0; | |
1386 | config->loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; | |
1387 | config->loglevel = NULL; | |
1388 | config->probe = NULL; | |
1389 | config->function = NULL; | |
1390 | config->channel_name = NULL; | |
1391 | config->filter = NULL; | |
1392 | config->exclude = NULL; | |
1393 | config->event_list = NULL; | |
1394 | end: | |
1395 | return config; | |
1396 | } | |
de044b7a | 1397 | |
721b94fe JR |
1398 | int cmd_enable_events(int argc, const char **argv) |
1399 | { | |
1400 | int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1; | |
1401 | static poptContext pc; | |
1402 | char *session_name = NULL; | |
1403 | int event_type = -1; | |
f3ed775e | 1404 | |
721b94fe | 1405 | struct domain_configuration *jul_config = NULL; |
c7dfbc0e | 1406 | struct domain_configuration *kernel_config = NULL; |
721b94fe JR |
1407 | struct domain_configuration *log4j_config = NULL; |
1408 | struct domain_configuration *python_config = NULL; | |
c7dfbc0e | 1409 | struct domain_configuration *ust_config = NULL; |
3ecec76a | 1410 | |
89476427 JRJ |
1411 | /* Mi check */ |
1412 | if (lttng_opt_mi) { | |
1413 | writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi); | |
1414 | if (!writer) { | |
1415 | ret = -LTTNG_ERR_NOMEM; | |
1416 | goto end; | |
1417 | } | |
1418 | ||
1419 | /* Open command element */ | |
1420 | ret = mi_lttng_writer_command_open(writer, | |
1421 | mi_lttng_element_command_enable_event); | |
1422 | if (ret) { | |
1423 | ret = CMD_ERROR; | |
1424 | goto end; | |
1425 | } | |
1426 | ||
1427 | /* Open output element */ | |
1428 | ret = mi_lttng_writer_open_element(writer, | |
1429 | mi_lttng_element_command_output); | |
1430 | if (ret) { | |
1431 | ret = CMD_ERROR; | |
1432 | goto end; | |
1433 | } | |
c7dfbc0e JR |
1434 | |
1435 | /* Open the domains element */ | |
1436 | ret = mi_lttng_domains_open(writer); | |
1437 | if (ret) { | |
1438 | ret = CMD_ERROR; | |
1439 | goto end; | |
1440 | } | |
1441 | ||
89476427 JRJ |
1442 | } |
1443 | ||
721b94fe JR |
1444 | /* Parse global arguments */ |
1445 | pc = poptGetContext(NULL, argc, argv, global_long_options, 0); | |
1446 | poptReadDefaultConfig(pc, 0); | |
1447 | ||
1448 | ||
1449 | while ((opt = poptGetNextOpt(pc)) != -1) { | |
1450 | switch (opt) { | |
1451 | case OPT_HELP: | |
1452 | SHOW_HELP(); | |
1453 | goto end; | |
1454 | case OPT_LIST_OPTIONS: | |
1455 | list_cmd_options(stdout, long_options); | |
1456 | goto end; | |
1457 | default: | |
1458 | break; | |
1459 | } | |
f3ed775e DG |
1460 | } |
1461 | ||
721b94fe JR |
1462 | /* Dispose the global arguments context */ |
1463 | poptFreeContext(pc); | |
1464 | pc = NULL; | |
1465 | ||
c7dfbc0e JR |
1466 | if (!opt_session_name) { |
1467 | session_name = get_session_name(); | |
1468 | if (session_name == NULL) { | |
1469 | command_ret = CMD_ERROR; | |
1470 | success = 0; | |
1471 | goto mi_closing; | |
1472 | } | |
1473 | } else { | |
1474 | session_name = opt_session_name; | |
1475 | } | |
1476 | ||
721b94fe JR |
1477 | /* Find the number of domain based on the passed arguments */ |
1478 | int i; | |
1479 | int args_tuple_count = 0; | |
1480 | int arg_state_looking_for_end = 0; | |
1481 | struct args_tuple *args_tuple_list = NULL; | |
1482 | for (i = 1; i < argc ; i++) { | |
1483 | ||
1484 | if (strcmp("-u", argv[i]) && strcmp("--userspace", argv[i]) && | |
1485 | strcmp("-j", argv[i]) && strcmp("--jul", argv[i]) && | |
1486 | strcmp("-l", argv[i]) && strcmp("--log4j", argv[i]) && | |
1487 | strcmp("-p", argv[i]) && strcmp("--python", argv[i]) && | |
1488 | strcmp("-k", argv[i]) && strcmp("--kernel", argv[i])) { | |
1489 | continue; | |
1490 | } | |
1491 | ||
1492 | ||
1493 | struct args_tuple *tmp_pointer = NULL; | |
1494 | args_tuple_count++; | |
1495 | tmp_pointer = realloc(args_tuple_list, sizeof(struct args_tuple) * args_tuple_count); | |
1496 | if (!tmp_pointer) { | |
1497 | ERR("Realoc of args tuple failed"); | |
1498 | ret = CMD_ERROR; | |
1499 | goto end; | |
1500 | } | |
1501 | args_tuple_list = tmp_pointer; | |
1502 | ||
1503 | if (!arg_state_looking_for_end) { | |
1504 | if (args_tuple_count -1 < 0) { | |
1505 | ERR("Args parsing illegal state"); | |
1506 | ret = CMD_ERROR; | |
1507 | goto end; | |
1508 | } | |
1509 | args_tuple_list[args_tuple_count-1].argv_index_start = i; | |
1510 | arg_state_looking_for_end = 1; | |
1511 | } else { | |
1512 | if (args_tuple_count - 2 < 0 || args_tuple_count -1 < 0) { | |
1513 | ERR("Args parsing illegal state"); | |
1514 | ret = CMD_ERROR; | |
1515 | goto end; | |
1516 | } | |
1517 | ||
1518 | /* Close the previous tuple */ | |
1519 | args_tuple_list[args_tuple_count-2].argv_index_end = i - 1; | |
1520 | ||
1521 | /* Start the new tuple */ | |
1522 | args_tuple_list[args_tuple_count-1].argv_index_start = i; | |
cd80958d | 1523 | } |
cd80958d DG |
1524 | } |
1525 | ||
721b94fe JR |
1526 | if (args_tuple_count <= 0) { |
1527 | ret = print_missing_or_multiple_domains(0); | |
1528 | if (ret) { | |
1529 | ret = CMD_ERROR; | |
1530 | goto end; | |
1531 | } | |
1532 | goto end; | |
89476427 JRJ |
1533 | } |
1534 | ||
721b94fe JR |
1535 | /* Close the last tuple */ |
1536 | args_tuple_list[args_tuple_count-1].argv_index_end = i - 1; | |
1537 | ||
1538 | if (args_tuple_count == 1) { | |
1539 | /* Preserve the old way with a domain flag that can be anywhere */ | |
1540 | args_tuple_list[0].argv_index_start = 1; | |
1541 | } | |
1542 | ||
1543 | for (i = 0; i < args_tuple_count; i++) { | |
1544 | struct args_tuple *tuple = &args_tuple_list[i]; | |
1545 | int cur_argc = tuple->argv_index_end - tuple-> argv_index_start + 1; | |
1546 | const char **cur_argv = &argv[tuple->argv_index_start]; | |
1547 | ||
1548 | /* Default options */ | |
1549 | ||
1550 | /* Domain */ | |
1551 | opt_domain = LTTNG_DOMAIN_NONE; | |
1552 | ||
1553 | /* Event options */ | |
1554 | opt_enable_all = 0; | |
1555 | opt_event_type = LTTNG_EVENT_ALL; | |
1556 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; | |
1557 | opt_event_list = NULL; | |
1558 | opt_loglevel = NULL; | |
1559 | opt_probe = NULL; | |
1560 | opt_function = NULL; | |
1561 | opt_channel_name = NULL; | |
1562 | opt_filter = NULL; | |
1563 | opt_exclude = NULL; | |
1564 | ||
1565 | pc = poptGetContext(NULL, cur_argc, cur_argv, long_options, POPT_CONTEXT_KEEP_FIRST); | |
1566 | poptReadDefaultConfig(pc, 0); | |
1567 | ||
1568 | /* Default event type */ | |
1569 | opt_event_type = LTTNG_EVENT_ALL; | |
1570 | ||
1571 | while ((opt = poptGetNextOpt(pc)) != -1) { | |
1572 | switch (opt) { | |
1573 | case OPT_HELP: | |
1574 | SHOW_HELP(); | |
1575 | goto end; | |
1576 | case OPT_TRACEPOINT: | |
1577 | opt_event_type = LTTNG_EVENT_TRACEPOINT; | |
1578 | break; | |
1579 | case OPT_PROBE: | |
1580 | opt_event_type = LTTNG_EVENT_PROBE; | |
1581 | break; | |
1582 | case OPT_FUNCTION: | |
1583 | opt_event_type = LTTNG_EVENT_FUNCTION; | |
1584 | break; | |
1585 | case OPT_SYSCALL: | |
1586 | opt_event_type = LTTNG_EVENT_SYSCALL; | |
1587 | break; | |
1588 | case OPT_LOGLEVEL: | |
1589 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_RANGE; | |
1590 | opt_loglevel = poptGetOptArg(pc); | |
1591 | break; | |
1592 | case OPT_LOGLEVEL_ONLY: | |
1593 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_SINGLE; | |
1594 | opt_loglevel = poptGetOptArg(pc); | |
1595 | break; | |
1596 | case OPT_LIST_OPTIONS: | |
1597 | list_cmd_options(stdout, long_options); | |
1598 | goto end; | |
1599 | case OPT_FILTER: | |
1600 | break; | |
1601 | case OPT_EXCLUDE: | |
1602 | break; | |
1603 | default: | |
1604 | ret = CMD_UNDEFINED; | |
1605 | goto end; | |
1606 | } | |
1607 | ||
1608 | /* Validate event type. Multiple event type are not supported. */ | |
1609 | if (event_type == -1) { | |
1610 | event_type = opt_event_type; | |
1611 | } else { | |
1612 | if (event_type != opt_event_type) { | |
1613 | ERR("Multiple event type not supported."); | |
1614 | ret = CMD_ERROR; | |
1615 | goto end; | |
1616 | } | |
1617 | } | |
1618 | } | |
1619 | ||
1620 | struct domain_configuration *tmp_config = initialize_domain_configuration(opt_domain); | |
1621 | ||
1622 | opt_event_list = (char*) poptGetArg(pc); | |
1623 | if (opt_event_list == NULL && opt_enable_all == 0) { | |
1624 | ERR("Missing event name(s).\n"); | |
1625 | ret = CMD_ERROR; | |
1626 | goto end; | |
1627 | } | |
1628 | ||
721b94fe | 1629 | |
c7dfbc0e JR |
1630 | tmp_config->event_type = opt_event_type ; |
1631 | tmp_config->enable_all = opt_enable_all; | |
1632 | tmp_config->loglevel_type = opt_loglevel_type; | |
1633 | tmp_config->loglevel = opt_loglevel; | |
1634 | tmp_config->probe = opt_probe; | |
1635 | tmp_config->function = opt_function; | |
1636 | tmp_config->channel_name = opt_channel_name; | |
1637 | tmp_config->filter = opt_filter; | |
1638 | tmp_config->exclude = opt_exclude; | |
1639 | tmp_config->event_list = opt_event_list; | |
721b94fe JR |
1640 | |
1641 | switch(opt_domain) { | |
1642 | case LTTNG_DOMAIN_KERNEL: | |
1643 | if (kernel_config) { | |
1644 | ERR("Only one -k option is permitted per command"); | |
1645 | ret = CMD_ERROR; | |
1646 | goto end; | |
1647 | } | |
1648 | kernel_config = tmp_config; | |
1649 | break; | |
1650 | case LTTNG_DOMAIN_UST: | |
1651 | if (ust_config) { | |
1652 | ERR("Only one -u option is permitted per command"); | |
1653 | ret = CMD_ERROR; | |
1654 | goto end; | |
1655 | } | |
1656 | ust_config = tmp_config; | |
1657 | break; | |
1658 | case LTTNG_DOMAIN_JUL: | |
1659 | if (jul_config) { | |
1660 | ERR("Only one -j option is permitted per command"); | |
1661 | ret = CMD_ERROR; | |
1662 | goto end; | |
1663 | } | |
1664 | jul_config = tmp_config; | |
1665 | break; | |
1666 | case LTTNG_DOMAIN_LOG4J: | |
1667 | if (log4j_config) { | |
1668 | ERR("Only one -l option is permitted per command"); | |
1669 | ret = CMD_ERROR; | |
1670 | goto end; | |
1671 | } | |
1672 | log4j_config = tmp_config; | |
1673 | break; | |
1674 | case LTTNG_DOMAIN_PYTHON: | |
1675 | if (python_config) { | |
1676 | ERR("Only one -p option is permitted per command"); | |
1677 | ret = CMD_ERROR; | |
1678 | goto end; | |
1679 | } | |
1680 | python_config = tmp_config; | |
1681 | break; | |
1682 | case LTTNG_DOMAIN_NONE: | |
1683 | default: | |
1684 | ret = CMD_ERROR; | |
c7dfbc0e JR |
1685 | goto end; |
1686 | } | |
721b94fe JR |
1687 | |
1688 | tmp_config = NULL; | |
1689 | ||
c7dfbc0e JR |
1690 | poptFreeContext(pc); |
1691 | pc = NULL; | |
721b94fe JR |
1692 | } |
1693 | ||
c7dfbc0e JR |
1694 | if (kernel_config) { |
1695 | command_ret = enable_events(session_name, kernel_config); | |
721b94fe JR |
1696 | if (command_ret) { |
1697 | success = 0; | |
1698 | } | |
c7dfbc0e JR |
1699 | } |
1700 | ||
1701 | if (ust_config) { | |
1702 | command_ret = enable_events(session_name, ust_config); | |
1703 | if (command_ret) { | |
1704 | success = 0; | |
1705 | } | |
1706 | } | |
1707 | ||
1708 | if (jul_config) { | |
1709 | command_ret = enable_events(session_name, jul_config); | |
1710 | if (command_ret) { | |
1711 | success = 0; | |
1712 | } | |
1713 | } | |
1714 | ||
1715 | if (log4j_config) { | |
1716 | command_ret = enable_events(session_name, log4j_config); | |
1717 | if (command_ret) { | |
1718 | success = 0; | |
1719 | } | |
1720 | } | |
1721 | ||
1722 | if (python_config) { | |
1723 | command_ret = enable_events(session_name, python_config); | |
1724 | if (command_ret) { | |
1725 | success = 0; | |
1726 | } | |
1727 | } | |
721b94fe | 1728 | |
89476427 JRJ |
1729 | mi_closing: |
1730 | /* Mi closing */ | |
1731 | if (lttng_opt_mi) { | |
c7dfbc0e JR |
1732 | /* Close domains and output element */ |
1733 | ret = mi_lttng_close_multi_element(writer, 2); | |
89476427 JRJ |
1734 | if (ret) { |
1735 | ret = CMD_ERROR; | |
1736 | goto end; | |
1737 | } | |
1738 | ||
1739 | ret = mi_lttng_writer_write_element_bool(writer, | |
1740 | mi_lttng_element_command_success, success); | |
1741 | if (ret) { | |
1742 | ret = CMD_ERROR; | |
1743 | goto end; | |
1744 | } | |
1745 | ||
1746 | /* Command element close */ | |
1747 | ret = mi_lttng_writer_command_close(writer); | |
1748 | if (ret) { | |
1749 | ret = CMD_ERROR; | |
1750 | goto end; | |
1751 | } | |
1752 | } | |
f3ed775e DG |
1753 | |
1754 | end: | |
89476427 JRJ |
1755 | /* Mi clean-up */ |
1756 | if (writer && mi_lttng_writer_destroy(writer)) { | |
1757 | /* Preserve original error code */ | |
1758 | ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL; | |
1759 | } | |
1760 | ||
cd80958d | 1761 | if (opt_session_name == NULL) { |
c7dfbc0e | 1762 | printf("Here\n"); |
cd80958d DG |
1763 | free(session_name); |
1764 | } | |
1765 | ||
89476427 JRJ |
1766 | /* Overwrite ret if an error occurred in enable_events */ |
1767 | ret = command_ret ? command_ret : ret; | |
1768 | ||
c7dfbc0e | 1769 | free(args_tuple_list); |
721b94fe | 1770 | free(jul_config); |
c7dfbc0e | 1771 | free(kernel_config); |
721b94fe JR |
1772 | free(log4j_config); |
1773 | free(python_config); | |
c7dfbc0e | 1774 | free(ust_config); |
ca1c3607 | 1775 | poptFreeContext(pc); |
f3ed775e DG |
1776 | return ret; |
1777 | } |