1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef IOPRIO_H
3 #define IOPRIO_H
4 
5 #include <linux/sched.h>
6 #include <linux/sched/rt.h>
7 #include <linux/iocontext.h>
8 
9 #include <uapi/linux/ioprio.h>
10 
11 /*
12  * Default IO priority.
13  */
14 #define IOPRIO_DEFAULT	IOPRIO_PRIO_VALUE(IOPRIO_CLASS_NONE, 0)
15 
16 /*
17  * Check that a priority value has a valid class.
18  */
ioprio_valid(unsigned short ioprio)19 static inline bool ioprio_valid(unsigned short ioprio)
20 {
21 	unsigned short class = IOPRIO_PRIO_CLASS(ioprio);
22 
23 	return class > IOPRIO_CLASS_NONE && class <= IOPRIO_CLASS_IDLE;
24 }
25 
26 /*
27  * if process has set io priority explicitly, use that. if not, convert
28  * the cpu scheduler nice value to an io priority
29  */
task_nice_ioprio(struct task_struct * task)30 static inline int task_nice_ioprio(struct task_struct *task)
31 {
32 	return (task_nice(task) + 20) / 5;
33 }
34 
35 /*
36  * This is for the case where the task hasn't asked for a specific IO class.
37  * Check for idle and rt task process, and return appropriate IO class.
38  */
task_nice_ioclass(struct task_struct * task)39 static inline int task_nice_ioclass(struct task_struct *task)
40 {
41 	if (task->policy == SCHED_IDLE)
42 		return IOPRIO_CLASS_IDLE;
43 	else if (task_is_realtime(task))
44 		return IOPRIO_CLASS_RT;
45 	else
46 		return IOPRIO_CLASS_BE;
47 }
48 
49 #ifdef CONFIG_BLOCK
50 int __get_task_ioprio(struct task_struct *p);
51 #else
__get_task_ioprio(struct task_struct * p)52 static inline int __get_task_ioprio(struct task_struct *p)
53 {
54 	return IOPRIO_DEFAULT;
55 }
56 #endif /* CONFIG_BLOCK */
57 
get_current_ioprio(void)58 static inline int get_current_ioprio(void)
59 {
60 	return __get_task_ioprio(current);
61 }
62 
63 extern int set_task_ioprio(struct task_struct *task, int ioprio);
64 
65 #ifdef CONFIG_BLOCK
66 extern int ioprio_check_cap(int ioprio);
67 #else
ioprio_check_cap(int ioprio)68 static inline int ioprio_check_cap(int ioprio)
69 {
70 	return -ENOTBLK;
71 }
72 #endif /* CONFIG_BLOCK */
73 
74 #endif
75