Fix: new syscall names in textdump
[lttngtop.git] / src / lttngtop.c
index 46493482de9890238685ed5bbc54ee2639f93f0f..c3596aa23f697a5c487ddf0004b03e578b279f96 100644 (file)
@@ -40,6 +40,7 @@
 #include <sys/mman.h>
 #include <sys/wait.h>
 #include <lttng/lttng.h>
+
 #ifdef LTTNGTOP_MMAP_LIVE
 #include <lttng/lttngtop-helper.h>
 #include <babeltrace/lttngtopmmappacketseek.h>
 #include "lttngtoptypes.h"
 #include "cputop.h"
 #include "iostreamtop.h"
-#include "cursesdisplay.h"
 #include "common.h"
 #include "network-live.h"
+#include "lttng-session.h"
+
+#ifdef HAVE_LIBNCURSES
+#include "cursesdisplay.h"
+#endif
 
 #define NET_URL_PREFIX "net://"
 #define NET4_URL_PREFIX        "net4://"
@@ -66,6 +71,8 @@ int opt_begin;
 int opt_all;
 
 int quit = 0;
+/* We need at least one valid trace to start processing. */
+int valid_trace = 0;
 
 struct lttngtop *copy;
 pthread_t display_thread;
@@ -94,11 +101,16 @@ enum {
        OPT_TEXTDUMP,
        OPT_PID,
        OPT_CHILD,
-       OPT_HOSTNAME,
+       OPT_PROCNAME,
        OPT_RELAY_HOSTNAME,
        OPT_KPROBES,
        OPT_BEGIN,
        OPT_ALL,
+       OPT_OUTPUT_FILE,
+       OPT_VERBOSE,
+       OPT_GUI_TEST,
+       OPT_CREATE_LOCAL_SESSION,
+       OPT_CREATE_LIVE_SESSION,
 };
 
 static struct poptOption long_options[] = {
@@ -108,11 +120,16 @@ static struct poptOption long_options[] = {
        { "child", 'f', POPT_ARG_NONE, NULL, OPT_CHILD, NULL, NULL },
        { "begin", 'b', POPT_ARG_NONE, NULL, OPT_BEGIN, NULL, NULL },
        { "pid", 'p', POPT_ARG_STRING, &opt_tid, OPT_PID, NULL, NULL },
-       { "hostname", 'n', POPT_ARG_STRING, &opt_hostname, OPT_HOSTNAME, NULL, NULL },
+       { "procname", 'n', POPT_ARG_STRING, &opt_procname, OPT_PROCNAME, NULL, NULL },
        { "relay-hostname", 'r', POPT_ARG_STRING, &opt_relay_hostname,
                OPT_RELAY_HOSTNAME, NULL, NULL },
        { "kprobes", 'k', POPT_ARG_STRING, &opt_kprobes, OPT_KPROBES, NULL, NULL },
        { "all", 'a', POPT_ARG_NONE, NULL, OPT_ALL, NULL, NULL },
+       { "output", 'o', POPT_ARG_STRING, &opt_output, OPT_OUTPUT_FILE, NULL, NULL },
+       { "verbose", 'v', POPT_ARG_NONE, NULL, OPT_VERBOSE, NULL, NULL },
+       { "gui-test", 'g', POPT_ARG_NONE, NULL, OPT_GUI_TEST, NULL, NULL },
+       { "create-local-session", 0, POPT_ARG_NONE, NULL, OPT_CREATE_LOCAL_SESSION, NULL, NULL },
+       { "create-live-session", 0, POPT_ARG_NONE, NULL, OPT_CREATE_LIVE_SESSION, NULL, NULL },
        { NULL, 0, 0, NULL, 0, NULL, NULL },
 };
 
@@ -147,6 +164,7 @@ void *refresh_thread(void *p)
        }
 }
 
+#ifdef HAVE_LIBNCURSES
 void *ncurses_display(void *p)
 {
        unsigned int current_display_index = 0;
@@ -178,6 +196,7 @@ void *ncurses_display(void *p)
                sem_post(&pause_sem);
        }
 }
