X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=src%2Flttng-context-need-reschedule.c;h=0ccbfc78793a2bdaee154430d6c38b3b30bf54a8;hb=f5ffbd770ec2b9d1ac9b7059eb33a01432043bce;hp=6984d1a44d7dfdac404c550e5cad042272da501f;hpb=2dc781e02eb156a76554ada092a181ab2916db57;p=lttng-modules.git

diff --git a/src/lttng-context-need-reschedule.c b/src/lttng-context-need-reschedule.c
index 6984d1a4..0ccbfc78 100644
--- a/src/lttng-context-need-reschedule.c
+++ b/src/lttng-context-need-reschedule.c
@@ -18,7 +18,7 @@
 #include <lttng/tracer.h>
 
 static
-size_t need_reschedule_get_size(void *priv, struct lttng_probe_ctx *probe_ctx, size_t offset)
+size_t need_reschedule_get_size(void *priv, struct lttng_kernel_probe_ctx *probe_ctx, size_t offset)
 {
 	size_t size = 0;
 
@@ -28,19 +28,18 @@ size_t need_reschedule_get_size(void *priv, struct lttng_probe_ctx *probe_ctx, s
 }
 
 static
-void need_reschedule_record(void *priv, struct lttng_probe_ctx *probe_ctx,
-		struct lib_ring_buffer_ctx *ctx,
-		struct lttng_channel *chan)
+void need_reschedule_record(void *priv, struct lttng_kernel_probe_ctx *probe_ctx,
+		struct lttng_kernel_ring_buffer_ctx *ctx,
+		struct lttng_kernel_channel_buffer *chan)
 {
 	uint8_t need_reschedule = test_tsk_need_resched(current);
 
-	lib_ring_buffer_align_ctx(ctx, lttng_alignof(need_reschedule));
-	chan->ops->event_write(ctx, &need_reschedule, sizeof(need_reschedule));
+	chan->ops->event_write(ctx, &need_reschedule, sizeof(need_reschedule), lttng_alignof(need_reschedule));
 }
 
 static
 void need_reschedule_get_value(void *priv,
-		struct lttng_probe_ctx *lttng_probe_ctx,
+		struct lttng_kernel_probe_ctx *lttng_probe_ctx,
 		struct lttng_ctx_value *value)
 {
 	value->u.s64 = test_tsk_need_resched(current);;