projects / lttng-tools.git / blob
? search:
summary | shortlog | log | commit | commitdiff | tree
blame | history | raw | HEAD
Remove unneeded RCU read lock
[lttng-tools.git] / src / bin / lttng-sessiond / save.c
1 /*
2 * Copyright (C) 2014 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License, version 2 only, as
6 * published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful, but WITHOUT
9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
11 * more details.
12 *
13 * You should have received a copy of the GNU General Public License along with
14 * this program; if not, write to the Free Software Foundation, Inc., 51
15 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
16 */
17
18 #define _GNU_SOURCE
19 #define _LGPL_SOURCE
20 #include <assert.h>
21 #include <inttypes.h>
22 #include <string.h>
23 #include <urcu/uatomic.h>
24 #include <unistd.h>
25
26 #include <common/defaults.h>
27 #include <common/error.h>
28 #include <common/config/config.h>
29 #include <common/utils.h>
30 #include <common/runas.h>
31 #include <lttng/save-internal.h>
32
33 #include "kernel.h"
34 #include "save.h"
35 #include "session.h"
36 #include "syscall.h"
37 #include "trace-ust.h"
38
39 static
40 int save_kernel_channel_attributes(struct config_writer *writer,
41 struct lttng_channel_attr *attr)
42 {
43 int ret;
44
45 ret = config_writer_write_element_string(writer,
46 config_element_overwrite_mode,
47 attr->overwrite ? config_overwrite_mode_overwrite :
48 config_overwrite_mode_discard);
49 if (ret) {
50 goto end;
51 }
52
53 ret = config_writer_write_element_unsigned_int(writer,
54 config_element_subbuf_size, attr->subbuf_size);
55 if (ret) {
56 goto end;
57 }
58
59 ret = config_writer_write_element_unsigned_int(writer,
60 config_element_num_subbuf,
61 attr->num_subbuf);
62 if (ret) {
63 goto end;
64 }
65
66 ret = config_writer_write_element_unsigned_int(writer,
67 config_element_switch_timer_interval,
68 attr->switch_timer_interval);
69 if (ret) {
70 goto end;
71 }
72
73 ret = config_writer_write_element_unsigned_int(writer,
74 config_element_read_timer_interval,
75 attr->read_timer_interval);
76 if (ret) {
77 goto end;
78 }
79
80 ret = config_writer_write_element_string(writer,
81 config_element_output_type,
82 attr->output == LTTNG_EVENT_SPLICE ?
83 config_output_type_splice : config_output_type_mmap);
84 if (ret) {
85 goto end;
86 }
87
88 ret = config_writer_write_element_unsigned_int(writer,
89 config_element_tracefile_size, attr->tracefile_size);
90 if (ret) {
91 goto end;
92 }
93
94 ret = config_writer_write_element_unsigned_int(writer,
95 config_element_tracefile_count,
96 attr->tracefile_count);
97 if (ret) {
98 goto end;
99 }
100
101 ret = config_writer_write_element_unsigned_int(writer,
102 config_element_live_timer_interval,
103 attr->live_timer_interval);
104 if (ret) {
105 goto end;
106 }
107 end:
108 return ret ? LTTNG_ERR_SAVE_IO_FAIL : 0;
109 }
110
111 static
112 int save_ust_channel_attributes(struct config_writer *writer,
113 struct lttng_ust_channel_attr *attr)
114 {
115 int ret;
116
117 ret = config_writer_write_element_string(writer,
118 config_element_overwrite_mode,
119 attr->overwrite ? config_overwrite_mode_overwrite :
120 config_overwrite_mode_discard);
121 if (ret) {
122 goto end;
123 }
124
125 ret = config_writer_write_element_unsigned_int(writer,
126 config_element_subbuf_size, attr->subbuf_size);
127 if (ret) {
128 goto end;
129 }
130
131 ret = config_writer_write_element_unsigned_int(writer,
132 config_element_num_subbuf,
133 attr->num_subbuf);
134 if (ret) {
135 goto end;
136 }
137
138 ret = config_writer_write_element_unsigned_int(writer,
139 config_element_switch_timer_interval,
140 attr->switch_timer_interval);
141 if (ret) {
142 goto end;
143 }
144
145 ret = config_writer_write_element_unsigned_int(writer,
146 config_element_read_timer_interval,
147 attr->read_timer_interval);
148 if (ret) {
149 goto end;
150 }
151
152 ret = config_writer_write_element_string(writer,
153 config_element_output_type,
154 attr->output == LTTNG_UST_MMAP ?
155 config_output_type_mmap : config_output_type_splice);
156 if (ret) {
157 goto end;
158 }
159 end:
160 return ret ? LTTNG_ERR_SAVE_IO_FAIL : 0;
161 }
162
163 static
164 const char *get_kernel_instrumentation_string(
165 enum lttng_kernel_instrumentation instrumentation)
166 {
167 const char *instrumentation_string;
168
169 switch (instrumentation) {
170 case LTTNG_KERNEL_ALL:
171 instrumentation_string = config_event_type_all;
172 break;
173 case LTTNG_KERNEL_TRACEPOINT:
174 instrumentation_string = config_event_type_tracepoint;
175 break;
176 case LTTNG_KERNEL_KPROBE:
177 instrumentation_string = config_event_type_kprobe;
178 break;
179 case LTTNG_KERNEL_FUNCTION:
180 instrumentation_string = config_event_type_function;
181 break;
182 case LTTNG_KERNEL_KRETPROBE:
183 instrumentation_string = config_event_type_kretprobe;
184 break;
185 case LTTNG_KERNEL_NOOP:
186 instrumentation_string = config_event_type_noop;
187 break;
188 case LTTNG_KERNEL_SYSCALL:
189 instrumentation_string = config_event_type_syscall;
190 break;
191 default:
192 instrumentation_string = NULL;
193 }
194
195 return instrumentation_string;
196 }
197
198 static
199 const char *get_kernel_context_type_string(
200 enum lttng_kernel_context_type context_type)
201 {
202 const char *context_type_string;
203
204 switch (context_type) {
205 case LTTNG_KERNEL_CONTEXT_PID:
206 context_type_string = config_event_context_pid;
207 break;
208 case LTTNG_KERNEL_CONTEXT_PROCNAME:
209 context_type_string = config_event_context_procname;
210 break;
211 case LTTNG_KERNEL_CONTEXT_PRIO:
212 context_type_string = config_event_context_prio;
213 break;
214 case LTTNG_KERNEL_CONTEXT_NICE:
215 context_type_string = config_event_context_nice;
216 break;
217 case LTTNG_KERNEL_CONTEXT_VPID:
218 context_type_string = config_event_context_vpid;
219 break;
220 case LTTNG_KERNEL_CONTEXT_TID:
221 context_type_string = config_event_context_tid;
222 break;
223 case LTTNG_KERNEL_CONTEXT_VTID:
224 context_type_string = config_event_context_vtid;
225 break;
226 case LTTNG_KERNEL_CONTEXT_PPID:
227 context_type_string = config_event_context_ppid;
228 break;
229 case LTTNG_KERNEL_CONTEXT_VPPID:
230 context_type_string = config_event_context_vppid;
231 break;
232 case LTTNG_KERNEL_CONTEXT_HOSTNAME:
233 context_type_string = config_event_context_hostname;
234 break;
235 default:
236 context_type_string = NULL;
237 }
238
239 return context_type_string;
240 }
241
242 static
243 const char *get_ust_context_type_string(
244 enum lttng_ust_context_type context_type)
245 {
246 const char *context_type_string;
247
248 switch (context_type) {
249 case LTTNG_UST_CONTEXT_PROCNAME:
250 context_type_string = config_event_context_procname;
251 break;
252 case LTTNG_UST_CONTEXT_VPID:
253 context_type_string = config_event_context_vpid;
254 break;
255 case LTTNG_UST_CONTEXT_VTID:
256 context_type_string = config_event_context_vtid;
257 break;
258 case LTTNG_UST_CONTEXT_IP:
259 context_type_string = config_event_context_ip;
260 break;
261 case LTTNG_UST_CONTEXT_PTHREAD_ID:
262 context_type_string = config_event_context_pthread_id;
263 break;
264 case LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER:
265 /*
266 * Error, should not be stored in the XML, perf contexts
267 * are stored as a node of type event_perf_context_type.
268 */
269 default:
270 context_type_string = NULL;
271 break;
272 }
273
274 return context_type_string;
275 }
276
277 static
278 const char *get_buffer_type_string(
279 enum lttng_buffer_type buffer_type)
280 {
281 const char *buffer_type_string;
282
283 switch (buffer_type) {
284 case LTTNG_BUFFER_PER_PID:
285 buffer_type_string = config_buffer_type_per_pid;
286 break;
287 case LTTNG_BUFFER_PER_UID:
288 buffer_type_string = config_buffer_type_per_uid;
289 break;
290 case LTTNG_BUFFER_GLOBAL:
291 buffer_type_string = config_buffer_type_global;
292 break;
293 default:
294 buffer_type_string = NULL;
295 }
296
297 return buffer_type_string;
298 }
299
300 static
301 const char *get_loglevel_type_string(
302 enum lttng_ust_loglevel_type loglevel_type)
303 {
304 const char *loglevel_type_string;
305
306 switch (loglevel_type) {
307 case LTTNG_UST_LOGLEVEL_ALL:
308 loglevel_type_string = config_loglevel_type_all;
309 break;
310 case LTTNG_UST_LOGLEVEL_RANGE:
311 loglevel_type_string = config_loglevel_type_range;
312 break;
313 case LTTNG_UST_LOGLEVEL_SINGLE:
314 loglevel_type_string = config_loglevel_type_single;
315 break;
316 default:
317 loglevel_type_string = NULL;
318 }
319
320 return loglevel_type_string;
321 }
322
323 static
324 int save_kernel_event(struct config_writer *writer,
325 struct ltt_kernel_event *event)
326 {
327 int ret;
328 const char *instrumentation_type;
329
330 ret = config_writer_open_element(writer, config_element_event);
331 if (ret) {
332 ret = LTTNG_ERR_SAVE_IO_FAIL;
333 goto end;
334 }
335
336 if (event->event->name[0]) {
337 ret = config_writer_write_element_string(writer,
338 config_element_name, event->event->name);
339 if (ret) {
340 ret = LTTNG_ERR_SAVE_IO_FAIL;
341 goto end;
342 }
343 }
344
345 ret = config_writer_write_element_bool(writer, config_element_enabled,
346 event->enabled);
347 if (ret) {
348 ret = LTTNG_ERR_SAVE_IO_FAIL;
349 goto end;
350 }
351
352 instrumentation_type = get_kernel_instrumentation_string(
353 event->event->instrumentation);
354 if (!instrumentation_type) {
355 ret = LTTNG_ERR_INVALID;
356 goto end;
357 }
358
359 ret = config_writer_write_element_string(writer, config_element_type,
360 instrumentation_type);
361 if (ret) {
362 ret = LTTNG_ERR_SAVE_IO_FAIL;
363 goto end;
364 }
365
366 if (event->event->instrumentation == LTTNG_KERNEL_FUNCTION ||
367 event->event->instrumentation == LTTNG_KERNEL_KPROBE ||
368 event->event->instrumentation == LTTNG_KERNEL_KRETPROBE) {
369
370 ret = config_writer_open_element(writer,
371 config_element_attributes);
372 if (ret) {
373 ret = LTTNG_ERR_SAVE_IO_FAIL;
374 goto end;
375 }
376
377 switch (event->event->instrumentation) {
378 case LTTNG_KERNEL_SYSCALL:
379 case LTTNG_KERNEL_FUNCTION:
380 ret = config_writer_open_element(writer,
381 config_element_function_attributes);
382 if (ret) {
383 ret = LTTNG_ERR_SAVE_IO_FAIL;
384 goto end;
385 }
386
387 ret = config_writer_write_element_string(writer,
388 config_element_name,
389 event->event->u.ftrace.symbol_name);
390 if (ret) {
391 ret = LTTNG_ERR_SAVE_IO_FAIL;
392 goto end;
393 }
394
395 /* /function attributes */
396 ret = config_writer_close_element(writer);
397 if (ret) {
398 ret = LTTNG_ERR_SAVE_IO_FAIL;
399 goto end;
400 }
401 break;
402 case LTTNG_KERNEL_KPROBE:
403 case LTTNG_KERNEL_KRETPROBE:
404 {
405 const char *symbol_name;
406 uint64_t addr;
407 uint64_t offset;
408
409 if (event->event->instrumentation ==
410 LTTNG_KERNEL_KPROBE) {
411 /*
412 * Comments in lttng-kernel.h mention that
413 * either addr or symbol_name are set, not both.
414 */
415 addr = event->event->u.kprobe.addr;
416 offset = event->event->u.kprobe.offset;
417 symbol_name = addr ? NULL :
418 event->event->u.kprobe.symbol_name;
419 } else {
420 symbol_name =
421 event->event->u.kretprobe.symbol_name;
422 addr = event->event->u.kretprobe.addr;
423 offset = event->event->u.kretprobe.offset;
424 }
425
426 ret = config_writer_open_element(writer,
427 config_element_probe_attributes);
428 if (ret) {
429 ret = LTTNG_ERR_SAVE_IO_FAIL;
430 goto end;
431 }
432
433 if (symbol_name) {
434 ret = config_writer_write_element_string(writer,
435 config_element_symbol_name,
436 symbol_name);
437 if (ret) {
438 ret = LTTNG_ERR_SAVE_IO_FAIL;
439 goto end;
440 }
441 }
442
443 if (addr) {
444 ret = config_writer_write_element_unsigned_int(
445 writer, config_element_address, addr);
446 if (ret) {
447 ret = LTTNG_ERR_SAVE_IO_FAIL;
448 goto end;
449 }
450 }
451
452 if (offset) {
453 ret = config_writer_write_element_unsigned_int(
454 writer, config_element_offset, offset);
455 if (ret) {
456 ret = LTTNG_ERR_SAVE_IO_FAIL;
457 goto end;
458 }
459 }
460
461 ret = config_writer_close_element(writer);
462 if (ret) {
463 ret = LTTNG_ERR_SAVE_IO_FAIL;
464 goto end;
465 }
466 break;
467 }
468 default:
469 ERR("Unsupported kernel instrumentation type.");
470 ret = LTTNG_ERR_INVALID;
471 goto end;
472 }
473
474 /* /attributes */
475 ret = config_writer_close_element(writer);
476 if (ret) {
477 ret = LTTNG_ERR_SAVE_IO_FAIL;
478 goto end;
479 }
480 }
481
482 /* /event */
483 ret = config_writer_close_element(writer);
484 if (ret) {
485 ret = LTTNG_ERR_SAVE_IO_FAIL;
486 goto end;
487 }
488 end:
489 return ret;
490 }
491
492 static
493 int save_kernel_syscall(struct config_writer *writer,
494 struct ltt_kernel_channel *kchan)
495 {
496 int ret, i;
497 ssize_t count;
498 struct lttng_event *events = NULL;
499
500 assert(writer);
501 assert(kchan);
502
503 count = syscall_list_channel(kchan, &events, 0);
504 if (!count) {
505 /* No syscalls, just gracefully return. */
506 ret = 0;
507 goto end;
508 }
509
510 for (i = 0; i < count; i++) {
511 struct ltt_kernel_event *kevent;
512
513 /* Create a temporary kevent in order to save it. */
514 /*
515 * TODO: struct lttng_event does not really work for a filter,
516 * but unfortunately, it is exposed as external API (and used as
517 * internal representation. Using NULL meanwhile.
518 */
519 kevent = trace_kernel_create_event(&events[i],
520 NULL, NULL);
521 if (!kevent) {
522 ret = -ENOMEM;
523 goto end;
524 }
525 /* Init list in order so the destroy call can del the node. */
526 CDS_INIT_LIST_HEAD(&kevent->list);
527
528 ret = save_kernel_event(writer, kevent);
529 trace_kernel_destroy_event(kevent);
530 if (ret) {
531 goto end;
532 }
533 }
534
535 /* Everything went well */
536 ret = 0;
537
538 end:
539 free(events);
540 return ret;
541 }
542
543 static
544 int save_kernel_events(struct config_writer *writer,
545 struct ltt_kernel_channel *kchan)
546 {
547 int ret;
548 struct ltt_kernel_event *event;
549
550 ret = config_writer_open_element(writer, config_element_events);
551 if (ret) {
552 ret = LTTNG_ERR_SAVE_IO_FAIL;
553 goto end;
554 }
555
556 cds_list_for_each_entry(event, &kchan->events_list.head, list) {
557 ret = save_kernel_event(writer, event);
558 if (ret) {
559 goto end;
560 }
561 }
562
563 /* Save syscalls if any. */
564 ret = save_kernel_syscall(writer, kchan);
565 if (ret) {
566 goto end;
567 }
568
569 /* /events */
570 ret = config_writer_close_element(writer);
571 if (ret) {
572 ret = LTTNG_ERR_SAVE_IO_FAIL;
573 goto end;
574 }
575 end:
576 return ret;
577 }
578
579 static
580 int save_ust_event(struct config_writer *writer,
581 struct ltt_ust_event *event)
582 {
583 int ret;
584 const char *loglevel_type_string;
585
586 ret = config_writer_open_element(writer, config_element_event);
587 if (ret) {
588 ret = LTTNG_ERR_SAVE_IO_FAIL;
589 goto end;
590 }
591
592 if (event->attr.name[0]) {
593 ret = config_writer_write_element_string(writer,
594 config_element_name, event->attr.name);
595 if (ret) {
596 ret = LTTNG_ERR_SAVE_IO_FAIL;
597 goto end;
598 }
599 }
600
601 ret = config_writer_write_element_bool(writer, config_element_enabled,
602 event->enabled);
603 if (ret) {
604 ret = LTTNG_ERR_SAVE_IO_FAIL;
605 goto end;
606 }
607
608 if (event->attr.instrumentation != LTTNG_UST_TRACEPOINT) {
609 ERR("Unsupported UST instrumentation type.");
610 ret = LTTNG_ERR_INVALID;
611 goto end;
612 }
613 ret = config_writer_write_element_string(writer, config_element_type,
614 config_event_type_tracepoint);
615 if (ret) {
616 ret = LTTNG_ERR_SAVE_IO_FAIL;
617 goto end;
618 }
619
620 loglevel_type_string = get_loglevel_type_string(
621 event->attr.loglevel_type);
622 if (!loglevel_type_string) {
623 ERR("Unsupported UST loglevel type.");
624 ret = LTTNG_ERR_INVALID;
625 goto end;
626 }
627
628 ret = config_writer_write_element_string(writer,
629 config_element_loglevel_type, loglevel_type_string);
630 if (ret) {
631 ret = LTTNG_ERR_SAVE_IO_FAIL;
632 goto end;
633 }
634
635 ret = config_writer_write_element_signed_int(writer,
636 config_element_loglevel, event->attr.loglevel);
637 if (ret) {
638 ret = LTTNG_ERR_SAVE_IO_FAIL;
639 goto end;
640 }
641
642 if (event->filter_expression) {
643 ret = config_writer_write_element_string(writer,
644 config_element_filter, event->filter_expression);
645 if (ret) {
646 ret = LTTNG_ERR_SAVE_IO_FAIL;
647 goto end;
648 }
649 }
650
651 if (event->exclusion && event->exclusion->count) {
652 uint32_t i;
653
654 ret = config_writer_open_element(writer,
655 config_element_exclusions);
656 if (ret) {
657 ret = LTTNG_ERR_SAVE_IO_FAIL;
658 goto end;
659 }
660
661 for (i = 0; i < event->exclusion->count; i++) {
662 ret = config_writer_write_element_string(writer,
663 config_element_exclusion,
664 &event->exclusion->names[0][i]);
665 if (ret) {
666 ret = LTTNG_ERR_SAVE_IO_FAIL;
667 goto end;
668 }
669 }
670
671 /* /exclusions */
672 ret = config_writer_close_element(writer);
673 if (ret) {
674 ret = LTTNG_ERR_SAVE_IO_FAIL;
675 goto end;
676 }
677 }
678
679 /* /event */
680 ret = config_writer_close_element(writer);
681 if (ret) {
682 ret = LTTNG_ERR_SAVE_IO_FAIL;
683 goto end;
684 }
685 end:
686 return ret;
687 }
688
689 static
690 int save_ust_events(struct config_writer *writer,
691 struct lttng_ht *events)
692 {
693 int ret;
694 struct ltt_ust_event *event;
695 struct lttng_ht_node_str *node;
696 struct lttng_ht_iter iter;
697
698 ret = config_writer_open_element(writer, config_element_events);
699 if (ret) {
700 ret = LTTNG_ERR_SAVE_IO_FAIL;
701 goto end;
702 }
703
704 rcu_read_lock();
705 cds_lfht_for_each_entry(events->ht, &iter.iter, node, node) {
706 event = caa_container_of(node, struct ltt_ust_event, node);
707
708 if (event->internal) {
709 /* Internal events must not be exposed to clients */
710 continue;
711 }
712 ret = save_ust_event(writer, event);
713 if (ret) {
714 rcu_read_unlock();
715 goto end;
716 }
717 }
718 rcu_read_unlock();
719
720 /* /events */
721 ret = config_writer_close_element(writer);
722 if (ret) {
723 ret = LTTNG_ERR_SAVE_IO_FAIL;
724 goto end;
725 }
726 end:
727 return ret;
728 }
729
730 static
731 int save_kernel_context(struct config_writer *writer,
732 struct lttng_kernel_context *ctx)
733 {
734 int ret = 0;
735
736 if (!ctx) {
737 goto end;
738 }
739
740 ret = config_writer_open_element(writer, config_element_context);
741 if (ret) {
742 ret = LTTNG_ERR_SAVE_IO_FAIL;
743 goto end;
744 }
745
746 if (ctx->ctx == LTTNG_KERNEL_CONTEXT_PERF_CPU_COUNTER) {
747 ret = config_writer_open_element(writer, config_element_perf);
748 if (ret) {
749 ret = LTTNG_ERR_SAVE_IO_FAIL;
750 goto end;
751 }
752
753 ret = config_writer_write_element_unsigned_int(writer,
754 config_element_type, ctx->u.perf_counter.type);
755 if (ret) {
756 ret = LTTNG_ERR_SAVE_IO_FAIL;
757 goto end;
758 }
759
760 ret = config_writer_write_element_unsigned_int(writer,
761 config_element_config, ctx->u.perf_counter.config);
762 if (ret) {
763 ret = LTTNG_ERR_SAVE_IO_FAIL;
764 goto end;
765 }
766
767 ret = config_writer_write_element_string(writer,
768 config_element_name, ctx->u.perf_counter.name);
769 if (ret) {
770 ret = LTTNG_ERR_SAVE_IO_FAIL;
771 goto end;
772 }
773
774 /* /perf */
775 ret = config_writer_close_element(writer);
776 if (ret) {
777 ret = LTTNG_ERR_SAVE_IO_FAIL;
778 goto end;
779 }
780 } else {
781 const char *context_type_string =
782 get_kernel_context_type_string(ctx->ctx);
783
784 if (!context_type_string) {
785 ERR("Unsupported kernel context type.");
786 ret = LTTNG_ERR_INVALID;
787 goto end;
788 }
789
790 ret = config_writer_write_element_string(writer,
791 config_element_type, context_type_string);
792 if (ret) {
793 ret = LTTNG_ERR_SAVE_IO_FAIL;
794 goto end;
795 }
796 }
797
798 /* /context */
799 ret = config_writer_close_element(writer);
800 if (ret) {
801 ret = LTTNG_ERR_SAVE_IO_FAIL;
802 goto end;
803 }
804
805 end:
806 return ret;
807 }
808
809 static
810 int save_kernel_contexts(struct config_writer *writer,
811 struct ltt_kernel_channel *kchan)
812 {
813 int ret;
814 struct ltt_kernel_context *ctx;
815
816 if (cds_list_empty(&kchan->ctx_list)) {
817 ret = 0;
818 goto end;
819 }
820
821 ret = config_writer_open_element(writer, config_element_contexts);
822 if (ret) {
823 ret = LTTNG_ERR_SAVE_IO_FAIL;
824 goto end;
825 }
826
827 cds_list_for_each_entry(ctx, &kchan->ctx_list, list) {
828 ret = save_kernel_context(writer, &ctx->ctx);
829 if (ret) {
830 goto end;
831 }
832 }
833
834 /* /contexts */
835 ret = config_writer_close_element(writer);
836 if (ret) {
837 ret = LTTNG_ERR_SAVE_IO_FAIL;
838 goto end;
839 }
840 end:
841 return ret;
842 }
843
844 static
845 int save_ust_context(struct config_writer *writer,
846 struct cds_list_head *ctx_list)
847 {
848 int ret;
849 struct ltt_ust_context *ctx;
850
851 assert(writer);
852 assert(ctx_list);
853
854 ret = config_writer_open_element(writer, config_element_contexts);
855 if (ret) {
856 ret = LTTNG_ERR_SAVE_IO_FAIL;
857 goto end;
858 }
859
860 cds_list_for_each_entry(ctx, ctx_list, list) {
861 const char *context_type_string;
862
863
864 ret = config_writer_open_element(writer,
865 config_element_context);
866 if (ret) {
867 ret = LTTNG_ERR_SAVE_IO_FAIL;
868 goto end;
869 }
870
871 if (ctx->ctx.ctx == LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER) {
872 /* Perf contexts are saved as event_perf_context_type */
873 ret = config_writer_open_element(writer,
874 config_element_perf);
875 if (ret) {
876 ret = LTTNG_ERR_SAVE_IO_FAIL;
877 goto end;
878 }
879
880 ret = config_writer_write_element_unsigned_int(writer,
881 config_element_type,
882 ctx->ctx.u.perf_counter.type);
883 if (ret) {
884 ret = LTTNG_ERR_SAVE_IO_FAIL;
885 goto end;
886 }
887
888 ret = config_writer_write_element_unsigned_int(writer,
889 config_element_config,
890 ctx->ctx.u.perf_counter.config);
891 if (ret) {
892 ret = LTTNG_ERR_SAVE_IO_FAIL;
893 goto end;
894 }
895
896 ret = config_writer_write_element_string(writer,
897 config_element_name,
898 ctx->ctx.u.perf_counter.name);
899 if (ret) {
900 ret = LTTNG_ERR_SAVE_IO_FAIL;
901 goto end;
902 }
903
904 /* /perf */
905 ret = config_writer_close_element(writer);
906 if (ret) {
907 ret = LTTNG_ERR_SAVE_IO_FAIL;
908 goto end;
909 }
910 } else {
911 /* Save context as event_context_type_type */
912 context_type_string = get_ust_context_type_string(
913 ctx->ctx.ctx);
914 if (!context_type_string) {
915 ERR("Unsupported UST context type.")
916 ret = LTTNG_ERR_INVALID;
917 goto end;
918 }
919
920 ret = config_writer_write_element_string(writer,
921 config_element_type, context_type_string);
922 if (ret) {
923 ret = LTTNG_ERR_SAVE_IO_FAIL;
924 goto end;
925 }
926 }
927
928 /* /context */
929 ret = config_writer_close_element(writer);
930 if (ret) {
931 ret = LTTNG_ERR_SAVE_IO_FAIL;
932 goto end;
933 }
934 }
935
936 /* /contexts */
937 ret = config_writer_close_element(writer);
938 if (ret) {
939 ret = LTTNG_ERR_SAVE_IO_FAIL;
940 goto end;
941 }
942 end:
943 return ret;
944 }
945
946 static
947 int save_kernel_channel(struct config_writer *writer,
948 struct ltt_kernel_channel *kchan)
949 {
950 int ret;
951
952 assert(writer);
953 assert(kchan);
954
955 ret = config_writer_open_element(writer, config_element_channel);
956 if (ret) {
957 ret = LTTNG_ERR_SAVE_IO_FAIL;
958 goto end;
959 }
960
961 ret = config_writer_write_element_string(writer, config_element_name,
962 kchan->channel->name);
963 if (ret) {
964 ret = LTTNG_ERR_SAVE_IO_FAIL;
965 goto end;
966 }
967
968 ret = config_writer_write_element_bool(writer, config_element_enabled,
969 kchan->channel->enabled);
970 if (ret) {
971 ret = LTTNG_ERR_SAVE_IO_FAIL;
972 goto end;
973 }
974
975 ret = save_kernel_channel_attributes(writer, &kchan->channel->attr);
976 if (ret) {
977 goto end;
978 }
979
980 ret = save_kernel_events(writer, kchan);
981 if (ret) {
982 goto end;
983 }
984
985 ret = save_kernel_contexts(writer, kchan);
986 if (ret) {
987 goto end;
988 }
989
990 /* /channel */
991 ret = config_writer_close_element(writer);
992 if (ret) {
993 ret = LTTNG_ERR_SAVE_IO_FAIL;
994 goto end;
995 }
996 end:
997 return ret;
998 }
999
1000 static
1001 int save_ust_channel(struct config_writer *writer,
1002 struct ltt_ust_channel *ust_chan,
1003 struct ltt_ust_session *session)
1004 {
1005 int ret;
1006
1007 assert(writer);
1008 assert(ust_chan);
1009 assert(session);
1010
1011 ret = config_writer_open_element(writer, config_element_channel);
1012 if (ret) {
1013 ret = LTTNG_ERR_SAVE_IO_FAIL;
1014 goto end;
1015 }
1016
1017 ret = config_writer_write_element_string(writer, config_element_name,
1018 ust_chan->name);
1019 if (ret) {
1020 ret = LTTNG_ERR_SAVE_IO_FAIL;
1021 goto end;
1022 }
1023
1024 ret = config_writer_write_element_bool(writer, config_element_enabled,
1025 ust_chan->enabled);
1026 if (ret) {
1027 ret = LTTNG_ERR_SAVE_IO_FAIL;
1028 goto end;
1029 }
1030
1031 ret = save_ust_channel_attributes(writer, &ust_chan->attr);
1032 if (ret) {
1033 goto end;
1034 }
1035
1036 ret = config_writer_write_element_unsigned_int(writer,
1037 config_element_tracefile_size, ust_chan->tracefile_size);
1038 if (ret) {
1039 ret = LTTNG_ERR_SAVE_IO_FAIL;
1040 goto end;
1041 }
1042
1043 ret = config_writer_write_element_unsigned_int(writer,
1044 config_element_tracefile_count, ust_chan->tracefile_count);
1045 if (ret) {
1046 ret = LTTNG_ERR_SAVE_IO_FAIL;
1047 goto end;
1048 }
1049
1050 ret = config_writer_write_element_unsigned_int(writer,
1051 config_element_live_timer_interval,
1052 session->live_timer_interval);
1053 if (ret) {
1054 ret = LTTNG_ERR_SAVE_IO_FAIL;
1055 goto end;
1056 }
1057
1058 ret = save_ust_events(writer, ust_chan->events);
1059 if (ret) {
1060 ret = LTTNG_ERR_SAVE_IO_FAIL;
1061 goto end;
1062 }
1063
1064 ret = save_ust_context(writer, &ust_chan->ctx_list);
1065 if (ret) {
1066 goto end;
1067 }
1068
1069 /* /channel */
1070 ret = config_writer_close_element(writer);
1071 if (ret) {
1072 ret = LTTNG_ERR_SAVE_IO_FAIL;
1073 goto end;
1074 }
1075 end:
1076 return ret;
1077 }
1078
1079 static
1080 int save_kernel_session(struct config_writer *writer,
1081 struct ltt_session *session)
1082 {
1083 int ret;
1084 struct ltt_kernel_channel *kchan;
1085
1086 assert(writer);
1087 assert(session);
1088
1089 ret = config_writer_write_element_string(writer, config_element_type,
1090 config_domain_type_kernel);
1091 if (ret) {
1092 ret = LTTNG_ERR_SAVE_IO_FAIL;
1093 goto end;
1094 }
1095
1096 ret = config_writer_write_element_string(writer,
1097 config_element_buffer_type, config_buffer_type_global);
1098 if (ret) {
1099 ret = LTTNG_ERR_SAVE_IO_FAIL;
1100 goto end;
1101 }
1102
1103 ret = config_writer_open_element(writer,
1104 config_element_channels);
1105 if (ret) {
1106 ret = LTTNG_ERR_SAVE_IO_FAIL;
1107 goto end;
1108 }
1109
1110 cds_list_for_each_entry(kchan, &session->kernel_session->channel_list.head,
1111 list) {
1112 ret = save_kernel_channel(writer, kchan);
1113 if (ret) {
1114 goto end;
1115 }
1116 }
1117
1118 /* /channels */
1119 ret = config_writer_close_element(writer);
1120 if (ret) {
1121 ret = LTTNG_ERR_SAVE_IO_FAIL;
1122 goto end;
1123 }
1124 end:
1125 return ret;
1126 }
1127
1128 static
1129 int save_ust_session(struct config_writer *writer,
1130 struct ltt_session *session, int save_agent)
1131 {
1132 int ret;
1133 struct ltt_ust_channel *ust_chan;
1134 const char *buffer_type_string;
1135 struct lttng_ht_node_str *node;
1136 struct lttng_ht_iter iter;
1137
1138 assert(writer);
1139 assert(session);
1140
1141 ret = config_writer_write_element_string(writer, config_element_type,
1142 save_agent ? config_domain_type_jul : config_domain_type_ust);
1143 if (ret) {
1144 ret = LTTNG_ERR_SAVE_IO_FAIL;
1145 goto end;
1146 }
1147
1148 buffer_type_string = get_buffer_type_string(
1149 session->ust_session->buffer_type);
1150 if (!buffer_type_string) {
1151 ERR("Unsupported buffer type.");
1152 ret = LTTNG_ERR_INVALID;
1153 goto end;
1154 }
1155
1156 ret = config_writer_write_element_string(writer,
1157 config_element_buffer_type, buffer_type_string);
1158 if (ret) {
1159 ret = LTTNG_ERR_SAVE_IO_FAIL;
1160 goto end;
1161 }
1162
1163 ret = config_writer_open_element(writer, config_element_channels);
1164 if (ret) {
1165 ret = LTTNG_ERR_SAVE_IO_FAIL;
1166 goto end;
1167 }
1168
1169 rcu_read_lock();
1170 cds_lfht_for_each_entry(session->ust_session->domain_global.channels->ht,
1171 &iter.iter, node, node) {
1172 int agent_channel;
1173
1174 ust_chan = caa_container_of(node, struct ltt_ust_channel, node);
1175 agent_channel = !strcmp(DEFAULT_JUL_CHANNEL_NAME, ust_chan->name) ||
1176 !strcmp(DEFAULT_LOG4J_CHANNEL_NAME, ust_chan->name) ||
1177 !strcmp(DEFAULT_PYTHON_CHANNEL_NAME, ust_chan->name);
1178 if (!(save_agent ^ agent_channel)) {
1179 ret = save_ust_channel(writer, ust_chan, session->ust_session);
1180 if (ret) {
1181 rcu_read_unlock();
1182 goto end;
1183 }
1184 }
1185 }
1186 rcu_read_unlock();
1187
1188 /* /channels */
1189 ret = config_writer_close_element(writer);
1190 if (ret) {
1191 ret = LTTNG_ERR_SAVE_IO_FAIL;
1192 goto end;
1193 }
1194 end:
1195 return ret;
1196 }
1197
1198 static
1199 int save_pid_tracker(struct config_writer *writer,
1200 struct ltt_session *sess, int domain)
1201 {
1202 int ret = 0;
1203 ssize_t nr_pids = 0, i;
1204 int32_t *pids = NULL;
1205
1206 switch (domain) {
1207 case LTTNG_DOMAIN_KERNEL:
1208 {
1209 nr_pids = kernel_list_tracker_pids(sess->kernel_session, &pids);
1210 if (nr_pids < 0) {
1211 ret = LTTNG_ERR_KERN_LIST_FAIL;
1212 goto end;
1213 }
1214 break;
1215 }
1216 case LTTNG_DOMAIN_UST:
1217 {
1218 nr_pids = trace_ust_list_tracker_pids(sess->ust_session, &pids);
1219 if (nr_pids < 0) {
1220 ret = LTTNG_ERR_UST_LIST_FAIL;
1221 goto end;
1222 }
1223 break;
1224 }
1225 case LTTNG_DOMAIN_JUL:
1226 case LTTNG_DOMAIN_LOG4J:
1227 case LTTNG_DOMAIN_PYTHON:
1228 default:
1229 ret = LTTNG_ERR_UNKNOWN_DOMAIN;
1230 goto end;
1231 }
1232
1233 /* Only create a pid_tracker if enabled or untrack all */
1234 if (nr_pids != 1 || (nr_pids == 1 && pids[0] != -1)) {
1235 ret = config_writer_open_element(writer,
1236 config_element_pid_tracker);
1237 if (ret) {
1238 ret = LTTNG_ERR_SAVE_IO_FAIL;
1239 goto end;
1240 }
1241
1242 ret = config_writer_open_element(writer,
1243 config_element_targets);
1244 if (ret) {
1245 ret = LTTNG_ERR_SAVE_IO_FAIL;
1246 goto end;
1247 }
1248
1249 for (i = 0; i < nr_pids; i++) {
1250 ret = config_writer_open_element(writer,
1251 config_element_target_pid);
1252 if (ret) {
1253 ret = LTTNG_ERR_SAVE_IO_FAIL;
1254 goto end;
1255 }
1256
1257 ret = config_writer_write_element_unsigned_int(writer,
1258 config_element_pid, pids[i]);
1259 if (ret) {
1260 ret = LTTNG_ERR_SAVE_IO_FAIL;
1261 goto end;
1262 }
1263
1264 /* /pid_target */
1265 ret = config_writer_close_element(writer);
1266 if (ret) {
1267 ret = LTTNG_ERR_SAVE_IO_FAIL;
1268 goto end;
1269 }
1270 }
1271
1272 /* /targets */
1273 ret = config_writer_close_element(writer);
1274 if (ret) {
1275 ret = LTTNG_ERR_SAVE_IO_FAIL;
1276 goto end;
1277 }
1278
1279 /* /pid_tracker */
1280 ret = config_writer_close_element(writer);
1281 if (ret) {
1282 ret = LTTNG_ERR_SAVE_IO_FAIL;
1283 goto end;
1284 }
1285 }
1286 end:
1287 free(pids);
1288 return ret;
1289 }
1290
1291 static
1292 int save_domains(struct config_writer *writer, struct ltt_session *session)
1293 {
1294 int ret = 0;
1295
1296 assert(writer);
1297 assert(session);
1298
1299 if (!session->kernel_session && !session->ust_session) {
1300 goto end;
1301 }
1302
1303 ret = config_writer_open_element(writer, config_element_domains);
1304 if (ret) {
1305 ret = LTTNG_ERR_SAVE_IO_FAIL;
1306 goto end;
1307 }
1308
1309
1310 if (session->kernel_session) {
1311 ret = config_writer_open_element(writer,
1312 config_element_domain);
1313 if (ret) {
1314 ret = LTTNG_ERR_SAVE_IO_FAIL;
1315 goto end;
1316 }
1317
1318 ret = save_kernel_session(writer, session);
1319 if (ret) {
1320 goto end;
1321 }
1322
1323 ret = config_writer_open_element(writer,
1324 config_element_trackers);
1325 if (ret) {
1326 ret = LTTNG_ERR_SAVE_IO_FAIL;
1327 goto end;
1328 }
1329
1330 ret = save_pid_tracker(writer, session, LTTNG_DOMAIN_KERNEL);
1331 if (ret) {
1332 goto end;
1333 }
1334
1335 /* /trackers */
1336 ret = config_writer_close_element(writer);
1337 if (ret) {
1338 ret = LTTNG_ERR_SAVE_IO_FAIL;
1339 goto end;
1340 }
1341 /* /domain */
1342 ret = config_writer_close_element(writer);
1343 if (ret) {
1344 ret = LTTNG_ERR_SAVE_IO_FAIL;
1345 goto end;
1346 }
1347 }
1348
1349 if (session->ust_session) {
1350 unsigned long agent_count;
1351
1352 ret = config_writer_open_element(writer,
1353 config_element_domain);
1354 if (ret) {
1355 ret = LTTNG_ERR_SAVE_IO_FAIL;
1356 goto end;
1357 }
1358
1359 ret = save_ust_session(writer, session, 0);
1360 if (ret) {
1361 goto end;
1362 }
1363
1364 ret = config_writer_open_element(writer,
1365 config_element_trackers);
1366 if (ret) {
1367 ret = LTTNG_ERR_SAVE_IO_FAIL;
1368 goto end;
1369 }
1370
1371 ret = save_pid_tracker(writer, session, LTTNG_DOMAIN_UST);
1372 if (ret) {
1373 goto end;
1374 }
1375
1376 /* /trackers */
1377 ret = config_writer_close_element(writer);
1378 if (ret) {
1379 ret = LTTNG_ERR_SAVE_IO_FAIL;
1380 goto end;
1381 }
1382 /* /domain */
1383 ret = config_writer_close_element(writer);
1384 if (ret) {
1385 ret = LTTNG_ERR_SAVE_IO_FAIL;
1386 goto end;
1387 }
1388
1389 agent_count = lttng_ht_get_count(session->ust_session->agents);
1390 if (agent_count > 0) {
1391 ret = config_writer_open_element(writer,
1392 config_element_domain);
1393 if (ret) {
1394 ret = LTTNG_ERR_SAVE_IO_FAIL;
1395 goto end;
1396 }
1397
1398 ret = save_ust_session(writer, session, 1);
1399 if (ret) {
1400 goto end;
1401 }
1402
1403 /* /domain */
1404 ret = config_writer_close_element(writer);
1405 if (ret) {
1406 ret = LTTNG_ERR_SAVE_IO_FAIL;
1407 goto end;
1408 }
1409 }
1410 }
1411
1412 /* /domains */
1413 ret = config_writer_close_element(writer);
1414 if (ret) {
1415 ret = LTTNG_ERR_SAVE_IO_FAIL;
1416 goto end;
1417 }
1418 end:
1419 return ret;
1420 }
1421
1422 static
1423 int save_consumer_output(struct config_writer *writer,
1424 struct consumer_output *output)
1425 {
1426 int ret;
1427
1428 assert(writer);
1429 assert(output);
1430
1431 ret = config_writer_open_element(writer, config_element_consumer_output);
1432 if (ret) {
1433 ret = LTTNG_ERR_SAVE_IO_FAIL;
1434 goto end;
1435 }
1436
1437 ret = config_writer_write_element_bool(writer, config_element_enabled,
1438 output->enabled);
1439 if (ret) {
1440 ret = LTTNG_ERR_SAVE_IO_FAIL;
1441 goto end;
1442 }
1443
1444 ret = config_writer_open_element(writer, config_element_destination);
1445 if (ret) {
1446 ret = LTTNG_ERR_SAVE_IO_FAIL;
1447 goto end;
1448 }
1449
1450 switch (output->type) {
1451 case CONSUMER_DST_LOCAL:
1452 ret = config_writer_write_element_string(writer,
1453 config_element_path, output->dst.trace_path);
1454 if (ret) {
1455 ret = LTTNG_ERR_SAVE_IO_FAIL;
1456 goto end;
1457 }
1458 break;
1459 case CONSUMER_DST_NET:
1460 {
1461 char *uri;
1462
1463 uri = zmalloc(PATH_MAX);
1464 if (!uri) {
1465 ret = LTTNG_ERR_NOMEM;
1466 goto end;
1467 }
1468
1469 ret = config_writer_open_element(writer, config_element_net_output);
1470 if (ret) {
1471 ret = LTTNG_ERR_SAVE_IO_FAIL;
1472 goto end_net_output;
1473 }
1474
1475 if (output->dst.net.control_isset &&
1476 output->dst.net.data_isset) {
1477 ret = uri_to_str_url(&output->dst.net.control, uri, PATH_MAX);
1478 if (ret < 0) {
1479 ret = LTTNG_ERR_INVALID;
1480 goto end_net_output;
1481 }
1482
1483 ret = config_writer_write_element_string(writer,
1484 config_element_control_uri, uri);
1485 if (ret) {
1486 ret = LTTNG_ERR_SAVE_IO_FAIL;
1487 goto end_net_output;
1488 }
1489
1490 ret = uri_to_str_url(&output->dst.net.data, uri, PATH_MAX);
1491 if (ret < 0) {
1492 ret = LTTNG_ERR_INVALID;
1493 goto end_net_output;
1494 }
1495
1496 ret = config_writer_write_element_string(writer,
1497 config_element_data_uri, uri);
1498 if (ret) {
1499 ret = LTTNG_ERR_SAVE_IO_FAIL;
1500 goto end_net_output;
1501 }
1502
1503 end_net_output:
1504 free(uri);
1505 if (ret) {
1506 goto end;
1507 }
1508 } else {
1509 ret = !output->dst.net.control_isset ?
1510 LTTNG_ERR_URL_CTRL_MISS :
1511 LTTNG_ERR_URL_DATA_MISS;
1512 free(uri);
1513 goto end;
1514 }
1515
1516 ret = config_writer_close_element(writer);
1517 if (ret) {
1518 ret = LTTNG_ERR_SAVE_IO_FAIL;
1519 goto end;
1520 }
1521 break;
1522 }
1523 default:
1524 ERR("Unsupported consumer output type.");
1525 ret = LTTNG_ERR_INVALID;
1526 goto end;
1527 }
1528
1529 /* /destination */
1530 ret = config_writer_close_element(writer);
1531 if (ret) {
1532 ret = LTTNG_ERR_SAVE_IO_FAIL;
1533 goto end;
1534 }
1535
1536 /* /consumer_output */
1537 ret = config_writer_close_element(writer);
1538 if (ret) {
1539 ret = LTTNG_ERR_SAVE_IO_FAIL;
1540 goto end;
1541 }
1542 end:
1543 return ret;
1544 }
1545
1546 static
1547 int save_snapshot_outputs(struct config_writer *writer,
1548 struct snapshot *snapshot)
1549 {
1550 int ret;
1551 struct lttng_ht_iter iter;
1552 struct snapshot_output *output;
1553
1554 assert(writer);
1555 assert(snapshot);
1556
1557 ret = config_writer_open_element(writer, config_element_snapshot_outputs);
1558 if (ret) {
1559 ret = LTTNG_ERR_SAVE_IO_FAIL;
1560 goto end;
1561 }
1562
1563 rcu_read_lock();
1564 cds_lfht_for_each_entry(snapshot->output_ht->ht, &iter.iter, output,
1565 node.node) {
1566 ret = config_writer_open_element(writer,
1567 config_element_output);
1568 if (ret) {
1569 ret = LTTNG_ERR_SAVE_IO_FAIL;
1570 goto end_unlock;
1571 }
1572
1573 ret = config_writer_write_element_string(writer,
1574 config_element_name, output->name);
1575 if (ret) {
1576 ret = LTTNG_ERR_SAVE_IO_FAIL;
1577 goto end_unlock;
1578 }
1579
1580 ret = config_writer_write_element_unsigned_int(writer,
1581 config_element_max_size, output->max_size);
1582 if (ret) {
1583 ret = LTTNG_ERR_SAVE_IO_FAIL;
1584 goto end_unlock;
1585 }
1586
1587 ret = save_consumer_output(writer, output->consumer);
1588 if (ret) {
1589 goto end_unlock;
1590 }
1591
1592 /* /output */
1593 ret = config_writer_close_element(writer);
1594 if (ret) {
1595 ret = LTTNG_ERR_SAVE_IO_FAIL;
1596 goto end_unlock;
1597 }
1598 }
1599 rcu_read_unlock();
1600
1601 /* /snapshot_outputs */
1602 ret = config_writer_close_element(writer);
1603 if (ret) {
1604 ret = LTTNG_ERR_SAVE_IO_FAIL;
1605 goto end;
1606 }
1607
1608 end:
1609 return ret;
1610 end_unlock:
1611 rcu_read_unlock();
1612 return ret;
1613 }
1614
1615 static
1616 int save_session_output(struct config_writer *writer,
1617 struct ltt_session *session)
1618 {
1619 int ret;
1620
1621 assert(writer);
1622 assert(session);
1623
1624 if ((session->snapshot_mode && session->snapshot.nb_output == 0) ||
1625 (!session->snapshot_mode && !session->consumer)) {
1626 /* Session is in no output mode */
1627 ret = 0;
1628 goto end;
1629 }
1630
1631 ret = config_writer_open_element(writer, config_element_output);
1632 if (ret) {
1633 ret = LTTNG_ERR_SAVE_IO_FAIL;
1634 goto end;
1635 }
1636
1637 if (session->snapshot_mode) {
1638 ret = save_snapshot_outputs(writer, &session->snapshot);
1639 if (ret) {
1640 goto end;
1641 }
1642 } else {
1643 if (session->consumer) {
1644 ret = save_consumer_output(writer, session->consumer);
1645 if (ret) {
1646 goto end;
1647 }
1648 }
1649 }
1650
1651 /* /output */
1652 ret = config_writer_close_element(writer);
1653 if (ret) {
1654 ret = LTTNG_ERR_SAVE_IO_FAIL;
1655 goto end;
1656 }
1657 end:
1658 return ret;
1659 }
1660
1661 /*
1662 * Save the given session.
1663 *
1664 * Return 0 on success else a LTTNG_ERR* code.
1665 */
1666 static
1667 int save_session(struct ltt_session *session,
1668 struct lttng_save_session_attr *attr, lttng_sock_cred *creds)
1669 {
1670 int ret, fd;
1671 unsigned int file_opened = 0; /* Indicate if the file has been opened */
1672 char config_file_path[PATH_MAX];
1673 size_t len;
1674 struct config_writer *writer = NULL;
1675 size_t session_name_len;
1676 const char *provided_path;
1677
1678 assert(session);
1679 assert(attr);
1680 assert(creds);
1681
1682 session_name_len = strlen(session->name);
1683 memset(config_file_path, 0, sizeof(config_file_path));
1684
1685 if (!session_access_ok(session,
1686 LTTNG_SOCK_GET_UID_CRED(creds),
1687 LTTNG_SOCK_GET_GID_CRED(creds))) {
1688 ret = LTTNG_ERR_EPERM;
1689 goto end;
1690 }
1691
1692 provided_path = lttng_save_session_attr_get_output_url(attr);
1693 if (provided_path) {
1694 DBG3("Save session in provided path %s", provided_path);
1695 len = strlen(provided_path);
1696 if (len >= sizeof(config_file_path)) {
1697 ret = LTTNG_ERR_SET_URL;
1698 goto end;
1699 }
1700 strncpy(config_file_path, provided_path, len);
1701 } else {
1702 ssize_t ret_len;
1703 char *home_dir = utils_get_user_home_dir(
1704 LTTNG_SOCK_GET_UID_CRED(creds));
1705 if (!home_dir) {
1706 ret = LTTNG_ERR_SET_URL;
1707 goto end;
1708 }
1709
1710 ret_len = snprintf(config_file_path, sizeof(config_file_path),
1711 DEFAULT_SESSION_HOME_CONFIGPATH, home_dir);
1712 free(home_dir);
1713 if (ret_len < 0) {
1714 PERROR("snprintf save session");
1715 ret = LTTNG_ERR_SET_URL;
1716 goto end;
1717 }
1718 len = ret_len;
1719 }
1720
1721 /*
1722 * Check the path fits in the config file path dst including the '/'
1723 * followed by trailing .lttng extension and the NULL terminated string.
1724 */
1725 if ((len + session_name_len + 2 +
1726 sizeof(DEFAULT_SESSION_CONFIG_FILE_EXTENSION))
1727 > sizeof(config_file_path)) {
1728 ret = LTTNG_ERR_SET_URL;
1729 goto end;
1730 }
1731
1732 ret = run_as_mkdir_recursive(config_file_path, S_IRWXU | S_IRWXG,
1733 LTTNG_SOCK_GET_UID_CRED(creds), LTTNG_SOCK_GET_GID_CRED(creds));
1734 if (ret) {
1735 ret = LTTNG_ERR_SET_URL;
1736 goto end;
1737 }
1738
1739 /*
1740 * At this point, we know that everything fits in the buffer. Validation
1741 * was done just above.
1742 */
1743 config_file_path[len++] = '/';
1744 strncpy(config_file_path + len, session->name, session_name_len);
1745 len += session_name_len;
1746 strcpy(config_file_path + len, DEFAULT_SESSION_CONFIG_FILE_EXTENSION);
1747 len += sizeof(DEFAULT_SESSION_CONFIG_FILE_EXTENSION);
1748 config_file_path[len] = '\0';
1749
1750 if (!access(config_file_path, F_OK) && !attr->overwrite) {
1751 /* File exists, notify the user since the overwrite flag is off. */
1752 ret = LTTNG_ERR_SAVE_FILE_EXIST;
1753 goto end;
1754 }
1755
1756 fd = run_as_open(config_file_path, O_CREAT | O_WRONLY | O_TRUNC,
1757 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP,
1758 LTTNG_SOCK_GET_UID_CRED(creds), LTTNG_SOCK_GET_GID_CRED(creds));
1759 if (fd < 0) {
1760 PERROR("Could not create configuration file");
1761 ret = LTTNG_ERR_SAVE_IO_FAIL;
1762 goto end;
1763 }
1764 file_opened = 1;
1765
1766 writer = config_writer_create(fd, 1);
1767 if (!writer) {
1768 ret = LTTNG_ERR_NOMEM;
1769 goto end;
1770 }
1771
1772 ret = config_writer_open_element(writer, config_element_sessions);
1773 if (ret) {
1774 ret = LTTNG_ERR_SAVE_IO_FAIL;
1775 goto end;
1776 }
1777
1778 ret = config_writer_open_element(writer, config_element_session);
1779 if (ret) {
1780 ret = LTTNG_ERR_SAVE_IO_FAIL;
1781 goto end;
1782 }
1783
1784 ret = config_writer_write_element_string(writer, config_element_name,
1785 session->name);
1786 if (ret) {
1787 ret = LTTNG_ERR_SAVE_IO_FAIL;
1788 goto end;
1789 }
1790
1791 if(session->shm_path[0] != '\0') {
1792 ret = config_writer_write_element_string(writer,
1793 config_element_shared_memory_path,
1794 session->shm_path);
1795 if (ret) {
1796 ret = LTTNG_ERR_SAVE_IO_FAIL;
1797 goto end;
1798 }
1799 }
1800
1801 ret = save_domains(writer, session);
1802 if (ret) {
1803 goto end;
1804 }
1805
1806 ret = config_writer_write_element_bool(writer, config_element_started,
1807 session->active);
1808 if (ret) {
1809 ret = LTTNG_ERR_SAVE_IO_FAIL;
1810 goto end;
1811 }
1812
1813 if (session->snapshot_mode || session->live_timer) {
1814 ret = config_writer_open_element(writer, config_element_attributes);
1815 if (ret) {
1816 ret = LTTNG_ERR_SAVE_IO_FAIL;
1817 goto end;
1818 }
1819
1820 if (session->snapshot_mode) {
1821 ret = config_writer_write_element_bool(writer,
1822 config_element_snapshot_mode, 1);
1823 if (ret) {
1824 ret = LTTNG_ERR_SAVE_IO_FAIL;
1825 goto end;
1826 }
1827 } else {
1828 ret = config_writer_write_element_unsigned_int(writer,
1829 config_element_live_timer_interval, session->live_timer);
1830 if (ret) {
1831 ret = LTTNG_ERR_SAVE_IO_FAIL;
1832 goto end;
1833 }
1834 }
1835
1836 /* /attributes */
1837 ret = config_writer_close_element(writer);
1838 if (ret) {
1839 ret = LTTNG_ERR_SAVE_IO_FAIL;
1840 goto end;
1841 }
1842 }
1843
1844 ret = save_session_output(writer, session);
1845 if (ret) {
1846 goto end;
1847 }
1848
1849 /* /session */
1850 ret = config_writer_close_element(writer);
1851 if (ret) {
1852 ret = LTTNG_ERR_SAVE_IO_FAIL;
1853 goto end;
1854 }
1855
1856 /* /sessions */
1857 ret = config_writer_close_element(writer);
1858 if (ret) {
1859 ret = LTTNG_ERR_SAVE_IO_FAIL;
1860 goto end;
1861 }
1862 end:
1863 if (writer && config_writer_destroy(writer)) {
1864 /* Preserve the original error code */
1865 ret = ret ? ret : LTTNG_ERR_SAVE_IO_FAIL;
1866 }
1867 if (ret) {
1868 /* Delete file in case of error */
1869 if (file_opened && unlink(config_file_path)) {
1870 PERROR("Unlinking XML session configuration.");
1871 }
1872 }
1873
1874 return ret;
1875 }
1876
1877 int cmd_save_sessions(struct lttng_save_session_attr *attr,
1878 lttng_sock_cred *creds)
1879 {
1880 int ret;
1881 const char *session_name;
1882 struct ltt_session *session;
1883
1884 session_lock_list();
1885
1886 session_name = lttng_save_session_attr_get_session_name(attr);
1887 if (session_name) {
1888 session = session_find_by_name(session_name);
1889 if (!session) {
1890 ret = LTTNG_ERR_SESS_NOT_FOUND;
1891 goto end;
1892 }
1893
1894 session_lock(session);
1895 ret = save_session(session, attr, creds);
1896 session_unlock(session);
1897 if (ret) {
1898 goto end;
1899 }
1900 } else {
1901 struct ltt_session_list *list = session_get_list();
1902
1903 cds_list_for_each_entry(session, &list->head, list) {
1904 session_lock(session);
1905 ret = save_session(session, attr, creds);
1906 session_unlock(session);
1907
1908 /* Don't abort if we don't have the required permissions. */
1909 if (ret && ret != LTTNG_ERR_EPERM) {
1910 goto end;
1911 }
1912 }
1913 }
1914 ret = LTTNG_OK;
1915
1916 end:
1917 session_unlock_list();
1918 return ret;
1919 }
LTTng tools - Deliverables
This page took 0.104447 seconds and 6 git commands to generate.