X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=probes%2Flttng-kretprobes.c;h=2da3143cd0d4d081170e988273faf452cd0d4105;hb=refs%2Ftags%2Fv2.12.5;hp=a6bcd21883468929378d556eac1a14209c95d661;hpb=57ccbfa6a8a79c7b84394c2097efaf7935607aa5;p=lttng-modules.git

diff --git a/probes/lttng-kretprobes.c b/probes/lttng-kretprobes.c
index a6bcd218..2da3143c 100644
--- a/probes/lttng-kretprobes.c
+++ b/probes/lttng-kretprobes.c
@@ -8,7 +8,7 @@
  */
 
 #include <linux/module.h>
-#include <linux/kprobes.h>
+#include <wrapper/kprobes.h>
 #include <linux/slab.h>
 #include <linux/kref.h>
 #include <lttng-events.h>
@@ -36,7 +36,7 @@ int _lttng_kretprobes_handler(struct kretprobe_instance *krpi,
 			      enum lttng_kretprobe_type type)
 {
 	struct lttng_krp *lttng_krp =
-		container_of(krpi->rp, struct lttng_krp, krp);
+		container_of(lttng_get_kretprobe(krpi), struct lttng_krp, krp);
 	struct lttng_event *event =
 		lttng_krp->event[type];
 	struct lttng_probe_ctx lttng_probe_ctx = {
@@ -58,7 +58,7 @@ int _lttng_kretprobes_handler(struct kretprobe_instance *krpi,
 	if (unlikely(!LTTNG_READ_ONCE(event->enabled)))
 		return 0;
 
-	payload.ip = (unsigned long) krpi->rp->kp.addr;
+	payload.ip = (unsigned long) lttng_get_kretprobe(krpi)->kp.addr;
 	payload.parent_ip = (unsigned long) krpi->ret_addr;
 
 	lib_ring_buffer_ctx_init(&ctx, chan->chan, &lttng_probe_ctx, sizeof(payload),