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 | */ |
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 | */ |
30 | static inline int task_nice_ioprio(struct task_struct *task) |
31 | { |
32 | return (task_nice(p: 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 | */ |
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(tsk: 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 |
52 | static inline int __get_task_ioprio(struct task_struct *p) |
53 | { |
54 | return IOPRIO_DEFAULT; |
55 | } |
56 | #endif /* CONFIG_BLOCK */ |
57 | |
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 |
68 | static inline int ioprio_check_cap(int ioprio) |
69 | { |
70 | return -ENOTBLK; |
71 | } |
72 | #endif /* CONFIG_BLOCK */ |
73 | |
74 | #endif |
75 |