at v6.2 1.6 kB view raw
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 */ 19static 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 */ 30static 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 */ 39static 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 50int __get_task_ioprio(struct task_struct *p); 51#else 52static inline int __get_task_ioprio(struct task_struct *p) 53{ 54 return IOPRIO_DEFAULT; 55} 56#endif /* CONFIG_BLOCK */ 57 58static inline int get_current_ioprio(void) 59{ 60 return __get_task_ioprio(current); 61} 62 63extern int set_task_ioprio(struct task_struct *task, int ioprio); 64 65#ifdef CONFIG_BLOCK 66extern int ioprio_check_cap(int ioprio); 67#else 68static inline int ioprio_check_cap(int ioprio) 69{ 70 return -ENOTBLK; 71} 72#endif /* CONFIG_BLOCK */ 73 74#endif