test mark update

[lttv.git] / tests / kernel / probe.c

diff --git a/tests/kernel/probe.c b/tests/kernel/probe.c
index 96d9ff4d0bc13ac26fcb07c4f335eda9c5b41360..f8916a1a77d05cc23c101bd57566d40ef8e30afc 100644 (file)
--- a/tests/kernel/probe.c
+++ b/tests/kernel/probe.c
@@ -13,21 +13,56 @@
 #include <linux/kallsyms.h>
 
 /* function to install */
-void do_mark1(const char *format, int value)
+#define DO_MARK1_FORMAT "%d"
+asmlinkage void do_mark1(const char *format, int value)
 {
+       __mark_check_format(DO_MARK1_FORMAT, value);
        printk("value is %d\n", value);
 }
 
+#define DO_MARK2_FORMAT "%d %s"
+asmlinkage void do_mark2(const char *format, int value, const char *string)
+{
+       __mark_check_format(DO_MARK2_FORMAT, value, string);
+       printk("value is %d %s\n", value, string);
+}
+
+#define DO_MARK3_FORMAT "%d %s %s"
+asmlinkage void do_mark3(const char *format, int value, const char *s1,
+                               const char *s2)
+{
+       __mark_check_format(DO_MARK3_FORMAT, value, s1, s2);
+       printk("value is %d %s %s\n", value, s1, s2);
+}
+
 int init_module(void)
 {
-       return marker_set_probe("subsys_mark1", (marker_probe)do_mark1,
-                       MARKER_CALL);
+       int result;
+       result = marker_set_probe("subsys_mark1", DO_MARK1_FORMAT,
+                       (marker_probe_func*)do_mark1);
+       if(!result) goto end;
+       result = marker_set_probe("subsys_mark2", DO_MARK2_FORMAT,
+                       (marker_probe_func*)do_mark2);
+       if(!result) goto cleanup1;
+       result = marker_set_probe("subsys_mark3", DO_MARK3_FORMAT,
+                       (marker_probe_func*)do_mark3);
+       if(!result) goto cleanup2;
+
+       return 0;
+
+cleanup2:
+       marker_remove_probe((marker_probe_func*)do_mark2);
+cleanup1:
+       marker_remove_probe((marker_probe_func*)do_mark1);
+end:
+       return -EPERM;
 }
 
 void cleanup_module(void)
 {
-       marker_disable_probe("subsys_mark1", (marker_probe)do_mark1,
-               MARKER_CALL);
+       marker_remove_probe((marker_probe_func*)do_mark1);
+       marker_remove_probe((marker_probe_func*)do_mark2);
+       marker_remove_probe((marker_probe_func*)do_mark3);
 }
 
 MODULE_LICENSE("GPL");