X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=lttng-events.c;h=eefee69b31a5cc59583fb57d7f074c5715aec654;hb=1c53e689434a6bdd7dc3f54c07bfb72750d1d24c;hp=ff28c9af559aecff2d5952a6f5fe9a43e2e343fe;hpb=0c95667685cf8612eb0d231883874c173b7bb5a1;p=lttng-modules.git

diff --git a/lttng-events.c b/lttng-events.c
index ff28c9af..eefee69b 100644
--- a/lttng-events.c
+++ b/lttng-events.c
@@ -400,7 +400,7 @@ struct lttng_event *lttng_event_create(struct lttng_channel *chan,
 			ret = -ENOENT;
 			goto register_error;
 		}
-		ret = kabi_2635_tracepoint_probe_register(event->desc->kname,
+		ret = lttng_wrapper_tracepoint_probe_register(event->desc->kname,
 				event->desc->probe_callback,
 				event);
 		if (ret) {
@@ -519,7 +519,7 @@ int _lttng_event_unregister(struct lttng_event *event)
 
 	switch (event->instrumentation) {
 	case LTTNG_KERNEL_TRACEPOINT:
-		ret = kabi_2635_tracepoint_probe_unregister(event->desc->kname,
+		ret = lttng_wrapper_tracepoint_probe_unregister(event->desc->kname,
 						  event->desc->probe_callback,
 						  event);
 		if (ret)
@@ -1255,9 +1255,14 @@ static int __init lttng_events_init(void)
 {
 	int ret;
 
+	ret = lttng_tracepoint_init();
+	if (ret)
+		return ret;
 	event_cache = KMEM_CACHE(lttng_event, 0);
-	if (!event_cache)
-		return -ENOMEM;
+	if (!event_cache) {
+		ret = -ENOMEM;
+		goto error_kmem;
+	}
 	ret = lttng_abi_init();
 	if (ret)
 		goto error_abi;
@@ -1270,6 +1275,8 @@ error_logger:
 	lttng_abi_exit();
 error_abi:
 	kmem_cache_destroy(event_cache);
+error_kmem:
+	lttng_tracepoint_exit();
 	return ret;
 }
 
@@ -1284,6 +1291,7 @@ static void __exit lttng_events_exit(void)
 	list_for_each_entry_safe(session, tmpsession, &sessions, list)
 		lttng_session_destroy(session);
 	kmem_cache_destroy(event_cache);
+	lttng_tracepoint_exit();
 }
 
 module_exit(lttng_events_exit);