2 * Copyright (C) 2011-2012 Julien Desfossez
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 as
6 * published by the Free Software Foundation;
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.
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.
22 #include <babeltrace/babeltrace.h>
23 #include <babeltrace/ctf/events.h>
24 #include <babeltrace/ctf/callbacks.h>
25 #include <babeltrace/ctf/iterator.h>
37 #include <sys/types.h>
41 #include <lttng/lttng.h>
42 #include <lttng/lttngtop-helper.h>
43 #include <babeltrace/lttngtopmmappacketseek.h>
45 #include "lttngtoptypes.h"
47 #include "iostreamtop.h"
48 #include "cursesdisplay.h"
51 #define DEFAULT_FILE_ARRAY_SIZE 1
53 const char *opt_input_path
;
54 static int opt_textdump
;
59 struct lttngtop
*copy
;
60 pthread_t display_thread
;
61 pthread_t timer_thread
;
63 unsigned long refresh_display
= 1 * NSEC_PER_SEC
;
64 unsigned long last_display_update
= 0;
66 /* list of FDs available for being read with snapshots */
67 struct mmap_stream_list mmap_list
;
68 GPtrArray
*lttng_consumer_stream_array
;
69 int sessiond_metadata
, consumerd_metadata
;
70 struct lttng_consumer_local_data
*ctx
= NULL
;
71 /* list of snapshots currently not consumed */
72 GPtrArray
*available_snapshots
;
73 sem_t metadata_available
;
76 int last_textdump_print_newline
= 1;
88 static struct poptOption long_options
[] = {
89 /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
90 { "help", 'h', POPT_ARG_NONE
, NULL
, OPT_HELP
, NULL
, NULL
},
91 { "textdump", 't', POPT_ARG_NONE
, NULL
, OPT_TEXTDUMP
, NULL
, NULL
},
92 { "child", 'f', POPT_ARG_NONE
, NULL
, OPT_CHILD
, NULL
, NULL
},
93 { "pid", 'p', POPT_ARG_STRING
, &opt_tid
, OPT_PID
, NULL
, NULL
},
94 { "hostname", 'n', POPT_ARG_STRING
, &opt_hostname
, OPT_HOSTNAME
, NULL
, NULL
},
95 { "kprobes", 'k', POPT_ARG_STRING
, &opt_kprobes
, OPT_KPROBES
, NULL
, NULL
},
96 { NULL
, 0, 0, NULL
, 0, NULL
, NULL
},
99 static void handle_textdump_sigterm(int signal
)
102 lttng_destroy_session("test");
105 void *refresh_thread(void *p
)
107 struct mmap_stream
*mmap_info
;
111 sem_post(&pause_sem
);
113 sem_post(&end_trace_sem
);
114 sem_post(&goodtodisplay
);
115 sem_post(&goodtoupdate
);
118 if (!opt_input_path
) {
119 bt_list_for_each_entry(mmap_info
, &mmap_list
.head
, list
)
120 helper_kernctl_buffer_flush(mmap_info
->fd
);
122 sem_wait(&pause_sem
);
123 sem_post(&pause_sem
);
125 sleep(refresh_display
/NSEC_PER_SEC
);
129 void *ncurses_display(void *p
)
131 unsigned int current_display_index
= 0;
133 sem_wait(&bootstrap
);
135 * Prevent the 1 second delay when we hit ESC
142 sem_wait(&goodtodisplay
);
143 sem_wait(&pause_sem
);
146 sem_post(&pause_sem
);
152 copy
= g_ptr_array_index(copies
, current_display_index
);
154 display(current_display_index
++);
156 sem_post(&goodtoupdate
);
157 sem_post(&pause_sem
);
161 void print_fields(struct bt_ctf_event
*event
)
164 const struct definition
*const * list
;
165 const struct definition
*scope
;
166 enum ctf_type_id type
;
169 scope
= bt_ctf_get_top_level_scope(event
, BT_EVENT_FIELDS
);
171 bt_ctf_get_field_list(event
, scope
, &list
, &cnt
);
172 for (i
= 0; i
< cnt
; i
++) {
175 printf("%s = ", bt_ctf_field_name(list
[i
]));
176 type
= bt_ctf_field_type(list
[i
]);
177 if (type
== CTF_TYPE_INTEGER
) {
178 if (bt_ctf_get_int_signedness(list
[i
]) == 0)
179 printf("%" PRIu64
"", bt_ctf_get_uint64(list
[i
]));
181 printf("%" PRId64
"", bt_ctf_get_int64(list
[i
]));
182 } else if (type
== CTF_TYPE_STRING
) {
183 printf("%s", bt_ctf_get_string(list
[i
]));
184 } else if (type
== CTF_TYPE_ARRAY
) {
185 str
= bt_ctf_get_char_array(list
[i
]);
186 if (!bt_ctf_field_get_error() && str
)
193 * hook on each event to check the timestamp and refresh the display if
196 enum bt_cb_ret
print_timestamp(struct bt_ctf_event
*call_data
, void *private_data
)
198 unsigned long timestamp
;
200 uint64_t ts_nsec_start
;
203 const struct definition
*scope
;
204 const char *hostname
, *procname
;
206 timestamp
= bt_ctf_get_timestamp(call_data
);
208 start
= format_timestamp(timestamp
);
209 ts_nsec_start
= timestamp
% NSEC_PER_SEC
;
211 pid
= get_context_pid(call_data
);
212 if (pid
== -1ULL && opt_tid
) {
216 hostname
= get_context_hostname(call_data
);
217 if (opt_tid
|| opt_hostname
)
218 if (!lookup_filter_tid_list(pid
))
221 cpu_id
= get_cpu_id(call_data
);
222 procname
= get_context_comm(call_data
);
224 if ((strcmp(bt_ctf_event_name(call_data
), "exit_syscall") == 0) &&
225 !last_textdump_print_newline
) {
226 scope
= bt_ctf_get_top_level_scope(call_data
,
228 syscall_ret
= bt_ctf_get_int64(bt_ctf_get_field(call_data
,
230 printf("= %ld\n", syscall_ret
);
231 last_textdump_print_newline
= 1;
233 /* we might have lost the exit_syscall event, so need to
234 * print the newline in this case */
235 if (last_textdump_print_newline
== 0)
237 printf("%02d:%02d:%02d.%09" PRIu64
" (%s) (cpu %d) [%s (%d)] %s (",
238 start
.tm_hour
, start
.tm_min
, start
.tm_sec
,
239 ts_nsec_start
, hostname
, cpu_id
, procname
, pid
,
240 bt_ctf_event_name(call_data
));
241 print_fields(call_data
);
243 if (strncmp(bt_ctf_event_name(call_data
), "sys_", 4) != 0) {
245 last_textdump_print_newline
= 1;
247 last_textdump_print_newline
= 0;
254 return BT_CB_ERROR_STOP
;
257 enum bt_cb_ret
handle_kprobes(struct bt_ctf_event
*call_data
, void *private_data
)
260 struct kprobes
*kprobe
;
263 for (i
= 0; i
< lttngtop
.kprobes_table
->len
; i
++) {
264 kprobe
= g_ptr_array_index(lttngtop
.kprobes_table
, i
);
265 if (strcmp(bt_ctf_event_name(call_data
), kprobe
->probe_name
) == 0) {
274 * hook on each event to check the timestamp and refresh the display if
277 enum bt_cb_ret
check_timestamp(struct bt_ctf_event
*call_data
, void *private_data
)
279 unsigned long timestamp
;
281 timestamp
= bt_ctf_get_timestamp(call_data
);
282 if (timestamp
== -1ULL)
285 if (last_display_update
== 0)
286 last_display_update
= timestamp
;
288 if (timestamp
- last_display_update
>= refresh_display
) {
289 sem_wait(&goodtoupdate
);
290 g_ptr_array_add(copies
, get_copy_lttngtop(last_display_update
,
292 sem_post(&goodtodisplay
);
293 sem_post(&bootstrap
);
294 last_display_update
= timestamp
;
299 fprintf(stderr
, "check_timestamp callback error\n");
300 return BT_CB_ERROR_STOP
;
304 * get_perf_counter : get or create and return a perf_counter struct for
305 * either a process or a cpu (only one of the 2 parameters mandatory)
307 struct perfcounter
*get_perf_counter(const char *name
, struct processtop
*proc
,
310 struct perfcounter
*ret
;
320 ret
= g_hash_table_lookup(table
, (gpointer
) name
);
324 ret
= g_new0(struct perfcounter
, 1);
325 /* by default, make it visible in the UI */
327 g_hash_table_insert(table
, (gpointer
) strdup(name
), ret
);
336 void update_perf_value(struct processtop
*proc
, struct cputime
*cpu
,
337 const char *name
, int value
)
339 struct perfcounter
*cpu_perf
, *process_perf
;
341 cpu_perf
= get_perf_counter(name
, NULL
, cpu
);
342 if (cpu_perf
->count
< value
) {
343 process_perf
= get_perf_counter(name
, proc
, NULL
);
344 process_perf
->count
+= value
- cpu_perf
->count
;
345 cpu_perf
->count
= value
;
349 void extract_perf_counter_scope(const struct bt_ctf_event
*event
,
350 const struct bt_definition
*scope
,
351 struct processtop
*proc
,
354 struct bt_definition
const * const *list
= NULL
;
355 const struct bt_definition
*field
;
357 struct perfcounter
*perfcounter
;
365 ret
= bt_ctf_get_field_list(event
, scope
, &list
, &count
);
372 g_hash_table_iter_init(&iter
, global_perf_liszt
);
373 while (g_hash_table_iter_next (&iter
, &key
, (gpointer
) &perfcounter
)) {
374 field
= bt_ctf_get_field(event
, scope
, (char *) key
);
376 int value
= bt_ctf_get_uint64(field
);
377 if (bt_ctf_field_get_error())
379 update_perf_value(proc
, cpu
, (char *) key
, value
);
387 void update_perf_counter(struct processtop
*proc
, const struct bt_ctf_event
*event
)
390 const struct bt_definition
*scope
;
392 cpu
= get_cpu(get_cpu_id(event
));
394 scope
= bt_ctf_get_top_level_scope(event
, BT_STREAM_EVENT_CONTEXT
);
395 extract_perf_counter_scope(event
, scope
, proc
, cpu
);
397 scope
= bt_ctf_get_top_level_scope(event
, BT_STREAM_PACKET_CONTEXT
);
398 extract_perf_counter_scope(event
, scope
, proc
, cpu
);
400 scope
= bt_ctf_get_top_level_scope(event
, BT_EVENT_CONTEXT
);
401 extract_perf_counter_scope(event
, scope
, proc
, cpu
);
404 enum bt_cb_ret
fix_process_table(struct bt_ctf_event
*call_data
,
407 int pid
, tid
, ppid
, vpid
, vtid
, vppid
;
408 char *comm
, *hostname
;
409 struct processtop
*parent
, *child
;
410 unsigned long timestamp
;
412 timestamp
= bt_ctf_get_timestamp(call_data
);
413 if (timestamp
== -1ULL)
416 pid
= get_context_pid(call_data
);
421 tid
= get_context_tid(call_data
);
425 ppid
= get_context_ppid(call_data
);
429 vpid
= get_context_vpid(call_data
);
433 vtid
= get_context_vtid(call_data
);
437 vppid
= get_context_vppid(call_data
);
441 comm
= get_context_comm(call_data
);
446 hostname
= get_context_hostname(call_data
);
448 /* find or create the current process */
449 child
= find_process_tid(<tngtop
, tid
, comm
);
451 child
= add_proc(<tngtop
, tid
, comm
, timestamp
, hostname
);
454 update_proc(child
, pid
, tid
, ppid
, vpid
, vtid
, vppid
, comm
, hostname
);
457 /* find or create the parent */
458 parent
= find_process_tid(<tngtop
, pid
, comm
);
460 parent
= add_proc(<tngtop
, pid
, comm
, timestamp
, hostname
);
465 /* attach the parent to the current process */
466 child
->threadparent
= parent
;
467 add_thread(parent
, child
);
470 update_perf_counter(child
, call_data
);
476 return BT_CB_ERROR_STOP
;
481 copies
= g_ptr_array_new();
482 global_perf_liszt
= g_hash_table_new(g_str_hash
, g_str_equal
);
483 global_filter_list
= g_hash_table_new(g_str_hash
, g_str_equal
);
485 sem_init(&goodtodisplay
, 0, 0);
486 sem_init(&goodtoupdate
, 0, 1);
487 sem_init(&timer
, 0, 1);
488 sem_init(&bootstrap
, 0, 0);
489 sem_init(&pause_sem
, 0, 1);
490 sem_init(&end_trace_sem
, 0, 0);
492 reset_global_counters();
494 lttngtop
.nbthreads
= 0;
495 lttngtop
.nbfiles
= 0;
497 lttngtop
.process_hash_table
= g_hash_table_new(g_direct_hash
,
499 lttngtop
.process_table
= g_ptr_array_new();
500 lttngtop
.files_table
= g_ptr_array_new();
501 lttngtop
.cpu_table
= g_ptr_array_new();
508 fprintf(fp
, "LTTngTop %s\n\n", VERSION
);
509 fprintf(fp
, "Usage : lttngtop [OPTIONS] [TRACE]\n");
510 fprintf(fp
, " TRACE Path to the trace to analyse (no trace path for live tracing)\n");
511 fprintf(fp
, " -h, --help This help message\n");
512 fprintf(fp
, " -t, --textdump Display live events in text-only\n");
513 fprintf(fp
, " -p, --pid Comma-separated list of PIDs to display\n");
514 fprintf(fp
, " -f, --child Follow threads associated with selected PIDs\n");
515 fprintf(fp
, " -n, --hostname Comma-separated list of hostnames to display (require hostname context in trace)\n");
516 fprintf(fp
, " -k, --kprobes Comma-separated list of kprobes to insert (same format as lttng enable-event)\n");
520 * Parse probe options.
521 * Shamelessly stolen from lttng-tools :
522 * src/bin/lttng/commands/enable_events.c
524 static struct kprobes
*parse_probe_opts(char *opt
)
527 char name
[LTTNG_SYMBOL_NAME_LEN
];
528 struct kprobes
*kprobe
;
532 kprobe->probe_addr = 0;
533 kprobe->probe_offset = 0;
534 asprintf(&kprobe->probe_name, "probe_sys_open");
535 asprintf(&kprobe->symbol_name, "sys_open");
543 kprobe
= g_new0(struct kprobes
, 1);
545 /* Check for symbol+offset */
546 ret
= sscanf(opt
, "%[^'+']+%s", name
, s_hex
);
548 asprintf(&kprobe
->probe_name
, "probe_%s", name
);
549 asprintf(&kprobe
->symbol_name
, "%s", name
);
551 if (strlen(s_hex
) == 0) {
552 fprintf(stderr
, "Invalid probe offset %s", s_hex
);
556 kprobe
->probe_offset
= strtoul(s_hex
, NULL
, 0);
557 kprobe
->probe_addr
= 0;
561 /* Check for symbol */
562 if (isalpha(name
[0])) {
563 ret
= sscanf(opt
, "%s", name
);
565 asprintf(&kprobe
->probe_name
, "probe_%s", name
);
566 asprintf(&kprobe
->symbol_name
, "%s", name
);
567 kprobe
->probe_offset
= 0;
568 kprobe
->probe_addr
= 0;
573 /* Check for address */
574 ret
= sscanf(opt
, "%s", s_hex
);
576 if (strlen(s_hex
) == 0) {
577 fprintf(stderr
, "Invalid probe address %s", s_hex
);
581 asprintf(&kprobe
->probe_name
, "probe_%s", s_hex
);
582 kprobe
->probe_offset
= 0;
583 kprobe
->probe_addr
= strtoul(s_hex
, NULL
, 0);
595 * Return 0 if caller should continue, < 0 if caller should return
596 * error, > 0 if caller should exit without reporting error.
598 static int parse_options(int argc
, char **argv
)
605 pc
= poptGetContext(NULL
, argc
, (const char **) argv
, long_options
, 0);
606 poptReadDefaultConfig(pc
, 0);
608 while ((opt
= poptGetNextOpt(pc
)) != -1) {
612 ret
= 1; /* exit cleanly */
623 tid_filter_list
= g_hash_table_new(g_str_hash
,
625 tmp_str
= strtok(opt_tid
, ",");
627 tid
= malloc(sizeof(int));
628 *tid
= atoi(tmp_str
);
629 g_hash_table_insert(tid_filter_list
,
630 (gpointer
) tid
, tid
);
631 tmp_str
= strtok(NULL
, ",");
636 hostname_filter_list
= g_hash_table_new(g_str_hash
,
638 tmp_str
= strtok(opt_hostname
, ",");
640 char *new_str
= strdup(tmp_str
);
641 g_hash_table_insert(hostname_filter_list
,
644 tmp_str
= strtok(NULL
, ",");
648 lttngtop
.kprobes_table
= g_ptr_array_new();
649 tmp_str
= strtok(opt_kprobes
, ",");
651 struct kprobes
*kprobe
;
653 kprobe
= parse_probe_opts(tmp_str
);
656 lttngtop
.kprobes_table
,
662 tmp_str
= strtok(NULL
, ",");
671 opt_input_path
= poptGetArg(pc
);
680 void iter_trace(struct bt_context
*bt_ctx
)
682 struct bt_ctf_iter
*iter
;
683 struct bt_iter_pos begin_pos
;
684 struct kprobes
*kprobe
;
685 const struct bt_ctf_event
*event
;
689 begin_pos
.type
= BT_SEEK_BEGIN
;
690 iter
= bt_ctf_iter_create(bt_ctx
, &begin_pos
, NULL
);
692 /* at each event, verify the status of the process table */
693 bt_ctf_iter_add_callback(iter
, 0, NULL
, 0,
697 bt_ctf_iter_add_callback(iter
, 0, NULL
, 0,
701 /* at each event check if we need to refresh */
702 bt_ctf_iter_add_callback(iter
, 0, NULL
, 0,
705 /* to handle the scheduling events */
706 bt_ctf_iter_add_callback(iter
,
707 g_quark_from_static_string("sched_switch"),
708 NULL
, 0, handle_sched_switch
, NULL
, NULL
, NULL
);
709 /* to clean up the process table */
710 bt_ctf_iter_add_callback(iter
,
711 g_quark_from_static_string("sched_process_free"),
712 NULL
, 0, handle_sched_process_free
, NULL
, NULL
, NULL
);
713 /* to get all the process from the statedumps */
714 bt_ctf_iter_add_callback(iter
,
715 g_quark_from_static_string(
716 "lttng_statedump_process_state"),
717 NULL
, 0, handle_statedump_process_state
,
721 bt_ctf_iter_add_callback(iter
,
722 g_quark_from_static_string("exit_syscall"),
723 NULL
, 0, handle_exit_syscall
, NULL
, NULL
, NULL
);
724 bt_ctf_iter_add_callback(iter
,
725 g_quark_from_static_string("sys_write"),
726 NULL
, 0, handle_sys_write
, NULL
, NULL
, NULL
);
727 bt_ctf_iter_add_callback(iter
,
728 g_quark_from_static_string("sys_read"),
729 NULL
, 0, handle_sys_read
, NULL
, NULL
, NULL
);
730 bt_ctf_iter_add_callback(iter
,
731 g_quark_from_static_string("sys_open"),
732 NULL
, 0, handle_sys_open
, NULL
, NULL
, NULL
);
733 bt_ctf_iter_add_callback(iter
,
734 g_quark_from_static_string("sys_close"),
735 NULL
, 0, handle_sys_close
, NULL
, NULL
, NULL
);
736 bt_ctf_iter_add_callback(iter
,
737 g_quark_from_static_string(
738 "lttng_statedump_file_descriptor"),
739 NULL
, 0, handle_statedump_file_descriptor
,
743 if (lttngtop
.kprobes_table
) {
744 for (i
= 0; i
< lttngtop
.kprobes_table
->len
; i
++) {
745 kprobe
= g_ptr_array_index(lttngtop
.kprobes_table
, i
);
746 bt_ctf_iter_add_callback(iter
,
747 g_quark_from_static_string(
749 NULL
, 0, handle_kprobes
,
755 while ((event
= bt_ctf_iter_read_event(iter
)) != NULL
) {
756 if (quit
|| reload_trace
)
758 ret
= bt_iter_next(bt_ctf_get_iter(iter
));
763 /* block until quit, we reached the end of the trace */
764 sem_wait(&end_trace_sem
);
767 bt_ctf_iter_destroy(iter
);
771 * bt_context_add_traces_recursive: Open a trace recursively
772 * (copied from BSD code in converter/babeltrace.c)
774 * Find each trace present in the subdirectory starting from the given
775 * path, and add them to the context. The packet_seek parameter can be
776 * NULL: this specify to use the default format packet_seek.
778 * Return: 0 on success, nonzero on failure.
779 * Unable to open toplevel: failure.
780 * Unable to open some subdirectory or file: warn and continue;
782 int bt_context_add_traces_recursive(struct bt_context
*ctx
, const char *path
,
783 const char *format_str
,
784 void (*packet_seek
)(struct bt_stream_pos
*pos
,
785 size_t offset
, int whence
))
790 char lpath
[PATH_MAX
];
791 char * const paths
[2] = { lpath
, NULL
};
795 * Need to copy path, because fts_open can change it.
796 * It is the pointer array, not the strings, that are constant.
798 strncpy(lpath
, path
, PATH_MAX
);
799 lpath
[PATH_MAX
- 1] = '\0';
801 tree
= fts_open(paths
, FTS_NOCHDIR
| FTS_LOGICAL
, 0);
803 fprintf(stderr
, "[error] [Context] Cannot traverse \"%s\" for reading.\n",
808 trace_ids
= g_array_new(FALSE
, TRUE
, sizeof(int));
810 while ((node
= fts_read(tree
))) {
813 if (!(node
->fts_info
& FTS_D
))
816 dirfd
= open(node
->fts_accpath
, 0);
818 fprintf(stderr
, "[error] [Context] Unable to open trace "
819 "directory file descriptor.\n");
823 metafd
= openat(dirfd
, "metadata", O_RDONLY
);
842 trace_id
= bt_context_add_trace(ctx
,
843 node
->fts_accpath
, format_str
,
844 packet_seek
, NULL
, NULL
);
846 fprintf(stderr
, "[warning] [Context] opening trace \"%s\" from %s "
847 "for reading.\n", node
->fts_accpath
, path
);
848 /* Allow to skip erroneous traces. */
851 g_array_append_val(trace_ids
, trace_id
);
855 g_array_free(trace_ids
, TRUE
);
862 static int check_field_requirements(const struct bt_ctf_field_decl
*const * field_list
,
863 int field_cnt
, int *tid_check
, int *pid_check
,
864 int *procname_check
, int *ppid_check
)
867 struct perfcounter
*global
;
870 for (j
= 0; j
< field_cnt
; j
++) {
871 name
= bt_ctf_get_decl_field_name(field_list
[j
]);
872 if (*tid_check
== 0) {
873 if (strncmp(name
, "tid", 3) == 0)
876 if (*pid_check
== 0) {
877 if (strncmp(name
, "pid", 3) == 0)
880 if (*ppid_check
== 0) {
881 if (strncmp(name
, "ppid", 4) == 0)
884 if (*procname_check
== 0) {
885 if (strncmp(name
, "procname", 8) == 0)
888 if (strncmp(name
, "perf_", 5) == 0) {
889 global
= g_hash_table_lookup(global_perf_liszt
, (gpointer
) name
);
891 global
= g_new0(struct perfcounter
, 1);
892 /* by default, sort on the first perf context */
893 if (g_hash_table_size(global_perf_liszt
) == 0)
896 g_hash_table_insert(global_perf_liszt
, (gpointer
) strdup(name
), global
);
901 if (*tid_check
== 1 && *pid_check
== 1 && *ppid_check
== 1 &&
902 *procname_check
== 1)
909 * check_requirements: check if the required context informations are available
911 * If each mandatory context information is available for at least in one
912 * event, return 0 otherwise return -1.
914 int check_requirements(struct bt_context
*ctx
)
916 unsigned int i
, evt_cnt
, field_cnt
;
917 struct bt_ctf_event_decl
*const * evt_list
;
918 const struct bt_ctf_field_decl
*const * field_list
;
921 int procname_check
= 0;
925 bt_ctf_get_event_decl_list(0, ctx
, &evt_list
, &evt_cnt
);
926 for (i
= 0; i
< evt_cnt
; i
++) {
927 bt_ctf_get_decl_fields(evt_list
[i
], BT_STREAM_EVENT_CONTEXT
,
928 &field_list
, &field_cnt
);
929 ret
= check_field_requirements(field_list
, field_cnt
,
930 &tid_check
, &pid_check
, &procname_check
,
933 bt_ctf_get_decl_fields(evt_list
[i
], BT_EVENT_CONTEXT
,
934 &field_list
, &field_cnt
);
935 ret
= check_field_requirements(field_list
, field_cnt
,
936 &tid_check
, &pid_check
, &procname_check
,
939 bt_ctf_get_decl_fields(evt_list
[i
], BT_STREAM_PACKET_CONTEXT
,
940 &field_list
, &field_cnt
);
941 ret
= check_field_requirements(field_list
, field_cnt
,
942 &tid_check
, &pid_check
, &procname_check
,
946 if (tid_check
== 0) {
948 fprintf(stderr
, "[error] missing tid context information\n");
950 if (pid_check
== 0) {
952 fprintf(stderr
, "[error] missing pid context information\n");
954 if (ppid_check
== 0) {
956 fprintf(stderr
, "[error] missing ppid context information\n");
958 if (procname_check
== 0) {
960 fprintf(stderr
, "[error] missing procname context information\n");
966 ssize_t
read_subbuffer(struct lttng_consumer_stream
*kconsumerd_fd
,
967 struct lttng_consumer_local_data
*ctx
)
972 int infd
= helper_get_lttng_consumer_stream_wait_fd(kconsumerd_fd
);
974 if (helper_get_lttng_consumer_stream_output(kconsumerd_fd
) == LTTNG_EVENT_SPLICE
) {
975 /* Get the next subbuffer */
976 err
= helper_kernctl_get_next_subbuf(infd
);
979 perror("Reserving sub buffer failed (everything is normal, "
980 "it is due to concurrency)");
983 /* read the whole subbuffer */
984 err
= helper_kernctl_get_padded_subbuf_size(infd
, &len
);
987 perror("Getting sub-buffer len failed.");
991 /* splice the subbuffer to the tracefile */
992 ret
= helper_lttng_consumer_on_read_subbuffer_splice(ctx
, kconsumerd_fd
, len
);
995 * display the error but continue processing to try
996 * to release the subbuffer
998 fprintf(stderr
,"Error splicing to tracefile\n");
1000 err
= helper_kernctl_put_next_subbuf(infd
);
1003 perror("Reserving sub buffer failed (everything is normal, "
1004 "it is due to concurrency)");
1007 sem_post(&metadata_available
);
1014 int on_update_fd(int key
, uint32_t state
)
1016 /* let the lib handle the metadata FD */
1017 if (key
== sessiond_metadata
)
1022 int on_recv_fd(struct lttng_consumer_stream
*kconsumerd_fd
)
1025 struct mmap_stream
*new_mmap_stream
;
1027 /* Opening the tracefile in write mode */
1028 if (helper_get_lttng_consumer_stream_path_name(kconsumerd_fd
) != NULL
) {
1029 ret
= open(helper_get_lttng_consumer_stream_path_name(kconsumerd_fd
),
1030 O_WRONLY
|O_CREAT
|O_TRUNC
, S_IRWXU
|S_IRWXG
|S_IRWXO
);
1035 helper_set_lttng_consumer_stream_out_fd(kconsumerd_fd
, ret
);
1038 if (helper_get_lttng_consumer_stream_output(kconsumerd_fd
) == LTTNG_EVENT_MMAP
) {
1039 new_mmap_stream
= malloc(sizeof(struct mmap_stream
));
1040 new_mmap_stream
->fd
= helper_get_lttng_consumer_stream_wait_fd(
1042 bt_list_add(&new_mmap_stream
->list
, &mmap_list
.head
);
1044 g_ptr_array_add(lttng_consumer_stream_array
, kconsumerd_fd
);
1045 /* keep mmap FDs internally */
1048 consumerd_metadata
= helper_get_lttng_consumer_stream_wait_fd(kconsumerd_fd
);
1049 sessiond_metadata
= helper_get_lttng_consumer_stream_key(kconsumerd_fd
);
1059 void live_consume(struct bt_context
**bt_ctx
)
1064 sem_wait(&metadata_available
);
1065 if (access("/tmp/livesession/kernel/metadata", F_OK
) != 0) {
1066 fprintf(stderr
,"no metadata\n");
1069 metadata_fp
= fopen("/tmp/livesession/kernel/metadata", "r");
1071 *bt_ctx
= bt_context_create();
1072 ret
= bt_context_add_trace(*bt_ctx
, NULL
, "ctf",
1073 lttngtop_ctf_packet_seek
, &mmap_list
, metadata_fp
);
1075 printf("Error adding trace\n");
1083 int setup_consumer(char *command_sock_path
, pthread_t
*threads
,
1084 struct lttng_consumer_local_data
*ctx
)
1088 ctx
= helper_lttng_consumer_create(HELPER_LTTNG_CONSUMER_KERNEL
,
1089 read_subbuffer
, NULL
, on_recv_fd
, on_update_fd
);
1093 unlink(command_sock_path
);
1094 helper_lttng_consumer_set_command_sock_path(ctx
, command_sock_path
);
1095 helper_lttng_consumer_init();
1097 /* Create the thread to manage the receive of fd */
1098 ret
= pthread_create(&threads
[0], NULL
, helper_lttng_consumer_thread_receive_fds
,
1101 perror("pthread_create receive fd");
1104 /* Create thread to manage the polling/writing of traces */
1105 ret
= pthread_create(&threads
[1], NULL
, helper_lttng_consumer_thread_poll_fds
,
1108 perror("pthread_create poll fd");
1116 int enable_kprobes(struct lttng_handle
*handle
, char *channel_name
)
1118 struct lttng_event ev
;
1119 struct kprobes
*kprobe
;
1123 for (i
= 0; i
< lttngtop
.kprobes_table
->len
; i
++) {
1124 kprobe
= g_ptr_array_index(lttngtop
.kprobes_table
, i
);
1126 memset(&ev
, '\0', sizeof(struct lttng_event
));
1127 ev
.type
= LTTNG_EVENT_PROBE
;
1128 if (kprobe
->symbol_name
)
1129 sprintf(ev
.attr
.probe
.symbol_name
, "%s", kprobe
->symbol_name
);
1130 sprintf(ev
.name
, "%s", kprobe
->probe_name
);
1131 ev
.attr
.probe
.addr
= kprobe
->probe_addr
;
1132 ev
.attr
.probe
.offset
= kprobe
->probe_offset
;
1133 if ((ret
= lttng_enable_event(handle
, &ev
, channel_name
)) < 0) {
1134 fprintf(stderr
,"error enabling kprobes : %s\n",
1135 helper_lttcomm_get_readable_code(ret
));
1144 int setup_live_tracing()
1146 struct lttng_domain dom
;
1147 struct lttng_channel chan
;
1148 char *channel_name
= "mmapchan";
1149 struct lttng_event ev
;
1151 char *command_sock_path
= "/tmp/consumerd_sock";
1152 static pthread_t threads
[2]; /* recv_fd, poll */
1153 struct lttng_event_context kctxpid
, kctxcomm
, kctxppid
, kctxtid
;
1155 struct lttng_handle
*handle
;
1157 BT_INIT_LIST_HEAD(&mmap_list
.head
);
1159 lttng_consumer_stream_array
= g_ptr_array_new();
1161 if ((ret
= setup_consumer(command_sock_path
, threads
, ctx
)) < 0) {
1162 fprintf(stderr
,"error setting up consumer\n");
1166 available_snapshots
= g_ptr_array_new();
1168 /* setup the session */
1169 dom
.type
= LTTNG_DOMAIN_KERNEL
;
1171 ret
= unlink("/tmp/livesession");
1173 lttng_destroy_session("test");
1174 if ((ret
= lttng_create_session("test", "/tmp/livesession")) < 0) {
1175 fprintf(stderr
,"error creating the session : %s\n",
1176 helper_lttcomm_get_readable_code(ret
));
1180 if ((handle
= lttng_create_handle("test", &dom
)) == NULL
) {
1181 fprintf(stderr
,"error creating handle\n");
1186 * FIXME : need to let the
1187 * helper_lttng_consumer_thread_receive_fds create the
1189 * Cleaner solution ?
1191 while (access(command_sock_path
, F_OK
)) {
1195 if ((ret
= lttng_register_consumer(handle
, command_sock_path
)) < 0) {
1196 fprintf(stderr
,"error registering consumer : %s\n",
1197 helper_lttcomm_get_readable_code(ret
));
1201 strcpy(chan
.name
, channel_name
);
1202 chan
.attr
.overwrite
= 0;
1203 if (opt_tid
&& opt_textdump
) {
1204 chan
.attr
.subbuf_size
= 32768;
1205 chan
.attr
.num_subbuf
= 8;
1207 //chan.attr.subbuf_size = 1048576; /* 1MB */
1208 chan
.attr
.subbuf_size
= 2097152; /* 1MB */
1209 chan
.attr
.num_subbuf
= 4;
1211 chan
.attr
.switch_timer_interval
= 0;
1212 chan
.attr
.read_timer_interval
= 200;
1213 chan
.attr
.output
= LTTNG_EVENT_MMAP
;
1215 if ((ret
= lttng_enable_channel(handle
, &chan
)) < 0) {
1216 fprintf(stderr
,"error creating channel : %s\n",
1217 helper_lttcomm_get_readable_code(ret
));
1221 memset(&ev
, '\0', sizeof(struct lttng_event
));
1222 //sprintf(ev.name, "sched_switch");
1223 ev
.type
= LTTNG_EVENT_TRACEPOINT
;
1224 if ((ret
= lttng_enable_event(handle
, &ev
, channel_name
)) < 0) {
1225 fprintf(stderr
,"error enabling event : %s\n",
1226 helper_lttcomm_get_readable_code(ret
));
1230 memset(&ev
, '\0', sizeof(struct lttng_event
));
1231 ev
.type
= LTTNG_EVENT_SYSCALL
;
1232 if ((ret
= lttng_enable_event(handle
, &ev
, channel_name
)) < 0) {
1233 fprintf(stderr
,"error enabling syscalls : %s\n",
1234 helper_lttcomm_get_readable_code(ret
));
1238 if (lttngtop
.kprobes_table
) {
1239 ret
= enable_kprobes(handle
, channel_name
);
1245 kctxpid
.ctx
= LTTNG_EVENT_CONTEXT_PID
;
1246 lttng_add_context(handle
, &kctxpid
, NULL
, NULL
);
1247 kctxtid
.ctx
= LTTNG_EVENT_CONTEXT_TID
;
1248 lttng_add_context(handle
, &kctxtid
, NULL
, NULL
);
1249 kctxppid
.ctx
= LTTNG_EVENT_CONTEXT_PPID
;
1250 lttng_add_context(handle
, &kctxppid
, NULL
, NULL
);
1251 kctxcomm
.ctx
= LTTNG_EVENT_CONTEXT_PROCNAME
;
1252 lttng_add_context(handle
, &kctxcomm
, NULL
, NULL
);
1253 kctxpid
.ctx
= LTTNG_EVENT_CONTEXT_VPID
;
1254 lttng_add_context(handle
, &kctxpid
, NULL
, NULL
);
1255 kctxtid
.ctx
= LTTNG_EVENT_CONTEXT_VTID
;
1256 lttng_add_context(handle
, &kctxtid
, NULL
, NULL
);
1257 kctxtid
.ctx
= LTTNG_EVENT_CONTEXT_HOSTNAME
;
1258 lttng_add_context(handle
, &kctxtid
, NULL
, NULL
);
1260 if ((ret
= lttng_start_tracing("test")) < 0) {
1261 fprintf(stderr
,"error starting tracing : %s\n",
1262 helper_lttcomm_get_readable_code(ret
));
1266 helper_kernctl_buffer_flush(consumerd_metadata
);
1268 /* block until metadata is ready */
1269 sem_init(&metadata_available
, 0, 0);
1274 lttng_destroy_session("test");
1279 int main(int argc
, char **argv
)
1282 struct bt_context
*bt_ctx
= NULL
;
1283 struct mmap_stream
*mmap_info
;
1284 unsigned long mmap_len
;
1287 ret
= parse_options(argc
, argv
);
1289 fprintf(stdout
, "Error parsing options.\n\n");
1292 } else if (ret
> 0) {
1296 if (!opt_input_path
) {
1298 signal(SIGTERM
, handle_textdump_sigterm
);
1299 signal(SIGINT
, handle_textdump_sigterm
);
1301 ret
= setup_live_tracing();
1305 if (!opt_textdump
) {
1306 pthread_create(&display_thread
, NULL
, ncurses_display
, (void *) NULL
);
1307 pthread_create(&timer_thread
, NULL
, refresh_thread
, (void *) NULL
);
1311 live_consume(&bt_ctx
);
1313 ret
= bt_context_remove_trace(bt_ctx
, 0);
1315 fprintf(stderr
, "error removing trace\n");
1317 bt_context_put(bt_ctx
);
1321 * since we receive all FDs every time there is an
1322 * update and the FD number is different every time,
1323 * we don't know which one are valid.
1324 * so we check if all FDs are usable with a simple
1327 bt_list_for_each_entry(mmap_info
, &mmap_list
.head
, list
) {
1328 ret
= helper_kernctl_get_mmap_len(mmap_info
->fd
, &mmap_len
);
1330 bt_list_del(&mmap_info
->list
);
1333 sem_post(&metadata_available
);
1336 pthread_join(timer_thread
, NULL
);
1338 pthread_join(display_thread
, NULL
);
1340 lttng_stop_tracing("test");
1341 lttng_destroy_session("test");
1347 bt_ctx
= bt_context_create();
1348 ret
= bt_context_add_traces_recursive(bt_ctx
, opt_input_path
, "ctf", NULL
);
1350 fprintf(stderr
, "[error] Opening the trace\n");
1354 ret
= check_requirements(bt_ctx
);
1356 fprintf(stderr
, "[error] some mandatory contexts were missing, exiting.\n");
1359 pthread_create(&display_thread
, NULL
, ncurses_display
, (void *) NULL
);
1360 pthread_create(&timer_thread
, NULL
, refresh_thread
, (void *) NULL
);
1364 pthread_join(display_thread
, NULL
);
1366 pthread_join(timer_thread
, NULL
);
1371 bt_context_put(bt_ctx
);