Commit | Line | Data |
---|---|---|
20fe2104 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. | |
20fe2104 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. | |
20fe2104 DG |
16 | */ |
17 | ||
6c1c0768 | 18 | #define _LGPL_SOURCE |
7b395890 | 19 | #include <fcntl.h> |
20fe2104 DG |
20 | #include <stdlib.h> |
21 | #include <stdio.h> | |
f34daff7 | 22 | #include <string.h> |
8c0faa1d | 23 | #include <unistd.h> |
77c7c900 | 24 | #include <inttypes.h> |
20fe2104 | 25 | |
990570ed | 26 | #include <common/common.h> |
db758600 | 27 | #include <common/kernel-ctl/kernel-ctl.h> |
c052142c | 28 | #include <common/kernel-ctl/kernel-ioctl.h> |
42224349 | 29 | #include <common/sessiond-comm/sessiond-comm.h> |
1e307fab | 30 | |
2f77fc4b | 31 | #include "consumer.h" |
4771f025 | 32 | #include "kernel.h" |
6dc3064a | 33 | #include "kernel-consumer.h" |
096102bd | 34 | #include "kern-modules.h" |
834978fd | 35 | #include "utils.h" |
5c408ad8 | 36 | #include "rotate.h" |
20fe2104 | 37 | |
e1f3997a JD |
38 | /* |
39 | * Key used to reference a channel between the sessiond and the consumer. This | |
40 | * is only read and updated with the session_list lock held. | |
41 | */ | |
42 | static uint64_t next_kernel_channel_key; | |
43 | ||
d65106b1 | 44 | /* |
050349bb | 45 | * Add context on a kernel channel. |
df3c77c8 JG |
46 | * |
47 | * Assumes the ownership of ctx. | |
d65106b1 DG |
48 | */ |
49 | int kernel_add_channel_context(struct ltt_kernel_channel *chan, | |
645328ae | 50 | struct ltt_kernel_context *ctx) |
d65106b1 DG |
51 | { |
52 | int ret; | |
53 | ||
0525e9ae DG |
54 | assert(chan); |
55 | assert(ctx); | |
56 | ||
d65106b1 | 57 | DBG("Adding context to channel %s", chan->channel->name); |
645328ae | 58 | ret = kernctl_add_context(chan->fd, &ctx->ctx); |
d65106b1 | 59 | if (ret < 0) { |
32af2c95 | 60 | switch (-ret) { |
1ae5e83e JD |
61 | case ENOSYS: |
62 | /* Exists but not available for this kernel */ | |
63 | ret = LTTNG_ERR_KERN_CONTEXT_UNAVAILABLE; | |
64 | goto error; | |
65 | case EEXIST: | |
b579acd9 DG |
66 | /* If EEXIST, we just ignore the error */ |
67 | ret = 0; | |
1ae5e83e JD |
68 | goto end; |
69 | default: | |
70 | PERROR("add context ioctl"); | |
71 | ret = LTTNG_ERR_KERN_CONTEXT_FAIL; | |
72 | goto error; | |
b579acd9 | 73 | } |
d65106b1 | 74 | } |
21ed98c1 | 75 | ret = 0; |
d65106b1 | 76 | |
1ae5e83e | 77 | end: |
645328ae | 78 | cds_list_add_tail(&ctx->list, &chan->ctx_list); |
ba985c3a | 79 | ctx->in_list = true; |
df3c77c8 | 80 | ctx = NULL; |
d65106b1 | 81 | error: |
df3c77c8 JG |
82 | if (ctx) { |
83 | trace_kernel_destroy_context(ctx); | |
84 | } | |
d65106b1 DG |
85 | return ret; |
86 | } | |
87 | ||
20fe2104 | 88 | /* |
050349bb DG |
89 | * Create a new kernel session, register it to the kernel tracer and add it to |
90 | * the session daemon session. | |
20fe2104 | 91 | */ |
8c0faa1d | 92 | int kernel_create_session(struct ltt_session *session, int tracer_fd) |
20fe2104 DG |
93 | { |
94 | int ret; | |
95 | struct ltt_kernel_session *lks; | |
96 | ||
0525e9ae DG |
97 | assert(session); |
98 | ||
54012638 | 99 | /* Allocate data structure */ |
dec56f6c | 100 | lks = trace_kernel_create_session(); |
20fe2104 | 101 | if (lks == NULL) { |
54012638 | 102 | ret = -1; |
20fe2104 DG |
103 | goto error; |
104 | } | |
105 | ||
54012638 | 106 | /* Kernel tracer session creation */ |
20fe2104 DG |
107 | ret = kernctl_create_session(tracer_fd); |
108 | if (ret < 0) { | |
df0f840b | 109 | PERROR("ioctl kernel create session"); |
20fe2104 DG |
110 | goto error; |
111 | } | |
112 | ||
20fe2104 | 113 | lks->fd = ret; |
7b395890 DG |
114 | /* Prevent fd duplication after execlp() */ |
115 | ret = fcntl(lks->fd, F_SETFD, FD_CLOEXEC); | |
116 | if (ret < 0) { | |
df0f840b | 117 | PERROR("fcntl session fd"); |
7b395890 DG |
118 | } |
119 | ||
53632229 | 120 | lks->id = session->id; |
3bd1e081 | 121 | lks->consumer_fds_sent = 0; |
8c0faa1d | 122 | session->kernel_session = lks; |
8c0faa1d DG |
123 | |
124 | DBG("Kernel session created (fd: %d)", lks->fd); | |
20fe2104 DG |
125 | |
126 | return 0; | |
127 | ||
128 | error: | |
5f62c685 DG |
129 | if (lks) { |
130 | trace_kernel_destroy_session(lks); | |
131 | } | |
20fe2104 DG |
132 | return ret; |
133 | } | |
134 | ||
135 | /* | |
050349bb DG |
136 | * Create a kernel channel, register it to the kernel tracer and add it to the |
137 | * kernel session. | |
20fe2104 | 138 | */ |
050349bb | 139 | int kernel_create_channel(struct ltt_kernel_session *session, |
fdd9eb17 | 140 | struct lttng_channel *chan) |
20fe2104 DG |
141 | { |
142 | int ret; | |
143 | struct ltt_kernel_channel *lkc; | |
20fe2104 | 144 | |
0525e9ae DG |
145 | assert(session); |
146 | assert(chan); | |
0525e9ae | 147 | |
54012638 | 148 | /* Allocate kernel channel */ |
fdd9eb17 | 149 | lkc = trace_kernel_create_channel(chan); |
54012638 | 150 | if (lkc == NULL) { |
20fe2104 DG |
151 | goto error; |
152 | } | |
153 | ||
ecc48a90 | 154 | DBG3("Kernel create channel %s with attr: %d, %" PRIu64 ", %" PRIu64 ", %u, %u, %d, %d", |
fdd9eb17 | 155 | chan->name, lkc->channel->attr.overwrite, |
173af62f DG |
156 | lkc->channel->attr.subbuf_size, lkc->channel->attr.num_subbuf, |
157 | lkc->channel->attr.switch_timer_interval, lkc->channel->attr.read_timer_interval, | |
ecc48a90 | 158 | lkc->channel->attr.live_timer_interval, lkc->channel->attr.output); |
173af62f | 159 | |
54012638 | 160 | /* Kernel tracer channel creation */ |
f3ed775e | 161 | ret = kernctl_create_channel(session->fd, &lkc->channel->attr); |
20fe2104 | 162 | if (ret < 0) { |
df0f840b | 163 | PERROR("ioctl kernel create channel"); |
20fe2104 DG |
164 | goto error; |
165 | } | |
166 | ||
54012638 | 167 | /* Setup the channel fd */ |
20fe2104 | 168 | lkc->fd = ret; |
7b395890 DG |
169 | /* Prevent fd duplication after execlp() */ |
170 | ret = fcntl(lkc->fd, F_SETFD, FD_CLOEXEC); | |
171 | if (ret < 0) { | |
df0f840b | 172 | PERROR("fcntl session fd"); |
7b395890 DG |
173 | } |
174 | ||
54012638 | 175 | /* Add channel to session */ |
8c0faa1d DG |
176 | cds_list_add(&lkc->list, &session->channel_list.head); |
177 | session->channel_count++; | |
fb5f35b6 | 178 | lkc->session = session; |
e1f3997a | 179 | lkc->key = ++next_kernel_channel_key; |
20fe2104 | 180 | |
e1f3997a JD |
181 | DBG("Kernel channel %s created (fd: %d, key: %" PRIu64 ")", |
182 | lkc->channel->name, lkc->fd, lkc->key); | |
20fe2104 DG |
183 | |
184 | return 0; | |
185 | ||
186 | error: | |
5f62c685 DG |
187 | if (lkc) { |
188 | free(lkc->channel); | |
189 | free(lkc); | |
190 | } | |
54012638 | 191 | return -1; |
20fe2104 | 192 | } |
f34daff7 DG |
193 | |
194 | /* | |
050349bb DG |
195 | * Create a kernel event, enable it to the kernel tracer and add it to the |
196 | * channel event list of the kernel session. | |
49d21f93 | 197 | * We own filter_expression and filter. |
f34daff7 | 198 | */ |
050349bb | 199 | int kernel_create_event(struct lttng_event *ev, |
00a62084 MD |
200 | struct ltt_kernel_channel *channel, |
201 | char *filter_expression, | |
202 | struct lttng_filter_bytecode *filter) | |
f34daff7 DG |
203 | { |
204 | int ret; | |
205 | struct ltt_kernel_event *event; | |
f34daff7 | 206 | |
0525e9ae DG |
207 | assert(ev); |
208 | assert(channel); | |
209 | ||
a969e101 | 210 | /* We pass ownership of filter_expression and filter */ |
00a62084 MD |
211 | event = trace_kernel_create_event(ev, filter_expression, |
212 | filter); | |
54012638 | 213 | if (event == NULL) { |
d87bfb32 | 214 | ret = -1; |
f34daff7 DG |
215 | goto error; |
216 | } | |
217 | ||
f3ed775e DG |
218 | ret = kernctl_create_event(channel->fd, event->event); |
219 | if (ret < 0) { | |
32af2c95 | 220 | switch (-ret) { |
bd29c13d DG |
221 | case EEXIST: |
222 | break; | |
223 | case ENOSYS: | |
224 | WARN("Event type not implemented"); | |
225 | break; | |
8197a339 DG |
226 | case ENOENT: |
227 | WARN("Event %s not found!", ev->name); | |
228 | break; | |
bd29c13d | 229 | default: |
d87bfb32 DG |
230 | PERROR("create event ioctl"); |
231 | } | |
e953ef25 | 232 | goto free_event; |
8c0faa1d | 233 | } |
f34daff7 | 234 | |
d0ae4ea8 | 235 | event->type = ev->type; |
f3ed775e | 236 | event->fd = ret; |
7b395890 DG |
237 | /* Prevent fd duplication after execlp() */ |
238 | ret = fcntl(event->fd, F_SETFD, FD_CLOEXEC); | |
239 | if (ret < 0) { | |
df0f840b | 240 | PERROR("fcntl session fd"); |
7b395890 DG |
241 | } |
242 | ||
00a62084 MD |
243 | if (filter) { |
244 | ret = kernctl_filter(event->fd, filter); | |
245 | if (ret) { | |
246 | goto filter_error; | |
247 | } | |
248 | } | |
249 | ||
250 | ret = kernctl_enable(event->fd); | |
251 | if (ret < 0) { | |
32af2c95 | 252 | switch (-ret) { |
00a62084 MD |
253 | case EEXIST: |
254 | ret = LTTNG_ERR_KERN_EVENT_EXIST; | |
255 | break; | |
256 | default: | |
257 | PERROR("enable kernel event"); | |
258 | break; | |
259 | } | |
260 | goto enable_error; | |
261 | } | |
262 | ||
f3ed775e DG |
263 | /* Add event to event list */ |
264 | cds_list_add(&event->list, &channel->events_list.head); | |
cbbbb275 DG |
265 | channel->event_count++; |
266 | ||
e953ef25 DG |
267 | DBG("Event %s created (fd: %d)", ev->name, event->fd); |
268 | ||
269 | return 0; | |
270 | ||
00a62084 MD |
271 | enable_error: |
272 | filter_error: | |
273 | { | |
274 | int closeret; | |
275 | ||
276 | closeret = close(event->fd); | |
277 | if (closeret) { | |
278 | PERROR("close event fd"); | |
279 | } | |
280 | } | |
e953ef25 DG |
281 | free_event: |
282 | free(event); | |
283 | error: | |
d87bfb32 | 284 | return ret; |
e953ef25 DG |
285 | } |
286 | ||
26cc6b4e | 287 | /* |
050349bb | 288 | * Disable a kernel channel. |
26cc6b4e DG |
289 | */ |
290 | int kernel_disable_channel(struct ltt_kernel_channel *chan) | |
291 | { | |
292 | int ret; | |
293 | ||
0525e9ae DG |
294 | assert(chan); |
295 | ||
26cc6b4e DG |
296 | ret = kernctl_disable(chan->fd); |
297 | if (ret < 0) { | |
df0f840b | 298 | PERROR("disable chan ioctl"); |
26cc6b4e DG |
299 | goto error; |
300 | } | |
301 | ||
302 | chan->enabled = 0; | |
e1f3997a JD |
303 | DBG("Kernel channel %s disabled (fd: %d, key: %" PRIu64 ")", |
304 | chan->channel->name, chan->fd, chan->key); | |
26cc6b4e DG |
305 | |
306 | return 0; | |
307 | ||
308 | error: | |
309 | return ret; | |
310 | } | |
311 | ||
d36b8583 | 312 | /* |
050349bb | 313 | * Enable a kernel channel. |
d36b8583 DG |
314 | */ |
315 | int kernel_enable_channel(struct ltt_kernel_channel *chan) | |
316 | { | |
317 | int ret; | |
318 | ||
0525e9ae DG |
319 | assert(chan); |
320 | ||
d36b8583 | 321 | ret = kernctl_enable(chan->fd); |
32af2c95 | 322 | if (ret < 0 && ret != -EEXIST) { |
df0f840b | 323 | PERROR("Enable kernel chan"); |
d36b8583 DG |
324 | goto error; |
325 | } | |
326 | ||
327 | chan->enabled = 1; | |
e1f3997a JD |
328 | DBG("Kernel channel %s enabled (fd: %d, key: %" PRIu64 ")", |
329 | chan->channel->name, chan->fd, chan->key); | |
d36b8583 DG |
330 | |
331 | return 0; | |
332 | ||
333 | error: | |
334 | return ret; | |
335 | } | |
336 | ||
19e70852 | 337 | /* |
050349bb | 338 | * Enable a kernel event. |
19e70852 DG |
339 | */ |
340 | int kernel_enable_event(struct ltt_kernel_event *event) | |
341 | { | |
342 | int ret; | |
343 | ||
0525e9ae DG |
344 | assert(event); |
345 | ||
19e70852 | 346 | ret = kernctl_enable(event->fd); |
42224349 | 347 | if (ret < 0) { |
32af2c95 | 348 | switch (-ret) { |
42224349 | 349 | case EEXIST: |
f73fabfd | 350 | ret = LTTNG_ERR_KERN_EVENT_EXIST; |
42224349 DG |
351 | break; |
352 | default: | |
353 | PERROR("enable kernel event"); | |
354 | break; | |
355 | } | |
19e70852 DG |
356 | goto error; |
357 | } | |
358 | ||
359 | event->enabled = 1; | |
360 | DBG("Kernel event %s enabled (fd: %d)", event->event->name, event->fd); | |
361 | ||
362 | return 0; | |
363 | ||
364 | error: | |
d36b8583 | 365 | return ret; |
19e70852 DG |
366 | } |
367 | ||
e953ef25 | 368 | /* |
050349bb | 369 | * Disable a kernel event. |
e953ef25 | 370 | */ |
19e70852 | 371 | int kernel_disable_event(struct ltt_kernel_event *event) |
e953ef25 DG |
372 | { |
373 | int ret; | |
19e70852 | 374 | |
0525e9ae DG |
375 | assert(event); |
376 | ||
19e70852 | 377 | ret = kernctl_disable(event->fd); |
42224349 | 378 | if (ret < 0) { |
32af2c95 | 379 | switch (-ret) { |
42224349 | 380 | case EEXIST: |
f73fabfd | 381 | ret = LTTNG_ERR_KERN_EVENT_EXIST; |
42224349 DG |
382 | break; |
383 | default: | |
384 | PERROR("disable kernel event"); | |
385 | break; | |
386 | } | |
19e70852 | 387 | goto error; |
e953ef25 | 388 | } |
f3ed775e | 389 | |
19e70852 DG |
390 | event->enabled = 0; |
391 | DBG("Kernel event %s disabled (fd: %d)", event->event->name, event->fd); | |
392 | ||
f34daff7 DG |
393 | return 0; |
394 | ||
395 | error: | |
d36b8583 | 396 | return ret; |
f34daff7 | 397 | } |
aaf26714 | 398 | |
a24248b3 MD |
399 | static |
400 | struct lttng_tracker_list *get_id_tracker_list(struct ltt_kernel_session *session, | |
401 | enum lttng_tracker_type tracker_type) | |
402 | { | |
403 | switch (tracker_type) { | |
404 | case LTTNG_TRACKER_PID: | |
405 | return session->tracker_list_pid; | |
406 | case LTTNG_TRACKER_VPID: | |
407 | return session->tracker_list_vpid; | |
408 | case LTTNG_TRACKER_UID: | |
409 | return session->tracker_list_uid; | |
410 | case LTTNG_TRACKER_VUID: | |
411 | return session->tracker_list_vuid; | |
412 | case LTTNG_TRACKER_GID: | |
413 | return session->tracker_list_gid; | |
414 | case LTTNG_TRACKER_VGID: | |
415 | return session->tracker_list_vgid; | |
416 | default: | |
417 | return NULL; | |
418 | } | |
419 | } | |
6e911cad | 420 | |
a24248b3 MD |
421 | int kernel_track_id(enum lttng_tracker_type tracker_type, |
422 | struct ltt_kernel_session *session, | |
423 | struct lttng_tracker_id *id) | |
ccf10263 | 424 | { |
a24248b3 MD |
425 | int ret, value; |
426 | struct lttng_tracker_list *tracker_list; | |
21b4a71c MD |
427 | struct lttng_tracker_id *saved_ids; |
428 | ssize_t saved_ids_count, i; | |
a24248b3 MD |
429 | |
430 | ret = lttng_tracker_id_lookup_string(tracker_type, | |
431 | id, &value); | |
432 | if (ret != LTTNG_OK) { | |
433 | return ret; | |
434 | } | |
435 | ||
a24248b3 MD |
436 | tracker_list = get_id_tracker_list(session, tracker_type); |
437 | if (!tracker_list) { | |
438 | return LTTNG_ERR_INVALID; | |
439 | } | |
21b4a71c MD |
440 | /* Save list for restore on error. */ |
441 | saved_ids_count = lttng_tracker_id_get_list(tracker_list, &saved_ids); | |
442 | if (saved_ids_count < 0) { | |
443 | return LTTNG_ERR_INVALID; | |
444 | } | |
445 | /* Add to list. */ | |
a24248b3 MD |
446 | ret = lttng_tracker_list_add(tracker_list, id); |
447 | if (ret != LTTNG_OK) { | |
21b4a71c | 448 | goto end; |
a24248b3 | 449 | } |
7c493d31 | 450 | |
a24248b3 MD |
451 | switch (tracker_type) { |
452 | case LTTNG_TRACKER_PID: | |
453 | DBG("Kernel track PID %d for session id %" PRIu64 ".", | |
454 | value, session->id); | |
455 | ret = kernctl_track_pid(session->fd, value); | |
456 | if (!ret) { | |
21b4a71c MD |
457 | ret = LTTNG_OK; |
458 | goto end; | |
a24248b3 MD |
459 | } |
460 | break; | |
461 | case LTTNG_TRACKER_VPID: | |
462 | DBG("Kernel track VPID %d for session id %" PRIu64 ".", | |
463 | value, session->id); | |
464 | ret = kernctl_track_id(session->fd, LTTNG_TRACKER_VPID, value); | |
465 | if (!ret) { | |
21b4a71c MD |
466 | ret = LTTNG_OK; |
467 | goto end; | |
a24248b3 MD |
468 | } |
469 | break; | |
470 | case LTTNG_TRACKER_UID: | |
471 | DBG("Kernel track UID %d for session id %" PRIu64 ".", | |
472 | value, session->id); | |
473 | ret = kernctl_track_id(session->fd, LTTNG_TRACKER_UID, value); | |
474 | if (!ret) { | |
21b4a71c MD |
475 | ret = LTTNG_OK; |
476 | goto end; | |
a24248b3 MD |
477 | } |
478 | break; | |
479 | case LTTNG_TRACKER_GID: | |
480 | DBG("Kernel track GID %d for session id %" PRIu64 ".", | |
481 | value, session->id); | |
482 | ret = kernctl_track_id(session->fd, LTTNG_TRACKER_GID, value); | |
483 | if (!ret) { | |
21b4a71c MD |
484 | ret = LTTNG_OK; |
485 | goto end; | |
a24248b3 MD |
486 | } |
487 | break; | |
488 | case LTTNG_TRACKER_VUID: | |
489 | DBG("Kernel track VUID %d for session id %" PRIu64 ".", | |
490 | value, session->id); | |
491 | ret = kernctl_track_id(session->fd, LTTNG_TRACKER_VUID, value); | |
492 | if (!ret) { | |
21b4a71c MD |
493 | ret = LTTNG_OK; |
494 | goto end; | |
a24248b3 MD |
495 | } |
496 | break; | |
497 | case LTTNG_TRACKER_VGID: | |
498 | DBG("Kernel track VGID %d for session id %" PRIu64 ".", | |
499 | value, session->id); | |
500 | ret = kernctl_track_id(session->fd, LTTNG_TRACKER_VGID, value); | |
501 | if (!ret) { | |
21b4a71c MD |
502 | ret = LTTNG_OK; |
503 | goto end; | |
a24248b3 MD |
504 | } |
505 | break; | |
506 | default: | |
21b4a71c MD |
507 | ret = -EINVAL; |
508 | break; | |
7c493d31 | 509 | } |
21b4a71c | 510 | /* Error handling. */ |
32af2c95 | 511 | switch (-ret) { |
7c493d31 | 512 | case EINVAL: |
21b4a71c MD |
513 | ret = LTTNG_ERR_INVALID; |
514 | break; | |
7c493d31 | 515 | case ENOMEM: |
21b4a71c MD |
516 | ret = LTTNG_ERR_NOMEM; |
517 | break; | |
7c493d31 | 518 | case EEXIST: |
21b4a71c MD |
519 | ret = LTTNG_ERR_ID_TRACKED; |
520 | break; | |
7c493d31 | 521 | default: |
21b4a71c MD |
522 | ret = LTTNG_ERR_UNK; |
523 | break; | |
7c493d31 | 524 | } |
21b4a71c MD |
525 | if (lttng_tracker_id_set_list(tracker_list, saved_ids, saved_ids_count) != LTTNG_OK) { |
526 | ERR("Error on tracker add error handling.\n"); | |
527 | } | |
528 | end: | |
529 | for (i = 0; i < saved_ids_count; i++) { | |
530 | free(saved_ids[i].string); | |
531 | } | |
532 | free(saved_ids); | |
533 | return ret; | |
ccf10263 MD |
534 | } |
535 | ||
a24248b3 MD |
536 | int kernel_untrack_id(enum lttng_tracker_type tracker_type, |
537 | struct ltt_kernel_session *session, | |
538 | struct lttng_tracker_id *id) | |
ccf10263 | 539 | { |
a24248b3 MD |
540 | int ret, value; |
541 | struct lttng_tracker_list *tracker_list; | |
21b4a71c MD |
542 | struct lttng_tracker_id *saved_ids; |
543 | ssize_t saved_ids_count, i; | |
a24248b3 MD |
544 | |
545 | ret = lttng_tracker_id_lookup_string(tracker_type, | |
546 | id, &value); | |
547 | if (ret != LTTNG_OK) { | |
548 | return ret; | |
549 | } | |
7c493d31 | 550 | |
a24248b3 MD |
551 | tracker_list = get_id_tracker_list(session, tracker_type); |
552 | if (!tracker_list) { | |
553 | return LTTNG_ERR_INVALID; | |
554 | } | |
21b4a71c MD |
555 | /* Save list for restore on error. */ |
556 | saved_ids_count = lttng_tracker_id_get_list(tracker_list, &saved_ids); | |
557 | if (saved_ids_count < 0) { | |
558 | return LTTNG_ERR_INVALID; | |
559 | } | |
560 | /* Remove from list. */ | |
a24248b3 MD |
561 | ret = lttng_tracker_list_remove(tracker_list, id); |
562 | if (ret != LTTNG_OK) { | |
21b4a71c | 563 | goto end; |
7c493d31 | 564 | } |
a24248b3 MD |
565 | |
566 | switch (tracker_type) { | |
567 | case LTTNG_TRACKER_PID: | |
568 | DBG("Kernel untrack PID %d for session id %" PRIu64 ".", | |
569 | value, session->id); | |
570 | ret = kernctl_untrack_pid(session->fd, value); | |
571 | if (!ret) { | |
21b4a71c MD |
572 | ret = LTTNG_OK; |
573 | goto end; | |
a24248b3 MD |
574 | } |
575 | break; | |
576 | case LTTNG_TRACKER_VPID: | |
577 | DBG("Kernel untrack VPID %d for session id %" PRIu64 ".", | |
578 | value, session->id); | |
579 | ret = kernctl_untrack_id(session->fd, LTTNG_TRACKER_VPID, value); | |
580 | if (!ret) { | |
21b4a71c MD |
581 | ret = LTTNG_OK; |
582 | goto end; | |
a24248b3 MD |
583 | } |
584 | break; | |
585 | case LTTNG_TRACKER_UID: | |
586 | DBG("Kernel untrack UID %d for session id %" PRIu64 ".", | |
587 | value, session->id); | |
588 | ret = kernctl_untrack_id(session->fd, LTTNG_TRACKER_UID, value); | |
589 | if (!ret) { | |
21b4a71c MD |
590 | ret = LTTNG_OK; |
591 | goto end; | |
a24248b3 MD |
592 | } |
593 | break; | |
594 | case LTTNG_TRACKER_GID: | |
595 | DBG("Kernel untrack GID %d for session id %" PRIu64 ".", | |
596 | value, session->id); | |
597 | ret = kernctl_untrack_id(session->fd, LTTNG_TRACKER_GID, value); | |
598 | if (!ret) { | |
21b4a71c MD |
599 | ret = LTTNG_OK; |
600 | goto end; | |
a24248b3 MD |
601 | } |
602 | break; | |
603 | case LTTNG_TRACKER_VUID: | |
604 | DBG("Kernel untrack VUID %d for session id %" PRIu64 ".", | |
605 | value, session->id); | |
606 | ret = kernctl_untrack_id(session->fd, LTTNG_TRACKER_VUID, value); | |
607 | if (!ret) { | |
21b4a71c MD |
608 | ret = LTTNG_OK; |
609 | goto end; | |
a24248b3 MD |
610 | } |
611 | break; | |
612 | case LTTNG_TRACKER_VGID: | |
613 | DBG("Kernel untrack VGID %d for session id %" PRIu64 ".", | |
614 | value, session->id); | |
615 | ret = kernctl_untrack_id(session->fd, LTTNG_TRACKER_VGID, value); | |
616 | if (!ret) { | |
21b4a71c MD |
617 | ret = LTTNG_OK; |
618 | goto end; | |
a24248b3 MD |
619 | } |
620 | break; | |
621 | default: | |
21b4a71c MD |
622 | ret = -EINVAL; |
623 | break; | |
a24248b3 MD |
624 | } |
625 | ||
21b4a71c | 626 | /* Error handling. */ |
32af2c95 | 627 | switch (-ret) { |
7c493d31 | 628 | case EINVAL: |
21b4a71c MD |
629 | ret = LTTNG_ERR_INVALID; |
630 | break; | |
7c493d31 | 631 | case ENOMEM: |
21b4a71c MD |
632 | ret = LTTNG_ERR_NOMEM; |
633 | break; | |
634 | case EEXIST: | |
635 | ret = LTTNG_ERR_ID_TRACKED; | |
636 | break; | |
7c493d31 | 637 | default: |
21b4a71c MD |
638 | ret = LTTNG_ERR_UNK; |
639 | break; | |
7c493d31 | 640 | } |
21b4a71c MD |
641 | |
642 | if (lttng_tracker_id_set_list(tracker_list, saved_ids, saved_ids_count) != LTTNG_OK) { | |
643 | ERR("Error on tracker remove error handling.\n"); | |
644 | } | |
645 | end: | |
646 | for (i = 0; i < saved_ids_count; i++) { | |
647 | free(saved_ids[i].string); | |
648 | } | |
649 | free(saved_ids); | |
650 | return ret; | |
ccf10263 MD |
651 | } |
652 | ||
a24248b3 MD |
653 | /* |
654 | * Called with session lock held. | |
655 | */ | |
656 | ssize_t kernel_list_tracker_ids(enum lttng_tracker_type tracker_type, | |
657 | struct ltt_kernel_session *session, | |
658 | struct lttng_tracker_id **_ids) | |
a5dfbb9d | 659 | { |
a24248b3 | 660 | struct lttng_tracker_list *tracker_list; |
a5dfbb9d | 661 | |
a24248b3 MD |
662 | tracker_list = get_id_tracker_list(session, tracker_type); |
663 | if (!tracker_list) { | |
664 | return -LTTNG_ERR_INVALID; | |
a5dfbb9d | 665 | } |
a24248b3 | 666 | return lttng_tracker_id_get_list(tracker_list, _ids); |
a5dfbb9d MD |
667 | } |
668 | ||
aaf26714 | 669 | /* |
050349bb DG |
670 | * Create kernel metadata, open from the kernel tracer and add it to the |
671 | * kernel session. | |
aaf26714 | 672 | */ |
a4b92340 | 673 | int kernel_open_metadata(struct ltt_kernel_session *session) |
aaf26714 DG |
674 | { |
675 | int ret; | |
74024a21 | 676 | struct ltt_kernel_metadata *lkm = NULL; |
aaf26714 | 677 | |
0525e9ae DG |
678 | assert(session); |
679 | ||
54012638 | 680 | /* Allocate kernel metadata */ |
a4b92340 | 681 | lkm = trace_kernel_create_metadata(); |
54012638 | 682 | if (lkm == NULL) { |
aaf26714 DG |
683 | goto error; |
684 | } | |
685 | ||
54012638 | 686 | /* Kernel tracer metadata creation */ |
f3ed775e | 687 | ret = kernctl_open_metadata(session->fd, &lkm->conf->attr); |
aaf26714 | 688 | if (ret < 0) { |
74024a21 | 689 | goto error_open; |
aaf26714 DG |
690 | } |
691 | ||
8c0faa1d | 692 | lkm->fd = ret; |
d40f0359 | 693 | lkm->key = ++next_kernel_channel_key; |
7b395890 DG |
694 | /* Prevent fd duplication after execlp() */ |
695 | ret = fcntl(lkm->fd, F_SETFD, FD_CLOEXEC); | |
696 | if (ret < 0) { | |
df0f840b | 697 | PERROR("fcntl session fd"); |
7b395890 DG |
698 | } |
699 | ||
aaf26714 | 700 | session->metadata = lkm; |
8c0faa1d | 701 | |
00e2e675 | 702 | DBG("Kernel metadata opened (fd: %d)", lkm->fd); |
8c0faa1d DG |
703 | |
704 | return 0; | |
705 | ||
74024a21 DG |
706 | error_open: |
707 | trace_kernel_destroy_metadata(lkm); | |
8c0faa1d | 708 | error: |
54012638 | 709 | return -1; |
8c0faa1d DG |
710 | } |
711 | ||
712 | /* | |
050349bb | 713 | * Start tracing session. |
8c0faa1d DG |
714 | */ |
715 | int kernel_start_session(struct ltt_kernel_session *session) | |
716 | { | |
717 | int ret; | |
718 | ||
0525e9ae DG |
719 | assert(session); |
720 | ||
8c0faa1d DG |
721 | ret = kernctl_start_session(session->fd); |
722 | if (ret < 0) { | |
df0f840b | 723 | PERROR("ioctl start session"); |
8c0faa1d DG |
724 | goto error; |
725 | } | |
726 | ||
727 | DBG("Kernel session started"); | |
728 | ||
729 | return 0; | |
730 | ||
731 | error: | |
732 | return ret; | |
733 | } | |
734 | ||
f3ed775e | 735 | /* |
050349bb | 736 | * Make a kernel wait to make sure in-flight probe have completed. |
f3ed775e DG |
737 | */ |
738 | void kernel_wait_quiescent(int fd) | |
739 | { | |
740 | int ret; | |
741 | ||
742 | DBG("Kernel quiescent wait on %d", fd); | |
743 | ||
744 | ret = kernctl_wait_quiescent(fd); | |
745 | if (ret < 0) { | |
df0f840b | 746 | PERROR("wait quiescent ioctl"); |
f3ed775e DG |
747 | ERR("Kernel quiescent wait failed"); |
748 | } | |
749 | } | |
750 | ||
751 | /* | |
f3ed775e DG |
752 | * Force flush buffer of metadata. |
753 | */ | |
754 | int kernel_metadata_flush_buffer(int fd) | |
755 | { | |
756 | int ret; | |
757 | ||
169d2cb7 DG |
758 | DBG("Kernel flushing metadata buffer on fd %d", fd); |
759 | ||
f3ed775e DG |
760 | ret = kernctl_buffer_flush(fd); |
761 | if (ret < 0) { | |
00e2e675 | 762 | ERR("Fail to flush metadata buffers %d (ret: %d)", fd, ret); |
f3ed775e DG |
763 | } |
764 | ||
765 | return 0; | |
766 | } | |
767 | ||
768 | /* | |
050349bb | 769 | * Force flush buffer for channel. |
f3ed775e DG |
770 | */ |
771 | int kernel_flush_buffer(struct ltt_kernel_channel *channel) | |
772 | { | |
773 | int ret; | |
774 | struct ltt_kernel_stream *stream; | |
775 | ||
0525e9ae DG |
776 | assert(channel); |
777 | ||
f3ed775e DG |
778 | DBG("Flush buffer for channel %s", channel->channel->name); |
779 | ||
780 | cds_list_for_each_entry(stream, &channel->stream_list.head, list) { | |
781 | DBG("Flushing channel stream %d", stream->fd); | |
782 | ret = kernctl_buffer_flush(stream->fd); | |
783 | if (ret < 0) { | |
df0f840b | 784 | PERROR("ioctl"); |
f3ed775e DG |
785 | ERR("Fail to flush buffer for stream %d (ret: %d)", |
786 | stream->fd, ret); | |
787 | } | |
788 | } | |
789 | ||
790 | return 0; | |
791 | } | |
792 | ||
8c0faa1d | 793 | /* |
050349bb | 794 | * Stop tracing session. |
8c0faa1d DG |
795 | */ |
796 | int kernel_stop_session(struct ltt_kernel_session *session) | |
797 | { | |
798 | int ret; | |
799 | ||
0525e9ae DG |
800 | assert(session); |
801 | ||
8c0faa1d DG |
802 | ret = kernctl_stop_session(session->fd); |
803 | if (ret < 0) { | |
804 | goto error; | |
805 | } | |
806 | ||
807 | DBG("Kernel session stopped"); | |
808 | ||
809 | return 0; | |
810 | ||
811 | error: | |
812 | return ret; | |
813 | } | |
814 | ||
815 | /* | |
050349bb DG |
816 | * Open stream of channel, register it to the kernel tracer and add it |
817 | * to the stream list of the channel. | |
8c0faa1d | 818 | * |
1cfb4b98 MD |
819 | * Note: given that the streams may appear in random order wrt CPU |
820 | * number (e.g. cpu hotplug), the index value of the stream number in | |
821 | * the stream name is not necessarily linked to the CPU number. | |
822 | * | |
050349bb | 823 | * Return the number of created stream. Else, a negative value. |
8c0faa1d | 824 | */ |
f3ed775e | 825 | int kernel_open_channel_stream(struct ltt_kernel_channel *channel) |
8c0faa1d | 826 | { |
1cfb4b98 | 827 | int ret; |
8c0faa1d DG |
828 | struct ltt_kernel_stream *lks; |
829 | ||
0525e9ae DG |
830 | assert(channel); |
831 | ||
5a47c6a2 | 832 | while ((ret = kernctl_create_stream(channel->fd)) >= 0) { |
1cfb4b98 MD |
833 | lks = trace_kernel_create_stream(channel->channel->name, |
834 | channel->stream_count); | |
8c0faa1d | 835 | if (lks == NULL) { |
799e2c4f MD |
836 | ret = close(ret); |
837 | if (ret) { | |
838 | PERROR("close"); | |
839 | } | |
8c0faa1d DG |
840 | goto error; |
841 | } | |
842 | ||
843 | lks->fd = ret; | |
7b395890 DG |
844 | /* Prevent fd duplication after execlp() */ |
845 | ret = fcntl(lks->fd, F_SETFD, FD_CLOEXEC); | |
846 | if (ret < 0) { | |
df0f840b | 847 | PERROR("fcntl session fd"); |
7b395890 DG |
848 | } |
849 | ||
1624d5b7 JD |
850 | lks->tracefile_size = channel->channel->attr.tracefile_size; |
851 | lks->tracefile_count = channel->channel->attr.tracefile_count; | |
852 | ||
1cfb4b98 | 853 | /* Add stream to channel stream list */ |
8c0faa1d DG |
854 | cds_list_add(&lks->list, &channel->stream_list.head); |
855 | channel->stream_count++; | |
8c0faa1d | 856 | |
00e2e675 DG |
857 | DBG("Kernel stream %s created (fd: %d, state: %d)", lks->name, lks->fd, |
858 | lks->state); | |
54012638 | 859 | } |
8c0faa1d DG |
860 | |
861 | return channel->stream_count; | |
862 | ||
863 | error: | |
54012638 | 864 | return -1; |
8c0faa1d DG |
865 | } |
866 | ||
867 | /* | |
050349bb | 868 | * Open the metadata stream and set it to the kernel session. |
8c0faa1d | 869 | */ |
f3ed775e | 870 | int kernel_open_metadata_stream(struct ltt_kernel_session *session) |
8c0faa1d DG |
871 | { |
872 | int ret; | |
873 | ||
0525e9ae DG |
874 | assert(session); |
875 | ||
8c0faa1d DG |
876 | ret = kernctl_create_stream(session->metadata->fd); |
877 | if (ret < 0) { | |
df0f840b | 878 | PERROR("kernel create metadata stream"); |
8c0faa1d DG |
879 | goto error; |
880 | } | |
881 | ||
882 | DBG("Kernel metadata stream created (fd: %d)", ret); | |
883 | session->metadata_stream_fd = ret; | |
7b395890 DG |
884 | /* Prevent fd duplication after execlp() */ |
885 | ret = fcntl(session->metadata_stream_fd, F_SETFD, FD_CLOEXEC); | |
886 | if (ret < 0) { | |
df0f840b | 887 | PERROR("fcntl session fd"); |
7b395890 | 888 | } |
aaf26714 DG |
889 | |
890 | return 0; | |
891 | ||
892 | error: | |
54012638 | 893 | return -1; |
aaf26714 | 894 | } |
2ef84c95 DG |
895 | |
896 | /* | |
9f19cc17 | 897 | * Get the event list from the kernel tracer and return the number of elements. |
2ef84c95 | 898 | */ |
9f19cc17 | 899 | ssize_t kernel_list_events(int tracer_fd, struct lttng_event **events) |
2ef84c95 | 900 | { |
53efb85a | 901 | int fd, ret; |
9f19cc17 DG |
902 | char *event; |
903 | size_t nbmem, count = 0; | |
2ef84c95 | 904 | FILE *fp; |
9f19cc17 | 905 | struct lttng_event *elist; |
2ef84c95 | 906 | |
0525e9ae DG |
907 | assert(events); |
908 | ||
2ef84c95 DG |
909 | fd = kernctl_tracepoint_list(tracer_fd); |
910 | if (fd < 0) { | |
df0f840b | 911 | PERROR("kernel tracepoint list"); |
2ef84c95 DG |
912 | goto error; |
913 | } | |
914 | ||
915 | fp = fdopen(fd, "r"); | |
916 | if (fp == NULL) { | |
df0f840b | 917 | PERROR("kernel tracepoint list fdopen"); |
61b73b12 | 918 | goto error_fp; |
2ef84c95 DG |
919 | } |
920 | ||
921 | /* | |
922 | * Init memory size counter | |
923 | * See kernel-ctl.h for explanation of this value | |
924 | */ | |
6725fe19 | 925 | nbmem = KERNEL_EVENT_INIT_LIST_SIZE; |
ba7f0ae5 | 926 | elist = zmalloc(sizeof(struct lttng_event) * nbmem); |
3b870559 MD |
927 | if (elist == NULL) { |
928 | PERROR("alloc list events"); | |
929 | count = -ENOMEM; | |
930 | goto end; | |
931 | } | |
2ef84c95 | 932 | |
53efb85a | 933 | while (fscanf(fp, "event { name = %m[^;]; };\n", &event) == 1) { |
6725fe19 | 934 | if (count >= nbmem) { |
3b870559 | 935 | struct lttng_event *new_elist; |
53efb85a | 936 | size_t new_nbmem; |
3b870559 | 937 | |
53efb85a MD |
938 | new_nbmem = nbmem << 1; |
939 | DBG("Reallocating event list from %zu to %zu bytes", | |
940 | nbmem, new_nbmem); | |
941 | new_elist = realloc(elist, new_nbmem * sizeof(struct lttng_event)); | |
3b870559 | 942 | if (new_elist == NULL) { |
df0f840b | 943 | PERROR("realloc list events"); |
3b870559 MD |
944 | free(event); |
945 | free(elist); | |
61b73b12 MD |
946 | count = -ENOMEM; |
947 | goto end; | |
2ef84c95 | 948 | } |
53efb85a MD |
949 | /* Zero the new memory */ |
950 | memset(new_elist + nbmem, 0, | |
951 | (new_nbmem - nbmem) * sizeof(struct lttng_event)); | |
952 | nbmem = new_nbmem; | |
3b870559 | 953 | elist = new_elist; |
2ef84c95 | 954 | } |
99497cd0 MD |
955 | strncpy(elist[count].name, event, LTTNG_SYMBOL_NAME_LEN); |
956 | elist[count].name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; | |
67b9d018 | 957 | elist[count].enabled = -1; |
9f19cc17 | 958 | count++; |
3b870559 | 959 | free(event); |
2ef84c95 DG |
960 | } |
961 | ||
9f19cc17 | 962 | *events = elist; |
ced2f820 | 963 | DBG("Kernel list events done (%zu events)", count); |
61b73b12 | 964 | end: |
799e2c4f MD |
965 | ret = fclose(fp); /* closes both fp and fd */ |
966 | if (ret) { | |
967 | PERROR("fclose"); | |
968 | } | |
9f19cc17 | 969 | return count; |
2ef84c95 | 970 | |
61b73b12 | 971 | error_fp: |
799e2c4f MD |
972 | ret = close(fd); |
973 | if (ret) { | |
974 | PERROR("close"); | |
975 | } | |
2ef84c95 DG |
976 | error: |
977 | return -1; | |
978 | } | |
096102bd DG |
979 | |
980 | /* | |
981 | * Get kernel version and validate it. | |
982 | */ | |
88076e89 JD |
983 | int kernel_validate_version(int tracer_fd, |
984 | struct lttng_kernel_tracer_version *version, | |
985 | struct lttng_kernel_tracer_abi_version *abi_version) | |
096102bd DG |
986 | { |
987 | int ret; | |
096102bd | 988 | |
88076e89 | 989 | ret = kernctl_tracer_version(tracer_fd, version); |
096102bd | 990 | if (ret < 0) { |
521dd134 | 991 | ERR("Failed to retrieve the lttng-modules version"); |
096102bd DG |
992 | goto error; |
993 | } | |
994 | ||
995 | /* Validate version */ | |
88076e89 | 996 | if (version->major != VERSION_MAJOR) { |
c052142c | 997 | ERR("Kernel tracer major version (%d) is not compatible with lttng-tools major version (%d)", |
88076e89 | 998 | version->major, VERSION_MAJOR); |
096102bd | 999 | goto error_version; |
096102bd | 1000 | } |
88076e89 | 1001 | ret = kernctl_tracer_abi_version(tracer_fd, abi_version); |
c052142c | 1002 | if (ret < 0) { |
521dd134 | 1003 | ERR("Failed to retrieve lttng-modules ABI version"); |
c052142c MD |
1004 | goto error; |
1005 | } | |
88076e89 | 1006 | if (abi_version->major != LTTNG_MODULES_ABI_MAJOR_VERSION) { |
521dd134 | 1007 | ERR("Kernel tracer ABI version (%d.%d) does not match the expected ABI major version (%d.*)", |
88076e89 | 1008 | abi_version->major, abi_version->minor, |
c052142c MD |
1009 | LTTNG_MODULES_ABI_MAJOR_VERSION); |
1010 | goto error; | |
1011 | } | |
1012 | DBG2("Kernel tracer version validated (%d.%d, ABI %d.%d)", | |
88076e89 JD |
1013 | version->major, version->minor, |
1014 | abi_version->major, abi_version->minor); | |
096102bd DG |
1015 | return 0; |
1016 | ||
1017 | error_version: | |
096102bd DG |
1018 | ret = -1; |
1019 | ||
1020 | error: | |
521dd134 | 1021 | ERR("Kernel tracer version check failed; kernel tracing will not be available"); |
096102bd DG |
1022 | return ret; |
1023 | } | |
335a95b7 MD |
1024 | |
1025 | /* | |
1026 | * Kernel work-arounds called at the start of sessiond main(). | |
1027 | */ | |
1028 | int init_kernel_workarounds(void) | |
1029 | { | |
8936c33a | 1030 | int ret; |
335a95b7 MD |
1031 | FILE *fp; |
1032 | ||
1033 | /* | |
1034 | * boot_id needs to be read once before being used concurrently | |
1035 | * to deal with a Linux kernel race. A fix is proposed for | |
1036 | * upstream, but the work-around is needed for older kernels. | |
1037 | */ | |
1038 | fp = fopen("/proc/sys/kernel/random/boot_id", "r"); | |
1039 | if (!fp) { | |
1040 | goto end_boot_id; | |
1041 | } | |
1042 | while (!feof(fp)) { | |
1043 | char buf[37] = ""; | |
1044 | ||
8936c33a DG |
1045 | ret = fread(buf, 1, sizeof(buf), fp); |
1046 | if (ret < 0) { | |
1047 | /* Ignore error, we don't really care */ | |
1048 | } | |
335a95b7 | 1049 | } |
799e2c4f MD |
1050 | ret = fclose(fp); |
1051 | if (ret) { | |
1052 | PERROR("fclose"); | |
1053 | } | |
335a95b7 | 1054 | end_boot_id: |
335a95b7 MD |
1055 | return 0; |
1056 | } | |
2f77fc4b DG |
1057 | |
1058 | /* | |
1059 | * Complete teardown of a kernel session. | |
1060 | */ | |
1061 | void kernel_destroy_session(struct ltt_kernel_session *ksess) | |
1062 | { | |
1063 | if (ksess == NULL) { | |
1064 | DBG3("No kernel session when tearing down session"); | |
1065 | return; | |
1066 | } | |
1067 | ||
1068 | DBG("Tearing down kernel session"); | |
1069 | ||
07b86b52 | 1070 | /* |
15dc512a DG |
1071 | * Destroy channels on the consumer if at least one FD has been sent and we |
1072 | * are in no output mode because the streams are in *no* monitor mode so we | |
1073 | * have to send a command to clean them up or else they leaked. | |
07b86b52 | 1074 | */ |
15dc512a | 1075 | if (!ksess->output_traces && ksess->consumer_fds_sent) { |
07b86b52 JD |
1076 | int ret; |
1077 | struct consumer_socket *socket; | |
1078 | struct lttng_ht_iter iter; | |
1079 | ||
1080 | /* For each consumer socket. */ | |
d069d577 | 1081 | rcu_read_lock(); |
07b86b52 JD |
1082 | cds_lfht_for_each_entry(ksess->consumer->socks->ht, &iter.iter, |
1083 | socket, node.node) { | |
1084 | struct ltt_kernel_channel *chan; | |
1085 | ||
1086 | /* For each channel, ask the consumer to destroy it. */ | |
1087 | cds_list_for_each_entry(chan, &ksess->channel_list.head, list) { | |
1088 | ret = kernel_consumer_destroy_channel(socket, chan); | |
1089 | if (ret < 0) { | |
1090 | /* Consumer is probably dead. Use next socket. */ | |
1091 | continue; | |
1092 | } | |
1093 | } | |
1094 | } | |
d069d577 | 1095 | rcu_read_unlock(); |
07b86b52 JD |
1096 | } |
1097 | ||
2f77fc4b DG |
1098 | /* Close any relayd session */ |
1099 | consumer_output_send_destroy_relayd(ksess->consumer); | |
1100 | ||
1101 | trace_kernel_destroy_session(ksess); | |
1102 | } | |
fb5f35b6 DG |
1103 | |
1104 | /* | |
1105 | * Destroy a kernel channel object. It does not do anything on the tracer side. | |
1106 | */ | |
1107 | void kernel_destroy_channel(struct ltt_kernel_channel *kchan) | |
1108 | { | |
1109 | struct ltt_kernel_session *ksess = NULL; | |
1110 | ||
1111 | assert(kchan); | |
1112 | assert(kchan->channel); | |
1113 | ||
1114 | DBG3("Kernel destroy channel %s", kchan->channel->name); | |
1115 | ||
1116 | /* Update channel count of associated session. */ | |
1117 | if (kchan->session) { | |
1118 | /* Keep pointer reference so we can update it after the destroy. */ | |
1119 | ksess = kchan->session; | |
1120 | } | |
1121 | ||
1122 | trace_kernel_destroy_channel(kchan); | |
1123 | ||
1124 | /* | |
1125 | * At this point the kernel channel is not visible anymore. This is safe | |
1126 | * since in order to work on a visible kernel session, the tracing session | |
1127 | * lock (ltt_session.lock) MUST be acquired. | |
1128 | */ | |
1129 | if (ksess) { | |
1130 | ksess->channel_count--; | |
1131 | } | |
1132 | } | |
6dc3064a DG |
1133 | |
1134 | /* | |
1135 | * Take a snapshot for a given kernel session. | |
1136 | * | |
2a06df8d | 1137 | * Return 0 on success or else return a LTTNG_ERR code. |
6dc3064a DG |
1138 | */ |
1139 | int kernel_snapshot_record(struct ltt_kernel_session *ksess, | |
d07ceecd MD |
1140 | struct snapshot_output *output, int wait, |
1141 | uint64_t nb_packets_per_stream) | |
6dc3064a | 1142 | { |
2a06df8d | 1143 | int err, ret, saved_metadata_fd; |
6dc3064a DG |
1144 | struct consumer_socket *socket; |
1145 | struct lttng_ht_iter iter; | |
1146 | struct ltt_kernel_metadata *saved_metadata; | |
e098433c JG |
1147 | struct ltt_session *session; |
1148 | uint64_t trace_archive_id; | |
6dc3064a DG |
1149 | |
1150 | assert(ksess); | |
1151 | assert(ksess->consumer); | |
1152 | assert(output); | |
1153 | ||
1154 | DBG("Kernel snapshot record started"); | |
1155 | ||
e098433c JG |
1156 | session = session_find_by_id(ksess->id); |
1157 | assert(session); | |
1158 | assert(pthread_mutex_trylock(&session->lock)); | |
1159 | assert(session_trylock_list()); | |
1160 | trace_archive_id = session->current_archive_id; | |
1161 | ||
6dc3064a DG |
1162 | /* Save current metadata since the following calls will change it. */ |
1163 | saved_metadata = ksess->metadata; | |
1164 | saved_metadata_fd = ksess->metadata_stream_fd; | |
1165 | ||
1166 | rcu_read_lock(); | |
1167 | ||
1168 | ret = kernel_open_metadata(ksess); | |
1169 | if (ret < 0) { | |
1170 | ret = LTTNG_ERR_KERN_META_FAIL; | |
1171 | goto error; | |
1172 | } | |
1173 | ||
1174 | ret = kernel_open_metadata_stream(ksess); | |
1175 | if (ret < 0) { | |
1176 | ret = LTTNG_ERR_KERN_META_FAIL; | |
1177 | goto error_open_stream; | |
1178 | } | |
1179 | ||
1180 | /* Send metadata to consumer and snapshot everything. */ | |
1181 | cds_lfht_for_each_entry(ksess->consumer->socks->ht, &iter.iter, | |
1182 | socket, node.node) { | |
1183 | struct consumer_output *saved_output; | |
1184 | struct ltt_kernel_channel *chan; | |
6dc3064a DG |
1185 | |
1186 | /* | |
1187 | * Temporarly switch consumer output for our snapshot output. As long | |
1188 | * as the session lock is taken, this is safe. | |
1189 | */ | |
1190 | saved_output = ksess->consumer; | |
1191 | ksess->consumer = output->consumer; | |
1192 | ||
1193 | pthread_mutex_lock(socket->lock); | |
1194 | /* This stream must not be monitored by the consumer. */ | |
07b86b52 | 1195 | ret = kernel_consumer_add_metadata(socket, ksess, 0); |
6dc3064a | 1196 | pthread_mutex_unlock(socket->lock); |
07b86b52 | 1197 | /* Put back the saved consumer output into the session. */ |
6dc3064a DG |
1198 | ksess->consumer = saved_output; |
1199 | if (ret < 0) { | |
1200 | ret = LTTNG_ERR_KERN_CONSUMER_FAIL; | |
1201 | goto error_consumer; | |
1202 | } | |
1203 | ||
1204 | /* For each channel, ask the consumer to snapshot it. */ | |
1205 | cds_list_for_each_entry(chan, &ksess->channel_list.head, list) { | |
0a28e553 | 1206 | ret = consumer_snapshot_channel(socket, chan->key, output, 0, |
5c786ded JD |
1207 | ksess->uid, ksess->gid, |
1208 | DEFAULT_KERNEL_TRACE_DIR, wait, | |
e098433c JG |
1209 | nb_packets_per_stream, |
1210 | trace_archive_id); | |
6dc3064a DG |
1211 | if (ret < 0) { |
1212 | ret = LTTNG_ERR_KERN_CONSUMER_FAIL; | |
2a06df8d DG |
1213 | (void) kernel_consumer_destroy_metadata(socket, |
1214 | ksess->metadata); | |
6dc3064a DG |
1215 | goto error_consumer; |
1216 | } | |
1217 | } | |
1218 | ||
1219 | /* Snapshot metadata, */ | |
d40f0359 | 1220 | ret = consumer_snapshot_channel(socket, ksess->metadata->key, output, |
5c786ded | 1221 | 1, ksess->uid, ksess->gid, |
e098433c JG |
1222 | DEFAULT_KERNEL_TRACE_DIR, wait, 0, |
1223 | trace_archive_id); | |
6dc3064a DG |
1224 | if (ret < 0) { |
1225 | ret = LTTNG_ERR_KERN_CONSUMER_FAIL; | |
1226 | goto error_consumer; | |
1227 | } | |
07b86b52 JD |
1228 | |
1229 | /* | |
1230 | * The metadata snapshot is done, ask the consumer to destroy it since | |
1231 | * it's not monitored on the consumer side. | |
1232 | */ | |
1233 | (void) kernel_consumer_destroy_metadata(socket, ksess->metadata); | |
6dc3064a DG |
1234 | } |
1235 | ||
fac41e72 DG |
1236 | ret = LTTNG_OK; |
1237 | ||
6dc3064a DG |
1238 | error_consumer: |
1239 | /* Close newly opened metadata stream. It's now on the consumer side. */ | |
2a06df8d DG |
1240 | err = close(ksess->metadata_stream_fd); |
1241 | if (err < 0) { | |
6dc3064a DG |
1242 | PERROR("close snapshot kernel"); |
1243 | } | |
1244 | ||
1245 | error_open_stream: | |
1246 | trace_kernel_destroy_metadata(ksess->metadata); | |
1247 | error: | |
1248 | /* Restore metadata state.*/ | |
1249 | ksess->metadata = saved_metadata; | |
1250 | ksess->metadata_stream_fd = saved_metadata_fd; | |
1251 | ||
1252 | rcu_read_unlock(); | |
1253 | return ret; | |
1254 | } | |
834978fd DG |
1255 | |
1256 | /* | |
1257 | * Get the syscall mask array from the kernel tracer. | |
1258 | * | |
1259 | * Return 0 on success else a negative value. In both case, syscall_mask should | |
1260 | * be freed. | |
1261 | */ | |
1262 | int kernel_syscall_mask(int chan_fd, char **syscall_mask, uint32_t *nr_bits) | |
1263 | { | |
1264 | assert(syscall_mask); | |
1265 | assert(nr_bits); | |
1266 | ||
1267 | return kernctl_syscall_mask(chan_fd, syscall_mask, nr_bits); | |
1268 | } | |
6e21424e JR |
1269 | |
1270 | /* | |
1271 | * Check for the support of the RING_BUFFER_SNAPSHOT_SAMPLE_POSITIONS via abi | |
1272 | * version number. | |
1273 | * | |
1274 | * Return 1 on success, 0 when feature is not supported, negative value in case | |
1275 | * of errors. | |
1276 | */ | |
1277 | int kernel_supports_ring_buffer_snapshot_sample_positions(int tracer_fd) | |
1278 | { | |
1279 | int ret = 0; // Not supported by default | |
1280 | struct lttng_kernel_tracer_abi_version abi; | |
1281 | ||
1282 | ret = kernctl_tracer_abi_version(tracer_fd, &abi); | |
1283 | if (ret < 0) { | |
1284 | ERR("Failed to retrieve lttng-modules ABI version"); | |
1285 | goto error; | |
1286 | } | |
1287 | ||
1288 | /* | |
1289 | * RING_BUFFER_SNAPSHOT_SAMPLE_POSITIONS was introduced in 2.3 | |
1290 | */ | |
1291 | if (abi.major >= 2 && abi.minor >= 3) { | |
1292 | /* Supported */ | |
1293 | ret = 1; | |
1294 | } else { | |
1295 | /* Not supported */ | |
1296 | ret = 0; | |
1297 | } | |
1298 | error: | |
1299 | return ret; | |
1300 | } | |
5c408ad8 JD |
1301 | |
1302 | /* | |
1303 | * Rotate a kernel session. | |
1304 | * | |
1305 | * Return 0 on success or else return a LTTNG_ERR code. | |
1306 | */ | |
1307 | int kernel_rotate_session(struct ltt_session *session) | |
1308 | { | |
1309 | int ret; | |
1310 | struct consumer_socket *socket; | |
1311 | struct lttng_ht_iter iter; | |
1312 | struct ltt_kernel_session *ksess = session->kernel_session; | |
1313 | ||
1314 | assert(ksess); | |
1315 | assert(ksess->consumer); | |
1316 | ||
1317 | DBG("Rotate kernel session %s started (session %" PRIu64 ")", | |
1318 | session->name, session->id); | |
1319 | ||
1320 | rcu_read_lock(); | |
1321 | ||
1322 | /* | |
1323 | * Note that this loop will end after one iteration given that there is | |
1324 | * only one kernel consumer. | |
1325 | */ | |
1326 | cds_lfht_for_each_entry(ksess->consumer->socks->ht, &iter.iter, | |
1327 | socket, node.node) { | |
1328 | struct ltt_kernel_channel *chan; | |
1329 | ||
1330 | /* | |
1331 | * Account the metadata channel first to make sure the | |
1332 | * number of channels waiting for a rotation cannot | |
1333 | * reach 0 before we complete the iteration over all | |
1334 | * the channels. | |
1335 | */ | |
22a1b931 | 1336 | ret = rotate_add_channel_pending(ksess->metadata->key, |
5c408ad8 JD |
1337 | LTTNG_DOMAIN_KERNEL, session); |
1338 | if (ret < 0) { | |
1339 | ret = LTTNG_ERR_KERN_CONSUMER_FAIL; | |
1340 | goto error; | |
1341 | } | |
1342 | ||
1343 | /* For each channel, ask the consumer to rotate it. */ | |
1344 | cds_list_for_each_entry(chan, &ksess->channel_list.head, list) { | |
1345 | ret = rotate_add_channel_pending(chan->key, | |
1346 | LTTNG_DOMAIN_KERNEL, session); | |
1347 | if (ret < 0) { | |
1348 | ret = LTTNG_ERR_KERN_CONSUMER_FAIL; | |
1349 | goto error; | |
1350 | } | |
1351 | ||
1352 | DBG("Rotate channel %" PRIu64 ", session %s", chan->key, session->name); | |
1353 | ret = consumer_rotate_channel(socket, chan->key, | |
1354 | ksess->uid, ksess->gid, ksess->consumer, | |
1355 | ksess->consumer->subdir, | |
1356 | /* is_metadata_channel */ false, | |
ad9f5c17 | 1357 | session->current_archive_id, |
5c408ad8 JD |
1358 | &session->rotate_pending_relay); |
1359 | if (ret < 0) { | |
1360 | ret = LTTNG_ERR_KERN_CONSUMER_FAIL; | |
1361 | goto error; | |
1362 | } | |
1363 | } | |
1364 | ||
1365 | /* | |
1366 | * Rotate the metadata channel. | |
1367 | */ | |
22a1b931 | 1368 | ret = consumer_rotate_channel(socket, ksess->metadata->key, |
5c408ad8 JD |
1369 | ksess->uid, ksess->gid, ksess->consumer, |
1370 | ksess->consumer->subdir, | |
1371 | /* is_metadata_channel */ true, | |
ad9f5c17 | 1372 | session->current_archive_id, |
5c408ad8 JD |
1373 | &session->rotate_pending_relay); |
1374 | if (ret < 0) { | |
1375 | ret = LTTNG_ERR_KERN_CONSUMER_FAIL; | |
1376 | goto error; | |
1377 | } | |
1378 | } | |
1379 | ||
1380 | ret = LTTNG_OK; | |
1381 | ||
1382 | error: | |
1383 | rcu_read_unlock(); | |
1384 | return ret; | |
1385 | } |