X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;ds=inline;f=probes%2Flttng-ftrace.c;h=ec086902892d644fcee8c57f70d78231cc4944f2;hb=edeb3137b6064818f310567ded5c60e442a933e6;hp=7637e4b54dff1b1f1effda651ee3587e335ec03c;hpb=1695dc9a055c3383067cccac91cb8918cb2eaab1;p=lttng-modules.git

diff --git a/probes/lttng-ftrace.c b/probes/lttng-ftrace.c
index 7637e4b5..ec086902 100644
--- a/probes/lttng-ftrace.c
+++ b/probes/lttng-ftrace.c
@@ -92,6 +92,7 @@ int lttng_create_ftrace_event(const char *name, struct ltt_event *event)
 	fields[1].type.u.basic.integer.base = 16;
 	fields[1].type.u.basic.integer.encoding = lttng_encode_none;
 
+	desc->owner = THIS_MODULE;
 	event->desc = desc;
 
 	return 0;
@@ -118,7 +119,7 @@ int lttng_ftrace_register(const char *name,
 	if (ret)
 		goto error;
 
-	event->u.ftrace.symbol_name = kstrdup(name, GFP_KERNEL);
+	event->u.ftrace.symbol_name = kstrdup(symbol_name, GFP_KERNEL);
 	if (!event->u.ftrace.symbol_name)
 		goto name_error;
 
@@ -127,7 +128,7 @@ int lttng_ftrace_register(const char *name,
 
 	ret = wrapper_register_ftrace_function_probe(event->u.ftrace.symbol_name,
 			&lttng_ftrace_ops, event);
-	if (ret)
+	if (ret < 0)
 		goto register_error;
 	return 0;
 
@@ -145,12 +146,17 @@ void lttng_ftrace_unregister(struct ltt_event *event)
 {
 	wrapper_unregister_ftrace_function_probe(event->u.ftrace.symbol_name,
 			&lttng_ftrace_ops, event);
+}
+EXPORT_SYMBOL_GPL(lttng_ftrace_unregister);
+
+void lttng_ftrace_destroy_private(struct ltt_event *event)
+{
 	kfree(event->u.ftrace.symbol_name);
 	kfree(event->desc->fields);
 	kfree(event->desc->name);
 	kfree(event->desc);
 }
-EXPORT_SYMBOL_GPL(lttng_ftrace_unregister);
+EXPORT_SYMBOL_GPL(lttng_ftrace_destroy_private);
 
 int lttng_ftrace_init(void)
 {