X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=lttng-sessiond%2Fmain.c;h=7e58ff781ee86d481458657ddba725f30c102d49;hb=ebaeda9495ef97ec81d9f7dd66eccdb32f116cee;hp=8850fc0f9288463307f14c5bcd79f20c28ab09ff;hpb=5edd7e09e892604b00ca937c0f1bc887bb5a842c;p=lttng-tools.git diff --git a/lttng-sessiond/main.c b/lttng-sessiond/main.c index 8850fc0f9..7e58ff781 100644 --- a/lttng-sessiond/main.c +++ b/lttng-sessiond/main.c @@ -57,6 +57,8 @@ #include "ust-ctl.h" #include "utils.h" +#define CONSUMERD_FILE "lttng-consumerd" + struct consumer_data { enum lttng_consumer_type type; @@ -169,31 +171,38 @@ static struct ltt_session_list *session_list_ptr; int ust_consumerd64_fd = -1; int ust_consumerd32_fd = -1; -static const char *consumerd64_prog = "lttng-consumerd"; -static const char *consumerd32_prog = "lttng-consumerd"; - -static const char *consumerd64_bindir = - __stringify(CONFIG_64BIT_BINDIR); -static const char *consumerd32_bindir = - __stringify(CONFIG_32BIT_BINDIR); +static const char *consumerd32_path = + __stringify(CONFIG_CONSUMERD32_PATH); +static const char *consumerd64_path = + __stringify(CONFIG_CONSUMERD64_PATH); +static const char *consumerd32_libdir = + __stringify(CONFIG_CONSUMERD32_LIBDIR); +static const char *consumerd64_libdir = + __stringify(CONFIG_CONSUMERD64_LIBDIR); static void setup_consumerd_path(void) { - const char *bindir; + const char *path, *libdir; /* * Allow INSTALL_BIN_PATH to be used as a target path for the - * native architecture size consumer if CONFIG_NBIT_BINDIR as - * not been defined. + * native architecture size consumer if CONFIG_CONSUMER*_PATH + * has not been defined. */ -#if (CAA_BITS_PER_LONG == 64) - if (!consumerd64_bindir[0]) { - consumerd64_bindir = INSTALL_BIN_PATH; +#if (CAA_BITS_PER_LONG == 32) + if (!consumerd32_path[0]) { + consumerd32_bindir = INSTALL_BIN_PATH "/" CONSUMERD_FILE; + } + if (!consumerd32_libdir[0]) { + consumerd32_libdir = INSTALL_LIB_PATH; } -#elif (CAA_BITS_PER_LONG == 32) - if (!consumerd32_bindir[0]) { - consumerd32_bindir = INSTALL_BIN_PATH; +#elif (CAA_BITS_PER_LONG == 64) + if (!consumerd64_path[0]) { + consumerd64_path = INSTALL_BIN_PATH "/" CONSUMERD_FILE; + } + if (!consumerd64_libdir[0]) { + consumerd64_libdir = INSTALL_LIB_PATH; } #else #error "Unknown bitness" @@ -202,13 +211,21 @@ void setup_consumerd_path(void) /* * runtime env. var. overrides the build default. */ - bindir = getenv("LTTNG_TOOLS_64BIT_BINDIR"); - if (bindir) { - consumerd64_bindir = bindir; + path = getenv("LTTNG_CONSUMERD32_PATH"); + if (path) { + consumerd32_path = path; + } + path = getenv("LTTNG_CONSUMERD64_PATH"); + if (path) { + consumerd64_path = path; } - bindir = getenv("LTTNG_TOOLS_32BIT_BINDIR"); - if (bindir) { - consumerd32_bindir = bindir; + libdir = getenv("LTTNG_TOOLS_CONSUMERD32_LIBDIR"); + if (libdir) { + consumerd32_libdir = libdir; + } + libdir = getenv("LTTNG_TOOLS_CONSUMERD64_LIBDIR"); + if (libdir) { + consumerd64_libdir = libdir; } } @@ -1515,11 +1532,7 @@ static pid_t spawn_consumerd(struct consumer_data *consumer_data) break; case LTTNG_CONSUMER64_UST: { - char path[PATH_MAX]; - - snprintf(path, PATH_MAX, "%s/%s", - consumerd64_bindir, consumerd64_prog); - execl(path, verbosity, "-u", + execl(consumerd64_path, verbosity, "-u", "--consumerd-cmd-sock", consumer_data->cmd_unix_sock_path, "--consumerd-err-sock", consumer_data->err_unix_sock_path, NULL); @@ -1527,11 +1540,7 @@ static pid_t spawn_consumerd(struct consumer_data *consumer_data) } case LTTNG_CONSUMER32_UST: { - char path[PATH_MAX]; - - snprintf(path, PATH_MAX, "%s/%s", - consumerd32_bindir, consumerd32_prog); - execl(path, verbosity, "-u", + execl(consumerd32_path, verbosity, "-u", "--consumerd-cmd-sock", consumer_data->cmd_unix_sock_path, "--consumerd-err-sock", consumer_data->err_unix_sock_path, NULL); @@ -2276,15 +2285,17 @@ static int cmd_disable_event(struct ltt_session *session, int domain, case LTTNG_DOMAIN_KERNEL: { struct ltt_kernel_channel *kchan; + struct ltt_kernel_session *ksess; - kchan = trace_kernel_get_channel_by_name(channel_name, - session->kernel_session); + ksess = session->kernel_session; + + kchan = trace_kernel_get_channel_by_name(channel_name, ksess); if (kchan == NULL) { ret = LTTCOMM_KERN_CHAN_NOT_FOUND; goto error; } - ret = event_kernel_disable_tracepoint(session->kernel_session, kchan, event_name); + ret = event_kernel_disable_tracepoint(ksess, kchan, event_name); if (ret != LTTCOMM_OK) { goto error; } @@ -2293,11 +2304,34 @@ static int cmd_disable_event(struct ltt_session *session, int domain, break; } case LTTNG_DOMAIN_UST: + { + struct ltt_ust_session *usess; + struct ltt_ust_channel *uchan; + + usess = session->ust_session; + + uchan = trace_ust_find_channel_by_name(usess->domain_global.channels, + channel_name); + if (uchan == NULL) { + ret = LTTCOMM_UST_CHAN_NOT_FOUND; + goto error; + } + + ret = ust_app_disable_event(usess, uchan, event_name); + if (ret < 0) { + ret = LTTCOMM_UST_DISABLE_FAIL; + goto error; + } + + DBG2("Disable UST event %s in channel %s completed", event_name, + channel_name); + + break; + } case LTTNG_DOMAIN_UST_EXEC_NAME: case LTTNG_DOMAIN_UST_PID: case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN: default: - /* TODO: Other UST domains */ ret = LTTCOMM_NOT_IMPLEMENTED; goto error; } @@ -2315,26 +2349,57 @@ static int cmd_disable_event_all(struct ltt_session *session, int domain, char *channel_name) { int ret; - struct ltt_kernel_channel *kchan; switch (domain) { case LTTNG_DOMAIN_KERNEL: - kchan = trace_kernel_get_channel_by_name(channel_name, - session->kernel_session); + { + struct ltt_kernel_session *ksess; + struct ltt_kernel_channel *kchan; + + ksess = session->kernel_session; + + kchan = trace_kernel_get_channel_by_name(channel_name, ksess); if (kchan == NULL) { ret = LTTCOMM_KERN_CHAN_NOT_FOUND; goto error; } - ret = event_kernel_disable_all(session->kernel_session, kchan); + ret = event_kernel_disable_all(ksess, kchan); if (ret != LTTCOMM_OK) { goto error; } kernel_wait_quiescent(kernel_tracer_fd); break; + } + case LTTNG_DOMAIN_UST: + { + struct ltt_ust_session *usess; + struct ltt_ust_channel *uchan; + + usess = session->ust_session; + + uchan = trace_ust_find_channel_by_name(usess->domain_global.channels, + channel_name); + if (uchan == NULL) { + ret = LTTCOMM_UST_CHAN_NOT_FOUND; + goto error; + } + + ret = ust_app_disable_event_all(usess, uchan); + if (ret < 0) { + ret = LTTCOMM_UST_DISABLE_FAIL; + goto error; + } + + DBG2("Disable all UST event in channel %s completed", channel_name); + + break; + } + case LTTNG_DOMAIN_UST_EXEC_NAME: + case LTTNG_DOMAIN_UST_PID: + case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN: default: - /* TODO: Userspace tracing */ ret = LTTCOMM_NOT_IMPLEMENTED; goto error; } @@ -2418,8 +2483,10 @@ static int cmd_enable_event(struct ltt_session *session, int domain, ret = channel_kernel_create(session->kernel_session, attr, kernel_poll_pipe[1]); if (ret != LTTCOMM_OK) { + free(attr); goto error; } + free(attr); } /* Get the newly created kernel channel pointer */ @@ -2460,6 +2527,7 @@ static int cmd_enable_event(struct ltt_session *session, int domain, /* Use the internal command enable channel */ ret = cmd_enable_channel(session, domain, attr); if (ret < 0) { + free(attr); goto error; } @@ -3121,7 +3189,7 @@ static int process_client_msg(struct command_ctx *cmd_ctx) /* Start the UST consumer daemons */ /* 64-bit */ pthread_mutex_lock(&ustconsumer64_data.pid_mutex); - if (consumerd64_bindir[0] != '\0' && + if (consumerd64_path[0] != '\0' && ustconsumer64_data.pid == 0 && cmd_ctx->lsm->cmd_type != LTTNG_REGISTER_CONSUMER) { pthread_mutex_unlock(&ustconsumer64_data.pid_mutex); @@ -3137,7 +3205,7 @@ static int process_client_msg(struct command_ctx *cmd_ctx) pthread_mutex_unlock(&ustconsumer64_data.pid_mutex); } /* 32-bit */ - if (consumerd32_bindir[0] != '\0' && + if (consumerd32_path[0] != '\0' && ustconsumer32_data.pid == 0 && cmd_ctx->lsm->cmd_type != LTTNG_REGISTER_CONSUMER) { pthread_mutex_unlock(&ustconsumer32_data.pid_mutex); @@ -3577,8 +3645,10 @@ static void usage(void) fprintf(stderr, " --ustconsumerd64-err-sock PATH Specify path for the 64-bit UST consumer error socket\n"); fprintf(stderr, " --ustconsumerd32-cmd-sock PATH Specify path for the 32-bit UST consumer command socket\n"); fprintf(stderr, " --ustconsumerd64-cmd-sock PATH Specify path for the 64-bit UST consumer command socket\n"); - fprintf(stderr, " --ustconsumerd32 PATH Specify path for the 32-bit UST consumer daemon binary\n"); - fprintf(stderr, " --ustconsumerd64 PATH Specify path for the 64-bit UST consumer daemon binary\n"); + fprintf(stderr, " --consumerd32-path PATH Specify path for the 32-bit UST consumer daemon binary\n"); + fprintf(stderr, " --consumerd32-libdir PATH Specify path for the 32-bit UST consumer daemon libraries\n"); + fprintf(stderr, " --consumerd64-path PATH Specify path for the 64-bit UST consumer daemon binary\n"); + fprintf(stderr, " --consumerd64-libdir PATH Specify path for the 64-bit UST consumer daemon libraries\n"); fprintf(stderr, " -d, --daemonize Start as a daemon.\n"); fprintf(stderr, " -g, --group NAME Specify the tracing group name. (default: tracing)\n"); fprintf(stderr, " -V, --version Show version number.\n"); @@ -3600,12 +3670,14 @@ static int parse_args(int argc, char **argv) { "apps-sock", 1, 0, 'a' }, { "kconsumerd-cmd-sock", 1, 0, 'C' }, { "kconsumerd-err-sock", 1, 0, 'E' }, - { "ustconsumerd64", 1, 0, 't' }, - { "ustconsumerd64-cmd-sock", 1, 0, 'D' }, - { "ustconsumerd64-err-sock", 1, 0, 'F' }, - { "ustconsumerd32", 1, 0, 'u' }, { "ustconsumerd32-cmd-sock", 1, 0, 'G' }, { "ustconsumerd32-err-sock", 1, 0, 'H' }, + { "ustconsumerd64-cmd-sock", 1, 0, 'D' }, + { "ustconsumerd64-err-sock", 1, 0, 'F' }, + { "consumerd32-path", 1, 0, 'u' }, + { "consumerd32-libdir", 1, 0, 'U' }, + { "consumerd64-path", 1, 0, 't' }, + { "consumerd64-libdir", 1, 0, 'T' }, { "daemonize", 0, 0, 'd' }, { "sig-parent", 0, 0, 'S' }, { "help", 0, 0, 'h' }, @@ -3682,10 +3754,16 @@ static int parse_args(int argc, char **argv) opt_verbose_consumer += 1; break; case 'u': - consumerd32_bindir = optarg; + consumerd32_path= optarg; + break; + case 'U': + consumerd32_libdir = optarg; break; case 't': - consumerd64_bindir = optarg; + consumerd64_path = optarg; + break; + case 'T': + consumerd64_libdir = optarg; break; default: /* Unknown option or other error.