X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=liblttng-ust%2Flttng-context-provider.c;h=fe831366779a0332531fcfad63769742481549a0;hb=cab88ff8cb13622233f1e0ed338d8d10a229370d;hp=6c067e2fecae06548a3d4495ef9477b2495a6a5b;hpb=797b05f6cf195a6699b6aff0a76b8e1274d5b51f;p=lttng-ust.git

diff --git a/liblttng-ust/lttng-context-provider.c b/liblttng-ust/lttng-context-provider.c
index 6c067e2f..fe831366 100644
--- a/liblttng-ust/lttng-context-provider.c
+++ b/liblttng-ust/lttng-context-provider.c
@@ -20,11 +20,17 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
+#define _LGPL_SOURCE
+#include <stddef.h>
+#include <stdint.h>
 #include <sys/types.h>
 #include <unistd.h>
+
 #include <lttng/ust-context-provider.h>
+
 #include "lttng-tracer-core.h"
 #include "jhash.h"
+#include "context-provider-internal.h"
 #include <helper.h>
 
 #define CONTEXT_PROVIDER_HT_BITS	12
@@ -67,10 +73,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,9 +92,14 @@ 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;
@@ -94,11 +107,18 @@ end:
 
 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,
 		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();