X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=lttng%2Fcommands%2Fenable_events.c;h=f6bca6e94823fd28e997614ba55bf67b499b7ba7;hp=f5e24315629e3b02b918eb4000291a4b7f4f54de;hb=775182faeba273d9c28db45f7d17cbe238aadaab;hpb=3000dc785dc25a265485f7b3f7bc3c023d2e67dd diff --git a/lttng/commands/enable_events.c b/lttng/commands/enable_events.c index f5e243156..f6bca6e94 100644 --- a/lttng/commands/enable_events.c +++ b/lttng/commands/enable_events.c @@ -25,6 +25,7 @@ #include #include #include +#include #include "../cmd.h" #include "../conf.h" @@ -40,7 +41,8 @@ static int opt_userspace; static int opt_enable_all; static pid_t opt_pid; static char *opt_probe; -static char *opt_function_symbol; +static char *opt_function; +static char *opt_function_entry_symbol; static char *opt_channel_name; enum { @@ -50,8 +52,11 @@ enum { OPT_MARKER, OPT_PROBE, OPT_FUNCTION, + OPT_FUNCTION_ENTRY, }; +static struct lttng_handle *handle; + static struct poptOption long_options[] = { /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */ {"help", 'h', POPT_ARG_NONE, 0, OPT_HELP, 0, 0}, @@ -66,6 +71,7 @@ static struct poptOption long_options[] = { {"marker", 0, POPT_ARG_NONE, 0, OPT_MARKER, 0, 0}, {"probe", 0, POPT_ARG_STRING, 0, OPT_PROBE, 0, 0}, {"function", 0, POPT_ARG_STRING, 0, OPT_FUNCTION, 0, 0}, + {"function:entry", 0, POPT_ARG_STRING, 0, OPT_FUNCTION_ENTRY, 0, 0}, {0, 0, 0, 0, 0, 0, 0} }; @@ -87,11 +93,16 @@ static void usage(FILE *ofp) fprintf(ofp, "\n"); fprintf(ofp, "Event options:\n"); fprintf(ofp, " --tracepoint Tracepoint event (default)\n"); - fprintf(ofp, " --probe [addr | symbol+offset]\n"); + fprintf(ofp, " --probe [addr | symbol | symbol+offset]\n"); fprintf(ofp, " Dynamic probe.\n"); fprintf(ofp, " Addr and offset can be octal (0NNN...),\n"); fprintf(ofp, " decimal (NNN...) or hexadecimal (0xNNN...)\n"); - fprintf(ofp, " --function SYMBOL Function tracer event\n"); + fprintf(ofp, " --function [addr | symbol | symbol+offset]\n"); + fprintf(ofp, " Dynamic function entry/return probe.\n"); + fprintf(ofp, " Addr and offset can be octal (0NNN...),\n"); + fprintf(ofp, " decimal (NNN...) or hexadecimal (0xNNN...)\n"); + fprintf(ofp, " --function:entry symbol\n"); + fprintf(ofp, " Function tracer event\n"); fprintf(ofp, " --marker User-space marker (deprecated)\n"); fprintf(ofp, "\n"); } @@ -109,23 +120,38 @@ static int parse_probe_opts(struct lttng_event *ev, char *opt) if (opt == NULL) { ret = -1; - goto error; + goto end; } /* Check for symbol+offset */ ret = sscanf(opt, "%[^'+']+%s", name, s_hex); if (ret == 2) { strncpy(ev->attr.probe.symbol_name, name, LTTNG_SYMBOL_NAME_LEN); + ev->attr.probe.symbol_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; DBG("probe symbol %s", ev->attr.probe.symbol_name); if (strlen(s_hex) == 0) { ERR("Invalid probe offset %s", s_hex); ret = -1; - goto error; + goto end; } ev->attr.probe.offset = strtoul(s_hex, NULL, 0); DBG("probe offset %" PRIu64, ev->attr.probe.offset); ev->attr.probe.addr = 0; - goto error; + goto end; + } + + /* Check for symbol */ + if (isalpha(name[0])) { + ret = sscanf(opt, "%s", name); + if (ret == 1) { + strncpy(ev->attr.probe.symbol_name, name, LTTNG_SYMBOL_NAME_LEN); + ev->attr.probe.symbol_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; + DBG("probe symbol %s", ev->attr.probe.symbol_name); + ev->attr.probe.offset = 0; + DBG("probe offset %" PRIu64, ev->attr.probe.offset); + ev->attr.probe.addr = 0; + goto end; + } } /* Check for address */ @@ -134,19 +160,19 @@ static int parse_probe_opts(struct lttng_event *ev, char *opt) if (strlen(s_hex) == 0) { ERR("Invalid probe address %s", s_hex); ret = -1; - goto error; + goto end; } ev->attr.probe.addr = strtoul(s_hex, NULL, 0); DBG("probe addr %" PRIu64, ev->attr.probe.addr); ev->attr.probe.offset = 0; memset(ev->attr.probe.symbol_name, 0, LTTNG_SYMBOL_NAME_LEN); - goto error; + goto end; } /* No match */ ret = -1; -error: +end: return ret; } @@ -155,18 +181,13 @@ error: * * Enabling event using the lttng API. */ -static int enable_events(void) +static int enable_events(char *session_name) { int err, ret = CMD_SUCCESS; char *event_name, *channel_name = NULL; struct lttng_event ev; struct lttng_domain dom; - if (set_session_name(opt_session_name) < 0) { - ret = CMD_ERROR; - goto error; - } - if (opt_channel_name == NULL) { err = asprintf(&channel_name, DEFAULT_CHANNEL_NAME); if (err < 0) { @@ -182,9 +203,15 @@ static int enable_events(void) dom.type = LTTNG_DOMAIN_KERNEL; } + handle = lttng_create_handle(session_name, &dom); + if (handle == NULL) { + ret = -1; + goto error; + } + if (opt_enable_all) { if (opt_kernel) { - ret = lttng_enable_event(&dom, NULL, channel_name); + ret = lttng_enable_event(handle, NULL, channel_name); if (ret == 0) { MSG("All kernel events are enabled in channel %s", channel_name); } @@ -203,6 +230,7 @@ static int enable_events(void) event_name, channel_name); /* Copy name and type of the event */ strncpy(ev.name, event_name, LTTNG_SYMBOL_NAME_LEN); + ev.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; ev.type = opt_event_type; switch (opt_event_type) { @@ -217,14 +245,25 @@ static int enable_events(void) } break; case LTTNG_EVENT_FUNCTION: - strncpy(ev.attr.ftrace.symbol_name, opt_function_symbol, LTTNG_SYMBOL_NAME_LEN); + ret = parse_probe_opts(&ev, opt_function); + if (ret < 0) { + ERR("Unable to parse function probe options"); + ret = 0; + goto error; + } + break; + case LTTNG_EVENT_FUNCTION_ENTRY: + strncpy(ev.attr.ftrace.symbol_name, + opt_function_entry_symbol, + LTTNG_SYMBOL_NAME_LEN); + ev.attr.ftrace.symbol_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; break; default: ret = CMD_NOT_IMPLEMENTED; goto error; } - ret = lttng_enable_event(&dom, &ev, channel_name); + ret = lttng_enable_event(handle, &ev, channel_name); if (ret == 0) { MSG("Kernel event %s created in channel %s", event_name, channel_name); } @@ -238,7 +277,7 @@ static int enable_events(void) ret = CMD_NOT_IMPLEMENTED; goto error; } else { - ERR("Please specify a tracer (kernel or user-space)"); + ERR("Please specify a tracer (--kernel or --userspace)"); goto error; } @@ -250,6 +289,8 @@ error: if (opt_channel_name == NULL) { free(channel_name); } + lttng_destroy_handle(handle); + return ret; } @@ -262,6 +303,7 @@ int cmd_enable_events(int argc, const char **argv) { int opt, ret; static poptContext pc; + char *session_name = NULL; pc = poptGetContext(NULL, argc, argv, long_options, 0); poptReadDefaultConfig(pc, 0); @@ -291,7 +333,11 @@ int cmd_enable_events(int argc, const char **argv) break; case OPT_FUNCTION: opt_event_type = LTTNG_EVENT_FUNCTION; - opt_function_symbol = poptGetOptArg(pc); + opt_function = poptGetOptArg(pc); + break; + case OPT_FUNCTION_ENTRY: + opt_event_type = LTTNG_EVENT_FUNCTION_ENTRY; + opt_function_entry_symbol = poptGetOptArg(pc); break; default: usage(stderr); @@ -308,8 +354,22 @@ int cmd_enable_events(int argc, const char **argv) goto end; } - ret = enable_events(); + if (!opt_session_name) { + session_name = get_session_name(); + if (session_name == NULL) { + ret = -1; + goto end; + } + } else { + session_name = opt_session_name; + } + + ret = enable_events(session_name); end: + if (opt_session_name == NULL) { + free(session_name); + } + return ret; }