lttng-ctl and lttng-sessiond use serialized communication for messages using lttng_event_context.
Signed-off-by: Yannick Lamarre <ylama...@efficios.com> --- include/lttng/event-internal.h | 4 ++++ src/bin/lttng-sessiond/client.c | 16 +++++++++------- src/common/sessiond-comm/sessiond-comm.h | 2 +- src/lib/lttng-ctl/lttng-ctl.c | 12 +----------- 4 files changed, 15 insertions(+), 19 deletions(-) diff --git a/include/lttng/event-internal.h b/include/lttng/event-internal.h index 25a88448..09b4d232 100644 --- a/include/lttng/event-internal.h +++ b/include/lttng/event-internal.h @@ -57,4 +57,8 @@ struct lttng_event_extended { LTTNG_HIDDEN struct lttng_event *lttng_event_copy(const struct lttng_event *event); +int lttng_event_context_serialize(struct lttng_event_context_serialized *dst, const struct lttng_event_context *src); +int lttng_event_context_deserialize(struct lttng_event_context *dst, const struct lttng_event_context_serialized *src); +int lttng_event_perf_counter_ctx_serialize(struct lttng_event_perf_counter_ctx_serialized *dst, const struct lttng_event_perf_counter_ctx *src); +int lttng_event_perf_counter_ctx_deserialize(struct lttng_event_perf_counter_ctx *dst, const struct lttng_event_perf_counter_ctx_serialized *src); #endif /* LTTNG_EVENT_INTERNAL_H */ diff --git a/src/bin/lttng-sessiond/client.c b/src/bin/lttng-sessiond/client.c index 0e36e5ad..24e688c5 100644 --- a/src/bin/lttng-sessiond/client.c +++ b/src/bin/lttng-sessiond/client.c @@ -1069,6 +1069,8 @@ skip_domain: switch (cmd_ctx->lsm->cmd_type) { case LTTNG_ADD_CONTEXT: { + struct lttng_event_context evt_ctx; + lttng_event_context_deserialize(&evt_ctx, &cmd_ctx->lsm->u.context.ctx); /* * An LTTNG_ADD_CONTEXT command might have a supplementary * payload if the context being added is an application context. @@ -1095,7 +1097,7 @@ skip_domain: ret = -LTTNG_ERR_NOMEM; goto error; } - cmd_ctx->lsm->u.context.ctx.u.app_ctx.provider_name = + evt_ctx.u.app_ctx.provider_name = provider_name; context_name = zmalloc(context_name_len + 1); @@ -1103,7 +1105,7 @@ skip_domain: ret = -LTTNG_ERR_NOMEM; goto error_add_context; } - cmd_ctx->lsm->u.context.ctx.u.app_ctx.ctx_name = + evt_ctx.u.app_ctx.ctx_name = context_name; ret = lttcomm_recv_unix_sock(sock, provider_name, @@ -1126,14 +1128,14 @@ skip_domain: ret = cmd_add_context(cmd_ctx->session, cmd_ctx->lsm->domain.type, cmd_ctx->lsm->u.context.channel_name, - &cmd_ctx->lsm->u.context.ctx, + &evt_ctx, kernel_poll_pipe[1]); - cmd_ctx->lsm->u.context.ctx.u.app_ctx.provider_name = NULL; - cmd_ctx->lsm->u.context.ctx.u.app_ctx.ctx_name = NULL; + evt_ctx.u.app_ctx.provider_name = NULL; + evt_ctx.u.app_ctx.ctx_name = NULL; error_add_context: - free(cmd_ctx->lsm->u.context.ctx.u.app_ctx.provider_name); - free(cmd_ctx->lsm->u.context.ctx.u.app_ctx.ctx_name); + free(evt_ctx.u.app_ctx.provider_name); + free(evt_ctx.u.app_ctx.ctx_name); if (ret < 0) { goto error; } diff --git a/src/common/sessiond-comm/sessiond-comm.h b/src/common/sessiond-comm/sessiond-comm.h index 2760af3a..78b18185 100644 --- a/src/common/sessiond-comm/sessiond-comm.h +++ b/src/common/sessiond-comm/sessiond-comm.h @@ -410,7 +410,7 @@ struct lttcomm_session_msg { /* Context */ struct { char channel_name[LTTNG_SYMBOL_NAME_LEN]; - struct lttng_event_context ctx LTTNG_PACKED; + struct lttng_event_context_serialized ctx; uint32_t provider_name_len; uint32_t context_name_len; } LTTNG_PACKED context; diff --git a/src/lib/lttng-ctl/lttng-ctl.c b/src/lib/lttng-ctl/lttng-ctl.c index 789534c8..686a98ef 100644 --- a/src/lib/lttng-ctl/lttng-ctl.c +++ b/src/lib/lttng-ctl/lttng-ctl.c @@ -821,17 +821,7 @@ int lttng_add_context(struct lttng_handle *handle, memcpy(buf, provider_name, provider_len); memcpy(buf + provider_len, ctx_name, ctx_len); } - memcpy(&lsm.u.context.ctx, ctx, sizeof(struct lttng_event_context)); - - if (ctx->ctx == LTTNG_EVENT_CONTEXT_APP_CONTEXT) { - /* - * Don't leak application addresses to the sessiond. - * This is only necessary when ctx is for an app ctx otherwise - * the values inside the union (type & config) are overwritten. - */ - lsm.u.context.ctx.u.app_ctx.provider_name = NULL; - lsm.u.context.ctx.u.app_ctx.ctx_name = NULL; - } + lttng_event_context_serialize(&lsm.u.context.ctx, ctx); ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, buf, len, NULL); end: -- 2.11.0 _______________________________________________ lttng-dev mailing list lttng-dev@lists.lttng.org https://lists.lttng.org/cgi-bin/mailman/listinfo/lttng-dev