X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=wrapper%2Fsched.h;h=a4cce74d3b3cbca8474785beb8336d673a031339;hb=5d0b3afd73a0dda761eafd5e01a881561baaea4b;hp=1332cfe63ef8cf32606b578854c4f987e60c33a5;hpb=8fefc8a2d93a06a032c7983cf621ab3fb0c2acde;p=lttng-modules.git

diff --git a/wrapper/sched.h b/wrapper/sched.h
index 1332cfe6..a4cce74d 100644
--- a/wrapper/sched.h
+++ b/wrapper/sched.h
@@ -1,32 +1,27 @@
-#ifndef _LTT_WRAPPER_VMALLOC_H
-#define _LTT_WRAPPER_VMALLOC_H
-
-/*
- * Copyright (C) 2011 Mathieu Desnoyers (mathieu.desnoyers@efficios.com)
+/* SPDX-License-Identifier: (GPL-2.0-only or LGPL-2.1-only)
  *
- * Wrapper around task_prio call.
+ * src/wrapper/kprobes.h
  *
- * Dual LGPL v2.1/GPL v2 license.
+ * Copyright (C) 2021 Michael Jeanson <mjeanson@efficios.com>
  */
 
-#ifdef CONFIG_KALLSYMS
+#ifndef _LTTNG_WRAPPER_SCHED_H
+#define _LTTNG_WRAPPER_SCHED_H
 
-#include "../ltt-events.h"
+#include <linux/sched.h>
+#include <lttng-kernel-version.h>
 
-static inline
-int wrapper_task_prio(struct task_struct *t)
-{
-	return wrapper_task_prio_sym(t);
-}
-#else
+#if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(5,14,0) || \
+	LTTNG_RHEL_KERNEL_RANGE(4,18,0,372,0,0, 4,19,0,0,0,0))
 
-#include <linux/sched.h>
+#define lttng_get_task_state(task)	READ_ONCE((task)->__state)
+#define lttng_task_is_running(task)	task_is_running(task)
+
+#else /* LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(5,14,0) */
+
+#define lttng_get_task_state(task)	((task)->state)
+#define lttng_task_is_running(task)	(lttng_get_task_state(task) == TASK_RUNNING)
 
-static inline
-int wrapper_task_prio(struct task_struct *t)
-{
-	return task_prio(t);
-}
-#endif
+#endif /* LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(5,14,0) */
 
-#endif /* _LTT_WRAPPER_VMALLOC_H */
+#endif /* _LTTNG_WRAPPER_SCHED_H */