X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=liblttng-ust%2Flttng-context-provider.c;h=b3fc21defd48bedecf3d60859c0c0a24675cf07f;hb=2b6f437483769831e50886b9f3bc4710b28f4ac9;hp=e7462cdabec945522c697576c632511f97043567;hpb=53569322d40ed45abe0368ddb08eb4a2738afc37;p=lttng-ust.git

diff --git a/liblttng-ust/lttng-context-provider.c b/liblttng-ust/lttng-context-provider.c
index e7462cda..b3fc21de 100644
--- a/liblttng-ust/lttng-context-provider.c
+++ b/liblttng-ust/lttng-context-provider.c
@@ -1,31 +1,23 @@
 /*
- * lttng-context-provider.c
- *
- * LTTng UST application context provider.
+ * SPDX-License-Identifier: LGPL-2.1-only
  *
  * Copyright (C) 2016 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
  *
- * 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 application context provider.
  */
 
+#define _LGPL_SOURCE
+#include <stddef.h>
+#include <stdint.h>
 #include <sys/types.h>
 #include <unistd.h>
-#include <lttng/ust-context-provider.h>
+
+#include <ust-context-provider.h>
+
 #include "lttng-tracer-core.h"
 #include "jhash.h"
-#include <helper.h>
+#include "context-provider-internal.h"
+#include <ust-helper.h>
 
 #define CONTEXT_PROVIDER_HT_BITS	12
 #define CONTEXT_PROVIDER_HT_SIZE	(1U << CONTEXT_PROVIDER_HT_BITS)
@@ -67,10 +59,12 @@ int lttng_ust_context_provider_register(struct lttng_ust_context_provider *provi
 	uint32_t hash;
 	int ret = 0;
 
+	lttng_ust_fixup_tls();
+
 	/* Provider name starts with "$app.". */
-	if (strncmp("$app.", provider->name, strlen("$app.") != 0))
+	if (strncmp("$app.", provider->name, strlen("$app.")) != 0)
 		return -EINVAL;
-	/* Provider name cannot contain a column character. */
+	/* Provider name cannot contain a colon character. */
 	if (strchr(provider->name, ':'))
 		return -EINVAL;
 	if (ust_lock()) {
@@ -84,48 +78,33 @@ int lttng_ust_context_provider_register(struct lttng_ust_context_provider *provi
 	hash = jhash(provider->name, name_len, 0);
 	head = &context_provider_ht.table[hash & (CONTEXT_PROVIDER_HT_SIZE - 1)];
 	cds_hlist_add_head(&provider->node, head);
+
 	lttng_ust_context_set_session_provider(provider->name,
 		provider->get_size, provider->record,
 		provider->get_value);
+
+	lttng_ust_context_set_event_notifier_group_provider(provider->name,
+		provider->get_size, provider->record,
+		provider->get_value);
 end:
 	ust_unlock();
 	return ret;
 }
 
-static
-size_t dummy_get_size(struct lttng_ctx_field *field, size_t offset)
-{
-	size_t size = 0;
-
-	size += lib_ring_buffer_align(offset, lttng_alignof(char));
-	size += sizeof(char);		/* tag */
-	return size;
-}
-
-static
-void dummy_record(struct lttng_ctx_field *field,
-		 struct lttng_ust_lib_ring_buffer_ctx *ctx,
-		 struct lttng_channel *chan)
-{
-	char sel_char = (char) LTTNG_UST_DYNAMIC_TYPE_NONE;
-
-	lib_ring_buffer_align_ctx(ctx, lttng_alignof(sel_char));
-	chan->ops->event_write(ctx, &sel_char, sizeof(sel_char));
-}
-
-static
-void dummy_get_value(struct lttng_ctx_field *field,
-		struct lttng_ctx_value *value)
-{
-	value->sel = LTTNG_UST_DYNAMIC_TYPE_NONE;
-}
-
 void lttng_ust_context_provider_unregister(struct lttng_ust_context_provider *provider)
 {
+	lttng_ust_fixup_tls();
+
 	if (ust_lock())
 		goto end;
 	lttng_ust_context_set_session_provider(provider->name,
-		dummy_get_size, dummy_record, dummy_get_value);
+		lttng_ust_dummy_get_size, lttng_ust_dummy_record,
+		lttng_ust_dummy_get_value);
+
+	lttng_ust_context_set_event_notifier_group_provider(provider->name,
+		lttng_ust_dummy_get_size, lttng_ust_dummy_record,
+		lttng_ust_dummy_get_value);
+
 	cds_hlist_del(&provider->node);
 end:
 	ust_unlock();
@@ -168,9 +147,9 @@ int lttng_ust_add_app_context_to_ctx_rcu(const char *name,
 		new_field.record = provider->record;
 		new_field.get_value = provider->get_value;
 	} else {
-		new_field.get_size = dummy_get_size;
-		new_field.record = dummy_record;
-		new_field.get_value = dummy_get_value;
+		new_field.get_size = lttng_ust_dummy_get_size;
+		new_field.record = lttng_ust_dummy_record;
+		new_field.get_value = lttng_ust_dummy_get_value;
 	}
 	ret = lttng_context_add_rcu(ctx, &new_field);
 	if (ret) {