X-Git-Url: https://git.lttng.org/?p=lttng-modules.git;a=blobdiff_plain;f=wrapper%2Fuprobes.h;h=d76a7881500f045e9deae61f9130dd3ce0c8d0be;hp=6ea9df189878bd48603dd86c3ab7bfa53d99a00e;hb=60c64030e811165e62a62eb551459dd2416d6ec2;hpb=3bcb1fb34fc8904413476afd87afbdeab3806e56

diff --git a/wrapper/uprobes.h b/wrapper/uprobes.h
index 6ea9df18..d76a7881 100644
--- a/wrapper/uprobes.h
+++ b/wrapper/uprobes.h
@@ -52,6 +52,19 @@ int wrapper_uprobe_register(struct inode *inode, loff_t offset, struct uprobe_co
 	}
 }
 
+/*
+ * Canary function to check for 'uprobe_register()' at compile time.
+ *
+ * From 'include/linux/uprobes.h':
+ *
+ *   extern int uprobe_register(struct inode *inode, loff_t offset, struct uprobe_consumer *uc);
+ */
+static inline
+int __canary__uprobe_register(struct inode *inode, loff_t offset, struct uprobe_consumer *uc)
+{
+	return uprobe_register(inode, offset, uc);
+}
+
 static inline
 void wrapper_uprobe_unregister(struct inode *inode, loff_t offset, struct uprobe_consumer *uc)
 {
@@ -66,6 +79,20 @@ void wrapper_uprobe_unregister(struct inode *inode, loff_t offset, struct uprobe
                 WARN_ON(1);
         }
 }
+
+/*
+ * Canary function to check for 'uprobe_unregister()' at compile time.
+ *
+ * From 'include/linux/uprobes.h':
+ *
+ *   extern void uprobe_unregister(struct inode *inode, loff_t offset, struct uprobe_consumer *uc);
+ */
+static inline
+int __canary__uprobe_unregister(struct inode *inode, loff_t offset, struct uprobe_consumer *uc)
+{
+	return uprobe_unregister(inode, offset, uc);
+}
+
 #endif
 #else
 /* Version <  3.5, before uprobe was added. */