Commit | Line | Data |
---|---|---|
f3ed775e | 1 | /* |
21cf9b6b | 2 | * Copyright (C) 2011 EfficiOS Inc. |
f3ed775e | 3 | * |
ab5be9fa | 4 | * SPDX-License-Identifier: GPL-2.0-only |
f3ed775e | 5 | * |
f3ed775e DG |
6 | */ |
7 | ||
6c1c0768 | 8 | #define _LGPL_SOURCE |
28ab034a JG |
9 | #include <common/compat/getenv.hpp> |
10 | #include <common/compat/string.hpp> | |
8884bbc1 | 11 | #include <common/make-unique-wrapper.hpp> |
27b5aa57 | 12 | #include <common/scope-exit.hpp> |
28ab034a JG |
13 | #include <common/sessiond-comm/sessiond-comm.hpp> |
14 | #include <common/string-utils/string-utils.hpp> | |
15 | #include <common/utils.hpp> | |
16 | ||
17 | #include <ctype.h> | |
18 | #include <inttypes.h> | |
f3ed775e | 19 | #include <popt.h> |
bf1413aa | 20 | #include <sstream> |
f3ed775e DG |
21 | #include <stdio.h> |
22 | #include <stdlib.h> | |
bf1413aa | 23 | #include <string> |
f3ed775e DG |
24 | #include <sys/stat.h> |
25 | #include <sys/types.h> | |
26 | #include <unistd.h> | |
bf1413aa | 27 | #include <vector> |
f3ed775e | 28 | |
89476427 | 29 | /* Mi dependancy */ |
28ab034a JG |
30 | #include "../command.hpp" |
31 | #include "../loglevel.hpp" | |
32 | #include "../uprobe.hpp" | |
33 | ||
c9e313bc | 34 | #include <common/mi-lttng.hpp> |
89476427 | 35 | |
c9e313bc SM |
36 | #include <lttng/domain-internal.hpp> |
37 | #include <lttng/event-internal.hpp> | |
7e8f2e9c | 38 | |
8ab7c0d9 | 39 | #if (LTTNG_SYMBOL_NAME_LEN == 256) |
28ab034a | 40 | #define LTTNG_SYMBOL_NAME_LEN_SCANF_IS_A_BROKEN_API "255" |
8ab7c0d9 MD |
41 | #endif |
42 | ||
d512607f | 43 | namespace { |
8884bbc1 JG |
44 | void _mi_lttng_writer_deleter_func(mi_writer *writer) |
45 | { | |
46 | if (writer && mi_lttng_writer_destroy(writer)) { | |
47 | LTTNG_THROW_ERROR("Failed to destroy mi_writer instance"); | |
48 | } | |
49 | } | |
50 | ||
51 | using mi_writer_uptr = std::unique_ptr< | |
52 | mi_writer, | |
53 | lttng::memory::create_deleter_class<mi_writer, _mi_lttng_writer_deleter_func>::deleter>; | |
bf1413aa | 54 | using event_rule_patterns = std::vector<std::string>; |
d512607f JG |
55 | |
56 | int opt_event_type; | |
57 | const char *opt_loglevel; | |
58 | int opt_loglevel_type; | |
59 | int opt_kernel; | |
60 | char *opt_session_name; | |
61 | int opt_userspace; | |
62 | int opt_jul; | |
63 | int opt_log4j; | |
64 | int opt_python; | |
65 | int opt_enable_all; | |
66 | char *opt_probe; | |
67 | char *opt_userspace_probe; | |
68 | char *opt_function; | |
69 | char *opt_channel_name; | |
70 | char *opt_filter; | |
71 | char *opt_exclude; | |
f3ed775e | 72 | |
8884bbc1 JG |
73 | struct lttng_handle *handle; |
74 | mi_writer_uptr writer; | |
75 | ||
4fc83d94 PP |
76 | #ifdef LTTNG_EMBED_HELP |
77 | static const char help_msg[] = | |
78 | #include <lttng-enable-event.1.h> | |
28ab034a | 79 | ; |
4fc83d94 PP |
80 | #endif |
81 | ||
f3ed775e DG |
82 | enum { |
83 | OPT_HELP = 1, | |
f3ed775e | 84 | OPT_TRACEPOINT, |
cf0e5467 | 85 | OPT_PROBE, |
dcabc190 | 86 | OPT_USERSPACE_PROBE, |
f3ed775e | 87 | OPT_FUNCTION, |
a54bd42d | 88 | OPT_SYSCALL, |
eeac7d46 | 89 | OPT_USERSPACE, |
0cda4f28 MD |
90 | OPT_LOGLEVEL, |
91 | OPT_LOGLEVEL_ONLY, | |
679b4943 | 92 | OPT_LIST_OPTIONS, |
53a80697 | 93 | OPT_FILTER, |
fac3366c | 94 | OPT_EXCLUDE, |
f3ed775e DG |
95 | }; |
96 | ||
d512607f | 97 | struct poptOption long_options[] = { |
f3ed775e | 98 | /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */ |
cd9adb8b JG |
99 | { "help", 'h', POPT_ARG_NONE, nullptr, OPT_HELP, nullptr, nullptr }, |
100 | { "session", 's', POPT_ARG_STRING, &opt_session_name, 0, nullptr, nullptr }, | |
101 | { "all", 'a', POPT_ARG_VAL, &opt_enable_all, 1, nullptr, nullptr }, | |
102 | { "channel", 'c', POPT_ARG_STRING, &opt_channel_name, 0, nullptr, nullptr }, | |
103 | { "kernel", 'k', POPT_ARG_VAL, &opt_kernel, 1, nullptr, nullptr }, | |
104 | { "userspace", 'u', POPT_ARG_NONE, nullptr, OPT_USERSPACE, nullptr, nullptr }, | |
105 | { "jul", 'j', POPT_ARG_VAL, &opt_jul, 1, nullptr, nullptr }, | |
106 | { "log4j", 'l', POPT_ARG_VAL, &opt_log4j, 1, nullptr, nullptr }, | |
107 | { "python", 'p', POPT_ARG_VAL, &opt_python, 1, nullptr, nullptr }, | |
108 | { "tracepoint", 0, POPT_ARG_NONE, nullptr, OPT_TRACEPOINT, nullptr, nullptr }, | |
109 | { "probe", 0, POPT_ARG_STRING, &opt_probe, OPT_PROBE, nullptr, nullptr }, | |
110 | { "userspace-probe", | |
111 | 0, | |
112 | POPT_ARG_STRING, | |
113 | &opt_userspace_probe, | |
114 | OPT_USERSPACE_PROBE, | |
115 | nullptr, | |
116 | nullptr }, | |
117 | { "function", 0, POPT_ARG_STRING, &opt_function, OPT_FUNCTION, nullptr, nullptr }, | |
118 | { "syscall", 0, POPT_ARG_NONE, nullptr, OPT_SYSCALL, nullptr, nullptr }, | |
119 | { "loglevel", 0, POPT_ARG_STRING, nullptr, OPT_LOGLEVEL, nullptr, nullptr }, | |
120 | { "loglevel-only", 0, POPT_ARG_STRING, nullptr, OPT_LOGLEVEL_ONLY, nullptr, nullptr }, | |
121 | { "list-options", 0, POPT_ARG_NONE, nullptr, OPT_LIST_OPTIONS, nullptr, nullptr }, | |
122 | { "filter", 'f', POPT_ARG_STRING, &opt_filter, OPT_FILTER, nullptr, nullptr }, | |
123 | { "exclude", 'x', POPT_ARG_STRING, &opt_exclude, OPT_EXCLUDE, nullptr, nullptr }, | |
124 | { nullptr, 0, 0, nullptr, 0, nullptr, nullptr } | |
f3ed775e DG |
125 | }; |
126 | ||
0d63dd19 | 127 | /* |
6181537c | 128 | * Parse probe options. |
0d63dd19 | 129 | */ |
d512607f | 130 | int parse_probe_opts(struct lttng_event *ev, char *opt) |
0d63dd19 | 131 | { |
49d4e302 JRJ |
132 | int ret = CMD_SUCCESS; |
133 | int match; | |
8ff0bbd0 | 134 | char s_hex[19]; |
28ab034a | 135 | #define S_HEX_LEN_SCANF_IS_A_BROKEN_API "18" /* 18 is (19 - 1) (\0 is extra) */ |
0d63dd19 DG |
136 | char name[LTTNG_SYMBOL_NAME_LEN]; |
137 | ||
cd9adb8b | 138 | if (opt == nullptr) { |
49d4e302 | 139 | ret = CMD_ERROR; |
8f0d098b | 140 | goto end; |
0d63dd19 DG |
141 | } |
142 | ||
143 | /* Check for symbol+offset */ | |
28ab034a JG |
144 | match = sscanf(opt, |
145 | "%" LTTNG_SYMBOL_NAME_LEN_SCANF_IS_A_BROKEN_API | |
146 | "[^'+']+%" S_HEX_LEN_SCANF_IS_A_BROKEN_API "s", | |
147 | name, | |
148 | s_hex); | |
49d4e302 | 149 | if (match == 2) { |
7d29a247 | 150 | strncpy(ev->attr.probe.symbol_name, name, LTTNG_SYMBOL_NAME_LEN); |
99497cd0 | 151 | ev->attr.probe.symbol_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; |
cf0e5467 | 152 | DBG("probe symbol %s", ev->attr.probe.symbol_name); |
9d035200 | 153 | if (*s_hex == '\0') { |
8ff0bbd0 | 154 | ERR("Invalid probe offset %s", s_hex); |
49d4e302 | 155 | ret = CMD_ERROR; |
8f0d098b | 156 | goto end; |
0d63dd19 | 157 | } |
cd9adb8b | 158 | ev->attr.probe.offset = strtoull(s_hex, nullptr, 0); |
cf0e5467 | 159 | DBG("probe offset %" PRIu64, ev->attr.probe.offset); |
3000dc78 | 160 | ev->attr.probe.addr = 0; |
8f0d098b MD |
161 | goto end; |
162 | } | |
163 | ||
164 | /* Check for symbol */ | |
88c91427 | 165 | if (isalpha(name[0]) || name[0] == '_') { |
28ab034a | 166 | match = sscanf(opt, "%" LTTNG_SYMBOL_NAME_LEN_SCANF_IS_A_BROKEN_API "s", name); |
49d4e302 | 167 | if (match == 1) { |
8f0d098b | 168 | strncpy(ev->attr.probe.symbol_name, name, LTTNG_SYMBOL_NAME_LEN); |
99497cd0 | 169 | ev->attr.probe.symbol_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; |
8f0d098b MD |
170 | DBG("probe symbol %s", ev->attr.probe.symbol_name); |
171 | ev->attr.probe.offset = 0; | |
172 | DBG("probe offset %" PRIu64, ev->attr.probe.offset); | |
173 | ev->attr.probe.addr = 0; | |
174 | goto end; | |
175 | } | |
0d63dd19 DG |
176 | } |
177 | ||
178 | /* Check for address */ | |
49d4e302 JRJ |
179 | match = sscanf(opt, "%" S_HEX_LEN_SCANF_IS_A_BROKEN_API "s", s_hex); |
180 | if (match > 0) { | |
531be721 FD |
181 | /* |
182 | * Return an error if the first character of the tentative | |
183 | * address is NULL or not a digit. It can be "0" if the address | |
184 | * is in hexadecimal and can be 1 to 9 if it's in decimal. | |
185 | */ | |
186 | if (*s_hex == '\0' || !isdigit(*s_hex)) { | |
13c25a5c | 187 | ERR("Invalid probe description `%s`", s_hex); |
49d4e302 | 188 | ret = CMD_ERROR; |
8f0d098b | 189 | goto end; |
0d63dd19 | 190 | } |
cd9adb8b | 191 | ev->attr.probe.addr = strtoull(s_hex, nullptr, 0); |
cf0e5467 | 192 | DBG("probe addr %" PRIu64, ev->attr.probe.addr); |
3000dc78 DG |
193 | ev->attr.probe.offset = 0; |
194 | memset(ev->attr.probe.symbol_name, 0, LTTNG_SYMBOL_NAME_LEN); | |
8f0d098b | 195 | goto end; |
0d63dd19 DG |
196 | } |
197 | ||
198 | /* No match */ | |
49d4e302 | 199 | ret = CMD_ERROR; |
0d63dd19 | 200 | |
8f0d098b | 201 | end: |
0d63dd19 DG |
202 | return ret; |
203 | } | |
204 | ||
d512607f | 205 | const char *print_channel_name(const char *name) |
85076754 | 206 | { |
28ab034a | 207 | return name ?: DEFAULT_CHANNEL_NAME; |
85076754 MD |
208 | } |
209 | ||
d512607f | 210 | const char *print_raw_channel_name(const char *name) |
85076754 | 211 | { |
28ab034a | 212 | return name ?: "<default>"; |
85076754 MD |
213 | } |
214 | ||
89476427 JRJ |
215 | /* |
216 | * Mi print exlcusion list | |
217 | */ | |
d512607f | 218 | int mi_print_exclusion(const struct lttng_dynamic_pointer_array *exclusions) |
89476427 | 219 | { |
e358ddd5 JG |
220 | int ret; |
221 | size_t i; | |
222 | const size_t count = lttng_dynamic_pointer_array_get_count(exclusions); | |
89476427 | 223 | |
a0377dfe | 224 | LTTNG_ASSERT(writer); |
89476427 JRJ |
225 | |
226 | if (count == 0) { | |
227 | ret = 0; | |
228 | goto end; | |
229 | } | |
e358ddd5 | 230 | |
8884bbc1 | 231 | ret = mi_lttng_writer_open_element(writer.get(), config_element_exclusions); |
89476427 JRJ |
232 | if (ret) { |
233 | goto end; | |
234 | } | |
235 | ||
236 | for (i = 0; i < count; i++) { | |
28ab034a JG |
237 | const char *exclusion = |
238 | (const char *) lttng_dynamic_pointer_array_get_pointer(exclusions, i); | |
e358ddd5 | 239 | |
28ab034a | 240 | ret = mi_lttng_writer_write_element_string( |
8884bbc1 | 241 | writer.get(), config_element_exclusion, exclusion); |
89476427 JRJ |
242 | if (ret) { |
243 | goto end; | |
244 | } | |
245 | } | |
246 | ||
247 | /* Close exclusions element */ | |
8884bbc1 | 248 | ret = mi_lttng_writer_close_element(writer.get()); |
89476427 JRJ |
249 | |
250 | end: | |
251 | return ret; | |
252 | } | |
253 | ||
9c48cab3 JI |
254 | /* |
255 | * Return allocated string for pretty-printing exclusion names. | |
256 | */ | |
d512607f | 257 | char *print_exclusions(const struct lttng_dynamic_pointer_array *exclusions) |
9c48cab3 JI |
258 | { |
259 | int length = 0; | |
e358ddd5 | 260 | size_t i; |
bfe36393 | 261 | const char preamble[] = " excluding "; |
9c48cab3 | 262 | char *ret; |
e358ddd5 | 263 | const size_t count = lttng_dynamic_pointer_array_get_count(exclusions); |
9c48cab3 JI |
264 | |
265 | if (count == 0) { | |
266 | return strdup(""); | |
267 | } | |
268 | ||
e358ddd5 | 269 | /* Calculate total required length. */ |
9c48cab3 | 270 | for (i = 0; i < count; i++) { |
28ab034a JG |
271 | const char *exclusion = |
272 | (const char *) lttng_dynamic_pointer_array_get_pointer(exclusions, i); | |
e358ddd5 JG |
273 | |
274 | length += strlen(exclusion) + 4; | |
9c48cab3 JI |
275 | } |
276 | ||
bfe36393 | 277 | length += sizeof(preamble); |
64803277 | 278 | ret = calloc<char>(length); |
fd591b18 | 279 | if (!ret) { |
cd9adb8b | 280 | return nullptr; |
fd591b18 | 281 | } |
e358ddd5 | 282 | |
9c48cab3 JI |
283 | strncpy(ret, preamble, length); |
284 | for (i = 0; i < count; i++) { | |
28ab034a JG |
285 | const char *exclusion = |
286 | (const char *) lttng_dynamic_pointer_array_get_pointer(exclusions, i); | |
e358ddd5 | 287 | |
13c25a5c | 288 | strcat(ret, "`"); |
e358ddd5 | 289 | strcat(ret, exclusion); |
13c25a5c | 290 | strcat(ret, "`"); |
9c48cab3 | 291 | if (i != count - 1) { |
9f449915 | 292 | strcat(ret, ", "); |
9c48cab3 JI |
293 | } |
294 | } | |
89476427 | 295 | |
9c48cab3 JI |
296 | return ret; |
297 | } | |
298 | ||
bf1413aa | 299 | int check_exclusion_subsets(const char *pattern, const char *exclusion) |
748bde76 | 300 | { |
9f449915 PP |
301 | bool warn = false; |
302 | int ret = 0; | |
bf1413aa | 303 | const char *e = pattern; |
9f449915 PP |
304 | const char *x = exclusion; |
305 | ||
306 | /* Scan both the excluder and the event letter by letter */ | |
307 | while (true) { | |
308 | if (*e == '\\') { | |
309 | if (*x != *e) { | |
310 | warn = true; | |
311 | goto end; | |
312 | } | |
748bde76 | 313 | |
9f449915 PP |
314 | e++; |
315 | x++; | |
316 | goto cmp_chars; | |
317 | } | |
318 | ||
319 | if (*x == '*') { | |
320 | /* Event is a subset of the excluder */ | |
13c25a5c JG |
321 | ERR("Exclusion pattern `%s` excludes all events from `%s`", |
322 | exclusion, | |
323 | pattern); | |
9f449915 PP |
324 | goto error; |
325 | } | |
326 | ||
327 | if (*e == '*') { | |
328 | /* | |
329 | * Reached the end of the event name before the | |
330 | * end of the exclusion: this is valid. | |
331 | */ | |
332 | goto end; | |
333 | } | |
334 | ||
28ab034a | 335 | cmp_chars: |
9f449915 PP |
336 | if (*x != *e) { |
337 | warn = true; | |
338 | break; | |
339 | } | |
340 | ||
341 | x++; | |
342 | e++; | |
748bde76 JI |
343 | } |
344 | ||
9f449915 | 345 | goto end; |
748bde76 | 346 | |
9f449915 PP |
347 | error: |
348 | ret = -1; | |
748bde76 | 349 | |
9f449915 PP |
350 | end: |
351 | if (warn) { | |
13c25a5c | 352 | WARN("Exclusion pattern `%s` does not exclude any event from `%s`", |
28ab034a | 353 | exclusion, |
bf1413aa | 354 | pattern); |
9f449915 | 355 | } |
748bde76 | 356 | |
9f449915 PP |
357 | return ret; |
358 | } | |
359 | ||
bf1413aa | 360 | int create_exclusion_list_and_validate(const char *pattern, |
d512607f JG |
361 | const char *exclusions_arg, |
362 | struct lttng_dynamic_pointer_array *exclusions) | |
b03a81fb SM |
363 | { |
364 | int ret = 0; | |
b03a81fb SM |
365 | |
366 | /* Split exclusions. */ | |
e358ddd5 JG |
367 | ret = strutils_split(exclusions_arg, ',', true, exclusions); |
368 | if (ret < 0) { | |
b03a81fb SM |
369 | goto error; |
370 | } | |
371 | ||
bf1413aa | 372 | if (validate_exclusion_list(pattern, exclusions) != 0) { |
b03a81fb SM |
373 | goto error; |
374 | } | |
375 | ||
748bde76 | 376 | goto end; |
9f449915 | 377 | |
748bde76 | 378 | error: |
9f449915 | 379 | ret = -1; |
e358ddd5 | 380 | lttng_dynamic_pointer_array_reset(exclusions); |
9f449915 | 381 | |
748bde76 | 382 | end: |
748bde76 JI |
383 | return ret; |
384 | } | |
502bbe89 | 385 | |
d512607f JG |
386 | void warn_on_truncated_exclusion_names(const struct lttng_dynamic_pointer_array *exclusions, |
387 | int *warn) | |
502bbe89 | 388 | { |
e358ddd5 JG |
389 | size_t i; |
390 | const size_t num_exclusions = lttng_dynamic_pointer_array_get_count(exclusions); | |
391 | ||
392 | for (i = 0; i < num_exclusions; i++) { | |
28ab034a JG |
393 | const char *const exclusion = |
394 | (const char *) lttng_dynamic_pointer_array_get_pointer(exclusions, i); | |
502bbe89 | 395 | |
e358ddd5 | 396 | if (strlen(exclusion) >= LTTNG_SYMBOL_NAME_LEN) { |
28ab034a | 397 | WARN("Event exclusion \"%s\" will be truncated", exclusion); |
502bbe89 PP |
398 | *warn = 1; |
399 | } | |
400 | } | |
401 | } | |
402 | ||
f3ed775e | 403 | /* |
6181537c | 404 | * Enabling event using the lttng API. |
89476427 | 405 | * Note: in case of error only the last error code will be return. |
f3ed775e | 406 | */ |
bf1413aa | 407 | int enable_events(const std::string& session_name, const event_rule_patterns& patterns) |
f3ed775e | 408 | { |
89476427 JRJ |
409 | int ret = CMD_SUCCESS, command_ret = CMD_SUCCESS; |
410 | int error_holder = CMD_SUCCESS, warn = 0, error = 0, success = 1; | |
bf1413aa | 411 | char *channel_name = nullptr; |
91744e14 | 412 | struct lttng_event *ev; |
e358ddd5 JG |
413 | struct lttng_domain dom = {}; |
414 | struct lttng_dynamic_pointer_array exclusions; | |
cd9adb8b | 415 | struct lttng_userspace_probe_location *uprobe_loc = nullptr; |
f3ed775e | 416 | |
cd9adb8b | 417 | lttng_dynamic_pointer_array_init(&exclusions, nullptr); |
441c16a7 | 418 | |
91744e14 FD |
419 | ev = lttng_event_create(); |
420 | if (!ev) { | |
421 | ret = CMD_ERROR; | |
422 | goto error; | |
423 | } | |
424 | ||
53a80697 | 425 | if (opt_kernel) { |
67b58630 | 426 | if (opt_loglevel) { |
13c25a5c | 427 | WARN("Kernel log levels are not supported"); |
67b58630 | 428 | } |
53a80697 MD |
429 | } |
430 | ||
7d29a247 DG |
431 | /* Create lttng domain */ |
432 | if (opt_kernel) { | |
433 | dom.type = LTTNG_DOMAIN_KERNEL; | |
7972aab2 | 434 | dom.buf_type = LTTNG_BUFFER_GLOBAL; |
d78d6610 | 435 | } else if (opt_userspace) { |
2bdd86d4 | 436 | dom.type = LTTNG_DOMAIN_UST; |
7972aab2 | 437 | /* Default. */ |
8692d4e5 | 438 | dom.buf_type = LTTNG_BUFFER_PER_UID; |
b9dfb167 DG |
439 | } else if (opt_jul) { |
440 | dom.type = LTTNG_DOMAIN_JUL; | |
441 | /* Default. */ | |
442 | dom.buf_type = LTTNG_BUFFER_PER_UID; | |
5cdb6027 DG |
443 | } else if (opt_log4j) { |
444 | dom.type = LTTNG_DOMAIN_LOG4J; | |
445 | /* Default. */ | |
446 | dom.buf_type = LTTNG_BUFFER_PER_UID; | |
0e115563 DG |
447 | } else if (opt_python) { |
448 | dom.type = LTTNG_DOMAIN_PYTHON; | |
449 | /* Default. */ | |
450 | dom.buf_type = LTTNG_BUFFER_PER_UID; | |
6181537c | 451 | } else { |
3ecec76a | 452 | /* Checked by the caller. */ |
a0377dfe | 453 | abort(); |
2bdd86d4 | 454 | } |
7d29a247 | 455 | |
8b08c525 AB |
456 | if (opt_exclude) { |
457 | switch (dom.type) { | |
458 | case LTTNG_DOMAIN_KERNEL: | |
459 | case LTTNG_DOMAIN_JUL: | |
460 | case LTTNG_DOMAIN_LOG4J: | |
461 | case LTTNG_DOMAIN_PYTHON: | |
13c25a5c | 462 | ERR("Event name exclusions are not supported for %s event rules", |
28ab034a | 463 | lttng_domain_type_str(dom.type)); |
8b08c525 AB |
464 | ret = CMD_ERROR; |
465 | goto error; | |
466 | case LTTNG_DOMAIN_UST: | |
467 | /* Exclusions supported */ | |
468 | break; | |
469 | default: | |
a0377dfe | 470 | abort(); |
8b08c525 | 471 | } |
d5dd17fd JI |
472 | } |
473 | ||
4fd2697f FD |
474 | /* |
475 | * Adding a filter to a probe, function or userspace-probe would be | |
476 | * denied by the kernel tracer as it's not supported at the moment. We | |
477 | * do an early check here to warn the user. | |
478 | */ | |
479 | if (opt_filter && opt_kernel) { | |
480 | switch (opt_event_type) { | |
481 | case LTTNG_EVENT_ALL: | |
482 | case LTTNG_EVENT_TRACEPOINT: | |
483 | case LTTNG_EVENT_SYSCALL: | |
484 | break; | |
485 | case LTTNG_EVENT_PROBE: | |
486 | case LTTNG_EVENT_USERSPACE_PROBE: | |
487 | case LTTNG_EVENT_FUNCTION: | |
13c25a5c | 488 | ERR("Filter expressions are not supported for %s event rules", |
28ab034a | 489 | get_event_type_str((lttng_event_type) opt_event_type)); |
4fd2697f FD |
490 | ret = CMD_ERROR; |
491 | goto error; | |
492 | default: | |
493 | ret = CMD_UNDEFINED; | |
494 | goto error; | |
495 | } | |
496 | } | |
497 | ||
85076754 | 498 | channel_name = opt_channel_name; |
ae856491 | 499 | |
bf1413aa | 500 | handle = lttng_create_handle(session_name.c_str(), &dom); |
cd9adb8b | 501 | if (handle == nullptr) { |
cd80958d DG |
502 | ret = -1; |
503 | goto error; | |
504 | } | |
1aef21b6 | 505 | |
89476427 JRJ |
506 | /* Prepare Mi */ |
507 | if (lttng_opt_mi) { | |
508 | /* Open a events element */ | |
8884bbc1 | 509 | ret = mi_lttng_writer_open_element(writer.get(), config_element_events); |
89476427 JRJ |
510 | if (ret) { |
511 | ret = CMD_ERROR; | |
512 | goto error; | |
513 | } | |
514 | } | |
515 | ||
bf1413aa | 516 | for (const auto& pattern : patterns) { |
6181537c | 517 | /* Copy name and type of the event */ |
bf1413aa | 518 | strncpy(ev->name, pattern.c_str(), LTTNG_SYMBOL_NAME_LEN); |
91744e14 | 519 | ev->name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; |
48a40005 | 520 | ev->type = (lttng_event_type) opt_event_type; |
6181537c | 521 | |
f3ed775e DG |
522 | /* Kernel tracer action */ |
523 | if (opt_kernel) { | |
13c25a5c | 524 | DBG_FMT("Enabling kernel event rule: pattern=`{}`, channel_name=`{}`", |
bf1413aa JG |
525 | pattern, |
526 | print_channel_name(channel_name)); | |
f3ed775e DG |
527 | |
528 | switch (opt_event_type) { | |
28ab034a | 529 | case LTTNG_EVENT_ALL: /* Enable tracepoints and syscalls */ |
29c62722 | 530 | /* If event name differs from *, select tracepoint. */ |
5c7248cd | 531 | if (strcmp(ev->name, "*") != 0) { |
91744e14 | 532 | ev->type = LTTNG_EVENT_TRACEPOINT; |
29c62722 MD |
533 | } |
534 | break; | |
e6ddca71 | 535 | case LTTNG_EVENT_TRACEPOINT: |
f3ed775e | 536 | break; |
7d29a247 | 537 | case LTTNG_EVENT_PROBE: |
91744e14 | 538 | ret = parse_probe_opts(ev, opt_probe); |
49d4e302 | 539 | if (ret) { |
cf0e5467 | 540 | ERR("Unable to parse probe options"); |
91744e14 | 541 | ret = CMD_ERROR; |
0d63dd19 DG |
542 | goto error; |
543 | } | |
f3ed775e | 544 | break; |
dcabc190 | 545 | case LTTNG_EVENT_USERSPACE_PROBE: |
a0377dfe | 546 | LTTNG_ASSERT(ev->type == LTTNG_EVENT_USERSPACE_PROBE); |
0c4e727b JR |
547 | |
548 | ret = parse_userspace_probe_opts(opt_userspace_probe, &uprobe_loc); | |
dcabc190 | 549 | if (ret) { |
36aa2f64 FD |
550 | switch (ret) { |
551 | case CMD_UNSUPPORTED: | |
552 | /* | |
553 | * Error message describing | |
554 | * what is not supported was | |
555 | * printed in the function. | |
556 | */ | |
557 | break; | |
558 | case CMD_ERROR: | |
559 | default: | |
13c25a5c | 560 | ERR("Unable to parse user space probe options"); |
36aa2f64 FD |
561 | break; |
562 | } | |
dcabc190 FD |
563 | goto error; |
564 | } | |
0c4e727b JR |
565 | |
566 | ret = lttng_event_set_userspace_probe_location(ev, uprobe_loc); | |
567 | if (ret) { | |
568 | WARN("Failed to set probe location on event"); | |
569 | ret = CMD_ERROR; | |
570 | goto error; | |
571 | } | |
572 | ||
573 | /* Ownership of the uprobe location was transferred to the event. */ | |
cd9adb8b | 574 | uprobe_loc = nullptr; |
dcabc190 | 575 | break; |
f3ed775e | 576 | case LTTNG_EVENT_FUNCTION: |
91744e14 | 577 | ret = parse_probe_opts(ev, opt_function); |
49d4e302 | 578 | if (ret) { |
8f0d098b | 579 | ERR("Unable to parse function probe options"); |
91744e14 | 580 | ret = CMD_ERROR; |
8f0d098b MD |
581 | goto error; |
582 | } | |
583 | break; | |
a54bd42d | 584 | case LTTNG_EVENT_SYSCALL: |
91744e14 | 585 | ev->type = LTTNG_EVENT_SYSCALL; |
c6aa2d41 | 586 | break; |
f3ed775e | 587 | default: |
1ab1ea0b | 588 | ret = CMD_UNDEFINED; |
f3ed775e DG |
589 | goto error; |
590 | } | |
0cda4f28 | 591 | |
0cda4f28 | 592 | /* kernel loglevels not implemented */ |
91744e14 | 593 | ev->loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; |
28ab034a | 594 | } else if (opt_userspace) { /* User-space tracer action */ |
13c25a5c JG |
595 | DBG_FMT("Enabling user space event rule: pattern=`{}`, channel_name=`{}`, log_level=`{}`", |
596 | pattern.c_str(), | |
597 | print_channel_name(channel_name), | |
598 | opt_loglevel ?: "all"); | |
2bdd86d4 MD |
599 | |
600 | switch (opt_event_type) { | |
28ab034a | 601 | case LTTNG_EVENT_ALL: /* Default behavior is tracepoint */ |
2bdd86d4 MD |
602 | /* Fall-through */ |
603 | case LTTNG_EVENT_TRACEPOINT: | |
e4baff1e | 604 | /* Copy name and type of the event */ |
91744e14 | 605 | ev->type = LTTNG_EVENT_TRACEPOINT; |
bf1413aa | 606 | strncpy(ev->name, pattern.c_str(), LTTNG_SYMBOL_NAME_LEN); |
91744e14 | 607 | ev->name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; |
2bdd86d4 MD |
608 | break; |
609 | case LTTNG_EVENT_PROBE: | |
610 | case LTTNG_EVENT_FUNCTION: | |
2bdd86d4 | 611 | case LTTNG_EVENT_SYSCALL: |
dcabc190 | 612 | case LTTNG_EVENT_USERSPACE_PROBE: |
2bdd86d4 | 613 | default: |
13c25a5c JG |
614 | ERR("Instrumentation point type not supported for the %s domain", |
615 | lttng_domain_type_str(dom.type)); | |
4ce78777 | 616 | ret = CMD_UNSUPPORTED; |
2bdd86d4 MD |
617 | goto error; |
618 | } | |
0cda4f28 | 619 | |
7ed70bc9 | 620 | if (opt_exclude) { |
91744e14 | 621 | ev->exclusion = 1; |
28ab034a JG |
622 | if (opt_event_type != LTTNG_EVENT_ALL && |
623 | opt_event_type != LTTNG_EVENT_TRACEPOINT) { | |
d5dd17fd JI |
624 | ERR("Exclusion option can only be used with tracepoint events"); |
625 | ret = CMD_ERROR; | |
626 | goto error; | |
627 | } | |
e358ddd5 JG |
628 | /* Free previously allocated items. */ |
629 | lttng_dynamic_pointer_array_reset(&exclusions); | |
9f449915 | 630 | ret = create_exclusion_list_and_validate( |
bf1413aa | 631 | pattern.c_str(), opt_exclude, &exclusions); |
9f449915 PP |
632 | if (ret) { |
633 | ret = CMD_ERROR; | |
7ed70bc9 JI |
634 | goto error; |
635 | } | |
502bbe89 | 636 | |
28ab034a | 637 | warn_on_truncated_exclusion_names(&exclusions, &warn); |
7ed70bc9 JI |
638 | } |
639 | ||
48a40005 | 640 | ev->loglevel_type = (lttng_loglevel_type) opt_loglevel_type; |
ed7f4083 | 641 | if (opt_loglevel) { |
7e8f2e9c | 642 | enum lttng_loglevel loglevel; |
28ab034a JG |
643 | const int name_search_ret = |
644 | loglevel_name_to_value(opt_loglevel, &loglevel); | |
7e8f2e9c JG |
645 | |
646 | if (name_search_ret == -1) { | |
13c25a5c | 647 | ERR("Unknown log level `%s`", opt_loglevel); |
2f70b271 | 648 | ret = -LTTNG_ERR_INVALID; |
8005f29a MD |
649 | goto error; |
650 | } | |
7e8f2e9c JG |
651 | |
652 | ev->loglevel = (int) loglevel; | |
22e25b71 | 653 | } else { |
91744e14 | 654 | ev->loglevel = -1; |
ed7f4083 | 655 | } |
0e115563 | 656 | } else if (opt_jul || opt_log4j || opt_python) { |
b9dfb167 | 657 | if (opt_event_type != LTTNG_EVENT_ALL && |
28ab034a | 658 | opt_event_type != LTTNG_EVENT_TRACEPOINT) { |
13c25a5c JG |
659 | ERR("Instrumentation point type not supported for the %s domain", |
660 | lttng_domain_type_str(dom.type)); | |
b9dfb167 DG |
661 | ret = CMD_UNSUPPORTED; |
662 | goto error; | |
663 | } | |
b2064f54 | 664 | |
48a40005 | 665 | ev->loglevel_type = (lttng_loglevel_type) opt_loglevel_type; |
b2064f54 | 666 | if (opt_loglevel) { |
7e8f2e9c JG |
667 | int name_search_ret; |
668 | ||
5cdb6027 | 669 | if (opt_jul) { |
7e8f2e9c JG |
670 | enum lttng_loglevel_jul loglevel; |
671 | ||
28ab034a JG |
672 | name_search_ret = |
673 | loglevel_jul_name_to_value(opt_loglevel, &loglevel); | |
7e8f2e9c | 674 | ev->loglevel = (int) loglevel; |
5cdb6027 | 675 | } else if (opt_log4j) { |
7e8f2e9c JG |
676 | enum lttng_loglevel_log4j loglevel; |
677 | ||
28ab034a JG |
678 | name_search_ret = loglevel_log4j_name_to_value(opt_loglevel, |
679 | &loglevel); | |
7e8f2e9c | 680 | ev->loglevel = (int) loglevel; |
7fefa907 JG |
681 | } else { |
682 | /* python domain. */ | |
7e8f2e9c JG |
683 | enum lttng_loglevel_python loglevel; |
684 | ||
28ab034a JG |
685 | name_search_ret = loglevel_python_name_to_value( |
686 | opt_loglevel, &loglevel); | |
7e8f2e9c | 687 | ev->loglevel = (int) loglevel; |
5cdb6027 | 688 | } |
7e8f2e9c JG |
689 | |
690 | if (name_search_ret) { | |
13c25a5c | 691 | ERR("Unknown log level `%s`", opt_loglevel); |
b2064f54 DG |
692 | ret = -LTTNG_ERR_INVALID; |
693 | goto error; | |
694 | } | |
695 | } else { | |
5cdb6027 | 696 | if (opt_jul) { |
91744e14 | 697 | ev->loglevel = LTTNG_LOGLEVEL_JUL_ALL; |
5cdb6027 | 698 | } else if (opt_log4j) { |
91744e14 | 699 | ev->loglevel = LTTNG_LOGLEVEL_LOG4J_ALL; |
0e115563 | 700 | } else if (opt_python) { |
91744e14 | 701 | ev->loglevel = LTTNG_LOGLEVEL_PYTHON_DEBUG; |
5cdb6027 | 702 | } |
b2064f54 | 703 | } |
91744e14 | 704 | ev->type = LTTNG_EVENT_TRACEPOINT; |
bf1413aa | 705 | strncpy(ev->name, pattern.c_str(), LTTNG_SYMBOL_NAME_LEN); |
91744e14 | 706 | ev->name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; |
f3ed775e | 707 | } else { |
a0377dfe | 708 | abort(); |
f3ed775e DG |
709 | } |
710 | ||
025faf73 | 711 | if (!opt_filter) { |
28ab034a JG |
712 | command_ret = lttng_enable_event_with_exclusions( |
713 | handle, | |
714 | ev, | |
715 | channel_name, | |
cd9adb8b | 716 | nullptr, |
28ab034a JG |
717 | lttng_dynamic_pointer_array_get_count(&exclusions), |
718 | (char **) exclusions.array.buffer.data); | |
93928b6b | 719 | |
303ac4ed JG |
720 | auto exclusion_string = |
721 | lttng::make_unique_wrapper<char, lttng::memory::free>( | |
722 | print_exclusions(&exclusions)); | |
5ef79758 MD |
723 | if (!exclusion_string) { |
724 | PERROR("Cannot allocate exclusion_string"); | |
725 | error = 1; | |
726 | goto end; | |
727 | } | |
13c25a5c | 728 | |
89476427 | 729 | if (command_ret < 0) { |
025faf73 | 730 | /* Turn ret to positive value to handle the positive error code */ |
89476427 | 731 | switch (-command_ret) { |
025faf73 | 732 | case LTTNG_ERR_KERN_EVENT_EXIST: |
13c25a5c JG |
733 | WARN("Kernel event rule %s%s and attached to channel `%s` is already enabled (session `%s`)", |
734 | opt_enable_all ? "matching all events" : | |
735 | (std::string("with pattern `") + | |
736 | pattern + std::string("`")) | |
737 | .c_str(), | |
93928b6b | 738 | exclusion_string.get(), |
28ab034a | 739 | print_channel_name(channel_name), |
bf1413aa | 740 | session_name.c_str()); |
89476427 | 741 | warn = 1; |
025faf73 | 742 | break; |
45d5d421 CB |
743 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
744 | { | |
13c25a5c | 745 | ERR("Can't enable an event in a new domain for a session that has already been started once (session `%s`)", |
bf1413aa | 746 | session_name.c_str()); |
45d5d421 CB |
747 | error = 1; |
748 | break; | |
749 | } | |
dcabc190 | 750 | case LTTNG_ERR_SDT_PROBE_SEMAPHORE: |
13c25a5c | 751 | ERR("Failed to enable event rule with pattern `%s` and attached to channel `%s` as SDT probes guarded by semaphores are not supported (session `%s`)", |
bf1413aa | 752 | pattern.c_str(), |
28ab034a | 753 | print_channel_name(channel_name), |
bf1413aa | 754 | session_name.c_str()); |
dcabc190 FD |
755 | error = 1; |
756 | break; | |
025faf73 | 757 | default: |
13c25a5c JG |
758 | ERR("Failed to enable event rule %s%s and attached to channel `%s`: %s (session `%s`)", |
759 | opt_enable_all ? "matching all events" : | |
760 | (std::string("with pattern `") + | |
761 | pattern + std::string("`")) | |
762 | .c_str(), | |
93928b6b | 763 | exclusion_string.get(), |
28ab034a JG |
764 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME ? |
765 | print_raw_channel_name(channel_name) : | |
766 | print_channel_name(channel_name), | |
13c25a5c | 767 | lttng_strerror(command_ret), |
bf1413aa | 768 | session_name.c_str()); |
93928b6b JG |
769 | |
770 | if (opt_kernel) { | |
771 | print_kernel_tracer_status_error(); | |
772 | } | |
773 | ||
89476427 | 774 | error = 1; |
025faf73 DG |
775 | break; |
776 | } | |
89476427 | 777 | error_holder = command_ret; |
025faf73 | 778 | } else { |
8274eeba AB |
779 | switch (dom.type) { |
780 | case LTTNG_DOMAIN_KERNEL: | |
781 | case LTTNG_DOMAIN_UST: | |
13c25a5c JG |
782 | { |
783 | MSG("Enabled %s event rule %s%s and attached to channel `%s`", | |
28ab034a | 784 | lttng_domain_type_str(dom.type), |
13c25a5c JG |
785 | opt_enable_all ? "matching all events" : |
786 | (std::string("with pattern `") + | |
787 | pattern + std::string("`")) | |
788 | .c_str(), | |
93928b6b | 789 | exclusion_string.get(), |
28ab034a | 790 | print_channel_name(channel_name)); |
8274eeba | 791 | break; |
13c25a5c | 792 | } |
8274eeba AB |
793 | case LTTNG_DOMAIN_JUL: |
794 | case LTTNG_DOMAIN_LOG4J: | |
795 | case LTTNG_DOMAIN_PYTHON: | |
796 | /* | |
797 | * Don't print the default channel | |
798 | * name for agent domains. | |
799 | */ | |
13c25a5c | 800 | MSG("Enabled %s event rule %s%s", |
28ab034a | 801 | lttng_domain_type_str(dom.type), |
13c25a5c JG |
802 | opt_enable_all ? "matching all events" : |
803 | (std::string("with pattern `") + | |
804 | pattern + std::string("`")) | |
805 | .c_str(), | |
93928b6b | 806 | exclusion_string.get()); |
8274eeba AB |
807 | break; |
808 | default: | |
a0377dfe | 809 | abort(); |
49ceaa70 | 810 | } |
42224349 | 811 | } |
13c25a5c | 812 | } else { |
89476427 | 813 | /* Filter present */ |
91744e14 | 814 | ev->filter = 1; |
89476427 | 815 | |
28ab034a JG |
816 | command_ret = lttng_enable_event_with_exclusions( |
817 | handle, | |
818 | ev, | |
819 | channel_name, | |
820 | opt_filter, | |
821 | lttng_dynamic_pointer_array_get_count(&exclusions), | |
822 | (char **) exclusions.array.buffer.data); | |
93928b6b | 823 | |
303ac4ed JG |
824 | auto exclusion_string = |
825 | lttng::make_unique_wrapper<char, lttng::memory::free>( | |
826 | print_exclusions(&exclusions)); | |
5ef79758 | 827 | if (!exclusion_string) { |
13c25a5c | 828 | PERROR("Failed allocate exclusion string"); |
5ef79758 MD |
829 | error = 1; |
830 | goto end; | |
831 | } | |
89476427 JRJ |
832 | if (command_ret < 0) { |
833 | switch (-command_ret) { | |
7671f53c | 834 | case LTTNG_ERR_FILTER_EXIST: |
13c25a5c JG |
835 | WARN("An event rule %s%s and filter expression `%s` is already attached to channel `%s`", |
836 | opt_enable_all ? "matching all events" : | |
837 | (std::string("with pattern `") + | |
838 | pattern + std::string("`")) | |
839 | .c_str(), | |
93928b6b | 840 | exclusion_string.get(), |
13c25a5c JG |
841 | opt_filter, |
842 | print_channel_name(channel_name)); | |
89476427 | 843 | warn = 1; |
7671f53c | 844 | break; |
45d5d421 CB |
845 | case LTTNG_ERR_TRACE_ALREADY_STARTED: |
846 | { | |
13c25a5c JG |
847 | ERR("Can't enable an event in a new domain for a session that has already been started once (session `%s`)", |
848 | session_name.c_str()); | |
45d5d421 CB |
849 | error = 1; |
850 | break; | |
851 | } | |
7671f53c | 852 | default: |
13c25a5c JG |
853 | ERR("Failed to enable event rule %s%s, with filter expression `%s`, and attached to channel `%s`: %s (session `%s`)", |
854 | opt_enable_all ? "matching all events" : | |
855 | (std::string("with pattern `") + | |
856 | pattern + std::string("`")) | |
857 | .c_str(), | |
93928b6b | 858 | exclusion_string.get(), |
13c25a5c | 859 | opt_filter, |
28ab034a JG |
860 | command_ret == -LTTNG_ERR_NEED_CHANNEL_NAME ? |
861 | print_raw_channel_name(channel_name) : | |
862 | print_channel_name(channel_name), | |
13c25a5c JG |
863 | lttng_strerror(command_ret), |
864 | session_name.c_str()); | |
89476427 | 865 | error = 1; |
93928b6b JG |
866 | |
867 | if (opt_kernel) { | |
868 | print_kernel_tracer_status_error(); | |
869 | } | |
870 | ||
7671f53c CB |
871 | break; |
872 | } | |
89476427 | 873 | |
13c25a5c | 874 | error_holder = command_ret; |
16363652 | 875 | } else { |
13c25a5c JG |
876 | MSG("Enabled %s event rule %s%s and filter expression `%s`", |
877 | lttng_domain_type_str(dom.type), | |
878 | opt_enable_all ? "matching all events" : | |
879 | (std::string("with pattern `") + pattern + | |
880 | std::string("`")) | |
881 | .c_str(), | |
882 | ||
93928b6b | 883 | exclusion_string.get(), |
28ab034a | 884 | opt_filter); |
53a80697 MD |
885 | } |
886 | } | |
6181537c | 887 | |
89476427 JRJ |
888 | if (lttng_opt_mi) { |
889 | if (command_ret) { | |
890 | success = 0; | |
91744e14 | 891 | ev->enabled = 0; |
89476427 | 892 | } else { |
91744e14 | 893 | ev->enabled = 1; |
89476427 JRJ |
894 | } |
895 | ||
8884bbc1 | 896 | ret = mi_lttng_event(writer.get(), ev, 1, handle->domain.type); |
89476427 JRJ |
897 | if (ret) { |
898 | ret = CMD_ERROR; | |
899 | goto error; | |
900 | } | |
901 | ||
902 | /* print exclusion */ | |
e358ddd5 | 903 | ret = mi_print_exclusion(&exclusions); |
89476427 JRJ |
904 | if (ret) { |
905 | ret = CMD_ERROR; | |
906 | goto error; | |
907 | } | |
908 | ||
909 | /* Success ? */ | |
28ab034a | 910 | ret = mi_lttng_writer_write_element_bool( |
8884bbc1 | 911 | writer.get(), mi_lttng_element_command_success, success); |
89476427 JRJ |
912 | if (ret) { |
913 | ret = CMD_ERROR; | |
914 | goto end; | |
915 | } | |
916 | ||
917 | /* Close event element */ | |
8884bbc1 | 918 | ret = mi_lttng_writer_close_element(writer.get()); |
89476427 JRJ |
919 | if (ret) { |
920 | ret = CMD_ERROR; | |
921 | goto end; | |
922 | } | |
923 | } | |
924 | ||
89476427 JRJ |
925 | /* Reset warn, error and success */ |
926 | success = 1; | |
f3ed775e DG |
927 | } |
928 | ||
8c9ae521 | 929 | end: |
89476427 JRJ |
930 | /* Close Mi */ |
931 | if (lttng_opt_mi) { | |
932 | /* Close events element */ | |
8884bbc1 | 933 | ret = mi_lttng_writer_close_element(writer.get()); |
89476427 JRJ |
934 | if (ret) { |
935 | ret = CMD_ERROR; | |
936 | goto error; | |
937 | } | |
938 | } | |
f3ed775e | 939 | error: |
ae856491 DG |
940 | if (warn) { |
941 | ret = CMD_WARNING; | |
942 | } | |
89476427 JRJ |
943 | if (error) { |
944 | ret = CMD_ERROR; | |
945 | } | |
cd80958d | 946 | lttng_destroy_handle(handle); |
e358ddd5 | 947 | lttng_dynamic_pointer_array_reset(&exclusions); |
0c4e727b | 948 | lttng_userspace_probe_location_destroy(uprobe_loc); |
7ed70bc9 | 949 | |
89476427 JRJ |
950 | /* Overwrite ret with error_holder if there was an actual error with |
951 | * enabling an event. | |
952 | */ | |
953 | ret = error_holder ? error_holder : ret; | |
954 | ||
91744e14 | 955 | lttng_event_destroy(ev); |
f3ed775e DG |
956 | return ret; |
957 | } | |
958 | ||
c926bb36 JG |
959 | void _poptContextFree_deleter_func(poptContext ctx) |
960 | { | |
961 | poptFreeContext(ctx); | |
962 | } | |
963 | ||
d512607f JG |
964 | } /* namespace */ |
965 | ||
bf1413aa | 966 | int validate_exclusion_list(const char *pattern, |
d512607f JG |
967 | const struct lttng_dynamic_pointer_array *exclusions) |
968 | { | |
969 | int ret; | |
970 | ||
13c25a5c | 971 | /* Event name pattern must be a valid globbing pattern to allow exclusions. */ |
bf1413aa | 972 | if (!strutils_is_star_glob_pattern(pattern)) { |
13c25a5c | 973 | ERR("Event name pattern must contain wildcard characters to use exclusions"); |
d512607f JG |
974 | goto error; |
975 | } | |
976 | ||
977 | /* | |
978 | * If the event name is a star-at-end only globbing pattern, | |
979 | * then we can validate the individual exclusions. Otherwise | |
980 | * all exclusions are passed to the session daemon. | |
981 | */ | |
bf1413aa | 982 | if (strutils_is_star_at_the_end_only_glob_pattern(pattern)) { |
d512607f JG |
983 | size_t i, num_exclusions; |
984 | ||
985 | num_exclusions = lttng_dynamic_pointer_array_get_count(exclusions); | |
986 | ||
987 | for (i = 0; i < num_exclusions; i++) { | |
988 | const char *exclusion = | |
989 | (const char *) lttng_dynamic_pointer_array_get_pointer(exclusions, | |
990 | i); | |
991 | ||
992 | if (!strutils_is_star_glob_pattern(exclusion) || | |
993 | strutils_is_star_at_the_end_only_glob_pattern(exclusion)) { | |
bf1413aa | 994 | ret = check_exclusion_subsets(pattern, exclusion); |
d512607f JG |
995 | if (ret) { |
996 | goto error; | |
997 | } | |
998 | } | |
999 | } | |
1000 | } | |
1001 | ||
1002 | ret = 0; | |
1003 | goto end; | |
1004 | ||
1005 | error: | |
1006 | ret = -1; | |
1007 | ||
1008 | end: | |
1009 | return ret; | |
1010 | } | |
1011 | ||
f3ed775e | 1012 | /* |
6181537c | 1013 | * Add event to trace session |
f3ed775e DG |
1014 | */ |
1015 | int cmd_enable_events(int argc, const char **argv) | |
1016 | { | |
89476427 | 1017 | int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1; |
bf1413aa | 1018 | std::string session_name; |
cd9adb8b JG |
1019 | const char *arg_event_list = nullptr; |
1020 | const char *leftover = nullptr; | |
de044b7a | 1021 | int event_type = -1; |
bf1413aa | 1022 | event_rule_patterns patterns; |
f3ed775e | 1023 | |
c926bb36 JG |
1024 | auto pc = lttng::make_unique_wrapper<poptContext_s, _poptContextFree_deleter_func>( |
1025 | poptGetContext(nullptr, argc, argv, long_options, 0)); | |
1026 | poptReadDefaultConfig(pc.get(), 0); | |
f3ed775e DG |
1027 | |
1028 | /* Default event type */ | |
7a3d1328 | 1029 | opt_event_type = LTTNG_EVENT_ALL; |
f3ed775e | 1030 | |
c926bb36 | 1031 | while ((opt = poptGetNextOpt(pc.get())) != -1) { |
f3ed775e DG |
1032 | switch (opt) { |
1033 | case OPT_HELP: | |
4ba92f18 | 1034 | SHOW_HELP(); |
27b5aa57 | 1035 | return CMD_SUCCESS; |
f3ed775e | 1036 | case OPT_TRACEPOINT: |
e6ddca71 | 1037 | opt_event_type = LTTNG_EVENT_TRACEPOINT; |
f3ed775e | 1038 | break; |
cf0e5467 | 1039 | case OPT_PROBE: |
7d29a247 | 1040 | opt_event_type = LTTNG_EVENT_PROBE; |
f3ed775e | 1041 | break; |
dcabc190 FD |
1042 | case OPT_USERSPACE_PROBE: |
1043 | opt_event_type = LTTNG_EVENT_USERSPACE_PROBE; | |
1044 | break; | |
f3ed775e DG |
1045 | case OPT_FUNCTION: |
1046 | opt_event_type = LTTNG_EVENT_FUNCTION; | |
8f0d098b | 1047 | break; |
a54bd42d MD |
1048 | case OPT_SYSCALL: |
1049 | opt_event_type = LTTNG_EVENT_SYSCALL; | |
0133c199 | 1050 | break; |
eeac7d46 MD |
1051 | case OPT_USERSPACE: |
1052 | opt_userspace = 1; | |
eeac7d46 | 1053 | break; |
0cda4f28 | 1054 | case OPT_LOGLEVEL: |
8005f29a | 1055 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_RANGE; |
c926bb36 | 1056 | opt_loglevel = poptGetOptArg(pc.get()); |
0cda4f28 MD |
1057 | break; |
1058 | case OPT_LOGLEVEL_ONLY: | |
8005f29a | 1059 | opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_SINGLE; |
c926bb36 | 1060 | opt_loglevel = poptGetOptArg(pc.get()); |
13dce3b7 | 1061 | break; |
679b4943 SM |
1062 | case OPT_LIST_OPTIONS: |
1063 | list_cmd_options(stdout, long_options); | |
27b5aa57 | 1064 | return CMD_SUCCESS; |
53a80697 MD |
1065 | case OPT_FILTER: |
1066 | break; | |
fac3366c JI |
1067 | case OPT_EXCLUDE: |
1068 | break; | |
f3ed775e | 1069 | default: |
27b5aa57 | 1070 | return CMD_UNDEFINED; |
f3ed775e | 1071 | } |
de044b7a DG |
1072 | |
1073 | /* Validate event type. Multiple event type are not supported. */ | |
1074 | if (event_type == -1) { | |
1075 | event_type = opt_event_type; | |
1076 | } else { | |
1077 | if (event_type != opt_event_type) { | |
13c25a5c | 1078 | ERR("Only one event type may be enabled at once"); |
27b5aa57 | 1079 | return CMD_ERROR; |
de044b7a DG |
1080 | } |
1081 | } | |
f3ed775e DG |
1082 | } |
1083 | ||
3ecec76a | 1084 | ret = print_missing_or_multiple_domains( |
28ab034a | 1085 | opt_kernel + opt_userspace + opt_jul + opt_log4j + opt_python, true); |
3ecec76a | 1086 | if (ret) { |
27b5aa57 | 1087 | return CMD_ERROR; |
3ecec76a PP |
1088 | } |
1089 | ||
89476427 JRJ |
1090 | /* Mi check */ |
1091 | if (lttng_opt_mi) { | |
8884bbc1 | 1092 | writer = mi_writer_uptr(mi_lttng_writer_create(fileno(stdout), lttng_opt_mi)); |
89476427 | 1093 | if (!writer) { |
27b5aa57 JG |
1094 | LTTNG_THROW_ERROR(lttng::format( |
1095 | "Failed to create MI writer: format_code={}", lttng_opt_mi)); | |
89476427 JRJ |
1096 | } |
1097 | ||
1098 | /* Open command element */ | |
8884bbc1 JG |
1099 | ret = mi_lttng_writer_command_open(writer.get(), |
1100 | mi_lttng_element_command_enable_event); | |
89476427 | 1101 | if (ret) { |
27b5aa57 JG |
1102 | LTTNG_THROW_ERROR(lttng::format( |
1103 | "Failed to open MI command element: command_name=`{}`", | |
1104 | mi_lttng_element_command_enable_event)); | |
89476427 JRJ |
1105 | } |
1106 | ||
1107 | /* Open output element */ | |
8884bbc1 | 1108 | ret = mi_lttng_writer_open_element(writer.get(), mi_lttng_element_command_output); |
89476427 | 1109 | if (ret) { |
27b5aa57 JG |
1110 | LTTNG_THROW_ERROR( |
1111 | lttng::format("Failed to open MI element: element_name=`{}`", | |
1112 | mi_lttng_element_command_output)); | |
89476427 JRJ |
1113 | } |
1114 | } | |
1115 | ||
27b5aa57 JG |
1116 | /* Close the MI command context when leaving the function, no matter the result. */ |
1117 | const auto close_mi_on_exit = lttng::make_scope_exit([&success]() noexcept { | |
1118 | if (!lttng_opt_mi) { | |
1119 | return; | |
1120 | } | |
1121 | ||
1122 | /* Close output element. */ | |
1123 | if (mi_lttng_writer_close_element(writer.get())) { | |
1124 | ERR_FMT("Failed to close MI output element"); | |
1125 | return; | |
1126 | } | |
1127 | ||
1128 | if (mi_lttng_writer_write_element_bool( | |
1129 | writer.get(), mi_lttng_element_command_success, success)) { | |
1130 | ERR_FMT("Failed to write MI element: element_name=`{}`, value={}", | |
1131 | mi_lttng_element_command_success, | |
1132 | success); | |
1133 | return; | |
1134 | } | |
1135 | ||
1136 | /* Command element close. */ | |
1137 | if (mi_lttng_writer_command_close(writer.get())) { | |
1138 | ERR_FMT("Failed to close MI command element"); | |
1139 | return; | |
1140 | } | |
1141 | }); | |
1142 | ||
c926bb36 | 1143 | arg_event_list = poptGetArg(pc.get()); |
cd9adb8b | 1144 | if (arg_event_list == nullptr && opt_enable_all == 0) { |
13c25a5c | 1145 | ERR("Missing event name pattern(s)"); |
27b5aa57 | 1146 | return CMD_ERROR; |
f3ed775e DG |
1147 | } |
1148 | ||
77878c17 JG |
1149 | if (opt_enable_all) { |
1150 | patterns.emplace_back("*"); | |
1151 | } else { | |
1152 | std::stringstream event_list_arg_stream(arg_event_list); | |
1153 | ||
1154 | for (std::string line; std::getline(event_list_arg_stream, line, ',');) { | |
1155 | patterns.emplace_back(std::move(line)); | |
1156 | } | |
5b915816 MJ |
1157 | } |
1158 | ||
c926bb36 | 1159 | leftover = poptGetArg(pc.get()); |
68c7f6e5 JD |
1160 | if (leftover) { |
1161 | ERR("Unknown argument: %s", leftover); | |
27b5aa57 | 1162 | return CMD_ERROR; |
68c7f6e5 JD |
1163 | } |
1164 | ||
cd80958d | 1165 | if (!opt_session_name) { |
bf1413aa | 1166 | const auto rc_file_session_name = |
303ac4ed | 1167 | lttng::make_unique_wrapper<char, lttng::memory::free>(get_session_name()); |
bf1413aa JG |
1168 | |
1169 | if (!rc_file_session_name) { | |
27b5aa57 | 1170 | return CMD_ERROR; |
cd80958d | 1171 | } |
bf1413aa JG |
1172 | |
1173 | session_name = rc_file_session_name.get(); | |
cd80958d DG |
1174 | } else { |
1175 | session_name = opt_session_name; | |
1176 | } | |
1177 | ||
bf1413aa | 1178 | command_ret = enable_events(session_name, patterns); |
89476427 | 1179 | if (command_ret) { |
27b5aa57 | 1180 | return CMD_ERROR; |
89476427 | 1181 | } |
f3ed775e | 1182 | |
27b5aa57 | 1183 | return CMD_SUCCESS; |
f3ed775e | 1184 | } |