X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=libust%2Fltt-probes.c;h=76156f03d5bb7490ba67e8468c3b499d9ee99700;hb=8165c8da23fb8a3395d7e829f3d5734e18dd7db9;hp=5b3a0cb217c208878d957f114aae2c7cf787b1bb;hpb=83610856fc8fea9e25911b0ca9345970cc39ff17;p=lttng-ust.git

diff --git a/libust/ltt-probes.c b/libust/ltt-probes.c
index 5b3a0cb2..76156f03 100644
--- a/libust/ltt-probes.c
+++ b/libust/ltt-probes.c
@@ -14,8 +14,12 @@
 #include <ust/core.h>
 #include <ust/lttng-events.h>
 
+#include "ltt-tracer-core.h"
+
+/*
+ * probe list is protected by lock_ust()/unlock_ust().
+ */
 static CDS_LIST_HEAD(probe_list);
-static DEFINE_MUTEX(probe_mutex);
 
 static
 const struct lttng_event_desc *find_event(const char *name)
@@ -37,7 +41,7 @@ int ltt_probe_register(struct lttng_probe_desc *desc)
 	int ret = 0;
 	int i;
 
-	pthread_mutex_lock(&probe_mutex);
+	lock_ust();
 	/*
 	 * TODO: This is O(N^2). Turn into a hash table when probe registration
 	 * overhead becomes an issue.
@@ -49,25 +53,34 @@ int ltt_probe_register(struct lttng_probe_desc *desc)
 		}
 	}
 	cds_list_add(&desc->head, &probe_list);
+
+	/*
+	 * fix the events awaiting probe load.
+	 */
+	for (i = 0; i < desc->nr_events; i++) {
+		ret = pending_probe_fix_events(&desc->event_desc[i]);
+		assert(!ret);
+	}
 end:
-	pthread_mutex_unlock(&probe_mutex);
+	unlock_ust();
 	return ret;
 }
 
 void ltt_probe_unregister(struct lttng_probe_desc *desc)
 {
-	pthread_mutex_lock(&probe_mutex);
+	lock_ust();
 	cds_list_del(&desc->head);
-	pthread_mutex_unlock(&probe_mutex);
+	unlock_ust();
 }
 
+/*
+ * called with UST lock held.
+ */
 const struct lttng_event_desc *ltt_event_get(const char *name)
 {
 	const struct lttng_event_desc *event;
 
-	pthread_mutex_lock(&probe_mutex);
 	event = find_event(name);
-	pthread_mutex_unlock(&probe_mutex);
 	if (!event)
 		return NULL;
 	return event;