X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=probes%2Flttng-kretprobes.c;fp=probes%2Flttng-kretprobes.c;h=52b3f78138f94978427eedc8e197175527270274;hb=a0493bef1367458230b0319401f8533b0400cea2;hp=d71090a2bc321edcac3695c774508c1d0d025655;hpb=f6da044fcd41afe52612b72376672912b2118ce4;p=lttng-modules.git

diff --git a/probes/lttng-kretprobes.c b/probes/lttng-kretprobes.c
index d71090a2..52b3f781 100644
--- a/probes/lttng-kretprobes.c
+++ b/probes/lttng-kretprobes.c
@@ -285,6 +285,26 @@ void lttng_kretprobes_destroy_private(struct lttng_event *event)
 }
 EXPORT_SYMBOL_GPL(lttng_kretprobes_destroy_private);
 
+int lttng_kretprobes_event_enable_state(struct lttng_event *event,
+		int enable)
+{
+	struct lttng_event *event_return;
+	struct lttng_krp *lttng_krp;
+
+	if (event->instrumentation != LTTNG_KERNEL_KRETPROBE) {
+		return -EINVAL;
+	}
+	if (event->enabled == enable) {
+		return -EBUSY;
+	}
+	lttng_krp = event->u.kretprobe.lttng_krp;
+	event_return = lttng_krp->event[EVENT_RETURN];
+	ACCESS_ONCE(event->enabled) = enable;
+	ACCESS_ONCE(event_return->enabled) = enable;
+	return 0;
+}
+EXPORT_SYMBOL_GPL(lttng_kretprobes_event_enable_state);
+
 MODULE_LICENSE("GPL and additional rights");
 MODULE_AUTHOR("Mathieu Desnoyers");
 MODULE_DESCRIPTION("Linux Trace Toolkit Kretprobes Support");