+#endif /* HAVE_LIBNCURSES */
 
 void print_fields(struct bt_ctf_event *event, const char *procname,
                int pid)
@@ -197,8 +216,8 @@ void print_fields(struct bt_ctf_event *event, const char *procname,
        bt_ctf_get_field_list(event, scope, &list, &cnt);
        for (i = 0; i < cnt; i++) {
                if (i != 0)
-                       printf(", ");
-               printf("%s = ", bt_ctf_field_name(list[i]));
+                       fprintf(output, ", ");
+               fprintf(output, "%s = ", bt_ctf_field_name(list[i]));
                l = bt_ctf_get_decl_from_def(list[i]);
                if (strncmp(bt_ctf_field_name(list[i]), "fd", 2) == 0)
                        fd = 1;
@@ -208,17 +227,17 @@ void print_fields(struct bt_ctf_event *event, const char *procname,
                if (type == CTF_TYPE_INTEGER) {
                        if (bt_ctf_get_int_signedness(l) == 0) {
                                fd_value = bt_ctf_get_uint64(list[i]);
-                               printf("%" PRIu64, bt_ctf_get_uint64(list[i]));
+                               fprintf(output, "%" PRIu64, bt_ctf_get_uint64(list[i]));
                        } else {
                                fd_value = bt_ctf_get_int64(list[i]);
-                               printf("%" PRId64, bt_ctf_get_int64(list[i]));
+                               fprintf(output, "%" PRId64, bt_ctf_get_int64(list[i]));
                        }
                } else if (type == CTF_TYPE_STRING) {
-                       printf("%s", bt_ctf_get_string(list[i]));
+                       fprintf(output, "%s", bt_ctf_get_string(list[i]));
                } else if (type == CTF_TYPE_ARRAY) {
                        str = bt_ctf_get_char_array(list[i]);
                        if (!bt_ctf_field_get_error() && str)
-                               printf("%s", str);
+                               fprintf(output, "%s", str);
                }
                if (fd) {
                        current_proc = find_process_tid(&lttngtop, pid, procname);
@@ -227,7 +246,7 @@ void print_fields(struct bt_ctf_event *event, const char *procname,
                        current_file = get_file(current_proc, fd_value);
                        if (!current_file || !current_file->name)
                                continue;
-                       printf("<%s>", current_file->name);
+                       fprintf(output, "<%s>", current_file->name);
                }
        }
 }
@@ -247,6 +266,7 @@ enum bt_cb_ret textdump(struct bt_ctf_event *call_data, void *private_data)
        const char *hostname, *procname;
        struct cputime *cpu;
        char *from_syscall = NULL;
+       int syscall_exit = 0;
 
        timestamp = bt_ctf_get_timestamp(call_data);
 
@@ -271,7 +291,7 @@ enum bt_cb_ret textdump(struct bt_ctf_event *call_data, void *private_data)
                lookup = pid;
        else
                lookup = tid;
-       if (opt_tid || opt_hostname || opt_exec_name) {
+       if (opt_tid || opt_procname || opt_exec_name) {
                if (!lookup_filter_tid_list(lookup)) {
                        /* To display when a process of ours in getting scheduled in */
                        if (strcmp(bt_ctf_event_name(call_data), "sched_switch") == 0) {
@@ -290,26 +310,33 @@ enum bt_cb_ret textdump(struct bt_ctf_event *call_data, void *private_data)
                                                goto end;
                                } else {
                                        if (opt_all)
-                                               printf("%c[1m", 27);
+                                               fprintf(output, "%c[1m", 27);
                                }
                        } else if (!opt_all) {
                                goto end;
                        }
                } else {
                        if (opt_all)
-                               printf("%c[1m", 27);
+                               fprintf(output, "%c[1m", 27);
                }
        }
 
-       if (last_syscall && (strncmp(bt_ctf_event_name(call_data),
-                                "exit_syscall", 12)) != 0) {
+       if (((strncmp(bt_ctf_event_name(call_data),
+                                               "exit_syscall", 12)) == 0) ||
+                       ((strncmp(bt_ctf_event_name(call_data),
+                                "syscall_exit", 12)) == 0)) {
+               syscall_exit = 1;
+       }
+
+       if (last_syscall && !syscall_exit) {
                last_syscall = NULL;
-               printf(" ...interrupted...\n");
+               fprintf(output, " ...interrupted...\n");
        }
 
        cpu_id = get_cpu_id(call_data);
        procname = get_context_comm(call_data);
-       if (strncmp(bt_ctf_event_name(call_data), "sys_", 4) == 0) {
+       if ((strncmp(bt_ctf_event_name(call_data), "sys_", 4) == 0) ||
+                       (strncmp(bt_ctf_event_name(call_data), "syscall_entry", 13) == 0)){
                cpu = get_cpu(cpu_id);
                cpu->current_syscall = g_new0(struct syscall, 1);
                cpu->current_syscall->name = strdup(bt_ctf_event_name(call_data));
@@ -317,7 +344,7 @@ enum bt_cb_ret textdump(struct bt_ctf_event *call_data, void *private_data)
                cpu->current_syscall->cpu_id = cpu_id;
                last_syscall = cpu->current_syscall;
                current_syscall = 1;
-       } else if ((strncmp(bt_ctf_event_name(call_data), "exit_syscall", 12)) == 0) {
+       } else if (syscall_exit) {
                struct tm start_ts;
 
                /* Return code of a syscall if it was the last displayed event. */
@@ -331,14 +358,14 @@ enum bt_cb_ret textdump(struct bt_ctf_event *call_data, void *private_data)
                                syscall_ret = bt_ctf_get_int64(bt_ctf_get_field(call_data,
                                                        scope, "_ret"));
 
-                               printf("= %" PRId64 " (+%" PRIu64 ".%09" PRIu64 ")\n",
+                               fprintf(output, "= %" PRId64 " (%" PRIu64 ".%09" PRIu64 "s)\n",
                                                syscall_ret, delta / NSEC_PER_SEC,
                                                delta % NSEC_PER_SEC);
                                last_syscall = NULL;
                                goto end;
                        } else {
                                last_syscall = NULL;
-                               printf(" ...interrupted...\n");
+                               fprintf(output, " ...interrupted...\n");
                        }
                }
 
@@ -367,7 +394,7 @@ enum bt_cb_ret textdump(struct bt_ctf_event *call_data, void *private_data)
        delta = timestamp - prev_ts;
        prev_ts = timestamp;
 
-       printf("%02d:%02d:%02d.%09" PRIu64 " (+%" PRIu64 ".%09" PRIu64 ") %s%s"
+       fprintf(output, "%02d:%02d:%02d.%09" PRIu64 " (+%" PRIu64 ".%09" PRIu64 ") %s%s"
                        "(cpu %d) [%s (%d/%d)] %s (",
                        start.tm_hour, start.tm_min, start.tm_sec,
                        ts_nsec_start, delta / NSEC_PER_SEC,
@@ -375,12 +402,12 @@ enum bt_cb_ret textdump(struct bt_ctf_event *call_data, void *private_data)
                        (hostname) ? " ": "", cpu_id, procname, pid, tid,
                        bt_ctf_event_name(call_data));
        print_fields(call_data, procname, pid);
-       printf(")%s%c", (from_syscall) ? from_syscall : "",
+       fprintf(output, ")%s%c", (from_syscall) ? from_syscall : "",
                        (!current_syscall) ? '\n' : ' ');
 
        free(from_syscall);
-       if (opt_all && (opt_tid || opt_hostname || opt_exec_name))
-               printf("%c[0m", 27);
+       if (opt_all && (opt_tid || opt_procname || opt_exec_name))
+               fprintf(output, "%c[0m", 27);
 
 end:
        return BT_CB_OK;
@@ -569,7 +596,7 @@ enum bt_cb_ret fix_process_table(struct bt_ctf_event *call_data,
        }
        ppid = get_context_ppid(call_data);
        if (ppid == -1ULL) {
-               goto error;
+               goto end;
        }
        vpid = get_context_vpid(call_data);
        if (pid == -1ULL) {
@@ -598,6 +625,17 @@ enum bt_cb_ret fix_process_table(struct bt_ctf_event *call_data,
                goto end;
        update_proc(child, pid, tid, ppid, vpid, vtid, vppid, comm, hostname);
 
+       if (opt_procname && lookup_procname(comm) &&
+                       !lookup_filter_tid_list(tid)) {
+               int *tmp_tid;
+
+               tmp_tid = malloc(sizeof(int));
+               *tmp_tid = tid;
+               printf("ADDING %s %d\n", comm, tid);
+               g_hash_table_insert(global_filter_list,
+                               (gpointer) tmp_tid, tmp_tid);
+       }
+
        if (pid != tid) {
                /* find or create the parent */
                parent = find_process_tid(&lttngtop, pid, comm);
@@ -627,6 +665,7 @@ void init_lttngtop()
        global_perf_liszt = g_hash_table_new(g_str_hash, g_str_equal);
        global_filter_list = g_hash_table_new(g_str_hash, g_str_equal);
        global_host_list = g_hash_table_new(g_str_hash, g_str_equal);
+       global_procname_list = g_hash_table_new(g_str_hash, g_str_equal);
        tid_filter_list = g_hash_table_new(g_str_hash,
                        g_str_equal);
 
@@ -660,11 +699,15 @@ void usage(FILE *fp)
        fprintf(fp, "  -t, --textdump           Display live events in text-only\n");
        fprintf(fp, "  -p, --pid                Comma-separated list of PIDs to display\n");
        fprintf(fp, "  -f, --child              Follow threads associated with selected PIDs\n");
-       fprintf(fp, "  -n, --hostname           Comma-separated list of hostnames to display (require hostname context in trace)\n");
+       fprintf(fp, "  -n, --procname           Comma-separated list of procnames to display (require procname context in trace)\n");
        fprintf(fp, "  -a, --all                In textdump mode, display all events but write in bold the processes we are interested in (-f, -p and -n)\n");
        fprintf(fp, "  -k, --kprobes            Comma-separated list of kprobes to insert (same format as lttng enable-event)\n");
        fprintf(fp, "  -r, --relay-hostname     Network live streaming : hostname of the lttng-relayd (default port)\n");
        fprintf(fp, "  -b, --begin              Network live streaming : read the trace for the beginning of the recording\n");
+       fprintf(fp, "  -o, --output <filename>  In textdump, output the log in <filename>\n");
+       fprintf(fp, "  -g, --gui-test           Test if the ncurses support is compiled in (return 0 if it is)\n");
+       fprintf(fp, "  --create-local-session   Setup a LTTng local session with all the right parameters\n");
+       fprintf(fp, "  --create-live-session    Setup a LTTng live session on localhost with all the right parameters\n");
 }
 
 /*
@@ -765,6 +808,19 @@ static int parse_options(int argc, char **argv)
                                usage(stdout);
                                ret = 1;    /* exit cleanly */
                                goto end;
+                       case OPT_GUI_TEST:
+#ifdef HAVE_LIBNCURSES
+                               exit(EXIT_SUCCESS);
+#else
+                               exit(EXIT_FAILURE);
+#endif
+                               goto end;
+                       case OPT_CREATE_LOCAL_SESSION:
+                               ret = create_local_session();
+                               exit(ret);
+                       case OPT_CREATE_LIVE_SESSION:
+                               ret = create_live_local_session(NULL, NULL, 1);
+                               exit(ret);
                        case OPT_TEXTDUMP:
                                opt_textdump = 1;
                                break;
@@ -789,11 +845,11 @@ static int parse_options(int argc, char **argv)
                                /* start reading the live trace from the beginning */
                                opt_begin = 1;
                                break;
-                       case OPT_HOSTNAME:
+                       case OPT_PROCNAME:
                                toggle_filter = 1;
-                               tmp_str = strtok(opt_hostname, ",");
+                               tmp_str = strtok(opt_procname, ",");
                                while (tmp_str) {
-                                       add_hostname_list(tmp_str, 1);
+                                       add_procname_list(tmp_str, 1);
                                        tmp_str = strtok(NULL, ",");
                                }
                                break;
@@ -818,6 +874,11 @@ static int parse_options(int argc, char **argv)
                                        tmp_str = strtok(NULL, ",");
                                }
                                break;
+                       case OPT_OUTPUT_FILE:
+                               break;
+                       case OPT_VERBOSE:
+                               babeltrace_verbose = 1;
+                               break;
                        default:
                                ret = -EINVAL;
                                goto end;
@@ -836,6 +897,15 @@ static int parse_options(int argc, char **argv)
        if (!opt_exec_name) {
                opt_input_path = poptGetArg(pc);
        }
+       if (!opt_output) {
+               opt_output = strdup("/dev/stdout");
+       }
+       output = fopen(opt_output, "w");
+       if (!output) {
+               perror("Error opening output file");
+               ret = -1;
+               goto end;
+       }
 
 end:
        if (pc) {
@@ -882,12 +952,36 @@ void iter_trace(struct bt_context *bt_ctx)
        bt_ctf_iter_add_callback(iter,
                        g_quark_from_static_string("sys_open"),
                        NULL, 0, handle_sys_open, NULL, NULL, NULL);
+       bt_ctf_iter_add_callback(iter,
+                       g_quark_from_static_string("syscall_entry_open"),
+                       NULL, 0, handle_sys_open, NULL, NULL, NULL);
+
+       bt_ctf_iter_add_callback(iter,
+                       g_quark_from_static_string("sys_socket"),
+                       NULL, 0, handle_sys_socket, NULL, NULL, NULL);
+       bt_ctf_iter_add_callback(iter,
+                       g_quark_from_static_string("syscall_entry_socket"),
+                       NULL, 0, handle_sys_socket, NULL, NULL, NULL);
+
        bt_ctf_iter_add_callback(iter,
                        g_quark_from_static_string("sys_close"),
                        NULL, 0, handle_sys_close, NULL, NULL, NULL);
+       bt_ctf_iter_add_callback(iter,
+                       g_quark_from_static_string("syscall_entry_close"),
+                       NULL, 0, handle_sys_close, NULL, NULL, NULL);
+
        bt_ctf_iter_add_callback(iter,
                        g_quark_from_static_string("exit_syscall"),
                        NULL, 0, handle_exit_syscall, NULL, NULL, NULL);
+       bt_ctf_iter_add_callback(iter,
+                       g_quark_from_static_string("syscall_exit_open"),
+                       NULL, 0, handle_exit_syscall, NULL, NULL, NULL);
+       bt_ctf_iter_add_callback(iter,
+                       g_quark_from_static_string("syscall_exit_socket"),
+                       NULL, 0, handle_exit_syscall, NULL, NULL, NULL);
+       bt_ctf_iter_add_callback(iter,
+                       g_quark_from_static_string("syscall_exit_close"),
+                       NULL, 0, handle_exit_syscall, NULL, NULL, NULL);
        if (opt_textdump) {
                bt_ctf_iter_add_callback(iter, 0, NULL, 0,
                                textdump,
@@ -905,9 +999,22 @@ void iter_trace(struct bt_context *bt_ctx)
                bt_ctf_iter_add_callback(iter,
                                g_quark_from_static_string("sys_write"),
                                NULL, 0, handle_sys_write, NULL, NULL, NULL);
+               bt_ctf_iter_add_callback(iter,
+                               g_quark_from_static_string("syscall_entry_write"),
+                               NULL, 0, handle_sys_write, NULL, NULL, NULL);
+               bt_ctf_iter_add_callback(iter,
+                               g_quark_from_static_string("syscall_exit_write"),
+                               NULL, 0, handle_exit_syscall, NULL, NULL, NULL);
+
                bt_ctf_iter_add_callback(iter,
                                g_quark_from_static_string("sys_read"),
                                NULL, 0, handle_sys_read, NULL, NULL, NULL);
+               bt_ctf_iter_add_callback(iter,
+                               g_quark_from_static_string("syscall_entry_read"),
+                               NULL, 0, handle_sys_read, NULL, NULL, NULL);
+               bt_ctf_iter_add_callback(iter,
+                               g_quark_from_static_string("syscall_exit_read"),
+                               NULL, 0, handle_exit_syscall, NULL, NULL, NULL);
 
                /* for kprobes */
                if (lttngtop.kprobes_table) {
@@ -1163,6 +1270,9 @@ int check_requirements(struct bt_context *ctx)
                ret = -1;
                fprintf(stderr, "[error] missing procname context information\n");
        }
+       if (ret == 0) {
+               valid_trace = 1;
+       }
 
 end:
        return ret;
@@ -1179,8 +1289,8 @@ int main(int argc, char **argv, char **envp)
 {
        int ret;
        struct bt_context *bt_ctx = NULL;
+       char *live_session_name = NULL;
 
-       //babeltrace_verbose = 1;
        init_lttngtop();
        ret = parse_options(argc, argv);
        if (ret < 0) {
@@ -1198,28 +1308,13 @@ int main(int argc, char **argv, char **envp)
 
        if (!opt_input_path && !remote_live && !opt_exec_name) {
                /* mmap live */
-#ifdef LTTNGTOP_MMAP_LIVE
-               if (opt_textdump) {
-                       signal(SIGTERM, handle_textdump_sigterm);
-                       signal(SIGINT, handle_textdump_sigterm);
-               }
-               mmap_live_loop(bt_ctx);
-               pthread_join(timer_thread, NULL);
-               quit = 1;
-               pthread_join(display_thread, NULL);
-
-               lttng_stop_tracing("test");
-               lttng_destroy_session("test");
-
-               goto end;
-#else
-               fprintf(stderr, "[ERROR] Mmap live support not compiled, specify a "
-                               "trace directory or -r <relayd hostname/IP>\n");
-               usage(stdout);
-               ret = -1;
-               goto end;
-#endif /* LTTNGTOP_MMAP_LIVE */
-       } else if (!opt_input_path && remote_live) {
+               ret = create_live_local_session(&opt_relay_hostname,
+                               &live_session_name, 0);
+               if (ret < 0)
+                       goto end;
+               remote_live = 1;
+       }
+       if (!opt_input_path && remote_live) {
                /* network live */
                bt_ctx = bt_context_create();
                ret = bt_context_add_traces_recursive(bt_ctx, opt_relay_hostname,
@@ -1229,8 +1324,6 @@ int main(int argc, char **argv, char **envp)
                        goto end;
                }
        } else {
-               //init_lttngtop();
-
                bt_ctx = bt_context_create();
                ret = bt_context_add_traces_recursive(bt_ctx, opt_input_path, "ctf", NULL);
                if (ret < 0) {
@@ -1239,17 +1332,23 @@ int main(int argc, char **argv, char **envp)
                }
 
                ret = check_requirements(bt_ctx);
-               if (ret < 0) {
+               if (ret < 0 && !valid_trace) {
                        fprintf(stderr, "[error] some mandatory contexts "
                                        "were missing, exiting.\n");
-                       goto end;
+                       //goto end;
                }
 
                if (!opt_textdump) {
+#ifdef HAVE_LIBNCURSES
                        pthread_create(&display_thread, NULL, ncurses_display,
                                        (void *) NULL);
                        pthread_create(&timer_thread, NULL, refresh_thread,
                                        (void *) NULL);
+#else
+                       printf("Ncurses support not compiled, please install "
+                                       "the missing dependencies and recompile\n");
+                       goto end;
+#endif
                }
 
                iter_trace(bt_ctx);
@@ -1266,5 +1365,12 @@ end:
        if (bt_ctx)
                bt_context_put(bt_ctx);
 
+       if (live_session_name) {
+               ret = destroy_live_local_session(live_session_name);
+               if (ret < 0) {
+                       fprintf(stderr, "Error destroying %s\n", live_session_name);
+               }
+       }
+
        return ret;
 }
This page took 0.033313 seconds and 4 git commands to generate.