X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=include%2Flttng%2Ftracepoint.h;h=2942c175e7cf0580ecf39edd99b78a1eb47d9c9c;hb=943be3a0046d008e90ecf503be582540a463091d;hp=148590b8e9d5c76810cbb773907d112bf2317b08;hpb=8fce9e2bd2103f4b413db52e7a6d970a7b057843;p=lttng-ust.git

diff --git a/include/lttng/tracepoint.h b/include/lttng/tracepoint.h
index 148590b8..2942c175 100644
--- a/include/lttng/tracepoint.h
+++ b/include/lttng/tracepoint.h
@@ -14,10 +14,11 @@
  * modified is included with the above copyright notice.
  */
 
-#include <urcu-bp.h>
 #include <lttng/tracepoint-types.h>
+#include <lttng/tracepoint-rcu.h>
 #include <urcu/compiler.h>
 #include <dlfcn.h>	/* for dlopen */
+#include <assert.h>
 
 #ifdef __cplusplus
 extern "C" {
@@ -119,10 +120,10 @@ static inline void __tracepoint_cb_##provider##___##name(_TP_ARGS_PROTO(__VA_ARG
 {											\
 	struct tracepoint_probe *__tp_probe;						\
 											\
-	if (!rcu_read_lock_bp)								\
+	if (!TP_RCU_LINK_TEST())							\
 		return;									\
-	rcu_read_lock_bp();								\
-	__tp_probe = rcu_dereference_bp(__tracepoint_##provider##___##name.probes);	\
+	tp_rcu_read_lock_bp();								\
+	__tp_probe = tp_rcu_dereference_bp(__tracepoint_##provider##___##name.probes);	\
 	if (caa_unlikely(!__tp_probe))							\
 		goto end;								\
 	do {										\
@@ -133,7 +134,7 @@ static inline void __tracepoint_cb_##provider##___##name(_TP_ARGS_PROTO(__VA_ARG
 				(_TP_ARGS_DATA_VAR(__VA_ARGS__));			\
 	} while ((++__tp_probe)->func);							\
 end:											\
-	rcu_read_unlock_bp();								\
+	tp_rcu_read_unlock_bp();							\
 }											\
 static inline void __tracepoint_register_##provider##___##name(char *name,		\
 		void *func, void *data)							\
@@ -200,6 +201,20 @@ static void __attribute__((constructor)) __tracepoints__init(void)
 		URCU_FORCE_CAST(int (*)(struct tracepoint * const *),
 				dlsym(liblttngust_handle,
 					"tracepoint_unregister_lib"));
+#ifndef _LGPL_SOURCE
+	tp_rcu_read_lock_bp =
+		URCU_FORCE_CAST(void (*)(void),
+				dlsym(liblttngust_handle,
+					"tp_rcu_read_lock_bp"));
+	tp_rcu_read_unlock_bp =
+		URCU_FORCE_CAST(void (*)(void),
+				dlsym(liblttngust_handle,
+					"tp_rcu_read_unlock_bp"));
+	tp_rcu_dereference_sym_bp =
+		URCU_FORCE_CAST(void *(*)(void *p),
+				dlsym(liblttngust_handle,
+					"tp_rcu_dereference_sym_bp"));
+#endif
 	tracepoint_register_lib(__start___tracepoints_ptrs,
 				__stop___tracepoints_ptrs -
 				__start___tracepoints_ptrs);