X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=lttng-context-prio.c;h=e23115ef60a34f614766013be131253fc86e8db0;hb=1d830a69689518a0561213f8d8c042d444e209fc;hp=9cd843d5437e45af1a2afe9fd924cff247dcc254;hpb=ceabb767180e064629b5a9ab4ed14449da864763;p=lttng-modules.git

diff --git a/lttng-context-prio.c b/lttng-context-prio.c
index 9cd843d5..e23115ef 100644
--- a/lttng-context-prio.c
+++ b/lttng-context-prio.c
@@ -11,24 +11,9 @@
 #include <linux/slab.h>
 #include <linux/sched.h>
 #include <lttng-events.h>
-#include <wrapper/ringbuffer/frontend_types.h>
-#include <wrapper/vmalloc.h>
-#include <wrapper/kallsyms.h>
+#include <include/ringbuffer/frontend_types.h>
 #include <lttng-tracer.h>
 
-static
-int (*wrapper_task_prio_sym)(struct task_struct *t);
-
-int wrapper_task_prio_init(void)
-{
-	wrapper_task_prio_sym = (void *) kallsyms_lookup_funcptr("task_prio");
-	if (!wrapper_task_prio_sym) {
-		printk(KERN_WARNING "LTTng: task_prio symbol lookup failed.\n");
-		return -EINVAL;
-	}
-	return 0;
-}
-
 static
 size_t prio_get_size(size_t offset)
 {
@@ -46,7 +31,7 @@ void prio_record(struct lttng_ctx_field *field,
 {
 	int prio;
 
-	prio = wrapper_task_prio_sym(current);
+	prio = task_prio(current);
 	lib_ring_buffer_align_ctx(ctx, lttng_alignof(prio));
 	chan->ops->event_write(ctx, &prio, sizeof(prio));
 }
@@ -56,19 +41,12 @@ void prio_get_value(struct lttng_ctx_field *field,
 		struct lttng_probe_ctx *lttng_probe_ctx,
 		union lttng_ctx_value *value)
 {
-	value->s64 = wrapper_task_prio_sym(current);
+	value->s64 = task_prio(current);
 }
 
 int lttng_add_prio_to_ctx(struct lttng_ctx **ctx)
 {
 	struct lttng_ctx_field *field;
-	int ret;
-
-	if (!wrapper_task_prio_sym) {
-		ret = wrapper_task_prio_init();
-		if (ret)
-			return ret;
-	}
 
 	field = lttng_append_context(ctx);
 	if (!field)
@@ -89,7 +67,6 @@ int lttng_add_prio_to_ctx(struct lttng_ctx **ctx)
 	field->record = prio_record;
 	field->get_value = prio_get_value;
 	lttng_context_update(*ctx);
-	wrapper_vmalloc_sync_all();
 	return 0;
 }
 EXPORT_SYMBOL_GPL(lttng_add_prio_to_ctx);