Fix: prio context NULL pointer exception

[lttng-modules.git] / src / lttng-context-prio.c

diff --git a/src/lttng-context-prio.c b/src/lttng-context-prio.c
index 3eb3ff9056b3a3022d4cb759efd22089121b5cbf..aac10a6399ac8618a42e23c299edd8d41a0c7ee7 100644 (file)
--- a/src/lttng-context-prio.c
+++ b/src/lttng-context-prio.c
@@ -29,6 +29,7 @@ int wrapper_task_prio_init(void)
        }
        return 0;
 }
+EXPORT_SYMBOL_GPL(wrapper_task_prio_init);
 
 /*
  * Canary function to check for 'task_prio()' at compile time.
@@ -44,7 +45,7 @@ int __canary__task_prio(const struct task_struct *p)
 }
 
 static
-size_t prio_get_size(void *priv, struct lttng_probe_ctx *probe_ctx, size_t offset)
+size_t prio_get_size(void *priv, struct lttng_kernel_probe_ctx *probe_ctx, size_t offset)
 {
        size_t size = 0;
 
@@ -54,20 +55,19 @@ size_t prio_get_size(void *priv, struct lttng_probe_ctx *probe_ctx, size_t offse
 }
 
 static
-void prio_record(void *priv, struct lttng_probe_ctx *probe_ctx,
-               struct lib_ring_buffer_ctx *ctx,
-               struct lttng_channel *chan)
+void prio_record(void *priv, struct lttng_kernel_probe_ctx *probe_ctx,
+               struct lttng_kernel_ring_buffer_ctx *ctx,
+               struct lttng_kernel_channel_buffer *chan)
 {
        int prio;
 
        prio = wrapper_task_prio_sym(current);
-       lib_ring_buffer_align_ctx(ctx, lttng_alignof(prio));
-       chan->ops->event_write(ctx, &prio, sizeof(prio));
+       chan->ops->event_write(ctx, &prio, sizeof(prio), lttng_alignof(prio));
 }
 
 static
 void prio_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 = wrapper_task_prio_sym(current);
@@ -76,7 +76,7 @@ void prio_get_value(void *priv,
 static const struct lttng_kernel_ctx_field *ctx_field = lttng_kernel_static_ctx_field(
        lttng_kernel_static_event_field("prio",
                lttng_kernel_static_type_integer_from_type(int, __BYTE_ORDER, 10),
-               false, false, false),
+               false, false),
        prio_get_size,
        prio_record,
        prio_get_value,