X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;ds=sidebyside;f=wrapper%2Fftrace.h;h=ace33c54100daf91235b611ee275d4727d4c575e;hb=cff67ebe90fc083a014bf8b2dae0c120b3201d1d;hp=4401469251a27205e0bdfc8bff2292818019c48f;hpb=5a9479dcba0b43d326569fa0bd48318222d23fcb;p=lttng-modules.git

diff --git a/wrapper/ftrace.h b/wrapper/ftrace.h
index 44014692..ace33c54 100644
--- a/wrapper/ftrace.h
+++ b/wrapper/ftrace.h
@@ -16,6 +16,7 @@
 #ifdef CONFIG_KALLSYMS
 
 #include <linux/kallsyms.h>
+#include "kallsyms.h"
 
 static inline
 int wrapper_register_ftrace_function_probe(char *glob,
@@ -24,7 +25,7 @@ int wrapper_register_ftrace_function_probe(char *glob,
 	int (*register_ftrace_function_probe_sym)(char *glob,
 			struct ftrace_probe_ops *ops, void *data);
 
-	register_ftrace_function_probe_sym = (void *) kallsyms_lookup_name("register_ftrace_function_probe_sym");
+	register_ftrace_function_probe_sym = (void *) kallsyms_lookup_funcptr("register_ftrace_function_probe");
 	if (register_ftrace_function_probe_sym) {
 		return register_ftrace_function_probe_sym(glob, ops, data);
 	} else {
@@ -40,7 +41,7 @@ void wrapper_unregister_ftrace_function_probe(char *glob,
 	void (*unregister_ftrace_function_probe_sym)(char *glob,
 			struct ftrace_probe_ops *ops, void *data);
 
-	unregister_ftrace_function_probe_sym = (void *) kallsyms_lookup_name("unregister_ftrace_function_probe_sym");
+	unregister_ftrace_function_probe_sym = (void *) kallsyms_lookup_funcptr("unregister_ftrace_function_probe");
 	if (unregister_ftrace_function_probe_sym) {
 		unregister_ftrace_function_probe_sym(glob, ops, data);
 	} else {
@@ -53,17 +54,16 @@ void wrapper_unregister_ftrace_function_probe(char *glob,
 
 static inline
 int wrapper_register_ftrace_function_probe(char *glob,
-		struct ftrace_probe_ops *ops, void *data);
-
+		struct ftrace_probe_ops *ops, void *data)
 {
-	return unregister_ftrace_function_probe();
+	return register_ftrace_function_probe(glob, ops, data);
 }
 
 static inline
 void wrapper_unregister_ftrace_function_probe(char *glob,
-		struct ftrace_probe_ops *ops, void *data);
+		struct ftrace_probe_ops *ops, void *data)
 {
-	return unregister_ftrace_function_probe();
+	return unregister_ftrace_function_probe(glob, ops, data);
 }
 #endif