X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=liblttng-ust%2Flttng-context-procname.c;h=96a1b2ca2c36083b9d921220f1173f551256025d;hb=6ba6fd60507f8e045bdc4f1be14e9d99c6a15f7f;hp=96cd1ee2c8f4979f7c7fbaf7c9367c2d9b59061b;hpb=009745db8ca05f7a3abbb37558b08eae0107f7e1;p=lttng-ust.git diff --git a/liblttng-ust/lttng-context-procname.c b/liblttng-ust/lttng-context-procname.c index 96cd1ee2..96a1b2ca 100644 --- a/liblttng-ust/lttng-context-procname.c +++ b/liblttng-ust/lttng-context-procname.c @@ -1,30 +1,25 @@ /* - * lttng-context-procname.c - * - * LTTng UST procname context. + * SPDX-License-Identifier: LGPL-2.1-only * * Copyright (C) 2009-2012 Mathieu Desnoyers * - * This library is free software; you can redistribute it and/or - * modify it under the terms of the GNU Lesser General Public - * License as published by the Free Software Foundation; only - * version 2.1 of the License. - * - * This library is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - * Lesser General Public License for more details. - * - * You should have received a copy of the GNU Lesser General Public - * License along with this library; if not, write to the Free Software - * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + * LTTng UST procname context. */ +#define _LGPL_SOURCE +#include #include #include -#include +#include +#include #include #include "compat.h" +#include "lttng-tracer-core.h" + +#include "context-internal.h" + +/* Maximum number of nesting levels for the procname cache. */ +#define PROCNAME_NESTING_MAX 2 /* * We cache the result to ensure we don't trigger a system call for @@ -35,67 +30,90 @@ * be set for a thread before the first event is logged within this * thread. */ -static __thread char cached_procname[17]; +typedef char procname_array[PROCNAME_NESTING_MAX][17]; + +static DEFINE_URCU_TLS(procname_array, cached_procname); + +static DEFINE_URCU_TLS(int, procname_nesting); static inline -char *wrapper_getprocname(void) +const char *wrapper_getprocname(void) { - if (caa_unlikely(!cached_procname[0])) { - lttng_ust_getprocname(cached_procname); - cached_procname[LTTNG_UST_PROCNAME_LEN - 1] = '\0'; + int nesting = CMM_LOAD_SHARED(URCU_TLS(procname_nesting)); + + if (caa_unlikely(nesting >= PROCNAME_NESTING_MAX)) + return ""; + if (caa_unlikely(!URCU_TLS(cached_procname)[nesting][0])) { + CMM_STORE_SHARED(URCU_TLS(procname_nesting), nesting + 1); + /* Increment nesting before updating cache. */ + cmm_barrier(); + lttng_pthread_getname_np(URCU_TLS(cached_procname)[nesting], LTTNG_UST_ABI_PROCNAME_LEN); + URCU_TLS(cached_procname)[nesting][LTTNG_UST_ABI_PROCNAME_LEN - 1] = '\0'; + /* Decrement nesting after updating cache. */ + cmm_barrier(); + CMM_STORE_SHARED(URCU_TLS(procname_nesting), nesting); } - return cached_procname; + return URCU_TLS(cached_procname)[nesting]; } -void lttng_context_procname_reset(void) +/* Reset should not be called from a signal handler. */ +void lttng_ust_context_procname_reset(void) { - cached_procname[0] = '\0'; + CMM_STORE_SHARED(URCU_TLS(cached_procname)[1][0], '\0'); + CMM_STORE_SHARED(URCU_TLS(procname_nesting), 1); + CMM_STORE_SHARED(URCU_TLS(cached_procname)[0][0], '\0'); + CMM_STORE_SHARED(URCU_TLS(procname_nesting), 0); } static -size_t procname_get_size(size_t offset) +size_t procname_get_size(void *priv __attribute__((unused)), + size_t offset __attribute__((unused))) { - size_t size = 0; - - size += LTTNG_UST_PROCNAME_LEN; - return size; + return LTTNG_UST_ABI_PROCNAME_LEN; } static -void procname_record(struct lttng_ctx_field *field, +void procname_record(void *priv __attribute__((unused)), struct lttng_ust_lib_ring_buffer_ctx *ctx, - struct ltt_channel *chan) + struct lttng_ust_channel_buffer *chan) { - char *procname; + const char *procname; procname = wrapper_getprocname(); - chan->ops->event_write(ctx, procname, LTTNG_UST_PROCNAME_LEN); + chan->ops->event_write(ctx, procname, LTTNG_UST_ABI_PROCNAME_LEN, 1); } -int lttng_add_procname_to_ctx(struct lttng_ctx **ctx) +static +void procname_get_value(void *priv __attribute__((unused)), + struct lttng_ust_ctx_value *value) { - struct lttng_ctx_field *field; - - field = lttng_append_context(ctx); - if (!field) - return -ENOMEM; - if (lttng_find_context(*ctx, "procname")) { - lttng_remove_context_field(ctx, field); - return -EEXIST; + value->u.str = wrapper_getprocname(); +} + +static const struct lttng_ust_ctx_field *ctx_field = lttng_ust_static_ctx_field( + lttng_ust_static_event_field("procname", + lttng_ust_static_type_array_text(LTTNG_UST_ABI_PROCNAME_LEN), + false, false), + procname_get_size, + procname_record, + procname_get_value, + NULL, NULL); + +int lttng_add_procname_to_ctx(struct lttng_ust_ctx **ctx) +{ + int ret; + + if (lttng_find_context(*ctx, ctx_field->event_field->name)) { + ret = -EEXIST; + goto error_find_context; } - field->event_field.name = "procname"; - field->event_field.type.atype = atype_array; - field->event_field.type.u.array.elem_type.atype = atype_integer; - field->event_field.type.u.array.elem_type.u.basic.integer.size = sizeof(char) * CHAR_BIT; - field->event_field.type.u.array.elem_type.u.basic.integer.alignment = lttng_alignof(char) * CHAR_BIT; - field->event_field.type.u.array.elem_type.u.basic.integer.signedness = lttng_is_signed_type(char); - field->event_field.type.u.array.elem_type.u.basic.integer.reverse_byte_order = 0; - field->event_field.type.u.array.elem_type.u.basic.integer.base = 10; - field->event_field.type.u.array.elem_type.u.basic.integer.encoding = lttng_encode_UTF8; - field->event_field.type.u.array.length = LTTNG_UST_PROCNAME_LEN; - field->get_size = procname_get_size; - field->record = procname_record; + ret = lttng_ust_context_append(ctx, ctx_field); + if (ret) + return ret; return 0; + +error_find_context: + return ret; } /* @@ -103,5 +121,5 @@ int lttng_add_procname_to_ctx(struct lttng_ctx **ctx) */ void lttng_fixup_procname_tls(void) { - asm volatile ("" : : "m" (cached_procname[0])); + asm volatile ("" : : "m" (URCU_TLS(cached_procname)[0])); }